摘要：
本文介绍在套接字的I/O操作上设置超时的三种方法。
图片可能有点宽，看不到的童鞋可以点击图片查看完整图片。。
1 调用alarm
使用SIGALRM为connect设置超时
设置方法：
- 监听SIGALRM信号，
- 设置sig_alrm处理函数，
- 在阻塞函数前调用alarm函数设置超时时间，
- 正常返回后，重置超时事件为0
void handle_msg(int sockfd) { char sendbuf[BUFSIZE]; char recvbuf[BUFSIZE]; signal(SIGALRM, sig_alrm); //监听SIGALRM信号 while(1) { memset( sendbuf, ‘\0‘, BUFSIZE ); memset( recvbuf, ‘\0‘, BUFSIZE ); printf("%s", "send msg:"); gets(sendbuf); if (strlen(sendbuf) > 0) send(sockfd,sendbuf,strlen(sendbuf),0); if ( !strcmp(sendbuf, "exit")) break; alarm(5); //设置超时事件为5s，同时设置服务器回射前sleep 10秒，以让recv函数超时 if (recv(sockfd,recvbuf,BUFSIZE,0) > 0) { alarm(0); printf("recv back:%s\n\n", recvbuf); } else { if (errno == EINTR) fprintf(stderr, "socket timeout\n"); else fprintf(stderr, "receive error\n"); } } close( sockfd ); return; } static void sig_alrm(int signo) { fprintf(stderr, "recv SIGALRM, return.\n"); return; }
运行截图：
虽然设置了SIGALRM信号处理函数，但是如图所示，本例依然可以等待读取回射信息，因为信号处理函数里只是return。
2 使用select阻塞等待I/O
设置方法：
使用select的内置时间限制，阻塞在select代替recv函数的阻塞。
void handle_msg(int sockfd) { char sendbuf[BUFSIZE]; char recvbuf[BUFSIZE]; while(1) { memset( sendbuf, ‘\0‘, BUFSIZE ); memset( recvbuf, ‘\0‘, BUFSIZE ); printf("%s", "send msg:"); gets(sendbuf); if (strlen(sendbuf) > 0) send(sockfd,sendbuf,strlen(sendbuf),0); if ( !strcmp(sendbuf, "exit")) break; if (readable_timeo(sockfd, 5) == 0) { fprintf(stderr, "socket timeout\n"); } else{ recv(sockfd,recvbuf,BUFSIZE,0); printf("recv back:%s\n\n", recvbuf); } } close( sockfd ); return; } int readable_timeo(int fd, int sec) { fd_set rset; struct timeval tv; FD_ZERO(&rset); FD_SET(fd, &rset); tv.tv_sec = sec; tv.tv_usec = 0; return select(fd+1, &rset, NULL, NULL, &tv); }
运行截图：
由运行截图可以看到，超时警告运行正常。
需要注意一点的是，虽然客户端在超时之后继续发送消息，但是服务器回射的消息（hello world）依然被接收，这导致我们再次发送消息时，从缓冲区中读出了延迟收到的hello world。
这里只是演示超时技术，因此对此并不做进一步处理。
3 使用SO_RCVTIMEO套接字选项
使用SO_RCVTIMEO套接字选项为recv设置超时
设置方法：
- 使用setsockopt函数对套接字进行设置
- 一旦设置了某个描述符，其超时设置将应用于该描述符上的所有读操作
- SO_RCVTIMEO仅用于读操作，SO_SNDTIMEO仅用于写操作，两者都不能用于为connect设置超时
- 如果套接字超时，被阻塞的函数将返回一个EWOULDBLOCK错误
void handle_msg(int sockfd) { char sendbuf[BUFSIZE]; char recvbuf[BUFSIZE + 1]; int n; struct timeval tv; tv.tv_sec = 5; tv.tv_usec = 0; setsockopt( sockfd, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv) ); while(1) { memset( sendbuf, ‘\0‘, BUFSIZE ); memset( recvbuf, ‘\0‘, BUFSIZE ); printf("%s", "send msg:"); gets(sendbuf); if (strlen(sendbuf) > 0) send(sockfd,sendbuf,strlen(sendbuf),0); if ( !strcmp(sendbuf, "exit")) break; if ( (n=recv(sockfd,recvbuf,BUFSIZE,0)) < 0 ) { if (errno == EWOULDBLOCK) { fprintf(stderr, "socket timeout\n"); continue; } else fprintf(stderr, "recv error"); } else{ printf("recv back:%s\n\n", recvbuf); } } close( sockfd ); return; }
运行截图：
可以看到，超时警报成功运行，但依然有上一例中的延迟接收的情况发生，不作处理。
示例源码上传到了github上，地址：https://github.com/zs634134578/UNP/tree/tryTimeout
参考资料：
《UNIX网络编程 卷1：套接字联网API（第3版）》