UNP学习之多线程与多进程的回射服务器与客户端实现

UNP学习之多线程与多进程的回射服务器与客户端实现

回射程序是指：
（1）客户端从标准输入读入一个字符串，并写给服务器；
（2）服务器从网络读取字符串，并回射给用户；
（3）客户端从网络读入这个字符串并显示在标准输出上。

客户端：

// File Name: tcp_client.c
// Author: AlexanderGan
// Created Time: Thu 09 Jul 2020 04:05:50 PM CST

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<unistd.h>
#include<fcntl.h>
#include<arpa/inet.h>

int main(int argc, char* argv[]){
    if(argc < 3)
    {
        printf("eg: ./a.out 8888(port) 192.168.x.x \n");
        exit(1);
    }
    int port = atoi(argv[1]);//字符串转整数
    //创建套接字
    int sfd = socket(AF_INET,SOCK_STREAM,0);
    //连接服务器
    struct sockaddr_in serv;
    memset(&serv,0,sizeof(serv));
    serv.sin_family = AF_INET;
    serv.sin_port = htons(port);
    inet_pton(AF_INET,argv[2],&serv.sin_addr.s_addr);//点分十进制表达转数值
    connect(sfd,(struct sockaddr*)&serv,sizeof(serv));
    //通信
    while(1)
    {
        //发送数据
        char buf[1024];
        printf("请输入要发送的字符串:\n");
        fgets(buf,sizeof(buf),stdin);
        write(sfd,buf,strlen(buf)+1);
        //等待接收数据
        int len = read(sfd,buf,sizeof(buf));
        if(len < 0)
        {
            perror("read error!");
            exit(1);
        }
        else if(len == 0)
        {
            printf("服务器端关闭连接");
            break;
        }
        else
        {
            printf("收到的数据为:%s\n",buf);
        }
    }
    close(sfd);
    
  return 0 ;
}

多进程服务器：

// File Name: tcp_server.c
// Author: AlexanderGan
// Created Time: Sat 11 Jul 2020 10:55:32 PM CST

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<unistd.h>
#include<fcntl.h>
#include<signal.h>
#include<sys/wait.h>
#include<arpa/inet.h>
#include<errno.h>

void recyle(int arg)
{
    pid_t pid;
    while(pid = waitpid(-1,NULL,WNOHANG) > 0)
    {
        printf("child died ,pid = %d\n",pid);
    }
}

int main(int argc, char* argv[]){
    if(argc < 2)
    {
        printf("eg: ./a.out IP port\n");
    }
    int port = atoi(argv[1]);

    struct sockaddr_in serv_addr, client_addr;
    socklen_t serv_len = sizeof(serv_addr);
    socklen_t cli_len = sizeof(client_addr);
    //创建套接字
    int lfd = socket(AF_INET,SOCK_STREAM,0);
    //初始化服务器
    memset(&serv_addr,0,serv_len);
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_addr.sin_port = htons(port);
    //绑定IP和端口
    bind(lfd,(struct sockaddr*)&serv_addr,serv_len);
    //设置同时监听的最大个数
    listen(lfd,36);
    printf("Start accept !\n");
    //使用信号回收子进程pcb
    struct sigaction act;
    act.sa_handler = recyle;
    act.sa_flags = 0;
    sigemptyset(&act.sa_mask);
    sigaction(SIGCHLD,&act,NULL);

    while(1){

       //accept阻塞的时候被信号中断，处理信号之后
       //直接返回-1，errornum被设置为EINTR
       
        //父进程接受连接请求
        int cfd = accept(lfd,(struct sockaddr*)&client_addr,&cli_len);
        //如果被信号中断，那么继续accept
        while(cfd == -1 && errno == EINTR){
            int fd = accept(lfd,(struct sockaddr*)&client_addr,&cli_len);
        }
        printf("Connect Success!\n");

        pid_t pid = fork();
        if(pid == 0){
            close(lfd);
            //子进程
            char ip[64];
            printf("client IP : %s, port : %d\n",
            inet_ntop(AF_INET,&client_addr.sin_addr.s_addr,ip,sizeof(ip)),
            ntohs(client_addr.sin_port)
                );     //通信
            for(;;)
            {
                char buf[1024];
                int len = read(cfd,buf,sizeof(buf));
                if(len == -1 ){
                    perror("read error");
                    exit(1);
                }
                else if(len == 0){
                    printf("客户端断开连接");
                    close(cfd);
                    break;
                }
                else{
                    printf("recv buf: %s \n",buf);
                    write(cfd,buf,len);
                }
            }
            //子进程结束
            return 0;
        }
        else{
            //父进程
            close(cfd);//关闭cfd
        }
    }
    

  return 0 ;
}

多线程服务器：

// File Name: tcp_server.c
// Author: AlexanderGan
// Created Time: Sat 11 Jul 2020 10:55:32 PM CST

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<unistd.h>
#include<pthread.h>
#include<signal.h>
#include<sys/wait.h>
#include<arpa/inet.h>
#include<errno.h>

typedef struct SockInfo{
    int fd;
    struct sockaddr_in addr;
    pthread_t tid;
} sockinfo;

void* worker(void* arg)
{
    char buf[1024];
    char ip[64];
    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("客户端已经断开连接");
            close(info->fd);
            break;
        }
        else{
            printf("recv buf: %s\n",buf);
            write(info->fd,buf,len);
        }

    }
    return NULL;
}

int main(int argc, char* argv[]){
    if(argc < 2)
    {
        printf("eg: ./a.out IP port\n");
    }
    int port = atoi(argv[1]);

    struct sockaddr_in serv_addr, client_addr;
    socklen_t serv_len = sizeof(serv_addr);
    socklen_t cli_len = sizeof(client_addr);
    //创建套接字
    int lfd = socket(AF_INET,SOCK_STREAM,0);
    //初始化服务器
    memset(&serv_addr,0,serv_len);
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_addr.sin_port = htons(port);
    //端口复用
    int flag = 1;
    setsockopt(lfd,SOL_SOCKET,SO_REUSEADDR,&flag,sizeof(flag));

    //绑定IP和端口
    if(bind(lfd,(struct sockaddr*)&serv_addr,serv_len) < 0){
        perror("bind error");
        exit(1);
    }
    //设置同时监听的最大个数
    listen(lfd,36);
    printf("Start accept !\n");
    
    sockinfo info[256];
    int i = 0;
    //fd == -1,则未使用
    for(i = 0;i < sizeof(info)/sizeof(info[0]);++i){
        info[i].fd = -1;
    }
    while(1){
		//寻找最小未使用的fd对应的info
       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].tid,NULL,worker,&info[i]);
        //线程分离
        pthread_detach(info[i].tid);
        
        
    }
    //只退出主线程
    pthread_exit(NULL);
    return 0;
}