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