connect 非阻塞 超時機制 轉

阻塞套接字的一般情況下,connect ()直到客戶端對SYN消息的ACK消息到達之前纔會返回。使connect()調用具有超時機制的一個方法是讓套接字成爲非阻塞的套接字體,然後用select()來等待它完成。
[code:1:7901c37cf2]
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)
}

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()失敗,進行錯處理
    }

    //到這裏說明connect()正確返回
    //下面恢復套接字阻塞狀態
    if (fcntl(s, F_SETFL, flags) < 0) {
        //錯誤處理
    }

    //下面是連接成功後要執行的代碼
    
    exit(0)
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章