本文为senlie原创，转载请保留此地址：http://blog.csdn.net/zhengsenlie
/**
* 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);
}