正文

Linux系统下select函数的原理与实战应用详解

/0 浏览量
0529

在Linux系统编程中，select函数是一个非常重要的函数，它允许程序监视多个文件描述符（如socket、文件等）上的I/O事件。本文将详细介绍select函数的原理，并展示如何在实战中应用它。

一、select函数简介

select函数的原型如下：

int select(int maxfdp1, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout);
  • maxfdp1：指定要监视的文件描述符中最大的一个加1。
  • readfds：指向fd_set类型的指针，用于指定需要监视读操作的文件描述符集合。
  • writefds：指向fd_set类型的指针，用于指定需要监视写操作的文件描述符集合。
  • exceptfds：指向fd_set类型的指针，用于指定需要监视异常操作的文件描述符集合。
  • timeout：指向timeval类型的指针，用于指定等待事件发生的超时时间。

select函数会阻塞调用进程，直到至少有一个指定的文件描述符就绪（可读、可写或异常），或者超时。

二、select函数原理

select函数通过内部数据结构来跟踪每个文件描述符的状态，并在调用时检查这些状态。以下是select函数的基本原理：

  1. 创建内部数据结构：select函数会创建一个内部数据结构来跟踪每个文件描述符的状态。
  2. 遍历文件描述符：select函数会遍历所有指定的文件描述符，检查它们的状态。
  3. 检查文件描述符状态：对于每个文件描述符，select函数会检查它是否就绪（可读、可写或异常）。
  4. 返回就绪文件描述符：当至少有一个文件描述符就绪时，select函数会返回，并将就绪的文件描述符集合填充到readfds、writefds和exceptfds中。

三、实战应用

下面是一个使用select函数的简单示例，该示例演示了如何使用select函数监视两个socket的读写事件。

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/select.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#define PORT 8080
#define MAX_CLIENTS 10

int main() {
    int server_fd, client_fd;
    struct sockaddr_in server_addr, client_addr;
    socklen_t client_addr_len;
    fd_set master_set, working_set;
    int max_sd;

    // 创建socket
    server_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (server_fd == -1) {
        perror("socket creation failed");
        exit(EXIT_FAILURE);
    }

    // 绑定socket
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = INADDR_ANY;
    server_addr.sin_port = htons(PORT);
    if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
        perror("bind failed");
        exit(EXIT_FAILURE);
    }

    // 监听socket
    if (listen(server_fd, MAX_CLIENTS) < 0) {
        perror("listen failed");
        exit(EXIT_FAILURE);
    }

    // 初始化文件描述符集合
    FD_ZERO(&master_set);
    FD_SET(server_fd, &master_set);
    max_sd = server_fd;

    while (1) {
        // 复制master_set到working_set
        FD_ZERO(&working_set);
        FD_SET(server_fd, &working_set);

        // 等待事件发生
        int activity = select(max_sd + 1, &working_set, NULL, NULL, NULL);

        if (activity < 0) {
            perror("select error");
            exit(EXIT_FAILURE);
        }

        if (FD_ISSET(server_fd, &working_set)) {
            client_addr_len = sizeof(client_addr);
            client_fd = accept(server_fd, (struct sockaddr *)&client_addr, &client_addr_len);
            if (client_fd < 0) {
                perror("accept error");
                continue;
            }
            FD_SET(client_fd, &master_set);
            if (client_fd > max_sd) {
                max_sd = client_fd;
            }
        }

        // 遍历所有连接的客户端
        for (int i = 0; i <= max_sd; i++) {
            if (FD_ISSET(i, &working_set)) {
                if (i == server_fd) {
                    continue;
                }
                char buffer[1024];
                int read_size = read(i, buffer, sizeof(buffer));
                if (read_size <= 0) {
                    close(i);
                    FD_CLR(i, &master_set);
                } else {
                    write(i, buffer, read_size);
                }
            }
        }
    }

    close(server_fd);
    return 0;
}

在这个示例中，我们创建了一个socket，并使用select函数监视该socket的读写事件。当有新的客户端连接时，我们接受连接并将其添加到master_set集合中。然后，我们遍历所有连接的客户端，读取它们的输入并回显它们的输出。

四、总结

select函数是一个强大的工具，可以帮助我们在Linux系统编程中监视多个文件描述符的I/O事件。通过本文的介绍，你应该已经了解了select函数的原理和实战应用。希望本文能帮助你更好地理解select函数，并在实际项目中使用它。

-- 展开阅读全文 --