最近项目中,有个需求是检测某ip地址是否是通的,使用了socket的connect函数。但是,当ip地址写错的话,connect就会一直阻塞在那里,大概2、3分钟才能返回连接失败。这对于用户来说是不可接受的。下面的文章介绍了两种方法实现这种超时设置:
转自http://blog.csdn.net/ast_224/article/details/2957294
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