select使用实例——str_cli函数（修订版）

我们可以使用select函数重写/article/6791685.html中的str_cli函数，这样服务器进程一终止，客户就能马上得到通知。早先那个版本的问题在于：当套接口上发生某些事件时，客户可能阻塞于fgets调用。新版本改为阻塞于select调用，等待要么标准输入可读，要么套接口可读。下图展示了调用select所处理的各种条件。





客户的套接口上的三个条件处理如下：


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


（ii）如果对端TCP发送一个FIN（对端进程终止），那么该头接口变为可读，并且read返回0（EOF）。


（iii）如果对端TCP发送一个RST（对端主机崩溃并重新启动），那么该套接口变为可读，并且read返回-1，而errno中含有确切的错误代码。

新版本源代码：

#include <stdio.h>
#include <sys/select.h>
#include <sys/time.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>

int
max(int a, int b)
{
return(a >= b ? a : b);
}

void
str_cli(FILE *fp, int sockfd)
{
int       maxfdpl;
fd_set    rset;
char      sendline[4096], recvline[4096];

FD_ZERO(&rset);
for(;;)
{
FD_SET(fileno(fp), &rset);
FD_SET(sockfd, &rset);
maxfdpl = max(fileno(fp), sockfd) + 1;
if(select(maxfdpl, &rset, NULL, NULL, NULL) < 0)
{
perror("select");
exit(1);
}

if(FD_ISSET(sockfd, &rset))    /* socket is readable */
{
if(readline(sockfd, recvline, 4096) == 0)
{
printf("str_cli: server terminated prematurely\n");
exit(1);
}
fputs(recvline, stdout);
}

if(FD_ISSET(fileno(fp), &rset)) /* input is readable */
{
if(fgets(sendline, 4096, fp) == NULL)
return;
writen(sockfd, sendline, strlen(sendline));
}
}
}

调用select

我们只需要一个用于检查可读性的描述字集。该集合由FD_ZERO初始化，并用FD_SET打开两位：一位对应于标准I/O文件指针fp，一位对应于套接口sockfd。fileno函数把标准I/O文件指针转换为对应的描述字。select和poll只工作在描述字上。

计算出两个描述字中的较大值后，调用select。在该调用中，写集合指针和异常集合指针都是空指针。最后一个参数（时间限制）也是空指针，因为我们希望本调用阻塞到某个描述字就绪为止。