Socket编程之多进程实现并发服务器

1 实现思路

1. 在父进程中循环accept阻塞等待客户端连接，每接收一个连接请求就创建一个子进程；
2. 在子进程中实现客户端与服务端的通信；
3. 子进程结束后发送SIGCHLD信号，通过信号捕捉函数捕捉信号终止的信息，在回调函数中完成子进程的回收，避免产生僵尸进程。

2 服务端程序

#include <iostream>
#include <arpa/inet.h>
#include <sys/wait.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <string.h>
#include <errno.h>

#define SERV_PORT 9998

using namespace std;

void recycleChild(int sig) { //回调函数，当子进程退出时执行，用来回收子进程，防止僵尸进程
    while(waitpid(-1, NULL, WNOHANG) > 0); //父进程非阻塞式等待所有子进程结束
}

int main() {
    int lfd, cfd;
    struct sockaddr_in sarrd, caddr; //创建地址结构体，分别用来存放服务端地址信息和客户端地址信息
    pid_t pid;         //创建子进程

    //创建临时阻塞信号集
    struct sigaction act; 
    act.sa_flags =0;
    act.sa_handler = recycleChild;
    sigemptyset(&act.sa_mask); //清空临时阻塞信号集

    //注册信号
    sigaction(SIGCHLD, &act, NULL);

    //创建服务端套接字
    if((lfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
        perror("socket");
        exit(-1);
    }

    //为服务端地址结构赋值
    sarrd.sin_family = AF_INET;
    sarrd.sin_port = htons(SERV_PORT);
    sarrd.sin_addr.s_addr = htonl(INADDR_ANY);

    //设置端口复用
    socklen_t optval = 1;
    setsockopt(lfd, SOL_SOCKET, SO_REUSEPORT, &optval, sizeof(optval));

    //绑定服务端地址信息
    if(bind(lfd, (struct sockaddr *)&sarrd, sizeof(sarrd)) == -1) {
        perror("bind");
        exit(-1);
    }

    //监听socket上的连接
    if(listen(lfd, 8) == -1) {
        perror("listen");
        exit(-1);
    }

    char ServerIP[16];
    inet_ntop(AF_INET, &sarrd.sin_addr.s_addr, ServerIP, 16);
    cout << "init success" << endl;
    cout << "host ip: " << ServerIP << ", port: " << ntohs(sarrd.sin_port) << endl;

    //不断循环等待客户端连接
    while(1) {
        socklen_t len = sizeof(caddr);
        if((cfd = accept(lfd, (struct sockaddr*)&caddr, &len)) == -1) {
            if(errno == EINTR) continue;
            /*
            进程在一个慢系统调用(slow system call)中阻塞时，当捕获到某个信号且相应信号处理函数
            返回时，这个系统调用被中断，调用返回错误，设置errno为EINTR
            参考：https://blog.csdn.net/benkaoya/article/details/17262053
            */
            perror("accept");
            exit(-1);
        }
        //输出连接客户端信息
        char ClientIP[16];
        inet_ntop(AF_INET, &caddr.sin_addr.s_addr, ClientIP, 16); 
        cout << "client ip: " << ClientIP << ", port: " << ntohs(caddr.sin_port) << endl;

        if((pid = fork()) == 0) {
            while(1) {
                char recvBuf[1024];
                int len = 0;
                if((len = read(cfd, recvBuf, sizeof(recvBuf))) > 0) {
                    cout << "(SERVER) recv data: " << recvBuf << "   (from port: " << ntohs(caddr.sin_port)  << ")"<< endl;
                    int i = 0;
                    for(i = 0; i < len; i++) {
                        recvBuf[i] = toupper(recvBuf[i]);
                    }
                    recvBuf[i] = '\0';
                    write(cfd, recvBuf, strlen(recvBuf)); //转为大写后发回
                }
                else if(len == 0) {
                    cout << "(SERVER) client, from port: " << ntohs(caddr.sin_port) << " closed..." << endl;
                    break;
                }
                else if(len == -1) {
                    perror("read");
                    exit(-1);
                }
            }            
            close(cfd);
            exit(0);  //退出子进程
        }
        else if(pid > 0) {
            close(cfd);
        }
        else {
            perror("fork");
            exit(-1);
        }
    }
    close(lfd);
    return 0;
}

3 运行结果

客户端程序不变，先启动服务端，然后运行两个客户端进行测试：
客户端连接：

客户端断开：