作業要求：①使用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;}

效果圖