Linux Socket Select 函数实战代码

念区

于 2024-08-26 13:03:05 发布

阅读量946

点赞数 29

本文链接：https://blog.csdn.net/weixin_42602368/article/details/141609607

版权

本文还有配套的精品资源，点击获取 menu-r.4af5f7ec.gif

简介：Linux Socket Select 函数是用于处理多个 I/O 事件（包括 Socket 读写就绪状态）的系统调用。本代码示例演示了如何将 Select 函数与 Socket 配合使用，以监控文件描述符，等待它们变为可读、可写或出现错误状态。该代码包括服务器端和客户端，展示了 Socket 创建、绑定、监听、连接和事件处理的完整流程。通过分析和运行此代码，开发者可以深入理解 Linux 网络编程和 I/O 多路复用技术。

1. Socket 简介

Socket 是一个网络编程接口，它允许应用程序通过网络与其他计算机进行通信。Socket 本质上是一个端点，它表示网络连接的一端。应用程序可以使用 Socket 来发送和接收数据，从而实现网络通信。Socket 编程是网络编程的基础，它为应用程序提供了与网络进行交互的接口。

2. Select 函数语法和参数

2.1 Select 函数的语法

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

参数说明：

nfds ：需要监视的文件描述符数量，必须大于等于 0。
readfds ：指向一个 fd_set 结构体，用于指定需要监视可读事件的文件描述符。
writefds ：指向一个 fd_set 结构体，用于指定需要监视可写事件的文件描述符。
exceptfds ：指向一个 fd_set 结构体，用于指定需要监视异常事件的文件描述符。
timeout ：指向一个 timeval 结构体，用于指定等待时间。如果 timeout 为 NULL ，则 select 函数将一直阻塞，直到至少有一个文件描述符满足监视条件。

2.2 Select 函数的参数

2.2.1 readfds 参数

readfds 参数是一个 fd_set 结构体，用于指定需要监视可读事件的文件描述符。 fd_set 结构体是一个位图，其中每个位对应一个文件描述符。如果某个位被置位，则表示该文件描述符需要监视可读事件。

2.2.2 writefds 参数

writefds 参数是一个 fd_set 结构体，用于指定需要监视可写事件的文件描述符。 fd_set 结构体是一个位图，其中每个位对应一个文件描述符。如果某个位被置位，则表示该文件描述符需要监视可写事件。

2.2.3 exceptfds 参数

exceptfds 参数是一个 fd_set 结构体，用于指定需要监视异常事件的文件描述符。 fd_set 结构体是一个位图，其中每个位对应一个文件描述符。如果某个位被置位，则表示该文件描述符需要监视异常事件。

2.2.4 timeout 参数

timeout 参数是一个 timeval 结构体，用于指定等待时间。 timeval 结构体包含两个成员：

tv_sec ：以秒为单位的等待时间。
tv_usec ：以微秒为单位的等待时间。

如果 timeout 为 NULL ，则 select 函数将一直阻塞，直到至少有一个文件描述符满足监视条件。如果 timeout 不为 NULL ，则 select 函数将在指定的时间内阻塞，如果在此时间内没有任何文件描述符满足监视条件，则 select 函数将返回 0。

3. FD_SET 集合操作

3.1 FD_SET 集合的定义

FD_SET 集合是用于管理文件描述符集合的一种数据结构。它是一个包含文件描述符的整型数组。每个文件描述符都对应数组中的一个位。如果文件描述符在集合中，则相应的位被设置为 1；否则，该位被设置为 0。

3.2 FD_SET 集合的操作函数

3.2.1 FD_ZERO

FD_ZERO 函数用于初始化一个 FD_SET 集合，将集合中的所有位都设置为 0。

void FD_ZERO(fd_set *set);

参数说明：

set ：要初始化的 FD_SET 集合。

3.2.2 FD_SET

FD_SET 函数将一个文件描述符添加到 FD_SET 集合中。如果文件描述符已经在集合中，则该函数不会执行任何操作。

void FD_SET(int fd, fd_set *set);

参数说明：

fd ：要添加到集合的文件描述符。
set ：要添加文件描述符的 FD_SET 集合。

3.2.3 FD_CLR

FD_CLR 函数将一个文件描述符从 FD_SET 集合中删除。如果文件描述符不在集合中，则该函数不会执行任何操作。

void FD_CLR(int fd, fd_set *set);

参数说明：

fd ：要从集合中删除的文件描述符。
set ：要删除文件描述符的 FD_SET 集合。

3.2.4 FD_ISSET

FD_ISSET 函数检查一个文件描述符是否在 FD_SET 集合中。如果文件描述符在集合中，则该函数返回 1；否则，返回 0。

int FD_ISSET(int fd, fd_set *set);

参数说明：

fd ：要检查的文件描述符。
set ：要检查的文件描述符的 FD_SET 集合。

代码块：

#include <sys/select.h>

int main() {
    fd_set readfds;
    FD_ZERO(&readfds);  // 初始化 readfds 集合

    int fd = 0;
    FD_SET(fd, &readfds);  // 将 fd 添加到 readfds 集合

    if (FD_ISSET(fd, &readfds)) {  // 检查 fd 是否在 readfds 集合中
        // fd 在集合中，可以进行读操作
    }

    return 0;
}

代码逻辑分析：

初始化一个 FD_SET 集合 readfds 。
将文件描述符 fd 添加到 readfds 集合中。
使用 FD_ISSET 函数检查 fd 是否在 readfds 集合中。
如果 fd 在集合中，则可以对 fd 进行读操作。

4. 实践应用

4.1 服务器端 Socket 创建、绑定、监听

在建立网络通信之前，服务器端需要创建 Socket、绑定地址和端口，并开始监听来自客户端的连接请求。

#include <sys/socket.h>
#include <netinet/in.h>

int main() {
    // 创建 Socket
    int server_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (server_fd == -1) {
        perror("socket");
        return -1;
    }

    // 绑定地址和端口
    struct sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_port = htons(8080);
    if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
        perror("bind");
        return -1;
    }

    // 开始监听
    if (listen(server_fd, 5) == -1) {
        perror("listen");
        return -1;
    }

    // ... 后续代码
}

代码逻辑分析：

socket 函数创建一个新的 Socket，并返回 Socket 描述符 server_fd 。
bind 函数将 Socket 绑定到指定的 IP 地址和端口。
listen 函数将 Socket 设置为监听模式，并指定允许排队的最大连接数。

4.2 客户端 Socket 创建、连接

客户端需要创建 Socket 并连接到服务器端的地址和端口。

#include <sys/socket.h>
#include <netinet/in.h>

int main() {
    // 创建 Socket
    int client_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (client_fd == -1) {
        perror("socket");
        return -1;
    }

    // 连接到服务器
    struct sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    server_addr.sin_port = htons(8080);
    if (connect(client_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
        perror("connect");
        return -1;
    }

    // ... 后续代码
}

代码逻辑分析：

socket 函数创建一个新的 Socket，并返回 Socket 描述符 client_fd 。
connect 函数将 Socket 连接到指定的服务器地址和端口。

4.3 Select 函数监控 Socket 事件

Select 函数用于监控多个 Socket 的事件，包括可读、可写和异常事件。

#include <sys/select.h>

int main() {
    // ... 前面创建 Socket 的代码

    // 初始化 FD_SET 集合
    fd_set readfds;
    FD_ZERO(&readfds);
    FD_SET(server_fd, &readfds);

    // 循环监控 Socket 事件
    while (1) {
        // 设置 select 超时时间
        struct timeval timeout;
        timeout.tv_sec = 5;
        timeout.tv_usec = 0;

        // 调用 select 函数
        int num_fds = select(server_fd + 1, &readfds, NULL, NULL, &timeout);
        if (num_fds == -1) {
            perror("select");
            return -1;
        }

        // 处理事件
        if (FD_ISSET(server_fd, &readfds)) {
            // 服务器端有新的连接请求
            // ... 处理连接请求的代码
        }
    }

    // ... 后续代码
}

代码逻辑分析：

初始化 readfds FD_SET 集合，并设置服务器端 Socket 的可读事件。
调用 select 函数，监控 readfds 集合中的 Socket 事件，并设置超时时间。
select 函数返回可读事件的 Socket 数量。
检查 server_fd 是否在可读事件的 Socket 中，如果是，则处理新的连接请求。

4.4 事件处理和读写操作

在监控到 Socket 事件后，需要进行相应的事件处理和读写操作。

// 处理新的连接请求
if (FD_ISSET(server_fd, &readfds)) {
    // 接受连接请求，创建新的客户端 Socket
    int client_fd = accept(server_fd, NULL, NULL);
    if (client_fd == -1) {
        perror("accept");
        continue;
    }

    // 将新的客户端 Socket 加入到 FD_SET 集合中
    FD_SET(client_fd, &readfds);
}

// 处理客户端数据读取
if (FD_ISSET(client_fd, &readfds)) {
    // 从客户端读取数据
    char buffer[1024];
    int num_bytes = read(client_fd, buffer, sizeof(buffer));
    if (num_bytes == -1) {
        perror("read");
        continue;
    }

    // 处理读取到的数据
    // ...
}

// 处理客户端数据写入
if (FD_ISSET(client_fd, &writefds)) {
    // 向客户端写入数据
    char *data = "Hello from server";
    int num_bytes = write(client_fd, data, strlen(data));
    if (num_bytes == -1) {
        perror("write");
        continue;
    }
}

代码逻辑分析：

处理新的连接请求，接受连接并创建新的客户端 Socket。
将新的客户端 Socket 加入到 FD_SET 集合中，以便后续监控其事件。
处理客户端数据读取，从客户端读取数据并进行处理。
处理客户端数据写入，向客户端写入数据。

4.5 错误处理和性能优化

在 Socket 编程中，错误处理和性能优化至关重要。

错误处理

// 错误处理示例
if (server_fd == -1) {
    perror("socket");
    return -1;
}

代码逻辑分析：

如果 socket 函数创建 Socket 失败，则打印错误信息并返回 -1。

性能优化

// 性能优化示例
// 使用较大的 FD_SET 大小来提高性能
fd_set readfds;
FD_SETSIZE = 1024;
FD_ZERO(&readfds);

代码逻辑分析：

增加 FD_SET 的大小可以提高 select 函数的性能，因为可以同时监控更多的 Socket。

5. Socket Select 函数的局限性和替代方案

5.1 Socket Select 函数的局限性

Select 函数虽然在实现多路复用方面具有较好的兼容性，但它也存在一些局限性：

可扩展性差： Select 函数在监控大量文件描述符时，效率会显著下降。这是因为 Select 函数使用轮询的方式遍历所有文件描述符，当文件描述符数量较多时，轮询的开销会变得非常大。
缺乏状态信息： Select 函数只能告知应用程序哪些文件描述符已准备好进行读写操作，但无法提供有关这些文件描述符的具体状态信息，例如连接是否已断开或错误类型。
不支持边缘触发： Select 函数只支持水平触发，即当文件描述符准备好进行读写操作时，它会一直触发事件。这可能会导致应用程序处理重复的事件，从而降低性能。