如何向Python绑定的C++库启动的子进程广播SIGINT信号？

问题根源 #

Python的信号处理依赖解释器主循环调度，当程序进入C库代码路径后，若C代码长期占用CPU且未释放GIL（全局解释器锁），或未实现信号处理逻辑，Python的信号处理函数会被延迟到C++代码执行完毕、回到Python上下文后才触发。

信号广播的正确流程 #

1. 消除C++代码对Python信号处理的阻塞 #

• 若可修改C库：在循环或阻塞操作中，通过Py_BEGIN_ALLOW_THREADS和Py_END_ALLOW_THREADS宏定期释放GIL，让Python解释器有机会处理信号；同时可在C层监听SIGINT，触发子进程清理并退出。
• 若无法修改C++库：通过Python层线程绕开主线程阻塞限制，实现信号的实时处理。

2. 捕获信号并广播至所有子进程 #

步骤1：获取C++启动的子进程PID

• 若C++库提供API返回子进程PID，直接留存；若无API，可通过进程树遍历获取Python进程的所有子进程：

import psutil
import os

def get_cpp_child_pids():
    parent_proc = psutil.Process(os.getpid())
    # 递归获取所有子进程PID，可根据C++子进程特征过滤
    return [child.pid for child in parent_proc.children(recursive=True)]

步骤2：自定义信号处理逻辑

在Python中注册SIGINT处理函数，收到信号后先向所有子进程广播信号，再执行自身清理。为避免主线程被C++代码阻塞，用守护线程监听信号：

import signal
import os
import threading
import time

child_pids = []

def handle_sigint(signum, frame):
    # 向所有子进程发送SIGINT
    for pid in child_pids:
        try:
            os.kill(pid, signal.SIGINT)
        except OSError:
            # 进程已退出则忽略
            pass
    # 执行Python侧清理逻辑
    print("Python进程启动清理流程...")
    exit(0)

def signal_monitor():
    signal.signal(signal.SIGINT, handle_sigint)
    while True:
        time.sleep(1)

# 启动信号监听守护线程
threading.Thread(target=signal_monitor, daemon=True).start()

# 调用C++库启动子进程
# cpp_library.launch_processes()
# 获取子进程PID列表
child_pids = get_cpp_child_pids()

# 执行C++库的阻塞操作
# cpp_library.run_blocking_task()

3. 修复C++库的信号屏蔽问题 #

部分C库会主动屏蔽SIGINT，导致Python进程无法接收信号。可在调用C库前后调整信号掩码：

import signal

# 保存原始信号掩码
original_mask = signal.pthread_sigmask(signal.SIG_BLOCK, [])
# 解除SIGINT屏蔽
signal.pthread_sigmask(signal.SIG_UNBLOCK, [signal.SIGINT])

# 执行C++库代码
# cpp_library.execute()

# 恢复原始信号掩码
signal.pthread_sigmask(signal.SIG_SETMASK, original_mask)

注意事项 #

• 若C++子进程忽略SIGINT，可改用SIGTERM或SIGKILL强制终止。
• 确保信号监听线程为守护线程，避免阻碍主进程正常退出。
• 若C代码完全阻塞主线程且无法释放GIL，可将C库调用封装为独立subprocess进程，Python主进程可直接向该进程发送信号并管理子进程树。

内容的提问来源于stack exchange，提问作者Adrian Sevcenco