国庆放假作业2

1、select实现服务器并发

#include <myhead.h>
 
#define PORT 7373
#define IP "192.168.1.9"
 
int main(int argc,const char *argv[])
{
	//创建报式套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket error");
		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");
 
	//地址信息结构体
	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");
 
	//将sfd设置为监听文件描述符
	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 s_res=-1;   //定义一个整形变量接收select函数的返回值
	int maxfd=sfd;  //设置一个最大文件描述符
	int newfd=-1;
 
	char buf[128]="";
	struct sockaddr_in cin; //存储客户端的地址信息
	socklen_t addrlen = sizeof(cin); //结构体大小
	struct sockaddr_in savecin[1024];
	//备份连接成功的客户端的地址信息，用下标来对应文件描述符
 
	ssize_t res=-1;  //定义一个变量接收recv函数的返回值
 
	while(1)
	{
		//将readfds备份一份给tempfds
		tempfds=readfds;
 
		//阻塞函数
		if((s_res=select(maxfd+1,&tempfds,NULL,NULL,NULL))<0)
		{
			ERR_MSG("select error");
			return -1;
		}
		else if(0==s_res)
		{
			printf("time over\n");
			break;  //设置定时，如果阻塞时间超过，则跳出循环
		}
 
		for(int i=0;i<=maxfd;i++)
		{
			if(FD_ISSET(i,&tempfds)==0)
			continue;
			
			//能运行到次，说明i所代表的文件描述符在集合中
			//触发键盘输入事件
			if(0==i)
			{
				printf("请输入：");
 
				bzero(buf,sizeof(buf));  //刷新一下数组容器
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]=0;
 
				printf("触发键盘输入事件完成 :%s\n",buf);  //输出buf中的内容
			}
			//触发客户端连接事件
			else if(sfd==i)
			{
				printf("触发客户端连接事件\n");
				
				if((newfd=accept(i,(struct sockaddr*)&cin,&addrlen))<0)
				{
					ERR_MSG("accept error");
					return -1;
				}
				printf("[%s:%d]客户端连接成功 newfd=%d\n",\
						inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
				savecin[newfd]=cin;  //将cin另存到newfd对应的下标位置去
 
				//将newfd加入到readfds文件描述符集合中
				FD_SET(newfd,&readfds);
 
				//刷新最大文件描述符maxfd
				maxfd = maxfd>newfd?maxfd:newfd;
 
			}
			//触发客户端交互事件
			else
			{
				printf("触发客户端交互事件\n");
 
				//清空字符串,因为需要接收客户端发来的信息
				bzero(buf,sizeof(buf));
				
				//接收
				res=recv(i,buf,sizeof(buf),0);
				if(res<0)
				{
					ERR_MSG("recv error");
					return -1;
				}
				else if(0==res)
				{
 
					printf("[%s:%d] 客户端下线 newfd=%d\n",\
							inet_ntoa(savecin[i].sin_addr),\
							ntohs(savecin[i].sin_port),i);
					//关闭此时的i文件描述符
					close(i);
 
					//清除此时的i文件描述符
					FD_CLR(i,&readfds);
 
					//如果i是最大文件描述符,更新最大文件描述符maxfd
					while(FD_ISSET(maxfd,&readfds)==0&&maxfd>=0);
					//FD_ISSET(maxfd,&readfds)==0
					//表示如果maxfd是i（此时最大文件描述符已清除不存在）就循环，否则就跳出循环
					//maxfd>=0
					//表示如果遍历到0就跳出循环
				}
				printf("[%s:%d] newfd=%d:%s\n",\
						inet_ntoa(savecin[i].sin_addr),\
							ntohs(savecin[i].sin_port),\
							i,buf);
				//发送
				strcat(buf,">_<");
				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 error");
		return -1;
	}
	return 0;
}