VC中将SOCKET类的connect函数设置为非堵塞

2013-07-25 06:25 ⁄ 工业·编程 ⁄ 共 2423字 ⁄ 字号小中大 ⁄ 暂无评论

windows的socket在创建后，默认是阻塞调用的，也就是说函数recv，recvfrom，send，sendto等函数都是阻塞的；那么我们如何将他们设置成非阻塞调用呢？我们可以通过windows为我们提供的ioctlsocket 函数实现；先给出一个例子：

BOOL LoadSocketSystem(void)
{
WORD wVersionRequested;
SOCKADDR_IN addrSrv;
BOOL bRet = FALSE;
WSADATA wsaData;
SOCKET sockClient;
int err;
int iMode;
wVersionRequested = MAKEWORD( 1, 1 );
err = WSAStartup( wVersionRequested, &wsaData );
if ( err != 0 ) {
return bRet;
}

if ( LOBYTE( wsaData.wVersion ) != 1 ||
        HIBYTE( wsaData.wVersion ) != 1 ) {
   WSACleanup( );
   return bRet;
}

s_socketStatus.m_hsocket = socket(AF_INET,SOCK_STREAM,0);

if( INVALID_SOCKET == s_socketStatus.m_hsocket )
{
   WSACleanup( );
   return bRet;
}
//-------------------------
// Set the socket I/O mode: In this case FIONBIO
// enables or disables the blocking mode for the
// socket based on the numerical value of iMode.
// If iMode = 0, blocking is enabled;
// If iMode != 0, non-blocking mode is enabled.
    iMode = 1;
ioctlsocket(s_socketStatus.m_hsocket,FIONBIO,(u_long FAR*) &iMode);

s_socketStatus.m_isConnected = TRUE;
sockClient = s_socketStatus.m_hsocket;

addrSrv.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
addrSrv.sin_family=AF_INET;
addrSrv.sin_port=htons(6000);
connect(sockClient,(SOCKADDR*)&addrSrv,sizeof(SOCKADDR));

return TRUE;
}

上面红色表示的就是将刚才创建的socket设置成非阻塞操作。

当然我们这样做是最简单的，但是如果对于一个复杂的系统，系统的其他部分可能已经对该socket调用了另两个windows函数--WSAAsyncSelect or WSAEventSelect ，这个个函数都是自动的将socket设置成非阻塞形式，所有在这样的情况下你调用ioctlsocket 想将其设置回阻塞形式是不会成功的；要想成功，你必须将由于调用上面两个函数对socket系统产生的影响clear掉，然后才能成功调用ioctlsocket 函数；如果clear呢？

如下调用WSAEventSelect函数：

rc = WSAEventSelect(s, hEventObject, 0); 如果想详细了解上述几个函数，请参阅MSDN。

int PASCAL FAR connect (SOCKET s, const struct sockaddr FAR *name, int namelen);

(1)如果服务器端程序已经运行, 客户端执行connect()函数,正常没问题,
(2)如果服务端程序没有运行的时候,客户端执行connect()函数,会在此函数停留很长时间,整个程序界面几乎死掉,要3秒左右才能执行完connect()函数并返回结果,
请问如何在(2)的情况下让connect函数立刻返回?

//设置非阻塞方式连接
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;