首页 > 代码库 > UNIX网络编程卷1 时间获取程序client TCP 使用非堵塞connect

UNIX网络编程卷1 时间获取程序client 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 表示连接已经建立(客户和server在同一台主机时,这样的情况更有可能发生)
	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);
}


UNIX网络编程卷1 时间获取程序client TCP 使用非堵塞connect