2023/08/31 day5 网络编程

#include <myhead.h>
#define PORT 8888  //端口号，范围1024-49151
#define IP "192.168.114.57"  //本机IP，ifconfig
int main(int argc, const char *argv[])
{
	//创建流式套接字 socket
	int sfd; //定义服务器文件描述符
	if((sfd = socket(AF_INET,SOCK_STREAM,0))<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d\n",sfd);
	//允许端口快速的被复用
	int reuse =1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt error");
		return -1;
	}
	printf("允许端口快速的被复用成功\n");
	//填充地址信息结构体给bind函数绑定
	//真实的地址信息结构体根据地址族指定 AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET; //必须填AF_INET;
	sin.sin_port   = htons(PORT); //端口的网络字节序
	sin.sin_addr.s_addr   = inet_addr(IP); //本机IP
	//绑定服务器的地址信息———》必须绑定 bind
	if(bind(sfd,(struct sockaddr *)&sin, sizeof(sin))<0)
	{
		ERR_MSG("bind error");
		return -1;
	}
	printf("bind success\n");
	//将套接字设置为被动监听状态————listen 
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen error");
		return -1;
	}
	printf("listen success\n");	
	//创建一个读集合
	fd_set readfds,tempfds;  //临时集合
	FD_ZERO(&readfds); //清空集合
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);
	int res = -1;	
	int newfd = -1;
	struct sockaddr_in cin;   //存储客户端的地址信息
	socklen_t addrlen = sizeof(cin); //真实的地址信息结构体的大小
	//定义搬运工
	char buf[128] = "";
	int ret =0;
	int maxfd = sfd; //存储最大的文件描述符
	struct sockaddr_in savecin[1024]; //备份连接的客户端的地址信息，用下标来应对文件描述符
	while(1)
	{ 	
		//将集合赋值给临时集合
		tempfds = readfds; 
		//执行IO多路复用函数
		res= select(maxfd+1,&tempfds,NULL,NULL,NULL);				
		if(res<0)
		{
			ERR_MSG("select error");
			return -1;
		}
		else if(res == 0)
		{
			printf("time out。。\n");
			break;
		}
		//能够允许到当前位置，则代表select函数解除阻塞，即代表集合中有文件描述符准备就绪
		//此时集合中只会剩下产生事件的文件描述符，例如：
		//0号准备就绪，则集合中只会剩下0号
		//sfd准备就绪，则集合中只会剩下sfd
		//0和sfd均准备就绪，则集合会剩下0和sfd；
		//只需要判断集合中剩下那个文件描述符，走对应处理函数就可以了
		
			
			for(int i=0;i<=maxfd;i++)
			{
				if(FD_ISSET(i,&tempfds)==0)
					continue;
				//能运行到当前位置，则说明i所代表的文件描述符在集合中
				if(0 == i)
				{
					printf("触发键盘输入事件\n");
					
					bzero(buf,sizeof(buf));
					fgets(buf,sizeof(buf),stdin);
					buf[strlen(buf)-1] = '\0';
					printf("%s\n",buf);

				}
				else if(sfd ==i)
				{
					//获取一个已经完成的客户端信息，生产一个新的文件描述符
					printf("触发客户端连接事件\n");
					newfd = accept(sfd,(struct sockaddr *)&cin,&addrlen);
					if(newfd <0)
					{
						ERR_MSG("newfd error");
						return -1;
					}
					printf("[%s:%d]客户号端连接成功 newfd = %d\n",\
							inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
					savecin[newfd] = cin;
					FD_SET(newfd,&readfds);   //将newfd添加到集合中
					maxfd = maxfd>newfd?maxfd:newfd;    //更新maxfd

				}
				else
				{
					printf("触发客户端交互事件\n");
					//清空字符串
					bzero(buf,sizeof(buf));   //memset
					//接收
					ret = recv(i,buf,sizeof(buf),0);
					if(ret <0)
					{			
						break;
					}
					else if(ret == 0)
					{
						printf("[%s:%d]客户号端下线 newfd = %d\n",\
								inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i);
						//关闭文件描述符
						close(i);
						//将文件描述符从集合中删除
						FD_CLR(i,&readfds);
						//由于剔除的文件描述符可能是最大文件描述符，所以要更新maxfd

						while(FD_ISSET(maxfd,&readfds)==0 && maxfd-->=0);
					}
					printf("[%s:%d] %s\n",inet_ntoa(savecin[i].sin_addr),ntohs(savecin[i].sin_port),buf);
				/*	//发送	
					bzero(buf,sizeof(buf));
					fgets(buf,sizeof(buf),stdin);
					buf[strlen(buf)-1] ='\0';
					if(send(i,buf,sizeof(buf),0)<0)
					{
						ERR_MSG("send error");
						return -1;
					}
					printf("send success\n"); */
				}
		}

	}	

	//关闭文件描述符
	if(close(sfd)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	return 0;
	
}