2 files changed, 5 insertions, 1 deletions

diff --git a/arch/x86/lib/thunk_32.S b/arch/x86/lib/thunk_32.S
index 2930ae05d773..28f85c916712 100644
--- a/arch/x86/lib/thunk_32.S
+++ b/arch/x86/lib/thunk_32.S
@@ -4,8 +4,8 @@
  *  (inspired by Andi Kleen's thunk_64.S)
  * Subject to the GNU public license, v.2. No warranty of any kind.
  */
-
         #include <linux/linkage.h>
+        #include <asm/asm.h>
 
 #ifdef CONFIG_TRACE_IRQFLAGS
         /* put return address in eax (arg1) */
@@ -22,6 +22,7 @@
         popl %ecx
         popl %eax
         ret
+        _ASM_NOKPROBE(\name)
         .endm
 
         thunk_ra trace_hardirqs_on_thunk,trace_hardirqs_on_caller
diff --git a/arch/x86/lib/thunk_64.S b/arch/x86/lib/thunk_64.S
index a63efd6bb6a5..92d9feaff42b 100644
--- a/arch/x86/lib/thunk_64.S
+++ b/arch/x86/lib/thunk_64.S
@@ -8,6 +8,7 @@
 #include <linux/linkage.h>
 #include <asm/dwarf2.h>
 #include <asm/calling.h>
+#include <asm/asm.h>
 
         /* rdi: arg1 ... normal C conventions. rax is saved/restored. */
         .macro THUNK name, func, put_ret_addr_in_rdi=0
@@ -25,6 +26,7 @@
         call \func
         jmp  restore
         CFI_ENDPROC
+        _ASM_NOKPROBE(\name)
         .endm
 
 #ifdef CONFIG_TRACE_IRQFLAGS
@@ -43,3 +45,4 @@ restore:
         RESTORE_ARGS
         ret
         CFI_ENDPROC
+        _ASM_NOKPROBE(restore)