将阻塞模式的socket连结变为非阻塞模式

发信人: landyhorse (马儿), 信区: NetPRG
标 题: ---将阻塞模式的socket连结变为非阻塞模式---
发信站: BBS 水木清华站 (Fri Dec 7 15:45:40 2001)

前两天在linux下作了这方面的应用，
当时也是找了好半天才搞出来，
看到有人问，就讲讲基本原理，和一段代码，免得大家再走弯路
该方法最适合线程使用

1。原理(来自linuxforum)

创建socket后，先把connect()用的socket的模式保存，
再将其设成nonblock模式（使用fcntl函数）,
然后进行connect()。
connect()会return两种情况，一是success,
那就restore socket的以前mode, again use fcntl(), 然后
就算connect好了，二则是-1, 那么就要看errno, 如果是
EINPROGRESS，则就用把socket加到FD_SET里，然后用select()
select socket for writting.
因为select()可以用timeout, 然后如果select() return的话，
如果return 0, 那就是timeout, -1就要看你根据errno如何处理了，
如果是>0, 则用FD_ISSET()看是否是socket connect好了，这时
还有个condition是connection refused, 对方送了你一个RST,
你需要用getsockopt(sockfd, SOL_SOCKET,SO_ERROR,....)去
查一下，如果没有问题那就是connect success了

2.例程代码(来自白云黄鹤)

/************************************************************************
* tcpconnect.c, by digger, 带超时限制的非阻塞式connect *
************************************************************************/

# include <stdio.h>
# include <string.h>
# include <sys/types.h>
# include <sys/socket.h>
# include <sys/time.h>
# include <fcntl.h>
# include <unistd.h>
# include <netinet/in.h>
# include <arpa/inet.h>
# include <netdb.h>
# include <errno.h>

/************************************************************************
* 函数: tcpconnect( ) *
* 输入: char * server_addr, u_short server_port, int limit_seconds *
* 成功: 返回创建并连接服务端成功的socket文件号, 失败: 返回-1 *
************************************************************************/

int tcpconnect( char * server_addr, /* 服务端地址, 域名或IP地址 */
u_short server_port, /* 服务端口 */
int limit_seconds /* 超时限制, 单位为秒 */
)
{
struct hostent * hp; /* host entry pointer */
struct sockaddr_in server; /* server address structure */
int sock; /* socket to create */
fd_set fdW; /* fileds to be select */
struct timeval timeout; /* timeout parameter */
int fflag; /* file flag for fcntl */
int errcode; /* error code for getsockopt */
int errlen; /* length of errcode */

/* fill in server address structure with ip address and port */

bzero(&server, sizeof(server));
server.sin_family = AF_INET;
server.sin_port = htons(server_port);
if ((hp = gethostbyname(server_addr)) == NULL) {
if ((server.sin_addr.s_addr = inet_addr(server_addr)) == -1) {
fprintf(stderr, "%s: unknown host/n", server_addr);
exit(1);
}
} else {
bcopy(hp->h_addr_list[0], &server.sin_addr, hp->h_length);
}

/* create client socket and connect it to server */

if ((sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
perror("socket");
return(-1);
}

/* get original sock flag */

if ((fflag = fcntl(sock, F_GETFL, 0)) < 0) {
perror("fcntl get file flag");
exit(-1);
}

/* set O_NDELAY flag to sock */

if (fcntl(sock, F_SETFL, fflag|O_NDELAY) < 0) {
perror("fcntl set file flag O_NDELAY");
exit(-1);
}

/* connect to server using nonblock mode */

connect(sock, (struct sockaddr *)&server, sizeof(server));

/* set timeout structure */

timeout.tv_sec = limit_seconds;
timeout.tv_usec = 0;

FD_ZERO(&fdW);
FD_SET(sock, &fdW);

CNNT_AGAIN:

switch(select(sock + 1, NULL, &fdW, NULL, &timeout)) {

case -1:
if (errno == EINTR) goto CNNT_AGAIN; /* call interrupted */
close(sock); /* select error */
exit(-1);

case 0:
close(sock);
return(-1);

default:
if (FD_ISSET(sock, &fdW)) { /* check if socket writable */
errlen = sizeof(errcode);
getsockopt(sock, SOL_SOCKET, SO_ERROR,
(char *)&errcode,&errlen);
if(errcode == 0) { /* check if connect right */
/* reset original sock flag */
if (fcntl(sock, F_SETFL, fflag) < 0) {
perror("fcntl reset file flag");
exit(-1);
}
return(sock);
} else {
close(sock);
return(-1);
}
}
}