From: Jinchao Wang <[email protected]>

Wprobe needs to move an active per-CPU watchpoint without releasing and
reserving its hardware slot.

Add arch_modify_local_hw_breakpoint_addr() to find the installed local
slot and update only its address shadow and hardware debug address
register. Publish the shadow first so hw_breakpoint_restore() observes
the new address if an NMI interrupts the update.

The caller must provide an installed local event and a valid address.
Slot ownership, breakpoint type, length, mask and DR7 remain unchanged.

Link: 
https://lore.kernel.org/all/59637b96946653393a7ad3c7de094094796b39c2.1785067572.git.wangjinchao...@gmail.com/

Signed-off-by: Jinchao Wang <[email protected]>
Signed-off-by: Masami Hiramatsu (Google) <[email protected]>
---
Changes in v11:
 - Return int error code instead of void.
 - Validate the new address using hw_breakpoint_arch_parse() before
   updating registers.
 - Increment cpu_dr7_seq in a sequence loop for NMI protection.
 - Return -ENOENT if the breakpoint slot is not found on the local CPU.
---
 arch/x86/include/asm/hw_breakpoint.h |    2 ++
 arch/x86/kernel/hw_breakpoint.c      |   33 +++++++++++++++++++++++++++++++++
 2 files changed, 35 insertions(+)

diff --git a/arch/x86/include/asm/hw_breakpoint.h 
b/arch/x86/include/asm/hw_breakpoint.h
index 0bc931cd0698..f35ec491f6dd 100644
--- a/arch/x86/include/asm/hw_breakpoint.h
+++ b/arch/x86/include/asm/hw_breakpoint.h
@@ -59,6 +59,8 @@ extern int hw_breakpoint_exceptions_notify(struct 
notifier_block *unused,
 
 
 int arch_install_hw_breakpoint(struct perf_event *bp);
+int arch_modify_local_hw_breakpoint_addr(struct perf_event *bp,
+                                        unsigned long addr);
 void arch_uninstall_hw_breakpoint(struct perf_event *bp);
 void hw_breakpoint_pmu_read(struct perf_event *bp);
 void hw_breakpoint_pmu_unthrottle(struct perf_event *bp);
diff --git a/arch/x86/kernel/hw_breakpoint.c b/arch/x86/kernel/hw_breakpoint.c
index 9ef24b55737f..c89f6e71c3a9 100644
--- a/arch/x86/kernel/hw_breakpoint.c
+++ b/arch/x86/kernel/hw_breakpoint.c
@@ -128,6 +128,39 @@ int arch_install_hw_breakpoint(struct perf_event *bp)
        return 0;
 }
 
+int arch_modify_local_hw_breakpoint_addr(struct perf_event *bp,
+                                         unsigned long addr)
+{
+       struct arch_hw_breakpoint hw;
+       struct perf_event_attr attr = bp->attr;
+       unsigned int seq;
+       int i, ret;
+
+       lockdep_assert_irqs_disabled();
+
+       attr.bp_addr = addr;
+       ret = hw_breakpoint_arch_parse(bp, &attr, &hw);
+       if (ret)
+               return ret;
+
+       for (i = 0; i < HBP_NUM; i++) {
+               if (this_cpu_read(bp_per_reg[i]) == bp)
+                       break;
+       }
+
+       if (WARN_ONCE(i == HBP_NUM, "Can't find any breakpoint slot"))
+               return -ENOENT;
+
+       do {
+               seq = this_cpu_inc_return(cpu_dr7_seq);
+               this_cpu_write(cpu_debugreg[i], addr);
+               barrier();
+               set_debugreg(addr, i);
+       } while (seq != this_cpu_read(cpu_dr7_seq));
+
+       return 0;
+}
+
 /*
  * Uninstall the breakpoint contained in the given counter.
  *


Reply via email to