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

我们可以使用select函数重写http://www.CUOXin.com/nufangrensheng/p/3587962.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);}voidstr_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。在该调用中，写集合指针和异常集合指针都是空指针。最后一个参数（时间限制）也是空指针，因为我们希望本调用阻塞到某个描述字就绪为止。