与父母一起杀死孩子

Question

我有一个程序生成并与 CPU 繁重、不稳定的进程通信，这不是我创建的。如果我的应用程序崩溃或被 SIGKILL 杀死，我希望子进程也被杀死，这样用户就不必跟踪它们并手动杀死它们。

我知道这个主题以前已经讨论过，但我已经尝试了所描述的所有方法，但似乎没有一个能够经受住考验。

我知道这一定是可能的，因为终端一直在这样做。如果我在终端中运行某些东西，然后终止该终端，那么这些东西总是会死掉。

我尝试过atexit、double fork 和ptys。 atexit 不适用于 sigkill；双叉根本不起作用；和 ptys 我没有找到使用 python 的方法。

今天，我发现了 prctl(PR_SET_PDEATHSIG, SIGKILL)，这应该是子进程在父进程死亡时命令自身终止的一种方式。 我尝试将它与 popen 一起使用，但它似乎根本没有任何效果：

import ctypes, subprocess
libc = ctypes.CDLL('/lib/libc.so.6')
PR_SET_PDEATHSIG = 1; TERM = 15
implant_bomb = lambda: libc.prctl(PR_SET_PDEATHSIG, TERM)
subprocess.Popen(['gnuchess'], preexec_fn=implant_bomb)

在上面，子级被创建，父级退出。现在，您可能期望 gnuchess 收到 SIGKILL 并死亡，但事实并非如此。我仍然可以在进程管理器中使用 100% CPU 找到它。

谁能告诉我我使用 prctl 是否有问题？ 或者你知道终端是如何杀死它们的孩子的吗？

Answer 1

我知道已经过去很多年了，但我找到了解决这个问题的一个简单（有点老套）的方法。在您的父进程中，将所有调用包装在一个非常简单的 C 程序中，该程序调用 prctl()，然后调用 exec() 在 Linux 上解决了这个问题。我称之为“yeshup”：

#include <linux/prctl.h>
#include <signal.h>
#include <unistd.h>

int main(int argc, char **argv) {
     if(argc < 2)
          return 1;
     prctl(PR_SET_PDEATHSIG, SIGHUP, 0, 0, 0);
     return execvp(argv[1], &argv[1]);
}

当从 Python（或任何其他语言）生成子进程时，您可以运行“yeshup gnuchess [argments]”。您会发现，当父进程被终止时，所有子进程（应该）都会很好地发出 SIGHUP 信号。

这是有效的，因为即使在调用 execvp 之后，Linux 也会尊重对 prctl 的调用（不清除它）（这有效地将 yeshup 进程“转换”为 gnuchess 进程，或您在那里指定的任何命令），这与 fork() 不同。

Answer 2

prctl 的 PR_SET_DEATHSIG 只能设置这个正在调用 prctl 的进程——不适用于任何其他进程，包括该特定进程的子进程。我指向的手册页表达这一点的方式是“这个值在 fork() 上被清除”——fork，当然，是生成其他进程的方式（在 Linux 和任何其他进程中）其他 Unix-y 操作系统）。

如果您无法控制要在子进程中运行的代码（本质上，对于您的 gnuchess 示例来说就是这种情况），我建议您首先使用以下命令生成一个单独的小型“监视器”进程跟踪其所有兄弟进程（您的父进程可以让监视器在生成这些兄弟进程时了解这些兄弟进程的 pid），并在公共父进程死亡时向它们发送杀手信号（监视器需要轮询，唤醒每个兄弟进程） N 秒，用于检查父级是否还活着；使用 select 等待来自父级的更多信息，超时时间为 N 秒，循环内）。

这并非微不足道，但此类系统任务通常并非微不足道。终端的做法有所不同（通过进程组的“控制终端”的概念），但当然，任何子进程都可以轻松阻止它（双叉、nohup 等）。

Answer 3

实际上，我发现您原来的方法对我来说效果很好 - 这是我测试过的确切示例代码：

echoer.py

#!/bin/env python

import time
import sys
i = 0
try:
    while True:
        i += 1
        print i
        time.sleep(1)
except KeyboardInterrupt:
    print "\nechoer caught KeyboardInterrupt"
    exit(0)

parentProc.py

#!/bin/env python

import ctypes
import subprocess
import time

libc = ctypes.CDLL('/lib64/libc.so.6')
PR_SET_PDEATHSIG = 1
SIGINT = 2
SIGTERM = 15

def set_death_signal(signal):
    libc.prctl(PR_SET_PDEATHSIG, signal)

def set_death_signal_int():
    set_death_signal(SIGINT)

def set_death_signal_term():
    set_death_signal(SIGTERM)

#subprocess.Popen(['./echoer.py'], preexec_fn=set_death_signal_term)
subprocess.Popen(['./echoer.py'], preexec_fn=set_death_signal_int)
time.sleep(1.5)
print "parentProc exiting..."

Answer 4

我以为双叉是从控制终端上分离出来的。我不确定你是如何尝试使用它的。

这是一种黑客行为，但您始终可以调用“ps”并搜索您试图杀死的进程名称。

Answer 5

我见过使用诸如 ps xuawww | 这样的东西来“清理”的非常令人讨厌的方法。 grep 我的应用程序 | awk '{ 打印 $1}' | xargs -n1 Kill -9

客户端进程如果弹出，可以捕获 SIG_PIPE 并终止。有很多方法可以解决这个问题，但这实际上取决于很多因素。如果你在子进程中抛出一些 ping 代码（ping 到父进程），你可以确保在死亡时发出 SIG_PIPE。如果它捕获了它，它应该捕获它，它就会终止。您需要双向通信才能正常工作......或者始终阻止客户端作为通信的发起者。如果您不想修改子项，请忽略此操作。

假设您不希望实际的 Python 解释器出现段错误，您可以将每个 PID 添加到一个序列中，然后在退出时终止。这对于退出甚至未捕获的异常来说应该是安全的。 Python 有执行退出代码的工具......以进行清理。

这里有一些更安全的讨厌的方法：将每个子 PID 附加到一个文件，包括您的主进程（单独的文件）。使用文件锁定。构建一个看门狗守护进程，用于查看主 pid 的集群（）状态。如果没有锁定，则杀死子 PID 列表中的每个 PID。在启动时运行相同的代码。

更糟糕的是：将 PID 写入文件，如上所述，然后在子 shell 中调用您的应用程序：(./myMaster; ./killMyChildren)

Answer 6

我想知道 PR_SET_PDEATHSIG 标志是否被清除，即使您在 fork 之后（以及 exec 之前）设置了它，所以看起来从类似的文档中，它不应该被清除。

为了测试该理论，您可以尝试以下操作：使用相同的代码运行用 C 编写的子进程，基本上只调用 prctl(PR_GET_PDEATHSIG, &result) 并打印结果。

您可以尝试的另一件事是：在调用 prctl 时为 arg3、arg4 和 arg5 添加显式零。 IE：

>>> implant_bomb = lambda: libc.prctl(PR_SET_PDEATHSIG, TERM, 0, 0, 0)

Answer 7

有一些安全限制需要考虑，因为如果我们在 execv 之后调用 setuid，他的孩子就无法接收信号。此限制的完整列表位于此处

祝你好运！
/穆罕默德

Answer 8

其他答案提到了 prctl 的 PR_SET_DEATHSIG 但忽略了这样一个事实，即可以使用

setpriv --pdeathsig HUP [command] &