作业要求：①使用IO多路复用中的select函数实现TCP并发服务器客户端
②使用IO多路复用中的poll函数实现TCP并发服务器的服务器端
一、
代码
#include <myhead.h>#define SERPORT 8888 //服务器端口号
#define SERIP "192.168.114.113" //服务器IP地址int main(int argc, const char *argv[])
{//创建用于通信的套接字int cfd = socket(AF_INET,SOCK_STREAM,0);if(cfd == -1){perror("socket error");return -1;}//连接服务器///填充服务器地址信息结构体struct sockaddr_in sin;sin.sin_family = AF_INET;sin.sin_port = htons(SERPORT);sin.sin_addr.s_addr = inet_addr(SERIP);///连接服务器if(connect(cfd,(struct sockaddr *)&sin,sizeof(sin)) == -1){perror("connect error");return -1;}//创建用于检测文件描述符的集合fd_set readfds,tempfds;//清空集合FD_ZERO(&readfds);//将要检测的文件描述符放入集合中FD_SET(cfd,&readfds);FD_SET(0,&readfds);int res = 0; //接收select的返回值int maxfd = cfd; //集合中值最大的文件描述符//向服务器进行数据的收发char buf[128] = "";int ret = 0; //接收recv的返回值while(1){tempfds = readfds;res = select(maxfd+1,&tempfds,NULL,NULL,NULL);if(res == -1){perror("select error");return -1;}else if(res == 0){printf("time out\n");return -1; }//遍历集合中所有的文件描述符for(int i = 0;i <= maxfd;i++){//判断当前文件描述符是否在集合中if(!FD_ISSET(i,&readfds)){continue;}//判断0号文件描述符是否还在集合中if(0 == i){//从标准输入中读取数据fgets(buf,sizeof(buf),stdin);buf[strlen(buf)-1] == 0;//将数据发送到服务器if(send(cfd,buf,sizeof(buf),0) == -1){perror("send error");return -1;}}else if(cfd == i) //判断cfd是否还在集合中{//接收来自服务器的消息ret = recv(cfd,buf,sizeof(buf),0);if(ret == -1){perror("recv error");return -1;}else if(ret == 0){printf("服务器已关闭\n");return -1;}printf("服务器消息：%s\n",buf);}}}//关闭文件描述符close(cfd);return 0;
}
效果图
二、
代码
#include <myhead.h>#define IP "192.168.114.118"
#define PORT 8888int main(int argc, const char *argv[])
{//创建用于连接的套接字int sfd = socket(AF_INET,SOCK_STREAM,0);if(sfd == -1){perror("socket error");return -1;}//设置端口号快速重用int reuse = 1;if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1){perror("setsockopt error");return -1;}//绑定服务器IP和端口号///填充服务器地址信息结构体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)) == -1){perror("bind error");return -1;}printf("bind success\n");//将连接用套接字设置为被动监听状态if(listen(sfd,128) == -1){perror("listen error");return -1;}printf("listen success\n");//定义一个集合管理sfd和打开的通信用文件描述符struct pollfd fds[1024];int maxfd = 0;//手动放入sfdfds[0].fd = sfd;fds[0].events = POLLIN; //表明为读事件//将fds中其余元素初始化为-1for(int i = 4;i <= 1024;i++){fds[i].fd = -1;}//填充客户端地址信息结构体struct sockaddr_in cin;cin.sin_family = AF_INET;socklen_t socklen = sizeof(cin);char cbuf[128] = ""; //给客户端用的容器int nfd;int res = 0; //接收poll返回的结果while(1){res = poll(fds,maxfd+1,-1);if(res == -1){perror("select");return -1;}else if(res == 0){continue;;}else if(res > 0) //说明检测到了有文件描述符对应的缓冲区的数据发生了改变{if(fds[0].revents == POLLIN) //表明有新的客户连接进来了{int nfd = accept(sfd,(struct sockaddr*)&cin,&socklen); //阻塞在此处，直到有客户端连接上来if(nfd == -1) //增加这些错误的判断非常重要，可以帮助找到出现问题的地方{perror("accept");return -1;}//将新的文件描述符加入到集合中for(int i = 1;i < 1024;i++){if( fds[i].fd == -1){fds[i].fd = nfd;fds[i].events = POLLIN;break;}}//更新最大的文件描述符if(nfd > maxfd){maxfd = nfd;}}for(int i = 1;i <= maxfd;i++) //轮询客户端对应的文件描述符{if(fds[i].revents == POLLIN) //说明此文件描述符对应的客户端发送来了数据{int ret = read(fds[i].fd,cbuf,sizeof(cbuf));if(ret == -1){perror("read");exit(-1);}else if(ret == 0){printf("client closed\n");close(fds[i].fd); //关闭对应的文件描述符fds[i].fd = -1; //在fds中清空对应的文件描述符}else if(ret > 0){printf("read buf = %s\n",cbuf);write(fds[i].fd,cbuf,strlen(cbuf)+1);}}}}}//关闭所有套接字close(sfd);return 0;}
效果图
