python - 从bash调用脚本时终止子进程

Question

我有一个 python 脚本：zombie.py

from multiprocessing import Process
from time import sleep
import atexit

def foo():
    while True:
        sleep(10)

@atexit.register
def stop_foo():
    p.terminate()
    p.join()

if __name__ == '__main__':
    p = Process(target=foo)
    p.start()

    while True:
        sleep(10)

当我使用python zombie.py & 运行它并使用kill -2 终止父进程时，stop() 被正确调用并且两个进程都终止了。

现在，假设我有一个 bash 脚本zombie.sh：

#!/bin/sh

python zombie.py &

echo "done"

我从命令行运行./zombie.sh。

现在，当父级被杀死时，stop() 永远不会被调用。如果我在父进程上运行kill -2，什么也不会发生。 kill -15 或 kill -9 都只是杀死父进程，而不是子进程：

[foo@bar ~]$ ./zombie.sh 
done
[foo@bar ~]$ ps -ef | grep zombie | grep -v grep
foo 27220     1  0 17:57 pts/3    00:00:00 python zombie.py
foo 27221 27220  0 17:57 pts/3    00:00:00 python zombie.py
[foo@bar ~]$ kill -2 27220
[foo@bar ~]$ ps -ef | grep zombie | grep -v grep
foo 27220     1  0 17:57 pts/3    00:00:00 python zombie.py
foo 27221 27220  0 17:57 pts/3    00:00:00 python zombie.py
[foo@bar ~]$ kill 27220
[foo@bar ~]$ ps -ef | grep zombie | grep -v grep
foo 27221     1  0 17:57 pts/3    00:00:00 python zombie.py

这里发生了什么？如何确保子进程与父进程一起死亡？

Answer 1

atexit 和p.daemon = True 都不会真正确保子进程将与父亲一起死亡。接收 SIGTERM 不会触发 atexit 例程。

为了确保孩子在父亲去世时被杀死，您必须在父亲中安装一个信号处理程序。这样，您可以对大多数信号（SIGQUIT、SIGINT、SIGHUP、SIGTERM，...）做出反应，但不能对 SIGKILL；根本没有办法从接收它的进程中对该信号作出反应。

为所有有用的信号安装一个信号处理程序，并在该处理程序中杀死子进程。

Answer 2

更新：此解决方案不适用于被信号终止的进程。

---

您的子进程不是zombie。它是活的。

如果您希望子进程在其父进程正常退出时被终止，则在p.start() 之前设置p.daemon = True。来自the docs：

当一个进程退出时，它会尝试终止其所有守护进程。

查看the source code，很明显multiprocessing 使用atexit 回调来杀死其守护进程子进程，即如果父进程被信号杀死，它将无法工作。例如：

#!/usr/bin/env python
import logging
import os
import signal
import sys
from multiprocessing import Process, log_to_stderr
from threading import Timer
from time import sleep

def foo():
    while True:
        sleep(1)

if __name__ == '__main__':
    log_to_stderr().setLevel(logging.DEBUG)
    p = Process(target=foo)
    p.daemon = True
    p.start()

    # either kill itself or exit normally in 5 seconds
    if '--kill' in sys.argv:
        Timer(5, os.kill, [os.getpid(), signal.SIGTERM]).start()
    else: # exit normally
        sleep(5)

输出

$ python kill-orphan.py [INFO/Process-1] 子进程调用 self.run() [INFO/MainProcess] 进程关闭 [DEBUG/MainProcess] 运行优先级 >= 0 的所有“atexit”终结器 [INFO/MainProcess] 为守护进程 Process-1 调用 terminate() [INFO/MainProcess] 为进程 Process-1 调用 join() [DEBUG/MainProcess] 运行剩余的“atexit”终结器

注意“为守护进程调用 terminate()” 行。

输出（带有--kill）

$ python kill-orphan.py --kill [INFO/Process-1] 子进程调用 self.run()

日志显示，如果父母被信号杀死，则不会调用“atexit”回调（ps 表明在这种情况下孩子还活着）。另见Multiprocess Daemon Not Terminating on Parent Exit。