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

最新推荐文章于 2024-04-09 13:52:09 发布

原创最新推荐文章于 2024-04-09 13:52:09 发布 · 7.1k 阅读

5 ·

CC 4.0 BY-SA版权

Linux编程专栏收录该内容

0 篇文章

订阅专栏

本文介绍如何使用prctl系统调用让子进程在父进程意外退出时接收到信号，从而优雅地终止。通过示例代码展示了如何设置父进程死亡信号为SIGHUP，并在子进程中捕获该信号。

本文首先发表在 YYNOTE.COM .

原文地址： http://yynote.com/blog/detail/others/how-to-make-child-process-exit-after-parent-exits

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;
  }