xref: /openbmc/linux/arch/s390/kernel/uprobes.c (revision 8d1a2427)
1 /*
2  *  User-space Probes (UProbes) for s390
3  *
4  *    Copyright IBM Corp. 2014
5  *    Author(s): Jan Willeke,
6  */
7 
8 #include <linux/uaccess.h>
9 #include <linux/uprobes.h>
10 #include <linux/compat.h>
11 #include <linux/kdebug.h>
12 #include <asm/switch_to.h>
13 #include <asm/facility.h>
14 #include <asm/kprobes.h>
15 #include <asm/dis.h>
16 #include "entry.h"
17 
18 #define	UPROBE_TRAP_NR	UINT_MAX
19 
20 int arch_uprobe_analyze_insn(struct arch_uprobe *auprobe, struct mm_struct *mm,
21 			     unsigned long addr)
22 {
23 	return probe_is_prohibited_opcode(auprobe->insn);
24 }
25 
26 int arch_uprobe_pre_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
27 {
28 	if (psw_bits(regs->psw).eaba == PSW_AMODE_24BIT)
29 		return -EINVAL;
30 	if (!is_compat_task() && psw_bits(regs->psw).eaba == PSW_AMODE_31BIT)
31 		return -EINVAL;
32 	clear_pt_regs_flag(regs, PIF_PER_TRAP);
33 	auprobe->saved_per = psw_bits(regs->psw).r;
34 	auprobe->saved_int_code = regs->int_code;
35 	regs->int_code = UPROBE_TRAP_NR;
36 	regs->psw.addr = current->utask->xol_vaddr;
37 	set_tsk_thread_flag(current, TIF_UPROBE_SINGLESTEP);
38 	update_cr_regs(current);
39 	return 0;
40 }
41 
42 bool arch_uprobe_xol_was_trapped(struct task_struct *tsk)
43 {
44 	struct pt_regs *regs = task_pt_regs(tsk);
45 
46 	if (regs->int_code != UPROBE_TRAP_NR)
47 		return true;
48 	return false;
49 }
50 
51 static int check_per_event(unsigned short cause, unsigned long control,
52 			   struct pt_regs *regs)
53 {
54 	if (!(regs->psw.mask & PSW_MASK_PER))
55 		return 0;
56 	/* user space single step */
57 	if (control == 0)
58 		return 1;
59 	/* over indication for storage alteration */
60 	if ((control & 0x20200000) && (cause & 0x2000))
61 		return 1;
62 	if (cause & 0x8000) {
63 		/* all branches */
64 		if ((control & 0x80800000) == 0x80000000)
65 			return 1;
66 		/* branch into selected range */
67 		if (((control & 0x80800000) == 0x80800000) &&
68 		    regs->psw.addr >= current->thread.per_user.start &&
69 		    regs->psw.addr <= current->thread.per_user.end)
70 			return 1;
71 	}
72 	return 0;
73 }
74 
75 int arch_uprobe_post_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
76 {
77 	int fixup = probe_get_fixup_type(auprobe->insn);
78 	struct uprobe_task *utask = current->utask;
79 
80 	clear_tsk_thread_flag(current, TIF_UPROBE_SINGLESTEP);
81 	update_cr_regs(current);
82 	psw_bits(regs->psw).r = auprobe->saved_per;
83 	regs->int_code = auprobe->saved_int_code;
84 
85 	if (fixup & FIXUP_PSW_NORMAL)
86 		regs->psw.addr += utask->vaddr - utask->xol_vaddr;
87 	if (fixup & FIXUP_RETURN_REGISTER) {
88 		int reg = (auprobe->insn[0] & 0xf0) >> 4;
89 
90 		regs->gprs[reg] += utask->vaddr - utask->xol_vaddr;
91 	}
92 	if (fixup & FIXUP_BRANCH_NOT_TAKEN) {
93 		int ilen = insn_length(auprobe->insn[0] >> 8);
94 
95 		if (regs->psw.addr - utask->xol_vaddr == ilen)
96 			regs->psw.addr = utask->vaddr + ilen;
97 	}
98 	if (check_per_event(current->thread.per_event.cause,
99 			    current->thread.per_user.control, regs)) {
100 		/* fix per address */
101 		current->thread.per_event.address = utask->vaddr;
102 		/* trigger per event */
103 		set_pt_regs_flag(regs, PIF_PER_TRAP);
104 	}
105 	return 0;
106 }
107 
108 int arch_uprobe_exception_notify(struct notifier_block *self, unsigned long val,
109 				 void *data)
110 {
111 	struct die_args *args = data;
112 	struct pt_regs *regs = args->regs;
113 
114 	if (!user_mode(regs))
115 		return NOTIFY_DONE;
116 	if (regs->int_code & 0x200) /* Trap during transaction */
117 		return NOTIFY_DONE;
118 	switch (val) {
119 	case DIE_BPT:
120 		if (uprobe_pre_sstep_notifier(regs))
121 			return NOTIFY_STOP;
122 		break;
123 	case DIE_SSTEP:
124 		if (uprobe_post_sstep_notifier(regs))
125 			return NOTIFY_STOP;
126 	default:
127 		break;
128 	}
129 	return NOTIFY_DONE;
130 }
131 
132 void arch_uprobe_abort_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
133 {
134 	clear_thread_flag(TIF_UPROBE_SINGLESTEP);
135 	regs->int_code = auprobe->saved_int_code;
136 	regs->psw.addr = current->utask->vaddr;
137 	current->thread.per_event.address = current->utask->vaddr;
138 }
139 
140 unsigned long arch_uretprobe_hijack_return_addr(unsigned long trampoline,
141 						struct pt_regs *regs)
142 {
143 	unsigned long orig;
144 
145 	orig = regs->gprs[14];
146 	regs->gprs[14] = trampoline;
147 	return orig;
148 }
149 
150 /* Instruction Emulation */
151 
152 static void adjust_psw_addr(psw_t *psw, unsigned long len)
153 {
154 	psw->addr = __rewind_psw(*psw, -len);
155 }
156 
157 #define EMU_ILLEGAL_OP		1
158 #define EMU_SPECIFICATION	2
159 #define EMU_ADDRESSING		3
160 
161 #define emu_load_ril(ptr, output)			\
162 ({							\
163 	unsigned int mask = sizeof(*(ptr)) - 1;		\
164 	__typeof__(*(ptr)) input;			\
165 	int __rc = 0;					\
166 							\
167 	if (!test_facility(34))				\
168 		__rc = EMU_ILLEGAL_OP;			\
169 	else if ((u64 __force)ptr & mask)		\
170 		__rc = EMU_SPECIFICATION;		\
171 	else if (get_user(input, ptr))			\
172 		__rc = EMU_ADDRESSING;			\
173 	else						\
174 		*(output) = input;			\
175 	__rc;						\
176 })
177 
178 #define emu_store_ril(regs, ptr, input)			\
179 ({							\
180 	unsigned int mask = sizeof(*(ptr)) - 1;		\
181 	__typeof__(ptr) __ptr = (ptr);			\
182 	int __rc = 0;					\
183 							\
184 	if (!test_facility(34))				\
185 		__rc = EMU_ILLEGAL_OP;			\
186 	else if ((u64 __force)__ptr & mask)		\
187 		__rc = EMU_SPECIFICATION;		\
188 	else if (put_user(*(input), __ptr))		\
189 		__rc = EMU_ADDRESSING;			\
190 	if (__rc == 0)					\
191 		sim_stor_event(regs, __ptr, mask + 1);	\
192 	__rc;						\
193 })
194 
195 #define emu_cmp_ril(regs, ptr, cmp)			\
196 ({							\
197 	unsigned int mask = sizeof(*(ptr)) - 1;		\
198 	__typeof__(*(ptr)) input;			\
199 	int __rc = 0;					\
200 							\
201 	if (!test_facility(34))				\
202 		__rc = EMU_ILLEGAL_OP;			\
203 	else if ((u64 __force)ptr & mask)		\
204 		__rc = EMU_SPECIFICATION;		\
205 	else if (get_user(input, ptr))			\
206 		__rc = EMU_ADDRESSING;			\
207 	else if (input > *(cmp))			\
208 		psw_bits((regs)->psw).cc = 1;		\
209 	else if (input < *(cmp))			\
210 		psw_bits((regs)->psw).cc = 2;		\
211 	else						\
212 		psw_bits((regs)->psw).cc = 0;		\
213 	__rc;						\
214 })
215 
216 struct insn_ril {
217 	u8 opc0;
218 	u8 reg	: 4;
219 	u8 opc1 : 4;
220 	s32 disp;
221 } __packed;
222 
223 union split_register {
224 	u64 u64;
225 	u32 u32[2];
226 	u16 u16[4];
227 	s64 s64;
228 	s32 s32[2];
229 	s16 s16[4];
230 };
231 
232 /*
233  * If user per registers are setup to trace storage alterations and an
234  * emulated store took place on a fitting address a user trap is generated.
235  */
236 static void sim_stor_event(struct pt_regs *regs, void *addr, int len)
237 {
238 	if (!(regs->psw.mask & PSW_MASK_PER))
239 		return;
240 	if (!(current->thread.per_user.control & PER_EVENT_STORE))
241 		return;
242 	if ((void *)current->thread.per_user.start > (addr + len))
243 		return;
244 	if ((void *)current->thread.per_user.end < addr)
245 		return;
246 	current->thread.per_event.address = regs->psw.addr;
247 	current->thread.per_event.cause = PER_EVENT_STORE >> 16;
248 	set_pt_regs_flag(regs, PIF_PER_TRAP);
249 }
250 
251 /*
252  * pc relative instructions are emulated, since parameters may not be
253  * accessible from the xol area due to range limitations.
254  */
255 static void handle_insn_ril(struct arch_uprobe *auprobe, struct pt_regs *regs)
256 {
257 	union split_register *rx;
258 	struct insn_ril *insn;
259 	unsigned int ilen;
260 	void *uptr;
261 	int rc = 0;
262 
263 	insn = (struct insn_ril *) &auprobe->insn;
264 	rx = (union split_register *) &regs->gprs[insn->reg];
265 	uptr = (void *)(regs->psw.addr + (insn->disp * 2));
266 	ilen = insn_length(insn->opc0);
267 
268 	switch (insn->opc0) {
269 	case 0xc0:
270 		switch (insn->opc1) {
271 		case 0x00: /* larl */
272 			rx->u64 = (unsigned long)uptr;
273 			break;
274 		}
275 		break;
276 	case 0xc4:
277 		switch (insn->opc1) {
278 		case 0x02: /* llhrl */
279 			rc = emu_load_ril((u16 __user *)uptr, &rx->u32[1]);
280 			break;
281 		case 0x04: /* lghrl */
282 			rc = emu_load_ril((s16 __user *)uptr, &rx->u64);
283 			break;
284 		case 0x05: /* lhrl */
285 			rc = emu_load_ril((s16 __user *)uptr, &rx->u32[1]);
286 			break;
287 		case 0x06: /* llghrl */
288 			rc = emu_load_ril((u16 __user *)uptr, &rx->u64);
289 			break;
290 		case 0x08: /* lgrl */
291 			rc = emu_load_ril((u64 __user *)uptr, &rx->u64);
292 			break;
293 		case 0x0c: /* lgfrl */
294 			rc = emu_load_ril((s32 __user *)uptr, &rx->u64);
295 			break;
296 		case 0x0d: /* lrl */
297 			rc = emu_load_ril((u32 __user *)uptr, &rx->u32[1]);
298 			break;
299 		case 0x0e: /* llgfrl */
300 			rc = emu_load_ril((u32 __user *)uptr, &rx->u64);
301 			break;
302 		case 0x07: /* sthrl */
303 			rc = emu_store_ril(regs, (u16 __user *)uptr, &rx->u16[3]);
304 			break;
305 		case 0x0b: /* stgrl */
306 			rc = emu_store_ril(regs, (u64 __user *)uptr, &rx->u64);
307 			break;
308 		case 0x0f: /* strl */
309 			rc = emu_store_ril(regs, (u32 __user *)uptr, &rx->u32[1]);
310 			break;
311 		}
312 		break;
313 	case 0xc6:
314 		switch (insn->opc1) {
315 		case 0x02: /* pfdrl */
316 			if (!test_facility(34))
317 				rc = EMU_ILLEGAL_OP;
318 			break;
319 		case 0x04: /* cghrl */
320 			rc = emu_cmp_ril(regs, (s16 __user *)uptr, &rx->s64);
321 			break;
322 		case 0x05: /* chrl */
323 			rc = emu_cmp_ril(regs, (s16 __user *)uptr, &rx->s32[1]);
324 			break;
325 		case 0x06: /* clghrl */
326 			rc = emu_cmp_ril(regs, (u16 __user *)uptr, &rx->u64);
327 			break;
328 		case 0x07: /* clhrl */
329 			rc = emu_cmp_ril(regs, (u16 __user *)uptr, &rx->u32[1]);
330 			break;
331 		case 0x08: /* cgrl */
332 			rc = emu_cmp_ril(regs, (s64 __user *)uptr, &rx->s64);
333 			break;
334 		case 0x0a: /* clgrl */
335 			rc = emu_cmp_ril(regs, (u64 __user *)uptr, &rx->u64);
336 			break;
337 		case 0x0c: /* cgfrl */
338 			rc = emu_cmp_ril(regs, (s32 __user *)uptr, &rx->s64);
339 			break;
340 		case 0x0d: /* crl */
341 			rc = emu_cmp_ril(regs, (s32 __user *)uptr, &rx->s32[1]);
342 			break;
343 		case 0x0e: /* clgfrl */
344 			rc = emu_cmp_ril(regs, (u32 __user *)uptr, &rx->u64);
345 			break;
346 		case 0x0f: /* clrl */
347 			rc = emu_cmp_ril(regs, (u32 __user *)uptr, &rx->u32[1]);
348 			break;
349 		}
350 		break;
351 	}
352 	adjust_psw_addr(&regs->psw, ilen);
353 	switch (rc) {
354 	case EMU_ILLEGAL_OP:
355 		regs->int_code = ilen << 16 | 0x0001;
356 		do_report_trap(regs, SIGILL, ILL_ILLOPC, NULL);
357 		break;
358 	case EMU_SPECIFICATION:
359 		regs->int_code = ilen << 16 | 0x0006;
360 		do_report_trap(regs, SIGILL, ILL_ILLOPC , NULL);
361 		break;
362 	case EMU_ADDRESSING:
363 		regs->int_code = ilen << 16 | 0x0005;
364 		do_report_trap(regs, SIGSEGV, SEGV_MAPERR, NULL);
365 		break;
366 	}
367 }
368 
369 bool arch_uprobe_skip_sstep(struct arch_uprobe *auprobe, struct pt_regs *regs)
370 {
371 	if ((psw_bits(regs->psw).eaba == PSW_AMODE_24BIT) ||
372 	    ((psw_bits(regs->psw).eaba == PSW_AMODE_31BIT) &&
373 	     !is_compat_task())) {
374 		regs->psw.addr = __rewind_psw(regs->psw, UPROBE_SWBP_INSN_SIZE);
375 		do_report_trap(regs, SIGILL, ILL_ILLADR, NULL);
376 		return true;
377 	}
378 	if (probe_is_insn_relative_long(auprobe->insn)) {
379 		handle_insn_ril(auprobe, regs);
380 		return true;
381 	}
382 	return false;
383 }
384