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-generic/module.h>
10
11 #define RELA_STACK_DEPTH 16
12
13 struct mod_section {
14 Elf_Shdr *shdr;
15 int num_entries;
16 int max_entries;
17 };
18
19 struct mod_arch_specific {
20 struct mod_section got;
21 struct mod_section plt;
22 struct mod_section plt_idx;
23 };
24
25 struct got_entry {
26 Elf_Addr symbol_addr;
27 };
28
29 struct plt_entry {
30 u32 inst_lu12iw;
31 u32 inst_lu32id;
32 u32 inst_lu52id;
33 u32 inst_jirl;
34 };
35
36 struct plt_idx_entry {
37 Elf_Addr symbol_addr;
38 };
39
40 Elf_Addr module_emit_got_entry(struct module *mod, Elf_Addr val);
41 Elf_Addr module_emit_plt_entry(struct module *mod, Elf_Addr val);
42
emit_got_entry(Elf_Addr val)43 static inline struct got_entry emit_got_entry(Elf_Addr val)
44 {
45 return (struct got_entry) { val };
46 }
47
emit_plt_entry(unsigned long val)48 static inline struct plt_entry emit_plt_entry(unsigned long val)
49 {
50 u32 lu12iw, lu32id, lu52id, jirl;
51
52 lu12iw = (lu12iw_op << 25 | (((val >> 12) & 0xfffff) << 5) | LOONGARCH_GPR_T1);
53 lu32id = larch_insn_gen_lu32id(LOONGARCH_GPR_T1, ADDR_IMM(val, LU32ID));
54 lu52id = larch_insn_gen_lu52id(LOONGARCH_GPR_T1, LOONGARCH_GPR_T1, ADDR_IMM(val, LU52ID));
55 jirl = larch_insn_gen_jirl(0, LOONGARCH_GPR_T1, 0, (val & 0xfff));
56
57 return (struct plt_entry) { lu12iw, lu32id, lu52id, jirl };
58 }
59
emit_plt_idx_entry(unsigned long val)60 static inline struct plt_idx_entry emit_plt_idx_entry(unsigned long val)
61 {
62 return (struct plt_idx_entry) { val };
63 }
64
get_plt_idx(unsigned long val,const struct mod_section * sec)65 static inline int get_plt_idx(unsigned long val, const struct mod_section *sec)
66 {
67 int i;
68 struct plt_idx_entry *plt_idx = (struct plt_idx_entry *)sec->shdr->sh_addr;
69
70 for (i = 0; i < sec->num_entries; i++) {
71 if (plt_idx[i].symbol_addr == val)
72 return i;
73 }
74
75 return -1;
76 }
77
get_plt_entry(unsigned long val,const struct mod_section * sec_plt,const struct mod_section * sec_plt_idx)78 static inline struct plt_entry *get_plt_entry(unsigned long val,
79 const struct mod_section *sec_plt,
80 const struct mod_section *sec_plt_idx)
81 {
82 int plt_idx = get_plt_idx(val, sec_plt_idx);
83 struct plt_entry *plt = (struct plt_entry *)sec_plt->shdr->sh_addr;
84
85 if (plt_idx < 0)
86 return NULL;
87
88 return plt + plt_idx;
89 }
90
get_got_entry(Elf_Addr val,const struct mod_section * sec)91 static inline struct got_entry *get_got_entry(Elf_Addr val,
92 const struct mod_section *sec)
93 {
94 struct got_entry *got = (struct got_entry *)sec->shdr->sh_addr;
95 int i;
96
97 for (i = 0; i < sec->num_entries; i++)
98 if (got[i].symbol_addr == val)
99 return &got[i];
100 return NULL;
101 }
102
103 #endif /* _ASM_MODULE_H */
104