TCP连接connect等待时长控制的另一种方法

原文地址：http://blog.csdn.net/slin000/article/details/3971644

在TCP网络应用开发中，作为客户端的程序经常需要主动连接服务器，这时你就需要建立一个Socket，然后调用connect函数连接到服务器地址。正常情况下，这并没有什么问题，但当服务器主机不存在的时候，connect函数可能会等待一分多钟才能返回。如果在主线程中调用connect函数，就会产生长时间无法响应的状况。

在现代的互联网硬件环境中，一分钟的等待有点太长了，我们需要缩短等待时间。

在Linux环境下，可以用alarm调用定时唤醒正在等待的线程，使connect函数从等待中返回，但在Windows下我没有找到类似的函数。如何让connect函数返回呢？

经过实验，找到一个简单的方法：直接关闭connect函数使用的那个socket套接字，connect函数就会立即返回。这个方法感觉土了点，但确实管用。该方法的工作过程描述如下：
1） 创建socket
2） 启动定时关闭该socket的线程
3） 调用connect函数连接服务器
4） 取消定时关闭线程的工作
5） 检查定时关闭线程的关闭操作是否已经执行
6） 检查connect返回值是否有效

摘录一段示例代码如下：

SOCKET CTCPConnector::ConnectTo(int toIp, int toPort,
                                int localIp , int localPort,
                                int timeOut)

{
    SOCKET Socket = socket(AF_INET, SOCK_STREAM, 0);

    sockaddr_in InetAddr;
    InetAddr.sin_family      = AF_INET;
    InetAddr.sin_addr.s_addr = htonl(localIp);
    InetAddr.sin_port        = htons(localPort);

    if (localIp > 0 && localPort>0)
    {
        if (bind(Socket, (sockaddr *) &InetAddr, sizeof(InetAddr)) < 0)
            return INVALID_SOCKET;
    }

    InetAddr.sin_addr.s_addr = htonl(toIp);
    InetAddr.sin_port        = htons(toPort);

    CTimeOutClose Closer(Socket);  // 这个是超时关闭线程
    if (timeOut > 0)
        Closer.SetTimeOut(timeOut);// 设定超时时长

    int err = connect(Socket, (const sockaddr *)&InetAddr, sizeof(InetAddr));
    if (timeOut >= 0)
    {
        Closer.Cancel();     // 取消超时关闭
        if (Closer.HasDone() // 检查定时关闭线程的关闭操作是否已经执行
            && err >= 0)     //
        {
            err = -1;         
        }
    }

    if (err < 0) // 检查connect返回值是否有效
    {            
        return INVALID_SOCKET;
    }

    return Socket;

}

代码中CTimeOutClose类是启动关闭线程，等待一段时间后关闭指定的套接字。同时，该类还提供接口，用于取消操作和检查操作是否已经执行。

虽然该方法需要启动一个新的线程，但对于大多数的应用来说，主动发起建立TCP连接的量都不会太多，所以对程序的性能并不会产生明显影响。

（以上代码可以在我上传的资源“回城卷轴网络通讯架构源代码”中找到，下载地址为：http://download.csdn.net/source/1023342）

作者：苏林