如果用阻塞模式,调用connect 函数时 如果服务端IP地址或者端口号错误,要等待2-3分钟connect 函数才返回链接失败。这是无法忍受的。
改进实现如下:
connect socket的超时设置
最近项目中,有个需求是检测某ip地址是否是通的,使用了socket的connect函数。但是,当ip地址写错的话,connect就会一直阻塞在那里,大概2、3分钟才能返回连接失败。这对于用户来说是不可接受的。下面的文章介绍了两种方法实现这种超时设置:
connect超时:
目前各平台通用的设置socket connect超时的办法是通过select(),具体方法如下:
1.建立socket; 2.将该socket设置为非阻塞模式; 3.调用connect(); 4.使用select()检查该socket描述符是否可写; 5.根据select()返回的结果判断connect()结果; 6.将socket设回阻塞模式。
//设socket为非阻塞
unsigned long ul=1;
int rm=ioctl(sockfd,FIONBIO,&ul);
if(rm==-1)
{
close(sockfd);
return 0;
}
if (connect(sockfd, (struct sockaddr *)&their_addr, sizeof(struct sockaddr)) == 0)
{
printf("connected/n");//正常连接(小小:使用gdb调试时,从来走不到这一步"设置为非阻塞,connect调用后,无论连接是否建立立即返回-1")
}
if(errno!=EINPROGRESS)//若errno不是EINPROGRESS,则出错(EINPROGRESS:以非阻塞的方式来进行连接的时候,返回的结果如果是 -1,这并不代表这次连接发生了错误,如果它的返回结果是 EINPROGRESS,那么就代表连接还在进行中)
{
perror("connect");
printf("cannot connect:%s/n",server);
return 0;
}
//使用select设置超时
struct timeval timeout;
fd_set r;
FD_ZERO(&r);
FD_SET(sockfd,&r);
timeout.tv_sec=0;
timeout.tv_usec=100;
int retval = select(sockfd+1,NULL,&r,NULL,&timeout);
if(retval==-1)
{
perror("select");
return 0;
}
else if(retval == 0)
{
fprintf(stderr,"Timeout/n");
return 0;
}
printf("%sconnected/n",server);
//将socket设置回正常的阻塞模式
ul1=0;
rm=ioctl(sockfd,FIONBIO,(unsigned long*)&ul1);
if(rm==-1)
{
close(sockfd);
return 0;
}
以上代码工作的很好,并且也可以通过getsockopt()获得连接发生错误的确切信息,但这总方法难免觉得有些复杂,因为要涉及到阻塞状态的解除和回置。
这里有个简单的操作方法,同样可以设置连接超时:即通过SO_SNDTIMO套节字参数。
原因是:Linux内核源码中connect的超时参数和SO_SNDTIMO操作的参数一致。
因此,在linux平台下,可以通过connect之前设置SO_SNDTIMO来达到控制连接超时的目的。
struct timeval timeo;
socklen_t len = sizeof(timeo);
timeo.tv_sec = overtime;
if (setsockopt(sockfd, SOL_SOCKET, SO_SNDTIMEO, &timeo, len) == -1)
{
strcpy(reason,strerror(errno));
perror("setsockopt");
return 0;
}
their_addr.sin_family = AF_INET;
their_addr.sin_port = htons(serverStruct->port);
their_addr.sin_addr = *((struct in_addr *)he->h_addr);
bzero(&(their_addr.sin_zero), 8);
if (connect(sockfd, (struct sockaddr *)&their_addr, sizeof(struct sockaddr)) == -1)
{
if (errno == EINPROGRESS)
{
strcpy(reason,"timeout");
return 0;
}
strcpy(reason,strerror(errno));
perror("connect");
return 0;
}
小小:上述两种实现方式都可以达到目的。第二种比较简单,利用直接修改connect内部实现用到的参数。但是这种方式可能会产生其他问题吧?(我不确定),可能会影响到其他socket编程接口的超时设定。根据SO_SNDTIMO字面意思来看,是send time out的意思,感觉可能会影响某些发送数据的函数,比如send