huyutian

他强由他强,清风拂山岗;他横由他横,明月照大江。他自狠来他自恶,我自一口真气足

  C++博客 :: 首页 :: 新随笔 :: 联系 :: 聚合  :: 管理 ::
  20 随笔 :: 47 文章 :: 22 评论 :: 0 Trackbacks
对于socket编程一直有一种误解:socket的阻塞连接下应该调用send/recv来发送和接收数据,而无阻塞连接下应该用WSASend/WSARecv函数来发送和接收数据。其实socket工作模式与调用哪个收发函数没有任何关系,自己写段简单的代码测试下就知道了。无阻塞模式下,send/recv都是立即返回的。并不会等待发送或接收完成才返回。测试还发现选择模式下无阻塞连接,调用connect函数会立即返回,随后调用select,在连接建立时会设置fdWrite。
SOCKET s = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (s == INVALID_SOCKET)
{
    
return -1;
}


// Set the connection to non-blocking mode
unsigned long ulMode = 1;
if (::ioctlsocket(s, FIONBIO, &ulMode) != 0)
{
    s 
= INVALID_SOCKET;
    
return -1;
}


char szHost[128= "www.myhost.com";
int nPort = 8000;
char szBuffer[1024= "test socket function.\n"

addrinfo aiHints;
addrinfo 
*aiList = NULL;
memset(
&aiHints, 0sizeof(aiHints));
aiHints.ai_family 
= AF_INET;
aiHints.ai_socktype    
= SOCK_STREAM;
aiHints.ai_protocol    
= IPPROTO_TCP;
if (0 != ::getaddrinfo(szHost, NULL, &aiHints, &aiList))
{
    
//m_nLastError = WSAGetLastError();
    ::WSASetLastError(WSAEINVAL);
    
return -1;
}


fd_set fdRead, fdWrite;
FD_ZERO(
&fdRead);
FD_ZERO(
&fdWrite);
FD_SET(s, 
&fdRead);
FD_SET(s, 
&fdWrite);
struct timeval tv;
tv.tv_sec 
= 3;
tv.tv_usec 
= 0;

SOCKADDR_IN hostAddr;
hostAddr 
= *((SOCKADDR_IN*)aiList->ai_addr);
hostAddr.sin_port 
= ::htons(nPort);
int retCode = ::connect(s, (SOCKADDR*&hostAddr, sizeof(hostAddr));
if (retCode == SOCKET_ERROR)
{
    
int errCode = WSAGetLastError();
    
if (errCode != WSAEWOULDBLOCK)
    
{
        ::closesocket(s);
        s 
= INVALID_SOCKET;
        
return -1;
    }

}

retCode 
= ::select(0&fdRead, &fdWrite, NULL, &tv);
if (retCode > 0)
{
    
if (FD_ISSET(s, &fdRead))
    
{
        printf(
"error!\n");
    }

    
if (FD_ISSET(s, &fdWrite))
    
{
        
//connection has succeeded.
        printf("connect success!\n");
        FD_SET(s, 
&fdWrite);
        FD_ZERO(
&fdRead);
        retCode 
= ::select(0&fdRead, &fdWrite, NULL, &tv);
        
if (retCode > 0)
        
{
            
if (FD_ISSET(s, &fdRead))
            
{
                printf(
"error!\n");
            }

            
if (FD_ISSET(s, &fdWrite))
            
{
                retCode 
= send(s, szBuffer, strlen(szBuffer), 0);
                
if (retCode > 0)
                
{
                    retCode 
= recv(s, buf, 10230);
                    
if (retCode == SOCKET_ERROR)
                    
{
                        retCode 
= WSAGetLastError();
                    }

                    
else if (retCode > 0)
                    
{
                        printf(
"recv %d bytes.\n", retCode);
                    }

                }

            }

        }

    }

}

return 0;
很多问题要想深入理解最好的方法就是写一段短的测试程序验证一下,既不要想当然也不要道听途说。
posted on 2011-06-19 22:47 胡雨田 阅读(503) 评论(0)  编辑 收藏 引用

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   博问   Chat2DB   管理