【linux网络编程】epoll函数

介绍：

通常来说，实现处理tcp请求，为一个连接一个线程，在高并发的场景，这种多线程模型与Epoll相比就显得相形见绌了。epoll是linux2.6内核的一个新的系统调用，epoll在设计之初，就是为了替代select, poll线性复杂度的模型，epoll的时间复杂度为O(1), 也就意味着，epoll在高并发场景，随着文件描述符的增长，有良好的可扩展性。

• select 和 poll 监听文件描述符list，进行一个线性的查找 O(n)
• epoll: 使用了内核文件级别的回调机制O(1)

头文件：

 #include <sys/epoll.h>

关键函数：

• epoll_create1: 创建一个epoll实例，文件描述符
• epoll_ctl: 将监听的文件描述符添加到epoll实例中，实例代码为将标准输入文件描述符添加到epoll中
• epoll_wait: 等待epoll事件从epoll实例中发生， 并返回事件以及对应文件描述符l

函数原型：

int epoll_create(int size);

size是创建节点数，但是从Linux2.6.8开始这个值大于0即可，epoll会自动展；

返回值是epoll树的根节点。

 

int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);

epfd是epoll_create创建的树根节点；

op是对应的操作（EPOLL_CTL_ADD是添加文件描述符到树节点上，EPOLL_CTL_MOD将fd对应的事件设置为第三个参数对应的事件，EPOLL_CTL_DEL是将文件描述符fd从树节点上摘除）；

*event 对应要添加或者修改的事件，其对应的结构体如下。

struct epoll_event
{
  uint32_t events;  /* Epoll events */
  epoll_data_t data;    /* User data variable */
};
typedef union epoll_data
{
  void *ptr;
  int fd;
  uint32_t u32;
  uint64_t u64;
} epoll_data_t;

int epoll_wait(int epfd, struct epoll_event *events,
                      int maxevents, int timeout);

epfd也是树根节点；events是我们需要等待检测的epoll事件；maxevents表示最多检测多少个节点发生的事件；timeout为-1时该函数会永久阻塞，为0时该函数会立即返回。

 

epoll的两种触发方式：

epoll事件有两种模型，边沿触发：edge-triggered (ET)， 水平触发：level-triggered (LT)；

1. 水平触发(level-triggered)

• socket接收缓冲区不为空 有数据可读 读事件一直触发
• socket发送缓冲区不满 可以继续写入数据 写事件一直触发

2. 边沿触发(edge-triggered)

• socket的接收缓冲区状态变化时触发读事件，即空的接收缓冲区刚接收到数据时触发读事件
• socket的发送缓冲区状态变化时触发写事件，即满的缓冲区刚空出空间时触发读事件

边沿触发仅触发一次，水平触发会一直触发。

事件宏

• EPOLLIN ： 表示对应的文件描述符可以读（包括对端SOCKET正常关闭）；
• EPOLLOUT： 表示对应的文件描述符可以写；
• EPOLLPRI： 表示对应的文件描述符有紧急的数据可读（这里应该表示有带外数据到来）；
• EPOLLERR： 表示对应的文件描述符发生错误；
• EPOLLHUP： 表示对应的文件描述符被挂断；
• EPOLLET： 将 EPOLL设为边缘触发(Edge Triggered)模式（默认为水平触发），这是相对于水平触发(Level Triggered)来说的。
• EPOLLONESHOT： 只监听一次事件，当监听完这次事件之后，如果还需要继续监听这个socket的话，需要再次把这个socket加入到EPOLL队列里

libevent 采用水平触发， nginx 采用边沿触发

 

实例代码：

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <string.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <ctype.h>
#include <sys/epoll.h>


int main(int argc, const char* argv[])
{
    if(argc < 2)
    {
        printf("eg: ./a.out port\n");
        exit(1);
    }
    struct sockaddr_in serv_addr;
    socklen_t serv_len = sizeof(serv_addr);
    int port = atoi(argv[1]);

    // 创建套接字
    int lfd = socket(AF_INET, SOCK_STREAM, 0);
    // 初始化服务器 sockaddr_in 
    memset(&serv_addr, 0, serv_len);
    serv_addr.sin_family = AF_INET;                   // 地址族 
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);    // 监听本机所有的IP
    serv_addr.sin_port = htons(port);            // 设置端口 
    // 绑定IP和端口
    bind(lfd, (struct sockaddr*)&serv_addr, serv_len);

    // 设置同时监听的最大个数
    listen(lfd, 36);
    printf("Start accept ......\n");

    struct sockaddr_in client_addr;//用于保存客户端信息
    socklen_t cli_len = sizeof(client_addr);

    // 创建epoll树根节点
    int epfd = epoll_create(2000);
    // 初始化epoll树
    struct epoll_event ev;
    ev.events = EPOLLIN;
    ev.data.fd = lfd;
    epoll_ctl(epfd, EPOLL_CTL_ADD, lfd, &ev);

    struct epoll_event all[2000];//是一个传出参数,所有读写缓冲区有变化的节点都会存储在此
    while(1)
    {
        // 使用epoll通知内核fd 文件IO检测
        int ret = epoll_wait(epfd, all, sizeof(all)/sizeof(all[0]), -1);

        // 遍历all数组中的前ret个元素
        for(int i=0; i<ret; ++i)
        {
            int fd = all[i].data.fd;
            // 判断是否有新连接
            if(fd == lfd)
            {
                // 接受连接请求
                int cfd = accept(lfd, (struct sockaddr*)&client_addr, &cli_len);
                if(cfd == -1)
                {
                    perror("accept error");
                    exit(1);
                }
                // 将新得到的cfd挂到树上
                struct epoll_event temp;
                temp.events = EPOLLIN;
                temp.data.fd = cfd;
                epoll_ctl(epfd, EPOLL_CTL_ADD, cfd, &temp);
                
                // 打印客户端信息
                char ip[64] = {0};
                printf("New Client IP: %s, Port: %d\n",
                    inet_ntop(AF_INET, &client_addr.sin_addr.s_addr, ip, sizeof(ip)),
                    ntohs(client_addr.sin_port));
                
            }
            else
            {
                // 处理已经连接的客户端发送过来的数据
                if(!all[i].events & EPOLLIN)//注意有些客户端不是EPOLLIN标志位 
                {
                    continue;
                }

                // 读数据
                char buf[1024] = {0};
                int len = recv(fd, buf, sizeof(buf), 0);
                if(len == -1)
                {
                    perror("recv error");
                    exit(1);
                }
                else if(len == 0)
                {
                    printf("client disconnected ....\n");
                    // fd从epoll树上删除
                    ret = epoll_ctl(epfd, EPOLL_CTL_DEL, fd, NULL);
                    if(ret == -1)
                    {
                        perror("epoll_ctl - del error");
                        exit(1);
                    }
                    close(fd);
                    
                }
                else
                {
                    printf(" recv buf: %s\n", buf);
                    write(fd, buf, len);
                }
            }
        }
    }

    close(lfd);
    return 0;
}