1/* SPDX-License-Identifier: GPL-2.0 */
2/*
3 * Copyright (C) 2022 Loongson Technology Corporation Limited
4 */
5
6#ifndef _ASM_LOONGARCH_FTRACE_H
7#define _ASM_LOONGARCH_FTRACE_H
8
9#define FTRACE_PLT_IDX		0
10#define FTRACE_REGS_PLT_IDX	1
11#define NR_FTRACE_PLTS		2
12
13#ifdef CONFIG_FUNCTION_TRACER
14
15#define MCOUNT_INSN_SIZE 4		/* sizeof mcount call */
16
17#ifndef __ASSEMBLY__
18
19#ifndef CONFIG_DYNAMIC_FTRACE
20
21#define mcount _mcount
22extern void _mcount(void);
23extern void prepare_ftrace_return(unsigned long self_addr, unsigned long callsite_sp, unsigned long old);
24
25#else
26
27struct dyn_ftrace;
28struct dyn_arch_ftrace { };
29
30#define ARCH_SUPPORTS_FTRACE_OPS 1
31#define HAVE_FUNCTION_GRAPH_RET_ADDR_PTR
32
33#define ftrace_init_nop ftrace_init_nop
34int ftrace_init_nop(struct module *mod, struct dyn_ftrace *rec);
35
36static inline unsigned long ftrace_call_adjust(unsigned long addr)
37{
38	return addr;
39}
40
41void prepare_ftrace_return(unsigned long self_addr, unsigned long *parent);
42
43#endif /* CONFIG_DYNAMIC_FTRACE */
44
45#ifdef CONFIG_HAVE_DYNAMIC_FTRACE_WITH_ARGS
46struct ftrace_ops;
47
48struct ftrace_regs {
49	struct pt_regs regs;
50};
51
52static __always_inline struct pt_regs *arch_ftrace_get_regs(struct ftrace_regs *fregs)
53{
54	return &fregs->regs;
55}
56
57static __always_inline unsigned long
58ftrace_regs_get_instruction_pointer(struct ftrace_regs *fregs)
59{
60	return instruction_pointer(&fregs->regs);
61}
62
63static __always_inline void
64ftrace_regs_set_instruction_pointer(struct ftrace_regs *fregs, unsigned long ip)
65{
66	instruction_pointer_set(&fregs->regs, ip);
67}
68
69#define ftrace_regs_get_argument(fregs, n) \
70	regs_get_kernel_argument(&(fregs)->regs, n)
71#define ftrace_regs_get_stack_pointer(fregs) \
72	kernel_stack_pointer(&(fregs)->regs)
73#define ftrace_regs_return_value(fregs) \
74	regs_return_value(&(fregs)->regs)
75#define ftrace_regs_set_return_value(fregs, ret) \
76	regs_set_return_value(&(fregs)->regs, ret)
77#define ftrace_override_function_with_return(fregs) \
78	override_function_with_return(&(fregs)->regs)
79#define ftrace_regs_query_register_offset(name) \
80	regs_query_register_offset(name)
81
82#define ftrace_graph_func ftrace_graph_func
83void ftrace_graph_func(unsigned long ip, unsigned long parent_ip,
84		       struct ftrace_ops *op, struct ftrace_regs *fregs);
85
86#ifdef CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS
87static inline void
88__arch_ftrace_set_direct_caller(struct pt_regs *regs, unsigned long addr)
89{
90	regs->regs[13] = addr;	/* t1 */
91}
92
93#define arch_ftrace_set_direct_caller(fregs, addr) \
94	__arch_ftrace_set_direct_caller(&(fregs)->regs, addr)
95#endif /* CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS */
96
97#endif
98
99#endif /* __ASSEMBLY__ */
100
101#endif /* CONFIG_FUNCTION_TRACER */
102
103#ifndef __ASSEMBLY__
104#ifdef CONFIG_FUNCTION_GRAPH_TRACER
105struct fgraph_ret_regs {
106	/* a0 - a1 */
107	unsigned long regs[2];
108
109	unsigned long fp;
110	unsigned long __unused;
111};
112
113static inline unsigned long fgraph_ret_regs_return_value(struct fgraph_ret_regs *ret_regs)
114{
115	return ret_regs->regs[0];
116}
117
118static inline unsigned long fgraph_ret_regs_frame_pointer(struct fgraph_ret_regs *ret_regs)
119{
120	return ret_regs->fp;
121}
122#endif /* ifdef CONFIG_FUNCTION_GRAPH_TRACER */
123#endif
124
125#endif /* _ASM_LOONGARCH_FTRACE_H */
126