Day32

思维导图

select的TCP服务器模型

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr, "line:%d %s %s", __LINE__, __FILE__, __func__);\
	perror("msg");\
}while(0)

#define IP "192.168.2.54"
#define PORT 6666

int main(int argc, const char *argv[])
{
	//创建流式套接字
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	
	//绑定
	//填充地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}

	//监听
	if(listen(sfd, 128) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}


	//接收数据包从哪来
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	//新的文件描述符
	int newfd = -1;
	//存储客户端地址信息
	struct sockaddr_in saveCin[1024-3];
    //存储信息
	char buf[128] = "";
	//recv返回值
	ssize_t res = 0;

	//创建select监测集合
	fd_set readfds, tempfds;
	//清
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	//添加
	FD_SET(0, &readfds);
	FD_SET(sfd, &readfds);
	//select返回值
	int res_select = 0;
    //最大文件描述符	
	int maxfd = sfd;


	
	//接收accept
	while(1)
	{
		tempfds = readfds;
		//监测
		res_select = select(maxfd+1, &tempfds, NULL, NULL, NULL);
	
		if(res_select < 0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0 == res_select)
		{
			printf("time out...\n");
			break;
		}



		for(int i=0; i<=maxfd; i++)
		{
			if(!FD_ISSET(i, &tempfds))
				continue;
			if(0 == i)
			{
				//终端获取，send发送
				int sendfd;
				res = scanf("%d %s", &sendfd, buf);
				while(getchar() != '\n');

				if(res != 2)
				{
					fprintf(stderr, "请输入正确格式:  fd  string\n" );
					continue;
				}
			
				if(sendfd<3 || sendfd>1024 || FD_ISSET(sendfd, &readfds) == 0)
				{
					fprintf(stderr, "sndfd=%d文件描述符错误\n", sendfd);
					continue;
				}
				if(send(sendfd, buf, sizeof(buf), 0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");


			
			}
			else if(sfd == i)
			{
				//accept
				newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
				if(newfd < 0)
				{
					ERR_MSG("accept");
					return -1;
				}
				printf("[%s:%d] newfd:%d 客户端连接成功\n",\
						inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

				//存储客户端地址信息
				saveCin[newfd-3] = cin;
				//加入集合
				FD_SET(newfd, &readfds);
				//更新最大文件描述符
				maxfd = maxfd>newfd?maxfd:newfd;
			}
			else
			{
				//接收
				bzero(buf, sizeof(buf));
				res = recv(i, buf, sizeof(buf), 0);
				if(res < 0)
				{
					ERR_MSG("recv");
					return -1;
				}
				else if(0 == res)
				{
					//下线
					printf("[%s:%d] newfd:%d 客户端下线\n",\
							inet_ntoa(saveCin[i-3].sin_addr), ntohs(saveCin[i-3].sin_port), i);
					close(i);
					FD_CLR(i, &readfds);
					int j = maxfd;
					while(j-->=0 && !FD_ISSET(j, &readfds));
					maxfd = j;

					continue;
				}

				printf("[%s:%d] newfd:%d : %s\n",\
						inet_ntoa(saveCin[i-3].sin_addr), ntohs(saveCin[i-3].sin_port), newfd, buf);

			
			
			}
		}


	
	
	}
	close(sfd);
	
	return 0;
}

select的TCP客户端模型

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr, "line:%d %s %s", __LINE__, __FILE__, __func__);\
	perror(msg);\
}while(0)

#define IP "192.168.2.54"   //ifconfig出来的本机IP
#define PORT 6666            //1024~49151，网络字节序

int main(int argc, const char *argv[])
{
	//1> 创建流式套接字
	int cfd = socket(AF_INET, SOCK_STREAM, 0);
	if(cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	//绑定客户端自身的地址消息结构体--->非必须绑定
	//若若不绑定，则操作系统会自动帮cli绑定IP和端口
	//IP:客户端运行环境的IP地址
	//port:默认从49151~65535范围动态选择


	//填充服务器的地址信息结构体，给connect函数使用
	//客户端想连哪个服务器，就填哪个服务器绑定的IP和端口
	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;        //必须填AF_INET
	sin.sin_port        = htons(PORT);    //1024~491515，网络字节序
	sin.sin_addr.s_addr = inet_addr(IP);  //ifconfig出来的本机IP

	//2> 连接服务器
	if(connect(cfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success __%d__\n", __LINE__);


	//创建要监测的集合
	fd_set readfds, tempfds;
	//清空集合
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	//将需要的文件描述符设置到集合中
	FD_SET(0, &readfds);
	FD_SET(cfd, &readfds);

	int maxfd = cfd;

	int s_res = 0;
	char buf[128] = "";
	ssize_t res = 0;



	while(1)
	{
		tempfds = readfds;
		s_res = select(maxfd+1, &tempfds,NULL, NULL, NULL);
		if(s_res < 0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0 == s_res)
		{
			printf( "time out...\n");
			break;
		}
		if(FD_ISSET(0, &tempfds))
		{

			res = scanf( "%s" , buf);
			while(getchar() != 10);

			if(send(cfd, buf, sizeof(buf), 0) < 0)
			{
				ERR_MSG("send");
				return -1;
			}
		//	printf("send success_%d___\n", __LINE__);

		}
		if(FD_ISSET(cfd, &tempfds))
		{
			res = recv(cfd, buf, sizeof(buf), 0);
			if(res < 0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(res == 0)
			{
				printf("服务器下线\n");
				break;
			}
			printf("[服务器]：%s\n", buf);
		}
	}


	//关闭套接字文件描述符
	close(cfd);
	return 0;
}