socket编程之pthread多线程通信

ゝ一纸荒年。 2022-09-24 10:26 224阅读 0赞

/*************************************************************************
     *  Copyright (C): 1540999272@qq.com
     *  Filename:      server.c
     *  Author:        Lu Zengmeng
     *  Description:   创建子线程接受客户端数据
     *  Creat Time:    2016-07-19 12:08
     ************************************************************************/
    #include <stdio.h> 
    #include <stdlib.h> 
    #include <string.h> 
    #include <sys/types.h> 
    #include <sys/socket.h> 
    #include <unistd.h> 
    #include <arpa/inet.h> 
    #include <netinet/in.h> 
    #include <strings.h> 
    #include <errno.h> 
    #include <pthread.h> 
    
    #define PORT            8003
    #define BACKLOG     10
    #define BUF_LEN     128
    
    void *recv_data(void *fd);
    
    int main(int argc, char **argv)
    {
            int      sockfd;
            int      *conn_fd;
            struct   sockaddr_in addr;
    
            if(-1==(sockfd=socket(AF_INET,SOCK_STREAM,0)))
            {
                    perror("socket");
                    exit(1);
            }
    
            addr.sin_family = AF_INET;
            addr.sin_port = htons(PORT);
            addr.sin_addr.s_addr = htons(INADDR_ANY);
            bzero(&(addr.sin_zero),8);
    
            if(-1==(bind(sockfd,(struct sockaddr*)&addr,sizeof(addr))))
            {
                    perror("bind");
                    exit(1);
            }
    
            if(-1==listen(sockfd, BACKLOG))
            {
                    perror("listen");
                    exit(1);
            }
    
            while(1)
            {
    
            pthread_t thread;
            conn_fd = (int*)malloc(sizeof(int));
    
                    if (-1 == (*conn_fd = accept(sockfd, NULL, NULL)))
                    {
                            perror("accept");
                        continue;
                    }
    
            if (0!= pthread_create(&thread,NULL,recv_data,conn_fd))
            {
                perror("pthread_create");
                break;
            }
            }
    
        shutdown(sockfd,2);
        shutdown(*conn_fd,2);
        return 0;
    
    }
    
    void *recv_data(void *fd)
    {
            int               conn_fd;
            int               recv_len;
            unsigned char     recv_buf[BUF_LEN];
    
        conn_fd = *((int*)fd);
        for(;;)
        {
            memset(recv_buf,0,BUF_LEN);
                recv_len = recv(conn_fd,recv_buf,BUF_LEN,0);
    
            if(-1==recv_len)
            {
                perror("recv");
                exit(1);
            }
                recv_buf[recv_len] = '\0';
    
            if(strncmp(recv_buf,"exit",4)==0)
            {
                printf("%s\n",recv_buf);
                break;
            }
    
                printf("Receive data:%s\n", recv_buf);
        }
    
        free(fd);
            close(conn_fd);
        pthread_exit(NULL);
    }

编译服务器时需要链接phtread库

gcc -lpthread server.c -o server

上述代码中pthread\_create函数用来创建线程并执行线程函数，函数原型为int pthread\_create(pthread\_t \*thread, const pthread\_attr\_t \*attr,void \*(\*start\_routine) (void \*), void \*arg)，第一个参数thread为线程标识符，第二个参数attr是线程属性设置，第三个参数start\_routine为线程函数地址，第四个参数arg为传给线程函数的参数。该函数声明在pthread.h头文件中，实现在pthread库。函数成功执行返回0，失败返回-1。

pthread\_exit()函数用来结束该线程，相当于进程中的exit()函数。

/*************************************************************************
     *  Copyright (C): 1540999272@qq.com
     *  Filename:      server.c
     *  Author:        Lu Zengmeng
     *  Description:   客户端程序，向服务器发送数据
     *  Creat Time:    2016-07-19 12:08
     ************************************************************************/
    #include <stdio.h>
    #include <sys/types.h>
    #include <sys/socket.h>
    #include <arpa/inet.h>
    #include <netinet/in.h>
    #include <strings.h>
    #include <stdlib.h>
    #include <unistd.h>
    #include <string.h>
    #include <fcntl.h>
    #include <sys/stat.h>
    #include <time.h>
    
    #define PORT             8003
    #define BUF_LEN          100
    
    int main(int argc, char *argv[])
    {
            int                    sockfd;
            unsigned char          send_buf[BUF_LEN];
            unsigned char          send_len;
            struct sockaddr_in     serv_addr;
    
        if (argc != 2)
        {
            printf("Usage: %s ServerIP\n",argv[0]);
            exit(1);
        }
    
        bzero(&serv_addr,sizeof(serv_addr));
        serv_addr.sin_family = AF_INET;
        serv_addr.sin_port   = htons(PORT);
    
        if ((inet_pton(AF_INET,argv[1],&serv_addr.sin_addr)) < 0)
        {
            perror("inet_pton");
            exit(1);
        }
    
        if (-1 == (sockfd = socket(AF_INET, SOCK_STREAM, 0)))
        {
            perror("socket");
            exit(1);
        }
    
        if (-1 == connect(sockfd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)))
        {
            perror("connect");
            exit(1);
        }
    
        for(;;)
        {
            memset(send_buf,0,BUF_LEN);
            printf("Input your data:\n");
            fgets(send_buf,BUF_LEN,stdin);
            fflush(stdin);
    
            if(strncmp(send_buf,"exit",4)==0) 
            {
                if (-1 == (send(sockfd, send_buf, BUF_LEN,0)))
                    {
                        perror("read");
                        exit(1);
                    }
                break;
            }
    
            if (-1 == (send(sockfd, send_buf, BUF_LEN,0)))
                {
                perror("read");
                exit(1);
                }
        }
    
        close(sockfd);
            return 0;
    }

int strncmp(const char \*s1, const char \*s2, size\_t n);该函数用来比较字符串s1和s2的前n个字符，相同返回0，s1大于s2返回正值，s1小于s2返回负值。