如何在套接字IO操作上设置超时机制

主要有三种方案：
（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
= 0;
fputs(recvline, stdout);
}