socket 并发服务器 多线程模式

server.c 端

#include<stdio.h>

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


#define PORT 1234
#define BACKLOG 5
#define MAXDATASIZE 1000


typedef struct _ARG
{
       int connfd;
       struct sockaddr_in client;
}ARG;


void process_cli(int connectfd,struct sockaddr_in client);
void* start_routine(void* arg);


int main()
{
    int listenfd,connectfd;
    pthread_t thread;
    ARG *arg;


    struct sockaddr_in server;
    struct sockaddr_in client;
    int sin_size;
    
    if((listenfd=socket(AF_INET,SOCK_STREAM,0))==-1)
    {
        perror("socket failed");
        exit(1);
    }
    int opt=SO_REUSEADDR;
    setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));
    bzero(&server,sizeof(server));
    server.sin_family=AF_INET;
    server.sin_port=htons(PORT);
    server.sin_addr.s_addr=htonl(INADDR_ANY);
    if(bind(listenfd,(struct sockaddr*)&server,sizeof(struct sockaddr))==-1)
    {
        perror("bind error");
        exit(1);
    }
    if(listen(listenfd,BACKLOG)==-1)
    {
        perror("listen error");
        exit(1);
    }
    sin_size =sizeof(struct sockaddr_in);
    while(1)
    {
        if((connectfd=accept(listenfd,(struct sockaddr*)&client,&sin_size))==-1)
        {
            perror("accept failed");
            exit(1);
        }


       // ARG  arg;
        arg=(void*)malloc(sizeof(arg));
        arg->connfd=connectfd;
        memcpy((void *)&arg->client,&client,sizeof(client));
        if(pthread_create(&thread,NULL,start_routine,(void*)arg))
        {
            perror("pthread_create error");
            exit(1);
        }
    }
    close(listenfd);
}


void process_cli(int connectfd,struct sockaddr_in client)
{
    int num,i;
    char recvbuf[MAXDATASIZE],sendbuf[MAXDATASIZE],cli_name[MAXDATASIZE];
    printf("you got a connection from %s.\n",inet_ntoa(client.sin_addr));
    send(connectfd,"welcome to my server.\n",22,0);
    num=recv(connectfd,cli_name,MAXDATASIZE,0);
    if(num==0)
    {   
        close(connectfd);
        perror("client connected failed.");
        exit(1);
    }
    cli_name[num]='\0';
    printf("client'name is %s.\n",cli_name);
    while(num=recv(connectfd,recvbuf,MAXDATASIZE,0))
    {
        recvbuf[num]='\0';
        printf("recv client(%s)message:%s\n",cli_name,recvbuf);
        for(i=0;i<num;i++)
        {
            sendbuf[i]=recvbuf[i];
        }
        sendbuf[num]='\0';
        send(connectfd,sendbuf,strlen(sendbuf),0);
    }
    close(connectfd);
}


void* start_routine(void *arg)
{
    ARG *info;
    info=(ARG*)arg;
    process_cli(info->connfd,info->client);
    free(info);
    pthread_exit(NULL);

}

client.c 端

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netdb.h>


#define PORT 1234
#define MAXDATASIZE 1000


int main(int argc,char *argv[])
{
    int fd,numbytes;
    char buf[MAXDATASIZE];
    struct hostent *he;
    struct sockaddr_in server;
    int i=1;
    if((he=gethostbyname(argv[1]))==NULL)
    {
        perror("gethostbyname error");
        exit(1);
    }
    if((fd=socket(AF_INET,SOCK_STREAM,0))==-1)
    {
        perror("socket failed");
        exit(1);
    }
    bzero(&server,sizeof(server));
    server.sin_family =AF_INET;
    server.sin_port =htons(PORT);
    server.sin_addr =*((struct in_addr*)he->h_addr);
    if(connect(fd,(struct sockaddr*)&server,sizeof(struct sockaddr))==-1)
    {
        perror("bind error");
        exit(1);
    }
    if((numbytes=recv(fd,buf,MAXDATASIZE,0))==-1)
    {
        perror("recv error");
        exit(1);
    }
    buf[numbytes]='\0';
    printf("%s\n",buf);
    printf("input your name:");
    scanf("%s",buf);
    if((numbytes=send(fd,buf,strlen(buf),0))==-1)
    {
        perror("send failed");
        exit(1);
    }
    while(i)
    {
        printf("input me message(max char:%d\n.):",MAXDATASIZE);
        scanf("%s",buf);
        if(strlen(buf)<1)
        {
            i=0;
        }
        if((numbytes=send(fd,buf,strlen(buf),0))==-1)
        {
            perror("send failed");
            exit(1);
        }


        if((numbytes=recv(fd,buf,MAXDATASIZE,0))==-1)
        {   
            perror("recv error");
            exit(1);
        }
        buf[numbytes]='\0';
        printf("server message:%s\n",buf);
        printf("\n");
    }
    close(fd);
    return 0;
}