C++实现UDP多线程文件传输

下面是一个简单的C++实现UDP多线程文件传输的示例代码：

// UDP多线程文件传输服务器端代码
#include <iostream>
#include <fstream>
#include <string>
#include <thread>
#include <mutex>
#include <chrono>
#include <WinSock2.h>

#pragma comment(lib, "Ws2_32.lib")

using namespace std;

// 接收数据线程函数
void recv_data_thread(SOCKET sock, sockaddr_in& client_addr, mutex& mtx, string& file_name)
{
    char recv_buf[1024];
    int recv_len = 0;

    // 接收文件名
    recv_len = recvfrom(sock, recv_buf, sizeof(recv_buf), 0, (sockaddr*)&client_addr, NULL);
    if (recv_len <= 0)
    {
        cerr << "Error: Failed to receive file name!" << endl;
        return;
    }

    mtx.lock();
    file_name = recv_buf;
    mtx.unlock();

    // 接收文件内容
    ofstream outfile(file_name, ios::out | ios::binary);
    if (!outfile)
    {
        cerr << "Error: Failed to open file " << file_name << " for write!" << endl;
        return;
    }

    while (true)
    {
        recv_len = recvfrom(sock, recv_buf, sizeof(recv_buf), 0, (sockaddr*)&client_addr, NULL);
        if (recv_len <= 0)
        {
            break;
        }

        outfile.write(recv_buf, recv_len);
    }

    outfile.close();
}

int main()
{
    // 初始化WinSock
    WSADATA wsa_data;
    if (WSAStartup(MAKEWORD(2, 2), &wsa_data) != 0)
    {
        cerr << "Error: Failed to initialize WinSock library!" << endl;
        return -1;
    }

    // 创建UDP socket
    SOCKET udp_sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
    if (udp_sock == INVALID_SOCKET)
    {
        cerr << "Error: Failed to create UDP socket!" << endl;
        WSACleanup();
        return -1;
    }

    // 绑定socket
    sockaddr_in server_addr;
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_port = htons(8000);

    if (bind(udp_sock, (sockaddr*)&server_addr, sizeof(server_addr)) == SOCKET_ERROR)
    {
        cerr << "Error: Failed to bind UDP socket!" << endl;
        closesocket(udp_sock);
        WSACleanup();
        return -1;
    }

    cout << "UDP server started..." << endl;

    while (true)
    {
        // 接收客户端请求
        sockaddr_in client_addr;
        memset(&client_addr, 0, sizeof(client_addr));
        int addr_len = sizeof(client_addr);
        mutex mtx;
        string file_name;

        thread recv_thread(recv_data_thread, udp_sock, ref(client_addr), ref(mtx), ref(file_name));
        recv_thread.detach();

        // 休眠一段时间，等待接收线程处理完成
        this_thread::sleep_for(chrono::milliseconds(1000));

        if (!file_name.empty())
        {
            cout << "File " << file_name << " received!" << endl;
        }
    }

    // 关闭socket
    closesocket(udp_sock);

    // 清理WinSock
    WSACleanup();

    return 0;
}

在这个示例中，首先使用WSAStartup()函数初始化WinSock库，然后使用socket()函数创建一个UDP socket，并使用bind()函数将socket绑定到本地IP地址和端口号上。在一个无限循环中，使用recv_data_thread()函数作为接收数据的线程函数，使用thread()函数创建一个新的线程来处理客户端请求。每当接收到一个客户端请求时，就创建一个新的线程来接收客户端发送的文件内容。接收文件名和文件内容的过程在接收数据线程函数中实现，使用mutex来保证线程安全。在文件内容接收完成后，将文件名输出到控制台上。

客户端代码可以使用类似的方式实现。需要注意的是，在发送文件内容时，需要将文件内容分成多个数据包进行发送，并在每个数据包中包含文件名和文件偏移量等信息，以便在服务器端将多个数据包组装成完整的文件。