hc
2023-10-25 6c2073b7aa40e29d0eca7d571dd7bc590c7ecaa7
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/* SPDX-License-Identifier: GPL-2.0 */
#ifndef __X86_KERNEL_KPROBES_COMMON_H
#define __X86_KERNEL_KPROBES_COMMON_H
 
/* Kprobes and Optprobes common header */
 
#include <asm/asm.h>
 
#ifdef CONFIG_FRAME_POINTER
# define SAVE_RBP_STRING "    push %" _ASM_BP "\n" \
            "    mov  %" _ASM_SP ", %" _ASM_BP "\n"
#else
# define SAVE_RBP_STRING "    push %" _ASM_BP "\n"
#endif
 
#ifdef CONFIG_X86_64
#define SAVE_REGS_STRING            \
   /* Skip cs, ip, orig_ax. */        \
   "    subq $24, %rsp\n"        \
   "    pushq %rdi\n"            \
   "    pushq %rsi\n"            \
   "    pushq %rdx\n"            \
   "    pushq %rcx\n"            \
   "    pushq %rax\n"            \
   "    pushq %r8\n"            \
   "    pushq %r9\n"            \
   "    pushq %r10\n"            \
   "    pushq %r11\n"            \
   "    pushq %rbx\n"            \
   SAVE_RBP_STRING                \
   "    pushq %r12\n"            \
   "    pushq %r13\n"            \
   "    pushq %r14\n"            \
   "    pushq %r15\n"
#define RESTORE_REGS_STRING            \
   "    popq %r15\n"            \
   "    popq %r14\n"            \
   "    popq %r13\n"            \
   "    popq %r12\n"            \
   "    popq %rbp\n"            \
   "    popq %rbx\n"            \
   "    popq %r11\n"            \
   "    popq %r10\n"            \
   "    popq %r9\n"            \
   "    popq %r8\n"            \
   "    popq %rax\n"            \
   "    popq %rcx\n"            \
   "    popq %rdx\n"            \
   "    popq %rsi\n"            \
   "    popq %rdi\n"            \
   /* Skip orig_ax, ip, cs */        \
   "    addq $24, %rsp\n"
#else
#define SAVE_REGS_STRING            \
   /* Skip cs, ip, orig_ax and gs. */    \
   "    subl $16, %esp\n"        \
   "    pushl %fs\n"            \
   "    pushl %es\n"            \
   "    pushl %ds\n"            \
   "    pushl %eax\n"            \
   SAVE_RBP_STRING                \
   "    pushl %edi\n"            \
   "    pushl %esi\n"            \
   "    pushl %edx\n"            \
   "    pushl %ecx\n"            \
   "    pushl %ebx\n"
#define RESTORE_REGS_STRING            \
   "    popl %ebx\n"            \
   "    popl %ecx\n"            \
   "    popl %edx\n"            \
   "    popl %esi\n"            \
   "    popl %edi\n"            \
   "    popl %ebp\n"            \
   "    popl %eax\n"            \
   /* Skip ds, es, fs, gs, orig_ax, and ip. Note: don't pop cs here*/\
   "    addl $24, %esp\n"
#endif
 
/* Ensure if the instruction can be boostable */
extern int can_boost(struct insn *insn, void *orig_addr);
/* Recover instruction if given address is probed */
extern unsigned long recover_probed_instruction(kprobe_opcode_t *buf,
                    unsigned long addr);
/*
 * Copy an instruction and adjust the displacement if the instruction
 * uses the %rip-relative addressing mode.
 */
extern int __copy_instruction(u8 *dest, u8 *src, u8 *real, struct insn *insn);
 
/* Generate a relative-jump/call instruction */
extern void synthesize_reljump(void *dest, void *from, void *to);
extern void synthesize_relcall(void *dest, void *from, void *to);
 
#ifdef    CONFIG_OPTPROBES
extern int setup_detour_execution(struct kprobe *p, struct pt_regs *regs, int reenter);
extern unsigned long __recover_optprobed_insn(kprobe_opcode_t *buf, unsigned long addr);
#else    /* !CONFIG_OPTPROBES */
static inline int setup_detour_execution(struct kprobe *p, struct pt_regs *regs, int reenter)
{
   return 0;
}
static inline unsigned long __recover_optprobed_insn(kprobe_opcode_t *buf, unsigned long addr)
{
   return addr;
}
#endif
 
#endif