【网络编程实例】C++实现基于多进程的回射服务器

参考《TCP/IP网络编程第10章》

• 父进程fork子进程实现多客户端和服务器端通信；
• 利用信号量及信号处理函数实现子进程退出后用waitpid()回收，避免僵尸进程。

服务器端代码：echo_mlserver.cpp

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <sys/socket.h>
#include <sys/unistd.h>
#include <sys/types.h>
#include <sys/errno.h>
#include <netinet/in.h>
#include <signal.h>
#include <sys/wait.h>

#include <iostream>
#include <string>

using std::cout;
using std::cin;
using std::string;
using std::endl;

#define BUFFSIZE 30
#define DEFAULT_PORT 9190    // 指定端口为 9190


void read_childproc(int sig);

int main()
{ 
    
    int serv_sock, clnt_sock;
    struct sockaddr_in serv_adr, clnt_adr;

    pid_t pid;
    struct sigaction act;
    socklen_t adr_sz;
    int str_len, state;
    char buf[BUFFSIZE];

    act.sa_handler = read_childproc; //防止僵尸进程
    sigemptyset(&act.sa_mask);
    act.sa_flags = 0;
    state = sigaction(SIGCHLD, &act, 0);  //注册信号处理器,把成功的返回值给 state

    serv_sock = socket(PF_INET, SOCK_STREAM, 0);
    bzero(&serv_adr, sizeof(serv_adr));
    serv_adr.sin_family = AF_INET;
    serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_adr.sin_port = htons(DEFAULT_PORT);

    //调用 bind 函数分配ip地址和端口号
    if(bind(serv_sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)
       cout << "Bind error" << errno << ": " << strerror(errno) << endl;

     进入等待连接请求状态
    if(listen(serv_sock, 5) == -1)
       cout << "Listen error" <<  errno << ": " << strerror(errno) << endl;

    while(1)
    {
        adr_sz = sizeof(clnt_adr);
        clnt_sock = accept(serv_sock,(struct sockaddr *)&clnt_adr, &adr_sz);
         if (clnt_sock == -1)
            continue;
        else
           cout << "new client connected..." << endl;

        pid = fork();
        if(pid == -1)
        {
            close(clnt_sock);
            continue;
        }
        if(pid == 0)//子进程运行区域,此部分向客户端提供回声服务
        {
            close(serv_sock); //关闭服务器套接字，因为从父进程传递到了子进程
            while((str_len = read(clnt_sock, buf, BUFFSIZE)) != -1) 
                write(clnt_sock, buf, BUFFSIZE);
            
            close(clnt_sock);
            cout << "client disconnected..." << endl;
            return 0;
        }
        else
            close(clnt_sock);//通过 accept 函数创建的套接字文件描述符已经复制给子进程，因此服务器端要销毁自己拥有的文件描述符
    }
    close(serv_sock);
    return 0;
}

void read_childproc(int sig)
{
    pid_t pid;
    int status;
    pid = waitpid(-1, &status, WNOHANG);
    printf("removed proc id: %d \n", pid);
}

客户端代码：echo_clnt.cpp

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <sys/socket.h>
#include <sys/unistd.h>
#include <sys/types.h>
#include <sys/errno.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <iostream>
#include <string>

using std::cout;
using std::cin;
using std::string;
using std::endl;

#define BUFFSIZE 1024
#define SERVER_IP "127.0.0.1"    // 指定服务端的IP
#define SERVER_PORT 9190            // 指定服务端的port


int main()
{
    int sock;
    char message[BUFFSIZE];
    int str_len;
    struct sockaddr_in serv_adr;

    sock = socket(PF_INET, SOCK_STREAM, 0);
    if(sock == -1) cout << "socket() error" << endl;

    bzero(&serv_adr, sizeof(serv_adr));

    serv_adr.sin_family = AF_INET;
    serv_adr.sin_addr.s_addr = inet_addr(SERVER_IP);
    serv_adr.sin_port = htons(SERVER_PORT);

    if(connect(sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)
        cout << "connect() error: " << errno << " " << strerror(errno) << endl;
    else
        cout << "connected............" << endl;

    while(1)
    {
        cout << "Input message(Q to quit): " << endl;
        cin >> message;
        if(!strcmp(message, "q") || !strcmp(message, "Q"))
            break;

        write(sock, message, strlen(message));
        str_len = read(sock, message, BUFFSIZE - 1);
        message[str_len] = 0;
        cout << "Message from server:" << message << endl;
    }
    close(sock);
    return 0;
}