Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 1 | /* |
| 2 | * Code for replacing ftrace calls with jumps. |
| 3 | * |
| 4 | * Copyright (C) 2007-2008 Steven Rostedt <srostedt@redhat.com> |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 5 | * Copyright (C) 2009, 2010 DSLab, Lanzhou University, China |
Wu Zhangjin | f7a904d | 2010-01-04 17:16:51 +0800 | [diff] [blame] | 6 | * Author: Wu Zhangjin <wuzhangjin@gmail.com> |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 7 | * |
| 8 | * Thanks goes to Steven Rostedt for writing the original x86 version. |
| 9 | */ |
| 10 | |
| 11 | #include <linux/uaccess.h> |
| 12 | #include <linux/init.h> |
| 13 | #include <linux/ftrace.h> |
Ralf Baechle | 19e2e17 | 2012-07-13 23:38:17 +0200 | [diff] [blame] | 14 | #include <linux/syscalls.h> |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 15 | |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 16 | #include <asm/asm.h> |
| 17 | #include <asm/asm-offsets.h> |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 18 | #include <asm/cacheflush.h> |
Ralf Baechle | 19e2e17 | 2012-07-13 23:38:17 +0200 | [diff] [blame] | 19 | #include <asm/syscall.h> |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 20 | #include <asm/uasm.h> |
Ralf Baechle | 19e2e17 | 2012-07-13 23:38:17 +0200 | [diff] [blame] | 21 | #include <asm/unistd.h> |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 22 | |
Wu Zhangjin | d9cdb2f | 2011-01-20 03:28:29 +0800 | [diff] [blame] | 23 | #include <asm-generic/sections.h> |
Wu Zhangjin | c9f8487 | 2010-05-14 19:08:34 +0800 | [diff] [blame] | 24 | |
Thomas Gleixner | 49de830 | 2011-07-23 12:41:23 +0000 | [diff] [blame] | 25 | #if defined(KBUILD_MCOUNT_RA_ADDRESS) && defined(CONFIG_32BIT) |
| 26 | #define MCOUNT_OFFSET_INSNS 5 |
| 27 | #else |
| 28 | #define MCOUNT_OFFSET_INSNS 4 |
| 29 | #endif |
| 30 | |
Markos Chandras | cb2f993 | 2013-06-10 10:35:26 +0000 | [diff] [blame] | 31 | #ifdef CONFIG_DYNAMIC_FTRACE |
| 32 | |
Al Cooper | 58b6940 | 2013-01-16 22:43:28 +0000 | [diff] [blame] | 33 | /* Arch override because MIPS doesn't need to run this from stop_machine() */ |
| 34 | void arch_ftrace_update_code(int command) |
| 35 | { |
| 36 | ftrace_modify_all_code(command); |
| 37 | } |
| 38 | |
Markos Chandras | cb2f993 | 2013-06-10 10:35:26 +0000 | [diff] [blame] | 39 | #endif |
| 40 | |
Thomas Gleixner | 49de830 | 2011-07-23 12:41:23 +0000 | [diff] [blame] | 41 | /* |
| 42 | * Check if the address is in kernel space |
| 43 | * |
| 44 | * Clone core_kernel_text() from kernel/extable.c, but doesn't call |
| 45 | * init_kernel_text() for Ftrace doesn't trace functions in init sections. |
| 46 | */ |
| 47 | static inline int in_kernel_space(unsigned long ip) |
| 48 | { |
| 49 | if (ip >= (unsigned long)_stext && |
| 50 | ip <= (unsigned long)_etext) |
| 51 | return 1; |
| 52 | return 0; |
| 53 | } |
| 54 | |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 55 | #ifdef CONFIG_DYNAMIC_FTRACE |
| 56 | |
| 57 | #define JAL 0x0c000000 /* jump & link: ip --> ra, jump to target */ |
| 58 | #define ADDR_MASK 0x03ffffff /* op_code|addr : 31...26|25 ....0 */ |
David Daney | c54794d | 2010-12-28 13:21:37 -0800 | [diff] [blame] | 59 | #define JUMP_RANGE_MASK ((1UL << 28) - 1) |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 60 | |
Wu Zhangjin | 4d6829f | 2010-05-14 19:08:31 +0800 | [diff] [blame] | 61 | #define INSN_NOP 0x00000000 /* nop */ |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 62 | #define INSN_JAL(addr) \ |
| 63 | ((unsigned int)(JAL | (((addr) >> 2) & ADDR_MASK))) |
| 64 | |
| 65 | static unsigned int insn_jal_ftrace_caller __read_mostly; |
| 66 | static unsigned int insn_lui_v1_hi16_mcount __read_mostly; |
| 67 | static unsigned int insn_j_ftrace_graph_caller __maybe_unused __read_mostly; |
| 68 | |
| 69 | static inline void ftrace_dyn_arch_init_insns(void) |
| 70 | { |
| 71 | u32 *buf; |
| 72 | unsigned int v1; |
| 73 | |
| 74 | /* lui v1, hi16_mcount */ |
| 75 | v1 = 3; |
| 76 | buf = (u32 *)&insn_lui_v1_hi16_mcount; |
| 77 | UASM_i_LA_mostly(&buf, v1, MCOUNT_ADDR); |
| 78 | |
| 79 | /* jal (ftrace_caller + 8), jump over the first two instruction */ |
| 80 | buf = (u32 *)&insn_jal_ftrace_caller; |
David Daney | c54794d | 2010-12-28 13:21:37 -0800 | [diff] [blame] | 81 | uasm_i_jal(&buf, (FTRACE_ADDR + 8) & JUMP_RANGE_MASK); |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 82 | |
| 83 | #ifdef CONFIG_FUNCTION_GRAPH_TRACER |
| 84 | /* j ftrace_graph_caller */ |
| 85 | buf = (u32 *)&insn_j_ftrace_graph_caller; |
David Daney | c54794d | 2010-12-28 13:21:37 -0800 | [diff] [blame] | 86 | uasm_i_j(&buf, (unsigned long)ftrace_graph_caller & JUMP_RANGE_MASK); |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 87 | #endif |
| 88 | } |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 89 | |
| 90 | static int ftrace_modify_code(unsigned long ip, unsigned int new_code) |
| 91 | { |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 92 | int faulted; |
Leonid Yegoshin | 6ebda44 | 2013-12-16 12:06:55 +0000 | [diff] [blame] | 93 | mm_segment_t old_fs; |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 94 | |
| 95 | /* *(unsigned int *)ip = new_code; */ |
| 96 | safe_store_code(new_code, ip, faulted); |
| 97 | |
| 98 | if (unlikely(faulted)) |
| 99 | return -EFAULT; |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 100 | |
Leonid Yegoshin | 6ebda44 | 2013-12-16 12:06:55 +0000 | [diff] [blame] | 101 | old_fs = get_fs(); |
| 102 | set_fs(get_ds()); |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 103 | flush_icache_range(ip, ip + 8); |
Leonid Yegoshin | 6ebda44 | 2013-12-16 12:06:55 +0000 | [diff] [blame] | 104 | set_fs(old_fs); |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 105 | |
| 106 | return 0; |
| 107 | } |
| 108 | |
Al Cooper | 58b6940 | 2013-01-16 22:43:28 +0000 | [diff] [blame] | 109 | #ifndef CONFIG_64BIT |
| 110 | static int ftrace_modify_code_2(unsigned long ip, unsigned int new_code1, |
| 111 | unsigned int new_code2) |
| 112 | { |
| 113 | int faulted; |
| 114 | |
| 115 | safe_store_code(new_code1, ip, faulted); |
| 116 | if (unlikely(faulted)) |
| 117 | return -EFAULT; |
Viller Hsiao | a467109 | 2014-02-22 15:46:49 +0800 | [diff] [blame] | 118 | safe_store_code(new_code2, ip + 4, faulted); |
Al Cooper | 58b6940 | 2013-01-16 22:43:28 +0000 | [diff] [blame] | 119 | if (unlikely(faulted)) |
| 120 | return -EFAULT; |
Viller Hsiao | a467109 | 2014-02-22 15:46:49 +0800 | [diff] [blame] | 121 | flush_icache_range(ip, ip + 8); |
Al Cooper | 58b6940 | 2013-01-16 22:43:28 +0000 | [diff] [blame] | 122 | return 0; |
| 123 | } |
| 124 | #endif |
| 125 | |
Wu Zhangjin | 7f21a60 | 2011-01-20 03:28:31 +0800 | [diff] [blame] | 126 | /* |
| 127 | * The details about the calling site of mcount on MIPS |
| 128 | * |
| 129 | * 1. For kernel: |
| 130 | * |
| 131 | * move at, ra |
| 132 | * jal _mcount --> nop |
| 133 | * |
| 134 | * 2. For modules: |
| 135 | * |
| 136 | * 2.1 For KBUILD_MCOUNT_RA_ADDRESS and CONFIG_32BIT |
| 137 | * |
Ralf Baechle | 7034228 | 2013-01-22 12:59:30 +0100 | [diff] [blame] | 138 | * lui v1, hi_16bit_of_mcount --> b 1f (0x10000005) |
Wu Zhangjin | 7f21a60 | 2011-01-20 03:28:31 +0800 | [diff] [blame] | 139 | * addiu v1, v1, low_16bit_of_mcount |
| 140 | * move at, ra |
| 141 | * move $12, ra_address |
| 142 | * jalr v1 |
| 143 | * sub sp, sp, 8 |
Ralf Baechle | 7034228 | 2013-01-22 12:59:30 +0100 | [diff] [blame] | 144 | * 1: offset = 5 instructions |
Wu Zhangjin | 7f21a60 | 2011-01-20 03:28:31 +0800 | [diff] [blame] | 145 | * 2.2 For the Other situations |
| 146 | * |
Ralf Baechle | 7034228 | 2013-01-22 12:59:30 +0100 | [diff] [blame] | 147 | * lui v1, hi_16bit_of_mcount --> b 1f (0x10000004) |
Wu Zhangjin | 7f21a60 | 2011-01-20 03:28:31 +0800 | [diff] [blame] | 148 | * addiu v1, v1, low_16bit_of_mcount |
| 149 | * move at, ra |
| 150 | * jalr v1 |
| 151 | * nop | move $12, ra_address | sub sp, sp, 8 |
Ralf Baechle | 7034228 | 2013-01-22 12:59:30 +0100 | [diff] [blame] | 152 | * 1: offset = 4 instructions |
Wu Zhangjin | 7f21a60 | 2011-01-20 03:28:31 +0800 | [diff] [blame] | 153 | */ |
| 154 | |
Wu Zhangjin | 7f21a60 | 2011-01-20 03:28:31 +0800 | [diff] [blame] | 155 | #define INSN_B_1F (0x10000000 | MCOUNT_OFFSET_INSNS) |
| 156 | |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 157 | int ftrace_make_nop(struct module *mod, |
| 158 | struct dyn_ftrace *rec, unsigned long addr) |
| 159 | { |
| 160 | unsigned int new; |
| 161 | unsigned long ip = rec->ip; |
| 162 | |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 163 | /* |
Wu Zhangjin | d9cdb2f | 2011-01-20 03:28:29 +0800 | [diff] [blame] | 164 | * If ip is in kernel space, no long call, otherwise, long call is |
| 165 | * needed. |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 166 | */ |
Wu Zhangjin | 7f21a60 | 2011-01-20 03:28:31 +0800 | [diff] [blame] | 167 | new = in_kernel_space(ip) ? INSN_NOP : INSN_B_1F; |
Al Cooper | 58b6940 | 2013-01-16 22:43:28 +0000 | [diff] [blame] | 168 | #ifdef CONFIG_64BIT |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 169 | return ftrace_modify_code(ip, new); |
Al Cooper | 58b6940 | 2013-01-16 22:43:28 +0000 | [diff] [blame] | 170 | #else |
| 171 | /* |
| 172 | * On 32 bit MIPS platforms, gcc adds a stack adjust |
| 173 | * instruction in the delay slot after the branch to |
| 174 | * mcount and expects mcount to restore the sp on return. |
| 175 | * This is based on a legacy API and does nothing but |
| 176 | * waste instructions so it's being removed at runtime. |
| 177 | */ |
| 178 | return ftrace_modify_code_2(ip, new, INSN_NOP); |
| 179 | #endif |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 180 | } |
| 181 | |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 182 | int ftrace_make_call(struct dyn_ftrace *rec, unsigned long addr) |
| 183 | { |
| 184 | unsigned int new; |
| 185 | unsigned long ip = rec->ip; |
| 186 | |
Wu Zhangjin | d9cdb2f | 2011-01-20 03:28:29 +0800 | [diff] [blame] | 187 | new = in_kernel_space(ip) ? insn_jal_ftrace_caller : |
| 188 | insn_lui_v1_hi16_mcount; |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 189 | |
| 190 | return ftrace_modify_code(ip, new); |
| 191 | } |
| 192 | |
| 193 | #define FTRACE_CALL_IP ((unsigned long)(&ftrace_call)) |
| 194 | |
| 195 | int ftrace_update_ftrace_func(ftrace_func_t func) |
| 196 | { |
| 197 | unsigned int new; |
| 198 | |
Wu Zhangjin | 4d6829f | 2010-05-14 19:08:31 +0800 | [diff] [blame] | 199 | new = INSN_JAL((unsigned long)func); |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 200 | |
| 201 | return ftrace_modify_code(FTRACE_CALL_IP, new); |
| 202 | } |
| 203 | |
Jiri Slaby | 3a36cb1 | 2014-02-24 19:59:59 +0100 | [diff] [blame] | 204 | int __init ftrace_dyn_arch_init(void) |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 205 | { |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 206 | /* Encode the instructions when booting */ |
| 207 | ftrace_dyn_arch_init_insns(); |
| 208 | |
| 209 | /* Remove "b ftrace_stub" to ensure ftrace_caller() is executed */ |
| 210 | ftrace_modify_code(MCOUNT_ADDR, INSN_NOP); |
| 211 | |
Wu Zhangjin | 538f195 | 2009-11-20 20:34:32 +0800 | [diff] [blame] | 212 | return 0; |
| 213 | } |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 214 | #endif /* CONFIG_DYNAMIC_FTRACE */ |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 215 | |
| 216 | #ifdef CONFIG_FUNCTION_GRAPH_TRACER |
| 217 | |
Wu Zhangjin | e17ff5f | 2009-11-20 20:34:35 +0800 | [diff] [blame] | 218 | #ifdef CONFIG_DYNAMIC_FTRACE |
| 219 | |
| 220 | extern void ftrace_graph_call(void); |
Wu Zhangjin | e17ff5f | 2009-11-20 20:34:35 +0800 | [diff] [blame] | 221 | #define FTRACE_GRAPH_CALL_IP ((unsigned long)(&ftrace_graph_call)) |
| 222 | |
| 223 | int ftrace_enable_ftrace_graph_caller(void) |
| 224 | { |
| 225 | return ftrace_modify_code(FTRACE_GRAPH_CALL_IP, |
Wu Zhangjin | e424054 | 2010-05-14 19:08:32 +0800 | [diff] [blame] | 226 | insn_j_ftrace_graph_caller); |
Wu Zhangjin | e17ff5f | 2009-11-20 20:34:35 +0800 | [diff] [blame] | 227 | } |
| 228 | |
| 229 | int ftrace_disable_ftrace_graph_caller(void) |
| 230 | { |
Wu Zhangjin | 4d6829f | 2010-05-14 19:08:31 +0800 | [diff] [blame] | 231 | return ftrace_modify_code(FTRACE_GRAPH_CALL_IP, INSN_NOP); |
Wu Zhangjin | e17ff5f | 2009-11-20 20:34:35 +0800 | [diff] [blame] | 232 | } |
| 233 | |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 234 | #endif /* CONFIG_DYNAMIC_FTRACE */ |
Wu Zhangjin | e17ff5f | 2009-11-20 20:34:35 +0800 | [diff] [blame] | 235 | |
Wu Zhangjin | 7326c4e | 2009-11-20 20:34:38 +0800 | [diff] [blame] | 236 | #ifndef KBUILD_MCOUNT_RA_ADDRESS |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 237 | |
Ralf Baechle | 7034228 | 2013-01-22 12:59:30 +0100 | [diff] [blame] | 238 | #define S_RA_SP (0xafbf << 16) /* s{d,w} ra, offset(sp) */ |
| 239 | #define S_R_SP (0xafb0 << 16) /* s{d,w} R, offset(sp) */ |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 240 | #define OFFSET_MASK 0xffff /* stack offset range: 0 ~ PT_SIZE */ |
| 241 | |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 242 | unsigned long ftrace_get_parent_ra_addr(unsigned long self_ra, unsigned long |
| 243 | old_parent_ra, unsigned long parent_ra_addr, unsigned long fp) |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 244 | { |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 245 | unsigned long sp, ip, tmp; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 246 | unsigned int code; |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 247 | int faulted; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 248 | |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 249 | /* |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 250 | * For module, move the ip from the return address after the |
Wu Zhangjin | 9a620a5 | 2011-01-20 03:28:27 +0800 | [diff] [blame] | 251 | * instruction "lui v1, hi_16bit_of_mcount"(offset is 24), but for |
| 252 | * kernel, move after the instruction "move ra, at"(offset is 16) |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 253 | */ |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 254 | ip = self_ra - (in_kernel_space(self_ra) ? 16 : 24); |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 255 | |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 256 | /* |
| 257 | * search the text until finding the non-store instruction or "s{d,w} |
| 258 | * ra, offset(sp)" instruction |
| 259 | */ |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 260 | do { |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 261 | /* get the code at "ip": code = *(unsigned int *)ip; */ |
| 262 | safe_load_code(code, ip, faulted); |
| 263 | |
| 264 | if (unlikely(faulted)) |
| 265 | return 0; |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 266 | /* |
| 267 | * If we hit the non-store instruction before finding where the |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 268 | * ra is stored, then this is a leaf function and it does not |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 269 | * store the ra on the stack |
| 270 | */ |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 271 | if ((code & S_R_SP) != S_R_SP) |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 272 | return parent_ra_addr; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 273 | |
Wu Zhangjin | 9a620a5 | 2011-01-20 03:28:27 +0800 | [diff] [blame] | 274 | /* Move to the next instruction */ |
| 275 | ip -= 4; |
| 276 | } while ((code & S_RA_SP) != S_RA_SP); |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 277 | |
| 278 | sp = fp + (code & OFFSET_MASK); |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 279 | |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 280 | /* tmp = *(unsigned long *)sp; */ |
| 281 | safe_load_stack(tmp, sp, faulted); |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 282 | if (unlikely(faulted)) |
| 283 | return 0; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 284 | |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 285 | if (tmp == old_parent_ra) |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 286 | return sp; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 287 | return 0; |
| 288 | } |
| 289 | |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 290 | #endif /* !KBUILD_MCOUNT_RA_ADDRESS */ |
Wu Zhangjin | 7326c4e | 2009-11-20 20:34:38 +0800 | [diff] [blame] | 291 | |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 292 | /* |
| 293 | * Hook the return address and push it in the stack of return addrs |
| 294 | * in current thread info. |
| 295 | */ |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 296 | void prepare_ftrace_return(unsigned long *parent_ra_addr, unsigned long self_ra, |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 297 | unsigned long fp) |
| 298 | { |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 299 | unsigned long old_parent_ra; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 300 | struct ftrace_graph_ent trace; |
| 301 | unsigned long return_hooker = (unsigned long) |
| 302 | &return_to_handler; |
Wu Zhangjin | b9f07eb | 2011-01-22 02:01:53 +0800 | [diff] [blame] | 303 | int faulted, insns; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 304 | |
| 305 | if (unlikely(atomic_read(¤t->tracing_graph_pause))) |
| 306 | return; |
| 307 | |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 308 | /* |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 309 | * "parent_ra_addr" is the stack address saved the return address of |
| 310 | * the caller of _mcount. |
Wu Zhangjin | 7326c4e | 2009-11-20 20:34:38 +0800 | [diff] [blame] | 311 | * |
| 312 | * if the gcc < 4.5, a leaf function does not save the return address |
| 313 | * in the stack address, so, we "emulate" one in _mcount's stack space, |
| 314 | * and hijack it directly, but for a non-leaf function, it save the |
| 315 | * return address to the its own stack space, we can not hijack it |
| 316 | * directly, but need to find the real stack address, |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 317 | * ftrace_get_parent_addr() does it! |
Wu Zhangjin | 7326c4e | 2009-11-20 20:34:38 +0800 | [diff] [blame] | 318 | * |
| 319 | * if gcc>= 4.5, with the new -mmcount-ra-address option, for a |
| 320 | * non-leaf function, the location of the return address will be saved |
| 321 | * to $12 for us, and for a leaf function, only put a zero into $12. we |
| 322 | * do it in ftrace_graph_caller of mcount.S. |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 323 | */ |
| 324 | |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 325 | /* old_parent_ra = *parent_ra_addr; */ |
| 326 | safe_load_stack(old_parent_ra, parent_ra_addr, faulted); |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 327 | if (unlikely(faulted)) |
| 328 | goto out; |
Wu Zhangjin | 7326c4e | 2009-11-20 20:34:38 +0800 | [diff] [blame] | 329 | #ifndef KBUILD_MCOUNT_RA_ADDRESS |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 330 | parent_ra_addr = (unsigned long *)ftrace_get_parent_ra_addr(self_ra, |
| 331 | old_parent_ra, (unsigned long)parent_ra_addr, fp); |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 332 | /* |
| 333 | * If fails when getting the stack address of the non-leaf function's |
| 334 | * ra, stop function graph tracer and return |
| 335 | */ |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 336 | if (parent_ra_addr == 0) |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 337 | goto out; |
Wu Zhangjin | 7326c4e | 2009-11-20 20:34:38 +0800 | [diff] [blame] | 338 | #endif |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 339 | /* *parent_ra_addr = return_hooker; */ |
| 340 | safe_store_stack(return_hooker, parent_ra_addr, faulted); |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 341 | if (unlikely(faulted)) |
| 342 | goto out; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 343 | |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 344 | if (ftrace_push_return_trace(old_parent_ra, self_ra, &trace.depth, fp) |
| 345 | == -EBUSY) { |
| 346 | *parent_ra_addr = old_parent_ra; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 347 | return; |
| 348 | } |
| 349 | |
Wu Zhangjin | b9f07eb | 2011-01-22 02:01:53 +0800 | [diff] [blame] | 350 | /* |
| 351 | * Get the recorded ip of the current mcount calling site in the |
| 352 | * __mcount_loc section, which will be used to filter the function |
| 353 | * entries configured through the tracing/set_graph_function interface. |
| 354 | */ |
| 355 | |
| 356 | insns = in_kernel_space(self_ra) ? 2 : MCOUNT_OFFSET_INSNS + 1; |
| 357 | trace.func = self_ra - (MCOUNT_INSN_SIZE * insns); |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 358 | |
| 359 | /* Only trace if the calling function expects to */ |
| 360 | if (!ftrace_graph_entry(&trace)) { |
| 361 | current->curr_ret_stack--; |
Wu Zhangjin | 2816e32 | 2011-01-20 03:28:30 +0800 | [diff] [blame] | 362 | *parent_ra_addr = old_parent_ra; |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 363 | } |
Wu Zhangjin | 046199c | 2009-11-20 20:34:36 +0800 | [diff] [blame] | 364 | return; |
| 365 | out: |
| 366 | ftrace_graph_stop(); |
| 367 | WARN_ON(1); |
Wu Zhangjin | 29c5d34 | 2009-11-20 20:34:34 +0800 | [diff] [blame] | 368 | } |
Wu Zhangjin | 68ccf75 | 2010-05-14 19:08:33 +0800 | [diff] [blame] | 369 | #endif /* CONFIG_FUNCTION_GRAPH_TRACER */ |
Ralf Baechle | 19e2e17 | 2012-07-13 23:38:17 +0200 | [diff] [blame] | 370 | |
| 371 | #ifdef CONFIG_FTRACE_SYSCALLS |
| 372 | |
| 373 | #ifdef CONFIG_32BIT |
| 374 | unsigned long __init arch_syscall_addr(int nr) |
| 375 | { |
| 376 | return (unsigned long)sys_call_table[nr - __NR_O32_Linux]; |
| 377 | } |
| 378 | #endif |
| 379 | |
| 380 | #ifdef CONFIG_64BIT |
| 381 | |
| 382 | unsigned long __init arch_syscall_addr(int nr) |
| 383 | { |
| 384 | #ifdef CONFIG_MIPS32_N32 |
| 385 | if (nr >= __NR_N32_Linux && nr <= __NR_N32_Linux + __NR_N32_Linux_syscalls) |
Ralf Baechle | 46e12c0 | 2012-07-14 09:22:05 +0200 | [diff] [blame] | 386 | return (unsigned long)sysn32_call_table[nr - __NR_N32_Linux]; |
Ralf Baechle | 19e2e17 | 2012-07-13 23:38:17 +0200 | [diff] [blame] | 387 | #endif |
| 388 | if (nr >= __NR_64_Linux && nr <= __NR_64_Linux + __NR_64_Linux_syscalls) |
| 389 | return (unsigned long)sys_call_table[nr - __NR_64_Linux]; |
| 390 | #ifdef CONFIG_MIPS32_O32 |
| 391 | if (nr >= __NR_O32_Linux && nr <= __NR_O32_Linux + __NR_O32_Linux_syscalls) |
| 392 | return (unsigned long)sys32_call_table[nr - __NR_O32_Linux]; |
| 393 | #endif |
| 394 | |
| 395 | return (unsigned long) &sys_ni_syscall; |
| 396 | } |
| 397 | #endif |
| 398 | |
| 399 | #endif /* CONFIG_FTRACE_SYSCALLS */ |