设置超时的第二个技术是使用 select 函数。它使用了 select 函数的最后一个超时参数，对套接字描述符进行了“预读”。

1. 程序路径

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

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

如果你已经 clone 过这个代码了，请使用 Git pull 更新一下。本节程序所使用的程序路径是 unp/program/advcio/read_timeo/read_timeo_select.

2. 伪代码

用于判断是否可读的函数

// 该函数利用 select 函数判断在 nsec 时间内描述符 fd 是否可读 int readable_timeo(int fd, int nsec) {   int ret;   fd_set rfds;   struct timeval tv;    FD_ZERO(&rfds);   FD_SET(fd, &rfds);    tv.tv_sec = nsec;   tv.tv_usec = 0;    ret = select(fd + 1, &rfds, NULL, NULL, &tv);    return ret; }

带超时的 recvfrom 函数

int recvfrom_timeo(int sockfd, void *buf, size_t len, int flags, struct sockaddr *src_addr, socklen_t *addrlen, int nsec) {   int ret;    ret = readable_timeo(sockfd, nsec);    // select 函数返回 0，表示超时   if (ret == 0) {     errno = ETIMEDOUT;     ret = -1;   }   else {     ret = recvfrom(sockfd, buf, len, flags, src_addr, addrlen);   }    return ret; }

3. 实验结果

read_timeo_select 是一个 udp 客户端，实际上就是前面的 udp/basic 程序，只不过它将 recvfrom 修改成了带超时的 recvfrom.

read_timeo_select 向 mars 主机发送了一个报文后，阻塞在 recvfrom_timeo 函数上，因为 mars 主机上的服务器并未启动，因此 read_timeo_select 永远也不可能收到 mars 主机发回的数据。

经历 5 秒的时候后，recvfrom_timeo 函数超时返回。

图1 带超时的 recvfrom 函数

4. 总结

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