ftrace.c 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. #include <linux/spinlock.h>
  2. #include <linux/hardirq.h>
  3. #include <linux/ftrace.h>
  4. #include <linux/percpu.h>
  5. #include <linux/init.h>
  6. #include <linux/list.h>
  7. #include <asm/ftrace.h>
  8. static const u32 ftrace_nop = 0x01000000;
  9. unsigned char *ftrace_nop_replace(void)
  10. {
  11. return (char *)&ftrace_nop;
  12. }
  13. unsigned char *ftrace_call_replace(unsigned long ip, unsigned long addr)
  14. {
  15. static u32 call;
  16. s32 off;
  17. off = ((s32)addr - (s32)ip);
  18. call = 0x40000000 | ((u32)off >> 2);
  19. return (unsigned char *) &call;
  20. }
  21. int
  22. ftrace_modify_code(unsigned long ip, unsigned char *old_code,
  23. unsigned char *new_code)
  24. {
  25. u32 old = *(u32 *)old_code;
  26. u32 new = *(u32 *)new_code;
  27. u32 replaced;
  28. int faulted;
  29. __asm__ __volatile__(
  30. "1: cas [%[ip]], %[old], %[new]\n"
  31. " flush %[ip]\n"
  32. " mov 0, %[faulted]\n"
  33. "2:\n"
  34. " .section .fixup,#alloc,#execinstr\n"
  35. " .align 4\n"
  36. "3: sethi %%hi(2b), %[faulted]\n"
  37. " jmpl %[faulted] + %%lo(2b), %%g0\n"
  38. " mov 1, %[faulted]\n"
  39. " .previous\n"
  40. " .section __ex_table,\"a\"\n"
  41. " .align 4\n"
  42. " .word 1b, 3b\n"
  43. " .previous\n"
  44. : "=r" (replaced), [faulted] "=r" (faulted)
  45. : [new] "0" (new), [old] "r" (old), [ip] "r" (ip)
  46. : "memory");
  47. if (replaced != old && replaced != new)
  48. faulted = 2;
  49. return faulted;
  50. }
  51. int ftrace_update_ftrace_func(ftrace_func_t func)
  52. {
  53. unsigned long ip = (unsigned long)(&ftrace_call);
  54. unsigned char old[MCOUNT_INSN_SIZE], *new;
  55. memcpy(old, &ftrace_call, MCOUNT_INSN_SIZE);
  56. new = ftrace_call_replace(ip, (unsigned long)func);
  57. return ftrace_modify_code(ip, old, new);
  58. }
  59. int __init ftrace_dyn_arch_init(void *data)
  60. {
  61. ftrace_mcount_set(data);
  62. return 0;
  63. }