LINUX下select设置超时

最新推荐文章于 2024-06-29 22:25:40 发布

转载最新推荐文章于 2024-06-29 22:25:40 发布 · 5.1k 阅读

linux系统编程专栏收录该内容

69 篇文章

订阅专栏

本文深入探讨了Linux环境下通过非阻塞connect和select函数设置连接超时的方法，包括创建socket、设置非阻塞模式、调用connect进行连接尝试、使用select检查连接状态，以及超时处理策略。提供了从套接字创建到连接成功或超时的完整流程，适合网络编程和系统管理员使用。

LINUX设置连接超时方法：

非阻塞 connect：

在一个 TCP 套接字被设置为非阻塞之后调用 connect ，connect 会立即返回 EINPROGRESS 错误，表示连接操作正在进行中，但是仍未完成，与此同时 TCP 三次握手操作会同时进行。在这之后，我们可以通过调用 select 来检查这个链接是否建立成功。

在阻塞套接字的一般情况下，connect ()直到客户端对SYN消息的ACK消息到达之前才会返回。使connect()调用具有超时机制的一个方法是让套接字成为非阻塞的套接字体，然后用select()来等待它完成。
s = socket(AF_INET, SOCK_STREAM, 0);
//下面获取套接字的标志
if ((flags = fcntl(s, F_GETFL, 0)) < 0) {
//错误处理
}

//下面设置套接字为非阻塞
if (fcntl(s, F_SETFL, flags | O_NONBLOCK) < 0) {
//错误处理
}

if ((retcode = connect(s, (struct sockaddr*)&peer, sizeof(peer)) &&
    errno != EINPROGRESS) {
   //因为套接字设为NONBLOCK，通常情况下，连接在connect()返回
   //之前是不会建立的，因此它会返回EINPROGRESS错误，如果返回
   //任何其他错误，则要进行错误处理
}

if (0 == retcode) {  //如果connect()返回0则连接已建立
    //下面恢复套接字阻塞状态
    if (fcntl(s, F_SETFL, flags) < 0) {
        //错误处理
    }

    //下面是连接成功后要执行的代码

    exit(0)
}

（要设备send/recv超时只需从此处开始修改相应值，前面不用）
FD_ZERO(&rdevents);
FD_SET(s, &rdevents);  //把先前的套接字加到读集合里面
wrevents = rdevents;   //写集合
exevents = rdevents;   //异常集合

tv.tv_sec = 5; //设置时间为5秒
tv_tv_usec = 0;

retcode = select(s+1, &rdevents, &wrevents, &exevents, &tv);
if (retcode < 0) {  //select返回错误???
    //错误处理
}
else if (0 == retcode) {  //select 超时???
    //超时处理
}
esle {
    //套接字已经准备好
    if (!FD_ISSET(s, &rdevents) && !FD_ISSET(s, &wrevents)) {
        //connect()失败，进行错处理
    }

    if (getsockopt(s, SOL_SOCKET, SO_ERROR, &err, &len) < 0) {
        //getsockopt()失败，进行错处理
    }

    if (err != 0) {
        //connect()失败，进行错处理
    }
（send/recv超时到此为止，返回send()/recv()函数）

    //到这里说明connect()正确返回
    //下面恢复套接字阻塞状态
    if (fcntl(s, F_SETFL, flags) < 0) {
        //错误处理
    }

//下面是连接成功后要执行的代码
exit(0)

处理非阻塞 connect 的步骤：
第一步，创建 socket，返回套接字描述符；
第二步，调用 fcntl 或 ioctlsocket 把套接口描述符设置成非阻塞；
第三步，调用 connect 开始建立连接；
第四步，判断连接是否成功建立：
A)如果 connect 返回 0 ，表示连接成功(服务器和客户端在同一台机器上时就有可能发生这种情况)；
B)调用 select 来判定连接建立的是否成功；
如果 select 返回 0 ，则表示在 select 的超时时间内未能成功建立连接；我们需要返回超时错误给用户，同时关闭连接，以防止TCP三次握手继续进行下去；
如果 select 返回大于 0 的值，则说明检测到可读或可写或异常的套接字描述符存在；此时我们可以通过调用 getsockopt 来检测集合中的套接口上是否存在待处理的错误，如果连接建立是成功的，则通过 getsockopt(sockfd,SOL_SOCKET,SO_ERROR,(char *)&error,&len) 获取的 error 值将是 0 ，如果建立连接时遇到错误，则 error 的值是连接错误所对应的 errno 值，比如ECONNREFUSED，ETIMEDOUT等。