7.10 作业

1.IO复用TCP服务器

#include <myhead.h>
#define IP "192.168.122.183"

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

	//允许端口快速被重用
	int reuse = 1;                                     
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口快速被重用成功\n");

	struct sockaddr_in sai;
	sai.sin_family = AF_INET;//协议族
	sai.sin_port = htons(6666);//服务器端口
	sai.sin_addr.s_addr = inet_addr(IP);//服务器IP
	if(bind(sfd,(struct sockaddr *)&sai,sizeof(struct sockaddr_in)) < 0)//绑定ip和端口
	{
		ERR_MSG("bind");
		return -1;
	}
	puts("bind success");

	if(listen(sfd,128) < 0)//设置为被动监听状态，最大128
	{
		ERR_MSG("listen");
		return -1;
	}
	puts("listen success");

	struct sockaddr_in addr;
	socklen_t addrlen=sizeof(addr);
	struct sockaddr_in addrs[1024];//定义地址信息结构体数组

	fd_set readfds,tempfds;//定义集合
	FD_ZERO(&readfds);//清空
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);//放入sfd
	tempfds = readfds;//tempfds保存内容
	int nfds = sfd;//最大文件描述符
	while(1)
	{
		int num = select(nfds+1,&readfds,NULL,NULL,NULL);//等待文件描述符有数据
		if(num < 0)
		{
			ERR_MSG("select");
			return -1;
		}

		for(int i=0;i<=nfds;i++)//循环遍历文件描述符
		{
			if(FD_ISSET(i,&readfds))//文件描述符有内容
			{
				if(i == 0)//有键盘输入
				{
					printf("输入事件\n");
					char buf[64] = "";
					fgets(buf,sizeof(buf),stdin);
					buf[strlen(buf) - 1] = 0;
					printf("buf:%s\n",buf);		
				}
				else if(i == sfd)//有客户端连接
				{
					puts("客户端连接事件");
					int newfd = accept(sfd,(struct sockaddr *)&addr,&addrlen);//等待客户端连接，返回客户端的文件描述符和ip端口等
					if(newfd <0)
					{
						ERR_MSG("accept");
						return -1;
					}
					printf("[%s:%d]newfd = %d 客户端连接成功\n",inet_ntoa(addr.sin_addr),ntohs(addr.sin_port),newfd);
					FD_SET(newfd,&tempfds);//把newfd放入集合，用于接受客户端数据
					nfds = newfd>nfds?newfd:nfds;//设置最大的文件描述符
					addrs[newfd] = addr;//保存地址结构体信息
				}
				else//有数据从客户端发送
				{
					addr = addrs[i];//客户端地址
					char buf[64] = "";
					ssize_t res = recv(i,buf,sizeof(buf),0);//读取信息
					if(res < 0)
					{
						ERR_MSG("recv");
						continue;
					}
					else if(res == 0)
					{
						printf("[%s:%d] 客户端:%d 下线\n",inet_ntoa(addr.sin_addr),ntohs(addr.sin_port),i);
						close(i);//关闭对应客户端的文件描述符
						FD_CLR(i,&tempfds);//从集合中删除
						if(i == nfds)
							for(;FD_ISSET(nfds,&tempfds)&&nfds>-1;nfds--);//找到现在最大的文件描述符
						continue;
					}
					printf("[%s:%d] 客户端:%d 的消息:%s\n",inet_ntoa(addr.sin_addr),ntohs(addr.sin_port),i,buf);
					
					strcat(buf,"*");
					res = send(i,buf,sizeof(buf),0);//发送信息
					if(res < 0)
					{
						ERR_MSG("send");
						continue;
					}
				}
			}
		}
		readfds = tempfds;//修改为保存的集合
	}

	if(close(sfd) < 0)
	{
		ERR_MSG("close");
		return -1;
	}	
	return 0;
}

2.IO复用TCP客户端

#include <myhead.h>
#define IP "192.168.122.183"
int main(int argc,const char *argv[])
{
	int cfd = socket(AF_INET,SOCK_STREAM,0);//获得客户端套接字文件描述符
	if(cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

	struct sockaddr_in sai;
	sai.sin_family = AF_INET;//服务器地址族
	sai.sin_port = htons(6666);//服务器端口
	sai.sin_addr.s_addr = inet_addr(IP);//服务器ip
	if(connect(cfd,(struct sockaddr*)&sai,sizeof(sai)) < 0)//和服务器进行连接
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("连接服务器:%d\n",cfd);

	//创建集合
	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);
	tempfds = readfds;

	for(;;)
	{
		int snum = select(cfd+1,&readfds,NULL,NULL,NULL);//等待文件描述符有数据
		if(snum < 0)
		{
			ERR_MSG("select");
			return -1;
		}

		if(FD_ISSET(0,&readfds))
		{
			char buf[64] = "";
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1] = 0;
			if(!strcmp(buf,"quit"))
				break;
			send(cfd,buf,sizeof(buf),0);
		}

		if(FD_ISSET(cfd,&readfds))
		{
			char buf[64] = "";
			ssize_t res = recv(cfd,buf,sizeof(buf),0);//从服务器读
			if(res < 0)
			{
				ERR_MSG("recv");
				break;
			}
			else if(res == 0)
			{
				printf("服务器:%d 下线\n",cfd);
				break;
			}
			printf("服务器:%d 的消息:%s\n",cfd,buf);	
		}

		readfds = tempfds;
	}
	close(cfd);
	return 0;
}