xref: /openbmc/linux/arch/powerpc/kernel/uprobes.c (revision d2ba09c1)
1 /*
2  * User-space Probes (UProbes) for powerpc
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17  *
18  * Copyright IBM Corporation, 2007-2012
19  *
20  * Adapted from the x86 port by Ananth N Mavinakayanahalli <ananth@in.ibm.com>
21  */
22 #include <linux/kernel.h>
23 #include <linux/sched.h>
24 #include <linux/ptrace.h>
25 #include <linux/uprobes.h>
26 #include <linux/uaccess.h>
27 #include <linux/kdebug.h>
28 
29 #include <asm/sstep.h>
30 
31 #define UPROBE_TRAP_NR	UINT_MAX
32 
33 /**
34  * is_trap_insn - check if the instruction is a trap variant
35  * @insn: instruction to be checked.
36  * Returns true if @insn is a trap variant.
37  */
38 bool is_trap_insn(uprobe_opcode_t *insn)
39 {
40 	return (is_trap(*insn));
41 }
42 
43 /**
44  * arch_uprobe_analyze_insn
45  * @mm: the probed address space.
46  * @arch_uprobe: the probepoint information.
47  * @addr: vaddr to probe.
48  * Return 0 on success or a -ve number on error.
49  */
50 int arch_uprobe_analyze_insn(struct arch_uprobe *auprobe,
51 		struct mm_struct *mm, unsigned long addr)
52 {
53 	if (addr & 0x03)
54 		return -EINVAL;
55 
56 	return 0;
57 }
58 
59 /*
60  * arch_uprobe_pre_xol - prepare to execute out of line.
61  * @auprobe: the probepoint information.
62  * @regs: reflects the saved user state of current task.
63  */
64 int arch_uprobe_pre_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
65 {
66 	struct arch_uprobe_task *autask = &current->utask->autask;
67 
68 	autask->saved_trap_nr = current->thread.trap_nr;
69 	current->thread.trap_nr = UPROBE_TRAP_NR;
70 	regs->nip = current->utask->xol_vaddr;
71 
72 	user_enable_single_step(current);
73 	return 0;
74 }
75 
76 /**
77  * uprobe_get_swbp_addr - compute address of swbp given post-swbp regs
78  * @regs: Reflects the saved state of the task after it has hit a breakpoint
79  * instruction.
80  * Return the address of the breakpoint instruction.
81  */
82 unsigned long uprobe_get_swbp_addr(struct pt_regs *regs)
83 {
84 	return instruction_pointer(regs);
85 }
86 
87 /*
88  * If xol insn itself traps and generates a signal (SIGILL/SIGSEGV/etc),
89  * then detect the case where a singlestepped instruction jumps back to its
90  * own address. It is assumed that anything like do_page_fault/do_trap/etc
91  * sets thread.trap_nr != UINT_MAX.
92  *
93  * arch_uprobe_pre_xol/arch_uprobe_post_xol save/restore thread.trap_nr,
94  * arch_uprobe_xol_was_trapped() simply checks that ->trap_nr is not equal to
95  * UPROBE_TRAP_NR == UINT_MAX set by arch_uprobe_pre_xol().
96  */
97 bool arch_uprobe_xol_was_trapped(struct task_struct *t)
98 {
99 	if (t->thread.trap_nr != UPROBE_TRAP_NR)
100 		return true;
101 
102 	return false;
103 }
104 
105 /*
106  * Called after single-stepping. To avoid the SMP problems that can
107  * occur when we temporarily put back the original opcode to
108  * single-step, we single-stepped a copy of the instruction.
109  *
110  * This function prepares to resume execution after the single-step.
111  */
112 int arch_uprobe_post_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
113 {
114 	struct uprobe_task *utask = current->utask;
115 
116 	WARN_ON_ONCE(current->thread.trap_nr != UPROBE_TRAP_NR);
117 
118 	current->thread.trap_nr = utask->autask.saved_trap_nr;
119 
120 	/*
121 	 * On powerpc, except for loads and stores, most instructions
122 	 * including ones that alter code flow (branches, calls, returns)
123 	 * are emulated in the kernel. We get here only if the emulation
124 	 * support doesn't exist and have to fix-up the next instruction
125 	 * to be executed.
126 	 */
127 	regs->nip = utask->vaddr + MAX_UINSN_BYTES;
128 
129 	user_disable_single_step(current);
130 	return 0;
131 }
132 
133 /* callback routine for handling exceptions. */
134 int arch_uprobe_exception_notify(struct notifier_block *self,
135 				unsigned long val, void *data)
136 {
137 	struct die_args *args = data;
138 	struct pt_regs *regs = args->regs;
139 
140 	/* regs == NULL is a kernel bug */
141 	if (WARN_ON(!regs))
142 		return NOTIFY_DONE;
143 
144 	/* We are only interested in userspace traps */
145 	if (!user_mode(regs))
146 		return NOTIFY_DONE;
147 
148 	switch (val) {
149 	case DIE_BPT:
150 		if (uprobe_pre_sstep_notifier(regs))
151 			return NOTIFY_STOP;
152 		break;
153 	case DIE_SSTEP:
154 		if (uprobe_post_sstep_notifier(regs))
155 			return NOTIFY_STOP;
156 	default:
157 		break;
158 	}
159 	return NOTIFY_DONE;
160 }
161 
162 /*
163  * This function gets called when XOL instruction either gets trapped or
164  * the thread has a fatal signal, so reset the instruction pointer to its
165  * probed address.
166  */
167 void arch_uprobe_abort_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
168 {
169 	struct uprobe_task *utask = current->utask;
170 
171 	current->thread.trap_nr = utask->autask.saved_trap_nr;
172 	instruction_pointer_set(regs, utask->vaddr);
173 
174 	user_disable_single_step(current);
175 }
176 
177 /*
178  * See if the instruction can be emulated.
179  * Returns true if instruction was emulated, false otherwise.
180  */
181 bool arch_uprobe_skip_sstep(struct arch_uprobe *auprobe, struct pt_regs *regs)
182 {
183 	int ret;
184 
185 	/*
186 	 * emulate_step() returns 1 if the insn was successfully emulated.
187 	 * For all other cases, we need to single-step in hardware.
188 	 */
189 	ret = emulate_step(regs, auprobe->insn);
190 	if (ret > 0)
191 		return true;
192 
193 	return false;
194 }
195 
196 unsigned long
197 arch_uretprobe_hijack_return_addr(unsigned long trampoline_vaddr, struct pt_regs *regs)
198 {
199 	unsigned long orig_ret_vaddr;
200 
201 	orig_ret_vaddr = regs->link;
202 
203 	/* Replace the return addr with trampoline addr */
204 	regs->link = trampoline_vaddr;
205 
206 	return orig_ret_vaddr;
207 }
208 
209 bool arch_uretprobe_is_alive(struct return_instance *ret, enum rp_check ctx,
210 				struct pt_regs *regs)
211 {
212 	if (ctx == RP_CHECK_CHAIN_CALL)
213 		return regs->gpr[1] <= ret->stack;
214 	else
215 		return regs->gpr[1] < ret->stack;
216 }
217