#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/kprobes.h>
static int func_count = 0;
// 要跟踪的函数
static int example_function(int arg)
{
++func_count;
printk(KERN_INFO "Example function called with argument: %d\n", arg);
return arg + 1;
}
// Kprobe处理程序
static int handler_pre(struct kprobe *p, struct pt_regs *regs)
{
printk(KERN_INFO "Inside pre_handler: p->addr = 0x%p, ip = %lx\n", p->addr, regs->ip);
// 调试输出寄存器的值
printk(KERN_INFO "Register values: ax = %lx, bx = %lx, cx = %lx, dx = %lx\n",
regs->ax, regs->bx, regs->cx, regs->dx);
return 0;
}
static void handler_post(struct kprobe *p, struct pt_regs *regs, unsigned long flags)
{
printk(KERN_INFO "Inside post_handler: p->addr = 0x%p, flags = 0x%lx\n", p->addr, flags);
}
static struct kprobe kp = {
.symbol_name = "example_function", // 要跟踪的函数名
.pre_handler = handler_pre, // 函数执行前的处理程序
.post_handler = handler_post, // 函数执行后的处理程序
};
// 模块初始化函数
static int __init kprobe_debug_init(void)
{
int ret;
// 注册Kprobe
ret = register_kprobe(&kp);
if (ret < 0) {
printk(KERN_ERR "Failed to register kprobe: %d\n", ret);
return ret;
}
printk(KERN_INFO "Kprobe registered\n");
return 0;
}
// 模块清理函数
static void __exit kprobe_debug_exit(void)
{
// 注销Kprobe
unregister_kprobe(&kp);
printk(KERN_INFO "Kprobe unregistered\n");
// 输出函数调用次数
printk(KERN_INFO "Function example_function called %d times\n", func_count);
}
module_init(kprobe_debug_init);
module_exit(kprobe_debug_exit);
MODULE_LICENSE("GPL");`在这里插入代码片`
在这个例子中,我们跟踪了 example_function 函数,并在Kprobe的处理程序中添加了调试信息。在 handler_pre 处理程序中,我们输出了Kprobe的地址和调用指令指针的值,以及寄存器 ax、bx、cx 和 dx 的值。在模块清理函数中,我们输出了函数调用的次数。
使用此例子,当调用 example_function 函数时,Kprobe会在函数执行前后分别调用 handler_pre 和 handler_post 处理程序,并在内核日志中打印出相应的调试信息。最后,模块清理函数会输出函数调用的次数。
编译和加载此内核模块后,您可以通过调用 example_function 函数来触发Kprobe,并在内核日志中查看输出的调试信息和函数调用次数。
执行结果示例:
Inside pre_handler: p->addr = 0xffffffff814af000, ip = 00000000600a41a2
Register values: ax = 0000000000000000, bx = 0000000000000000, cx = 00000000ffffffff, dx = 000000009ffcfa12
Inside post_handler: p->addr = 0xffffffff814af000, flags = 0x0
Example function called with argument: 42
Function example_function called 1 times