如何在内存分配失败时直接终止Python进程而非抛出MemoryError

方法1：用ctypes劫持malloc函数强制退出 #

Python可以通过ctypes直接修改系统C库的malloc行为，在它返回NULL时直接终止进程，跳过Python的异常抛出逻辑。你可以把这段代码放在第三方程序的启动脚本最开头：

import ctypes
import sys

# 加载系统默认C库
libc = ctypes.CDLL(None)

# 保存原始malloc函数指针
original_malloc = libc.malloc

# 定义自定义的malloc钩子
def custom_malloc(size):
    mem_ptr = original_malloc(size)
    if not mem_ptr:
        # 直接调用系统exit终止进程，不触发Python异常
        libc.exit(1)
    return mem_ptr

# 替换系统malloc为自定义钩子
libc.malloc = ctypes.CFUNCTYPE(ctypes.c_void_p, ctypes.c_size_t)(custom_malloc)

这样只要malloc分配失败返回NULL，进程会立刻退出，不会进入Python的异常处理流程。

方法2：编译定制版Python解释器 #

如果能重新编译Python，直接修改源码里的内存分配逻辑是最彻底的方案：

• 找到Python源码中处理malloc失败的位置（比如Objects/obmalloc.c里的内存分配代码）
• 把原本抛出MemoryError的代码替换成直接调用abort()或exit()
• 编译生成定制版Python，用它来运行第三方程序

这种方式不会有钩子带来的兼容性问题，但需要具备Python源码编译的能力。

注意事项 #

• 钩子方法可能会和Python内部的内存管理机制产生冲突，建议先在测试环境验证后再投入生产
• 强制终止进程会丢失所有未保存的数据，确认你的业务场景允许这种无预警的退出

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