kprobe/arch/dbi_kprobes.c

   1 /*
   2  *  Kernel Probes (KProbes)
   3  *  arch/<arch>/kernel/kprobes.c
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  18  *
  19  * Copyright (C) IBM Corporation, 2002, 2004
  20  */
  21
  22 /*
  23  *  Dynamic Binary Instrumentation Module based on KProbes
  24  *  modules/kprobe/arch/dbi_kprobes.c
  25  *
  26  * This program is free software; you can redistribute it and/or modify
  27  * it under the terms of the GNU General Public License as published by
  28  * the Free Software Foundation; either version 2 of the License, or
  29  * (at your option) any later version.
  30  *
  31  * This program is distributed in the hope that it will be useful,
  32  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  33  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  34  * GNU General Public License for more details.
  35  *
  36  * You should have received a copy of the GNU General Public License
  37  * along with this program; if not, write to the Free Software
  38  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  39  *
  40  * Copyright (C) Samsung Electronics, 2006-2010
  41  *
  42  * 2006-2007    Ekaterina Gorelkina <e.gorelkina@samsung.com>: initial implementation for ARM and MIPS
  43  * 2008-2009    Alexey Gerenkov <a.gerenkov@samsung.com> User-Space
  44  *              Probes initial implementation; Support x86/ARM/MIPS for both user and kernel spaces.
  45  * 2010         Ekaterina Gorelkina <e.gorelkina@samsung.com>: redesign module for separating core and arch parts
  46  *
  47
  48  */
  49
  50 #include "dbi_kprobes.h"
  51 #include "../dbi_kprobes.h"
  52 #include "asm/dbi_kprobes.h"
  53
  54 #include "../dbi_kdebug.h"
  55 #include "../dbi_insn_slots.h"
  56 #include "../dbi_kprobes_deps.h"
  57
  58 #include <linux/module.h>
  59
  60 typedef unsigned long (*fp_kallsyms_lookup_name_t) (const char *name);
  61 //export by swap_kprobes.ko
  62 extern fp_kallsyms_lookup_name_t lookup_name;
  63
  64 extern unsigned long sched_addr;
  65 extern unsigned long fork_addr;
  66
  67 extern struct hlist_head kprobe_insn_pages;
  68 extern struct hlist_head uprobe_insn_pages;
  69
  70
  71 static int ksyms = INVALID_VALUE;
  72 module_param(ksyms, uint, 0);
  73 MODULE_PARM_DESC(ksyms, "kallsyms_lookup_name address");
  74
  75 fp_kallsyms_lookup_name_t lookup_name;
  76 EXPORT_SYMBOL_GPL(lookup_name);
  77
  78 extern unsigned long (*kallsyms_search) (const char *name);
  79
  80 void arch_remove_kprobe (struct kprobe *p, struct task_struct *task)
  81 {
  82         // TODO: check boostable for x86 and MIPS
  83         if (p->tgid) {
  84 #ifdef CONFIG_ARM
  85                 free_insn_slot(&uprobe_insn_pages, task, p->ainsn.insn_arm);
  86                 free_insn_slot(&uprobe_insn_pages, task, p->ainsn.insn_thumb);
  87 #else /* CONFIG_ARM */
  88                 free_insn_slot(&uprobe_insn_pages, task, p->ainsn.insn);
  89 #endif /* CONFIG_ARM */
  90         } else {
  91                 free_insn_slot(&kprobe_insn_pages, NULL, p->ainsn.insn);
  92         }
  93 }
  94
  95 void arch_arm_uprobe (struct kprobe *p, struct task_struct *tsk)
  96 {
  97         kprobe_opcode_t insn = BREAKPOINT_INSTRUCTION;
  98
  99         if (!write_proc_vm_atomic (tsk, (unsigned long) p->addr, &insn, sizeof (insn)))
 100                 panic ("failed to write memory %p!\n", p->addr);
 101 }
 102
 103 void arch_arm_uretprobe (struct kretprobe *p, struct task_struct *tsk)
 104 {
 105 }
 106
 107 void arch_disarm_uprobe (struct kprobe *p, struct task_struct *tsk)
 108 {
 109         if (!write_proc_vm_atomic (tsk, (unsigned long) p->addr, &p->opcode, sizeof (p->opcode))) {
 110                 panic ("failed to write memory: tgid=%u, addr=%p!\n", tsk->tgid, p->addr);
 111         }
 112 }
 113 EXPORT_SYMBOL_GPL(arch_disarm_uprobe);
 114
 115 void arch_disarm_uretprobe (struct kretprobe *p, struct task_struct *tsk)
 116 {
 117 }
 118
 119 int arch_init_module_dependencies()
 120 {
 121         lookup_name = (void *) ksyms;
 122         kallsyms_search = lookup_name;//(void *) ksyms;
 123         DBPRINTF ("kallsyms=0x%08x\n", kallsyms_search);
 124
 125         sched_addr = kallsyms_search("__switch_to");//"schedule");
 126         fork_addr = kallsyms_search("do_fork");
 127
 128         init_module_dependencies();
 129
 130         return asm_init_module_dependencies();
 131 }