如何设置socket函数的非阻塞调用?

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。

猜你喜欢

转载自chanshui.iteye.com/blog/799177