CGO如何捕获SIGTERM/SIGABRT避免C库断言失败导致Go程序退出

核心问题原因 #

你当前的实现无法生效有3个核心原因：

• signal.Notify注册时机错误：你将注册逻辑放到了执行C代码的goroutine内部，C代码触发断言发送信号时，可能还未完成信号监听注册
• Go的信号监听是进程级别的，并非goroutine私有，单独在子goroutine注册无法绑定该goroutine内的C代码产生的信号
• C侧raise发送的信号会直接递交给整个进程，就算捕获成功，C库的执行栈已经处于异常状态，无法自动恢复执行流

最优方案：C侧使用setjmp/longjmp接管断言流 #

完全不需要走信号捕获逻辑，直接在C层接管断言失败的执行流，跳回正常调用路径，是稳定性最高的方案：

第一步：修改CGO头定义，添加跳转逻辑 #

/*
#include <setjmp.h>

// 线程局部存储跳转缓冲区，避免多goroutine并发调用互相干扰
__thread jmp_buf poly2tri_jmp_buf;

// 重定义断言宏，失败直接跳转
#define MPE_Assert(_Expression) (void) ((!!(_Expression)) || (longjmp(poly2tri_jmp_buf, 1)))

// 替换原来的poly2Tri调用，声明包装函数
int wrapped_poly2tri(/* 填写你原来poly2Tri需要的参数 */);
*/
import "C"

第二步：实现C侧包装函数 #

在你的C代码文件中实现包装逻辑，正常调用返回0，断言触发返回1：

int wrapped_poly2tri(/* 你的参数 */) {
    if (setjmp(poly2tri_jmp_buf) == 0) {
        // 调用原始的poly2Tri逻辑
        original_poly2Tri(/* 传递参数 */);
        return 0;
    } else {
        // 断言触发，返回错误码
        // 注意：此处需要手动释放C库调用过程中已经分配的内存，避免内存泄漏
        return 1;
    }
}

第三步：Go侧直接调用包装函数 #

不需要复杂的信号监听逻辑，直接判断返回值即可：

func Poly2Tri(verts [][2]float32, holes [][][2]float32) [][2]float32 {
    // 转换参数为C类型后调用wrapped_poly2tri
    ret := C.wrapped_poly2tri(/* 传递转换后的参数 */)
    if ret != 0 {
        return [][2]float32{}
    }
    // 转换返回值为Go类型后返回
    return convertedResult
}

备选方案：修复现有信号捕获逻辑 #

如果坚持使用信号方案，按照如下方式修改代码即可生效：

/*
#include <signal.h>
#define MPE_Assert(_Expression) (void) ((!!(_Expression)) || (raise(SIGTERM)))
*/
import "C"
import (
    "os"
    "os/signal"
    "syscall"
)

func Poly2Tri(verts [][2]float32, holes [][][2]float32) [][2]float32 {
    sig := make(chan os.Signal, 1)
    // 提前在主逻辑注册信号监听，指定监听SIGTERM
    signal.Notify(sig, syscall.SIGTERM)
    defer signal.Stop(sig)

    result := make(chan [][2]float32, 1)
    go func() {
        defer func() {
            _ = recover()
        }()
        res := poly2Tri(verts, holes)
        result <- res
    }()

    select {
    case res := <-result:
        return res
    case <-sig:
        return [][2]float32{}
    }
}

注意：信号方案存在并发冲突风险，多goroutine同时调用时可能出现信号被错误捕获的问题，仅推荐单线程调用场景使用。

额外优化建议 #

• 调用C库之前提前对输入参数做合法性校验，尽量避免触发断言
• 如果开启多并发调用，注意对C库的调用加锁，避免多线程操作共享变量导致的异常

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