xref: /linux/arch/loongarch/include/asm/module.h (revision ae22a94997b8a03dcb3c922857c203246711f9d4)
1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3  * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
4  */
5 #ifndef _ASM_MODULE_H
6 #define _ASM_MODULE_H
7 
8 #include <asm/inst.h>
9 #include <asm/orc_types.h>
10 #include <asm-generic/module.h>
11 
12 #define RELA_STACK_DEPTH 16
13 
14 struct mod_section {
15 	int shndx;
16 	int num_entries;
17 	int max_entries;
18 };
19 
20 struct mod_arch_specific {
21 	struct mod_section got;
22 	struct mod_section plt;
23 	struct mod_section plt_idx;
24 
25 #ifdef CONFIG_UNWINDER_ORC
26 	unsigned int num_orcs;
27 	int *orc_unwind_ip;
28 	struct orc_entry *orc_unwind;
29 #endif
30 
31 	/* For CONFIG_DYNAMIC_FTRACE */
32 	struct plt_entry *ftrace_trampolines;
33 };
34 
35 struct got_entry {
36 	Elf_Addr symbol_addr;
37 };
38 
39 struct plt_entry {
40 	u32 inst_lu12iw;
41 	u32 inst_lu32id;
42 	u32 inst_lu52id;
43 	u32 inst_jirl;
44 };
45 
46 struct plt_idx_entry {
47 	Elf_Addr symbol_addr;
48 };
49 
50 Elf_Addr module_emit_got_entry(struct module *mod, Elf_Shdr *sechdrs, Elf_Addr val);
51 Elf_Addr module_emit_plt_entry(struct module *mod, Elf_Shdr *sechdrs, Elf_Addr val);
52 
53 static inline struct got_entry emit_got_entry(Elf_Addr val)
54 {
55 	return (struct got_entry) { val };
56 }
57 
58 static inline struct plt_entry emit_plt_entry(unsigned long val)
59 {
60 	u32 lu12iw, lu32id, lu52id, jirl;
61 
62 	lu12iw = larch_insn_gen_lu12iw(LOONGARCH_GPR_T1, ADDR_IMM(val, LU12IW));
63 	lu32id = larch_insn_gen_lu32id(LOONGARCH_GPR_T1, ADDR_IMM(val, LU32ID));
64 	lu52id = larch_insn_gen_lu52id(LOONGARCH_GPR_T1, LOONGARCH_GPR_T1, ADDR_IMM(val, LU52ID));
65 	jirl = larch_insn_gen_jirl(0, LOONGARCH_GPR_T1, ADDR_IMM(val, ORI));
66 
67 	return (struct plt_entry) { lu12iw, lu32id, lu52id, jirl };
68 }
69 
70 static inline struct plt_idx_entry emit_plt_idx_entry(unsigned long val)
71 {
72 	return (struct plt_idx_entry) { val };
73 }
74 
75 static inline int get_plt_idx(unsigned long val, Elf_Shdr *sechdrs, const struct mod_section *sec)
76 {
77 	int i;
78 	struct plt_idx_entry *plt_idx = (struct plt_idx_entry *)sechdrs[sec->shndx].sh_addr;
79 
80 	for (i = 0; i < sec->num_entries; i++) {
81 		if (plt_idx[i].symbol_addr == val)
82 			return i;
83 	}
84 
85 	return -1;
86 }
87 
88 static inline struct plt_entry *get_plt_entry(unsigned long val,
89 					      Elf_Shdr *sechdrs,
90 					      const struct mod_section *sec_plt,
91 					      const struct mod_section *sec_plt_idx)
92 {
93 	int plt_idx = get_plt_idx(val, sechdrs, sec_plt_idx);
94 	struct plt_entry *plt = (struct plt_entry *)sechdrs[sec_plt->shndx].sh_addr;
95 
96 	if (plt_idx < 0)
97 		return NULL;
98 
99 	return plt + plt_idx;
100 }
101 
102 static inline struct got_entry *get_got_entry(Elf_Addr val,
103 					      Elf_Shdr *sechdrs,
104 					      const struct mod_section *sec)
105 {
106 	int i;
107 	struct got_entry *got = (struct got_entry *)sechdrs[sec->shndx].sh_addr;
108 
109 	for (i = 0; i < sec->num_entries; i++)
110 		if (got[i].symbol_addr == val)
111 			return &got[i];
112 	return NULL;
113 }
114 
115 #endif /* _ASM_MODULE_H */
116