uprobe/swap_uprobes.h

   1 #ifndef _DBI_UPROBES_H
   2 #define _DBI_UPROBES_H
   3
   4 /*
   5  *  Dynamic Binary Instrumentation Module based on KProbes
   6  *  modules/kprobe/dbi_uprobes.h
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  21  *
  22  * Copyright (C) Samsung Electronics, 2006-2010
  23  *
  24  * 2008-2009    Alexey Gerenkov <a.gerenkov@samsung.com> User-Space
  25  *              Probes initial implementation; Support x86/ARM/MIPS for both user and kernel spaces.
  26  * 2010         Ekaterina Gorelkina <e.gorelkina@samsung.com>: redesign module for separating core and arch parts
  27  *
  28  */
  29
  30 #include "dbi_kprobes.h"
  31
  32 struct uprobe {
  33         struct kprobe kp;
  34         struct task_struct *task;
  35         struct slot_manager *sm;
  36 };
  37
  38 typedef unsigned long (*uprobe_pre_entry_handler_t)(void *priv_arg, struct pt_regs * regs);
  39
  40 struct ujprobe {
  41         struct uprobe up;
  42         /* probe handling code to jump to */
  43         void *entry;
  44         // handler whichw willb bec called before 'entry'
  45         uprobe_pre_entry_handler_t pre_entry;
  46         void *priv_arg;
  47 };
  48
  49 struct uretprobe_instance;
  50
  51 typedef int (*uretprobe_handler_t)(struct uretprobe_instance *, struct pt_regs *, void *);
  52
  53 /*
  54  * Function-return probe -
  55  * Note:
  56  * User needs to provide a handler function, and initialize maxactive.
  57  * maxactive - The maximum number of instances of the probed function that
  58  * can be active concurrently.
  59  * nmissed - tracks the number of times the probed function's return was
  60  * ignored, due to maxactive being too low.
  61  *
  62  */
  63 struct uretprobe {
  64         struct uprobe up;
  65         uretprobe_handler_t handler;
  66         void *priv_arg;
  67         int maxactive;
  68         int nmissed;
  69         struct hlist_head free_instances;
  70         struct hlist_head used_instances;
  71 };
  72
  73 struct uretprobe_instance {
  74         /* either on free list or used list */
  75         struct hlist_node uflist;
  76         struct hlist_node hlist;
  77         struct uretprobe *rp;
  78         kprobe_opcode_t *ret_addr;
  79         kprobe_opcode_t *sp;
  80         struct task_struct *task;
  81 };
  82
  83 int dbi_register_uprobe(struct uprobe *p, int atomic);
  84 void dbi_unregister_uprobe(struct uprobe *p, int atomic);
  85
  86 int dbi_register_ujprobe(struct ujprobe *jp, int atomic);
  87 void dbi_unregister_ujprobe(struct ujprobe *jp, int atomic);
  88
  89 int dbi_register_uretprobe(struct uretprobe *rp, int atomic);
  90 void dbi_unregister_uretprobe(struct uretprobe *rp, int atomic);
  91
  92 void dbi_unregister_all_uprobes(struct task_struct *task, int atomic);
  93
  94 void dbi_uprobe_return(void);
  95 struct kprobe *get_ukprobe(void *addr, pid_t tgid);
  96 struct kprobe *get_ukprobe_by_insn_slot(void *addr, pid_t tgid, struct pt_regs *regs);
  97
  98 static inline struct uprobe *kp2up(struct kprobe *p)
  99 {
 100         return container_of(p, struct uprobe, kp);
 101 }
 102
 103 static inline struct kprobe *up2kp(struct uprobe *p)
 104 {
 105         return &p->kp;
 106 }
 107
 108 void disarm_uprobe(struct uprobe *p);
 109
 110 int trampoline_uprobe_handler(struct kprobe *p, struct pt_regs *regs);
 111
 112 #endif /*  _DBI_UPROBES_H */