多线程socket通信

#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <arpa/inet.h>
#include <pthread.h>

#define PORT 8888
#define BUFFER_SIZE 1024

void *serverThread(void *arg) {
    int server_fd, new_socket;
    struct sockaddr_in address;
    int addrlen = sizeof(address);
    char buffer[BUFFER_SIZE] = {0};
    char *response = "Hello, client!";

    // 创建 Socket
    if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0) {
        perror("Socket creation failed");
        exit(EXIT_FAILURE);
    }
    //允许重用端口
    int opt = 1;
    if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) < 0) {
        perror("setsockopt failed");
        exit(EXIT_FAILURE);
    }
    address.sin_family = AF_INET;
    address.sin_addr.s_addr = INADDR_ANY;
    address.sin_port = htons(PORT);

    // 绑定 IP 地址和端口号
    if (bind(server_fd, (struct sockaddr *)&address, addrlen) < 0) {
        perror("Bind failed");
        exit(EXIT_FAILURE);
    }

    // 监听连接请求
    if (listen(server_fd, 1) < 0) {
        perror("Listen failed");
        exit(EXIT_FAILURE);
    }

    printf("服务器已启动，等待连接...\n");

    // 接受客户端连接
    if ((new_socket = accept(server_fd, (struct sockaddr *)&address, (socklen_t *)&addrlen)) < 0) {
        perror("Accept failed");
        exit(EXIT_FAILURE);
    }

    printf("客户端已连接\n");

    // 接收数据
    if (recv(new_socket, buffer, BUFFER_SIZE, 0) < 0) {
        perror("Receive failed");
        exit(EXIT_FAILURE);
    }

    printf("server接收到的数据: %s\n", buffer);

    // 发送响应数据
    if (send(new_socket, response, strlen(response), 0) < 0) {
        perror("Send failed");
        exit(EXIT_FAILURE);
    }

    // 关闭连接
    close(new_socket);
    close(server_fd);

    pthread_exit(NULL);
}

void *clientThread(void *arg) {
    int client_fd;
    struct sockaddr_in server_address;
    char *message = "Hello, server!";
    char buffer[BUFFER_SIZE] = {0};

    // 创建 Socket
    if ((client_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0) {
        perror("Socket creation failed");
        exit(EXIT_FAILURE);
    }

    server_address.sin_family = AF_INET;
    server_address.sin_port = htons(PORT);

    // 将 IP 地址转换为二进制格式
    if (inet_pton(AF_INET, "127.0.0.1", &(server_address.sin_addr)) <= 0) {
        perror("Invalid address/ Address not supported");
        exit(EXIT_FAILURE);
    }

    // 连接服务器
    if (connect(client_fd, (struct sockaddr *)&server_address, sizeof(server_address)) < 0) {
        perror("Connection failed");
        exit(EXIT_FAILURE);
    }

    // 发送数据
    if (send(client_fd, message, strlen(message), 0) < 0) {
        perror("Send failed");
        exit(EXIT_FAILURE);
    }

    printf("数据已发送\n");

    // 接收响应数据
    if (recv(client_fd, buffer, BUFFER_SIZE, 0) < 0) {
        perror("Receive failed");
        exit(EXIT_FAILURE);
    }

    printf("client接收到的数据: %s\n", buffer);

    // 关闭连接
    close(client_fd);

    pthread_exit(NULL);
}

int main() {
    pthread_t serverThreadID, clientThreadID;

    // 创建服务器线程
    if (pthread_create(&serverThreadID, NULL, serverThread, NULL) != 0) {
        perror("Failed to create server thread");
        exit(EXIT_FAILURE);
    }

    // 创建客户端线程
    if (pthread_create(&clientThreadID, NULL, clientThread, NULL) != 0) {
        perror("Failed to create client thread");
        exit(EXIT_FAILURE);
    }

    // 等待线程结束
    pthread_join(serverThreadID, NULL);
    pthread_join(clientThreadID, NULL);

    return 0;
}

gcc -o program program.c -pthread
./program