欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Linux socket多线程实现回声服务器

程序员文章站 2024-03-23 10:18:52
...

服务端

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

#define CONNNUM 10

int fd_connect[CONNNUM] = {0}; // 连接描述符数组


int fd_sock; // 套接字描述符

pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER; // 互斥锁

// 服务器初始化
void server_init(short port); 

// 服务器运行
void server_run();

// 服务器结束
void server_end();

// 数据处理
void *data_handle(int fd_conn);

// 线程入口函数
void *thread_func(void *arg);

int main(int argc, char const *argv[])
{
    if (argc != 2)
    {
        fprintf(stderr, "Usage: %s <server-port>\n", argv[0]);
        return 1;
    }

    server_init(atoi(argv[1]));

    server_run();

    server_end();

    return 0;
}

// 服务器端初始化
void server_init(short port)
{
     // 创建套接字
    fd_sock = socket(AF_INET, SOCK_STREAM, 0);
    if (fd_sock == -1)
    {
        perror("socket");
        exit(1);
    }

    // 准备通信地址
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(port);
    addr.sin_addr.s_addr = htonl(INADDR_ANY);

    // 绑定
    int res = bind(fd_sock, (const struct sockaddr *)&addr, sizeof(addr));
    if (res == -1)
    {
        perror("bind");
        exit(1);
    }
    printf("bind success!\n");

    // 监听
    res = listen(fd_sock, 10);
    if (res == -1)
    {
        perror("listen");
        exit(1);
    }

    printf("server init success!\n");
}

// 服务器端运行
void server_run()
{
    while (1)
    {
        // 接收连接
        int fd_conn = accept(fd_sock, NULL, NULL);
        if (fd_conn == -1)
        {
            perror("accept");
            exit(1);
        }

        printf("new connection!\n");
        pthread_t tid; // 创建新线程
        int res = pthread_create(&tid, NULL, thread_func, (void *)(long)fd_conn);
        if (res != 0)
        {
            errno = res;
            perror("pthread_creat");
            exit(1);
        }

        // 设置线程属性为分离
        pthread_detach(tid);
        // pid_t pid = fork(); // 创建子进程
        // if (pid == -1)
        // {
        //     perror("fork");
        //     exit(1);
        // }
        // else if (pid == 0) // 子进程
        // {
        //     // 数据处理
        //     data_handle(fd_conn);
        //     exit(0);
        // }
        
    }

}

// 数据处理
void *data_handle(int fd_conn)
{
    // 修改连接描述符数组
    int index;
    pthread_mutex_lock(&mutex);
    for (int i = 0; i < 10; i++)
    {
        if (fd_connect[i] == 0)
        {
            fd_connect[i] = fd_conn;
            index = i;
            break;
        } 
    }
    pthread_mutex_unlock(&mutex);

    // 接收数据
    char str[512] = {0};
    struct sockaddr_in client_addr;
    socklen_t addrlen = sizeof(client_addr);
    getpeername(fd_conn, (struct sockaddr *)&client_addr, &addrlen);
    printf("[%s : %d]  online!\n", inet_ntoa(client_addr.sin_addr), client_addr.sin_port);
    while (1)
    {
        int r = read(fd_conn, str, sizeof(str));
        if (r == -1)
        {
            perror("recv cilent");
            exit(1);
        }
        else if (r == 0) // 对端关闭连接
        {
            break;
        }
        printf("recv data: %s\n", str);
        if (!strcmp(str, "end"))
            break;
        
        pthread_mutex_lock(&mutex);
        for (int i = 0; i < 10; i++) // 群发消息
        {
            if (fd_connect[i] != 0)
            {
                r = write(fd_connect[i], str, r);
                if (r == -1)
                {
                    perror("send cilent");
                    exit(1);
                }
            }
        }
        pthread_mutex_unlock(&mutex);
    }
    printf("[%s : %d]  offline!\n", inet_ntoa(client_addr.sin_addr), client_addr.sin_port);

    pthread_mutex_lock(&mutex);
    fd_connect[index] = 0;
    pthread_mutex_unlock(&mutex);

    close(fd_conn);
}

void server_end()
{
    close(fd_sock);
}

// 线程函数
void *thread_func(void *arg)
{
    int fd_conn = (long)arg;
    data_handle(fd_conn);
}

客户端

/*
 * @Descripttion: 
 * @version: 
 * @Author: Chilk
 * @Date: 2020-07-29 17:01:23
 * @LastEditors: Chilk
 * @LastEditTime: 2020-08-01 09:54:54
 */
#include <stdio.h>
#include <assert.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>

int main(int argc, char const *argv[])
{

    //创建一个IPv4的流式套接字
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1)
    {
        perror("socket error");
        exit(1);
    }

    struct sockaddr_in info;
    info.sin_family = AF_INET;
    // info.sin_port = htons(54321);
    info.sin_port = htons(atol(argv[2]));
    info.sin_addr.s_addr = inet_addr(argv[1]);
    // info.sin_addr.s_addr = htonl(INADDR_ANY);

    if (connect(sockfd, (const struct sockaddr *)(&info), sizeof(struct sockaddr)) == -1)
    {
        perror("connect error");
        exit(1);
    }
    printf("连接成功\n");

    while (1)
    {
        char buf[256];
        gets(buf);
        write(sockfd, buf, strlen(buf)+1);

        char rev_buf[256];
        read(sockfd, rev_buf, 254);
        printf("%s\n", rev_buf);
    }

    close(sockfd);
    return 0;
}