LINUX下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等。