selectLinux中select函数的使用指南-猿码集

1. select函数概述

在Linux系统中，select函数是一种用于多路复用的IO模型。它在一组文件描述符上进行轮询，通知哪些文件描述符已经准备好进行读写操作。

select函数的定义如下：

int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout);

select函数接收五个参数：

nfds：要监听的文件描述符的最大值加1。

readfds：要监听的可读文件描述符集合。

writefds：要监听的可写文件描述符集合。

exceptfds：要监听的异常文件描述符集合。

timeout：超时时间，设置为NULL则阻塞直到有文件描述符就绪。

2. select函数使用步骤

下面是使用select函数进行多路复用的一般步骤：

2.1 创建文件描述符集合

fd_set readfds;
fd_set writefds;
FD_ZERO(&readfds);
FD_ZERO(&writefds);

上述代码创建了两个文件描述符集合readfds和writefds，并将它们初始化为空集合。

2.2 添加文件描述符到集合中

FD_SET(fd, &readfds); // 将文件描述符fd添加到可读集合中
FD_SET(fd, &writefds); // 将文件描述符fd添加到可写集合中

通过调用FD_SET宏，将待监听的文件描述符添加到对应的集合中。

2.3 设置超时时间

struct timeval timeout;
timeout.tv_sec = 5; // 设置超时时间为5秒
timeout.tv_usec = 0;

在超时时间结构体timeout中设置需要等待的时间。

2.4 调用select函数并等待就绪文件描述符

int ret = select(nfds, &readfds, &writefds, NULL, &timeout);

调用select函数，传入文件描述符的最大值加1、读文件描述符集合、写文件描述符集合和超时时间。

函数返回值ret表示就绪的文件描述符数量，如果返回值为-1，则表示出错。

2.5 处理就绪的文件描述符

if (FD_ISSET(fd, &readfds)) {
    // 文件描述符fd可读
}
if (FD_ISSET(fd, &writefds)) {
    // 文件描述符fd可写
}

通过调用FD_ISSET宏，判断文件描述符是否在就绪的集合中，从而执行相应的读写操作。

3. select函数注意事项

3.1 文件描述符可读和可写的含义

在使用select函数时，需要注意文件描述符可读和可写的含义。

对于套接字（socket）而言：

当socket可读时，表示该套接字中有数据到达，可以调用recv函数进行读取。

当socket可写时，表示该套接字的发送缓冲区有空闲空间，可以调用send函数进行写入。

所以，对于需要读取服务器发来的数据的客户端，应当将服务器的套接字添加到可读集合中；而对于需要向服务器发送数据的客户端，应当将服务器的套接字添加到可写集合中。

对于文件描述符而言：

当描述符可读时，表示从该描述符上可以读取数据。

当描述符可写时，表示可以向该描述符写入数据。

因此，需要根据具体情况将文件描述符添加到可读或可写的集合中。

3.2 select函数的阻塞和非阻塞模式

当timeout参数为NULL时，select函数将会一直阻塞，直到有文件描述符就绪。

当timeout参数为非NULL时，可以实现超时的效果，如果超过指定的时间没有文件描述符就绪，则select函数返回。

同时，可以通过设置文件描述符的阻塞和非阻塞模式来控制select函数的阻塞行为。

设置文件描述符fd为非阻塞模式：

int flags = fcntl(fd, F_GETFL, 0);
fcntl(fd, F_SETFL, flags | O_NONBLOCK);

通过fcntl函数可以获取文件描述符fd的标志位，并使用F_SETFL设置非阻塞标志位O_NONBLOCK。

恢复文件描述符fd的阻塞模式：

fcntl(fd, F_SETFL, flags & ~O_NONBLOCK);

使用位运算符对标志位进行与运算，可以将非阻塞标志位去除。

3.3 select函数的性能

select函数的性能在某些情况下可能不是最佳的选择。

当需要监听的文件描述符较多时，每次调用select函数都需要遍历所有的文件描述符，这样会带来较大的性能开销。

如果程序需要同时监听大量的文件描述符，可以考虑使用更高效的IO多路复用函数，如epoll、kqueue等。

4. 示例代码

下面是一个简单示例，演示如何使用select函数监听文件描述符的可读和可写事件：

#include <stdio.h>
#include <sys/select.h>
int main() {
    // 创建文件描述符集合
    fd_set readfds;
    fd_set writefds;
    FD_ZERO(&readfds);
    FD_ZERO(&writefds);
    int fd = fileno(stdin); // 获取标准输入的文件描述符
    // 将文件描述符添加到集合中
    FD_SET(fd, &readfds);
    FD_SET(fd, &writefds);
    // 设置超时时间
    struct timeval timeout;
    timeout.tv_sec = 5;
    timeout.tv_usec = 0;
    // 调用select函数并等待就绪文件描述符
    int ret = select(fd + 1, &readfds, &writefds, NULL, &timeout);
    if (ret == -1) {
        perror("select error");
        return -1;
    } else if (ret == 0) {
        printf("select timeout\n");
        return 0;
    }
    // 处理就绪的文件描述符
    if (FD_ISSET(fd, &readfds)) {
        printf("stdin is readable\n");
    }
    if (FD_ISSET(fd, &writefds)) {
        printf("stdin is writable\n");
    }
    return 0;
}

以上示例代码使用select函数监听标准输入的可读和可写事件，如果在超时时间内有事件发生，将会打印相应的提示信息。

总结

本文介绍了Linux中select函数的使用方法及注意事项。通过使用select函数，可以在一组文件描述符上进行轮询，并及时通知哪些文件描述符已经准备好进行读写操作。

在使用select函数时，需要注意文件描述符可读和可写的含义，以及select函数的阻塞和非阻塞模式。同时，当需要监听大量的文件描述符时，可以考虑使用更高效的IO多路复用函数。

通过实际示例代码，可以更好地理解select函数的用法。在实际开发中，根据业务需求和具体情况，合理使用select函数可以提高程序的性能和效率。

selectLinux中select函数的使用指南