如何在Linux内核模块中实现类似sigaction的自定义异常处理程序

内核态没有直接对标用户态sigaction的API，因为内核异常处理是全局/Per-CPU层面的，而非Per-Process，但你可以通过以下几种机制，处理用户传入无效指针引发的页错误等异常，避免用户进程直接被终止：

1. 利用内核异常表（Exception Table） #

内核允许模块通过__ex_table注册异常处理入口，当指定地址触发页错误时，会跳转到你定义的处理函数。这是内核模块处理自身代码异常的标准方式之一。

示例代码框架：

#include <linux/module.h>
#include <linux/uaccess.h>
#include <linux/exception.h>

// 可能触发页错误的函数（比如处理用户传入的指针）
static ssize_t unsafe_user_copy(char __user *user_buf, char *kernel_buf, size_t len) {
    return copy_from_user(kernel_buf, user_buf, len);
}

// 自定义页错误处理函数
static int handle_custom_fault(struct pt_regs *regs, unsigned long error_code) {
    // 确认异常是来自我们的目标函数（需要计算触发异常的指令偏移）
    if (regs->ip == (unsigned long)unsafe_user_copy + 0x10) { // 示例偏移，需根据实际汇编调整
        // 自定义处理：设置函数返回值为EFAULT，跳过出错指令
        regs->ax = -EFAULT;
        regs->ip += 2; // 跳过引发异常的汇编指令长度
        return 1; // 告知内核异常已处理，无需默认终止流程
    }
    return 0; // 非目标异常，交给内核默认处理
}

// 注册异常表项
static struct exception_table_entry my_ex_table[] __used = {
    {
        .insn = (unsigned long)unsafe_user_copy + 0x10, // 触发异常的指令地址
        .fixup = (unsigned long)handle_custom_fault,    // 处理函数地址
    },
};

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Custom fault handler for user parameter errors");

注意：你需要通过反汇编模块代码，准确找到copy_from_user内部触发页错误的指令偏移，否则无法正确匹配异常。

2. 前置检查+错误返回（有限场景） #

虽然你提到无法完全软件层面检查，但access_ok可以做初步的用户地址合法性校验（比如地址是否属于用户空间），配合get_user/put_user的返回值处理，能覆盖大部分常见的无效指针场景：

char user_val;
// 先检查地址是否属于用户空间可读范围
if (!access_ok(VERIFY_READ, user_ptr, sizeof(char))) {
    return -EFAULT;
}
// 尝试读取，失败则直接返回错误
if (get_user(user_val, user_ptr)) {
    pr_warn("Failed to read from user pointer\n");
    return -EFAULT;
}

这种方式的局限是：无法处理access_ok通过但实际访问时触发页错误的情况（比如地址存在但无权限），此时仍会进入内核默认处理流程。

3. 替换Per-CPU页错误处理钩子（高风险） #

部分架构（如x86）允许通过set_page_fault_handler注册全局页错误处理函数，拦截所有页错误并自定义处理。但这属于底层操作，会影响整个系统的稳定性，仅适合调试或特殊场景：

#include <asm/traps.h>

static void (*original_pf_handler)(struct pt_regs *, unsigned long);

static void custom_pf_handler(struct pt_regs *regs, unsigned long error_code) {
    // 判断异常是否来自当前模块的代码
    if (within_module_core(regs->ip, THIS_MODULE)) {
        // 自定义处理逻辑，比如记录日志、返回错误
        pr_info("Handled page fault from our module\n");
        return;
    }
    // 非当前模块的异常，交给原处理函数
    original_pf_handler(regs, error_code);
}

static int __init my_module_init(void) {
    original_pf_handler = set_page_fault_handler(custom_pf_handler);
    return 0;
}

static void __exit my_module_exit(void) {
    set_page_fault_handler(original_pf_handler);
}

module_init(my_module_init);
module_exit(my_module_exit);
MODULE_LICENSE("GPL");

注意：不同架构的API可能不同，且这种方式可能干扰内核正常的页错误处理（比如缺页异常），务必谨慎使用。

核心注意事项 #

• 内核异常处理函数必须保证原子性，禁止调用可能睡眠的API（比如kmalloc、mutex_lock），否则会引发死锁或系统崩溃。
• 自定义处理仅能拦截来自你模块代码引发的异常，无法处理其他模块或内核本身的异常。
• 优先通过access_ok等前置检查减少异常发生，异常处理仅作为最后兜底手段。

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