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