Cpp网络通信03-多进程服务端通信

修改Server.cpp

// include...

// 服务端实例
Net::TcpServer server;


// 子进程退出
void HandleChildExit(int sig) {
  // 防止信号处理函数在执行的过程中再次被信号中断
  signal(SIGINT, SIG_IGN);
  signal(SIGTERM, SIG_IGN);

  cout << "子进程" << getpid() << "退出, sig=" << sig << endl;

  // 此处增加释放资源的代码(仅当前子进程)
  server.CloseClient();
  exit(0);
}

// 父进程退出
void HandleMainExit(int sig) {
  // 防止信号处理函数在执行的过程中再次被信号中断
  signal(SIGINT, SIG_IGN);
  signal(SIGTERM, SIG_IGN);

  cout << "父进程退出, sig=" << sig << endl;

  kill(0, SIGTERM); // 向全部子进程发送终止信号，通知它们退出

  // 此处增加释放资源的代码(仅当前子进程)
  server.CloseListener();
  exit(0);
}

int main(int argc, char *argv[]) {

  if (argc != 2) {
    printf("usage:\n\tServer <port>\n");
    return -1;
  }

  // 忽略全部信号，不希望被打扰，顺便解决僵尸进程的问题
  for (int i = 1; i <= 64; i++)
    signal(i, SIG_IGN);
  // 设置信号，在shell状态下可用"kill 进程号"或"ctrl+c"正常终止
  // 但不要用"kill -9 进程号"强行终止
  signal(SIGTERM, HandleMainExit);
  signal(SIGINT, HandleMainExit);

  // 初始化服务端，并开始监听端口
  if (!server.Init(atoi(argv[1]))) {
    perror("TcpServer::Init()");
    return -1;
  }
  cout << "服务器端口: " << argv[1] << endl;

  while (true) {
    // 服务端开始受理客户端连接
    if (!server.Accept()) {
      perror("TcpServer::Accept()");
      return -1;
    }

    // fork一个子进程来处理与客户端的通讯
    int pid = fork();
    if (pid == -1) { // 系统资源不足
        perror("fork");
      return -1;
    }
    if (pid > 0) { 
      server.CloseClient(); // 父进程不需要和客户端交流，所以关闭客户端连接的socket
      continue; // 父进程返回Accept等待，子进程继续执行下面的逻辑
    }

    server.CloseListener(); // 子进程关闭监听用的socket

    // 子进程需要重新设置信号
    signal(SIGTERM, HandleChildExit);
    signal(SIGINT, SIG_IGN); // 子进程不需要捕获SIGINT(强行中断)信号

    cout << "客户端已连接: " << server.GetClientIp() << endl;

    string buffer;
    while (true) {
      // 接收对端的报文
      // ...
    }
    return 0; // 子进程数据处理完毕后，一定要退出，否则会回到Accept
  }
}