epoll 入门

来源:互联网 发布:3维旋转矩阵公式 编辑:程序博客网 时间:2024/06/13 19:43

epoll用到的所有函数都是在头文件sys/epoll.h中声明的,下面简要说明所用到的数据结构和函数:
所用到的数据结构

  1. typedef union epoll_data {  
  2.                 void *ptr;  
  3.                 int fd;  
  4.                 __uint32_t u32;  
  5.                 __uint64_t u64;  
  6.         } epoll_data_t;  
  7.   
  8.         struct epoll_event {  
  9.                 __uint32_t events;      /* Epoll events */  
  10.                 epoll_data_t data;      /* User data variable */  
  11.         };  

结构体epoll_event 被用于注册所感兴趣的事件和回传所发生待处理的事件,其中epoll_data 联合体用来保存触发事件的某个文件描述符相关的数据,例如一个client连接到服务器,服务器通过调用accept函数可以得到于这个client对应的socket文件描述符,可以把这文件描述符赋给epoll_data的fd字段以便后面的读写操作在这个文件描述符上进行。epoll_event 结构体的events字段是表示感兴趣的事件和被触发的事件可能的取值为:

  1. EPOLLIN :表示对应的文件描述符可以读;  
  2. EPOLLOUT:表示对应的文件描述符可以写;  
  3. EPOLLPRI:表示对应的文件描述符有紧急的数据可读(这里应该表示有带外数据到来);  
  4. EPOLLERR:表示对应的文件描述符发生错误;  
  5. EPOLLHUP:表示对应的文件描述符被挂断;  
  6. EPOLLET:表示对应的文件描述符有事件发生;  

所用到的函数:
1、epoll_create函数
     函数声明:int epoll_create(int size) 
    该函数生成一个epoll专用的文件描述符,其中的参数是指定生成描述符的最大范围(我觉得这个参数和select函数的第一个参数应该是类似的但是该怎么设置才好,我也不太清楚)。
2、epoll_ctl函数
     函数声明:int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event)
     该函数用于控制某个文件描述符上的事件,可以注册事件,修改事件,删除事件。
    参数:epfd:由 epoll_create 生成的epoll专用的文件描述符;
                op:要进行的操作例如注册事件,可能的取值EPOLL_CTL_ADD 注册、EPOLL_CTL_MOD 修
                        改、EPOLL_CTL_DEL 删除
                fd:关联的文件描述符;
                event:指向epoll_event的指针;
    如果调用成功返回0,不成功返回-1
3、epoll_wait函数
函数声明:int epoll_wait(int epfd,struct epoll_event * events,int maxevents,int timeout)
该函数用于轮询I/O事件的发生;
参数:
epfd:由epoll_create 生成的epoll专用的文件描述符;
epoll_event:用于回传代处理事件的数组;
maxevents:每次能处理的事件数;
timeout:等待I/O事件发生的超时值;
返回发生事件数。

 

以代码来说明问题:
首先给出server的代码,需要说明的是每次accept的连接,加入可读集的时候采用的都是LT模式,而且接收缓冲区是5字节的,也就是每次只接收5字节的数据:

[cpp] view plaincopyprint?
  1. #include<iostream>   
  2. #include<sys/socket.h>   
  3. #include<sys/epoll.h>   
  4. #include<netinet/in.h>   
  5. #include<arpa/inet.h>   
  6. #include<fcntl.h>   
  7. #include<unistd.h>   
  8. #include<stdio.h>   
  9. #include<errno.h>   
  10. using namespace std;  
  11.   
  12. #define MAXLINE  5   
  13. #define OPEN_MAX  100   
  14. #define LISTENQ  20   
  15. #define SERV_PORT  5000   
  16. #define INFTIM  1000   
  17.   
  18. void setnonblocking(int sock)  
  19. {  
  20.     int opts;  
  21.     opts = fcntl(sock,F_GETFL);  
  22.     if(opts < 0)  
  23.     {  
  24.         perror(" fcntl(sock,GETFL)");  
  25.         exit (1);  
  26.     }  
  27.     opts = opts|O_NONBLOCK;  
  28.     if(fcntl(sock,F_SETFL,opts) < 0)  
  29.     {  
  30.         perror("fcntl(sock,SETFL,opts)");  
  31.         exit (1);  
  32.     }     
  33. }  
  34.   
  35. int main()  
  36. {  
  37.     int i, maxi, listenfd, connfd, sockfd, epfd, nfds;  
  38.     ssize_t n;  
  39.     char line[MAXLINE];  
  40.     socklen_t clilen;  
  41.   
  42.     // 声明epoll_event结构体的变量,ev用于注册事件,数组用于回传要处理的事件  
  43.     struct epoll_event ev,events[20];  
  44.   
  45.     // 生成用于处理accept的epoll专用的文件描述符  
  46.     epfd = epoll_create(256);  
  47.   
  48.     struct sockaddr_in clientaddr;  
  49.     struct sockaddr_in serveraddr;  
  50.     listenfd = socket(AF_INET, SOCK_STREAM,  0 );  
  51.     // 把socket设置为非阻塞方式  
  52.      setnonblocking(listenfd);  
  53.   
  54.     // 设置与要处理的事件相关的文件描述符  
  55.     ev.data.fd = listenfd;  
  56.   
  57.     // 设置要处理的事件类型  
  58.     ev.events = EPOLLIN;  
  59.     // 注册epoll事件  
  60.     epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev);  
  61.   
  62.     bzero(&serveraddr, sizeof(serveraddr));  
  63.     serveraddr.sin_family  =  AF_INET;  
  64.     char *local_addr = "127.0.0.1";  
  65.     inet_aton(local_addr, &(serveraddr.sin_addr)); // htons(SERV_PORT);  
  66.     serveraddr.sin_port = htons(SERV_PORT);  
  67.     bind(listenfd,(sockaddr *)&serveraddr, sizeof(serveraddr));  
  68.     listen(listenfd, LISTENQ);  
  69.     maxi = 0;  
  70.     for(;;)   
  71.     {  
  72.         // 等待epoll事件的发生  
  73.         nfds = epoll_wait(epfd, events, 20 , 500);  
  74.         // 处理所发生的所有事件       
  75.         for(i = 0 ;i < nfds; ++ i)  
  76.         {  
  77.             if(events[i].data.fd == listenfd)  
  78.             {                 
  79.                 clilen=sizeof(struct sockaddr);   
  80.                 connfd = accept(listenfd, (struct sockaddr *)&clientaddr, &clilen);  
  81.                 if (connfd < 0 )  
  82.                 {  
  83.                     perror("connfd<0");  
  84.                     exit ( 1 );  
  85.                 }  
  86.                 // setnonblocking(connfd);  
  87.                 char *str = inet_ntoa(clientaddr.sin_addr);  
  88.                 cout  <<   " accapt a connection from  "   <<  str  <<  endl;  
  89.   
  90.                 // 设置用于读操作的文件描述符  
  91.                 ev.data.fd = connfd;  
  92.                 // 设置用于注测的读操作事件  
  93.                 ev.events = EPOLLIN|EPOLLET;  
  94.                 // ev.events = EPOLLIN;  
  95.                 // 注册ev  
  96.                 epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev);  
  97.             }  
  98.             else if(events[i].events & EPOLLIN)  
  99.             {  
  100.                 cout  <<   " EPOLLIN "   <<  endl;  
  101.                 if((sockfd = events[i].data.fd) < 0)   
  102.                     continue;  
  103.                 if((n = read(sockfd, line, MAXLINE)) < 0)   
  104.                 {  
  105.                     if(errno == ECONNRESET)   
  106.                     {  
  107.                         close(sockfd);  
  108.                         events[i].data.fd  =   - 1 ;  
  109.                     }    
  110.                     else   
  111.                         std::cout << " readline error " << std::endl;  
  112.                 }    
  113.                 else if  (n  ==   0 )   
  114.                 {  
  115.                     close(sockfd);  
  116.                     events[i].data.fd = - 1 ;  
  117.                 }  
  118.                 line[n] = ' \0';   
  119.                 cout<<" read  "<<line<<endl;  
  120.   
  121.                 // 设置用于写操作的文件描述符  
  122.                 ev.data.fd = sockfd;  
  123.                 // 设置用于注测的写操作事件  
  124.                 ev.events = EPOLLOUT|EPOLLET;  
  125.                 // 修改sockfd上要处理的事件为EPOLLOUT  
  126.                 // epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd, & ev);  
  127.             }  
  128.             else if (events[i].events & EPOLLOUT)  
  129.             {     
  130.                 sockfd = events[i].data.fd;  
  131.                 write(sockfd, line, n);  
  132.   
  133.                 // 设置用于读操作的文件描述符  
  134.                 ev.data.fd = sockfd;  
  135.                 // 设置用于注测的读操作事件  
  136.                 ev.events = EPOLLIN|EPOLLET;  
  137.                 // 修改sockfd上要处理的事件为EPOLIN  
  138.                 epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd, & ev);  
  139.             }  
  140.         }  
  141.     }  
  142.     return  0 ;  
  143. }