用 Go 编写的 Python 扩展会忽略 Ctrl+C

PHP小编苹果发现，使用Go编写的python扩展在处理Ctrl+C信号时可能会出现问题。通常情况下，当我们在命令行中按下Ctrl+C时，会发送一个SIGINT信号给正在运行的程序，以请求其停止执行。然而，使用Go编写的Python扩展似乎会忽略这个信号，导致程序无法正常停止。这个问题对于使用Go编写Python扩展的开发者来说是一个潜在的隐患，需要特别注意。所以，在开发过程中，我们应该寻找解决方案来解决这个问题，以确保程序的正常退出。

问题内容

我正在使用一个用go编写的python驱动程序，该驱动程序连接到某个服务并进行一些处理。我们发现服务的“不健康”实例存在问题，这导致驱动程序卡住并且无法终止，除非我们 kill 进程。

做了一些实验，我发现当执行用 go 编写的扩展时，程序会忽略发出的 ctrl+c 命令，直到控制权返回到 python，而使用 c 编写的扩展则不会发生这种情况，其中在执行时引发 keyboardinterrupt c代码。我的问题是为什么会发生这种情况以及是否有办法规避这个问题或发出某种超时。尝试使用 threading.timer 引发异常来模拟超时，但问题是该异常是在它自己的线程中引发的，并且不会中断主线程。在python（cpython）3.9和3.10中测试。至于我写的poc扩展，go版本是1.20.4，c编译器是9.4.0。

我会在下面留下一个小poc。

python 代码：

import ctypes

import ctypes
go_library = ctypes.cdll.loadlibrary('./go_library.so')
hello_go = go_library.helloworld

c_library = ctypes.cdll.loadlibrary("./c_library.so")
hello_c = c_library.helloworld

try:
    print("calling golang code")
    hello_go()

    print("calling c code")
    hello_c()
except keyboardinterrupt:
    print("ctrl+c issued dd:")
finally:
    print("done")

go 扩展

package main

import (
   "c"
   "log"
   "time"
)

func helloworld(){
   log.println("hello world")
   time.sleep(10 * time.second)
   log.println("done sleeping")
}

func main(){
}

c 扩展

#include 

int helloWorld() {
    printf("Hello from C\n");
    sleep(10);
    printf("Done sleeping from C\n");

    return 0;
}

解决方法

我也遇到过同样的情况。经过一番研究，我找到了解决方案。

根据这个答案：

python 在 sigint 上安装了一个信号处理程序，它只是设置一个由主解释器循环检查的标志。为了使该处理程序正常工作，python 解释器必须运行 python 代码。

似乎在运行golang代码时，golang调用python安装的sigint信号处理程序来设置标志。因此，当运行回 python 时，会检测到键盘中断。

来自 golang 文档：

（当非go程序调用go代码时） 如果为异步信号调用 notify，则会为该信号安装 go 信号处理程序。如果稍后对该信号调用 reset，则该信号的原始处理将被重新安装，恢复非 go 信号处理程序（如果有）。

因此，下面的代码将在 go 代码中捕获 sigint，并在 sigint 时立即返回该函数。

func helloWorld() {
    c := make(chan os.Signal)
    signal.Notify(c, syscall.SIGINT)

    // the original handling for that signal will be reinstalled, restoring the non-Go signal handler if any.
    defer signal.Reset(syscall.SIGINT)

    go func() {
        log.Println("Hello World")
        time.Sleep(10 * time.Second)
        log.Println("Done sleeping")
        c <- nil

    }()

    // wait go routine to finish or signal received
    <-c
}

以上就是用 Go 编写的 Python 扩展会忽略 Ctrl+C的详细内容，更多请关注编程网其它相关文章！