4. kprobe与中断上下文

中断上下文是Linux内核中一种特殊的执行环境，与进程上下文有着本质区别。当kprobe探针插入到中断处理路径时，探针处理函数将在中断上下文中执行，这带来了严格的限制和独特的行为特征。理解这些概念是安全使用kprobe进行驱动调试的前提。

4.1 中断上下文定义

中断上下文（Interrupt Context）是指CPU响应硬件中断或软件中断时，内核所处的执行状态。在这种状态下，当前执行路径不属于任何用户进程，也不属于内核线程，而是直接服务于中断处理程序（ISR）或底半部机制（如softirq、tasklet）。

对比维度	进程上下文	中断上下文
关联进程	有明确的current进程	无关联进程，current指向被中断的进程
可睡眠	可以（调用schedule()）	禁止（会导致内核死锁）
可访问用户空间	可以（copy_from/to_user）	禁止（页表可能无效）
持有锁	可以使用互斥锁（mutex）	只能使用自旋锁（spinlock）
栈空间	进程内核栈（通常8KB/16KB）	中断栈（独立，通常4KB/8KB）
抢占状态	可被抢占（preempt_count=0）	不可抢占（preempt_count>0）

4.2 kprobe在中断上下文中的行为

当kprobe注册的探测点位于中断处理函数内部时，探针的pre_handler和post_handler将在中断上下文中被调用。kprobe框架本身对中断上下文有特殊的处理机制：

• 单步执行模式：kprobe在中断上下文中仍然使用单步调试异常（int1）来执行原始指令，但会临时禁用当前CPU的中断，防止嵌套中断导致递归。
• 中断屏蔽：在pre_handler执行期间，本地CPU的中断被屏蔽（local_irq_save），确保探针处理函数不会被同一CPU上的其他中断打断。
• 栈使用：kprobe在中断上下文中使用当前中断栈，不会切换到进程内核栈。因此探针处理函数必须严格控制栈使用量。
• 递归保护：kprobe内部维护每CPU的递归计数器，防止在同一个CPU上递归触发kprobe导致栈溢出。

4.3 中断上下文限制

在中断上下文中编写kprobe处理函数时，必须严格遵守以下限制。违反这些限制可能导致内核崩溃、死锁或数据损坏。

4.3.1 可调用函数白名单

函数类别	允许调用	禁止调用
内存分配	kmalloc(GFP_ATOMIC)、kzalloc(GFP_ATOMIC)	kmalloc(GFP_KERNEL)、vmalloc
锁操作	spin_lock_irqsave、atomic_*	mutex_lock、down_interruptible
调度相关	无（禁止任何调度）	schedule()、msleep()、wait_event
用户空间访问	无	copy_from_user、put_user
调试输出	printk_deferred()、trace_printk()	printk()（可能死锁）、WARN_ON
时间相关	ktime_get()、jiffies 读取	do_gettimeofday()（可能睡眠）

4.3.2 栈使用限制

中断栈的大小通常为4KB（x86_64）或8KB（ARM64）。kprobe处理函数本身会占用一部分栈空间，加上中断处理函数的已有栈使用，必须确保总栈使用不超过中断栈大小。建议：

• 避免在kprobe处理函数中定义大型局部数组（如 char buf[1024]）。
• 避免深层函数调用链（递归或超过3层调用）。
• 使用 __builtin_frame_address() 在调试时检查栈使用量。

4.3.3 嵌套与递归限制

kprobe框架虽然提供了递归保护，但以下情况仍需注意：

1. 同CPU嵌套：如果在kprobe处理函数中再次触发同一CPU上的断点（例如访问了被kprobe监控的内存区域），kprobe会检测到递归并跳过处理，但会丢失该次探测数据。
2. 跨CPU并发：多个CPU可能同时进入同一个kprobe处理函数，必须使用每CPU变量或原子操作保护共享数据。
3. NMI嵌套：如果kprobe插入到NMI（不可屏蔽中断）处理路径中，需要额外小心，因为NMI可以中断普通中断上下文。

4.3.4 时间约束

中断上下文中的kprobe处理函数必须快速执行，因为它在中断屏蔽状态下运行。过长的执行时间会导致：

• 中断延迟增加：同CPU上的其他中断被延迟处理，可能导致设备超时。
• 实时性破坏：在实时系统中，过长的中断屏蔽会导致任务错过截止时间。
• 看门狗超时：如果中断上下文执行超过1秒（取决于配置），内核软锁检测（soft lockup）可能触发。

4.3.5 代码示例：安全的中断上下文kprobe处理函数

#include <linux/kprobes.h>
#include <linux/ktime.h>
#include <linux/percpu.h>

/* 每CPU变量，避免锁竞争 */
static DEFINE_PER_CPU(u64, irq_entry_time);

/* 中断上下文中的kprobe pre_handler */
static int __kprobes irq_handler_pre(struct kprobe *p, struct pt_regs *regs)
{
    u64 now;
    unsigned long flags;

    /* 保存当前时间戳到每CPU变量 */
    now = ktime_get_ns();
    this_cpu_write(irq_entry_time, now);

    /* 使用原子操作更新全局计数器 */
    atomic64_inc(&probe_hit_count);

    /* 注意：不能调用printk！使用trace_printk替代 */
    trace_printk("kprobe hit at %pF, CPU=%d\n",
                 (void *)regs->ip, smp_processor_id());

    return 0;  /* 0表示继续正常执行 */
}

/* 注册kprobe结构 */
static struct kprobe irq_kp = {
    .symbol_name    = "handle_irq_event_percpu",
    .pre_handler    = irq_handler_pre,
};

static int __init kprobe_init(void)
{
    int ret;

    ret = register_kprobe(&irq_kp);
    if (ret < 0) {
        pr_err("register_kprobe failed: %d\n", ret);
        return ret;
    }
    pr_info("kprobe registered on %s\n", irq_kp.symbol_name);
    return 0;
}

static void __exit kprobe_exit(void)
{
    unregister_kprobe(&irq_kp);
    pr_info("kprobe unregistered\n");
}

module_init(kprobe_init);
module_exit(kprobe_exit);
MODULE_LICENSE("GPL");