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
14struct mod_section {
15	int shndx;
16	int num_entries;
17	int max_entries;
18};
19
20struct 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
35struct got_entry {
36	Elf_Addr symbol_addr;
37};
38
39struct plt_entry {
40	u32 inst_lu12iw;
41	u32 inst_lu32id;
42	u32 inst_lu52id;
43	u32 inst_jirl;
44};
45
46struct plt_idx_entry {
47	Elf_Addr symbol_addr;
48};
49
50Elf_Addr module_emit_got_entry(struct module *mod, Elf_Shdr *sechdrs, Elf_Addr val);
51Elf_Addr module_emit_plt_entry(struct module *mod, Elf_Shdr *sechdrs, Elf_Addr val);
52
53static inline struct got_entry emit_got_entry(Elf_Addr val)
54{
55	return (struct got_entry) { val };
56}
57
58static 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
70static inline struct plt_idx_entry emit_plt_idx_entry(unsigned long val)
71{
72	return (struct plt_idx_entry) { val };
73}
74
75static 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
88static 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
102static 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