xref: /linux-6.15/arch/powerpc/kernel/module_64.c (revision 8a0edc72)
11a59d1b8SThomas Gleixner // SPDX-License-Identifier: GPL-2.0-or-later
27568cb4eSPaul Mackerras /*  Kernel module help for PPC64.
37568cb4eSPaul Mackerras     Copyright (C) 2001, 2003 Rusty Russell IBM Corporation.
47568cb4eSPaul Mackerras 
57568cb4eSPaul Mackerras */
6c7d1f6afSAnton Blanchard 
7c7d1f6afSAnton Blanchard #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
8c7d1f6afSAnton Blanchard 
97568cb4eSPaul Mackerras #include <linux/module.h>
107568cb4eSPaul Mackerras #include <linux/elf.h>
117568cb4eSPaul Mackerras #include <linux/moduleloader.h>
127568cb4eSPaul Mackerras #include <linux/err.h>
137568cb4eSPaul Mackerras #include <linux/vmalloc.h>
14f48cb8b4SSteven Rostedt #include <linux/ftrace.h>
1573c9ceabSJeremy Fitzhardinge #include <linux/bug.h>
1683775b85SAnton Blanchard #include <linux/uaccess.h>
17*8a0edc72SGuo Zhengkui #include <linux/kernel.h>
187568cb4eSPaul Mackerras #include <asm/module.h>
1921c4ff80SBenjamin Herrenschmidt #include <asm/firmware.h>
20b7bcda63SMichael Ellerman #include <asm/code-patching.h>
21eda09fbdSEmil Medve #include <linux/sort.h>
22b88c4767SRobert Jennings #include <asm/setup.h>
23336a7b5dSMichael Ellerman #include <asm/sections.h>
2475346251SJordan Niethe #include <asm/inst.h>
257568cb4eSPaul Mackerras 
267568cb4eSPaul Mackerras /* FIXME: We don't do .init separately.  To do this, we'd need to have
277568cb4eSPaul Mackerras    a separate r2 value in the init and core section, and stub between
287568cb4eSPaul Mackerras    them, too.
297568cb4eSPaul Mackerras 
307568cb4eSPaul Mackerras    Using a magic allocator which places modules within 32MB solves
317568cb4eSPaul Mackerras    this, and makes other things simpler.  Anton?
327568cb4eSPaul Mackerras    --RR.  */
337568cb4eSPaul Mackerras 
34f55d9665SMichael Ellerman #ifdef PPC64_ELF_ABI_v2
35008d7a91SRusty Russell 
36008d7a91SRusty Russell /* An address is simply the address of the function. */
37008d7a91SRusty Russell typedef unsigned long func_desc_t;
38008d7a91SRusty Russell 
39008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr)
40008d7a91SRusty Russell {
41008d7a91SRusty Russell 	return addr;
42008d7a91SRusty Russell }
43008d7a91SRusty Russell static unsigned long func_addr(unsigned long addr)
44008d7a91SRusty Russell {
45008d7a91SRusty Russell 	return addr;
46008d7a91SRusty Russell }
47008d7a91SRusty Russell static unsigned long stub_func_addr(func_desc_t func)
48008d7a91SRusty Russell {
49008d7a91SRusty Russell 	return func;
50008d7a91SRusty Russell }
51008d7a91SRusty Russell 
52008d7a91SRusty Russell /* PowerPC64 specific values for the Elf64_Sym st_other field.  */
53008d7a91SRusty Russell #define STO_PPC64_LOCAL_BIT	5
54008d7a91SRusty Russell #define STO_PPC64_LOCAL_MASK	(7 << STO_PPC64_LOCAL_BIT)
55008d7a91SRusty Russell #define PPC64_LOCAL_ENTRY_OFFSET(other)					\
56008d7a91SRusty Russell  (((1 << (((other) & STO_PPC64_LOCAL_MASK) >> STO_PPC64_LOCAL_BIT)) >> 2) << 2)
57008d7a91SRusty Russell 
58008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym)
59008d7a91SRusty Russell {
60008d7a91SRusty Russell 	/* sym->st_other indicates offset to local entry point
61008d7a91SRusty Russell 	 * (otherwise it will assume r12 is the address of the start
62008d7a91SRusty Russell 	 * of function and try to derive r2 from it). */
63008d7a91SRusty Russell 	return PPC64_LOCAL_ENTRY_OFFSET(sym->st_other);
64008d7a91SRusty Russell }
65d2fae548SRusty Russell #else
66008d7a91SRusty Russell 
67008d7a91SRusty Russell /* An address is address of the OPD entry, which contains address of fn. */
68008d7a91SRusty Russell typedef struct ppc64_opd_entry func_desc_t;
69008d7a91SRusty Russell 
70008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr)
71008d7a91SRusty Russell {
72008d7a91SRusty Russell 	return *(struct ppc64_opd_entry *)addr;
73008d7a91SRusty Russell }
74008d7a91SRusty Russell static unsigned long func_addr(unsigned long addr)
75008d7a91SRusty Russell {
76008d7a91SRusty Russell 	return func_desc(addr).funcaddr;
77008d7a91SRusty Russell }
78008d7a91SRusty Russell static unsigned long stub_func_addr(func_desc_t func)
79008d7a91SRusty Russell {
80008d7a91SRusty Russell 	return func.funcaddr;
81008d7a91SRusty Russell }
82008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym)
83008d7a91SRusty Russell {
84008d7a91SRusty Russell 	return 0;
85008d7a91SRusty Russell }
865633e85bSSergey Senozhatsky 
875633e85bSSergey Senozhatsky void *dereference_module_function_descriptor(struct module *mod, void *ptr)
885633e85bSSergey Senozhatsky {
895633e85bSSergey Senozhatsky 	if (ptr < (void *)mod->arch.start_opd ||
905633e85bSSergey Senozhatsky 			ptr >= (void *)mod->arch.end_opd)
915633e85bSSergey Senozhatsky 		return ptr;
925633e85bSSergey Senozhatsky 
935633e85bSSergey Senozhatsky 	return dereference_function_descriptor(ptr);
945633e85bSSergey Senozhatsky }
95d2fae548SRusty Russell #endif
96d2fae548SRusty Russell 
97f17c4e01SMichael Ellerman #define STUB_MAGIC 0x73747562 /* stub */
98f17c4e01SMichael Ellerman 
997568cb4eSPaul Mackerras /* Like PPC32, we need little trampolines to do > 24-bit jumps (into
1007568cb4eSPaul Mackerras    the kernel itself).  But on PPC64, these need to be used for every
1017568cb4eSPaul Mackerras    jump, actually, to reset r2 (TOC+0x8000). */
1027568cb4eSPaul Mackerras struct ppc64_stub_entry
1037568cb4eSPaul Mackerras {
10483775b85SAnton Blanchard 	/* 28 byte jump instruction sequence (7 instructions). We only
10583775b85SAnton Blanchard 	 * need 6 instructions on ABIv2 but we always allocate 7 so
10683775b85SAnton Blanchard 	 * so we don't have to modify the trampoline load instruction. */
1070e60e46eSRusty Russell 	u32 jump[7];
108f17c4e01SMichael Ellerman 	/* Used by ftrace to identify stubs */
109f17c4e01SMichael Ellerman 	u32 magic;
1107568cb4eSPaul Mackerras 	/* Data for the above code */
111008d7a91SRusty Russell 	func_desc_t funcdata;
1127568cb4eSPaul Mackerras };
1137568cb4eSPaul Mackerras 
1145c729a11SRusty Russell /*
1155c729a11SRusty Russell  * PPC64 uses 24 bit jumps, but we need to jump into other modules or
1165c729a11SRusty Russell  * the kernel which may be further.  So we jump to a stub.
1175c729a11SRusty Russell  *
1185c729a11SRusty Russell  * For ELFv1 we need to use this to set up the new r2 value (aka TOC
1195c729a11SRusty Russell  * pointer).  For ELFv2 it's the callee's responsibility to set up the
1205c729a11SRusty Russell  * new r2, but for both we need to save the old r2.
1215c729a11SRusty Russell  *
1225c729a11SRusty Russell  * We could simply patch the new r2 value and function pointer into
1235c729a11SRusty Russell  * the stub, but it's significantly shorter to put these values at the
1245c729a11SRusty Russell  * end of the stub code, and patch the stub address (32-bits relative
1255c729a11SRusty Russell  * to the TOC ptr, r2) into the stub.
1265c729a11SRusty Russell  */
12783775b85SAnton Blanchard static u32 ppc64_stub_insns[] = {
12847b04699SChristophe Leroy 	PPC_RAW_ADDIS(_R11, _R2, 0),
12947b04699SChristophe Leroy 	PPC_RAW_ADDI(_R11, _R11, 0),
130fed8393eSEugene Surovegin 	/* Save current r2 value in magic place on the stack. */
13147b04699SChristophe Leroy 	PPC_RAW_STD(_R2, _R1, R2_STACK_OFFSET),
13247b04699SChristophe Leroy 	PPC_RAW_LD(_R12, _R11, 32),
133f55d9665SMichael Ellerman #ifdef PPC64_ELF_ABI_v1
1345c729a11SRusty Russell 	/* Set up new r2 from function descriptor */
13547b04699SChristophe Leroy 	PPC_RAW_LD(_R2, _R11, 40),
1365c729a11SRusty Russell #endif
13747b04699SChristophe Leroy 	PPC_RAW_MTCTR(_R12),
13847b04699SChristophe Leroy 	PPC_RAW_BCTR(),
13983775b85SAnton Blanchard };
14083775b85SAnton Blanchard 
1417568cb4eSPaul Mackerras /* Count how many different 24-bit relocations (different symbol,
1427568cb4eSPaul Mackerras    different addend) */
1437568cb4eSPaul Mackerras static unsigned int count_relocs(const Elf64_Rela *rela, unsigned int num)
1447568cb4eSPaul Mackerras {
145eda09fbdSEmil Medve 	unsigned int i, r_info, r_addend, _count_relocs;
1467568cb4eSPaul Mackerras 
1477568cb4eSPaul Mackerras 	/* FIXME: Only count external ones --RR */
148eda09fbdSEmil Medve 	_count_relocs = 0;
149eda09fbdSEmil Medve 	r_info = 0;
150eda09fbdSEmil Medve 	r_addend = 0;
151eda09fbdSEmil Medve 	for (i = 0; i < num; i++)
1527568cb4eSPaul Mackerras 		/* Only count 24-bit relocs, others don't need stubs */
153eda09fbdSEmil Medve 		if (ELF64_R_TYPE(rela[i].r_info) == R_PPC_REL24 &&
154eda09fbdSEmil Medve 		    (r_info != ELF64_R_SYM(rela[i].r_info) ||
155eda09fbdSEmil Medve 		     r_addend != rela[i].r_addend)) {
156eda09fbdSEmil Medve 			_count_relocs++;
157eda09fbdSEmil Medve 			r_info = ELF64_R_SYM(rela[i].r_info);
158eda09fbdSEmil Medve 			r_addend = rela[i].r_addend;
1597568cb4eSPaul Mackerras 		}
160eda09fbdSEmil Medve 
161eda09fbdSEmil Medve 	return _count_relocs;
1627568cb4eSPaul Mackerras }
1637568cb4eSPaul Mackerras 
164eda09fbdSEmil Medve static int relacmp(const void *_x, const void *_y)
165eda09fbdSEmil Medve {
166eda09fbdSEmil Medve 	const Elf64_Rela *x, *y;
167eda09fbdSEmil Medve 
168eda09fbdSEmil Medve 	y = (Elf64_Rela *)_x;
169eda09fbdSEmil Medve 	x = (Elf64_Rela *)_y;
170eda09fbdSEmil Medve 
171eda09fbdSEmil Medve 	/* Compare the entire r_info (as opposed to ELF64_R_SYM(r_info) only) to
172eda09fbdSEmil Medve 	 * make the comparison cheaper/faster. It won't affect the sorting or
173eda09fbdSEmil Medve 	 * the counting algorithms' performance
174eda09fbdSEmil Medve 	 */
175eda09fbdSEmil Medve 	if (x->r_info < y->r_info)
176eda09fbdSEmil Medve 		return -1;
177eda09fbdSEmil Medve 	else if (x->r_info > y->r_info)
178eda09fbdSEmil Medve 		return 1;
179eda09fbdSEmil Medve 	else if (x->r_addend < y->r_addend)
180eda09fbdSEmil Medve 		return -1;
181eda09fbdSEmil Medve 	else if (x->r_addend > y->r_addend)
182eda09fbdSEmil Medve 		return 1;
183eda09fbdSEmil Medve 	else
184eda09fbdSEmil Medve 		return 0;
185eda09fbdSEmil Medve }
186eda09fbdSEmil Medve 
1877568cb4eSPaul Mackerras /* Get size of potential trampolines required. */
1887568cb4eSPaul Mackerras static unsigned long get_stubs_size(const Elf64_Ehdr *hdr,
1897568cb4eSPaul Mackerras 				    const Elf64_Shdr *sechdrs)
1907568cb4eSPaul Mackerras {
1917568cb4eSPaul Mackerras 	/* One extra reloc so it's always 0-funcaddr terminated */
1927568cb4eSPaul Mackerras 	unsigned long relocs = 1;
1937568cb4eSPaul Mackerras 	unsigned i;
1947568cb4eSPaul Mackerras 
1957568cb4eSPaul Mackerras 	/* Every relocated section... */
1967568cb4eSPaul Mackerras 	for (i = 1; i < hdr->e_shnum; i++) {
1977568cb4eSPaul Mackerras 		if (sechdrs[i].sh_type == SHT_RELA) {
198c7d1f6afSAnton Blanchard 			pr_debug("Found relocations in section %u\n", i);
199c7d1f6afSAnton Blanchard 			pr_debug("Ptr: %p.  Number: %Lu\n",
2007568cb4eSPaul Mackerras 			       (void *)sechdrs[i].sh_addr,
2017568cb4eSPaul Mackerras 			       sechdrs[i].sh_size / sizeof(Elf64_Rela));
202eda09fbdSEmil Medve 
203eda09fbdSEmil Medve 			/* Sort the relocation information based on a symbol and
204eda09fbdSEmil Medve 			 * addend key. This is a stable O(n*log n) complexity
205eda09fbdSEmil Medve 			 * alogrithm but it will reduce the complexity of
206eda09fbdSEmil Medve 			 * count_relocs() to linear complexity O(n)
207eda09fbdSEmil Medve 			 */
208eda09fbdSEmil Medve 			sort((void *)sechdrs[i].sh_addr,
209eda09fbdSEmil Medve 			     sechdrs[i].sh_size / sizeof(Elf64_Rela),
210bac7ca7bSAndrey Abramov 			     sizeof(Elf64_Rela), relacmp, NULL);
211eda09fbdSEmil Medve 
2127568cb4eSPaul Mackerras 			relocs += count_relocs((void *)sechdrs[i].sh_addr,
2137568cb4eSPaul Mackerras 					       sechdrs[i].sh_size
2147568cb4eSPaul Mackerras 					       / sizeof(Elf64_Rela));
2157568cb4eSPaul Mackerras 		}
2167568cb4eSPaul Mackerras 	}
2177568cb4eSPaul Mackerras 
218f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
219f48cb8b4SSteven Rostedt 	/* make the trampoline to the ftrace_caller */
220f48cb8b4SSteven Rostedt 	relocs++;
221ae30cc05SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
222ae30cc05SNaveen N. Rao 	/* an additional one for ftrace_regs_caller */
223ae30cc05SNaveen N. Rao 	relocs++;
224ae30cc05SNaveen N. Rao #endif
225f48cb8b4SSteven Rostedt #endif
226f48cb8b4SSteven Rostedt 
227c7d1f6afSAnton Blanchard 	pr_debug("Looks like a total of %lu stubs, max\n", relocs);
2287568cb4eSPaul Mackerras 	return relocs * sizeof(struct ppc64_stub_entry);
2297568cb4eSPaul Mackerras }
2307568cb4eSPaul Mackerras 
2315b12c5c6SRusty Russell /* Still needed for ELFv2, for .TOC. */
2327568cb4eSPaul Mackerras static void dedotify_versions(struct modversion_info *vers,
2337568cb4eSPaul Mackerras 			      unsigned long size)
2347568cb4eSPaul Mackerras {
2357568cb4eSPaul Mackerras 	struct modversion_info *end;
2367568cb4eSPaul Mackerras 
2377568cb4eSPaul Mackerras 	for (end = (void *)vers + size; vers < end; vers++)
238c2cbcf53SLaurent Dufour 		if (vers->name[0] == '.') {
2397568cb4eSPaul Mackerras 			memmove(vers->name, vers->name+1, strlen(vers->name));
240c2cbcf53SLaurent Dufour 		}
2417568cb4eSPaul Mackerras }
2427568cb4eSPaul Mackerras 
243c153693dSAlan Modra /*
244c153693dSAlan Modra  * Undefined symbols which refer to .funcname, hack to funcname. Make .TOC.
245c153693dSAlan Modra  * seem to be defined (value set later).
246c153693dSAlan Modra  */
2477568cb4eSPaul Mackerras static void dedotify(Elf64_Sym *syms, unsigned int numsyms, char *strtab)
2487568cb4eSPaul Mackerras {
2497568cb4eSPaul Mackerras 	unsigned int i;
2507568cb4eSPaul Mackerras 
2517568cb4eSPaul Mackerras 	for (i = 1; i < numsyms; i++) {
2527568cb4eSPaul Mackerras 		if (syms[i].st_shndx == SHN_UNDEF) {
2537568cb4eSPaul Mackerras 			char *name = strtab + syms[i].st_name;
254c153693dSAlan Modra 			if (name[0] == '.') {
255c153693dSAlan Modra 				if (strcmp(name+1, "TOC.") == 0)
256c153693dSAlan Modra 					syms[i].st_shndx = SHN_ABS;
257f15838e9SAndreas Schwab 				syms[i].st_name++;
2587568cb4eSPaul Mackerras 			}
2597568cb4eSPaul Mackerras 		}
2607568cb4eSPaul Mackerras 	}
261c153693dSAlan Modra }
2627568cb4eSPaul Mackerras 
2634edebbeaSRusty Russell static Elf64_Sym *find_dot_toc(Elf64_Shdr *sechdrs,
2644edebbeaSRusty Russell 			       const char *strtab,
2654edebbeaSRusty Russell 			       unsigned int symindex)
2664edebbeaSRusty Russell {
2674edebbeaSRusty Russell 	unsigned int i, numsyms;
2684edebbeaSRusty Russell 	Elf64_Sym *syms;
2694edebbeaSRusty Russell 
2704edebbeaSRusty Russell 	syms = (Elf64_Sym *)sechdrs[symindex].sh_addr;
2714edebbeaSRusty Russell 	numsyms = sechdrs[symindex].sh_size / sizeof(Elf64_Sym);
2724edebbeaSRusty Russell 
2734edebbeaSRusty Russell 	for (i = 1; i < numsyms; i++) {
274c153693dSAlan Modra 		if (syms[i].st_shndx == SHN_ABS
275008d7a91SRusty Russell 		    && strcmp(strtab + syms[i].st_name, "TOC.") == 0)
2764edebbeaSRusty Russell 			return &syms[i];
2774edebbeaSRusty Russell 	}
2784edebbeaSRusty Russell 	return NULL;
2794edebbeaSRusty Russell }
2804edebbeaSRusty Russell 
281d4be60feSWedson Almeida Filho bool module_init_section(const char *name)
282d4be60feSWedson Almeida Filho {
283d4be60feSWedson Almeida Filho 	/* We don't handle .init for the moment: always return false. */
284d4be60feSWedson Almeida Filho 	return false;
285d4be60feSWedson Almeida Filho }
286d4be60feSWedson Almeida Filho 
2877568cb4eSPaul Mackerras int module_frob_arch_sections(Elf64_Ehdr *hdr,
2887568cb4eSPaul Mackerras 			      Elf64_Shdr *sechdrs,
2897568cb4eSPaul Mackerras 			      char *secstrings,
2907568cb4eSPaul Mackerras 			      struct module *me)
2917568cb4eSPaul Mackerras {
2927568cb4eSPaul Mackerras 	unsigned int i;
2937568cb4eSPaul Mackerras 
2947568cb4eSPaul Mackerras 	/* Find .toc and .stubs sections, symtab and strtab */
2957568cb4eSPaul Mackerras 	for (i = 1; i < hdr->e_shnum; i++) {
2967568cb4eSPaul Mackerras 		if (strcmp(secstrings + sechdrs[i].sh_name, ".stubs") == 0)
2977568cb4eSPaul Mackerras 			me->arch.stubs_section = i;
2985c45b528SAlan Modra 		else if (strcmp(secstrings + sechdrs[i].sh_name, ".toc") == 0) {
2997568cb4eSPaul Mackerras 			me->arch.toc_section = i;
3005c45b528SAlan Modra 			if (sechdrs[i].sh_addralign < 8)
3015c45b528SAlan Modra 				sechdrs[i].sh_addralign = 8;
3025c45b528SAlan Modra 		}
3037568cb4eSPaul Mackerras 		else if (strcmp(secstrings+sechdrs[i].sh_name,"__versions")==0)
3047568cb4eSPaul Mackerras 			dedotify_versions((void *)hdr + sechdrs[i].sh_offset,
3057568cb4eSPaul Mackerras 					  sechdrs[i].sh_size);
3067568cb4eSPaul Mackerras 
3077568cb4eSPaul Mackerras 		if (sechdrs[i].sh_type == SHT_SYMTAB)
3087568cb4eSPaul Mackerras 			dedotify((void *)hdr + sechdrs[i].sh_offset,
3097568cb4eSPaul Mackerras 				 sechdrs[i].sh_size / sizeof(Elf64_Sym),
3107568cb4eSPaul Mackerras 				 (void *)hdr
3117568cb4eSPaul Mackerras 				 + sechdrs[sechdrs[i].sh_link].sh_offset);
3127568cb4eSPaul Mackerras 	}
313f749edaeSAlan Modra 
314f749edaeSAlan Modra 	if (!me->arch.stubs_section) {
315c7d1f6afSAnton Blanchard 		pr_err("%s: doesn't contain .stubs.\n", me->name);
3167568cb4eSPaul Mackerras 		return -ENOEXEC;
3177568cb4eSPaul Mackerras 	}
3187568cb4eSPaul Mackerras 
319f749edaeSAlan Modra 	/* If we don't have a .toc, just use .stubs.  We need to set r2
320f749edaeSAlan Modra 	   to some reasonable value in case the module calls out to
321f749edaeSAlan Modra 	   other functions via a stub, or if a function pointer escapes
322f749edaeSAlan Modra 	   the module by some means.  */
323f749edaeSAlan Modra 	if (!me->arch.toc_section)
324f749edaeSAlan Modra 		me->arch.toc_section = me->arch.stubs_section;
325f749edaeSAlan Modra 
3267568cb4eSPaul Mackerras 	/* Override the stubs size */
3277568cb4eSPaul Mackerras 	sechdrs[me->arch.stubs_section].sh_size = get_stubs_size(hdr, sechdrs);
3287568cb4eSPaul Mackerras 	return 0;
3297568cb4eSPaul Mackerras }
3307568cb4eSPaul Mackerras 
331bd55e792SNaveen N. Rao #ifdef CONFIG_MPROFILE_KERNEL
332bd55e792SNaveen N. Rao 
333bd55e792SNaveen N. Rao static u32 stub_insns[] = {
33447b04699SChristophe Leroy 	PPC_RAW_LD(_R12, _R13, offsetof(struct paca_struct, kernel_toc)),
33547b04699SChristophe Leroy 	PPC_RAW_ADDIS(_R12, _R12, 0),
33647b04699SChristophe Leroy 	PPC_RAW_ADDI(_R12, _R12, 0),
33747b04699SChristophe Leroy 	PPC_RAW_MTCTR(_R12),
33847b04699SChristophe Leroy 	PPC_RAW_BCTR(),
339bd55e792SNaveen N. Rao };
340bd55e792SNaveen N. Rao 
341bd55e792SNaveen N. Rao /*
342bd55e792SNaveen N. Rao  * For mprofile-kernel we use a special stub for ftrace_caller() because we
343bd55e792SNaveen N. Rao  * can't rely on r2 containing this module's TOC when we enter the stub.
344bd55e792SNaveen N. Rao  *
345bd55e792SNaveen N. Rao  * That can happen if the function calling us didn't need to use the toc. In
346bd55e792SNaveen N. Rao  * that case it won't have setup r2, and the r2 value will be either the
347bd55e792SNaveen N. Rao  * kernel's toc, or possibly another modules toc.
348bd55e792SNaveen N. Rao  *
349bd55e792SNaveen N. Rao  * To deal with that this stub uses the kernel toc, which is always accessible
350bd55e792SNaveen N. Rao  * via the paca (in r13). The target (ftrace_caller()) is responsible for
351bd55e792SNaveen N. Rao  * saving and restoring the toc before returning.
352bd55e792SNaveen N. Rao  */
353bd55e792SNaveen N. Rao static inline int create_ftrace_stub(struct ppc64_stub_entry *entry,
354bd55e792SNaveen N. Rao 					unsigned long addr,
355bd55e792SNaveen N. Rao 					struct module *me)
356bd55e792SNaveen N. Rao {
357bd55e792SNaveen N. Rao 	long reladdr;
358bd55e792SNaveen N. Rao 
359bd55e792SNaveen N. Rao 	memcpy(entry->jump, stub_insns, sizeof(stub_insns));
360bd55e792SNaveen N. Rao 
361bd55e792SNaveen N. Rao 	/* Stub uses address relative to kernel toc (from the paca) */
362bd55e792SNaveen N. Rao 	reladdr = addr - kernel_toc_addr();
363bd55e792SNaveen N. Rao 	if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) {
364bd55e792SNaveen N. Rao 		pr_err("%s: Address of %ps out of range of kernel_toc.\n",
365bd55e792SNaveen N. Rao 							me->name, (void *)addr);
366bd55e792SNaveen N. Rao 		return 0;
367bd55e792SNaveen N. Rao 	}
368bd55e792SNaveen N. Rao 
369bd55e792SNaveen N. Rao 	entry->jump[1] |= PPC_HA(reladdr);
370bd55e792SNaveen N. Rao 	entry->jump[2] |= PPC_LO(reladdr);
371bd55e792SNaveen N. Rao 
372bd55e792SNaveen N. Rao 	/* Eventhough we don't use funcdata in the stub, it's needed elsewhere. */
373bd55e792SNaveen N. Rao 	entry->funcdata = func_desc(addr);
374bd55e792SNaveen N. Rao 	entry->magic = STUB_MAGIC;
375bd55e792SNaveen N. Rao 
376bd55e792SNaveen N. Rao 	return 1;
377bd55e792SNaveen N. Rao }
378bd55e792SNaveen N. Rao 
379bd55e792SNaveen N. Rao static bool is_mprofile_ftrace_call(const char *name)
380bd55e792SNaveen N. Rao {
381bd55e792SNaveen N. Rao 	if (!strcmp("_mcount", name))
382bd55e792SNaveen N. Rao 		return true;
383bd55e792SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE
384bd55e792SNaveen N. Rao 	if (!strcmp("ftrace_caller", name))
385bd55e792SNaveen N. Rao 		return true;
386bd55e792SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
387bd55e792SNaveen N. Rao 	if (!strcmp("ftrace_regs_caller", name))
388bd55e792SNaveen N. Rao 		return true;
389bd55e792SNaveen N. Rao #endif
390bd55e792SNaveen N. Rao #endif
391bd55e792SNaveen N. Rao 
392bd55e792SNaveen N. Rao 	return false;
393bd55e792SNaveen N. Rao }
394bd55e792SNaveen N. Rao #else
395bd55e792SNaveen N. Rao static inline int create_ftrace_stub(struct ppc64_stub_entry *entry,
396bd55e792SNaveen N. Rao 					unsigned long addr,
397bd55e792SNaveen N. Rao 					struct module *me)
398bd55e792SNaveen N. Rao {
399bd55e792SNaveen N. Rao 	return 0;
400bd55e792SNaveen N. Rao }
401bd55e792SNaveen N. Rao 
402bd55e792SNaveen N. Rao static bool is_mprofile_ftrace_call(const char *name)
403bd55e792SNaveen N. Rao {
404bd55e792SNaveen N. Rao 	return false;
405bd55e792SNaveen N. Rao }
406bd55e792SNaveen N. Rao #endif
407bd55e792SNaveen N. Rao 
4085c45b528SAlan Modra /*
4095c45b528SAlan Modra  * r2 is the TOC pointer: it actually points 0x8000 into the TOC (this gives the
4105c45b528SAlan Modra  * value maximum span in an instruction which uses a signed offset). Round down
4115c45b528SAlan Modra  * to a 256 byte boundary for the odd case where we are setting up r2 without a
4125c45b528SAlan Modra  * .toc section.
4135c45b528SAlan Modra  */
414136cd345SMichael Ellerman static inline unsigned long my_r2(const Elf64_Shdr *sechdrs, struct module *me)
4157568cb4eSPaul Mackerras {
4165c45b528SAlan Modra 	return (sechdrs[me->arch.toc_section].sh_addr & ~0xfful) + 0x8000;
4177568cb4eSPaul Mackerras }
4187568cb4eSPaul Mackerras 
4197568cb4eSPaul Mackerras /* Patch stub to reference function and correct r2 value. */
420136cd345SMichael Ellerman static inline int create_stub(const Elf64_Shdr *sechdrs,
4217568cb4eSPaul Mackerras 			      struct ppc64_stub_entry *entry,
422008d7a91SRusty Russell 			      unsigned long addr,
423bd55e792SNaveen N. Rao 			      struct module *me,
424bd55e792SNaveen N. Rao 			      const char *name)
4257568cb4eSPaul Mackerras {
4267568cb4eSPaul Mackerras 	long reladdr;
4278734b41bSRussell Currey 	func_desc_t desc;
4288734b41bSRussell Currey 	int i;
4297568cb4eSPaul Mackerras 
430bd55e792SNaveen N. Rao 	if (is_mprofile_ftrace_call(name))
431bd55e792SNaveen N. Rao 		return create_ftrace_stub(entry, addr, me);
432bd55e792SNaveen N. Rao 
433*8a0edc72SGuo Zhengkui 	for (i = 0; i < ARRAY_SIZE(ppc64_stub_insns); i++) {
4348734b41bSRussell Currey 		if (patch_instruction(&entry->jump[i],
4358734b41bSRussell Currey 				      ppc_inst(ppc64_stub_insns[i])))
4368734b41bSRussell Currey 			return 0;
4378734b41bSRussell Currey 	}
4387568cb4eSPaul Mackerras 
4397568cb4eSPaul Mackerras 	/* Stub uses address relative to r2. */
4407568cb4eSPaul Mackerras 	reladdr = (unsigned long)entry - my_r2(sechdrs, me);
4417568cb4eSPaul Mackerras 	if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) {
442c7d1f6afSAnton Blanchard 		pr_err("%s: Address %p of stub out of range of %p.\n",
4437568cb4eSPaul Mackerras 		       me->name, (void *)reladdr, (void *)my_r2);
4447568cb4eSPaul Mackerras 		return 0;
4457568cb4eSPaul Mackerras 	}
446c7d1f6afSAnton Blanchard 	pr_debug("Stub %p get data from reladdr %li\n", entry, reladdr);
4477568cb4eSPaul Mackerras 
4488734b41bSRussell Currey 	if (patch_instruction(&entry->jump[0],
4498734b41bSRussell Currey 			      ppc_inst(entry->jump[0] | PPC_HA(reladdr))))
4508734b41bSRussell Currey 		return 0;
4518734b41bSRussell Currey 
4528734b41bSRussell Currey 	if (patch_instruction(&entry->jump[1],
4538734b41bSRussell Currey 			  ppc_inst(entry->jump[1] | PPC_LO(reladdr))))
4548734b41bSRussell Currey 		return 0;
4558734b41bSRussell Currey 
4568734b41bSRussell Currey 	// func_desc_t is 8 bytes if ABIv2, else 16 bytes
4578734b41bSRussell Currey 	desc = func_desc(addr);
4588734b41bSRussell Currey 	for (i = 0; i < sizeof(func_desc_t) / sizeof(u32); i++) {
4598734b41bSRussell Currey 		if (patch_instruction(((u32 *)&entry->funcdata) + i,
4608734b41bSRussell Currey 				      ppc_inst(((u32 *)(&desc))[i])))
4618734b41bSRussell Currey 			return 0;
4628734b41bSRussell Currey 	}
4638734b41bSRussell Currey 
4648734b41bSRussell Currey 	if (patch_instruction(&entry->magic, ppc_inst(STUB_MAGIC)))
4658734b41bSRussell Currey 		return 0;
466f17c4e01SMichael Ellerman 
4677568cb4eSPaul Mackerras 	return 1;
4687568cb4eSPaul Mackerras }
4697568cb4eSPaul Mackerras 
470008d7a91SRusty Russell /* Create stub to jump to function described in this OPD/ptr: we need the
4717568cb4eSPaul Mackerras    stub to set up the TOC ptr (r2) for the function. */
472136cd345SMichael Ellerman static unsigned long stub_for_addr(const Elf64_Shdr *sechdrs,
473008d7a91SRusty Russell 				   unsigned long addr,
474bd55e792SNaveen N. Rao 				   struct module *me,
475bd55e792SNaveen N. Rao 				   const char *name)
4767568cb4eSPaul Mackerras {
4777568cb4eSPaul Mackerras 	struct ppc64_stub_entry *stubs;
4787568cb4eSPaul Mackerras 	unsigned int i, num_stubs;
4797568cb4eSPaul Mackerras 
4807568cb4eSPaul Mackerras 	num_stubs = sechdrs[me->arch.stubs_section].sh_size / sizeof(*stubs);
4817568cb4eSPaul Mackerras 
4827568cb4eSPaul Mackerras 	/* Find this stub, or if that fails, the next avail. entry */
4837568cb4eSPaul Mackerras 	stubs = (void *)sechdrs[me->arch.stubs_section].sh_addr;
484008d7a91SRusty Russell 	for (i = 0; stub_func_addr(stubs[i].funcdata); i++) {
4851c0437afSKamalesh Babulal 		if (WARN_ON(i >= num_stubs))
4861c0437afSKamalesh Babulal 			return 0;
4877568cb4eSPaul Mackerras 
488008d7a91SRusty Russell 		if (stub_func_addr(stubs[i].funcdata) == func_addr(addr))
4897568cb4eSPaul Mackerras 			return (unsigned long)&stubs[i];
4907568cb4eSPaul Mackerras 	}
4917568cb4eSPaul Mackerras 
492bd55e792SNaveen N. Rao 	if (!create_stub(sechdrs, &stubs[i], addr, me, name))
4937568cb4eSPaul Mackerras 		return 0;
4947568cb4eSPaul Mackerras 
4957568cb4eSPaul Mackerras 	return (unsigned long)&stubs[i];
4967568cb4eSPaul Mackerras }
4977568cb4eSPaul Mackerras 
4987568cb4eSPaul Mackerras /* We expect a noop next: if it is, replace it with instruction to
4997568cb4eSPaul Mackerras    restore r2. */
500250122baSNaveen N. Rao static int restore_r2(const char *name, u32 *instruction, struct module *me)
5017568cb4eSPaul Mackerras {
502b9eab08dSJosh Poimboeuf 	u32 *prev_insn = instruction - 1;
503b9eab08dSJosh Poimboeuf 
5041f2aaed2SNaveen N. Rao 	if (is_mprofile_ftrace_call(name))
505b9eab08dSJosh Poimboeuf 		return 1;
506b9eab08dSJosh Poimboeuf 
507b9eab08dSJosh Poimboeuf 	/*
508b9eab08dSJosh Poimboeuf 	 * Make sure the branch isn't a sibling call.  Sibling calls aren't
509b9eab08dSJosh Poimboeuf 	 * "link" branches and they don't return, so they don't need the r2
510b9eab08dSJosh Poimboeuf 	 * restore afterwards.
511b9eab08dSJosh Poimboeuf 	 */
51275346251SJordan Niethe 	if (!instr_is_relative_link_branch(ppc_inst(*prev_insn)))
51315308664STorsten Duwe 		return 1;
51431278b17SMichael Ellerman 
51547b04699SChristophe Leroy 	if (*instruction != PPC_RAW_NOP()) {
5161ea61ea2SJosh Poimboeuf 		pr_err("%s: Expected nop after call, got %08x at %pS\n",
5171ea61ea2SJosh Poimboeuf 			me->name, *instruction, instruction);
5187568cb4eSPaul Mackerras 		return 0;
5197568cb4eSPaul Mackerras 	}
5208734b41bSRussell Currey 
521d2fae548SRusty Russell 	/* ld r2,R2_STACK_OFFSET(r1) */
5228734b41bSRussell Currey 	if (patch_instruction(instruction, ppc_inst(PPC_INST_LD_TOC)))
5238734b41bSRussell Currey 		return 0;
5248734b41bSRussell Currey 
5257568cb4eSPaul Mackerras 	return 1;
5267568cb4eSPaul Mackerras }
5277568cb4eSPaul Mackerras 
5287568cb4eSPaul Mackerras int apply_relocate_add(Elf64_Shdr *sechdrs,
5297568cb4eSPaul Mackerras 		       const char *strtab,
5307568cb4eSPaul Mackerras 		       unsigned int symindex,
5317568cb4eSPaul Mackerras 		       unsigned int relsec,
5327568cb4eSPaul Mackerras 		       struct module *me)
5337568cb4eSPaul Mackerras {
5347568cb4eSPaul Mackerras 	unsigned int i;
5357568cb4eSPaul Mackerras 	Elf64_Rela *rela = (void *)sechdrs[relsec].sh_addr;
5367568cb4eSPaul Mackerras 	Elf64_Sym *sym;
5377568cb4eSPaul Mackerras 	unsigned long *location;
5387568cb4eSPaul Mackerras 	unsigned long value;
5397568cb4eSPaul Mackerras 
540c7d1f6afSAnton Blanchard 	pr_debug("Applying ADD relocate section %u to %u\n", relsec,
5417568cb4eSPaul Mackerras 	       sechdrs[relsec].sh_info);
5424edebbeaSRusty Russell 
5434edebbeaSRusty Russell 	/* First time we're called, we can fix up .TOC. */
5444edebbeaSRusty Russell 	if (!me->arch.toc_fixed) {
5454edebbeaSRusty Russell 		sym = find_dot_toc(sechdrs, strtab, symindex);
5464edebbeaSRusty Russell 		/* It's theoretically possible that a module doesn't want a
5474edebbeaSRusty Russell 		 * .TOC. so don't fail it just for that. */
5484edebbeaSRusty Russell 		if (sym)
5494edebbeaSRusty Russell 			sym->st_value = my_r2(sechdrs, me);
5504edebbeaSRusty Russell 		me->arch.toc_fixed = true;
5514edebbeaSRusty Russell 	}
5524edebbeaSRusty Russell 
5537568cb4eSPaul Mackerras 	for (i = 0; i < sechdrs[relsec].sh_size / sizeof(*rela); i++) {
5547568cb4eSPaul Mackerras 		/* This is where to make the change */
5557568cb4eSPaul Mackerras 		location = (void *)sechdrs[sechdrs[relsec].sh_info].sh_addr
5567568cb4eSPaul Mackerras 			+ rela[i].r_offset;
5577568cb4eSPaul Mackerras 		/* This is the symbol it is referring to */
5587568cb4eSPaul Mackerras 		sym = (Elf64_Sym *)sechdrs[symindex].sh_addr
5597568cb4eSPaul Mackerras 			+ ELF64_R_SYM(rela[i].r_info);
5607568cb4eSPaul Mackerras 
561c7d1f6afSAnton Blanchard 		pr_debug("RELOC at %p: %li-type as %s (0x%lx) + %li\n",
5627568cb4eSPaul Mackerras 		       location, (long)ELF64_R_TYPE(rela[i].r_info),
5637568cb4eSPaul Mackerras 		       strtab + sym->st_name, (unsigned long)sym->st_value,
5647568cb4eSPaul Mackerras 		       (long)rela[i].r_addend);
5657568cb4eSPaul Mackerras 
5667568cb4eSPaul Mackerras 		/* `Everything is relative'. */
5677568cb4eSPaul Mackerras 		value = sym->st_value + rela[i].r_addend;
5687568cb4eSPaul Mackerras 
5697568cb4eSPaul Mackerras 		switch (ELF64_R_TYPE(rela[i].r_info)) {
5707568cb4eSPaul Mackerras 		case R_PPC64_ADDR32:
5717568cb4eSPaul Mackerras 			/* Simply set it */
5727568cb4eSPaul Mackerras 			*(u32 *)location = value;
5737568cb4eSPaul Mackerras 			break;
5747568cb4eSPaul Mackerras 
5757568cb4eSPaul Mackerras 		case R_PPC64_ADDR64:
5767568cb4eSPaul Mackerras 			/* Simply set it */
5777568cb4eSPaul Mackerras 			*(unsigned long *)location = value;
5787568cb4eSPaul Mackerras 			break;
5797568cb4eSPaul Mackerras 
5807568cb4eSPaul Mackerras 		case R_PPC64_TOC:
5817568cb4eSPaul Mackerras 			*(unsigned long *)location = my_r2(sechdrs, me);
5827568cb4eSPaul Mackerras 			break;
5837568cb4eSPaul Mackerras 
5847568cb4eSPaul Mackerras 		case R_PPC64_TOC16:
585f749edaeSAlan Modra 			/* Subtract TOC pointer */
5867568cb4eSPaul Mackerras 			value -= my_r2(sechdrs, me);
5877568cb4eSPaul Mackerras 			if (value + 0x8000 > 0xffff) {
588c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16 relocation (0x%lx)\n",
5897568cb4eSPaul Mackerras 				       me->name, value);
5907568cb4eSPaul Mackerras 				return -ENOEXEC;
5917568cb4eSPaul Mackerras 			}
5927568cb4eSPaul Mackerras 			*((uint16_t *) location)
5937568cb4eSPaul Mackerras 				= (*((uint16_t *) location) & ~0xffff)
5947568cb4eSPaul Mackerras 				| (value & 0xffff);
5957568cb4eSPaul Mackerras 			break;
5967568cb4eSPaul Mackerras 
5971fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_LO:
5981fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
5991fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6001fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6011fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xffff)
6021fbe9cf2SAnton Blanchard 				| (value & 0xffff);
6031fbe9cf2SAnton Blanchard 			break;
6041fbe9cf2SAnton Blanchard 
6057568cb4eSPaul Mackerras 		case R_PPC64_TOC16_DS:
606f749edaeSAlan Modra 			/* Subtract TOC pointer */
6077568cb4eSPaul Mackerras 			value -= my_r2(sechdrs, me);
6087568cb4eSPaul Mackerras 			if ((value & 3) != 0 || value + 0x8000 > 0xffff) {
609c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16_DS relocation (0x%lx)\n",
6107568cb4eSPaul Mackerras 				       me->name, value);
6117568cb4eSPaul Mackerras 				return -ENOEXEC;
6127568cb4eSPaul Mackerras 			}
6137568cb4eSPaul Mackerras 			*((uint16_t *) location)
6147568cb4eSPaul Mackerras 				= (*((uint16_t *) location) & ~0xfffc)
6157568cb4eSPaul Mackerras 				| (value & 0xfffc);
6167568cb4eSPaul Mackerras 			break;
6177568cb4eSPaul Mackerras 
6181fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_LO_DS:
6191fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
6201fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6211fbe9cf2SAnton Blanchard 			if ((value & 3) != 0) {
622c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16_LO_DS relocation (0x%lx)\n",
6231fbe9cf2SAnton Blanchard 				       me->name, value);
6241fbe9cf2SAnton Blanchard 				return -ENOEXEC;
6251fbe9cf2SAnton Blanchard 			}
6261fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6271fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xfffc)
6281fbe9cf2SAnton Blanchard 				| (value & 0xfffc);
6291fbe9cf2SAnton Blanchard 			break;
6301fbe9cf2SAnton Blanchard 
6311fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_HA:
6321fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
6331fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6341fbe9cf2SAnton Blanchard 			value = ((value + 0x8000) >> 16);
6351fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6361fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xffff)
6371fbe9cf2SAnton Blanchard 				| (value & 0xffff);
6381fbe9cf2SAnton Blanchard 			break;
6391fbe9cf2SAnton Blanchard 
6407568cb4eSPaul Mackerras 		case R_PPC_REL24:
6417568cb4eSPaul Mackerras 			/* FIXME: Handle weak symbols here --RR */
642a443bf6eSKamalesh Babulal 			if (sym->st_shndx == SHN_UNDEF ||
643a443bf6eSKamalesh Babulal 			    sym->st_shndx == SHN_LIVEPATCH) {
6447568cb4eSPaul Mackerras 				/* External: go via stub */
645bd55e792SNaveen N. Rao 				value = stub_for_addr(sechdrs, value, me,
646bd55e792SNaveen N. Rao 						strtab + sym->st_name);
6477568cb4eSPaul Mackerras 				if (!value)
6487568cb4eSPaul Mackerras 					return -ENOENT;
649250122baSNaveen N. Rao 				if (!restore_r2(strtab + sym->st_name,
650250122baSNaveen N. Rao 							(u32 *)location + 1, me))
6517568cb4eSPaul Mackerras 					return -ENOEXEC;
652008d7a91SRusty Russell 			} else
653008d7a91SRusty Russell 				value += local_entry_offset(sym);
6547568cb4eSPaul Mackerras 
6557568cb4eSPaul Mackerras 			/* Convert value to relative */
6567568cb4eSPaul Mackerras 			value -= (unsigned long)location;
6577568cb4eSPaul Mackerras 			if (value + 0x2000000 > 0x3ffffff || (value & 3) != 0){
658c7d1f6afSAnton Blanchard 				pr_err("%s: REL24 %li out of range!\n",
6597568cb4eSPaul Mackerras 				       me->name, (long int)value);
6607568cb4eSPaul Mackerras 				return -ENOEXEC;
6617568cb4eSPaul Mackerras 			}
6627568cb4eSPaul Mackerras 
6637568cb4eSPaul Mackerras 			/* Only replace bits 2 through 26 */
6648734b41bSRussell Currey 			value = (*(uint32_t *)location & ~0x03fffffc)
6657568cb4eSPaul Mackerras 				| (value & 0x03fffffc);
6668734b41bSRussell Currey 
6678734b41bSRussell Currey 			if (patch_instruction((u32 *)location, ppc_inst(value)))
6688734b41bSRussell Currey 				return -EFAULT;
6698734b41bSRussell Currey 
6707568cb4eSPaul Mackerras 			break;
6717568cb4eSPaul Mackerras 
67221c4ff80SBenjamin Herrenschmidt 		case R_PPC64_REL64:
67321c4ff80SBenjamin Herrenschmidt 			/* 64 bits relative (used by features fixups) */
67421c4ff80SBenjamin Herrenschmidt 			*location = value - (unsigned long)location;
67521c4ff80SBenjamin Herrenschmidt 			break;
67621c4ff80SBenjamin Herrenschmidt 
6779f751b82SMichael Ellerman 		case R_PPC64_REL32:
6789f751b82SMichael Ellerman 			/* 32 bits relative (used by relative exception tables) */
679b851ba02SNicholas Piggin 			/* Convert value to relative */
680b851ba02SNicholas Piggin 			value -= (unsigned long)location;
681b851ba02SNicholas Piggin 			if (value + 0x80000000 > 0xffffffff) {
682b851ba02SNicholas Piggin 				pr_err("%s: REL32 %li out of range!\n",
683b851ba02SNicholas Piggin 				       me->name, (long int)value);
684b851ba02SNicholas Piggin 				return -ENOEXEC;
685b851ba02SNicholas Piggin 			}
686b851ba02SNicholas Piggin 			*(u32 *)location = value;
6879f751b82SMichael Ellerman 			break;
6889f751b82SMichael Ellerman 
689d247da0aSRusty Russell 		case R_PPC64_TOCSAVE:
690d247da0aSRusty Russell 			/*
691d247da0aSRusty Russell 			 * Marker reloc indicates we don't have to save r2.
692d247da0aSRusty Russell 			 * That would only save us one instruction, so ignore
693d247da0aSRusty Russell 			 * it.
694d247da0aSRusty Russell 			 */
695d247da0aSRusty Russell 			break;
696d247da0aSRusty Russell 
697a61674bdSUlrich Weigand 		case R_PPC64_ENTRY:
698a61674bdSUlrich Weigand 			/*
699a61674bdSUlrich Weigand 			 * Optimize ELFv2 large code model entry point if
700a61674bdSUlrich Weigand 			 * the TOC is within 2GB range of current location.
701a61674bdSUlrich Weigand 			 */
702a61674bdSUlrich Weigand 			value = my_r2(sechdrs, me) - (unsigned long)location;
703a61674bdSUlrich Weigand 			if (value + 0x80008000 > 0xffffffff)
704a61674bdSUlrich Weigand 				break;
705a61674bdSUlrich Weigand 			/*
706a61674bdSUlrich Weigand 			 * Check for the large code model prolog sequence:
707a61674bdSUlrich Weigand 		         *	ld r2, ...(r12)
708a61674bdSUlrich Weigand 			 *	add r2, r2, r12
709a61674bdSUlrich Weigand 			 */
71047b04699SChristophe Leroy 			if ((((uint32_t *)location)[0] & ~0xfffc) != PPC_RAW_LD(_R2, _R12, 0))
711a61674bdSUlrich Weigand 				break;
71247b04699SChristophe Leroy 			if (((uint32_t *)location)[1] != PPC_RAW_ADD(_R2, _R2, _R12))
713a61674bdSUlrich Weigand 				break;
714a61674bdSUlrich Weigand 			/*
715a61674bdSUlrich Weigand 			 * If found, replace it with:
716a61674bdSUlrich Weigand 			 *	addis r2, r12, (.TOC.-func)@ha
7172fb0a2c9SMichael Ellerman 			 *	addi  r2,  r2, (.TOC.-func)@l
718a61674bdSUlrich Weigand 			 */
71947b04699SChristophe Leroy 			((uint32_t *)location)[0] = PPC_RAW_ADDIS(_R2, _R12, PPC_HA(value));
72047b04699SChristophe Leroy 			((uint32_t *)location)[1] = PPC_RAW_ADDI(_R2, _R2, PPC_LO(value));
721a61674bdSUlrich Weigand 			break;
722a61674bdSUlrich Weigand 
7230906584aSRusty Russell 		case R_PPC64_REL16_HA:
7240906584aSRusty Russell 			/* Subtract location pointer */
7250906584aSRusty Russell 			value -= (unsigned long)location;
7260906584aSRusty Russell 			value = ((value + 0x8000) >> 16);
7270906584aSRusty Russell 			*((uint16_t *) location)
7280906584aSRusty Russell 				= (*((uint16_t *) location) & ~0xffff)
7290906584aSRusty Russell 				| (value & 0xffff);
7300906584aSRusty Russell 			break;
7310906584aSRusty Russell 
7320906584aSRusty Russell 		case R_PPC64_REL16_LO:
7330906584aSRusty Russell 			/* Subtract location pointer */
7340906584aSRusty Russell 			value -= (unsigned long)location;
7350906584aSRusty Russell 			*((uint16_t *) location)
7360906584aSRusty Russell 				= (*((uint16_t *) location) & ~0xffff)
7370906584aSRusty Russell 				| (value & 0xffff);
7380906584aSRusty Russell 			break;
7390906584aSRusty Russell 
7407568cb4eSPaul Mackerras 		default:
741c7d1f6afSAnton Blanchard 			pr_err("%s: Unknown ADD relocation: %lu\n",
7427568cb4eSPaul Mackerras 			       me->name,
7437568cb4eSPaul Mackerras 			       (unsigned long)ELF64_R_TYPE(rela[i].r_info));
7447568cb4eSPaul Mackerras 			return -ENOEXEC;
7457568cb4eSPaul Mackerras 		}
7467568cb4eSPaul Mackerras 	}
7477568cb4eSPaul Mackerras 
748136cd345SMichael Ellerman 	return 0;
749136cd345SMichael Ellerman }
750136cd345SMichael Ellerman 
751f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
75203b51416SNaveen N. Rao int module_trampoline_target(struct module *mod, unsigned long addr,
75303b51416SNaveen N. Rao 			     unsigned long *target)
75403b51416SNaveen N. Rao {
75503b51416SNaveen N. Rao 	struct ppc64_stub_entry *stub;
75603b51416SNaveen N. Rao 	func_desc_t funcdata;
75703b51416SNaveen N. Rao 	u32 magic;
75803b51416SNaveen N. Rao 
75903b51416SNaveen N. Rao 	if (!within_module_core(addr, mod)) {
76003b51416SNaveen N. Rao 		pr_err("%s: stub %lx not in module %s\n", __func__, addr, mod->name);
76103b51416SNaveen N. Rao 		return -EFAULT;
76203b51416SNaveen N. Rao 	}
76303b51416SNaveen N. Rao 
76403b51416SNaveen N. Rao 	stub = (struct ppc64_stub_entry *)addr;
76503b51416SNaveen N. Rao 
766fe557319SChristoph Hellwig 	if (copy_from_kernel_nofault(&magic, &stub->magic,
767fe557319SChristoph Hellwig 			sizeof(magic))) {
76803b51416SNaveen N. Rao 		pr_err("%s: fault reading magic for stub %lx for %s\n", __func__, addr, mod->name);
76903b51416SNaveen N. Rao 		return -EFAULT;
77003b51416SNaveen N. Rao 	}
77103b51416SNaveen N. Rao 
77203b51416SNaveen N. Rao 	if (magic != STUB_MAGIC) {
77303b51416SNaveen N. Rao 		pr_err("%s: bad magic for stub %lx for %s\n", __func__, addr, mod->name);
77403b51416SNaveen N. Rao 		return -EFAULT;
77503b51416SNaveen N. Rao 	}
77603b51416SNaveen N. Rao 
777fe557319SChristoph Hellwig 	if (copy_from_kernel_nofault(&funcdata, &stub->funcdata,
778fe557319SChristoph Hellwig 			sizeof(funcdata))) {
77903b51416SNaveen N. Rao 		pr_err("%s: fault reading funcdata for stub %lx for %s\n", __func__, addr, mod->name);
78003b51416SNaveen N. Rao                 return -EFAULT;
78103b51416SNaveen N. Rao 	}
78203b51416SNaveen N. Rao 
78303b51416SNaveen N. Rao 	*target = stub_func_addr(funcdata);
78403b51416SNaveen N. Rao 
78503b51416SNaveen N. Rao 	return 0;
78603b51416SNaveen N. Rao }
787336a7b5dSMichael Ellerman 
788136cd345SMichael Ellerman int module_finalize_ftrace(struct module *mod, const Elf_Shdr *sechdrs)
789136cd345SMichael Ellerman {
790bd55e792SNaveen N. Rao 	mod->arch.tramp = stub_for_addr(sechdrs,
791bd55e792SNaveen N. Rao 					(unsigned long)ftrace_caller,
792bd55e792SNaveen N. Rao 					mod,
793bd55e792SNaveen N. Rao 					"ftrace_caller");
794ae30cc05SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
795bd55e792SNaveen N. Rao 	mod->arch.tramp_regs = stub_for_addr(sechdrs,
796bd55e792SNaveen N. Rao 					(unsigned long)ftrace_regs_caller,
797bd55e792SNaveen N. Rao 					mod,
798bd55e792SNaveen N. Rao 					"ftrace_regs_caller");
799ae30cc05SNaveen N. Rao 	if (!mod->arch.tramp_regs)
800ae30cc05SNaveen N. Rao 		return -ENOENT;
801ae30cc05SNaveen N. Rao #endif
802136cd345SMichael Ellerman 
803136cd345SMichael Ellerman 	if (!mod->arch.tramp)
804136cd345SMichael Ellerman 		return -ENOENT;
805f48cb8b4SSteven Rostedt 
8067568cb4eSPaul Mackerras 	return 0;
8077568cb4eSPaul Mackerras }
808136cd345SMichael Ellerman #endif
809