Swift调用Python传递回调至布局函数时崩溃问题排查

你的问题核心是通过ctypes传入的Swift C风格回调没有持有Python全局解释器锁（GIL），导致回调内调用CPython API时无法访问正常的内部全局状态（比如你看到的SMALL_INT为空）。PythonKit在常规调用流程中会自动管理GIL，但ctypes回调不在这个管理链里，必须手动处理GIL的获取与释放。

修正方案 #

在回调函数内手动加GIL保护，确保所有CPython API操作都在锁的覆盖范围内：

import PythonKit

let nodeSizeFn: @convention(c) (UnsafeMutableRawPointer?) -> UnsafeMutableRawPointer? = { _ in
    // 获取GIL，确保当前线程能安全访问CPython内部状态
    let gilState = PyGILState_Ensure()
    defer {
        // 无论函数如何退出，都释放GIL，避免死锁
        PyGILState_Release(gilState)
    }
    
    // 在GIL保护下创建Python整数对象
    return PyInt_FromLong(30)
}

let nodeSizeFnPtr = unsafeBitCast(nodeSizeFn, to: UnsafeMutableRawPointer.self)
let nodeSizeFnAddr = UInt(bitPattern: nodeSizeFnPtr)

private let ctypes = Python.import("ctypes")
let sizefn = ctypes.CFUNCTYPE(ctypes.py_object)(nodeSizeFnAddr)

let (xs, ys, ws, hs) = graph.layout(sizefn).tuple4

关键说明 #

• PyGILState_Ensure()：自动获取GIL，如果当前线程未关联Python解释器，会自动初始化线程状态，是线程安全的GIL获取方式。
• defer块：保证GIL一定会被释放，就算回调内抛出异常也不会导致锁泄漏。
• 所有直接调用CPython C API的操作（比如PyInt_FromLong、PyTuple_New等）都必须放在GIL保护的代码块里，否则会触发未定义行为（比如段错误、内存损坏）。

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