// epoll2.cpp : 定義控制臺應用程序的入口點。 //#include <stdio.h> #include <sys/epoll.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <netdb.h> #include <sys/types.h> #include <signal.h> #include <unistd.h> #include <fcntl.h> #include <string.h> #include <errno.h> #include <stdlib.h> #include <pthread.h>//stl head#include <ext/hash_map> //包含hash_map 的頭文件//#include <map> //stl的mapusing namespace std; //std 命名空間using namespace __gnu_cxx; //而hash_map是在__gnu_cxx的命名空間里的int init_thread_pool(int threadNum); void *epoll_loop(void* para); void *check_connect_timeout(void* para);struct sockStruct {time_t time;unsigned int* recvBuf; };//hash-map//hash_map<int, unsigned int> sock_map;hash_map<int, sockStruct> sock_map; #define MAXRECVBUF 4096 #define MAXBUF MAXRECVBUF+10int fd_Setnonblocking(int fd) {int op;op=fcntl(fd,F_GETFL,0);fcntl(fd,F_SETFL,op|O_NONBLOCK);return op; }void on_sigint(int signal) {exit(0); }/* handle_message - 處理每個 socket 上的消息收發 */ int handle_message(int new_fd) {char buf[MAXBUF + 1];char sendbuf[MAXBUF+1];int len;/* 開始處理每個新連接上的數據收發 */bzero(buf, MAXBUF + 1);/* 接收客戶端的消息 *///len = recv(new_fd, buf, MAXBUF, 0);int nRecvBuf = MAXRECVBUF; //設置為32Ksetsockopt(new_fd, SOL_SOCKET, SO_RCVBUF, ( const char* )&nRecvBuf, sizeof(int));len=recv(new_fd,&buf, MAXBUF,0);//--------------------------------------------------------------------------------------------//這塊為了使用ab測試char bufSend[1000] = {0};sprintf(bufSend,"HTTP/1.0 200 OK\r\nContent-type: text/plain\r\n\r\n%s","Hello world!\n");send(new_fd,bufSend,strlen(buf),0);//--------------------------------------------------------------------------------------------if (len > 0){//printf ("%d接收消息成功:'%s'，共%d個字節的數據\n", new_fd, buf, len);//hash-maphash_map<int, sockStruct>::iterator it_find;it_find = sock_map.find(new_fd);if(it_find == sock_map.end()){//新的網絡連接，申請新的接收緩沖區，并放入map中//printf("new socket %d\n", new_fd);sockStruct newSockStruct;newSockStruct.time = time((time_t*)0);newSockStruct.recvBuf = (unsigned int*)malloc(1000);memset(newSockStruct.recvBuf, 0, 1000);strcat((char*)newSockStruct.recvBuf, buf);sock_map.insert(pair<int,sockStruct>(new_fd, newSockStruct));}else{//網絡連接已經存在，找到對應的數據緩沖區，將接收到的數據拼接到數據緩沖區中//printf("socket %d exist!\n", it_find->first);(it_find->second).time = time((time_t*)0); //時間更改char* bufSockMap = (char*)(it_find->second).recvBuf; //數據存儲strcat(bufSockMap, buf);//printf("bufSockMap:%s\n", bufSockMap);}}else {if (len < 0)printf ("消息接收失敗！錯誤代碼是%d，錯誤信息是'%s'\n",errno, strerror(errno));else {//將socket從map中移除/*hash_map<int, sockStruct>::iterator it_find;it_find = sock_map.find(new_fd);sock_map.erase(it_find);*/printf("client %d quit!\n",new_fd);}//close(new_fd);return -1;}/* 處理每個新連接上的數據收發結束 *///關閉socket的時候，要釋放接收緩沖區。hash_map<int, sockStruct>::iterator it_find;it_find = sock_map.find(new_fd);free((it_find->second).recvBuf);sock_map.erase(it_find);close(new_fd);return len; }int listenfd; int sock_op=1; struct sockaddr_in address; struct epoll_event event0; struct epoll_event events[1024]; int epfd; int n; int i; char buf[512]; int off; int result; char *p;int main(int argc,char* argv[]) {init_thread_pool(1);signal(SIGPIPE,SIG_IGN);signal(SIGCHLD,SIG_IGN);signal(SIGINT,&on_sigint);listenfd=socket(AF_INET,SOCK_STREAM,0);setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&sock_op,sizeof(sock_op));memset(&address,0,sizeof(address));address.sin_addr.s_addr=htonl(INADDR_ANY);address.sin_port=htons(8006);bind(listenfd,(struct sockaddr*)&address,sizeof(address));listen(listenfd,1024);fd_Setnonblocking(listenfd);epfd=epoll_create(65535);memset(&event0,0,sizeof(event0));event0.data.fd=listenfd;event0.events=EPOLLIN|EPOLLET;epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&event0);while(1){sleep(1000);}return 0; }/************************************************* * Function: * init_thread_pool * Description: * 初始化線程 * Input: * threadNum:用于處理epoll的線程數 * Output: * * Others: * 此函數為靜態static函數, *************************************************/ int init_thread_pool(int threadNum) {int i,ret;pthread_t threadId;//初始化epoll線程池for ( i = 0; i < threadNum; i++){ret = pthread_create(&threadId, 0, epoll_loop, (void *)0);if (ret != 0){printf("pthread create failed!\n");return(-1);}}ret = pthread_create(&threadId, 0, check_connect_timeout, (void *)0);return(0); } /************************************************* * Function: * epoll_loop * Description: * epoll檢測循環 * Input: * * Output: * * Others: * *************************************************/ static int count111 = 0; static time_t oldtime = 0, nowtime = 0; void *epoll_loop(void* para) {while(1){n=epoll_wait(epfd,events,4096,-1);//printf("n = %d\n", n);if(n>0){for(i=0;i<n;++i){if(events[i].data.fd==listenfd){while(1){event0.data.fd=accept(listenfd,NULL,NULL);if(event0.data.fd>0){fd_Setnonblocking(event0.data.fd);event0.events=EPOLLIN|EPOLLET;epoll_ctl(epfd,EPOLL_CTL_ADD,event0.data.fd,&event0);}else{if(errno==EAGAIN)break;}}}else{if(events[i].events&EPOLLIN){//handle_message(events[i].data.fd);char recvBuf[1024] = {0};int ret = 999;int rs = 1;while(rs){ret = recv(events[n].data.fd,recvBuf,1024,0);// 接受客戶端消息if(ret < 0){//由于是非阻塞的模式,所以當errno為EAGAIN時,表示當前緩沖區已無數據可//讀在這里就當作是該次事件已處理過。if(errno == EAGAIN){printf("EAGAIN\n");break;}else{printf("recv error!\n");epoll_ctl(epfd, EPOLL_CTL_DEL, events[i].data.fd, &event0);close(events[i].data.fd);break;}}else if(ret == 0){// 這里表示對端的socket已正常關閉.rs = 0;}if(ret == sizeof(recvBuf))rs = 1; // 需要再次讀取elsers = 0;}if(ret>0){count111 ++;struct tm *today;time_t ltime;time( &nowtime );if(nowtime != oldtime){printf("%d\n", count111);oldtime = nowtime;count111 = 0;}char buf[1000] = {0};sprintf(buf,"HTTP/1.0 200 OK\r\nContent-type: text/plain\r\n\r\n%s","Hello world!\n");send(events[i].data.fd,buf,strlen(buf),0);// CGelsServer Gelsserver;// Gelsserver.handle_message(events[i].data.fd);}epoll_ctl(epfd, EPOLL_CTL_DEL, events[i].data.fd, &event0);close(events[i].data.fd);}else if(events[i].events&EPOLLOUT){sprintf(buf,"HTTP/1.0 200 OK\r\nContent-type: text/plain\r\n\r\n%s","Hello world!\n");send(events[i].data.fd,buf,strlen(buf),0);/*if(p!=NULL){free(p);p=NULL;}*/close(events[i].data.fd);}else{close(events[i].data.fd);}}}}}} /************************************************* * Function: * check_connect_timeout * Description: * 檢測長時間沒反應的網絡連接，并關閉刪除 * Input: * * Output: * * Others: * *************************************************/ void *check_connect_timeout(void* para) {hash_map<int, sockStruct>::iterator it_find;for(it_find = sock_map.begin(); it_find!=sock_map.end(); ++it_find){if( time((time_t*)0) - (it_find->second).time > 120){ //時間更改free((it_find->second).recvBuf);sock_map.erase(it_find);close(it_find->first);}}}

代碼是在vs2017下建立linux工程編譯的，編譯的時候加入一個pthread多線程的庫

總結

以上是生活随笔為你收集整理的网络编程五种IO模型之epoll模型的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。