From 222257a20141dd947de933b0da7073d15bbc0db3 Mon Sep 17 00:00:00 2001 From: Vyacheslav Cherkashin Date: Thu, 14 Feb 2013 20:41:23 +0400 Subject: [PATCH] check for correct values --- kprobe/arch/dbi_kprobes.c | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/kprobe/arch/dbi_kprobes.c b/kprobe/arch/dbi_kprobes.c index c923035..959d1eb 100644 --- a/kprobe/arch/dbi_kprobes.c +++ b/kprobe/arch/dbi_kprobes.c @@ -106,11 +106,22 @@ void arch_disarm_uretprobe (struct kretprobe *p, struct task_struct *tsk) int arch_init_module_dependencies(void) { + int ret; + sched_addr = swap_ksyms("__switch_to"); fork_addr = swap_ksyms("do_fork"); exit_addr = swap_ksyms("do_exit"); - init_module_dependencies(); + if (sched_addr == NULL || + fork_addr == NULL || + exit_addr == NULL) { + return -ESRCH; + } + + ret = init_module_dependencies(); + if (ret) { + return ret; + } return asm_init_module_dependencies(); } -- 2.7.4