linux多线程并发服务器

我就是我 2022-11-29 13:29 196阅读 0赞

# linux多线程并发服务器 #

此为多线程并发服务器，[多进程版][Link 1]。

#include <stdio.h>
    #include <unistd.h>
    #include <stdlib.h>
    #include <sys/types.h>
    #include <string.h>
    #include <sys/socket.h>
    #include <arpa/inet.h>
    #include <ctype.h>
    #include <pthread.h>
    
    //自定义数据结构
    typedef struct SockInfo
    { 
        int fd;//文件描述符cfd
        struct sockaddr_in addr;//存放ip地址的结构体
        pthread_t id;//线程的id
    }SockInfo;
    
    //子线程处理函数
    void* worker(void* arg){ 
        char ip[64];
        char buf[1024];
        SockInfo* info = (SockInfo*)arg;
        //通信
        while(1)
        { 
            printf("client IP:%s,port:%d\n",
                    inet_ntop(AF_INET,&info->addr.sin_addr.s_addr,ip,sizeof(ip)),
                    ntohs(info->addr.sin_port));
            int len = read(info->fd,buf,sizeof(buf));
            if(len == -1){ 
                perror("read error");
                pthread_exit(NULL);
            }
            else if(len ==0){ 
                printf("客户端断开了连接\n");
                close(info->fd);
                break;
            }
            else{ 
                printf("recv buf: %s\n",buf);
                write(info->fd,buf,len);
            }        
        }
    
        return NULL;
    }
    
    int main(int argc, char const *argv[])
    { 
        if(argc<2){ 
            printf("eg: ./a.out port\n");
            exit(1);
        }
        struct sockaddr_in serv_addr;
        socklen_t serv_len = sizeof(serv_addr);
        int port = atoi(argv[1]);
    
        //创建套接字
        int lfd = socket(AF_INET,SOCK_STREAM,0);
        //初始化服务器 sockaddr_in
        memset(&serv_addr,0,serv_len);
        serv_addr.sin_family = AF_INET;                 //地址族
        serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);  //监听本机所有的IP
        serv_addr.sin_port = htons(port);               //设置端口
        //绑定IP和端口
        bind(lfd,(struct sockaddr*)&serv_addr,serv_len);
    
        //设置同时监听的最大个数
        listen(lfd,36);
        printf("Start accept ......\n");
    
        int i = 0;
        SockInfo info[256];//线程个数,限制最多为256
        //规定 数组里边的fd等于-1，表示当前结构体空闲
        for(i=0;i<sizeof(info)/sizeof(info[0]);++i){ 
            info[i].fd = -1;
        }
        struct sockaddr_in client_addr;
        socklen_t cli_len = sizeof(struct sockaddr_in);
        while(1){ 
            //选一个没有被使用的，最小的数组元素
            for(i=0;i<256;++i){ 
                if(info[i].fd == -1){ 
                    break;
                }
            }
            if(i == 256){ 
                break;
            }
    
            //主线程 - 等待接受连接请求
            info[i].fd = accept(lfd,(struct sockaddr*)&info[i].addr,&cli_len);
    
            //创建子线程 - 通信
            pthread_create(&info[i].id,NULL,worker,&info[i]);
            //设置线程分离
            pthread_detach(info[i].id);
    
    
        }
    
        close(lfd);
        //只退出主线程
        pthread_exit(NULL);
        return 0;
    }

终端1

$ gcc pthread_server.c -lpthread -o server
    $ ./server 9876
    Start accept ......
    client IP:127.0.0.1,port:39808
    recv buf: 321
    
    client IP:127.0.0.1,port:39808
    client IP:127.0.0.1,port:39810
    recv buf: 456
    
    client IP:127.0.0.1,port:39810
    client IP:127.0.0.1,port:39820
    recv buf: hahaha
    
    client IP:127.0.0.1,port:39820
    客户端断开了连接
    客户端断开了连接
    客户端断开了连接

终端2

$ nc 127.1 9876
    321 -------------客户端输入的
    321 -------------服务端返回的

终端3

$ nc 127.1 9876
    456 -------------客户端输入的
    456 -------------服务端返回的

终端4

$ nc 127.1 9876
    hahaha
    hahaha

[Link 1]: https://blog.csdn.net/e891377/article/details/108151329