本文为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); }
时间: 2024-10-12 03:42:06