Add find_kprobe_{insn,optinsn}_slot_entry() functions to find
corresponding entry address of the kprobe instrurction buffer
which includes given address.

Signed-off-by: Masami Hiramatsu <mhira...@kernel.org>
---
 include/linux/kprobes.h |    8 ++++++++
 kernel/kprobes.c        |   25 ++++++++++++++++++-------
 2 files changed, 26 insertions(+), 7 deletions(-)

diff --git a/include/linux/kprobes.h b/include/linux/kprobes.h
index f530f82a046d..08adfd6cf562 100644
--- a/include/linux/kprobes.h
+++ b/include/linux/kprobes.h
@@ -305,6 +305,8 @@ extern void __free_insn_slot(struct kprobe_insn_cache *c,
 /* sleep-less address checking routine  */
 extern bool __is_insn_slot_addr(struct kprobe_insn_cache *c,
                                unsigned long addr);
+extern unsigned long
+__find_insn_slot_entry(struct kprobe_insn_cache *c, unsigned long addr);
 
 #define DEFINE_INSN_CACHE_OPS(__name)                                  \
 extern struct kprobe_insn_cache kprobe_##__name##_slots;               \
@@ -322,6 +324,12 @@ static inline void free_##__name##_slot(kprobe_opcode_t 
*slot, int dirty)\
 static inline bool is_kprobe_##__name##_slot(unsigned long addr)       \
 {                                                                      \
        return __is_insn_slot_addr(&kprobe_##__name##_slots, addr);     \
+}                                                                      \
+                                                                       \
+static inline unsigned long                                            \
+find_kprobe_##__name##_slot_entry(unsigned long addr)                  \
+{                                                                      \
+       return __find_insn_slot_entry(&kprobe_##__name##_slots, addr);  \
 }
 #define KPROBE_INSN_PAGE_SYM           "kprobe_insn_page"
 #define KPROBE_OPTINSN_PAGE_SYM                "kprobe_optinsn_page"
diff --git a/kernel/kprobes.c b/kernel/kprobes.c
index 4ce3e6f5d28d..b62635969fa6 100644
--- a/kernel/kprobes.c
+++ b/kernel/kprobes.c
@@ -277,26 +277,37 @@ void __free_insn_slot(struct kprobe_insn_cache *c,
 }
 
 /*
- * Check given address is on the page of kprobe instruction slots.
- * This will be used for checking whether the address on a stack
- * is on a text area or not.
+ * Find the entry address of the kprobe instruction slots where the
+ * @addr points.
  */
-bool __is_insn_slot_addr(struct kprobe_insn_cache *c, unsigned long addr)
+unsigned long
+__find_insn_slot_entry(struct kprobe_insn_cache *c, unsigned long addr)
 {
        struct kprobe_insn_page *kip;
-       bool ret = false;
+       unsigned long entry = 0, index;
 
        rcu_read_lock();
        list_for_each_entry_rcu(kip, &c->pages, list) {
                if (addr >= (unsigned long)kip->insns &&
                    addr < (unsigned long)kip->insns + PAGE_SIZE) {
-                       ret = true;
+                       index = (addr - (unsigned long)kip->insns) / 
c->insn_size;
+                       entry = (unsigned long)kip->insns + (index * 
c->insn_size);
                        break;
                }
        }
        rcu_read_unlock();
 
-       return ret;
+       return entry;
+}
+
+/*
+ * Check given address is on the page of kprobe instruction slots.
+ * This will be used for checking whether the address on a stack
+ * is on a text area or not.
+ */
+bool __is_insn_slot_addr(struct kprobe_insn_cache *c, unsigned long addr)
+{
+       return __find_insn_slot_entry(c, addr) != 0;
 }
 
 int kprobe_cache_get_kallsym(struct kprobe_insn_cache *c, unsigned int *symnum,

Reply via email to