在 c/c++ 中通过 tcp 套接字发送时数据被切断

Question

当我通过 tcp 套接字发送长消息时，它们被截断了。它因目的地而异。在我的机器上本地发送和接收时，一切都会通过。在我的服务器上本地发送和接收时，它始终在第 21846 个字节后被切断。从我的机器发送到服务器时，它始终在第 1441 个字节处被切断。服务器在斯德哥尔摩，我在英国。当客户端在 Windows 上并使用 Windows' 网络代码时，也会出现同样的问题。

这里客户端应该发送 29 999 个零和一个空终止符，接收它们并打印它们。当我用 wc 计算它们时，我得到了我收到的实际字节数。所以这些数字代表了一种双向传输，但从测试中我可以说这个问题在单向传输中具有相同的属性。

据我所知，read/write 函数正在阻塞，所以问题不在于数据在函数退出之前未完全到达 - 手册页中描述的问题。

我该如何解决这个问题？

往下看解决方法

这是我用来测试的代码：

server/main.cpp

#include <filesystem>
#include <fstream>
#include <iomanip>
#include <iostream>
#include <sstream>
#include <string>

#include "aes/aes.hpp"
#include "network.hpp"
#include <thread>

using namespace std;

int main()
{
    int server_socket = InitServerSocket(26969);
    std::cout << "Listening..." << std::endl;

    while (true) {
        // accept incomming connections - blocking
        int client_socket = accept(server_socket, NULL, NULL);

        if (client_socket < 0) {
            std::cerr << "Unable to accept";
            close(server_socket);
            return 1;
        }

        char long_text[30000];
        read(client_socket, long_text, 30000);

        std::cout << long_text << std::endl;

        write(client_socket, long_text, 30000);

        close(client_socket);
    }

    return 0;
}

InitServerSocket():

int InitServerSocket(int port)
{
    struct sockaddr_in server_address;
    server_address.sin_family = AF_INET;
    server_address.sin_port = htons(port);
    server_address.sin_addr.s_addr = htonl(INADDR_ANY);

    int server_socket;
    server_socket = socket(AF_INET, SOCK_STREAM, 0);
    if (server_socket < 0) {
        perror("Unable to create socket");
        _exit(1);
    }

    int result = bind(
        server_socket,
        (struct sockaddr*)&server_address,
        sizeof(server_address));
    if (result < 0) {
        perror("Unable to bind");
        _exit(1);
    }

    if (listen(server_socket, 1000) < 0) {
        perror("Unable to listen");
        _exit(1);
    }

    return server_socket;
}

client/main.cpp

#include <arpa/inet.h>
#include <iostream>
#include <netdb.h>
#include <string.h>
#include <string>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

#include "network.hpp"

using namespace std;

int main()
{
    int socket = ConnectToHost("70.34.195.74", 26969);

    char long_text[30000];
    for (int i = 0; i < 30000; i++)
        long_text[i] = '0';
    long_text[29999] = '[=13=]';

    write(socket, long_text, 30000);

    read(socket, long_text, 30000);

    std::cout << long_text << std::endl;

    CloseConnection(socket);

    return 0;
}

ConnectToHost():

int ConnectToHost(char* IPAddress, int PortNo)
{
    // create a socket
    int network_socket; // socket descriptor ~= pointer ~= fd
    network_socket = socket(AF_INET, SOCK_STREAM, 0);
    // specify a destination address
    struct sockaddr_in server_address;
    server_address.sin_family = AF_INET; // specify protocol
    server_address.sin_port = htons(PortNo); // specify port
    // server_address.sin_addr.s_addr = a.s_addr; // specify resolved ip
    server_address.sin_addr.s_addr = inet_addr(IPAddress);

    // connect
    int connection_status = connect(
        network_socket,
        (struct sockaddr*)&server_address,
        sizeof(server_address));
    if (connection_status == -1)
        std::cout << "Failed to conect to remote host\n";
    return network_socket;
}

解决方案：

这是我为解决问题而编写的包装函数：

int Send(int soc, char* buffer, int size)
{
    int ret = -1;
    int index = 0;

    do {
        ret = write(soc, buffer + index, size - index);
        index += ret;
    } while (ret > 0);

    if (ret == -1)
        return -1;
    else
        return 1;
};

int Recv(int soc, char* buffer, int size)
{
    int ret = -1;
    int index = 0;

    do {
        ret = read(soc, buffer + index, size - index);
        index += ret;
    } while (ret > 0);

    if (ret == -1)
        return -1;
    else
        return 1;
};

Answer 1

write(client_socket, long_text, 30000);

您无法保证所有 30000 个字节都已写入，即使使用阻塞套接字也是如此。您必须检查 write() 的 return 值以确定实际写入了多少字节，然后实现所需的逻辑以重试，剩下的要写入，并以这种方式进行直到全部 30000字节被写入套接字。这就是套接字始终工作的方式。

read(socket, long_text, 30000);

同样如此，您必须检查 returned 值。如果套接字只有一百个字节的未读数据在等待，您将得到这 100 个字节，而 read() 将 return 100 个。如果阻塞套接字中没有未读数据，read() 将阻塞。如果套接字最终收到一个 1 字节的数据包，read() returns 1，这告诉你只读取了 1 个字节。

How can I fix this?

您必须始终检查每 read() 和 write() return 的内容，并相应地进行。如果您需要从套接字读取或写入更多数据，请再试一次。

so the problem is not that the data has not arrived fully before the functions exit - a problem described in the man page.

手册页还描述了 read() 和 write() 中的 returned 值的含义：负值表示错误，正值表示实际读取了多少字节或书面。只有读取和写入常规文件才能保证读取或写入请求的字节数（除非读取到达文件末尾）。

在 c/c++ 中通过 tcp 套接字发送时数据被切断

Data gets cut off when send through a tcp socket in c/c++

c++

linux

windows

tcp

解决方案：