eventfd以及epoll原理分析

来源:互联网 发布:表格删除筛选数据 编辑:程序博客网 时间:2024/06/09 17:36
  1. #include <stdio.h>
  2. #include <unistd.h>
  3. #include <sys/time.h>
  4. #include <stdint.h>
  5. #include <pthread.h>
  6. #include <sys/eventfd.h>
  7. #include <sys/epoll.h>

  8. int efd = -1;

  9. void *read_thread(void *dummy)
  10. {
  11.     int ret = 0;
  12.     uint64_t count = 0;
  13.     int ep_fd = -1;
  14.     struct epoll_event events[10];

  15.     if (efd < 0)
  16.     {
  17.         printf("efd not inited.\n");
  18.         goto fail;
  19.     }

  20.     ep_fd = epoll_create(1024);
  21.     if (ep_fd < 0)
  22.     {
  23.         perror("epoll_create fail: ");
  24.         goto fail;
  25.     }

  26.     {
  27.         struct epoll_event read_event;

  28.         read_event.events = EPOLLHUP | EPOLLERR | EPOLLIN;
  29.         read_event.data.fd = efd;

  30.         ret = epoll_ctl(ep_fd, EPOLL_CTL_ADD, efd, &read_event);
  31.         if (ret < 0)
  32.         {
  33.             perror("epoll ctl failed:");
  34.             goto fail;
  35.         }
  36.     }

  37.     while (1)
  38.     {
  39.         ret = epoll_wait(ep_fd, &events[0], 10, 5000);
  40.         if (ret > 0)
  41.         {
  42.             int i = 0;
  43.             for (; i < ret; i++)
  44.             {
  45.                 if (events[i].events & EPOLLHUP)
  46.                 {
  47.                     printf("epoll eventfd has epoll hup.\n");
  48.                     goto fail;
  49.                 }
  50.                 else if (events[i].events & EPOLLERR)
  51.                 {
  52.                     printf("epoll eventfd has epoll error.\n");
  53.                     goto fail;
  54.                 }
  55.                 else if (events[i].events & EPOLLIN)
  56.                 {
  57.                     int event_fd = events[i].data.fd;
  58.                     ret = read(event_fd, &count, sizeof(count));
  59.                     if (ret < 0)
  60.                     {
  61.                         perror("read fail:");
  62.                         goto fail;
  63.                     }
  64.                     else
  65.                     {
  66.                         struct timeval tv;

  67.                         gettimeofday(&tv, NULL);
  68.                         printf("success read from efd, read %d bytes(%llu) at %lds %ldus\n",
  69.                                ret, count, tv.tv_sec, tv.tv_usec);
  70.                     }
  71.                 }
  72.             }
  73.         }
  74.         else if (ret == 0)
  75.         {
  76.             /* time out */
  77.             printf("epoll wait timed out.\n");
  78.             break;
  79.         }
  80.         else
  81.         {
  82.             perror("epoll wait error:");
  83.             goto fail;
  84.         }
  85.     }

  86. fail:
  87.     if (ep_fd >= 0)
  88.     {
  89.         close(ep_fd);
  90.         ep_fd = -1;
  91.     }

  92.     return NULL;
  93. }

  94. int main(int argc, char *argv[])
  95. {
  96.     pthread_t pid = 0;
  97.     uint64_t count = 0;
  98.     int ret = 0;
  99.     int i = 0;

  100.     efd = eventfd(0, 0);
  101.     if (efd < 0)
  102.     {
  103.         perror("eventfd failed.");
  104.         goto fail;
  105.     }

  106.     ret = pthread_create(&pid, NULL, read_thread, NULL);
  107.     if (ret < 0)
  108.     {
  109.         perror("pthread create:");
  110.         goto fail;
  111.     }

  112.     for (= 0; i < 5; i++)
  113.     {
  114.         count = 4;
  115.         ret = write(efd, &count, sizeof(count));
  116.         if (ret < 0)
  117.         {
  118.             perror("write event fd fail:");
  119.             goto fail;
  120.         }
  121.         else
  122.         {
  123.             struct timeval tv;

  124.             gettimeofday(&tv, NULL);
  125.             printf("success write to efd, write %d bytes(%llu) at %lds %ldus\n",
  126.                    ret, count, tv.tv_sec, tv.tv_usec);
  127.         }

  128.         sleep(1);
  129.     }

  130. fail:
  131.     if (!= pid)
  132.     {
  133.         pthread_join(pid, NULL);
  134.         pid = 0;
  135.     }

  136.     if (efd >= 0)
  137.     {
  138.         close(efd);
  139.         efd = -1;
  140.     }
  141.     return ret;
  142. }
0 0
原创粉丝点击