1 /* SPDX-License-Identifier: GPL-2.0 */ <<
2 /* 1 /*
3 * LoongArch specific _mcount support !! 2 * MIPS specific _mcount support
4 * 3 *
5 * Copyright (C) 2022 Loongson Technology Corp !! 4 * This file is subject to the terms and conditions of the GNU General Public
>> 5 * License. See the file "COPYING" in the main directory of this archive for
>> 6 * more details.
>> 7 *
>> 8 * Copyright (C) 2009 Lemote Inc. & DSLab, Lanzhou University, China
>> 9 * Copyright (C) 2010 DSLab, Lanzhou University, China
>> 10 * Author: Wu Zhangjin <wuzhangjin@gmail.com>
6 */ 11 */
7 12
8 #include <linux/export.h> !! 13 #include <asm/export.h>
9 #include <asm/ftrace.h> <<
10 #include <asm/regdef.h> 14 #include <asm/regdef.h>
11 #include <asm/stackframe.h> 15 #include <asm/stackframe.h>
>> 16 #include <asm/ftrace.h>
12 17
13 .text 18 .text
14 !! 19 .set noreorder
15 #define MCOUNT_S0_OFFSET (0) !! 20 .set noat
16 #define MCOUNT_RA_OFFSET (SZREG) <<
17 #define MCOUNT_STACK_SIZE (2 * SZREG) <<
18 21
19 .macro MCOUNT_SAVE_REGS 22 .macro MCOUNT_SAVE_REGS
20 PTR_ADDI sp, sp, -MCOUNT_STACK_ !! 23 PTR_SUBU sp, PT_SIZE
21 PTR_S s0, sp, MCOUNT_S0_OFFS !! 24 PTR_S ra, PT_R31(sp)
22 PTR_S ra, sp, MCOUNT_RA_OFFS !! 25 PTR_S AT, PT_R1(sp)
23 move s0, a0 !! 26 PTR_S a0, PT_R4(sp)
>> 27 PTR_S a1, PT_R5(sp)
>> 28 PTR_S a2, PT_R6(sp)
>> 29 PTR_S a3, PT_R7(sp)
>> 30 #ifdef CONFIG_64BIT
>> 31 PTR_S a4, PT_R8(sp)
>> 32 PTR_S a5, PT_R9(sp)
>> 33 PTR_S a6, PT_R10(sp)
>> 34 PTR_S a7, PT_R11(sp)
>> 35 #endif
24 .endm 36 .endm
25 37
26 .macro MCOUNT_RESTORE_REGS 38 .macro MCOUNT_RESTORE_REGS
27 move a0, s0 !! 39 PTR_L ra, PT_R31(sp)
28 PTR_L ra, sp, MCOUNT_RA_OFFS !! 40 PTR_L AT, PT_R1(sp)
29 PTR_L s0, sp, MCOUNT_S0_OFFS !! 41 PTR_L a0, PT_R4(sp)
30 PTR_ADDI sp, sp, MCOUNT_STACK_S !! 42 PTR_L a1, PT_R5(sp)
>> 43 PTR_L a2, PT_R6(sp)
>> 44 PTR_L a3, PT_R7(sp)
>> 45 #ifdef CONFIG_64BIT
>> 46 PTR_L a4, PT_R8(sp)
>> 47 PTR_L a5, PT_R9(sp)
>> 48 PTR_L a6, PT_R10(sp)
>> 49 PTR_L a7, PT_R11(sp)
>> 50 #endif
>> 51 PTR_ADDIU sp, PT_SIZE
31 .endm 52 .endm
32 53
33 SYM_FUNC_START(_mcount) !! 54 .macro RETURN_BACK
34 la.pcrel t1, ftrace_stub !! 55 jr ra
35 la.pcrel t2, ftrace_trace_funct !! 56 move ra, AT
36 PTR_L t2, t2, 0 !! 57 .endm
37 beq t1, t2, fgraph_trace !! 58
>> 59 /*
>> 60 * The -mmcount-ra-address option of gcc 4.5 uses register $12 to pass
>> 61 * the location of the parent's return address.
>> 62 */
>> 63 #define MCOUNT_RA_ADDRESS_REG $12
>> 64
>> 65 #ifdef CONFIG_DYNAMIC_FTRACE
>> 66
>> 67 NESTED(ftrace_caller, PT_SIZE, ra)
>> 68 .globl _mcount
>> 69 _mcount:
>> 70 EXPORT_SYMBOL(_mcount)
>> 71 b ftrace_stub
>> 72 #ifdef CONFIG_32BIT
>> 73 addiu sp,sp,8
>> 74 #else
>> 75 nop
>> 76 #endif
>> 77
>> 78 /* When tracing is activated, it calls ftrace_caller+8 (aka here) */
>> 79 MCOUNT_SAVE_REGS
>> 80 #ifdef KBUILD_MCOUNT_RA_ADDRESS
>> 81 PTR_S MCOUNT_RA_ADDRESS_REG, PT_R12(sp)
>> 82 #endif
>> 83
>> 84 PTR_SUBU a0, ra, 8 /* arg1: self address */
>> 85 PTR_LA t1, _stext
>> 86 sltu t2, a0, t1 /* t2 = (a0 < _stext) */
>> 87 PTR_LA t1, _etext
>> 88 sltu t3, t1, a0 /* t3 = (a0 > _etext) */
>> 89 or t1, t2, t3
>> 90 beqz t1, ftrace_call
>> 91 nop
>> 92 #if defined(KBUILD_MCOUNT_RA_ADDRESS) && defined(CONFIG_32BIT)
>> 93 PTR_SUBU a0, a0, 16 /* arg1: adjust to module's recorded callsite */
>> 94 #else
>> 95 PTR_SUBU a0, a0, 12
>> 96 #endif
>> 97
>> 98 .globl ftrace_call
>> 99 ftrace_call:
>> 100 nop /* a placeholder for the call to a real tracing function */
>> 101 move a1, AT /* arg2: parent's return address */
>> 102
>> 103 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
>> 104 .globl ftrace_graph_call
>> 105 ftrace_graph_call:
>> 106 nop
>> 107 nop
>> 108 #endif
>> 109
>> 110 MCOUNT_RESTORE_REGS
>> 111 .globl ftrace_stub
>> 112 ftrace_stub:
>> 113 RETURN_BACK
>> 114 END(ftrace_caller)
>> 115
>> 116 #else /* ! CONFIG_DYNAMIC_FTRACE */
>> 117
>> 118 NESTED(_mcount, PT_SIZE, ra)
>> 119 EXPORT_SYMBOL(_mcount)
>> 120 PTR_LA t1, ftrace_stub
>> 121 PTR_L t2, ftrace_trace_function /* Prepare t2 for (1) */
>> 122 beq t1, t2, fgraph_trace
>> 123 nop
38 124
39 MCOUNT_SAVE_REGS 125 MCOUNT_SAVE_REGS
40 126
41 move a0, ra !! 127 move a0, ra /* arg1: self return address */
42 move a1, s0 !! 128 jalr t2 /* (1) call *ftrace_trace_function */
43 jirl ra, t2, 0 !! 129 move a1, AT /* arg2: parent's return address */
44 130
45 MCOUNT_RESTORE_REGS 131 MCOUNT_RESTORE_REGS
46 132
47 fgraph_trace: 133 fgraph_trace:
48 #ifdef CONFIG_FUNCTION_GRAPH_TRACER !! 134 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
49 la.pcrel t1, ftrace_stub !! 135 PTR_LA t1, ftrace_stub
50 la.pcrel t3, ftrace_graph_retur !! 136 PTR_L t3, ftrace_graph_return
51 PTR_L t3, t3, 0 !! 137 bne t1, t3, ftrace_graph_caller
52 bne t1, t3, ftrace_graph_c !! 138 nop
53 la.pcrel t1, ftrace_graph_entry !! 139 PTR_LA t1, ftrace_graph_entry_stub
54 la.pcrel t3, ftrace_graph_entry !! 140 PTR_L t3, ftrace_graph_entry
55 PTR_L t3, t3, 0 !! 141 bne t1, t3, ftrace_graph_caller
56 bne t1, t3, ftrace_graph_c !! 142 nop
57 #endif 143 #endif
58 144
59 SYM_INNER_LABEL(ftrace_stub, SYM_L_GLOBAL) !! 145 #ifdef CONFIG_32BIT
60 jr ra !! 146 addiu sp, sp, 8
61 #ifdef CONFIG_FUNCTION_GRAPH_TRACER <<
62 SYM_INNER_LABEL(ftrace_graph_func, SYM_L_GLOBA <<
63 bl ftrace_stub <<
64 #endif 147 #endif
65 SYM_FUNC_END(_mcount) !! 148
66 EXPORT_SYMBOL(_mcount) !! 149 .globl ftrace_stub
>> 150 ftrace_stub:
>> 151 RETURN_BACK
>> 152 END(_mcount)
>> 153
>> 154 #endif /* ! CONFIG_DYNAMIC_FTRACE */
67 155
68 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 156 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
69 SYM_FUNC_START(ftrace_graph_caller) !! 157
>> 158 NESTED(ftrace_graph_caller, PT_SIZE, ra)
>> 159 #ifndef CONFIG_DYNAMIC_FTRACE
70 MCOUNT_SAVE_REGS 160 MCOUNT_SAVE_REGS
>> 161 #endif
>> 162
>> 163 /* arg1: Get the location of the parent's return address */
>> 164 #ifdef KBUILD_MCOUNT_RA_ADDRESS
>> 165 #ifdef CONFIG_DYNAMIC_FTRACE
>> 166 PTR_L a0, PT_R12(sp)
>> 167 #else
>> 168 move a0, MCOUNT_RA_ADDRESS_REG
>> 169 #endif
>> 170 bnez a0, 1f /* non-leaf func: stored in MCOUNT_RA_ADDRESS_REG */
>> 171 nop
>> 172 #endif
>> 173 PTR_LA a0, PT_R1(sp) /* leaf func: the location in current stack */
>> 174 1:
71 175
72 PTR_ADDI a0, ra, -4 !! 176 /* arg2: Get self return address */
73 PTR_ADDI a1, sp, MCOUNT_STACK_S !! 177 #ifdef CONFIG_DYNAMIC_FTRACE
74 move a2, s0 !! 178 PTR_L a1, PT_R31(sp)
75 bl prepare_ftrace_return !! 179 #else
>> 180 move a1, ra
>> 181 #endif
76 182
>> 183 /* arg3: Get frame pointer of current stack */
>> 184 #ifdef CONFIG_64BIT
>> 185 PTR_LA a2, PT_SIZE(sp)
>> 186 #else
>> 187 PTR_LA a2, (PT_SIZE+8)(sp)
>> 188 #endif
>> 189
>> 190 jal prepare_ftrace_return
>> 191 nop
77 MCOUNT_RESTORE_REGS 192 MCOUNT_RESTORE_REGS
78 jr ra !! 193 #ifndef CONFIG_DYNAMIC_FTRACE
79 SYM_FUNC_END(ftrace_graph_caller) !! 194 #ifdef CONFIG_32BIT
>> 195 addiu sp, sp, 8
>> 196 #endif
>> 197 #endif
>> 198 RETURN_BACK
>> 199 END(ftrace_graph_caller)
80 200
81 SYM_FUNC_START(return_to_handler) !! 201 .align 2
82 PTR_ADDI sp, sp, -FGRET_REGS_SI !! 202 .globl return_to_handler
83 PTR_S a0, sp, FGRET_REGS_A0 !! 203 return_to_handler:
84 PTR_S a1, sp, FGRET_REGS_A1 !! 204 PTR_SUBU sp, PT_SIZE
85 PTR_S zero, sp, FGRET_REGS_F !! 205 PTR_S v0, PT_R2(sp)
86 !! 206
87 move a0, sp !! 207 jal ftrace_return_to_handler
88 bl ftrace_return_to_handl !! 208 PTR_S v1, PT_R3(sp)
89 !! 209
90 /* Restore the real parent address: a0 !! 210 /* restore the real parent address: v0 -> ra */
91 move ra, a0 !! 211 move ra, v0
92 !! 212
93 PTR_L a0, sp, FGRET_REGS_A0 !! 213 PTR_L v0, PT_R2(sp)
94 PTR_L a1, sp, FGRET_REGS_A1 !! 214 PTR_L v1, PT_R3(sp)
95 PTR_ADDI sp, sp, FGRET_REGS_SIZ !! 215 jr ra
96 jr ra !! 216 PTR_ADDIU sp, PT_SIZE
97 SYM_FUNC_END(return_to_handler) <<
98 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */ 217 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
>> 218
>> 219 .set at
>> 220 .set reorder
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.