UNIX网络编程笔记(6):I/O复用之select函数

向右看齐 2022-08-18 02:13 131阅读 0赞

上一讲中我们正确处理了僵尸子进程，使得这个简单的服务器更加健壮。不幸的是，这个程序仍然有问题。想象一下，如果一个客户正在和一个服务器子进程连接建立完毕正在通话，而服务器子进程意外终止（比如kill），服务器TCP向客户TCP发送一个FIN，但客户端正在调用fgets函数等待用户输入字符而得不到这个FIN，直到套接字读为止（可能过了很长时间）。

**1、来看看问题是什么**

首先启动服务器与客户，建立连接：

![20151204213239347][]

客户输入nihao后服务器正确回射，然后kill掉服务器子进程后客户输入and...，输出错误：

![20151204213122536][]

服务器端重启（上一讲中解决的问题）。

这个过程发生了什么？

（1）当kill掉服务器子进程后，服务器父进程正确处理了SIGCHLD信号。服务器也向客户发送一个FIN，客户回应一个ACK，TCP连接终止的前半部分完成；

（2）客户上没有发生任何特殊的事，客户TCP接收到服务器的FIN后回应一个ACK，但问题是客户正阻塞在fgets上，等待用户输入；

（3）在客户上再输入and...，导致错误发生：str\_cli调用write，客户TCP接着把数据发送给服务器，但是客户FIN的接收并没有告知客户TCP服务器进程已经终止。当服务器TCP接收到客户的数据时，发送一个RST；

（4）但是客户进程看不到这个RST，因为调用write后直接调用readline，并且由于之前接收的FIN，导致readline直接返回0（表示EOF）。但客户没有预期收到EOF，于是出错；

（5）客户终止，所有打开的描述符关闭；

问题的根本在于，当FIN到达套接字时，客户正阻塞在fgets调用上。客户实际上在应对两个描述符：套接字和用户输入，它不能单纯阻塞在某个特定的源上。select函数可以解决这个问题。

**2、select函数**

有些进程需要一种预先告知内核的能力，使得内核一旦发现进程指定的一个或多个I/O条件就绪时，它就通知进程。这个能力就是I/O复用。

select函数允许进程指示内核等待多个事件中的任何一个发生，并只在有一个或多个事件发生或经历一段指定的时间后才唤醒它。

下面的例子指出select可以告知内核在什么条件下发生时返回：

*  集合\{1,4,5\}中的任何描述符准备好读；
 *  集合\{2,7\}中的任何描述符准备好写；
 *  集合\{1,4\}中的任何描述符由异常条件待处理；
 *  已经过了12.8秒；

也就是说，调用select函数告知内核对哪些描述符（就读、写或异常条件）感兴趣以及等待多长时间。描述符不局限于套接字，任何描述符都可以使用select。

函数定义如下，包含在<sys/select.h>头文件中：

#include <sys/select.h>
    #include <sys/time.h>
    int select(int maxfdp1,fd_set *readset,fd_set *writeset,fd_set * exceptset,const struct timeval *timeout);

函数如果有描述符就绪就返回数量，超时返回0，出错返回-1。

参数含义如下：

maxfdp1：指定待测试描述符的个数，通常是待测试的最大描述符加1；

中间三个参数指定内核测试读、写和异常条件的描述符，这里设置成NULL即可；

timeval：告知内核等待所指定的时间，有三种情况：

（1）永远等待：设置为NULL；

（2）等待固定时间：在有一个描述符就绪后返回，但不超过参数所指定的时间；

（3）根本不等待：检查描述符后立刻返回；

**3、重写客户端中的str\_cli函数**

客户的套接字上的三个条件：

（1）如果对端TCP发送数据，那么该套接字变为可读，并且read返回一个大于0的值（即读入数据的字节数）；

（2）如果对端TCP发送一个FIN，那么该套接字变为可读，read返回0（EOF）；

（3）如果对端TCP发送一个RST（对端主机崩溃并重启），那么该套接字变为可读，并且read返回-1，errno中有错误码；

下面是修改后的str\_cli函数：

void str_cli(FILE *fp,int sockfd)
    {
    	int maxfdp1;
    	fd_set rset;
    	char sendline[MAXLINE],recvline[MAXLINE];
    	FD_ZERO(&rset);
    	int n;
    	for(;;)
    	{
    		FD_SET(fileno(fp),&rset);
    		FD_SET(sockfd,&rset);
    		maxfdp1=max(fileno(fp),sockfd)+1;
    		if((n=select(maxfdp1,&rset,NULL,NULL,NULL))==0)
    		{
    			printf("select timeout\n");
    			return;
    		}
    		else if(n<0)
    		{
    			printf("select error\n");
    			return;
    		}
    		if(FD_ISSET(sockfd,&rset))
    		{
    			if(readline(sockfd,recvline,MAXLINE)==0)
    			{
    				printf("str_cli:server terminated prematurely\n");
    				return;
    			}
    			fputs(recvline,stdout);
    		}
    		if(FD_ISSET(fileno(fp),&rset))
    		{
    			if(fgets(sendline,MAXLINE,fp)==NULL)
    				return;
    			write(sockfd,sendline,strlen(sendline));
    		}
    	}
    }

函数首先检查套接字，如果套接字是可读的，那么就用readline读入回射文本并输出；如果标准输入是可读的，就调用fgets读入一行，然后写入套接字。

运行结果：

（1）服务器进程终止后客户端直接终止：

![20151204220902593][]

（2）服务器端：

![20151204220848281][]

可以看到，程序运行良好。

[20151204213239347]: /images/20220731/8f1b170305eb437891547d6a80a5de8f.png
[20151204213122536]: /images/20220731/abb1afd519c747828d925a5c7e0474d8.png
[20151204220902593]: /images/20220731/8ab2b215463142cf83560c9d8b11b07d.png
[20151204220848281]: /images/20220731/a33dba00534342dcb6856620d87643b7.png