Socket中如何設置連接超時

Socket中如何設置連接超時
AntGhazi/2001.12.14 主頁：antghazi.yeah.net

把CSDN與中文yahoo翻了底朝天，也沒找到如何設置socket的連接超時的滿意方法，問此問題的兄弟已有一大堆，這裏偶就講一下win下如何設置socket的connect超時。
設置connect的超時很簡單，CSDN上也有人提到過使用select，但卻沒有一個令人滿意與完整的答案。偶所講的也正是select函數，此函數集成在winsock1.1中，簡單點講，"作用使那些想避免在套接字調用過程中被鎖定的應用程序，採取一種有序的方式，同時對多個套接字進行管理"(《Windows網絡編程技術》原話)。使用方法與解釋請見《Windows網絡編程技術》。
在使用此函數前，需先將socket設置爲非鎖定模式，這樣，在connect時，纔會立馬跳過，同時，通常也會產生一個WSAEWOULDBLOCK錯誤，這個錯誤沒關係。再執行select則是真正的超時。

WSADATA wsd;
SOCKET cClient;
int ret;
struct sockaddr_in server;
hostent *host=NULL;

if(WSAStartup(MAKEWORD(2,0),&wsd)){return 0;}
cClient=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
if(cClient==INVALID_SOCKET){return 0;}
//set Recv and Send time out
int TimeOut=6000; //設置發送超時6秒
if(::setsockopt(cClient,SOL_SOCKET,SO_SNDTIMEO,(char *)&TimeOut,sizeof(TimeOut))==SOCKET_ERROR){
return 0;
}
TimeOut=6000;//設置接收超時6秒
if(::setsockopt(cClient,SOL_SOCKET,SO_RCVTIMEO,(char *)&TimeOut,sizeof(TimeOut))==SOCKET_ERROR){
return 0;
}
//設置非阻塞方式連接
unsigned long ul = 1;
ret = ioctlsocket(cClient, FIONBIO, (unsigned long*)&ul);
if(ret==SOCKET_ERROR)return 0;

//連接
server.sin_family = AF_INET;
server.sin_port = htons(25);
server.sin_addr .s_addr = inet_addr((LPCSTR)pSmtp);
if(server.sin_addr.s_addr == INADDR_NONE){return 0;}

connect(cClient,(const struct sockaddr *)&server,sizeof(server));

//select 模型，即設置超時
struct timeval timeout ;
fd_set r;

FD_ZERO(&r);
FD_SET(cClient, &r);
timeout.tv_sec = 15; //連接超時15秒
timeout.tv_usec =0;
ret = select(0, 0, &r, 0, &timeout);
if ( ret <= 0 )
{
::closesocket(cClient);
return 0;
}
//一般非鎖定模式套接比較難控制，可以根據實際情況考慮 再設回阻塞模式
unsigned long ul1= 0 ;
ret = ioctlsocket(cClient, FIONBIO, (unsigned long*)&ul1);
if(ret==SOCKET_ERROR){
::closesocket (cClient);
return 0;
}

 

--------------------------------------------------------------------------------------------------------------

LINUX下的方法：

 

在阻塞套接字的一般情況下，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)