I/O多路复用-select函数详解

为什么要使用I/O多路复用

在原来的程序中，我们可以使用fgets方法等待标准输入，但是一旦这么做，就没有办法在套接字有数据的时候读数据(read)；我们也可以使用read方法等待套接字有数据时返回，但这么做了，就没有办法在标准输入(fget)有数据时，读入数据发送给对方。

而I/O多路复用的设计初衷就是为了解决这个问题。

什么是I/O多路复用

我们将标准输入、套接字等看做I/O的一路，多路复用的意思就是任何一路I/O有“事件”发生的情况下，通知应用程序去处理相应的I/O事件，这样就仿佛可以处理多个I/O事件。

select函数

select函数就是这样一种常见的I/O多路复用技术。使用select函数，通知内核挂起进程，当一个或多个I/O事件发生后，控制权还给应用程序，由应用程序进行I/O事件的处理。

I/O事件

标准输入文件描述符准备好可以读。
监听套接字准备好，新的连接已经建立成功。
已连接套接字准备好可以写。
如果一个 I/O 事件等待超过了 10 秒，发生了超时事件。

select函数参数

int select(int maxfd, fd_set *readset, fd_set *writeset,fd_set *exceptset, const struct timeval *timeout);

//返回：若有就绪描述符则为其数目，若超时则为 0，若出错则为 -1

maxfd参数：待测试的描述符基数(个数)，它的值是待测试的最大描述符加1。比如现在select待测试的描述符集合为{0,1,4}，那么maxfd就是5。
描述符集合readset
描述符集合writeset
异常描述符集合exceptset
上面三个参数分别通知内核，在哪些描述符上检测数据可读，可以写和有异常发生。

可以通过下面的宏设置描述符集合：

void FD_ZERO(fd_set *fdset);　　　    //将描述符集合设置为空　　
void FD_SET(int fd, fd_set *fdset);　//将对应套接字fd添加入套接字集合
void FD_CLR(int fd, fd_set *fdset);　//将对应套接字fd从套接字集合中删除
int  FD_ISSET(int fd, fd_set *fdset);//判断套接字集合中是否含有该套接字

我们可以这样理解，描述符集合就是一个数组，其中数组的长度是maxfd-1，数组的下标为0到最大描述符，数组的元素为0或1。描述符在描述符集合中则其对应的下标为1，不在则为0。

比如描述符集合{0,1,4}，待测试的描述符个数为5个，其对应的向量为：a[4],a[3],a[2],a[1],a[0]。

最后一个参数timeval结构体：

struct timeval {
  long tv_sec; /* seconds 秒*/
  long tv_usec; /* microseconds 毫秒*/
};

将这个参数设置为NULL：select函数阻塞，直到有I/O事件发生。
设置为非零值：等待固定的一段时间后从select阻塞调用中返回。
结构体中两个参数设置为0：不等待，检测完毕立即返回。

select例子：

下面的程序使用i/o多路复用，启用的i/o事件有两个:(1)标准输入中输入数据，(2)从服务端接收数据。

1. 一开始会阻塞在select函数上
2. 当感知到有i/o事件，则将对应的i/o事件写到readmask中
3. 通过FD_ISSET判断发生了那个i/o事件，从而执行对应的代码
   下面的代码可以直接从for(;;)附近读，前面的是客户端的socket，connect和其他参数设置。

#include<stdio.h>
#include<error.h>
#include<errno.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<string.h>
#include<stdlib.h>
#include<arpa/inet.h>
#include<sys/select.h>

#define MAXLINE 4096

int main(int argc,char **argv){
    //1. 设置接收参数，接收服务端的ip地址信息和端口号
    if(argc!=3){
        error(1,errno,"usage: tcp_client <IPaddress> <port>");
    }
    char *server_ip=argv[1];
    int server_port=atoi(argv[2]);
    //调用socket函数，创建套接字描述符
    int sockfd;
    sockfd=socket(AF_INET,SOCK_STREAM,0);

    //3. 初始化套接字地址信息
    struct sockaddr_in serveraddr;
    bzero(&serveraddr,sizeof(serveraddr));
    serveraddr.sin_family=AF_INET;
    serveraddr.sin_port=htons(server_port);
    inet_pton(AF_INET,server_ip,&serveraddr.sin_addr);

    //4. 调用connect函数，与服务器建立联系
    int connect_rt=connect(sockfd,(struct sockaddr *)&serveraddr,sizeof(serveraddr));
    if(connect_rt<0){
        error(1,errno,"connect failed");
    }

    char send_line[MAXLINE],recv_line[MAXLINE+1];
    int n;

    fd_set readmask;
    fd_set allreads;

    FD_ZERO(&allreads);
    FD_SET(0,&allreads);
    FD_SET(sockfd,&allreads);

    for(;;){
        readmask=allreads;//重置待测试集合
        int rc=select(sockfd+1,&readmask,NULL,NULL,NULL);//设置timeval为null，阻塞在这里，直到检测到有i/o事件发生。检测到有i/o事件后，会修改readmask。
        if(rc<=0){
            error(1,errno,"select failed");
        }
        //当连接套接字上有数据可读，将数据读入到程序缓冲区
        if(FD_ISSET(sockfd,&readmask)){//检测sockfd描述符是否在readmask中
            n = read(sockfd, recv_line, MAXLINE);
            if (n < 0) {
                error(1, errno, "read error");
            } else if (n == 0) {
                error(1, 0, "server terminated \n");
            }
            recv_line[n] = 0;
            fputs(recv_line, stdout);//将参数打印到stdout中
            fputs("\n", stdout);
        }
        //当标准输入上有数据可读时，读入后进行判断
        if(FD_ISSET(0,&readmask)){//检测标准输入描述符是否在readmask中
            if (fgets(send_line, MAXLINE, stdin) != NULL) {
                if (strncmp(send_line, "quit", 4) == 0) {//比较两个字符串的前8位，相等为0
                    if (shutdown(sockfd, 1)) {
                        error(1, errno, "shutdown failed");
                    }
                } else {
                    int i = strlen(send_line);
                    if (send_line[i - 1] == '\n') {
                        send_line[i - 1] = 0;
                    }

                    size_t rt = write(sockfd, send_line, strlen(send_line));
                    if (rt < 0) {
                        error(1, errno, "write failed ");
                    }
                }
            }
        }
    }

}

select函数缺点

select所支持的文件描述符的个数是有限制的。在linux系统中，select的默认最大值为1024。