套接字连接从分离的线程中止

Question

我正在构建一个套接字客户端，我需要在其中实现连接、读取、写入超时以及协议本身的超时（缺少答案等）。

我正在考虑在一个分离线程中使用一个简单的计时器，该计时器将在每个事务上启动，然后在事务完成时取消。同样的方法将用于使用不同超时的协议控制。

为了测试我做了以下简单的代码：

#include <string>
#include <sstream>
#include <map>
#include <iostream>
#include <cstring>
#include <thread>

#ifdef _WIN32
#include <io.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <Windows.h>
#else
#include <unistd.h>
#include <sys/socket.h>
#include <netdb.h>
#include <sys/types.h>
#endif

#include <stdio.h>
#include <stdlib.h>

bool timerOn = false;
int currentSocket = 0;


void Timer(int seconds)
{
    int tick = seconds;

    while (tick > 0)
    {
        std::this_thread::sleep_for(std::chrono::seconds(1));
        tick--;
    }

    if (timerOn)
        close(currentSocket);
}

void StartTimer(int seconds)
{
    timerOn = true;
    std::thread t(&Timer, seconds);
    t.detach();
}

void StopTimer()
{
    timerOn = false;
}


void Connect(std::string address, int port)
{
    struct addrinfo hints;
    struct addrinfo *result = NULL;
    struct addrinfo *rp = NULL;
    int sfd, s;

    std::memset(&hints, 0, sizeof(struct addrinfo));

    hints.ai_family = AF_UNSPEC;        /* Allow IPV4 or IPV6 */
    hints.ai_socktype = SOCK_STREAM;    
    hints.ai_flags = 0;
    hints.ai_protocol = 0;              

    std::string portStr;
    portStr = std::to_string(port);

    s = getaddrinfo(address.c_str(), portStr.c_str(), &hints, &result);

    if (s != 0)
    {
        std::stringstream ss;
        ss << "Cannot resolve hostname " << address << gai_strerror(s);
        throw std::runtime_error(ss.str());
    }

    for (rp = result; rp != NULL; rp = rp->ai_next)
    {
        sfd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol);

        if (sfd == -1)
            continue;

        StartTimer(10);
        int sts = connect(sfd, rp->ai_addr, rp->ai_addrlen);
        StopTimer();

        if (sts == 0)
            break;

        close(sfd);
    }

    freeaddrinfo(result); /* Object no longer needed */

    if (rp == NULL)
    {
        std::stringstream ss;
        ss << "Cannot find server address at " << address << " port " << port;
        throw std::runtime_error(ss.str());
    }

    currentSocket = sfd;
}


int main()
{
    try
    {
        Connect("192.168.0.187", 9090);
        std::cout << "Connected to server. Congrats!!!" << std::endl;
    }
    catch (std::exception& ex)
    {
        std::cout << "Error connecting to server. Aborting." << std::endl;
        std::cout << ex.what() << std::endl;
    }
}

在计时器上关闭套接字不会取消“连接”操作，而是强制它因错误而中止。我也试过shutdown(sfd, SHUT_RDWR);，但没有成功...

我的方法无效吗？为什么它不工作？

如何强制connect 在分离线程出错时中止？

Answer 1

在计时器上关闭套接字不会取消“连接”操作，而是强制它以错误中止。

哇！你绝对不能那样做。当您关闭套接字时，无法知道线程实际上在connect 中被阻塞（而不是即将调用connect）。在一个线程中释放资源，而另一个线程正在使用或可能正在使用它是灾难的根源。

想象一下会发生这种情况：

1. 一个线程即将调用connect，所以它安排了一个超时。

2. 超时，socket关闭。

3. 某个库中的线程出于某种原因创建了一个新套接字以供使用，并获得相同的套接字描述符。

4. 即将调用connect的线程终于被调度并调用connect——连接库的套接字！灾难。

你有两个选择：

1. 使用非阻塞connect 操作。

2. 使用信号之类的东西来中断调用connect的线程。

但我想知道你为什么要打扰。为什么需要中止连接？如果在超时前连接没有成功，你需要做其他事情，那就继续做吧。