tcp并发服务器基于多线程实现

head.h 

#ifndef _HEAD_H_
#define _HEAD_H_ 1

#define handler_error(msg) do{\
        perror(msg);exit(EXIT_FAILURE);}while(0)

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <pthread.h>
#include <semaphore.h>
#include <signal.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <dirent.h>
#include <sys/un.h>
#include <sqlite3.h>

#endif

server.c
    

#include "head.h"


void *do_client(void *arg)
{
    int n;
    char buf[1024];
    int sockfd = * ((int *)arg);

    while(1)
    {
        //接受数据
        n = read(sockfd,buf,sizeof(buf) - 1);
        buf[n] = '\0';

        printf("Recv %d bytes : %s.\n",n,buf);

        //发回去
        write(sockfd,buf,n);

        if(strncmp(buf,"quit",4) == 0)
            break;

    }

    free(arg);

    pthread_exit(NULL);
}


//./server ip port
int main(int argc, const char *argv[])
{
    int pid;
    struct sockaddr_in peer_addr;
    socklen_t addrlen = sizeof(struct sockaddr);
    int listen_fd;
    struct sockaddr_in server_addr;

    if(argc < 3)
    {
        fprintf(stderr,"Usage : %s ip port.\n",argv[1]);
        exit(EXIT_FAILURE);
    }

     listen_fd = socket(AF_INET,SOCK_STREAM,0);
     if(listen_fd < 0){
         handler_error("socket");
     }


     memset(&server_addr,0,sizeof(server_addr));
     server_addr.sin_family = AF_INET;
     server_addr.sin_port = htons(atoi(argv[2]));
     server_addr.sin_addr.s_addr = inet_addr(argv[1]);

     if(bind(listen_fd,(struct sockaddr *)&server_addr,sizeof(server_addr)) < 0)
     {
         handler_error("bind");
     }

     listen(listen_fd,10);

     printf("listen ...\n");
     int ret;
     pthread_t tid;

     int *pconnect_fd;


     while(1)
     {


        pconnect_fd = (int *)malloc(sizeof(int));

        *pconnect_fd = accept(listen_fd,(struct sockaddr *)&peer_addr,&addrlen);
        if(connect_fd < 0)
            handler_error("accept");

        printf("*****************************\n");
        printf("Peer Ip  :%s.\n",inet_ntoa(peer_addr.sin_addr));
        printf("Peer Port:%d.\n",ntohs(peer_addr.sin_port));
        printf("*****************************\n");

        //创建线程和客户端进行收发数据
        ret = pthread_create(&tid,NULL,do_client,pconnect_fd);
        if(ret != 0){
            fprintf(stderr,"Fail to pthread_create : %s.\n",strerror(ret));
            exit(EXIT_FAILURE);
        }

         //让线程结束的时候，系统自动回收未释放的资源
        pthread_detach(tid);
     }

    return 0;
}

client.c

#include "head.h"

//./client ip port
int main(int argc, const char *argv[])
{
    int n;
    char buf[1024];
    int sockfd;
    struct sockaddr_in server_addr;

    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0){
        handler_error("socket");
    }

    memset(&server_addr,0,sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(atoi(argv[2]));
    server_addr.sin_addr.s_addr = inet_addr(argv[1]);

    if(connect(sockfd,(struct sockaddr *)&server_addr,sizeof(server_addr)) < 0){
        handler_error("connect");
    }

    //connect成功返回，表示三次握手完成
    while(1)
    {
        printf(">");

        fgets(buf,sizeof(buf),stdin);
        buf[strlen(buf) - 1] = '\0';

        write(sockfd,buf,strlen(buf));


        memset(buf,0,sizeof(buf));
        n = read(sockfd,buf,sizeof(buf) - 1);
        buf[n] = '\0';

        printf("Recv %d bytes : %s.\n",n,buf);
    }

    return 0;
}