计算机网络【EPoll原理】

news2025/1/23 15:06:19

预备知识：内核poll钩子原理

内核函数poll_wait

把当前进程加入到驱动里自定义的等待队列上；

当驱动事件就绪后，就可以在驱动里自定义的等待队列上唤醒调用poll的进程；

故poll_wait作用：可以让驱动知道事件就绪的时候唤醒哪些等待进程；

钩子poll

内核f_op->poll必须配合驱动自己的等待队列才能用，不然驱动有事件产生后不知道哪些进程调用了poll来等待这个事件。

内核f_op->poll要做的事情。

调用poll_wait，将当前进程放入驱动设备的等待队列上，这样驱动就知道哪些进程在调用poll等待事件。
检查此时立刻已有的事件（POLLIN\POLLOUT\POLLERR…）并返回掩码表示。

f_op->poll是一个非阻塞的操作，立即返回，返回值以掩码形式表示当前已产生的事件集合。

预备知识：等待队列

等待队列对头：wait_queue_head_t ；

队列的成员：wait_queue_t；

wait_queue_t的成员：

void *private; /*指向进程描述符task_struct*/    
wait_queue_func_t  func;//唤醒时调用此函数，即钩子函数    
struct list_head  task_list;//队列链表指针

一般钩子函数func是内核默认函数default_wake_function，功能就是唤醒了进程。

我们也可以在把进程放入等待队列时主动设定钩子函数，使得在唤醒进程时自动执行我们需要的操作。

epoll就利用了队列钩子函数：把产生的事件内容copy到rdlist 。这样，事件来临时会自动把事件内容放到rdlist中，而不需要我们自己遍历监听句柄们查有谁产生了事件。

调用epoll_create1/epoll_create

创建了epoll句柄eventpoll，返回其文件表示的描述符epfd。

eventpoll内部有以下关键数据结构：

rbtree：红黑树，每个被加入到epoll监控的文件事件会创建一个epitem结构，作为rbtree节点。

使用rbtree的优点：可容纳大量文件事件，方便增删改（O(logN)）。

rdlist：内核链表，用于存放当前产生了期待事件产生的文件句柄们（这里的一个文件句柄可以理解为一个epoll_event）。
wq：当进程调用epoll_wait等待时，进程加入等待队列wq。
poll_wait：eventpoll本身的等待队列，由于eventpoll自己也被当做文件，这个队列用于自己被别人调用select/poll/epoll监听的情况（一般没啥用）。

poll_wait在啥时候用呢：

fd = socket(...);
efd1 = epoll_create();
efd2 = epoll_create();
epoll_ctl(efd1, EPOLL_CTL_ADD, fd, ...);
epoll_ctl(efd2, EPOLL_CTL_ADD, efd1, ...);

如上，efd1监控fd，而efd2监控了efd1，即嵌套的epoll监控：epoll监控另一个epoll句柄

efd2要监控efd1，将调用efd1的poll函数

回忆之前说过：文件f_op->poll需要配合驱动提供的等待队列

对于epollfd，等待队列就是poll_wait

efd2监听efd1，会调用efd1->f_op->poll，于是把当前进程放到efd1的poll_wait队列上

在epoll的内核实现中，当efd1本身监听到fd事件产生后，会顺便唤醒poll_wait上的进程

于是，“efd1监听到事件” 被通知到efd2。这样，就实现了epollfd被其他多路复用监听了！

故：poll_wait就是用于epoll句柄被另外的多路复用监听的，配合epoll自己的f_op->poll，看起来一般用不到

调用epoll_ctl操作句柄新增监控事件

epoll_ctl：EPOLL_CTL_ADD、EPOLL_CTL_MOD、EPOLL_CTL_DEL新增、修改、删除红黑树上的文件句柄。

其中epll_ctl：EPOLL_CTL_ADD新增句柄不仅仅新增红黑树节点，更关键的是对文件开始监控！

与select/poll的本质区别：并不是调用epoll_wait的时候才监听文件，而是EPOLL_CTL_ADD的时候就开始监听了。

epoll_ctl(epfd, EPOLL_CTL_ADD, fd, fdevent)核心流程：

对要注册的事件event->events追加关心事件：EPOLLERR | EPOLLHUP。

回忆epoll的使用中说过：EPOLLERR、EPOLLHUP事件会被自动监听，即使我们没设置。

创建epitem结构，加入到红黑树中。
【关键】revent = file->f_op->poll，即调用poll，把当前进程放到文件的等待队列上且设置回调函数ep_poll_callback，返回值revent是文件当前已产生事件掩码。
检查返回事件：如果revent与关心事件event->events有交集（说明ADD之前事件就准备好了）。
- 把此epitem节点拷贝到rdlist链表中；（就绪句柄拷贝到rdlist）。
- 如果有进程在wq等待队列上（即有进程在调用epoll_wait等待），则唤醒之！
- 顺便，如果有进程在poll_wait等待队列上（即有进程调用多路复用来监听当前epoll句柄），则唤醒之！

可以看到，如果在EPOLL_CTL_ADD一个文件之前，这个文件关心的事件就已经产生了的话，由于会唤醒wq队列上的进程，则此时EPOLL_CTL_ADD会使得epoll_wait函数从阻塞中返回。