【Linux】TCP server如何与多个client通信？

Linux

发布日期: 2022-06-22

文章字数: 775

阅读时长: 3 分

阅读次数:

大家好，我是LinuxZn。

上一篇文章我们分享了TCP常用应用接口的使用封装，其demo中我们只是简单地实现一对一的收发。

但是实际开发中，tcp server是要支持与多个client同时进行通信的，本篇分享一对多的收发，也即tcp server并发处理。

tcp server实现并发的方式有：多进程、多线程。

多进程开销比较大，不常用。本篇笔记我们分享多线程的方法。

简单的demo

tcp_server.c：

#include <pthread.h>
#include "tcp_socket.h"

static pthread_t cli_data_proce_thread_tid;

static void *process_client_data(void *arg)
{
    int client_fd = *(int*)arg;

    while (1)
    {
        char buf[128] = {0};
        
        int recv_len = tcp_blocking_recv(client_fd, buf, sizeof(buf));
        if (recv_len <= 0)
        {
            printf("recv error!\n");
            tcp_close(client_fd);
            return NULL;
        }
        printf("client_fd = %d, recv : %s\n", client_fd, buf);

        int send_len = tcp_send(client_fd, buf, strlen(buf));
        if (send_len <= 0)
        {
            printf("send error!\n");
            tcp_close(client_fd);
            return NULL;
        }
        else
        {
            printf("send success! send: %s, send_len: %d\n", buf, send_len);
        }
        usleep(10 * 1000);
    }
}

int main(int argc, char **argv)
{
    printf("==================tcp server==================\n");
    int server_fd = tcp_init(NULL, 4321);

    while (1)
    {
        int new_fd = tcp_accept(server_fd);

        // 创建客户端数据处理线程
        int ret = pthread_create(&cli_data_proce_thread_tid, NULL, process_client_data, (void*)&new_fd);
        if(ret != 0)
        {
            perror("pthread_create");
            exit(EXIT_FAILURE);
        }
    }

    tcp_close(server_fd);

    return 0;
}

主线程，监听客户端连接；cli_data_proce_thread_tid线程处理客户端数据。下面我们创建4个client与该server进行连接。

首先，需要注意的是，我们创建tcp_server的方式为：

int server_fd = tcp_init(NULL, 4321);

int tcp_init(const char* ip, int port)
{
    int optval = 1; 
    int server_fd = socket(AF_INET, SOCK_STREAM, 0);
    if (server_fd < 0)
    {
        perror("socket");
        return -1;
    }

    /* 解除端口占用 */
    if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)) < 0)
	{
		perror("setsockopt\n");
		return -1;
	}

    struct sockaddr_in server_addr;
    bzero(&server_addr, sizeof(struct sockaddr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(port);
    if (NULL == ip)
    {
        server_addr.sin_addr.s_addr = htonl(INADDR_ANY); 
    }
    else
    {
        server_addr.sin_addr.s_addr = inet_addr(ip); 
    }

    if (bind(server_fd, (struct sockaddr*)&server_addr,sizeof(struct sockaddr)) < 0)
    {
        perror("bind");
        close(server_fd);
        return -1;
    }

    if(listen(server_fd, MAX_CONNECT_NUM) < 0)
    {
        perror("listen");
        close(server_fd);
        return -1;
    }

    return server_fd;
}