文章会从网卡接收数据的流程讲起,串联起 CPU 中断、操作系统进程调度等知识;再一步步分析阻塞接收数据、select 到 epoll 的进化过程,最后探究 epoll 的实现细节。

一、从网卡接收数据说起

下边是一个典型的计算机结构图,计算机由 CPU、存储器(内存)与网络接口等部件组成,了解 epoll本质的第一步,要从硬件的角度看计算机怎样接收网络数据。

image-20240626170512773

下图展示了网卡接收数据的过程:

  • 在 1 阶段,网卡收到网线传来的数据。
  • 经过 2 阶段的硬件电路的传输。
  • 最终 3 阶段将数据写入到内存中的某个地址上。

这个过程涉及到 DMA 传输、IO 通路选择等硬件有关的知识,但我们只需知道:网卡会把接收到的数据写入内存。

image-20240626170546354

通过硬件传输,网卡接收的数据存放到内存中,操作系统就可以去读取它们。

二、如何知道接收了数据

了解 epoll 本质的第二步,要从 CPU 的角度来看数据接收。理解这个问题,要先了解一个概念:中断。

计算机执行程序时,会有优先级的需求。比如,当计算机收到断电信号时,它应立即去保存数据,保存数据的程序具有较高的优先级(电容可以保存少许电量,供 CPU 运行很短的一小段时间)。

一般而言,由硬件产生的信号需要 CPU 立马做出回应,不然数据可能就丢失了,所以它的优先级很高。

CPU 理应中断掉正在执行的程序,去做出响应,当 CPU 完成对硬件的响应后,再重新执行用户程序。

中断的过程如下图,它和函数调用差不多,只不过函数调用是事先定好位置,而中断的位置由“信号”决定。

image-20240626170632344

以键盘为例,当用户按下键盘某个按键时,键盘会给 CPU 的中断引脚发出一个高电平,CPU 能够捕获这个信号,然后执行键盘中断程序。

下图展示了各种硬件通过中断与 CPU 交互的过程:

image-20240626170650962

现在可以回答“如何知道接收了数据?”这个问题了:当网卡把数据写入到内存后,网卡向 CPU 发出一个中断信号,操作系统便能得知有新数据到来,再通过网卡中断程序去处理数据。

三、进程阻塞为什么不占用 CPU 资源

了解 epoll 本质的第三步,要从操作系统进程调度的角度来看数据接收。阻塞是进程调度的关键一环,指的是进程在等待某事件(如接收到网络数据)发生之前的等待状态,recv、select 和 epoll 都是阻塞方法。

下边分析一下进程阻塞为什么不占用 CPU 资源?为简单起见,我们从普通的 recv 接收开始分析,先看看下面代码:

1
2
3
4
5
6
7
8
9
10
11
12
//创建 socket
int s =socket(AF_INET, SOCK_STREAM, 0);
//绑定
bind(s, ...)
//监听
listen(s, ...)
//接受客户端连接
int c = accept(s, ...)
//接收客户端数据
recv(c, ...);
//将数据打印出来
printf(...)

这是一段最基础的网络编程代码,先新建 socket 对象,依次调用 bind、listen 与 accept,最后调用 recv 接收数据。

recv 是个阻塞方法,当程序运行到 recv 时,它会一直等待,直到接收到数据才往下执行。那么阻塞的原理是什么?

四、工作队列

操作系统为了支持多任务,实现了进程调度的功能,会把进程分为“运行”和“等待”等几种状态。

运行状态是进程获得 CPU 使用权,正在执行代码的状态;等待状态是阻塞状态,比如上述程序运行到recv 时,程序会从运行状态变为等待状态,接收到数据后又变回运行状态。

操作系统会分时执行各个运行状态的进程,由于速度很快,看上去就像是同时执行多个任务。

下图的计算机中运行着 A、B 与 C 三个进程,其中进程 A 执行着上述基础网络程序,一开始,这3 个进程都被操作系统的工作队列所引用,处于运行状态,会分时执行。

image-20240626170751776

五、等待队列

当进程 A 执行到创建 socket 的语句时,操作系统会创建一个由文件系统管理的 socket 对象(如下图)。

image-20240626170810525

这个 socket 对象包含了发送缓冲区、接收缓冲区与等待队列等成员。等待队列是个非常重要的结构,它指向所有需要等待该 socket 事件的进程。

当程序执行到 recv 时,操作系统会将进程 A 从工作队列移动到该 socket 的等待队列中(如下图)。

image-20240626170835637

由于工作队列只剩下了进程 B 和 C,依据进程调度,CPU 会轮流执行这两个进程的程序,不会执行进程 A 的程序。所以进程 A 被阻塞,不会往下执行代码,也不会占用 CPU 资源。

注:操作系统添加等待队列只是添加了对这个“等待中”进程的引用,以便在接收到数据时获取进程对象、将其唤醒,而非直接将进程管理纳入自己之下。上图为了方便说明,直接将进程挂到等待队列之下。

六、唤醒进程

当 socket 接收到数据后,操作系统将该 socket 等待队列上的进程重新放回到工作队列,该进程变成运行状态,继续执行代码。

同时由于 socket 的接收缓冲区已经有了数据,recv 可以返回接收到的数据。

七、内核接收网络数据全过程

这一步,贯穿网卡、中断与进程调度的知识,叙述阻塞 recv 下内核接收数据的全过程。

image-20240626170918164

如上图所示,进程在 recv 阻塞期间:

  • 计算机收到了对端传送的数据(步骤 ①)

  • 数据经由网卡传送到内存(步骤 ②)

  • 然后网卡通过中断信号通知 CPU 有数据到达,CPU 执行中断程序(步骤 ③)

此处的中断程序主要有两项功能,先将网络数据写入到对应 socket 的接收缓冲区里面(步骤 ④),

再唤醒进程 A(步骤 ⑤),重新将进程 A 放入工作队列中。

唤醒进程的过程如下图所示:

image-20240626170937542

以上是内核接收数据全过程,这里我们可能会思考两个问题:

  • 操作系统如何知道网络数据对应于哪个 socket?

  • 如何同时监视多个 socket 的数据?

第一个问题:因为一个 socket 对应着一个端口号,而网络数据包中包含了 IP 和端口的信息,内核可以通过端口号找到对应的 socket。

当然,为了提高处理速度,操作系统会维护端口号到 socket 的索引结构,以快速读取。

第二个问题是多路复用的重中之重,也正是本文后半部分的重点。

八、同时监视多个 socket 的简单方法

服务端需要管理多个客户端连接,而 recv 只能监视单个 socket,这种矛盾下,人们开始寻找监视多个 socket 的方法。epoll 的要义就是高效地监视多个 socket。

从历史发展角度看,必然先出现一种不太高效的方法,人们再加以改进,正如 select 之于 epoll。先理解不太高效的 select,才能够更好地理解 epoll 的本质。

假如能够预先传入一个 socket 列表,如果列表中的 socket 都没有数据,挂起进程,直到有一个 socket收到数据,唤醒进程。这种方法很直接,也是 select 的设计思想。

为方便理解,我们先复习 select 的用法。在下边的代码中,先准备一个数组 FDS,让 FDS 存放着所有需要监视的 socket。

然后调用 select,如果 FDS 中的所有 socket 都没有数据,select 会阻塞,直到有一个 socket 接收到数据,select 返回,唤醒进程。

用户可以遍历 FDS,通过 FD_ISSET 判断具体哪个 socket 收到数据,然后做出处理。

1
2
3
4
5
6
7
8
9
10
11
12
int s =socket(AF_INET, SOCK_STREAM, 0); 
bind(s, ...)
listen(s, ...)
int fds[] = 存放需要监听的 socket
while1){
int n = select(..., fds, ...)
forint i=0; i < fds.count; i++){
if(FD_ISSET(fds[i], ...)){
//fds[i]的数据处理
}
}
}

九、select 的流程

select 的实现思路很直接,假如程序同时监视如下图的 sock1、sock2 和 sock3 三个 socket,那

么在调用 select 之后,操作系统把进程 A 分别加入这三个 socket 的等待队列中。

当任何一个 socket 收到数据后,中断程序将唤起进程。下图展示了 sock2 接收到了数据的处理流程:

image-20240626171047235

注:recv 和 relect 的中断回调可以设置成不同的内容。

所谓唤起进程,就是将进程从所有的等待队列中移除,加入到工作队列里面,如下图所示:

image-20240626171100759

经由这些步骤,当进程 A 被唤醒后,它知道至少有一个 socket 接收了数据。程序只需遍历一遍 socket列表,就可以得到就绪的 socket。

这种简单方式行之有效,在几乎所有操作系统都有对应的实现。但是简单的方法往往有缺点,主要是:

  • 每次调用 select 都需要将进程加入到所有监视 socket 的等待队列,每次唤醒都需要从每个队列中移除。这里涉及了两次遍历,而且每次都要将整个 FDS 列表传递给内核,有一定的开销。正是因为遍历操作开销大,出于效率的考量,才会规定 select 的最大监视数量,默认只能监视 1024个 socket。

  • 进程被唤醒后,程序并不知道哪些 socket 收到数据,还需要遍历一次。

那么,有没有减少遍历的方法?有没有保存就绪 socket 的方法?这两个问题便是 epoll 技术要解决的。

补充说明:本节只解释了 select 的一种情形。当程序调用 select 时,内核会先遍历一遍 socket,

如果有一个以上的 socket 接收缓冲区有数据,那么 select 直接返回,不会阻塞。

这也是为什么 select 的返回值有可能大于 1 的原因之一。如果没有 socket 有数据,进程才会阻塞。

十、epoll 的设计思路

epoll 是在 select 出现 N 多年后才被发明的,是 select 和 poll(poll 和 select 基本一样,有少

量改进)的增强版本。epoll 通过以下一些措施来改进效率:

1、措施一:功能分离

select 低效的原因之一是将“维护等待队列”和“阻塞进程”两个步骤合二为一。

image-20240626171131861

如上图所示,每次调用 select 都需要这两步操作,然而大多数应用场景中,需要监视的 socket 相对固定,并不需要每次都修改。

epoll 将这两个操作分开,先用 epoll_ctl 维护等待队列,再调用 epoll_wait 阻塞进程。显而易见,效率就能得到提升。

为方便理解后续的内容,我们先了解一下 epoll 的用法。如下的代码中,先用 epoll_create 创建一个 epoll 对象 epfd,再通过 epoll_ctl 将需要监视的 socket 添加到 epfd 中,最后调用 epoll_wait 等待数据:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
int s =socket(AF_INET, SOCK_STREAM, 0); 

bind(s, ...)

listen(s, ...)

int epfd =epoll_create(...);

epoll_ctl(epfd, ...); //将所有需要监听的 socket 添加到 epfd 中

while1){

int n =epoll_wait(...)

for(接收到数据的 socket){

//处理

}

功能分离,使得 epoll 有了优化的可能。

2、措施二:就绪列表

select 低效的另一个原因在于程序不知道哪些 socket 收到数据,只能一个个遍历。如果内核维护一个“就绪列表”,引用收到数据的 socket,就能避免遍历。

image-20240626171211493

如上图所示,计算机共有三个 socket,收到数据的 sock2 和 sock3 被就绪列表 rdlist 所引用。

当进程被唤醒后,只要获取 rdlist 的内容,就能够知道哪些 socket 收到数据。

image-20240626171243774

Q:epoll底层红黑树的Key和val是什么?

A:

epoll 的底层实现中,红黑树主要用于管理活跃的文件描述符(file descriptors,FDs),即那些已经注册到 epoll 实例中的文件描述符。红黑树的作用是快速查找、插入和删除文件描述符,以便高效地管理大量的并发连接。

  1. Key:文件描述符(FD)
  • 红黑树中的 Key 是文件描述符(fd)。
  • fd 是一个非负整数,用于唯一标识一个打开的文件、网络连接、管道等资源。epoll 通过 fd 来标识要监控的目标。
  • 当文件描述符被注册到 epoll 实例时,它会被作为红黑树的 Key 插入到树中,方便后续的查找和管理。
  1. Valueepoll 事件信息(epitem
  • 红黑树中的 Value 是一个与文件描述符关联的结构体,通常称为 epitem,包含与该文件描述符相关的所有事件信息。
  • 这个结构体保存了文件描述符的监控事件类型(例如可读、可写、错误等),以及用户自定义的数据(通过 epoll_ctl 注册时传递的 epoll_data_t)。
  • 核心内容包括:
    • 文件描述符(fd):被监控的目标。
    • 事件掩码:表示要监控的事件类型(如 EPOLLINEPOLLOUT 等)。
    • 回调数据:用户在注册文件描述符时自定义的 epoll_data_t,可以是一个指针、整数等,用于在事件触发时返回给用户。

例子:

假设你将文件描述符 fd = 5 注册到 epoll 实例中,并监控它的可读事件(EPOLLIN),且附带了一些自定义数据。对于这次注册,epoll 会在红黑树中插入一个新的节点,其中:

  • Keyfd = 5
  • Value:一个包含事件信息的结构体epitem,其中包括:
    • fd = 5
    • 事件掩码 EPOLLIN
    • 用户自定义数据(例如指针或整数)

这棵红黑树使得 epoll 能够快速查找和管理成千上万的文件描述符。当你尝试修改或删除某个文件描述符的事件时,epoll 能利用红黑树的 O(log N) 时间复杂度高效地完成这些操作。

总结

  • Key:文件描述符(FD),唯一标识每个需要被 epoll 监控的资源。
  • Value:与文件描述符相关的事件信息(epitem 结构体),包含事件掩码和自定义的用户数据。