8

我在 Visual Studio Pro C++ 中创建了一个套接字 (Winsock2) 来侦听端口上的连接 (TCP)。它工作得很好,但我让它在自己的线程中运行,我希望能够关闭它,希望以后重新启动它。我可以毫无问题地终止线程,但是这样做并不会阻止套接字接受新客户端(也就是说,它会在我关闭线程之前一直在接受它)。我可以将新客户连接到它,但没有任何反应......它只是接受,就是这样。我想要的是阻止它监听和接受,然后能够告诉它稍后在同一个端口上重新启动。现在尝试重新启动它只会告诉我该端口已被占用。

这是监听线程函数:

DWORD WINAPI ListeningThread(void* parameter){
TCPServer *server = (TCPServer*)parameter;

try{
    server = new TCPServer(listen_port);
}catch(char* err){
    cout<<"ERROR: "<<err<<endl;
    return -1;
}

int result = server->start_listening();
if(result < 0){
    cout<<"ERROR: WSA Err # "<<WSAGetLastError()<<endl;
    return result;
}
cout<<"LISTENING: "<<result<<endl<<endl;
while(true){
    TCPClientProtocol *cl= new TCPClientProtocol(server->waitAndAccept());
    HANDLE clientThread = CreateThread(0, 0, AcceptThread, cl, 0, 0);
    cout<<"Connection spawned."<<endl;
}

return 0;
}

TCPServer中的相关函数如下:

TCPServer::TCPServer(int port){
listening = false;
is_bound = false;

//setup WSA
int result = WSAStartup(MAKEWORD(2, 2), (LPWSADATA) &wsaData);
if(result < 0){
    throw "WSAStartup ERROR.";
    return;
}

//create the socket
result = (serverSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP));
if(result < 0){
    throw "Socket Connect ERROR.";
    return;
}

//bind socket to address/port
SOCKADDR_IN sin;
sin.sin_family = PF_INET;
sin.sin_port = htons(port);
sin.sin_addr.s_addr = INADDR_ANY;

result = bind(serverSocket, (LPSOCKADDR) &sin, sizeof(sin));
if(result < 0){
    throw "Could not Bind socket - Make sure your selected PORT is available.";
    return;
}

is_bound = true;
}

int TCPServer::start_listening(){
int result = -1;
if(is_bound){
    //SOMAXCONN parameter (max) is a backlog:
    //  how many connections can be queued at any time.
    result = listen(serverSocket, SOMAXCONN);
    if(result >= 0)
        listening = true;
}
return result;
}

SOCKET TCPServer::waitAndAccept(){
if(listening)
    return accept(serverSocket, NULL, NULL);
else
    return NULL;
}

我已经尝试过 closesocket() 和 shutdown() 但这两个都抛出了错误。

谢谢大家的时间和帮助!

4

1 回答 1

10

首先,确保您SO_REUSEADDR在服务器套接字上设置了能够重新开始监听的选项。

然后,我猜,你的问题是accept()阻塞,你不能在需要的时候停止它,所以你正在杀死线程。糟糕的解决方案。这里的正确答案是异步 I/O,即select()poll()它们的 Windows 对应物。查看高级 Winsock 示例

多线程应用程序中一个快速而简单的解决方案是is_it_time_to_stop_accepting_connections在 each 之前检查一些标志accept(),然后在需要停止时,翻转标志并连接到侦听端口(是的,在同一个程序中)。这将解除阻塞accept()并允许您做适当的事情closesocket()或其他任何事情。

但说真的,请阅读异步 I/O

于 2011-08-07T04:16:03.810 回答