温馨提示×

linux socket5可设置多线程吗

小樊
81
2024-12-29 06:12:20
栏目: 智能运维

在Linux中,套接字(socket)本身并不直接支持多线程

以下是一个简单的示例,展示了如何在C语言中使用多线程处理TCP连接:

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

#define PORT 8080
#define BUFFER_SIZE 1024

void *handle_client(void *arg);

int main() {
    int server_fd, client_fd;
    struct sockaddr_in server_addr, client_addr;
    socklen_t addr_len = sizeof(client_addr);
    pthread_t thread_id;

    // 创建套接字
    if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0) {
        perror("socket failed");
        exit(EXIT_FAILURE);
    }

    // 配置服务器地址
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = INADDR_ANY;
    server_addr.sin_port = htons(PORT);

    // 绑定套接字
    if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
        perror("bind failed");
        exit(EXIT_FAILURE);
    }

    // 监听连接
    if (listen(server_fd, 3) < 0) {
        perror("listen");
        exit(EXIT_FAILURE);
    }

    printf("Server is listening on port %d\n", PORT);

    while (1) {
        // 接受客户端连接
        if ((client_fd = accept(server_fd, (struct sockaddr *)&client_addr, (socklen_t *)&addr_len)) < 0) {
            perror("accept");
            exit(EXIT_FAILURE);
        }

        // 为每个客户端连接创建一个新线程
        if (pthread_create(&thread_id, NULL, handle_client, (void *)&client_fd) != 0) {
            perror("pthread_create");
            exit(EXIT_FAILURE);
        }

        // 等待线程结束
        pthread_join(thread_id, NULL);
    }

    close(server_fd);
    return 0;
}

void *handle_client(void *arg) {
    int client_fd = *((int *)arg);
    char buffer[BUFFER_SIZE] = {0};

    // 读取客户端发送的数据
    read(client_fd, buffer, BUFFER_SIZE);
    printf("Received from client: %s\n", buffer);

    // 向客户端发送数据
    send(client_fd, "Hello from server!", strlen("Hello from server!"), 0);
    printf("Sent to client: Hello from server!\n");

    // 关闭套接字
    close(client_fd);
    return NULL;
}

在这个示例中,我们创建了一个TCP服务器,监听8080端口。当客户端连接时,服务器会为每个客户端创建一个新线程来处理连接。这样,多个客户端可以同时与服务器通信,实现了多线程处理。

0