c++ - 解除阻止 WSAccept 以阻止 TCP 服务器套接字

Question

我正在编写一个 TCP 服务器(阻塞套接字模型)。 当服务器正在等待(阻塞)接受新的连接尝试(我使用 WSAccept)时，我无法实现有效的正常程序退出。 服务器的监听套接字的代码是这样的(我省略了错误处理和其他不相关的代码):

int ErrCode = WSAStartup(MAKEWORD(2,2), &m_wsaData) ;


// Create a new socket to listen and accept new connection attempts
struct addrinfo hints, *res = NULL, *ptr = NULL ;
int rc, count = 0 ;
memset(&hints, 0, sizeof(hints)) ;

hints.ai_family = AF_UNSPEC ;
hints.ai_socktype = SOCK_STREAM ;
hints.ai_protocol = IPPROTO_TCP ;
hints.ai_flags = AI_PASSIVE ;

CString strPort ;
strPort.Format("%d", Port) ;

getaddrinfo(pLocalIp, strPort.GetBuffer(), &hints, &res) ;

strPort.ReleaseBuffer() ;

ptr = res ;

if ((m_Socket = WSASocket(res->ai_family, res->ai_socktype, res->ai_protocol, NULL, 0, 0)) == INVALID_SOCKET)
{
    // some error   
} 

if(bind(m_Socket, (SOCKADDR *)res->ai_addr, res->ai_addrlen) == SOCKET_ERROR)
{
    // some error
}

if (listen(m_Socket, SOMAXCONN) == SOCKET_ERROR)
{
    // some error
}

到目前为止一切顺利...然后我在这样的线程中实现了 WSAccept 调用:

SOCKADDR_IN ClientAddr ;
int ClientAddrLen = sizeof(ClientAddr) ;

SOCKET TempS = WSAAccept(m_Socket, (SOCKADDR*) &ClientAddr, &ClientAddrLen, NULL, NULL);

当然 WSAccept 会阻塞，直到进行新的连接尝试，但如果我想退出 该程序然后我需要一些方法来使 WSAccept 退出。我尝试了几种不同的方法:

1. 尝试从另一个线程中使用 m_Socket 调用 shutdown 和/或 closesocket 失败(程序只是挂起)。
2. 使用 WSAEventSelect 确实解决了这个问题，但是 WSAccept 只提供非阻塞套接字——这不是我的意图。 (有没有办法让套接字阻塞？)
3. 我阅读了有关 APC 的内容并尝试使用类似 QueueUserAPC(MyAPCProc, m_hThread, 1)) 的方法，但它也没有用。

我做错了什么？ 是否有更好的方法使这个阻塞的 WSAccept 退出？

Answer 1

使用带超时的 select() 来检测客户端连接何时真正挂起，然后调用 WSAAccept() 接受它。它使用阻塞套接字而不将它们置于非阻塞模式。这将使您的代码有更多机会检查应用程序是否正在关闭。