UNIX网络编程卷1 时间获取程序客户端 TCP 使用非阻塞connect

本文为senlie原创，转载请保留此地址：http://blog.csdn.net/zhengsenlie

1.当在一个非阻塞的 TCP 套接字(可使用 fcntl 把套接字变成非阻塞的)上调用 connect时，

connect 将立即返回一个 EINPROGRESS 错误，不过已经发起的 TCP 三路握手继续进行。

接着使用 select 检测 这个连接是否建立成功

2.用途：

1)在TCP三路握手的时候进行其他处理

2)同时建立多个连接

3)使用 select 设置超时

/**
* TCP，使用非阻塞 connect
**/

#include	"unp.h"

int
connect_nonb(int sockfd, const SA *saptr, socklen_t salen, int nsec)
{
int				flags, n, error;
socklen_t		len;
fd_set			rset, wset;
struct timeval	tval;

//1.调用 fcntl 设置套接字为非阻塞
flags = Fcntl(sockfd, F_GETFL, 0);
Fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);

//2.发起非阻塞连接
//期望的错误是 EINPROGRESS ，表示连接建立已经启动但尚未完成。
// n > 0 是什么情况
error = 0;
if ( (n = connect(sockfd, saptr, salen)) < 0)
if (errno != EINPROGRESS)
return(-1);

/*在连接建立期间，可以做其他事 */

//3.检查连接是否建立
// n == 0 表示连接已经建立(客户和服务器在同一台主机时，这种情况更有可能发生)
if (n == 0)
goto done;	/* 连接建立完毕 */

//4.调用 select
FD_ZERO(&rset);//清零 rset
FD_SET(sockfd, &rset); //打开 rset 中对应 sockfd 的位
wset = rset; //将 rset 复制到 wset
tval.tv_sec = nsec; //设置超时时间
tval.tv_usec = 0;
//调用 select 等待套接字变为可读或可写
if ( (n = Select(sockfd+1, &rset, &wset, NULL,
nsec ? &tval : NULL)) == 0) {
//处理超时，如果 select 返回 0，那么超时发生
close(sockfd);
errno = ETIMEDOUT;
return(-1);
}

//5.检查可读或可写条件
//如果套接字可写不可读，则连接建立成功
//如果套接字可写又可读，则可能 select 之前套接字已经建立成功，或连接建立失败
//如果套接字既不可写又不可读，则 select 出错，因为 sockfd 没有设置
if (FD_ISSET(sockfd, &rset) || FD_ISSET(sockfd, &wset)) {
len = sizeof(error);
//使用 getsockopt 取得套接字的待处理错误。如果连接成功建立，该值为 0.
if (getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &error, &len) < 0)
return(-1);			/* Solaris pending error */
} else
err_quit("select error: sockfd not set");

//6.关闭非阻塞状态并返回
done:
Fcntl(sockfd, F_SETFL, flags);	//恢复套接字的文件状态标志并返回

if (error) {
close(sockfd);		/* just in case */
errno = error;
return(-1);
}
return(0);
}

#include    "unp.h"

int
main(int argc, char **argv)
{
int					sockfd, n;
char				recvline[MAXLINE + 1];
struct sockaddr_in	servaddr;

if (argc != 2)
err_quit("usage: a.out <IPaddress>");

if ( (sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
err_sys("socket error");

bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port   = htons(13);
if (inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0)
err_quit("inet_pton error for %s", argv[1]);

//这里从 connect 改为 connect_nonb
if (connect_nonb(sockfd, (SA *) &servaddr, sizeof(servaddr)) < 0)
err_sys("connect error");

while ( (n = read(sockfd, recvline, MAXLINE)) > 0) {
recvline
= 0;	/* null terminate */
if (fputs(recvline, stdout) == EOF)
err_sys("fputs error");
}
if (n < 0)
err_sys("read error");

exit(0);
}