GetQueuedCompletionStatus过早退出(或者我认为)

时间:2016-06-19 01:06:43

标签: c++ sockets winsock2

首先让我概述一下。我通过三个端口收到数据。我有一个套接字,一个完成端口和一个工作线程。我调用WSARecv,工作线程进程调用GetQueuedCompletionStatus,然后调用我的解析例程ReadMsgs。当调用ReadMsgs并且在ReadMsgs处理缓冲区时更新缓冲区时,有时会发生缓冲区未更改的情况。 GetQueuedCompletionStatus返回的处理字节数在发生更新时是​​正确的。

有谁知道为什么会出现这种情况以及我做错了什么。让我告诉你看起来最相关的代码。如果您需要查看更多代码,请具体说明。我的基本套接字类看起来像这样(我省略了似乎与我无关的细节。我也省略了所有错误检查。)

class Socket_Base : public OVERLAPPED
{
public:
Socket_Base()
{
    // Initialize base OVERLAPPED object
    Internal = 0;
    InternalHigh = 0;
    Offset = 0;
    OffsetHigh = 0;
    hEvent = WSACreateEvent();

    // Initialize addr structure
    ZeroMemory( &addr, sizeof(struct sockaddr_in));

    // Create the completion port
    hCP = CreateIoCompletionPort( INVALID_HANDLE_VALUE, NULL, 0, 1);

    // Create the worker thread and bind it to the callback function and the completion port
    hThread = (HANDLE)_beginthreadex( NULL, 0, Callback_Socket, hCP, 0, NULL);

    // Create the socket
    Sock = WSASocket( AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);

    // Bind the socket to the completion port
    CreateIoCompletionPort( (HANDLE)Sock, hCP, 0, 0);
}
void Connect() { WSAConnect( Sock, (SOCKADDR*)(&addr), sizeof(addr), NULL, NULL, NULL, NULL);}
void StartRecv()
{
    DWORD Flags = 0;
    DWORD numBytes = 0;
    if (WSARecv( Sock, &wsaBuf, 1, &numBytes, &Flags, (OVERLAPPED*)this, NULL) == 0) ReadMsgs( numBytes);
}
int ReadMsgs( int NumBytes);

protected:
    virtual ~Socket_Base() {}
    virtual void ProcessMsg() = 0;
    struct sockaddr_in addr;
    SOCKET Sock;
    HANDLE hCP;
    WSABUF wsaBuf;
    HANDLE hThread;
    char *readBuf;
    int bufsize;
};

每个端口都有自己的派生套接字类,由端口号和虚拟ProcessMsg函数(在解析每条消息时由ReadMsgs调用)区分。这是一个这样的课程:

class Socket_Admin : public Socket_Base
{
public:
static const int bufcap = 1024;
Socket_Admin::Socket_Admin() : Socket_Base()
{
    // Buffer
    readBuf = new char[ bufcap];

    // The socket
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    addr.sin_port = htons(9300);
    wsaBuf.buf = readBuf;
    wsaBuf.len = bufcap;
}
~Socket_Admin();
void ProcessMsg();
};

工作线程进程

unsigned int Callback_Socket( void *lpParameter)
{
    HANDLE hCP = (HANDLE)lpParameter;
    DWORD NumBytes = 0;
    ULONGLONG CompletionKey;
    WSAOVERLAPPED *pOverlapped;
    while (GetQueuedCompletionStatus( hCP, &NumBytes, &CompletionKey, &pOverlapped, INFINITE) && CompletionKey == 0)
    {
        Socket_Base *pTCP = (Socket_Base*)pOverlapped;
        if (NumBytes > 0) pTCP->ReadMsgs( NumBytes);
        NumBytes = 0;
    }
    return 0;
}

还有一件事我应该解释一下。 ReadMsgs进行了解析。服务器使用最终换行符分隔消息,并使用逗号分隔消息中的字段。 ReadMsgs在找到它们时用空字符替换逗号和换行符,并记下每个字段在指向缓冲区中位置的单独指针数组中的位置。现在,当ReadMsgs到达最后填充的缓冲区的末尾时,它有时会发现一条不完整的消息。这将被复制到缓冲区的开头,期望下一次读取完成消息,并相应地修改wsaBuf。因此,ReadMsgs的结尾如下所示:

wsaBuf.buf = pchar;
wsaBuf.len = remsize;
StartRecv();

其中pchar指向超出部分消息的字符,remsize是剩余缓冲区的大小。

我知道从详细的服务器日志发送到我的应用程序的消息。用空字符替换分隔符也可以很容易地看到缓冲区的哪个部分已被处理。通过将缓冲区保存到文件并检查它,我可以告诉它在调用ReadMsgs后更新了。此外,通过上面代码中未显示的日志消息,我知道在这些情况下,工作线程调用了ReadMsgs。它不会每次都发生,但确实会发生。

如果有人能告诉我我的错误是什么,我将不胜感激。

1 个答案:

答案 0 :(得分:0)

我可能有一个答案。我在StartRecv和Callback_Socket中都调用了ReadMsgs。在StartRecv中,当WSARecv返回0时,我调用ReadMsgs,表明传输已由WSARecv完成。我认为如果在WSARecv中完成传输,则不会涉及GetQueuedCompletion状态。但是,如果GetQueuedCompletionStatus确实返回以响应完成的读取,那么我将重复调用ReadMsgs,这将解释我收集的记录数据,以及我之前的假设。

我在StartRecv中删除了对ReadMsgs的调用。代码现在正常运行。

感谢那位给我负面投票的绅士。这告诉我,我之前没有观察到我所描述的行为,因此极不可能。这让我想到了一个新的方向。有时只是一个经验丰富的面包车的咕噜声说话。