1// SPDX-License-Identifier: GPL-2.0
2/*
3 *  User-space Probes (UProbes) for s390
4 *
5 *    Copyright IBM Corp. 2014
6 *    Author(s): Jan Willeke,
7 */
8
9#include <linux/uaccess.h>
10#include <linux/uprobes.h>
11#include <linux/compat.h>
12#include <linux/kdebug.h>
13#include <linux/sched/task_stack.h>
14
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
22int 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
28int arch_uprobe_pre_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
29{
30	if (psw_bits(regs->psw).eaba == PSW_BITS_AMODE_24BIT)
31		return -EINVAL;
32	if (!is_compat_task() && psw_bits(regs->psw).eaba == PSW_BITS_AMODE_31BIT)
33		return -EINVAL;
34	clear_thread_flag(TIF_PER_TRAP);
35	auprobe->saved_per = psw_bits(regs->psw).per;
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
44bool 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
53static 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
77int 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).per = 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_thread_flag(TIF_PER_TRAP);
106	}
107	return 0;
108}
109
110int 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		break;
129	default:
130		break;
131	}
132	return NOTIFY_DONE;
133}
134
135void arch_uprobe_abort_xol(struct arch_uprobe *auprobe, struct pt_regs *regs)
136{
137	clear_thread_flag(TIF_UPROBE_SINGLESTEP);
138	regs->int_code = auprobe->saved_int_code;
139	regs->psw.addr = current->utask->vaddr;
140	current->thread.per_event.address = current->utask->vaddr;
141}
142
143unsigned long arch_uretprobe_hijack_return_addr(unsigned long trampoline,
144						struct pt_regs *regs)
145{
146	unsigned long orig;
147
148	orig = regs->gprs[14];
149	regs->gprs[14] = trampoline;
150	return orig;
151}
152
153bool arch_uretprobe_is_alive(struct return_instance *ret, enum rp_check ctx,
154			     struct pt_regs *regs)
155{
156	if (ctx == RP_CHECK_CHAIN_CALL)
157		return user_stack_pointer(regs) <= ret->stack;
158	else
159		return user_stack_pointer(regs) < ret->stack;
160}
161
162/* Instruction Emulation */
163
164static void adjust_psw_addr(psw_t *psw, unsigned long len)
165{
166	psw->addr = __rewind_psw(*psw, -len);
167}
168
169#define EMU_ILLEGAL_OP		1
170#define EMU_SPECIFICATION	2
171#define EMU_ADDRESSING		3
172
173#define emu_load_ril(ptr, output)			\
174({							\
175	unsigned int mask = sizeof(*(ptr)) - 1;		\
176	__typeof__(*(ptr)) input;			\
177	int __rc = 0;					\
178							\
179	if ((u64 __force)ptr & mask)			\
180		__rc = EMU_SPECIFICATION;		\
181	else if (get_user(input, ptr))			\
182		__rc = EMU_ADDRESSING;			\
183	else						\
184		*(output) = input;			\
185	__rc;						\
186})
187
188#define emu_store_ril(regs, ptr, input)			\
189({							\
190	unsigned int mask = sizeof(*(ptr)) - 1;		\
191	__typeof__(ptr) __ptr = (ptr);			\
192	int __rc = 0;					\
193							\
194	if ((u64 __force)__ptr & mask)			\
195		__rc = EMU_SPECIFICATION;		\
196	else if (put_user(*(input), __ptr))		\
197		__rc = EMU_ADDRESSING;			\
198	if (__rc == 0)					\
199		sim_stor_event(regs,			\
200			       (void __force *)__ptr,	\
201			       mask + 1);		\
202	__rc;						\
203})
204
205#define emu_cmp_ril(regs, ptr, cmp)			\
206({							\
207	unsigned int mask = sizeof(*(ptr)) - 1;		\
208	__typeof__(*(ptr)) input;			\
209	int __rc = 0;					\
210							\
211	if ((u64 __force)ptr & mask)			\
212		__rc = EMU_SPECIFICATION;		\
213	else if (get_user(input, ptr))			\
214		__rc = EMU_ADDRESSING;			\
215	else if (input > *(cmp))			\
216		psw_bits((regs)->psw).cc = 1;		\
217	else if (input < *(cmp))			\
218		psw_bits((regs)->psw).cc = 2;		\
219	else						\
220		psw_bits((regs)->psw).cc = 0;		\
221	__rc;						\
222})
223
224struct insn_ril {
225	u8 opc0;
226	u8 reg	: 4;
227	u8 opc1 : 4;
228	s32 disp;
229} __packed;
230
231union split_register {
232	u64 u64;
233	u32 u32[2];
234	u16 u16[4];
235	s64 s64;
236	s32 s32[2];
237	s16 s16[4];
238};
239
240/*
241 * If user per registers are setup to trace storage alterations and an
242 * emulated store took place on a fitting address a user trap is generated.
243 */
244static void sim_stor_event(struct pt_regs *regs, void *addr, int len)
245{
246	if (!(regs->psw.mask & PSW_MASK_PER))
247		return;
248	if (!(current->thread.per_user.control & PER_EVENT_STORE))
249		return;
250	if ((void *)current->thread.per_user.start > (addr + len))
251		return;
252	if ((void *)current->thread.per_user.end < addr)
253		return;
254	current->thread.per_event.address = regs->psw.addr;
255	current->thread.per_event.cause = PER_EVENT_STORE >> 16;
256	set_thread_flag(TIF_PER_TRAP);
257}
258
259/*
260 * pc relative instructions are emulated, since parameters may not be
261 * accessible from the xol area due to range limitations.
262 */
263static void handle_insn_ril(struct arch_uprobe *auprobe, struct pt_regs *regs)
264{
265	union split_register *rx;
266	struct insn_ril *insn;
267	unsigned int ilen;
268	void *uptr;
269	int rc = 0;
270
271	insn = (struct insn_ril *) &auprobe->insn;
272	rx = (union split_register *) &regs->gprs[insn->reg];
273	uptr = (void *)(regs->psw.addr + (insn->disp * 2));
274	ilen = insn_length(insn->opc0);
275
276	switch (insn->opc0) {
277	case 0xc0:
278		switch (insn->opc1) {
279		case 0x00: /* larl */
280			rx->u64 = (unsigned long)uptr;
281			break;
282		}
283		break;
284	case 0xc4:
285		switch (insn->opc1) {
286		case 0x02: /* llhrl */
287			rc = emu_load_ril((u16 __user *)uptr, &rx->u32[1]);
288			break;
289		case 0x04: /* lghrl */
290			rc = emu_load_ril((s16 __user *)uptr, &rx->u64);
291			break;
292		case 0x05: /* lhrl */
293			rc = emu_load_ril((s16 __user *)uptr, &rx->u32[1]);
294			break;
295		case 0x06: /* llghrl */
296			rc = emu_load_ril((u16 __user *)uptr, &rx->u64);
297			break;
298		case 0x08: /* lgrl */
299			rc = emu_load_ril((u64 __user *)uptr, &rx->u64);
300			break;
301		case 0x0c: /* lgfrl */
302			rc = emu_load_ril((s32 __user *)uptr, &rx->u64);
303			break;
304		case 0x0d: /* lrl */
305			rc = emu_load_ril((u32 __user *)uptr, &rx->u32[1]);
306			break;
307		case 0x0e: /* llgfrl */
308			rc = emu_load_ril((u32 __user *)uptr, &rx->u64);
309			break;
310		case 0x07: /* sthrl */
311			rc = emu_store_ril(regs, (u16 __user *)uptr, &rx->u16[3]);
312			break;
313		case 0x0b: /* stgrl */
314			rc = emu_store_ril(regs, (u64 __user *)uptr, &rx->u64);
315			break;
316		case 0x0f: /* strl */
317			rc = emu_store_ril(regs, (u32 __user *)uptr, &rx->u32[1]);
318			break;
319		}
320		break;
321	case 0xc6:
322		switch (insn->opc1) {
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
373bool arch_uprobe_skip_sstep(struct arch_uprobe *auprobe, struct pt_regs *regs)
374{
375	if ((psw_bits(regs->psw).eaba == PSW_BITS_AMODE_24BIT) ||
376	    ((psw_bits(regs->psw).eaba == PSW_BITS_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