recv 函数在使用两次时失败

Question

大家好，我一直在从一个简单的客户端接收两个地址然后使用异步套接字和 WINAPI 将它们打印出来的服务器上工作，到目前为止我已经设置了窗口并且已经能够接受连接但是当我尝试向服务器发送两条消息时，它只收到一条消息，第二条失败。代码如下：

LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
    HWND hEdit = NULL;
    int len = sizeof(Server);
    switch (msg)
    {
    case WM_CREATE:


        hEdit = CreateWindowEx(WS_EX_CLIENTEDGE, "EDIT", "",
            WS_CHILD | WS_VISIBLE | WS_VSCROLL | WS_HSCROLL | ES_MULTILINE | ES_AUTOVSCROLL | ES_AUTOHSCROLL,
            0, 0, WIDTH, HEIGHT, hwnd, (HMENU)IDC_MAIN_EDIT, GetModuleHandle(NULL), NULL);
        break;
    case WM_DESTROY:
        PostQuitMessage(0);

        break;
    case 1111:

        if (LOWORD(lParam) == FD_ACCEPT)
        {

            socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
            socket = accept(sListen, (SOCKADDR*)&Server, &len);


            print_line(hwnd, "IT WAS ACCEPTED!!!!!!!!\r\n");
        }
        if (LOWORD(lParam) == FD_CLOSE)
        {
            print_line(hwnd, "Client left the server!\r\n");

        }
        if (LOWORD(lParam) == FD_READ)
        {
            char NICK[4096] = { 0 };
            char IP[4096] = { 0 };
            ZeroMemory(NICK, strlen(NICK));
            ZeroMemory(IP, strlen(IP));
            if (recv(socket, IP, sizeof(IP), NULL) == INVALID_SOCKET)//get the IP address
            {
                print_line(hwnd, "Failed to recieve IP Address from socket!");
                print_line(hwnd, "\r\n");
            }
            if (recv(socket, NICK, sizeof(NICK), NULL) == INVALID_SOCKET)//get the Nickname
            {
                print_line(hwnd, "Failed to recieve nickname from socket!");
                print_line(hwnd, "\r\n");
            }
            //prints the Username and IP address to the window screen 
            print_line(hwnd, "Username: ");
            print_line(hwnd, NICK);
            print_line(hwnd, "\r\n");
            print_line(hwnd, "IP Address: ");
            print_line(hwnd, IP);
            print_line(hwnd, "\r\n");

        }
        break;
    default:



        HWND hEdit;
        RECT rcClient;

        GetClientRect(hwnd, &rcClient);

        hEdit = GetDlgItem(hwnd, IDC_MAIN_EDIT);
        SetWindowPos(hEdit, NULL, 0, 0, rcClient.right, rcClient.bottom, SWP_NOZORDER);


        return (DefWindowProc(hwnd, msg, wParam, lParam));

    }
}

Answer 1

大概您正在使用WSAAsyncSelect()，但您没有显示创建侦听套接字或为其注册消息处理程序的代码。

您不应在代码中使用幻数。 1111 是WM_USER+87，因此您应该将其分配给一个常量以便于阅读，例如：const UINT WM_SOCKETMSG = WM_USER + 87;，然后在您的case 语句中使用该名称，例如：case WM_SOCKETMSG:。

您的套接字消息处理程序在调用accept() 之前正在调用socket()。 accept() 分配并返回一个新的套接字。因此，每次收到FD_ACCEPT 通知时，您都会泄漏一个套接字。如果多个客户端碰巧连接，您将失去对旧套接字的跟踪，因为您使用单个变量来跟踪所有套接字。

您没有考虑到 TCP 是字节流，recv() 返回的字节数可能比您要求的要少。它返回当前可用的任何数据，不超过您给它的缓冲区的大小。您正在使用异步套接字，但您编写阅读代码时就像使用同步套接字一样（即使那样，您显示的逻辑有时也会失败）。如果当前没有可用数据，recv() 将失败并出现 WSAEWOULDBLOCK 错误，您没有处理该错误。每当有新数据到达时，您需要将其读入滚动缓冲区，然后根据需要仅从该缓冲区中提取已完成的数据，将不完整的数据留在缓冲区中，以便后续读取完成。

你需要设计一个协议来控制你的数据流，你不能随便乱扔数据。我严重怀疑您想要 4KB 的用户名并在 IP 地址上浪费约 3KB。您需要对正在传输的值进行定界，不仅可以减少带宽使用，还可以减少保存它们所需的内存。对于像您所展示的那样简单的东西，您可以用 LF 字符分隔这些值。然后，您可以在处理滚动缓冲区时查找该字符。

最后，您根本没有处理错误。你需要这样做。

尝试类似的方法：

#include <map>
#include <string>

std::map<SOCKET, std::string> ClientBuffers;
typedef std::map<SOCKET, std::string>::iterator BufferIterator; 

const UINT WM_SOCKETMSG = WM_USER + 87;

LRESULT CALLBACK WndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
    ...
    switch (msg)
    {
        ...
        case WM_SOCKETMSG:
        {
            SOCKET socket = (SOCKET) wParam;
            int event = WSAGETSELECTEVENT(lParam);
            int error = WSAGETSELECTERROR(lParam);

            switch (event)
            {
                case FD_ACCEPT:
                {
                    if (error == 0)
                    {
                        sockaddr_in clientaddr = {0};
                        int len = sizeof(clientaddr);

                        if (accept(socket, (SOCKADDR*)&clientaddr, &len) != INVALID_SOCKET)
                        {
                            print_line(hwnd, "A client connected to the server!\r\n");
                            break;
                        }

                        error = WSAGetLastError();
                    }

                    print_line(hwnd, "Error accepting a client!\r\n");
                    // handle the error on the reported socket as needed...

                    break;
                }

                case FD_CLOSE:
                {
                    if (error == 0)
                        print_line(hwnd, "A client left the server!\r\n");
                    else
                        print_line(hwnd, "A client connection was lost unexpectedly!\r\n");

                    BufferIterator i = ClientBuffers.find(socket);
                    if (i != ClientBuffers.end())
                        ClientBuffers.erase(i);

                    break;
                }

                case FD_READ:
                {
                    char buf[1024];

                    int numRead = recv(socket, buf, sizeof(buf), NULL);
                    if (numRead == SOCKET_ERROR)
                    {
                        if (WSAGetLastError() != WSAEWOULDBLOCK)
                        {
                            print_line(hwnd, "Failed to read from a client!\r\n");
                            // handle the error on the reported socket as needed...
                        }
                        break;
                    }

                    if (numRead == 0)
                        break;

                    std::string &buffer = ClientBuffers[socket];
                    buffer += std::string(buf, numRead);

                    std::string::size_type idx1 = buffer.find('\n');
                    if (idx1 == std::string::npos)
                        break; // wait for more data

                    std::string::size_type idx2 = buffer.find('\n', idx1+1);
                    if (idx2 == std::string::npos)
                        break; // wait for more data

                    std::string IP = buffer.substr(0, idx1);
                    std::string NICK = buffer.substr(idx1+1, idx2-idx1-1);
                    buffer.erase(0, idx2+1);

                    //prints the Username and IP address to the window screen 
                    print_line(hwnd, "Username: ");
                    print_line(hwnd, NICK.c_str());
                    print_line(hwnd, "\r\n");
                    print_line(hwnd, "IP Address: ");
                    print_line(hwnd, IP.c_str());
                    print_line(hwnd, "\r\n");

                    break;
                }
            }

            break;
        }
    }
    ...
}