66-套接字超时(alarm)

我们知道,使用 read 或 recvfrom 函数从 socket 上读取数据时,可能会导致阻塞。特别是 UDP 服务器,稍有不甚,数据报就丢失就会直接导致程序假死(阻塞在 recvfrom)。

还有 connect 函数,如果去连接一个网络上不存在的主机,需要等待很长时间,可能长达 2 分钟。如果我们能控制它在指定的时间内连接不上,就直接报错,行不行呢?

很可惜,上面的问题 linux 并未直接提供带超时的函数帮我们解决,所以我们需要一些奇技淫巧来完成上面的功能。

1. 如何设置超时?

常用的有 3 种方法来设置超时:

  • 1) 使用 alarm 函数,一旦到期就产生 SIGALRM 信号,打断正在执行的系统调用。
  • 2) 使用 select 函数。
  • 3) 使用套接字选项 SO_RCVTIMEO 和 SO_SNDTIMEO.

关于这几个方案,有几点要说明:

  • 三个技术都可用于输入和输出(read 和 write)
  • 对于阻塞的 connect 只能使用方案一,对于非阻塞的 connect,只能使用 select.
  • 方案三对 connect 不适用。
  • 方案一和二适用于任何描述符,而方案三仅适用于套接字描述符。

2. 使用 alarm 设置超时

它的伪代码如下,这里以 connect 为例:

void sighandler(int sig) {   // 信号处理函数,什么也不用做   return; }  int connect_timeo(/*...*/, int nsec) {    // 注册信号    registSignal(SIGALRM, sighandler, &oldhandler);    // 开启一个 nsec 秒的 alarm    alarm(nsec);    ret = connect(/*...*/);    // 如果被信号打断,则表示超时    if (ret < 0 && errno == EINTR) {      errno = ETIMEDOUT;    }    // 删除 alarm    alarm(0);    registSignal(SIGALRM, oldhandler);    return ret; }

3. 实验代码

代码托管在 gitos 上,请使用下面的命令获取:

git clone https://git.oschina.net/ivan_allen/unp.git

如果你已经 clone 过这个代码了,请使用 git pull 更新一下。本节程序所使用的程序路径是 unp/program/advcio/conn_timeo/conn_timeo.cc,另一个使用 alarm 为 recvfrom 设置超时的例子是 unp/program/advcio/read_timeo/read_timeo_alarm.

4. 实验结果

ip 地址 192.168.166.4 在网络上并不存在,直接连接它就会导致 connect 函数阻塞,一直在发送 SYN 请求。

  • 不使用超时的 connect


这里写图片描述
图1 不使用超时的 connect

可以看到,默认的 connect 函数超时值约为 127 秒。

  • 使用超时的 connect

通过命令行 --timeo 选项传递超时值给 connec_timeo 函数。


这里写图片描述
图2 使用超时的 connect

5. 总结

  • 掌握使用 alarm 编写超时函数的方法

练习:使用 alarm 编写一个带超时的 recvfrom 函数(答案在read_timeo 文件下可以找到)。

说明:本文转自blog.csdn.net,用于学习交流分享,仅代表原文作者观点。如有疑问,请联系我们删除~