Unix网络编程之select版客户端实现

1.当客户端阻塞于某个输入操作时，服务器发送过来的终止连接的消息或者服务器崩溃的FIN，客户端将接受不到，而客户端以为服务器还在正常运行，一味的给服务器发消息。那么客户端将收到一个RST，但是客户端正忙于其他操作，忽略了这个RST，客户端继续调用read方法，如果read方法调用发生在RST到来之前，将返回一个0（EOF），否则返回一个ECONNRESET（对方复位连接错误），最终退出程序。

2.因此，我们可以用select 的i/o复用的方式重新编写我们的代码：

客户端改编之后的代码：

#include "unp.h"
int
main(int argc, char **argv){
int sockfd;
struct sockaddr_in servaddr;
if(argc != 2)
err_quit("usage : tcpcli <IPaddress>");
sockfd = Socket(AF_INET,SOCK_STREAM,0);
bzero(&servaddr,sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(SERV_PORT);
Inet_pton(AF_INET,argv[i],&servaddr.sin_addr);
Connect(sockfd,(struct sockaddr*)&servaddr,sizeof(servaddr));
strcli(stdin,sockfd);
exit(0);
}

接下来是strcli函数的select 实现方法：

#include "unp.h"
void
str_cli(FILE *fp, int sockfd){
int maxfdp1,stdineof;           //我们设置一个stdineof的标志位，为零代表它可以往sockfd里面写
fd_set rset;
char buf[MAXLINE]
int  n;

stdineof = 0;
FD_ZERO(&rset);
for(;;){
if(stdineof == 0)
FD_SET(fileno(fp),&rset);      //如果标志位为0 ， 那么我们把输入流的描述符设置一下。
FD_SET(sockfd,&rset);                //套接字的描述符是都应该设的，因为我们一般只是关闭客户端的写操作，对于服务器发送来的消息还是可以读到的。
maxfdp1 = max(fileno(fp),sockfd)+1;
Select(maxfdp1,&rset,NULL,NULL,NULL);
if(FD_ISSET(sockfd,&rset)){          //判断套接字是否可读，如果可读，读入到buf缓冲区里面。
if((n = Read(sockfd,buf,MAXLINE)) == 0){
if(stdineof == 1){    //如果我们正处在FIN-WAIT-1 状态下，我们就直接返回就行了，因为我们已经受到了来自服务器端的FIN。
return;
}else
err_quit("str_cli : server terminated");
}
Write(fileno(stdout,buf,MAXLINE));
}
if(FD_ISSET(fileno(stdin),&rset)){      //判断输入流里面是否可读，如果可读，把它读到buf缓冲区里面。
if((n = Read(fileno(stdin),buf,MAXLINE)) == 0){

stdineof = 1;              //如果我们读到了来自服务器端的FIN，那么我们就直接将stdineof标志位设置成 1
Shutdown(sockfd,SHUT_WR);   //然后我们向服务器发送ACK，并且还有我们的FIN。
FD_CLR(fileno(fp),&rset);   //接下来我们一定要将rset里面注册的监听描述符移除掉，不让内核再去判断他是否可读
continue;
}
Writen(sockfd,buf,n);
}
}
}