hw_breakpoint.c 9.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364
  1. /*
  2. * HW_breakpoint: a unified kernel/user-space hardware breakpoint facility,
  3. * using the CPU's debug registers. Derived from
  4. * "arch/x86/kernel/hw_breakpoint.c"
  5. *
  6. * This program is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU General Public License as published by
  8. * the Free Software Foundation; either version 2 of the License, or
  9. * (at your option) any later version.
  10. *
  11. * This program is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. * GNU General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU General Public License
  17. * along with this program; if not, write to the Free Software
  18. * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  19. *
  20. * Copyright 2010 IBM Corporation
  21. * Author: K.Prasad <prasad@linux.vnet.ibm.com>
  22. *
  23. */
  24. #include <linux/hw_breakpoint.h>
  25. #include <linux/notifier.h>
  26. #include <linux/kprobes.h>
  27. #include <linux/percpu.h>
  28. #include <linux/kernel.h>
  29. #include <linux/sched.h>
  30. #include <linux/init.h>
  31. #include <linux/smp.h>
  32. #include <asm/hw_breakpoint.h>
  33. #include <asm/processor.h>
  34. #include <asm/sstep.h>
  35. #include <asm/uaccess.h>
  36. /*
  37. * Stores the breakpoints currently in use on each breakpoint address
  38. * register for every cpu
  39. */
  40. static DEFINE_PER_CPU(struct perf_event *, bp_per_reg);
  41. /*
  42. * Returns total number of data or instruction breakpoints available.
  43. */
  44. int hw_breakpoint_slots(int type)
  45. {
  46. if (type == TYPE_DATA)
  47. return HBP_NUM;
  48. return 0; /* no instruction breakpoints available */
  49. }
  50. /*
  51. * Install a perf counter breakpoint.
  52. *
  53. * We seek a free debug address register and use it for this
  54. * breakpoint.
  55. *
  56. * Atomic: we hold the counter->ctx->lock and we only handle variables
  57. * and registers local to this cpu.
  58. */
  59. int arch_install_hw_breakpoint(struct perf_event *bp)
  60. {
  61. struct arch_hw_breakpoint *info = counter_arch_bp(bp);
  62. struct perf_event **slot = &__get_cpu_var(bp_per_reg);
  63. *slot = bp;
  64. /*
  65. * Do not install DABR values if the instruction must be single-stepped.
  66. * If so, DABR will be populated in single_step_dabr_instruction().
  67. */
  68. if (current->thread.last_hit_ubp != bp)
  69. set_dabr(info->address | info->type | DABR_TRANSLATION);
  70. return 0;
  71. }
  72. /*
  73. * Uninstall the breakpoint contained in the given counter.
  74. *
  75. * First we search the debug address register it uses and then we disable
  76. * it.
  77. *
  78. * Atomic: we hold the counter->ctx->lock and we only handle variables
  79. * and registers local to this cpu.
  80. */
  81. void arch_uninstall_hw_breakpoint(struct perf_event *bp)
  82. {
  83. struct perf_event **slot = &__get_cpu_var(bp_per_reg);
  84. if (*slot != bp) {
  85. WARN_ONCE(1, "Can't find the breakpoint");
  86. return;
  87. }
  88. *slot = NULL;
  89. set_dabr(0);
  90. }
  91. /*
  92. * Perform cleanup of arch-specific counters during unregistration
  93. * of the perf-event
  94. */
  95. void arch_unregister_hw_breakpoint(struct perf_event *bp)
  96. {
  97. /*
  98. * If the breakpoint is unregistered between a hw_breakpoint_handler()
  99. * and the single_step_dabr_instruction(), then cleanup the breakpoint
  100. * restoration variables to prevent dangling pointers.
  101. */
  102. if (bp->ctx->task)
  103. bp->ctx->task->thread.last_hit_ubp = NULL;
  104. }
  105. /*
  106. * Check for virtual address in kernel space.
  107. */
  108. int arch_check_bp_in_kernelspace(struct perf_event *bp)
  109. {
  110. struct arch_hw_breakpoint *info = counter_arch_bp(bp);
  111. return is_kernel_addr(info->address);
  112. }
  113. int arch_bp_generic_fields(int type, int *gen_bp_type)
  114. {
  115. switch (type) {
  116. case DABR_DATA_READ:
  117. *gen_bp_type = HW_BREAKPOINT_R;
  118. break;
  119. case DABR_DATA_WRITE:
  120. *gen_bp_type = HW_BREAKPOINT_W;
  121. break;
  122. case (DABR_DATA_WRITE | DABR_DATA_READ):
  123. *gen_bp_type = (HW_BREAKPOINT_W | HW_BREAKPOINT_R);
  124. break;
  125. default:
  126. return -EINVAL;
  127. }
  128. return 0;
  129. }
  130. /*
  131. * Validate the arch-specific HW Breakpoint register settings
  132. */
  133. int arch_validate_hwbkpt_settings(struct perf_event *bp)
  134. {
  135. int ret = -EINVAL;
  136. struct arch_hw_breakpoint *info = counter_arch_bp(bp);
  137. if (!bp)
  138. return ret;
  139. switch (bp->attr.bp_type) {
  140. case HW_BREAKPOINT_R:
  141. info->type = DABR_DATA_READ;
  142. break;
  143. case HW_BREAKPOINT_W:
  144. info->type = DABR_DATA_WRITE;
  145. break;
  146. case HW_BREAKPOINT_R | HW_BREAKPOINT_W:
  147. info->type = (DABR_DATA_READ | DABR_DATA_WRITE);
  148. break;
  149. default:
  150. return ret;
  151. }
  152. info->address = bp->attr.bp_addr;
  153. info->len = bp->attr.bp_len;
  154. /*
  155. * Since breakpoint length can be a maximum of HW_BREAKPOINT_LEN(8)
  156. * and breakpoint addresses are aligned to nearest double-word
  157. * HW_BREAKPOINT_ALIGN by rounding off to the lower address, the
  158. * 'symbolsize' should satisfy the check below.
  159. */
  160. if (info->len >
  161. (HW_BREAKPOINT_LEN - (info->address & HW_BREAKPOINT_ALIGN)))
  162. return -EINVAL;
  163. return 0;
  164. }
  165. /*
  166. * Restores the breakpoint on the debug registers.
  167. * Invoke this function if it is known that the execution context is
  168. * about to change to cause loss of MSR_SE settings.
  169. */
  170. void thread_change_pc(struct task_struct *tsk, struct pt_regs *regs)
  171. {
  172. struct arch_hw_breakpoint *info;
  173. if (likely(!tsk->thread.last_hit_ubp))
  174. return;
  175. info = counter_arch_bp(tsk->thread.last_hit_ubp);
  176. regs->msr &= ~MSR_SE;
  177. set_dabr(info->address | info->type | DABR_TRANSLATION);
  178. tsk->thread.last_hit_ubp = NULL;
  179. }
  180. /*
  181. * Handle debug exception notifications.
  182. */
  183. int __kprobes hw_breakpoint_handler(struct die_args *args)
  184. {
  185. int rc = NOTIFY_STOP;
  186. struct perf_event *bp;
  187. struct pt_regs *regs = args->regs;
  188. int stepped = 1;
  189. struct arch_hw_breakpoint *info;
  190. unsigned int instr;
  191. unsigned long dar = regs->dar;
  192. /* Disable breakpoints during exception handling */
  193. set_dabr(0);
  194. /*
  195. * The counter may be concurrently released but that can only
  196. * occur from a call_rcu() path. We can then safely fetch
  197. * the breakpoint, use its callback, touch its counter
  198. * while we are in an rcu_read_lock() path.
  199. */
  200. rcu_read_lock();
  201. bp = __get_cpu_var(bp_per_reg);
  202. if (!bp)
  203. goto out;
  204. info = counter_arch_bp(bp);
  205. /*
  206. * Return early after invoking user-callback function without restoring
  207. * DABR if the breakpoint is from ptrace which always operates in
  208. * one-shot mode. The ptrace-ed process will receive the SIGTRAP signal
  209. * generated in do_dabr().
  210. */
  211. if (bp->overflow_handler == ptrace_triggered) {
  212. perf_bp_event(bp, regs);
  213. rc = NOTIFY_DONE;
  214. goto out;
  215. }
  216. /*
  217. * Verify if dar lies within the address range occupied by the symbol
  218. * being watched to filter extraneous exceptions. If it doesn't,
  219. * we still need to single-step the instruction, but we don't
  220. * generate an event.
  221. */
  222. info->extraneous_interrupt = !((bp->attr.bp_addr <= dar) &&
  223. (dar - bp->attr.bp_addr < bp->attr.bp_len));
  224. /* Do not emulate user-space instructions, instead single-step them */
  225. if (user_mode(regs)) {
  226. bp->ctx->task->thread.last_hit_ubp = bp;
  227. regs->msr |= MSR_SE;
  228. goto out;
  229. }
  230. stepped = 0;
  231. instr = 0;
  232. if (!__get_user_inatomic(instr, (unsigned int *) regs->nip))
  233. stepped = emulate_step(regs, instr);
  234. /*
  235. * emulate_step() could not execute it. We've failed in reliably
  236. * handling the hw-breakpoint. Unregister it and throw a warning
  237. * message to let the user know about it.
  238. */
  239. if (!stepped) {
  240. WARN(1, "Unable to handle hardware breakpoint. Breakpoint at "
  241. "0x%lx will be disabled.", info->address);
  242. perf_event_disable(bp);
  243. goto out;
  244. }
  245. /*
  246. * As a policy, the callback is invoked in a 'trigger-after-execute'
  247. * fashion
  248. */
  249. if (!info->extraneous_interrupt)
  250. perf_bp_event(bp, regs);
  251. set_dabr(info->address | info->type | DABR_TRANSLATION);
  252. out:
  253. rcu_read_unlock();
  254. return rc;
  255. }
  256. /*
  257. * Handle single-step exceptions following a DABR hit.
  258. */
  259. int __kprobes single_step_dabr_instruction(struct die_args *args)
  260. {
  261. struct pt_regs *regs = args->regs;
  262. struct perf_event *bp = NULL;
  263. struct arch_hw_breakpoint *bp_info;
  264. bp = current->thread.last_hit_ubp;
  265. /*
  266. * Check if we are single-stepping as a result of a
  267. * previous HW Breakpoint exception
  268. */
  269. if (!bp)
  270. return NOTIFY_DONE;
  271. bp_info = counter_arch_bp(bp);
  272. /*
  273. * We shall invoke the user-defined callback function in the single
  274. * stepping handler to confirm to 'trigger-after-execute' semantics
  275. */
  276. if (!bp_info->extraneous_interrupt)
  277. perf_bp_event(bp, regs);
  278. set_dabr(bp_info->address | bp_info->type | DABR_TRANSLATION);
  279. current->thread.last_hit_ubp = NULL;
  280. /*
  281. * If the process was being single-stepped by ptrace, let the
  282. * other single-step actions occur (e.g. generate SIGTRAP).
  283. */
  284. if (test_thread_flag(TIF_SINGLESTEP))
  285. return NOTIFY_DONE;
  286. return NOTIFY_STOP;
  287. }
  288. /*
  289. * Handle debug exception notifications.
  290. */
  291. int __kprobes hw_breakpoint_exceptions_notify(
  292. struct notifier_block *unused, unsigned long val, void *data)
  293. {
  294. int ret = NOTIFY_DONE;
  295. switch (val) {
  296. case DIE_DABR_MATCH:
  297. ret = hw_breakpoint_handler(data);
  298. break;
  299. case DIE_SSTEP:
  300. ret = single_step_dabr_instruction(data);
  301. break;
  302. }
  303. return ret;
  304. }
  305. /*
  306. * Release the user breakpoints used by ptrace
  307. */
  308. void flush_ptrace_hw_breakpoint(struct task_struct *tsk)
  309. {
  310. struct thread_struct *t = &tsk->thread;
  311. unregister_hw_breakpoint(t->ptrace_bps[0]);
  312. t->ptrace_bps[0] = NULL;
  313. }
  314. void hw_breakpoint_pmu_read(struct perf_event *bp)
  315. {
  316. /* TODO */
  317. }