主要有三种方案:
(1)调用Alarm,它在指定超时期满时产生SIGALRM信号,此方法涉及信号处理,而信号处理在不同的实现上存在差异,而且可能干扰进程中现有的ALARM调用。
(2)在SELECT,即多路复用中阻塞等待IO,因为select有内置的时间限制,以此代替直接阻塞在read或write上的调用。
(3)使用较新的SO_RCVTIMEO和SO_SNDTIMEO套接字选项,这个方法的问题在于并非所有的实现都支持这两个套接字选项。
一、首先看一个如何调用Alarm,它主要是通过对慢系统调用产生中断信号来完成。
int connect_timeo(int sockfd, const SA *saptr, socklen_t salen, int nsec){
Sigfunc *sigfunc;
int n;
sigfunc = Signal(SIGALRM, connect_alarm);
if(alarm(nsec) != 0)
err_msg("Alarm has already been set!");
if( ( n = connect(sockfd, saptr, salen)) < 0){ //如果超过nsec秒数,会被中断产生EINTR
close(sockfd); //关闭此套接字,防止三次握手继续进行
if(errno == EINTR) //慢系统调用产生中断信号
errno = ETIMEDOUT;
}
alarm(0);
Signal(SIGALRM, sigfunc);
return n;
}
static void connect_alarm(int signo){
return;
}
二、再来看一Select多路复用是如何做到这一点的,它主要是通过第五个参数来设置所阻塞的描述符超时情况。
int readable_timeo(int fd, int sec){
fd_set rset;
struct timeval tv; //主要是设置此结构体,之后将该结构体以参数形式传给Select函数
FD_ZERO(&rset);
FD_SET(fd, &rset);
tv.tv_sec = sec;
tv.tv_usec = 0;
return (select(fd+1, &rset, NULL, NULL, &tv));
}
这样,在每次调用read之前就可以先调用此函数来设置超时秒数。如下:
int readable_timeo(int fd, int sec){
fd_set rset;
struct timeval tv; //主要是设置此结构体,之后将该结构体以参数形式传给Select函数
FD_ZERO(&rset);
FD_SET(fd, &rset);
tv.tv_sec = sec;
tv.tv_usec = 0;
return (select(fd+1, &rset, NULL, NULL, &tv));
}
三、使用SO_RCVTIMEO套接字选项为recvfrom设置超时
本选项一旦设置到某个描述符上,其超时设置将应用于该描述符上的所有读操作,本方法的优势在于一次设置,一直生效,而前面两种方法需要在每个操作发生之前设置。另外,本套接字选项仅用于读操作,类似SO_SNDTIMEO仅用于写操作,但两者均不能用于为connect设置超时,从名字就可以看出,一个是接收,一个是发送。
struct timeval tv;
tv.tv_sec = 5;
tv.tv_usec = 0;
Setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)); //设置超时,通过设定套接字选项
while(fgets(sendline, MAXLINE, stdin) != NULL){
sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);
n = recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL);
if( n < 0){
if(errno == EWOULDBLOCK){ //这里有点类似于非阻塞套接字,即无数据可读的返回值
fprintf(stderr, "Socket read timeout!");
continue;
}
else
err_sys("recvfrom error!");
}
recvline[n] = 0;
fputs(recvline, stdout);
}