*  Copyright (C) 2014  Steven Rostedt, Red Hat Inc
   */
  
 -#include <linux/linkage.h>
+ #include <linux/cfi_types.h>
 +#include <linux/linkage.h>
 +#include <asm/asm-offsets.h>
  #include <asm/ptrace.h>
  #include <asm/ftrace.h>
  #include <asm/export.h>
  
        .endm
  
+ SYM_TYPED_FUNC_START(ftrace_stub)
++      CALL_DEPTH_ACCOUNT
+       RET
+ SYM_FUNC_END(ftrace_stub)
+ 
+ SYM_TYPED_FUNC_START(ftrace_stub_graph)
++      CALL_DEPTH_ACCOUNT
+       RET
+ SYM_FUNC_END(ftrace_stub_graph)
+ 
  #ifdef CONFIG_DYNAMIC_FTRACE
  
  SYM_FUNC_START(__fentry__)
  #else /* ! CONFIG_DYNAMIC_FTRACE */
  
  SYM_FUNC_START(__fentry__)
 +      CALL_DEPTH_ACCOUNT
 +
        cmpq $ftrace_stub, ftrace_trace_function
        jnz trace
- 
- SYM_INNER_LABEL(ftrace_stub, SYM_L_GLOBAL)
-       ENDBR
        RET
  
  trace: