用多线程方式实现单一程序同时支持ipv4和ipv6

服务端代码

#include <iostream>
#include <thread>
#include <vector>
#include <cstring>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>

void handle_ipv4_connection(int clientSocket) {

    std::cout << "Handling IPv4 connection" << std::endl;

    char buffer[1024];
    memset(buffer, 0, sizeof(buffer));
    if (recv(clientSocket, buffer, sizeof(buffer)-1, 0) < 0) {
        std::cerr << "Failed to receive data from client" << std::endl;
        close(clientSocket);
        return;
    }
    std::cout << "Received data from IPv4 client: " << buffer << std::endl;

    if (send(clientSocket, buffer, strlen(buffer), 0) < 0) {
        std::cerr << "Failed to send data to IPv4 client" << std::endl;
        close(clientSocket);
        return;
    }

    close(clientSocket);
}

void handle_ipv6_connection(int clientSocket) {
    std::cout << "Handling IPv6 connection" << std::endl;

    char buffer[1024];
    memset(buffer, 0, sizeof(buffer));
    if (recv(clientSocket, buffer, sizeof(buffer)-1, 0) < 0) {
        std::cerr << "Failed to receive data from client" << std::endl;
        close(clientSocket);
        return;
    }
    std::cout << "Received data from IPv6 client: " << buffer << std::endl;

    if (send(clientSocket, buffer, strlen(buffer), 0) < 0) {
        std::cerr << "Failed to send data to IPv6 client" << std::endl;
        close(clientSocket);
        return;
    }
    close(clientSocket);
}

void ipv4_server() {
    int serverSocket = socket(AF_INET, SOCK_STREAM, 0);

    struct sockaddr_in serverAddress;
    std::memset(&serverAddress, 0, sizeof(serverAddress));
    serverAddress.sin_family = AF_INET;
    serverAddress.sin_port = htons(12345);
    serverAddress.sin_addr.s_addr = htonl(INADDR_ANY);

    bind(serverSocket, (struct sockaddr*)&serverAddress, sizeof(serverAddress));

    listen(serverSocket, 5);
    std::cout << "IPv4 server listening on port 12345" << std::endl;

    while (true) {
        struct sockaddr_in clientAddress;
        socklen_t clientAddressLen = sizeof(clientAddress);
        int clientSocket = accept(serverSocket, (struct sockaddr*)&clientAddress, &clientAddressLen);

        std::thread t(handle_ipv4_connection, clientSocket);
        t.detach(); 
    }
    
    close(serverSocket);
}

void ipv6_server() {
    int serverSocket = socket(AF_INET6, SOCK_STREAM, 0);

    struct sockaddr_in6 serverAddress;
    std::memset(&serverAddress, 0, sizeof(serverAddress));
    serverAddress.sin6_family = AF_INET6;
    serverAddress.sin6_port = htons(12346);
    serverAddress.sin6_addr = in6addr_any;

    bind(serverSocket, (struct sockaddr*)&serverAddress, sizeof(serverAddress));

    listen(serverSocket, 5);
    std::cout << "IPv6 server listening on port 12346" << std::endl;

    while (true) {
        struct sockaddr_in6 clientAddress;
        socklen_t clientAddressLen = sizeof(clientAddress);
        int clientSocket = accept(serverSocket, (struct sockaddr*)&clientAddress, &clientAddressLen);

        std::thread t(handle_ipv6_connection, clientSocket);
        t.detach(); 
    }
    close(serverSocket);
}

int main() {
    std::thread t1(ipv4_server);
    std::thread t2(ipv6_server);
    t1.join();
    t2.join();

    return 0;
}

ipv4客户端代码

#include <iostream>
#include <cstring>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>

int main() {

    int clientSocket = socket(AF_INET, SOCK_STREAM, 0);

    struct sockaddr_in serverAddress;
    std::memset(&serverAddress, 0, sizeof(serverAddress));
    serverAddress.sin_family = AF_INET;
    serverAddress.sin_port = htons(12345);
    serverAddress.sin_addr.s_addr = inet_addr("127.0.0.1");

    if (connect(clientSocket, (struct sockaddr*)&serverAddress, sizeof(serverAddress)) < 0) {
        std::cerr << "Failed to connect to server" << std::endl;
        return 1;
    }

    const char* message = "Hello, server!";
    if (send(clientSocket, message, strlen(message), 0) < 0) {
        std::cerr << "Failed to send data to server" << std::endl;
        close(clientSocket);
        return 1;
    }
    
    char buffer[1024];
    memset(buffer, 0, sizeof(buffer));
    if (recv(clientSocket, buffer, sizeof(buffer)-1, 0) < 0) {
        std::cerr << "Failed to receive data from server" << std::endl;
        close(clientSocket);
        return 1;
    }

    std::cout << "Received data from server: " << buffer << std::endl;

    close(clientSocket);

    return 0;
}

ipv6客户端代码

#include <iostream>
#include <cstring>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>

int main() {
    int clientSocket = socket(AF_INET6, SOCK_STREAM, 0);

    struct sockaddr_in6 serverAddress;
    std::memset(&serverAddress, 0, sizeof(serverAddress));
    serverAddress.sin6_family = AF_INET6;
    serverAddress.sin6_port = htons(12346);
    inet_pton(AF_INET6, "::1", &(serverAddress.sin6_addr));

    if (connect(clientSocket, (struct sockaddr*)&serverAddress, sizeof(serverAddress)) < 0) {
        std::cerr << "Failed to connect to server" << std::endl;
        return 1;
    }

    const char* message = "Hello, server!";
    if (send(clientSocket, message, strlen(message), 0) < 0) {
        std::cerr << "Failed to send data to server" << std::endl;
        close(clientSocket);
        return 1;
    }

    char buffer[1024];
    memset(buffer, 0, sizeof(buffer));
    if (recv(clientSocket, buffer, sizeof(buffer)-1, 0) < 0) {
        std::cerr << "Failed to receive data from server" << std::endl;
        close(clientSocket);
        return 1;
    }

    std::cout << "Received data from server: " << buffer << std::endl;

    close(clientSocket);

    return 0;
}

截图：