手机版
你好,游客 登录 注册
背景:
阅读新闻

Linux Epoll介绍和程序实例

[日期:2011-04-27] 来源:Linux社区  作者:Linux [字体: ]

8. 例子程序

下面是一个简单 Echo Server 的例子程序,麻雀虽小,五脏俱全,还包含了一个简单的超时检查机制,简洁起见没有做错误处理。

  1. //    
  2. // a simple echo server using epoll in linux   
  3. //    
  4. // 2009-11-05   
  5. // by sparkling   
  6. //    
  7. #include <sys/socket.h>   
  8. #include <sys/epoll.h>   
  9. #include <netinet/in.h>   
  10. #include <arpa/inet.h>   
  11. #include <fcntl.h>   
  12. #include <unistd.h>   
  13. #include <stdio.h>   
  14. #include <errno.h>   
  15. #include <iostream>   
  16. using namespace std;   
  17. #define MAX_EVENTS 500   
  18. struct myevent_s   
  19. {   
  20.     int fd;   
  21.     void (*call_back)(int fd, int events, void *arg);   
  22.     int events;   
  23.     void *arg;   
  24.     int status; // 1: in epoll wait list, 0 not in   
  25.     char buff[128]; // recv data buffer   
  26.     int len;   
  27.     long last_active; // last active time   
  28. };   
  29. // set event   
  30. void EventSet(myevent_s *ev, int fd, void (*call_back)(intintvoid*), void *arg)   
  31. {   
  32.     ev->fd = fd;   
  33.     ev->call_back = call_back;   
  34.     ev->events = 0;   
  35.     ev->arg = arg;   
  36.     ev->status = 0;   
  37.     ev->last_active = time(NULL);   
  38. }   
  39. // add/mod an event to epoll   
  40. void EventAdd(int epollFd, int events, myevent_s *ev)   
  41. {   
  42.     struct epoll_event epv = {0, {0}};   
  43.     int op;   
  44.     epv.data.ptr = ev;   
  45.     epv.events = ev->events = events;   
  46.     if(ev->status == 1){   
  47.         op = EPOLL_CTL_MOD;   
  48.     }   
  49.     else{   
  50.         op = EPOLL_CTL_ADD;   
  51.         ev->status = 1;   
  52.     }   
  53.     if(epoll_ctl(epollFd, op, ev->fd, &epv) < 0)   
  54.         printf("Event Add failed[fd=%d]\n", ev->fd);   
  55.     else  
  56.         printf("Event Add OK[fd=%d]\n", ev->fd);   
  57. }   
  58. // delete an event from epoll   
  59. void EventDel(int epollFd, myevent_s *ev)   
  60. {   
  61.     struct epoll_event epv = {0, {0}};   
  62.     if(ev->status != 1) return;   
  63.     epv.data.ptr = ev;   
  64.     ev->status = 0;   
  65.     epoll_ctl(epollFd, EPOLL_CTL_DEL, ev->fd, &epv);   
  66. }   
  67. int g_epollFd;   
  68. myevent_s g_Events[MAX_EVENTS+1]; // g_Events[MAX_EVENTS] is used by listen fd   
  69. void RecvData(int fd, int events, void *arg);   
  70. void SendData(int fd, int events, void *arg);   
  71. // accept new connections from clients   
  72. void AcceptConn(int fd, int events, void *arg)   
  73. {   
  74.     struct sockaddr_in sin;   
  75.     socklen_t len = sizeof(struct sockaddr_in);   
  76.     int nfd, i;   
  77.     // accept   
  78.     if((nfd = accept(fd, (struct sockaddr*)&sin, &len)) == -1)   
  79.     {   
  80.         if(errno != EAGAIN && errno != EINTR)   
  81.         {   
  82.             printf("%s: bad accept", __func__);   
  83.         }   
  84.         return;   
  85.     }   
  86.     do  
  87.     {   
  88.         for(i = 0; i < MAX_EVENTS; i++)   
  89.         {   
  90.             if(g_Events[i].status == 0)   
  91.             {   
  92.                 break;   
  93.             }   
  94.         }   
  95.         if(i == MAX_EVENTS)   
  96.         {   
  97.             printf("%s:max connection limit[%d].", __func__, MAX_EVENTS);   
  98.             break;   
  99.         }   
  100.         // set nonblocking   
  101.         if(fcntl(nfd, F_SETFL, O_NONBLOCK) < 0) break;   
  102.         // add a read event for receive data   
  103.         EventSet(&g_Events[i], nfd, RecvData, &g_Events[i]);   
  104.         EventAdd(g_epollFd, EPOLLIN|EPOLLET, &g_Events[i]);   
  105.         printf("new conn[%s:%d][time:%d]\n", inet_ntoa(sin.sin_addr), ntohs(sin.sin_port), g_Events[i].last_active);   
  106.     }while(0);   
  107. }   
  108. // receive data   
  109. void RecvData(int fd, int events, void *arg)   
  110. {   
  111.     struct myevent_s *ev = (struct myevent_s*)arg;   
  112.     int len;   
  113.     // receive data   
  114.     len = recv(fd, ev->buff, sizeof(ev->buff)-1, 0);     
  115.     EventDel(g_epollFd, ev);   
  116.     if(len > 0)   
  117.     {   
  118.         ev->len = len;   
  119.         ev->buff[len] = '\0';   
  120.         printf("C[%d]:%s\n", fd, ev->buff);   
  121.         // change to send event   
  122.         EventSet(ev, fd, SendData, ev);   
  123.         EventAdd(g_epollFd, EPOLLOUT|EPOLLET, ev);   
  124.     }   
  125.     else if(len == 0)   
  126.     {   
  127.         close(ev->fd);   
  128.         printf("[fd=%d] closed gracefully.\n", fd);   
  129.     }   
  130.     else  
  131.     {   
  132.         close(ev->fd);   
  133.         printf("recv[fd=%d] error[%d]:%s\n", fd, errno, strerror(errno));   
  134.     }   
  135. }   
  136. // send data   
  137. void SendData(int fd, int events, void *arg)   
  138. {   
  139.     struct myevent_s *ev = (struct myevent_s*)arg;   
  140.     int len;   
  141.     // send data   
  142.     len = send(fd, ev->buff, ev->len, 0);   
  143.     ev->len = 0;   
  144.     EventDel(g_epollFd, ev);   
  145.     if(len > 0)   
  146.     {   
  147.         // change to receive event   
  148.         EventSet(ev, fd, RecvData, ev);   
  149.         EventAdd(g_epollFd, EPOLLIN|EPOLLET, ev);   
  150.     }   
  151.     else  
  152.     {   
  153.         close(ev->fd);   
  154.         printf("recv[fd=%d] error[%d]\n", fd, errno);   
  155.     }   
  156. }   
  157. void InitListenSocket(int epollFd, short port)   
  158. {   
  159.     int listenFd = socket(AF_INET, SOCK_STREAM, 0);   
  160.     fcntl(listenFd, F_SETFL, O_NONBLOCK); // set non-blocking   
  161.     printf("server listen fd=%d\n", listenFd);   
  162.     EventSet(&g_Events[MAX_EVENTS], listenFd, AcceptConn, &g_Events[MAX_EVENTS]);   
  163.     // add listen socket   
  164.     EventAdd(epollFd, EPOLLIN|EPOLLET, &g_Events[MAX_EVENTS]);   
  165.     // bind & listen   
  166.     sockaddr_in sin;   
  167.     bzero(&sin, sizeof(sin));   
  168.     sin.sin_family = AF_INET;   
  169.     sin.sin_addr.s_addr = INADDR_ANY;   
  170.     sin.sin_port = htons(port);   
  171.     bind(listenFd, (const sockaddr*)&sin, sizeof(sin));   
  172.     listen(listenFd, 5);   
  173. }   
  174. int main(int argc, char **argv)   
  175. {   
  176.     short port = 12345; // default port   
  177.     if(argc == 2){   
  178.         port = atoi(argv[1]);   
  179.     }   
  180.     // create epoll   
  181.     g_epollFd = epoll_create(MAX_EVENTS);   
  182.     if(g_epollFd <= 0) printf("create epoll failed.%d\n", g_epollFd);   
  183.     // create & bind listen socket, and add to epoll, set non-blocking   
  184.     InitListenSocket(g_epollFd, port);   
  185.     // event loop   
  186.     struct epoll_event events[MAX_EVENTS];   
  187.     printf("server running:port[%d]\n", port);   
  188.     int checkPos = 0;   
  189.     while(1){   
  190.         // a simple timeout check here, every time 100, better to use a mini-heap, and add timer event   
  191.         long now = time(NULL);   
  192.         for(int i = 0; i < 100; i++, checkPos++) // doesn't check listen fd   
  193.         {   
  194.             if(checkPos == MAX_EVENTS) checkPos = 0; // recycle   
  195.             if(g_Events[checkPos].status != 1) continue;   
  196.             long duration = now - g_Events[checkPos].last_active;   
  197.             if(duration >= 60) // 60s timeout   
  198.             {   
  199.                 close(g_Events[checkPos].fd);   
  200.                 printf("[fd=%d] timeout[%d--%d].\n", g_Events[checkPos].fd, g_Events[checkPos].last_active, now);   
  201.                 EventDel(g_epollFd, &g_Events[checkPos]);   
  202.             }   
  203.         }   
  204.         // wait for events to happen   
  205.         int fds = epoll_wait(g_epollFd, events, MAX_EVENTS, 1000);   
  206.         if(fds < 0){   
  207.             printf("epoll_wait error, exit\n");   
  208.             break;   
  209.         }   
  210.         for(int i = 0; i < fds; i++){   
  211.             myevent_s *ev = (struct myevent_s*)events[i].data.ptr;   
  212.             if((events[i].events&EPOLLIN)&&(ev->events&EPOLLIN)) // read event   
  213.             {   
  214.                 ev->call_back(ev->fd, events[i].events, ev->arg);   
  215.             }   
  216.             if((events[i].events&EPOLLOUT)&&(ev->events&EPOLLOUT)) // write event   
  217.             {   
  218.                 ev->call_back(ev->fd, events[i].events, ev->arg);   
  219.             }   
  220.         }   
  221.     }   
  222.     // free resource   
  223.     return 0;   
  224. }   
linux
相关资讯       Linux 
本文评论   查看全部评论 (0)
表情: 表情 姓名: 字数

       

评论声明
  • 尊重网上道德,遵守中华人民共和国的各项有关法律法规
  • 承担一切因您的行为而直接或间接导致的民事或刑事法律责任
  • 本站管理人员有权保留或删除其管辖留言中的任意内容
  • 本站有权在网站内转载或引用您的评论
  • 参与本评论即表明您已经阅读并接受上述条款