父进程退出后如何通知子进程退出

最新推荐文章于 2024-03-23 21:57:41 发布

转载最新推荐文章于 2024-03-23 21:57:41 发布 · 7.5k 阅读

文章标签：

#linux

后端专栏收录该内容

39 篇文章

订阅专栏

本文介绍如何在Linux中使用prctl系统调用来使父进程能在意外终止时通知其子进程。通过设置特定信号，即使父进程崩溃也能确保子进程得到通知并采取相应措施。

Linux 中创建子进程是相当方便的，通过fork调用即可。当子进程退出时，要给父进程发送SIG_CHLD信号，是为了父进程回收子进程的资源，方便管理的目的。但是当父进程退出后，一般是不会通知子进程的，父进程会将自己的所有子进程过继给init进程。

但是，在实际的项目中，我们有这样的需求：如果父进程退出后，希望能通知子进程退出。

我们知道，可以利用进程间通信机制，在父进程退出前主动发送信号或pipe或其他手段告知子进程自己退出了。

但是，当父进程意外退出时，比如coredump了，根本没机会主动发送信号或pipe等消息的。

这时怎么办呢？

我们发现 prctl 这个调用，通过man prctl:

PR_SET_PDEATHSIG (since Linux 2.1.57)
          Set  the parent process death signal of the calling process to arg2    (either a signal value in the range
          1..maxsig, or 0 to clear).  This is the signal that the calling process
          will get when its parent  dies.

据此，我们可以在子进程中给父进程设置当父进程退出时子进程收到的信号。

prctl(PR_SET_PDEATHSIG, SIGHUP);

代码如下：

#include <stdio.h>
  #include <stdlib.h>
  #include <string.h>
  #include <sys/prctl.h>
  #include <signal.h>

  static int do_abort = 0;

  void handle_signal(int signo)
  {
      if (signo == SIGHUP)
      {                 
          printf("child recv SIGHUP..\n");
          do_abort = 1;
      }
  }

  int main(void)
  {
      pid_t pid;
      pid = fork();
      char *p = NULL;

      if (pid == 0) // child
      {
          signal(SIGHUP, handle_signal);
          prctl(PR_SET_PDEATHSIG, SIGHUP);
          while(!do_abort) {
              sleep(1);
              printf("in child...\n");
          }
          printf("child exit...\n");
          return 0;
      }
      else // parent
      {
          int times = 5;
          while(times -- > 0)
          {
              sleep(1);
              if (times == 3)
              {
                  printf("memcpy ...\n");
                  memcpy(p, "Hello", 5);
              }
              printf("in parent.\n");
          }

          printf("parent exit..\n");
          return 0;
      }

      return 0;
  }