Poll的原理及代码

2024-01-02 14:27:01

一、Select的存在问题及缺点

????????使用 select 时,需要在每次调用前重新设置位集的原因主要是 select 的工作方式。

? ?select 函数使用三个位集(fd_set)来表示文件描述符的状态,包括可读、可写和异常。在每次调用 select 之前,你需要告诉它哪些文件描述符你关心,以及关心它们的哪些事件。这是通过设置相应的位来实现的。

1. 初始化位集: 在每次调用 select 之前,你需要通过 FD_ZERO 宏将位集清零,然后通过 FD_SET 宏将你关心的文件描述符加入位集。

fd_set readfds;
FD_ZERO(&readfds);       // 清零
FD_SET(fd1, &readfds);   // 将 fd1 加入位集
FD_SET(fd2, &readfds);   // 将 fd2 加入位集

2.调用 select 然后,你将准备好的位集传递给 select 函数?

int ready = select(maxfd + 1, &readfds, NULL, NULL, NULL);

?????????这里 maxfd 是要监视的文件描述符中的最大值加一。select 将会修改传递给它的位集,指示哪些文件描述符处于就绪状态。

3.?检查位集: 调用 select 后,你需要检查位集,以确定哪些文件描述符处于就绪状态。由于 select 修改传递给它的位集,因此在每次调用之前都需要重新设置,以确保你关心的文件描述符和事件正确地反映在位集中。

if (FD_ISSET(fd1, &readfds)) {
    // fd1 可读
}
if (FD_ISSET(fd2, &readfds)) {
    // fd2 可读
}

1.文件描述符数量限制:

  • select 使用位集来表示文件描述符的状态,而这种表示方式有一个上限,通常由 FD_SETSIZE 宏定义。这意味着在某些系统上,你可能不能使用 select 监视超过该限制的文件描述符。

2. 线性扫描:

  • select 在监视文件描述符时需要线性扫描整个位集,这导致其时间复杂度为 O(n),其中 n 是文件描述符的数量。在大规模文件描述符集合上,性能可能较差。

3. 无法原子地更新:

  • select 的接口要求在调用之前重新设置位集,而在调用期间可能有新的文件描述符被动态添加或移除。这可能导致在多线程或多进程环境中无法原子地更新位集,从而引入竞态条件。

4. 不提供对文件描述符事件的详细信息:

  • select 只能告诉你文件描述符是否就绪,但不能提供更详细的信息,例如为什么文件描述符就绪以及具体的就绪事件。这可能导致在处理复杂场景时需要进行额外的查询或检查。

5.不支持超时精度:

  • select 使用 struct timeval 来表示超时时间,但其精度通常较低,只能精确到微秒级别。在某些应用场景下,可能需要更高精度的超时控制。

?二、Poll

??poll 是一种系统调用,用于实现多路复用,允许一个进程监视多个文件描述符的状态,以确定它们中是否有可以进行 I/O 操作的文件描述符。它是 POSIX 标准的一部分,提供了一种相对现代且更强大的多路复用机制。

poll 的工作方式:

  • poll 通过数组 fds 来指定要监视的文件描述符和关注的事件。
  • 当调用 poll 时,它会阻塞,直到指定的文件描述符中的至少一个就绪,或者超时发生。
  • poll 返回时,通过检查 revents 字段,可以确定哪些文件描述符处于就绪状态。
#include <poll.h>
struct pollfd {
int fd; /* 委托内核检测的文件描述符 */
short events; /* 委托内核检测文件描述符的什么事件 */
short revents; /* 文件描述符实际发生的事件 */
};
struct pollfd myfd;
myfd.fd = 5;
myfd.events = POLLIN | POLLOUT;
int poll(struct pollfd *fds, nfds_t nfds, int timeout);
- 参数:
- fds : 是一个struct pollfd 结构体数组,这是一个需要检测的文件描述符的集合
- nfds : 这个是第一个参数数组中最后一个有效元素的下标 + 1
- timeout : 阻塞时长
0 : 不阻塞
-1 : 阻塞,当检测到需要检测的文件描述符有变化,解除阻塞
>0 : 阻塞的时长
- 返回值:
-1 : 失败
>0(n) : 成功,n表示检测到集合中有n个文件描述符发生变化
优点:

1. poll 没有文件描述符数量的限制,因为它使用动态分配的数据结构来存储文件描述符。

2.?在使用 poll 函数时,不需要在每次调用前重新设置描述符集。与 select 不同,poll 使用一个数组(struct pollfd 数组)来传递要监视的文件描述符以及关注的事件,而不是位集。因此,不需要像 select 那样在每次调用前重新设置位集。

缺点:
  • poll 的性能可能在大规模文件描述符集合上较差,因为它仍然需要线性扫描 struct pollfd 数组。
服务器端代码:
#include <stdio.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/select.h>
#include<poll.h>
int main() {

    // 创建socket
    int lfd = socket(PF_INET, SOCK_STREAM, 0);
    struct sockaddr_in saddr;
    saddr.sin_port = htons(9999);
    saddr.sin_family = AF_INET;
    saddr.sin_addr.s_addr = INADDR_ANY;

    // 绑定
    bind(lfd, (struct sockaddr *)&saddr, sizeof(saddr));

    // 监听
    listen(lfd, 8);

    // 创建一个fd_set的集合,存放的是需要检测的文件描述符
//    fd_set rdset, tmp;
//    FD_ZERO(&rdset);
//    FD_SET(lfd, &rdset);
//    int maxfd = lfd;

    // 初始化检测的文件描述符数组
    struct pollfd fds[1024];
    int i=0;
    for(i;i<1024;i++){
        fds[i].fd = -1;
        fds[i].events = POLLIN;
    }
    fds[0].fd=lfd;
    int nfds =0;
    while(1) {

//        tmp = rdset;

        // 调用select系统函数,让内核帮检测哪些文件描述符有数据
//        int ret = select(maxfd + 1, &tmp, NULL, NULL, NULL);

        // 调用poll系统函数,让内核帮检测哪些文件描述符有数据
        int ret = poll(fds,nfds+1,-1);
        if(ret == -1) {
            perror("poll");
            exit(-1);
        } else if(ret == 0) {
            continue;
        } else if(ret > 0) {
            // 说明检测到了有文件描述符的对应的缓冲区的数据发生了改变
            if(fds[0].revents & POLLIN) {
                // 表示有新的客户端连接进来了
                struct sockaddr_in cliaddr;
                int len = sizeof(cliaddr);
                int cfd = accept(lfd, (struct sockaddr *)&cliaddr, &len);

                // 将新的文件描述符加入到集合中
//                FD_SET(cfd, &rdset);
                int i1=1;
                for ( i1;i1<1024;i1++){
                    if(fds[i1].fd==-1){
                        fds[i1].fd=cfd;
                        fds[i1].events=POLLIN;
                        break;
                    }
                }
                // 更新最大的文件描述符
                nfds = nfds > cfd ? nfds : cfd;
            }
            int i =  1;
            for(i; i <= nfds; i++) {
                if(fds[i].revents & POLLIN) {
                    // 说明这个文件描述符对应的客户端发来了数据
                    char buf[1024] = {0};
                    int len = read(fds[i].fd, buf, sizeof(buf));
                    if(len == -1) {
                        perror("read");
                        exit(-1);
                    } else if(len == 0) {
                        printf("client closed...\n");
                        close(fds[i].fd);
                        fds[i].fd=-1;
                    } else if(len > 0) {
                        printf("read buf = %s\n", buf);
                        write(fds[i].fd, buf, strlen(buf) + 1);
                    }
                }
            }

        }

    }
    close(lfd);
    return 0;
}

文章来源:https://blog.csdn.net/leimeili/article/details/135334521
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。