1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21#include <linux/cpu.h>
22#include <linux/debugfs.h>
23#include <linux/hardirq.h>
24#include <linux/init.h>
25#include <linux/ptrace.h>
26#include <linux/kprobes.h>
27#include <linux/stat.h>
28#include <linux/uaccess.h>
29#include <linux/sched/task_stack.h>
30
31#include <asm/cpufeature.h>
32#include <asm/cputype.h>
33#include <asm/daifflags.h>
34#include <asm/debug-monitors.h>
35#include <asm/system_misc.h>
36#include <asm/traps.h>
37
38
39u8 debug_monitors_arch(void)
40{
41 return cpuid_feature_extract_unsigned_field(read_sanitised_ftr_reg(SYS_ID_AA64DFR0_EL1),
42 ID_AA64DFR0_DEBUGVER_SHIFT);
43}
44
45
46
47
48static void mdscr_write(u32 mdscr)
49{
50 unsigned long flags;
51 flags = local_daif_save();
52 write_sysreg(mdscr, mdscr_el1);
53 local_daif_restore(flags);
54}
55NOKPROBE_SYMBOL(mdscr_write);
56
57static u32 mdscr_read(void)
58{
59 return read_sysreg(mdscr_el1);
60}
61NOKPROBE_SYMBOL(mdscr_read);
62
63
64
65
66
67static bool debug_enabled = true;
68
69static int create_debug_debugfs_entry(void)
70{
71 debugfs_create_bool("debug_enabled", 0644, NULL, &debug_enabled);
72 return 0;
73}
74fs_initcall(create_debug_debugfs_entry);
75
76static int __init early_debug_disable(char *buf)
77{
78 debug_enabled = false;
79 return 0;
80}
81
82early_param("nodebugmon", early_debug_disable);
83
84
85
86
87
88static DEFINE_PER_CPU(int, mde_ref_count);
89static DEFINE_PER_CPU(int, kde_ref_count);
90
91void enable_debug_monitors(enum dbg_active_el el)
92{
93 u32 mdscr, enable = 0;
94
95 WARN_ON(preemptible());
96
97 if (this_cpu_inc_return(mde_ref_count) == 1)
98 enable = DBG_MDSCR_MDE;
99
100 if (el == DBG_ACTIVE_EL1 &&
101 this_cpu_inc_return(kde_ref_count) == 1)
102 enable |= DBG_MDSCR_KDE;
103
104 if (enable && debug_enabled) {
105 mdscr = mdscr_read();
106 mdscr |= enable;
107 mdscr_write(mdscr);
108 }
109}
110NOKPROBE_SYMBOL(enable_debug_monitors);
111
112void disable_debug_monitors(enum dbg_active_el el)
113{
114 u32 mdscr, disable = 0;
115
116 WARN_ON(preemptible());
117
118 if (this_cpu_dec_return(mde_ref_count) == 0)
119 disable = ~DBG_MDSCR_MDE;
120
121 if (el == DBG_ACTIVE_EL1 &&
122 this_cpu_dec_return(kde_ref_count) == 0)
123 disable &= ~DBG_MDSCR_KDE;
124
125 if (disable) {
126 mdscr = mdscr_read();
127 mdscr &= disable;
128 mdscr_write(mdscr);
129 }
130}
131NOKPROBE_SYMBOL(disable_debug_monitors);
132
133
134
135
136static int clear_os_lock(unsigned int cpu)
137{
138 write_sysreg(0, oslar_el1);
139 isb();
140 return 0;
141}
142
143static int debug_monitors_init(void)
144{
145 return cpuhp_setup_state(CPUHP_AP_ARM64_DEBUG_MONITORS_STARTING,
146 "arm64/debug_monitors:starting",
147 clear_os_lock, NULL);
148}
149postcore_initcall(debug_monitors_init);
150
151
152
153
154static void set_regs_spsr_ss(struct pt_regs *regs)
155{
156 regs->pstate |= DBG_SPSR_SS;
157}
158NOKPROBE_SYMBOL(set_regs_spsr_ss);
159
160static void clear_regs_spsr_ss(struct pt_regs *regs)
161{
162 regs->pstate &= ~DBG_SPSR_SS;
163}
164NOKPROBE_SYMBOL(clear_regs_spsr_ss);
165
166
167static LIST_HEAD(step_hook);
168static DEFINE_SPINLOCK(step_hook_lock);
169
170void register_step_hook(struct step_hook *hook)
171{
172 spin_lock(&step_hook_lock);
173 list_add_rcu(&hook->node, &step_hook);
174 spin_unlock(&step_hook_lock);
175}
176
177void unregister_step_hook(struct step_hook *hook)
178{
179 spin_lock(&step_hook_lock);
180 list_del_rcu(&hook->node);
181 spin_unlock(&step_hook_lock);
182 synchronize_rcu();
183}
184
185
186
187
188
189
190
191static int call_step_hook(struct pt_regs *regs, unsigned int esr)
192{
193 struct step_hook *hook;
194 int retval = DBG_HOOK_ERROR;
195
196 rcu_read_lock();
197
198 list_for_each_entry_rcu(hook, &step_hook, node) {
199 retval = hook->fn(regs, esr);
200 if (retval == DBG_HOOK_HANDLED)
201 break;
202 }
203
204 rcu_read_unlock();
205
206 return retval;
207}
208NOKPROBE_SYMBOL(call_step_hook);
209
210static void send_user_sigtrap(int si_code)
211{
212 struct pt_regs *regs = current_pt_regs();
213
214 if (WARN_ON(!user_mode(regs)))
215 return;
216
217 if (interrupts_enabled(regs))
218 local_irq_enable();
219
220 arm64_force_sig_fault(SIGTRAP, si_code,
221 (void __user *)instruction_pointer(regs),
222 "User debug trap");
223}
224
225static int single_step_handler(unsigned long addr, unsigned int esr,
226 struct pt_regs *regs)
227{
228 bool handler_found = false;
229
230
231
232
233
234 if (!reinstall_suspended_bps(regs))
235 return 0;
236
237#ifdef CONFIG_KPROBES
238 if (kprobe_single_step_handler(regs, esr) == DBG_HOOK_HANDLED)
239 handler_found = true;
240#endif
241 if (!handler_found && call_step_hook(regs, esr) == DBG_HOOK_HANDLED)
242 handler_found = true;
243
244 if (!handler_found && user_mode(regs)) {
245 send_user_sigtrap(TRAP_TRACE);
246
247
248
249
250
251
252
253 user_rewind_single_step(current);
254 } else if (!handler_found) {
255 pr_warn("Unexpected kernel single-step exception at EL1\n");
256
257
258
259
260 set_regs_spsr_ss(regs);
261 }
262
263 return 0;
264}
265NOKPROBE_SYMBOL(single_step_handler);
266
267
268
269
270
271
272static LIST_HEAD(break_hook);
273static DEFINE_SPINLOCK(break_hook_lock);
274
275void register_break_hook(struct break_hook *hook)
276{
277 spin_lock(&break_hook_lock);
278 list_add_rcu(&hook->node, &break_hook);
279 spin_unlock(&break_hook_lock);
280}
281
282void unregister_break_hook(struct break_hook *hook)
283{
284 spin_lock(&break_hook_lock);
285 list_del_rcu(&hook->node);
286 spin_unlock(&break_hook_lock);
287 synchronize_rcu();
288}
289
290static int call_break_hook(struct pt_regs *regs, unsigned int esr)
291{
292 struct break_hook *hook;
293 int (*fn)(struct pt_regs *regs, unsigned int esr) = NULL;
294
295 rcu_read_lock();
296 list_for_each_entry_rcu(hook, &break_hook, node)
297 if ((esr & hook->esr_mask) == hook->esr_val)
298 fn = hook->fn;
299 rcu_read_unlock();
300
301 return fn ? fn(regs, esr) : DBG_HOOK_ERROR;
302}
303NOKPROBE_SYMBOL(call_break_hook);
304
305static int brk_handler(unsigned long addr, unsigned int esr,
306 struct pt_regs *regs)
307{
308 bool handler_found = false;
309
310#ifdef CONFIG_KPROBES
311 if ((esr & BRK64_ESR_MASK) == BRK64_ESR_KPROBES) {
312 if (kprobe_breakpoint_handler(regs, esr) == DBG_HOOK_HANDLED)
313 handler_found = true;
314 }
315#endif
316 if (!handler_found && call_break_hook(regs, esr) == DBG_HOOK_HANDLED)
317 handler_found = true;
318
319 if (!handler_found && user_mode(regs)) {
320 send_user_sigtrap(TRAP_BRKPT);
321 } else if (!handler_found) {
322 pr_warn("Unexpected kernel BRK exception at EL1\n");
323 return -EFAULT;
324 }
325
326 return 0;
327}
328NOKPROBE_SYMBOL(brk_handler);
329
330int aarch32_break_handler(struct pt_regs *regs)
331{
332 u32 arm_instr;
333 u16 thumb_instr;
334 bool bp = false;
335 void __user *pc = (void __user *)instruction_pointer(regs);
336
337 if (!compat_user_mode(regs))
338 return -EFAULT;
339
340 if (compat_thumb_mode(regs)) {
341
342 __le16 instr;
343 get_user(instr, (__le16 __user *)pc);
344 thumb_instr = le16_to_cpu(instr);
345 if (thumb_instr == AARCH32_BREAK_THUMB2_LO) {
346
347 get_user(instr, (__le16 __user *)(pc + 2));
348 thumb_instr = le16_to_cpu(instr);
349 bp = thumb_instr == AARCH32_BREAK_THUMB2_HI;
350 } else {
351 bp = thumb_instr == AARCH32_BREAK_THUMB;
352 }
353 } else {
354
355 __le32 instr;
356 get_user(instr, (__le32 __user *)pc);
357 arm_instr = le32_to_cpu(instr);
358 bp = (arm_instr & ~0xf0000000) == AARCH32_BREAK_ARM;
359 }
360
361 if (!bp)
362 return -EFAULT;
363
364 send_user_sigtrap(TRAP_BRKPT);
365 return 0;
366}
367NOKPROBE_SYMBOL(aarch32_break_handler);
368
369static int __init debug_traps_init(void)
370{
371 hook_debug_fault_code(DBG_ESR_EVT_HWSS, single_step_handler, SIGTRAP,
372 TRAP_TRACE, "single-step handler");
373 hook_debug_fault_code(DBG_ESR_EVT_BRK, brk_handler, SIGTRAP,
374 TRAP_BRKPT, "ptrace BRK handler");
375 return 0;
376}
377arch_initcall(debug_traps_init);
378
379
380void user_rewind_single_step(struct task_struct *task)
381{
382
383
384
385
386 if (test_ti_thread_flag(task_thread_info(task), TIF_SINGLESTEP))
387 set_regs_spsr_ss(task_pt_regs(task));
388}
389NOKPROBE_SYMBOL(user_rewind_single_step);
390
391void user_fastforward_single_step(struct task_struct *task)
392{
393 if (test_ti_thread_flag(task_thread_info(task), TIF_SINGLESTEP))
394 clear_regs_spsr_ss(task_pt_regs(task));
395}
396
397
398void kernel_enable_single_step(struct pt_regs *regs)
399{
400 WARN_ON(!irqs_disabled());
401 set_regs_spsr_ss(regs);
402 mdscr_write(mdscr_read() | DBG_MDSCR_SS);
403 enable_debug_monitors(DBG_ACTIVE_EL1);
404}
405NOKPROBE_SYMBOL(kernel_enable_single_step);
406
407void kernel_disable_single_step(void)
408{
409 WARN_ON(!irqs_disabled());
410 mdscr_write(mdscr_read() & ~DBG_MDSCR_SS);
411 disable_debug_monitors(DBG_ACTIVE_EL1);
412}
413NOKPROBE_SYMBOL(kernel_disable_single_step);
414
415int kernel_active_single_step(void)
416{
417 WARN_ON(!irqs_disabled());
418 return mdscr_read() & DBG_MDSCR_SS;
419}
420NOKPROBE_SYMBOL(kernel_active_single_step);
421
422
423void user_enable_single_step(struct task_struct *task)
424{
425 struct thread_info *ti = task_thread_info(task);
426
427 if (!test_and_set_ti_thread_flag(ti, TIF_SINGLESTEP))
428 set_regs_spsr_ss(task_pt_regs(task));
429}
430NOKPROBE_SYMBOL(user_enable_single_step);
431
432void user_disable_single_step(struct task_struct *task)
433{
434 clear_ti_thread_flag(task_thread_info(task), TIF_SINGLESTEP);
435}
436NOKPROBE_SYMBOL(user_disable_single_step);
437