module.h

  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#ifdef CONFIG_64BIT
 42	u32 inst_lu32id;
 43	u32 inst_lu52id;
 44#endif
 45	u32 inst_jirl;
 46};
 47
 48struct plt_idx_entry {
 49	Elf_Addr symbol_addr;
 50};
 51
 52Elf_Addr module_emit_got_entry(struct module *mod, Elf_Shdr *sechdrs, Elf_Addr val);
 53Elf_Addr module_emit_plt_entry(struct module *mod, Elf_Shdr *sechdrs, Elf_Addr val);
 54
 55static inline struct got_entry emit_got_entry(Elf_Addr val)
 56{
 57	return (struct got_entry) { val };
 58}
 59
 60static inline struct plt_entry emit_plt_entry(unsigned long val)
 61{
 62#ifdef CONFIG_32BIT
 63	u32 lu12iw, jirl;
 64
 65	lu12iw = larch_insn_gen_lu12iw(LOONGARCH_GPR_T1, ADDR_IMM(val, LU12IW));
 66	jirl = larch_insn_gen_jirl(0, LOONGARCH_GPR_T1, ADDR_IMM(val, ORI));
 67
 68	return (struct plt_entry) { lu12iw, jirl };
 69#else
 70	u32 lu12iw, lu32id, lu52id, jirl;
 71
 72	lu12iw = larch_insn_gen_lu12iw(LOONGARCH_GPR_T1, ADDR_IMM(val, LU12IW));
 73	lu32id = larch_insn_gen_lu32id(LOONGARCH_GPR_T1, ADDR_IMM(val, LU32ID));
 74	lu52id = larch_insn_gen_lu52id(LOONGARCH_GPR_T1, LOONGARCH_GPR_T1, ADDR_IMM(val, LU52ID));
 75	jirl = larch_insn_gen_jirl(0, LOONGARCH_GPR_T1, ADDR_IMM(val, ORI));
 76
 77	return (struct plt_entry) { lu12iw, lu32id, lu52id, jirl };
 78#endif
 79}
 80
 81static inline struct plt_idx_entry emit_plt_idx_entry(unsigned long val)
 82{
 83	return (struct plt_idx_entry) { val };
 84}
 85
 86static inline int get_plt_idx(unsigned long val, Elf_Shdr *sechdrs, const struct mod_section *sec)
 87{
 88	int i;
 89	struct plt_idx_entry *plt_idx = (struct plt_idx_entry *)sechdrs[sec->shndx].sh_addr;
 90
 91	for (i = 0; i < sec->num_entries; i++) {
 92		if (plt_idx[i].symbol_addr == val)
 93			return i;
 94	}
 95
 96	return -1;
 97}
 98
 99static inline struct plt_entry *get_plt_entry(unsigned long val,
100					      Elf_Shdr *sechdrs,
101					      const struct mod_section *sec_plt,
102					      const struct mod_section *sec_plt_idx)
103{
104	int plt_idx = get_plt_idx(val, sechdrs, sec_plt_idx);
105	struct plt_entry *plt = (struct plt_entry *)sechdrs[sec_plt->shndx].sh_addr;
106
107	if (plt_idx < 0)
108		return NULL;
109
110	return plt + plt_idx;
111}
112
113static inline struct got_entry *get_got_entry(Elf_Addr val,
114					      Elf_Shdr *sechdrs,
115					      const struct mod_section *sec)
116{
117	int i;
118	struct got_entry *got = (struct got_entry *)sechdrs[sec->shndx].sh_addr;
119
120	for (i = 0; i < sec->num_entries; i++)
121		if (got[i].symbol_addr == val)
122			return &got[i];
123	return NULL;
124}
125
126#endif /* _ASM_MODULE_H */