socket编程之pthread多线程通信

/*************************************************************************
 *  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返回负值。