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

windows的socket在创建后,默认是阻塞调用的,也就是说函数recv,recvfrom,send,sendto等函数都是阻塞的;那么我们如何将他们设置成非阻塞调用呢?我们可以通过windows为我们提供的ioctlsocket 函数实现;

 

CSocket类中的connect等函数都是默认为阻塞方式的,也就是说,它不达到目的就不走,在程序中就会出现像死机一样的状况,很是不爽。在这里给出一种设置connect函数为非阻塞方式的方法,代码很简单,且慢慢来看。
SOCKET sock = socket(AF_INET, SOCK_STREAM, 0);//先创建套接字
if(sock == INVALID_SOCKET)
{
  AfxMessageBox("创建套接字失败!");
  return 0;
}
SOCKADDR_IN addrTo;
addrTo.sin_family=AF_INET;
addrTo.sin_port=htons(4000);
addrTo.sin_addr.S_un.S_addr=inet_addr(ip);
    //--------------------------------------------------------------------------
    //设置为非阻塞方式连接
    unsigned long ul = 1;
    int err;
    int ret = ioctlsocket(sock, FIONBIO, (unsigned long*)&ul);
    if(ret == SOCKET_ERROR)
    {
        err = WSAGetLastError();
  closesocket(sock);
  sock = NULL;
  return FALSE;
    }

TIMEVAL timeval;
    fd_set r;                     

    FD_ZERO(&r);
    FD_SET(sock, &r);
    timeval.tv_sec = 1;   //秒
    timeval.tv_usec =200;  //毫秒

    //上面的代码是要connect在1.2秒之后返回,不管是否已经连接上,这样就不会阻塞啦,简单吧

    connect(sock, (SOCKADDR*)&addrTo, sizeof(SOCKADDR));

    ret = select(0, 0, &r, 0, &timeval); 
    if ( ret <= 0 )
    {
  char ch[20] = {0};
  sprintf(ch, "和%s连接超时, 取原参数失败", ip);
  MessageBox(ch, "错误", MB_ICONSTOP);
  err = WSAGetLastError();
  closesocket(sock);
  sock = NULL;
  return FALSE;
    }

//设回阻塞模式
    ul = 0 ;
    ret = ioctlsocket(sock, FIONBIO, (unsigned long*)&ul);  //返回之后,记得要设置为阻塞模式哟
    //--------------------------------------------------------------------------

如何设置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。

 

 

 

 

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;

 

使用WSAAsyncSelect,connect马上返回,在界面中响应消息WM_SOCKET消息

你可能感兴趣的:(VC中将SOCKET类的connect函数设置为非堵塞)