stacktrace.c 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. /*
  2. * Stack trace management functions
  3. *
  4. * Copyright (C) 2006-2009 Red Hat, Inc., Ingo Molnar <mingo@redhat.com>
  5. */
  6. #include <linux/sched.h>
  7. #include <linux/stacktrace.h>
  8. #include <linux/module.h>
  9. #include <linux/uaccess.h>
  10. #include <asm/stacktrace.h>
  11. static int save_stack_stack(void *data, char *name)
  12. {
  13. return 0;
  14. }
  15. static void
  16. __save_stack_address(void *data, unsigned long addr, bool reliable, bool nosched)
  17. {
  18. struct stack_trace *trace = data;
  19. #ifdef CONFIG_FRAME_POINTER
  20. if (!reliable)
  21. return;
  22. #endif
  23. if (nosched && in_sched_functions(addr))
  24. return;
  25. if (trace->skip > 0) {
  26. trace->skip--;
  27. return;
  28. }
  29. if (trace->nr_entries < trace->max_entries)
  30. trace->entries[trace->nr_entries++] = addr;
  31. }
  32. static void save_stack_address(void *data, unsigned long addr, int reliable)
  33. {
  34. return __save_stack_address(data, addr, reliable, false);
  35. }
  36. static void
  37. save_stack_address_nosched(void *data, unsigned long addr, int reliable)
  38. {
  39. return __save_stack_address(data, addr, reliable, true);
  40. }
  41. static const struct stacktrace_ops save_stack_ops = {
  42. .stack = save_stack_stack,
  43. .address = save_stack_address,
  44. .walk_stack = print_context_stack,
  45. };
  46. static const struct stacktrace_ops save_stack_ops_nosched = {
  47. .stack = save_stack_stack,
  48. .address = save_stack_address_nosched,
  49. .walk_stack = print_context_stack,
  50. };
  51. /*
  52. * Save stack-backtrace addresses into a stack_trace buffer.
  53. */
  54. void save_stack_trace(struct stack_trace *trace)
  55. {
  56. dump_trace(current, NULL, NULL, 0, &save_stack_ops, trace);
  57. if (trace->nr_entries < trace->max_entries)
  58. trace->entries[trace->nr_entries++] = ULONG_MAX;
  59. }
  60. EXPORT_SYMBOL_GPL(save_stack_trace);
  61. void save_stack_trace_regs(struct pt_regs *regs, struct stack_trace *trace)
  62. {
  63. dump_trace(current, regs, NULL, 0, &save_stack_ops, trace);
  64. if (trace->nr_entries < trace->max_entries)
  65. trace->entries[trace->nr_entries++] = ULONG_MAX;
  66. }
  67. void save_stack_trace_tsk(struct task_struct *tsk, struct stack_trace *trace)
  68. {
  69. dump_trace(tsk, NULL, NULL, 0, &save_stack_ops_nosched, trace);
  70. if (trace->nr_entries < trace->max_entries)
  71. trace->entries[trace->nr_entries++] = ULONG_MAX;
  72. }
  73. EXPORT_SYMBOL_GPL(save_stack_trace_tsk);
  74. /* Userspace stacktrace - based on kernel/trace/trace_sysprof.c */
  75. struct stack_frame_user {
  76. const void __user *next_fp;
  77. unsigned long ret_addr;
  78. };
  79. static int
  80. copy_stack_frame(const void __user *fp, struct stack_frame_user *frame)
  81. {
  82. int ret;
  83. if (!access_ok(VERIFY_READ, fp, sizeof(*frame)))
  84. return 0;
  85. ret = 1;
  86. pagefault_disable();
  87. if (__copy_from_user_inatomic(frame, fp, sizeof(*frame)))
  88. ret = 0;
  89. pagefault_enable();
  90. return ret;
  91. }
  92. static inline void __save_stack_trace_user(struct stack_trace *trace)
  93. {
  94. const struct pt_regs *regs = task_pt_regs(current);
  95. const void __user *fp = (const void __user *)regs->bp;
  96. if (trace->nr_entries < trace->max_entries)
  97. trace->entries[trace->nr_entries++] = regs->ip;
  98. while (trace->nr_entries < trace->max_entries) {
  99. struct stack_frame_user frame;
  100. frame.next_fp = NULL;
  101. frame.ret_addr = 0;
  102. if (!copy_stack_frame(fp, &frame))
  103. break;
  104. if ((unsigned long)fp < regs->sp)
  105. break;
  106. if (frame.ret_addr) {
  107. trace->entries[trace->nr_entries++] =
  108. frame.ret_addr;
  109. }
  110. if (fp == frame.next_fp)
  111. break;
  112. fp = frame.next_fp;
  113. }
  114. }
  115. void save_stack_trace_user(struct stack_trace *trace)
  116. {
  117. /*
  118. * Trace user stack if we are not a kernel thread
  119. */
  120. if (current->mm) {
  121. __save_stack_trace_user(trace);
  122. }
  123. if (trace->nr_entries < trace->max_entries)
  124. trace->entries[trace->nr_entries++] = ULONG_MAX;
  125. }