在 Linux 下,I/O 多路复用是一种高效的 I/O 模型,允许一个进程监视多个文件描述符上的 I/O 事件,从而实现同时处理多个连接或文件的目的。这种机制通常与非阻塞 I/O 结合使用,可以避免使用多线程或多进程的开销,提高系统的性能和资源利用率。
在 Linux 中,常见的实现 I/O 多路复用的机制包括 select、poll 和 epoll。这些机制在实现上有所不同,但它们的基本思想都是相似的,都是通过一组系统调用来监听多个文件描述符上的 I/O 事件。
select
select 是最早的一种 I/O 多路复用机制,它使用一个 fd_set 集合来保存要监听的文件描述符,并提供 select 函数来等待这些文件描述符上的事件。select 的缺点是,它使用了线性扫描的方式来检查每个文件描述符的状态,当文件描述符数量较大时,性能会下降。
- 准备文件描述符集合:将要监视的文件描述符添加到相应的文件描述符集合中。通常使用 FD_SET 宏来将文件描述符添加到读集合、写集合和异常集合中。
- 调用 select 函数:调用 select 函数来等待文件描述符上的事件发生。select 函数会阻塞进程,直到指定的文件描述符中的一个或多个发生了事件,或者超时时间到达。
- 检查文件描述符状态:当 select 函数返回时,需要遍历文件描述符集合,检查每个文件描述符的状态。可以使用 FD_ISSET 宏来检查文件描述符是否在集合中。
- 处理事件:根据文件描述符的状态,进行相应的处理。例如,如果某个文件描述符可读,则读取数据并处理;如果可写,则写入数据;如果有异常,则处理异常等。
poll
poll 是 select 的改进版本,它使用一个 pollfd 数组来保存要监听的文件描述符,并提供 poll 函数来等待这些文件描述符上的事件。与 select 不同,poll 使用了更加高效的数据结构来管理文件描述符,性能相对较好,但仍然存在一些限制。
- 准备 pollfd 数组:创建一个 pollfd 数组,每个元素对应一个文件描述符,用于保存要监视的文件描述符的信息。pollfd 结构体包含了文件描述符、监视的事件和发生的事件等信息。
- 填充 pollfd 数组:为每个文件描述符设置要监视的事件,包括可读、可写和异常等事件。可以使用 POLLIN、POLLOUT 和 POLLERR 等宏来设置相应的事件。
- 调用 poll 函数:调用 poll 函数来等待文件描述符上的事件发生。poll 函数会阻塞进程,直到指定的文件描述符中的一个或多个发生了事件,或者超时时间到达。
- 处理事件:当 poll 函数返回时,遍历 pollfd 数组,检查每个文件描述符上发生的事件。根据事件的类型,进行相应的处理。
- poll 和 select 在功能上类似,但在一些方面有所不同。相对于 select,poll 的优点包括:
- 没有文件描述符数量限制:poll 不受文件描述符数量的限制,可以监视任意数量的文件描述符。
- 不会修改文件描述符:poll 不会修改文件描述符的状态,因此不需要重新设置文件描述符集合。
- 没有超时参数的限制:poll 的超时参数是以毫秒为单位的整数,不像 select 那样受到秒数的限制。
epoll
epoll 是 Linux 下最先进和最高效的 I/O 多路复用机制,它引入了基于事件驱动的方式来管理文件描述符。epoll 使用一个事件表来保存要监听的文件描述符,并提供 epoll_create、epoll_ctl 和 epoll_wait 等系统调用来管理和等待事件。epoll 支持边缘触发和水平触发两种模式,性能比 select 和 poll 更好,特别是在处理大量并发连接时效果显著。
- 高效的事件通知机制:epoll 使用基于事件驱动的方式进行工作,只有在文件描述符上的事件发生时才会通知应用程序,避免了对所有文件描述符进行轮询的开销。
- 支持边缘触发和水平触发:epoll 支持边缘触发(Edge-Triggered)和水平触发(Level-Triggered)两种模式。在边缘触发模式下,只通知发生状态变化的文件描述符;在水平触发模式下,只要文件描述符处于可读或可写状态,就会通知应用程序。
- 单个系统调用监视多个文件描述符:epoll 可以同时监视多个文件描述符,而不需要为每个文件描述符创建一个线程。这使得 epoll 在处理大量并发连接时能够更加高效。
- 适用于大量并发连接:epoll 的性能优势在于它能够处理大量的并发连接,而不会因为文件描述符数量增加而降低性能。
- 更少的系统调用开销:相比于 select 和 poll,epoll 的系统调用开销更小,因为它使用了更高效的数据结构来管理文件描述符。
- epoll 的使用方式包括以下几个步骤:
- 创建 epoll 实例:使用 epoll_create 函数创建一个 epoll 实例。
- 向 epoll 实例注册文件描述符:使用 epoll_ctl 函数向 epoll 实例注册要监视的文件描述符,并指定要监视的事件类型(如可读、可写等)。
- 等待事件发生:使用 epoll_wait 函数等待文件描述符上的事件发生。epoll_wait 函数会阻塞进程,直到指定的文件描述符中的一个或多个发生了事件,或者超时时间到达。
- 处理事件:当 epoll_wait 返回时,遍历返回的事件列表,根据事件的类型进行相应的处理。
- 综上所述,epoll 是一种高效的事件通知机制,适用于处理大量并发连接的场景,特别是在网络服务器程序中能够发挥其优势。
epoll 通讯实例 (仅供参考)
- 这个示例程序创建了一个 TCP 服务器,监听端口号为 12345,使用 epoll 实例来进行事件监听。当有客户端连接时,将其加入 epoll 实例中,并在接收到客户端消息时进行回复。
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/epoll.h>
#include <sys/socket.h>
#define MAX_EVENTS 10
#define MAX_BUF_SIZE 1024
int main() {
int server_fd, client_fd, epoll_fd, nfds, n;
struct epoll_event ev, events[MAX_EVENTS];
struct sockaddr_in server_addr, client_addr;
socklen_t addr_len = sizeof(struct sockaddr_in);
char buf[MAX_BUF_SIZE];
// 创建 TCP 套接字
server_fd = socket(AF_INET, SOCK_STREAM, 0);
if (server_fd == -1) {
perror("socket");
exit(EXIT_FAILURE);
}
// 设置服务器地址和端口
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
server_addr.sin_port = htons(12345);
// 绑定地址和端口
if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
perror("bind");
exit(EXIT_FAILURE);
}
// 监听端口
if (listen(server_fd, SOMAXCONN) == -1) {
perror("listen");
exit(EXIT_FAILURE);
}
// 创建 epoll 实例
epoll_fd = epoll_create1(0);
if (epoll_fd == -1) {
perror("epoll_create1");
exit(EXIT_FAILURE);
}
// 将监听套接字加入 epoll 实例中
ev.events = EPOLLIN;
ev.data.fd = server_fd;
if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, server_fd, &ev) == -1) {
perror("epoll_ctl: server_fd");
exit(EXIT_FAILURE);
}
printf("Server started. Waiting for connections...\n");
while (1) {
// 等待事件发生
nfds = epoll_wait(epoll_fd, events, MAX_EVENTS, -1);
if (nfds == -1) {
perror("epoll_wait");
exit(EXIT_FAILURE);
}
// 处理所有就绪事件
for (int i = 0; i < nfds; ++i) {
if (events[i].data.fd == server_fd) {
// 新的连接请求
client_fd = accept(server_fd, (struct sockaddr *)&client_addr, &addr_len);
if (client_fd == -1) {
perror("accept");
exit(EXIT_FAILURE);
}
printf("New connection from %s:%d\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
// 将客户端套接字加入 epoll 实例中
ev.events = EPOLLIN;
ev.data.fd = client_fd;
if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, client_fd, &ev) == -1) {
perror("epoll_ctl: client_fd");
exit(EXIT_FAILURE);
}
} else {
// 客户端发送数据
memset(buf, 0, sizeof(buf));
n = read(events[i].data.fd, buf, sizeof(buf));
if (n <= 0) {
// 客户端断开连接
if (n == 0) {
printf("Connection closed by client\n");
} else {
perror("read");
}
// 关闭客户端套接字并从 epoll 实例中删除
close(events[i].data.fd);
epoll_ctl(epoll_fd, EPOLL_CTL_DEL, events[i].data.fd, NULL);
} else {
// 收到客户端消息,回复消息
printf("Received message from client: %s\n", buf);
write(events[i].data.fd, buf, strlen(buf));
}
}
}
}
// 关闭监听套接字和 epoll 实例
close(server_fd);
close(epoll_fd);
return 0;
}