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[n] = 0;	/* null terminate */
		if (fputs(recvline, stdout) == EOF)
			err_sys("fputs error");
	}
	if (n < 0)
		err_sys("read error");


    exit(0);
}