xref: /linux-6.15/arch/powerpc/kernel/module_64.c (revision d4be60fe)
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>
177568cb4eSPaul Mackerras #include <asm/module.h>
1821c4ff80SBenjamin Herrenschmidt #include <asm/firmware.h>
19b7bcda63SMichael Ellerman #include <asm/code-patching.h>
20eda09fbdSEmil Medve #include <linux/sort.h>
21b88c4767SRobert Jennings #include <asm/setup.h>
22336a7b5dSMichael Ellerman #include <asm/sections.h>
2375346251SJordan Niethe #include <asm/inst.h>
247568cb4eSPaul Mackerras 
257568cb4eSPaul Mackerras /* FIXME: We don't do .init separately.  To do this, we'd need to have
267568cb4eSPaul Mackerras    a separate r2 value in the init and core section, and stub between
277568cb4eSPaul Mackerras    them, too.
287568cb4eSPaul Mackerras 
297568cb4eSPaul Mackerras    Using a magic allocator which places modules within 32MB solves
307568cb4eSPaul Mackerras    this, and makes other things simpler.  Anton?
317568cb4eSPaul Mackerras    --RR.  */
327568cb4eSPaul Mackerras 
33f55d9665SMichael Ellerman #ifdef PPC64_ELF_ABI_v2
34008d7a91SRusty Russell 
35008d7a91SRusty Russell /* An address is simply the address of the function. */
36008d7a91SRusty Russell typedef unsigned long func_desc_t;
37008d7a91SRusty Russell 
38008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr)
39008d7a91SRusty Russell {
40008d7a91SRusty Russell 	return addr;
41008d7a91SRusty Russell }
42008d7a91SRusty Russell static unsigned long func_addr(unsigned long addr)
43008d7a91SRusty Russell {
44008d7a91SRusty Russell 	return addr;
45008d7a91SRusty Russell }
46008d7a91SRusty Russell static unsigned long stub_func_addr(func_desc_t func)
47008d7a91SRusty Russell {
48008d7a91SRusty Russell 	return func;
49008d7a91SRusty Russell }
50008d7a91SRusty Russell 
51008d7a91SRusty Russell /* PowerPC64 specific values for the Elf64_Sym st_other field.  */
52008d7a91SRusty Russell #define STO_PPC64_LOCAL_BIT	5
53008d7a91SRusty Russell #define STO_PPC64_LOCAL_MASK	(7 << STO_PPC64_LOCAL_BIT)
54008d7a91SRusty Russell #define PPC64_LOCAL_ENTRY_OFFSET(other)					\
55008d7a91SRusty Russell  (((1 << (((other) & STO_PPC64_LOCAL_MASK) >> STO_PPC64_LOCAL_BIT)) >> 2) << 2)
56008d7a91SRusty Russell 
57008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym)
58008d7a91SRusty Russell {
59008d7a91SRusty Russell 	/* sym->st_other indicates offset to local entry point
60008d7a91SRusty Russell 	 * (otherwise it will assume r12 is the address of the start
61008d7a91SRusty Russell 	 * of function and try to derive r2 from it). */
62008d7a91SRusty Russell 	return PPC64_LOCAL_ENTRY_OFFSET(sym->st_other);
63008d7a91SRusty Russell }
64d2fae548SRusty Russell #else
65008d7a91SRusty Russell 
66008d7a91SRusty Russell /* An address is address of the OPD entry, which contains address of fn. */
67008d7a91SRusty Russell typedef struct ppc64_opd_entry func_desc_t;
68008d7a91SRusty Russell 
69008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr)
70008d7a91SRusty Russell {
71008d7a91SRusty Russell 	return *(struct ppc64_opd_entry *)addr;
72008d7a91SRusty Russell }
73008d7a91SRusty Russell static unsigned long func_addr(unsigned long addr)
74008d7a91SRusty Russell {
75008d7a91SRusty Russell 	return func_desc(addr).funcaddr;
76008d7a91SRusty Russell }
77008d7a91SRusty Russell static unsigned long stub_func_addr(func_desc_t func)
78008d7a91SRusty Russell {
79008d7a91SRusty Russell 	return func.funcaddr;
80008d7a91SRusty Russell }
81008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym)
82008d7a91SRusty Russell {
83008d7a91SRusty Russell 	return 0;
84008d7a91SRusty Russell }
855633e85bSSergey Senozhatsky 
865633e85bSSergey Senozhatsky void *dereference_module_function_descriptor(struct module *mod, void *ptr)
875633e85bSSergey Senozhatsky {
885633e85bSSergey Senozhatsky 	if (ptr < (void *)mod->arch.start_opd ||
895633e85bSSergey Senozhatsky 			ptr >= (void *)mod->arch.end_opd)
905633e85bSSergey Senozhatsky 		return ptr;
915633e85bSSergey Senozhatsky 
925633e85bSSergey Senozhatsky 	return dereference_function_descriptor(ptr);
935633e85bSSergey Senozhatsky }
94d2fae548SRusty Russell #endif
95d2fae548SRusty Russell 
96f17c4e01SMichael Ellerman #define STUB_MAGIC 0x73747562 /* stub */
97f17c4e01SMichael Ellerman 
987568cb4eSPaul Mackerras /* Like PPC32, we need little trampolines to do > 24-bit jumps (into
997568cb4eSPaul Mackerras    the kernel itself).  But on PPC64, these need to be used for every
1007568cb4eSPaul Mackerras    jump, actually, to reset r2 (TOC+0x8000). */
1017568cb4eSPaul Mackerras struct ppc64_stub_entry
1027568cb4eSPaul Mackerras {
10383775b85SAnton Blanchard 	/* 28 byte jump instruction sequence (7 instructions). We only
10483775b85SAnton Blanchard 	 * need 6 instructions on ABIv2 but we always allocate 7 so
10583775b85SAnton Blanchard 	 * so we don't have to modify the trampoline load instruction. */
1060e60e46eSRusty Russell 	u32 jump[7];
107f17c4e01SMichael Ellerman 	/* Used by ftrace to identify stubs */
108f17c4e01SMichael Ellerman 	u32 magic;
1097568cb4eSPaul Mackerras 	/* Data for the above code */
110008d7a91SRusty Russell 	func_desc_t funcdata;
1117568cb4eSPaul Mackerras };
1127568cb4eSPaul Mackerras 
1135c729a11SRusty Russell /*
1145c729a11SRusty Russell  * PPC64 uses 24 bit jumps, but we need to jump into other modules or
1155c729a11SRusty Russell  * the kernel which may be further.  So we jump to a stub.
1165c729a11SRusty Russell  *
1175c729a11SRusty Russell  * For ELFv1 we need to use this to set up the new r2 value (aka TOC
1185c729a11SRusty Russell  * pointer).  For ELFv2 it's the callee's responsibility to set up the
1195c729a11SRusty Russell  * new r2, but for both we need to save the old r2.
1205c729a11SRusty Russell  *
1215c729a11SRusty Russell  * We could simply patch the new r2 value and function pointer into
1225c729a11SRusty Russell  * the stub, but it's significantly shorter to put these values at the
1235c729a11SRusty Russell  * end of the stub code, and patch the stub address (32-bits relative
1245c729a11SRusty Russell  * to the TOC ptr, r2) into the stub.
1255c729a11SRusty Russell  */
12683775b85SAnton Blanchard static u32 ppc64_stub_insns[] = {
12747b04699SChristophe Leroy 	PPC_RAW_ADDIS(_R11, _R2, 0),
12847b04699SChristophe Leroy 	PPC_RAW_ADDI(_R11, _R11, 0),
129fed8393eSEugene Surovegin 	/* Save current r2 value in magic place on the stack. */
13047b04699SChristophe Leroy 	PPC_RAW_STD(_R2, _R1, R2_STACK_OFFSET),
13147b04699SChristophe Leroy 	PPC_RAW_LD(_R12, _R11, 32),
132f55d9665SMichael Ellerman #ifdef PPC64_ELF_ABI_v1
1335c729a11SRusty Russell 	/* Set up new r2 from function descriptor */
13447b04699SChristophe Leroy 	PPC_RAW_LD(_R2, _R11, 40),
1355c729a11SRusty Russell #endif
13647b04699SChristophe Leroy 	PPC_RAW_MTCTR(_R12),
13747b04699SChristophe Leroy 	PPC_RAW_BCTR(),
13883775b85SAnton Blanchard };
13983775b85SAnton Blanchard 
1407568cb4eSPaul Mackerras /* Count how many different 24-bit relocations (different symbol,
1417568cb4eSPaul Mackerras    different addend) */
1427568cb4eSPaul Mackerras static unsigned int count_relocs(const Elf64_Rela *rela, unsigned int num)
1437568cb4eSPaul Mackerras {
144eda09fbdSEmil Medve 	unsigned int i, r_info, r_addend, _count_relocs;
1457568cb4eSPaul Mackerras 
1467568cb4eSPaul Mackerras 	/* FIXME: Only count external ones --RR */
147eda09fbdSEmil Medve 	_count_relocs = 0;
148eda09fbdSEmil Medve 	r_info = 0;
149eda09fbdSEmil Medve 	r_addend = 0;
150eda09fbdSEmil Medve 	for (i = 0; i < num; i++)
1517568cb4eSPaul Mackerras 		/* Only count 24-bit relocs, others don't need stubs */
152eda09fbdSEmil Medve 		if (ELF64_R_TYPE(rela[i].r_info) == R_PPC_REL24 &&
153eda09fbdSEmil Medve 		    (r_info != ELF64_R_SYM(rela[i].r_info) ||
154eda09fbdSEmil Medve 		     r_addend != rela[i].r_addend)) {
155eda09fbdSEmil Medve 			_count_relocs++;
156eda09fbdSEmil Medve 			r_info = ELF64_R_SYM(rela[i].r_info);
157eda09fbdSEmil Medve 			r_addend = rela[i].r_addend;
1587568cb4eSPaul Mackerras 		}
159eda09fbdSEmil Medve 
160eda09fbdSEmil Medve 	return _count_relocs;
1617568cb4eSPaul Mackerras }
1627568cb4eSPaul Mackerras 
163eda09fbdSEmil Medve static int relacmp(const void *_x, const void *_y)
164eda09fbdSEmil Medve {
165eda09fbdSEmil Medve 	const Elf64_Rela *x, *y;
166eda09fbdSEmil Medve 
167eda09fbdSEmil Medve 	y = (Elf64_Rela *)_x;
168eda09fbdSEmil Medve 	x = (Elf64_Rela *)_y;
169eda09fbdSEmil Medve 
170eda09fbdSEmil Medve 	/* Compare the entire r_info (as opposed to ELF64_R_SYM(r_info) only) to
171eda09fbdSEmil Medve 	 * make the comparison cheaper/faster. It won't affect the sorting or
172eda09fbdSEmil Medve 	 * the counting algorithms' performance
173eda09fbdSEmil Medve 	 */
174eda09fbdSEmil Medve 	if (x->r_info < y->r_info)
175eda09fbdSEmil Medve 		return -1;
176eda09fbdSEmil Medve 	else if (x->r_info > y->r_info)
177eda09fbdSEmil Medve 		return 1;
178eda09fbdSEmil Medve 	else if (x->r_addend < y->r_addend)
179eda09fbdSEmil Medve 		return -1;
180eda09fbdSEmil Medve 	else if (x->r_addend > y->r_addend)
181eda09fbdSEmil Medve 		return 1;
182eda09fbdSEmil Medve 	else
183eda09fbdSEmil Medve 		return 0;
184eda09fbdSEmil Medve }
185eda09fbdSEmil Medve 
1867568cb4eSPaul Mackerras /* Get size of potential trampolines required. */
1877568cb4eSPaul Mackerras static unsigned long get_stubs_size(const Elf64_Ehdr *hdr,
1887568cb4eSPaul Mackerras 				    const Elf64_Shdr *sechdrs)
1897568cb4eSPaul Mackerras {
1907568cb4eSPaul Mackerras 	/* One extra reloc so it's always 0-funcaddr terminated */
1917568cb4eSPaul Mackerras 	unsigned long relocs = 1;
1927568cb4eSPaul Mackerras 	unsigned i;
1937568cb4eSPaul Mackerras 
1947568cb4eSPaul Mackerras 	/* Every relocated section... */
1957568cb4eSPaul Mackerras 	for (i = 1; i < hdr->e_shnum; i++) {
1967568cb4eSPaul Mackerras 		if (sechdrs[i].sh_type == SHT_RELA) {
197c7d1f6afSAnton Blanchard 			pr_debug("Found relocations in section %u\n", i);
198c7d1f6afSAnton Blanchard 			pr_debug("Ptr: %p.  Number: %Lu\n",
1997568cb4eSPaul Mackerras 			       (void *)sechdrs[i].sh_addr,
2007568cb4eSPaul Mackerras 			       sechdrs[i].sh_size / sizeof(Elf64_Rela));
201eda09fbdSEmil Medve 
202eda09fbdSEmil Medve 			/* Sort the relocation information based on a symbol and
203eda09fbdSEmil Medve 			 * addend key. This is a stable O(n*log n) complexity
204eda09fbdSEmil Medve 			 * alogrithm but it will reduce the complexity of
205eda09fbdSEmil Medve 			 * count_relocs() to linear complexity O(n)
206eda09fbdSEmil Medve 			 */
207eda09fbdSEmil Medve 			sort((void *)sechdrs[i].sh_addr,
208eda09fbdSEmil Medve 			     sechdrs[i].sh_size / sizeof(Elf64_Rela),
209bac7ca7bSAndrey Abramov 			     sizeof(Elf64_Rela), relacmp, NULL);
210eda09fbdSEmil Medve 
2117568cb4eSPaul Mackerras 			relocs += count_relocs((void *)sechdrs[i].sh_addr,
2127568cb4eSPaul Mackerras 					       sechdrs[i].sh_size
2137568cb4eSPaul Mackerras 					       / sizeof(Elf64_Rela));
2147568cb4eSPaul Mackerras 		}
2157568cb4eSPaul Mackerras 	}
2167568cb4eSPaul Mackerras 
217f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
218f48cb8b4SSteven Rostedt 	/* make the trampoline to the ftrace_caller */
219f48cb8b4SSteven Rostedt 	relocs++;
220ae30cc05SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
221ae30cc05SNaveen N. Rao 	/* an additional one for ftrace_regs_caller */
222ae30cc05SNaveen N. Rao 	relocs++;
223ae30cc05SNaveen N. Rao #endif
224f48cb8b4SSteven Rostedt #endif
225f48cb8b4SSteven Rostedt 
226c7d1f6afSAnton Blanchard 	pr_debug("Looks like a total of %lu stubs, max\n", relocs);
2277568cb4eSPaul Mackerras 	return relocs * sizeof(struct ppc64_stub_entry);
2287568cb4eSPaul Mackerras }
2297568cb4eSPaul Mackerras 
2305b12c5c6SRusty Russell /* Still needed for ELFv2, for .TOC. */
2317568cb4eSPaul Mackerras static void dedotify_versions(struct modversion_info *vers,
2327568cb4eSPaul Mackerras 			      unsigned long size)
2337568cb4eSPaul Mackerras {
2347568cb4eSPaul Mackerras 	struct modversion_info *end;
2357568cb4eSPaul Mackerras 
2367568cb4eSPaul Mackerras 	for (end = (void *)vers + size; vers < end; vers++)
237c2cbcf53SLaurent Dufour 		if (vers->name[0] == '.') {
2387568cb4eSPaul Mackerras 			memmove(vers->name, vers->name+1, strlen(vers->name));
239c2cbcf53SLaurent Dufour 		}
2407568cb4eSPaul Mackerras }
2417568cb4eSPaul Mackerras 
242c153693dSAlan Modra /*
243c153693dSAlan Modra  * Undefined symbols which refer to .funcname, hack to funcname. Make .TOC.
244c153693dSAlan Modra  * seem to be defined (value set later).
245c153693dSAlan Modra  */
2467568cb4eSPaul Mackerras static void dedotify(Elf64_Sym *syms, unsigned int numsyms, char *strtab)
2477568cb4eSPaul Mackerras {
2487568cb4eSPaul Mackerras 	unsigned int i;
2497568cb4eSPaul Mackerras 
2507568cb4eSPaul Mackerras 	for (i = 1; i < numsyms; i++) {
2517568cb4eSPaul Mackerras 		if (syms[i].st_shndx == SHN_UNDEF) {
2527568cb4eSPaul Mackerras 			char *name = strtab + syms[i].st_name;
253c153693dSAlan Modra 			if (name[0] == '.') {
254c153693dSAlan Modra 				if (strcmp(name+1, "TOC.") == 0)
255c153693dSAlan Modra 					syms[i].st_shndx = SHN_ABS;
256f15838e9SAndreas Schwab 				syms[i].st_name++;
2577568cb4eSPaul Mackerras 			}
2587568cb4eSPaul Mackerras 		}
2597568cb4eSPaul Mackerras 	}
260c153693dSAlan Modra }
2617568cb4eSPaul Mackerras 
2624edebbeaSRusty Russell static Elf64_Sym *find_dot_toc(Elf64_Shdr *sechdrs,
2634edebbeaSRusty Russell 			       const char *strtab,
2644edebbeaSRusty Russell 			       unsigned int symindex)
2654edebbeaSRusty Russell {
2664edebbeaSRusty Russell 	unsigned int i, numsyms;
2674edebbeaSRusty Russell 	Elf64_Sym *syms;
2684edebbeaSRusty Russell 
2694edebbeaSRusty Russell 	syms = (Elf64_Sym *)sechdrs[symindex].sh_addr;
2704edebbeaSRusty Russell 	numsyms = sechdrs[symindex].sh_size / sizeof(Elf64_Sym);
2714edebbeaSRusty Russell 
2724edebbeaSRusty Russell 	for (i = 1; i < numsyms; i++) {
273c153693dSAlan Modra 		if (syms[i].st_shndx == SHN_ABS
274008d7a91SRusty Russell 		    && strcmp(strtab + syms[i].st_name, "TOC.") == 0)
2754edebbeaSRusty Russell 			return &syms[i];
2764edebbeaSRusty Russell 	}
2774edebbeaSRusty Russell 	return NULL;
2784edebbeaSRusty Russell }
2794edebbeaSRusty Russell 
280*d4be60feSWedson Almeida Filho bool module_init_section(const char *name)
281*d4be60feSWedson Almeida Filho {
282*d4be60feSWedson Almeida Filho 	/* We don't handle .init for the moment: always return false. */
283*d4be60feSWedson Almeida Filho 	return false;
284*d4be60feSWedson Almeida Filho }
285*d4be60feSWedson Almeida Filho 
2867568cb4eSPaul Mackerras int module_frob_arch_sections(Elf64_Ehdr *hdr,
2877568cb4eSPaul Mackerras 			      Elf64_Shdr *sechdrs,
2887568cb4eSPaul Mackerras 			      char *secstrings,
2897568cb4eSPaul Mackerras 			      struct module *me)
2907568cb4eSPaul Mackerras {
2917568cb4eSPaul Mackerras 	unsigned int i;
2927568cb4eSPaul Mackerras 
2937568cb4eSPaul Mackerras 	/* Find .toc and .stubs sections, symtab and strtab */
2947568cb4eSPaul Mackerras 	for (i = 1; i < hdr->e_shnum; i++) {
2957568cb4eSPaul Mackerras 		if (strcmp(secstrings + sechdrs[i].sh_name, ".stubs") == 0)
2967568cb4eSPaul Mackerras 			me->arch.stubs_section = i;
2975c45b528SAlan Modra 		else if (strcmp(secstrings + sechdrs[i].sh_name, ".toc") == 0) {
2987568cb4eSPaul Mackerras 			me->arch.toc_section = i;
2995c45b528SAlan Modra 			if (sechdrs[i].sh_addralign < 8)
3005c45b528SAlan Modra 				sechdrs[i].sh_addralign = 8;
3015c45b528SAlan Modra 		}
3027568cb4eSPaul Mackerras 		else if (strcmp(secstrings+sechdrs[i].sh_name,"__versions")==0)
3037568cb4eSPaul Mackerras 			dedotify_versions((void *)hdr + sechdrs[i].sh_offset,
3047568cb4eSPaul Mackerras 					  sechdrs[i].sh_size);
3057568cb4eSPaul Mackerras 
3067568cb4eSPaul Mackerras 		if (sechdrs[i].sh_type == SHT_SYMTAB)
3077568cb4eSPaul Mackerras 			dedotify((void *)hdr + sechdrs[i].sh_offset,
3087568cb4eSPaul Mackerras 				 sechdrs[i].sh_size / sizeof(Elf64_Sym),
3097568cb4eSPaul Mackerras 				 (void *)hdr
3107568cb4eSPaul Mackerras 				 + sechdrs[sechdrs[i].sh_link].sh_offset);
3117568cb4eSPaul Mackerras 	}
312f749edaeSAlan Modra 
313f749edaeSAlan Modra 	if (!me->arch.stubs_section) {
314c7d1f6afSAnton Blanchard 		pr_err("%s: doesn't contain .stubs.\n", me->name);
3157568cb4eSPaul Mackerras 		return -ENOEXEC;
3167568cb4eSPaul Mackerras 	}
3177568cb4eSPaul Mackerras 
318f749edaeSAlan Modra 	/* If we don't have a .toc, just use .stubs.  We need to set r2
319f749edaeSAlan Modra 	   to some reasonable value in case the module calls out to
320f749edaeSAlan Modra 	   other functions via a stub, or if a function pointer escapes
321f749edaeSAlan Modra 	   the module by some means.  */
322f749edaeSAlan Modra 	if (!me->arch.toc_section)
323f749edaeSAlan Modra 		me->arch.toc_section = me->arch.stubs_section;
324f749edaeSAlan Modra 
3257568cb4eSPaul Mackerras 	/* Override the stubs size */
3267568cb4eSPaul Mackerras 	sechdrs[me->arch.stubs_section].sh_size = get_stubs_size(hdr, sechdrs);
3277568cb4eSPaul Mackerras 	return 0;
3287568cb4eSPaul Mackerras }
3297568cb4eSPaul Mackerras 
330bd55e792SNaveen N. Rao #ifdef CONFIG_MPROFILE_KERNEL
331bd55e792SNaveen N. Rao 
332bd55e792SNaveen N. Rao static u32 stub_insns[] = {
33347b04699SChristophe Leroy 	PPC_RAW_LD(_R12, _R13, offsetof(struct paca_struct, kernel_toc)),
33447b04699SChristophe Leroy 	PPC_RAW_ADDIS(_R12, _R12, 0),
33547b04699SChristophe Leroy 	PPC_RAW_ADDI(_R12, _R12, 0),
33647b04699SChristophe Leroy 	PPC_RAW_MTCTR(_R12),
33747b04699SChristophe Leroy 	PPC_RAW_BCTR(),
338bd55e792SNaveen N. Rao };
339bd55e792SNaveen N. Rao 
340bd55e792SNaveen N. Rao /*
341bd55e792SNaveen N. Rao  * For mprofile-kernel we use a special stub for ftrace_caller() because we
342bd55e792SNaveen N. Rao  * can't rely on r2 containing this module's TOC when we enter the stub.
343bd55e792SNaveen N. Rao  *
344bd55e792SNaveen N. Rao  * That can happen if the function calling us didn't need to use the toc. In
345bd55e792SNaveen N. Rao  * that case it won't have setup r2, and the r2 value will be either the
346bd55e792SNaveen N. Rao  * kernel's toc, or possibly another modules toc.
347bd55e792SNaveen N. Rao  *
348bd55e792SNaveen N. Rao  * To deal with that this stub uses the kernel toc, which is always accessible
349bd55e792SNaveen N. Rao  * via the paca (in r13). The target (ftrace_caller()) is responsible for
350bd55e792SNaveen N. Rao  * saving and restoring the toc before returning.
351bd55e792SNaveen N. Rao  */
352bd55e792SNaveen N. Rao static inline int create_ftrace_stub(struct ppc64_stub_entry *entry,
353bd55e792SNaveen N. Rao 					unsigned long addr,
354bd55e792SNaveen N. Rao 					struct module *me)
355bd55e792SNaveen N. Rao {
356bd55e792SNaveen N. Rao 	long reladdr;
357bd55e792SNaveen N. Rao 
358bd55e792SNaveen N. Rao 	memcpy(entry->jump, stub_insns, sizeof(stub_insns));
359bd55e792SNaveen N. Rao 
360bd55e792SNaveen N. Rao 	/* Stub uses address relative to kernel toc (from the paca) */
361bd55e792SNaveen N. Rao 	reladdr = addr - kernel_toc_addr();
362bd55e792SNaveen N. Rao 	if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) {
363bd55e792SNaveen N. Rao 		pr_err("%s: Address of %ps out of range of kernel_toc.\n",
364bd55e792SNaveen N. Rao 							me->name, (void *)addr);
365bd55e792SNaveen N. Rao 		return 0;
366bd55e792SNaveen N. Rao 	}
367bd55e792SNaveen N. Rao 
368bd55e792SNaveen N. Rao 	entry->jump[1] |= PPC_HA(reladdr);
369bd55e792SNaveen N. Rao 	entry->jump[2] |= PPC_LO(reladdr);
370bd55e792SNaveen N. Rao 
371bd55e792SNaveen N. Rao 	/* Eventhough we don't use funcdata in the stub, it's needed elsewhere. */
372bd55e792SNaveen N. Rao 	entry->funcdata = func_desc(addr);
373bd55e792SNaveen N. Rao 	entry->magic = STUB_MAGIC;
374bd55e792SNaveen N. Rao 
375bd55e792SNaveen N. Rao 	return 1;
376bd55e792SNaveen N. Rao }
377bd55e792SNaveen N. Rao 
378bd55e792SNaveen N. Rao static bool is_mprofile_ftrace_call(const char *name)
379bd55e792SNaveen N. Rao {
380bd55e792SNaveen N. Rao 	if (!strcmp("_mcount", name))
381bd55e792SNaveen N. Rao 		return true;
382bd55e792SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE
383bd55e792SNaveen N. Rao 	if (!strcmp("ftrace_caller", name))
384bd55e792SNaveen N. Rao 		return true;
385bd55e792SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
386bd55e792SNaveen N. Rao 	if (!strcmp("ftrace_regs_caller", name))
387bd55e792SNaveen N. Rao 		return true;
388bd55e792SNaveen N. Rao #endif
389bd55e792SNaveen N. Rao #endif
390bd55e792SNaveen N. Rao 
391bd55e792SNaveen N. Rao 	return false;
392bd55e792SNaveen N. Rao }
393bd55e792SNaveen N. Rao #else
394bd55e792SNaveen N. Rao static inline int create_ftrace_stub(struct ppc64_stub_entry *entry,
395bd55e792SNaveen N. Rao 					unsigned long addr,
396bd55e792SNaveen N. Rao 					struct module *me)
397bd55e792SNaveen N. Rao {
398bd55e792SNaveen N. Rao 	return 0;
399bd55e792SNaveen N. Rao }
400bd55e792SNaveen N. Rao 
401bd55e792SNaveen N. Rao static bool is_mprofile_ftrace_call(const char *name)
402bd55e792SNaveen N. Rao {
403bd55e792SNaveen N. Rao 	return false;
404bd55e792SNaveen N. Rao }
405bd55e792SNaveen N. Rao #endif
406bd55e792SNaveen N. Rao 
4075c45b528SAlan Modra /*
4085c45b528SAlan Modra  * r2 is the TOC pointer: it actually points 0x8000 into the TOC (this gives the
4095c45b528SAlan Modra  * value maximum span in an instruction which uses a signed offset). Round down
4105c45b528SAlan Modra  * to a 256 byte boundary for the odd case where we are setting up r2 without a
4115c45b528SAlan Modra  * .toc section.
4125c45b528SAlan Modra  */
413136cd345SMichael Ellerman static inline unsigned long my_r2(const Elf64_Shdr *sechdrs, struct module *me)
4147568cb4eSPaul Mackerras {
4155c45b528SAlan Modra 	return (sechdrs[me->arch.toc_section].sh_addr & ~0xfful) + 0x8000;
4167568cb4eSPaul Mackerras }
4177568cb4eSPaul Mackerras 
4187568cb4eSPaul Mackerras /* Patch stub to reference function and correct r2 value. */
419136cd345SMichael Ellerman static inline int create_stub(const Elf64_Shdr *sechdrs,
4207568cb4eSPaul Mackerras 			      struct ppc64_stub_entry *entry,
421008d7a91SRusty Russell 			      unsigned long addr,
422bd55e792SNaveen N. Rao 			      struct module *me,
423bd55e792SNaveen N. Rao 			      const char *name)
4247568cb4eSPaul Mackerras {
4257568cb4eSPaul Mackerras 	long reladdr;
4268734b41bSRussell Currey 	func_desc_t desc;
4278734b41bSRussell Currey 	int i;
4287568cb4eSPaul Mackerras 
429bd55e792SNaveen N. Rao 	if (is_mprofile_ftrace_call(name))
430bd55e792SNaveen N. Rao 		return create_ftrace_stub(entry, addr, me);
431bd55e792SNaveen N. Rao 
4328734b41bSRussell Currey 	for (i = 0; i < sizeof(ppc64_stub_insns) / sizeof(u32); i++) {
4338734b41bSRussell Currey 		if (patch_instruction(&entry->jump[i],
4348734b41bSRussell Currey 				      ppc_inst(ppc64_stub_insns[i])))
4358734b41bSRussell Currey 			return 0;
4368734b41bSRussell Currey 	}
4377568cb4eSPaul Mackerras 
4387568cb4eSPaul Mackerras 	/* Stub uses address relative to r2. */
4397568cb4eSPaul Mackerras 	reladdr = (unsigned long)entry - my_r2(sechdrs, me);
4407568cb4eSPaul Mackerras 	if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) {
441c7d1f6afSAnton Blanchard 		pr_err("%s: Address %p of stub out of range of %p.\n",
4427568cb4eSPaul Mackerras 		       me->name, (void *)reladdr, (void *)my_r2);
4437568cb4eSPaul Mackerras 		return 0;
4447568cb4eSPaul Mackerras 	}
445c7d1f6afSAnton Blanchard 	pr_debug("Stub %p get data from reladdr %li\n", entry, reladdr);
4467568cb4eSPaul Mackerras 
4478734b41bSRussell Currey 	if (patch_instruction(&entry->jump[0],
4488734b41bSRussell Currey 			      ppc_inst(entry->jump[0] | PPC_HA(reladdr))))
4498734b41bSRussell Currey 		return 0;
4508734b41bSRussell Currey 
4518734b41bSRussell Currey 	if (patch_instruction(&entry->jump[1],
4528734b41bSRussell Currey 			  ppc_inst(entry->jump[1] | PPC_LO(reladdr))))
4538734b41bSRussell Currey 		return 0;
4548734b41bSRussell Currey 
4558734b41bSRussell Currey 	// func_desc_t is 8 bytes if ABIv2, else 16 bytes
4568734b41bSRussell Currey 	desc = func_desc(addr);
4578734b41bSRussell Currey 	for (i = 0; i < sizeof(func_desc_t) / sizeof(u32); i++) {
4588734b41bSRussell Currey 		if (patch_instruction(((u32 *)&entry->funcdata) + i,
4598734b41bSRussell Currey 				      ppc_inst(((u32 *)(&desc))[i])))
4608734b41bSRussell Currey 			return 0;
4618734b41bSRussell Currey 	}
4628734b41bSRussell Currey 
4638734b41bSRussell Currey 	if (patch_instruction(&entry->magic, ppc_inst(STUB_MAGIC)))
4648734b41bSRussell Currey 		return 0;
465f17c4e01SMichael Ellerman 
4667568cb4eSPaul Mackerras 	return 1;
4677568cb4eSPaul Mackerras }
4687568cb4eSPaul Mackerras 
469008d7a91SRusty Russell /* Create stub to jump to function described in this OPD/ptr: we need the
4707568cb4eSPaul Mackerras    stub to set up the TOC ptr (r2) for the function. */
471136cd345SMichael Ellerman static unsigned long stub_for_addr(const Elf64_Shdr *sechdrs,
472008d7a91SRusty Russell 				   unsigned long addr,
473bd55e792SNaveen N. Rao 				   struct module *me,
474bd55e792SNaveen N. Rao 				   const char *name)
4757568cb4eSPaul Mackerras {
4767568cb4eSPaul Mackerras 	struct ppc64_stub_entry *stubs;
4777568cb4eSPaul Mackerras 	unsigned int i, num_stubs;
4787568cb4eSPaul Mackerras 
4797568cb4eSPaul Mackerras 	num_stubs = sechdrs[me->arch.stubs_section].sh_size / sizeof(*stubs);
4807568cb4eSPaul Mackerras 
4817568cb4eSPaul Mackerras 	/* Find this stub, or if that fails, the next avail. entry */
4827568cb4eSPaul Mackerras 	stubs = (void *)sechdrs[me->arch.stubs_section].sh_addr;
483008d7a91SRusty Russell 	for (i = 0; stub_func_addr(stubs[i].funcdata); i++) {
4841c0437afSKamalesh Babulal 		if (WARN_ON(i >= num_stubs))
4851c0437afSKamalesh Babulal 			return 0;
4867568cb4eSPaul Mackerras 
487008d7a91SRusty Russell 		if (stub_func_addr(stubs[i].funcdata) == func_addr(addr))
4887568cb4eSPaul Mackerras 			return (unsigned long)&stubs[i];
4897568cb4eSPaul Mackerras 	}
4907568cb4eSPaul Mackerras 
491bd55e792SNaveen N. Rao 	if (!create_stub(sechdrs, &stubs[i], addr, me, name))
4927568cb4eSPaul Mackerras 		return 0;
4937568cb4eSPaul Mackerras 
4947568cb4eSPaul Mackerras 	return (unsigned long)&stubs[i];
4957568cb4eSPaul Mackerras }
4967568cb4eSPaul Mackerras 
4977568cb4eSPaul Mackerras /* We expect a noop next: if it is, replace it with instruction to
4987568cb4eSPaul Mackerras    restore r2. */
499250122baSNaveen N. Rao static int restore_r2(const char *name, u32 *instruction, struct module *me)
5007568cb4eSPaul Mackerras {
501b9eab08dSJosh Poimboeuf 	u32 *prev_insn = instruction - 1;
502b9eab08dSJosh Poimboeuf 
5031f2aaed2SNaveen N. Rao 	if (is_mprofile_ftrace_call(name))
504b9eab08dSJosh Poimboeuf 		return 1;
505b9eab08dSJosh Poimboeuf 
506b9eab08dSJosh Poimboeuf 	/*
507b9eab08dSJosh Poimboeuf 	 * Make sure the branch isn't a sibling call.  Sibling calls aren't
508b9eab08dSJosh Poimboeuf 	 * "link" branches and they don't return, so they don't need the r2
509b9eab08dSJosh Poimboeuf 	 * restore afterwards.
510b9eab08dSJosh Poimboeuf 	 */
51175346251SJordan Niethe 	if (!instr_is_relative_link_branch(ppc_inst(*prev_insn)))
51215308664STorsten Duwe 		return 1;
51331278b17SMichael Ellerman 
51447b04699SChristophe Leroy 	if (*instruction != PPC_RAW_NOP()) {
5151ea61ea2SJosh Poimboeuf 		pr_err("%s: Expected nop after call, got %08x at %pS\n",
5161ea61ea2SJosh Poimboeuf 			me->name, *instruction, instruction);
5177568cb4eSPaul Mackerras 		return 0;
5187568cb4eSPaul Mackerras 	}
5198734b41bSRussell Currey 
520d2fae548SRusty Russell 	/* ld r2,R2_STACK_OFFSET(r1) */
5218734b41bSRussell Currey 	if (patch_instruction(instruction, ppc_inst(PPC_INST_LD_TOC)))
5228734b41bSRussell Currey 		return 0;
5238734b41bSRussell Currey 
5247568cb4eSPaul Mackerras 	return 1;
5257568cb4eSPaul Mackerras }
5267568cb4eSPaul Mackerras 
5277568cb4eSPaul Mackerras int apply_relocate_add(Elf64_Shdr *sechdrs,
5287568cb4eSPaul Mackerras 		       const char *strtab,
5297568cb4eSPaul Mackerras 		       unsigned int symindex,
5307568cb4eSPaul Mackerras 		       unsigned int relsec,
5317568cb4eSPaul Mackerras 		       struct module *me)
5327568cb4eSPaul Mackerras {
5337568cb4eSPaul Mackerras 	unsigned int i;
5347568cb4eSPaul Mackerras 	Elf64_Rela *rela = (void *)sechdrs[relsec].sh_addr;
5357568cb4eSPaul Mackerras 	Elf64_Sym *sym;
5367568cb4eSPaul Mackerras 	unsigned long *location;
5377568cb4eSPaul Mackerras 	unsigned long value;
5387568cb4eSPaul Mackerras 
539c7d1f6afSAnton Blanchard 	pr_debug("Applying ADD relocate section %u to %u\n", relsec,
5407568cb4eSPaul Mackerras 	       sechdrs[relsec].sh_info);
5414edebbeaSRusty Russell 
5424edebbeaSRusty Russell 	/* First time we're called, we can fix up .TOC. */
5434edebbeaSRusty Russell 	if (!me->arch.toc_fixed) {
5444edebbeaSRusty Russell 		sym = find_dot_toc(sechdrs, strtab, symindex);
5454edebbeaSRusty Russell 		/* It's theoretically possible that a module doesn't want a
5464edebbeaSRusty Russell 		 * .TOC. so don't fail it just for that. */
5474edebbeaSRusty Russell 		if (sym)
5484edebbeaSRusty Russell 			sym->st_value = my_r2(sechdrs, me);
5494edebbeaSRusty Russell 		me->arch.toc_fixed = true;
5504edebbeaSRusty Russell 	}
5514edebbeaSRusty Russell 
5527568cb4eSPaul Mackerras 	for (i = 0; i < sechdrs[relsec].sh_size / sizeof(*rela); i++) {
5537568cb4eSPaul Mackerras 		/* This is where to make the change */
5547568cb4eSPaul Mackerras 		location = (void *)sechdrs[sechdrs[relsec].sh_info].sh_addr
5557568cb4eSPaul Mackerras 			+ rela[i].r_offset;
5567568cb4eSPaul Mackerras 		/* This is the symbol it is referring to */
5577568cb4eSPaul Mackerras 		sym = (Elf64_Sym *)sechdrs[symindex].sh_addr
5587568cb4eSPaul Mackerras 			+ ELF64_R_SYM(rela[i].r_info);
5597568cb4eSPaul Mackerras 
560c7d1f6afSAnton Blanchard 		pr_debug("RELOC at %p: %li-type as %s (0x%lx) + %li\n",
5617568cb4eSPaul Mackerras 		       location, (long)ELF64_R_TYPE(rela[i].r_info),
5627568cb4eSPaul Mackerras 		       strtab + sym->st_name, (unsigned long)sym->st_value,
5637568cb4eSPaul Mackerras 		       (long)rela[i].r_addend);
5647568cb4eSPaul Mackerras 
5657568cb4eSPaul Mackerras 		/* `Everything is relative'. */
5667568cb4eSPaul Mackerras 		value = sym->st_value + rela[i].r_addend;
5677568cb4eSPaul Mackerras 
5687568cb4eSPaul Mackerras 		switch (ELF64_R_TYPE(rela[i].r_info)) {
5697568cb4eSPaul Mackerras 		case R_PPC64_ADDR32:
5707568cb4eSPaul Mackerras 			/* Simply set it */
5717568cb4eSPaul Mackerras 			*(u32 *)location = value;
5727568cb4eSPaul Mackerras 			break;
5737568cb4eSPaul Mackerras 
5747568cb4eSPaul Mackerras 		case R_PPC64_ADDR64:
5757568cb4eSPaul Mackerras 			/* Simply set it */
5767568cb4eSPaul Mackerras 			*(unsigned long *)location = value;
5777568cb4eSPaul Mackerras 			break;
5787568cb4eSPaul Mackerras 
5797568cb4eSPaul Mackerras 		case R_PPC64_TOC:
5807568cb4eSPaul Mackerras 			*(unsigned long *)location = my_r2(sechdrs, me);
5817568cb4eSPaul Mackerras 			break;
5827568cb4eSPaul Mackerras 
5837568cb4eSPaul Mackerras 		case R_PPC64_TOC16:
584f749edaeSAlan Modra 			/* Subtract TOC pointer */
5857568cb4eSPaul Mackerras 			value -= my_r2(sechdrs, me);
5867568cb4eSPaul Mackerras 			if (value + 0x8000 > 0xffff) {
587c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16 relocation (0x%lx)\n",
5887568cb4eSPaul Mackerras 				       me->name, value);
5897568cb4eSPaul Mackerras 				return -ENOEXEC;
5907568cb4eSPaul Mackerras 			}
5917568cb4eSPaul Mackerras 			*((uint16_t *) location)
5927568cb4eSPaul Mackerras 				= (*((uint16_t *) location) & ~0xffff)
5937568cb4eSPaul Mackerras 				| (value & 0xffff);
5947568cb4eSPaul Mackerras 			break;
5957568cb4eSPaul Mackerras 
5961fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_LO:
5971fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
5981fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
5991fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6001fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xffff)
6011fbe9cf2SAnton Blanchard 				| (value & 0xffff);
6021fbe9cf2SAnton Blanchard 			break;
6031fbe9cf2SAnton Blanchard 
6047568cb4eSPaul Mackerras 		case R_PPC64_TOC16_DS:
605f749edaeSAlan Modra 			/* Subtract TOC pointer */
6067568cb4eSPaul Mackerras 			value -= my_r2(sechdrs, me);
6077568cb4eSPaul Mackerras 			if ((value & 3) != 0 || value + 0x8000 > 0xffff) {
608c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16_DS relocation (0x%lx)\n",
6097568cb4eSPaul Mackerras 				       me->name, value);
6107568cb4eSPaul Mackerras 				return -ENOEXEC;
6117568cb4eSPaul Mackerras 			}
6127568cb4eSPaul Mackerras 			*((uint16_t *) location)
6137568cb4eSPaul Mackerras 				= (*((uint16_t *) location) & ~0xfffc)
6147568cb4eSPaul Mackerras 				| (value & 0xfffc);
6157568cb4eSPaul Mackerras 			break;
6167568cb4eSPaul Mackerras 
6171fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_LO_DS:
6181fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
6191fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6201fbe9cf2SAnton Blanchard 			if ((value & 3) != 0) {
621c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16_LO_DS relocation (0x%lx)\n",
6221fbe9cf2SAnton Blanchard 				       me->name, value);
6231fbe9cf2SAnton Blanchard 				return -ENOEXEC;
6241fbe9cf2SAnton Blanchard 			}
6251fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6261fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xfffc)
6271fbe9cf2SAnton Blanchard 				| (value & 0xfffc);
6281fbe9cf2SAnton Blanchard 			break;
6291fbe9cf2SAnton Blanchard 
6301fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_HA:
6311fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
6321fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6331fbe9cf2SAnton Blanchard 			value = ((value + 0x8000) >> 16);
6341fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6351fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xffff)
6361fbe9cf2SAnton Blanchard 				| (value & 0xffff);
6371fbe9cf2SAnton Blanchard 			break;
6381fbe9cf2SAnton Blanchard 
6397568cb4eSPaul Mackerras 		case R_PPC_REL24:
6407568cb4eSPaul Mackerras 			/* FIXME: Handle weak symbols here --RR */
641a443bf6eSKamalesh Babulal 			if (sym->st_shndx == SHN_UNDEF ||
642a443bf6eSKamalesh Babulal 			    sym->st_shndx == SHN_LIVEPATCH) {
6437568cb4eSPaul Mackerras 				/* External: go via stub */
644bd55e792SNaveen N. Rao 				value = stub_for_addr(sechdrs, value, me,
645bd55e792SNaveen N. Rao 						strtab + sym->st_name);
6467568cb4eSPaul Mackerras 				if (!value)
6477568cb4eSPaul Mackerras 					return -ENOENT;
648250122baSNaveen N. Rao 				if (!restore_r2(strtab + sym->st_name,
649250122baSNaveen N. Rao 							(u32 *)location + 1, me))
6507568cb4eSPaul Mackerras 					return -ENOEXEC;
651008d7a91SRusty Russell 			} else
652008d7a91SRusty Russell 				value += local_entry_offset(sym);
6537568cb4eSPaul Mackerras 
6547568cb4eSPaul Mackerras 			/* Convert value to relative */
6557568cb4eSPaul Mackerras 			value -= (unsigned long)location;
6567568cb4eSPaul Mackerras 			if (value + 0x2000000 > 0x3ffffff || (value & 3) != 0){
657c7d1f6afSAnton Blanchard 				pr_err("%s: REL24 %li out of range!\n",
6587568cb4eSPaul Mackerras 				       me->name, (long int)value);
6597568cb4eSPaul Mackerras 				return -ENOEXEC;
6607568cb4eSPaul Mackerras 			}
6617568cb4eSPaul Mackerras 
6627568cb4eSPaul Mackerras 			/* Only replace bits 2 through 26 */
6638734b41bSRussell Currey 			value = (*(uint32_t *)location & ~0x03fffffc)
6647568cb4eSPaul Mackerras 				| (value & 0x03fffffc);
6658734b41bSRussell Currey 
6668734b41bSRussell Currey 			if (patch_instruction((u32 *)location, ppc_inst(value)))
6678734b41bSRussell Currey 				return -EFAULT;
6688734b41bSRussell Currey 
6697568cb4eSPaul Mackerras 			break;
6707568cb4eSPaul Mackerras 
67121c4ff80SBenjamin Herrenschmidt 		case R_PPC64_REL64:
67221c4ff80SBenjamin Herrenschmidt 			/* 64 bits relative (used by features fixups) */
67321c4ff80SBenjamin Herrenschmidt 			*location = value - (unsigned long)location;
67421c4ff80SBenjamin Herrenschmidt 			break;
67521c4ff80SBenjamin Herrenschmidt 
6769f751b82SMichael Ellerman 		case R_PPC64_REL32:
6779f751b82SMichael Ellerman 			/* 32 bits relative (used by relative exception tables) */
678b851ba02SNicholas Piggin 			/* Convert value to relative */
679b851ba02SNicholas Piggin 			value -= (unsigned long)location;
680b851ba02SNicholas Piggin 			if (value + 0x80000000 > 0xffffffff) {
681b851ba02SNicholas Piggin 				pr_err("%s: REL32 %li out of range!\n",
682b851ba02SNicholas Piggin 				       me->name, (long int)value);
683b851ba02SNicholas Piggin 				return -ENOEXEC;
684b851ba02SNicholas Piggin 			}
685b851ba02SNicholas Piggin 			*(u32 *)location = value;
6869f751b82SMichael Ellerman 			break;
6879f751b82SMichael Ellerman 
688d247da0aSRusty Russell 		case R_PPC64_TOCSAVE:
689d247da0aSRusty Russell 			/*
690d247da0aSRusty Russell 			 * Marker reloc indicates we don't have to save r2.
691d247da0aSRusty Russell 			 * That would only save us one instruction, so ignore
692d247da0aSRusty Russell 			 * it.
693d247da0aSRusty Russell 			 */
694d247da0aSRusty Russell 			break;
695d247da0aSRusty Russell 
696a61674bdSUlrich Weigand 		case R_PPC64_ENTRY:
697a61674bdSUlrich Weigand 			/*
698a61674bdSUlrich Weigand 			 * Optimize ELFv2 large code model entry point if
699a61674bdSUlrich Weigand 			 * the TOC is within 2GB range of current location.
700a61674bdSUlrich Weigand 			 */
701a61674bdSUlrich Weigand 			value = my_r2(sechdrs, me) - (unsigned long)location;
702a61674bdSUlrich Weigand 			if (value + 0x80008000 > 0xffffffff)
703a61674bdSUlrich Weigand 				break;
704a61674bdSUlrich Weigand 			/*
705a61674bdSUlrich Weigand 			 * Check for the large code model prolog sequence:
706a61674bdSUlrich Weigand 		         *	ld r2, ...(r12)
707a61674bdSUlrich Weigand 			 *	add r2, r2, r12
708a61674bdSUlrich Weigand 			 */
70947b04699SChristophe Leroy 			if ((((uint32_t *)location)[0] & ~0xfffc) != PPC_RAW_LD(_R2, _R12, 0))
710a61674bdSUlrich Weigand 				break;
71147b04699SChristophe Leroy 			if (((uint32_t *)location)[1] != PPC_RAW_ADD(_R2, _R2, _R12))
712a61674bdSUlrich Weigand 				break;
713a61674bdSUlrich Weigand 			/*
714a61674bdSUlrich Weigand 			 * If found, replace it with:
715a61674bdSUlrich Weigand 			 *	addis r2, r12, (.TOC.-func)@ha
7162fb0a2c9SMichael Ellerman 			 *	addi  r2,  r2, (.TOC.-func)@l
717a61674bdSUlrich Weigand 			 */
71847b04699SChristophe Leroy 			((uint32_t *)location)[0] = PPC_RAW_ADDIS(_R2, _R12, PPC_HA(value));
71947b04699SChristophe Leroy 			((uint32_t *)location)[1] = PPC_RAW_ADDI(_R2, _R2, PPC_LO(value));
720a61674bdSUlrich Weigand 			break;
721a61674bdSUlrich Weigand 
7220906584aSRusty Russell 		case R_PPC64_REL16_HA:
7230906584aSRusty Russell 			/* Subtract location pointer */
7240906584aSRusty Russell 			value -= (unsigned long)location;
7250906584aSRusty Russell 			value = ((value + 0x8000) >> 16);
7260906584aSRusty Russell 			*((uint16_t *) location)
7270906584aSRusty Russell 				= (*((uint16_t *) location) & ~0xffff)
7280906584aSRusty Russell 				| (value & 0xffff);
7290906584aSRusty Russell 			break;
7300906584aSRusty Russell 
7310906584aSRusty Russell 		case R_PPC64_REL16_LO:
7320906584aSRusty Russell 			/* Subtract location pointer */
7330906584aSRusty Russell 			value -= (unsigned long)location;
7340906584aSRusty Russell 			*((uint16_t *) location)
7350906584aSRusty Russell 				= (*((uint16_t *) location) & ~0xffff)
7360906584aSRusty Russell 				| (value & 0xffff);
7370906584aSRusty Russell 			break;
7380906584aSRusty Russell 
7397568cb4eSPaul Mackerras 		default:
740c7d1f6afSAnton Blanchard 			pr_err("%s: Unknown ADD relocation: %lu\n",
7417568cb4eSPaul Mackerras 			       me->name,
7427568cb4eSPaul Mackerras 			       (unsigned long)ELF64_R_TYPE(rela[i].r_info));
7437568cb4eSPaul Mackerras 			return -ENOEXEC;
7447568cb4eSPaul Mackerras 		}
7457568cb4eSPaul Mackerras 	}
7467568cb4eSPaul Mackerras 
747136cd345SMichael Ellerman 	return 0;
748136cd345SMichael Ellerman }
749136cd345SMichael Ellerman 
750f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
75103b51416SNaveen N. Rao int module_trampoline_target(struct module *mod, unsigned long addr,
75203b51416SNaveen N. Rao 			     unsigned long *target)
75303b51416SNaveen N. Rao {
75403b51416SNaveen N. Rao 	struct ppc64_stub_entry *stub;
75503b51416SNaveen N. Rao 	func_desc_t funcdata;
75603b51416SNaveen N. Rao 	u32 magic;
75703b51416SNaveen N. Rao 
75803b51416SNaveen N. Rao 	if (!within_module_core(addr, mod)) {
75903b51416SNaveen N. Rao 		pr_err("%s: stub %lx not in module %s\n", __func__, addr, mod->name);
76003b51416SNaveen N. Rao 		return -EFAULT;
76103b51416SNaveen N. Rao 	}
76203b51416SNaveen N. Rao 
76303b51416SNaveen N. Rao 	stub = (struct ppc64_stub_entry *)addr;
76403b51416SNaveen N. Rao 
765fe557319SChristoph Hellwig 	if (copy_from_kernel_nofault(&magic, &stub->magic,
766fe557319SChristoph Hellwig 			sizeof(magic))) {
76703b51416SNaveen N. Rao 		pr_err("%s: fault reading magic for stub %lx for %s\n", __func__, addr, mod->name);
76803b51416SNaveen N. Rao 		return -EFAULT;
76903b51416SNaveen N. Rao 	}
77003b51416SNaveen N. Rao 
77103b51416SNaveen N. Rao 	if (magic != STUB_MAGIC) {
77203b51416SNaveen N. Rao 		pr_err("%s: bad magic for stub %lx for %s\n", __func__, addr, mod->name);
77303b51416SNaveen N. Rao 		return -EFAULT;
77403b51416SNaveen N. Rao 	}
77503b51416SNaveen N. Rao 
776fe557319SChristoph Hellwig 	if (copy_from_kernel_nofault(&funcdata, &stub->funcdata,
777fe557319SChristoph Hellwig 			sizeof(funcdata))) {
77803b51416SNaveen N. Rao 		pr_err("%s: fault reading funcdata for stub %lx for %s\n", __func__, addr, mod->name);
77903b51416SNaveen N. Rao                 return -EFAULT;
78003b51416SNaveen N. Rao 	}
78103b51416SNaveen N. Rao 
78203b51416SNaveen N. Rao 	*target = stub_func_addr(funcdata);
78303b51416SNaveen N. Rao 
78403b51416SNaveen N. Rao 	return 0;
78503b51416SNaveen N. Rao }
786336a7b5dSMichael Ellerman 
787136cd345SMichael Ellerman int module_finalize_ftrace(struct module *mod, const Elf_Shdr *sechdrs)
788136cd345SMichael Ellerman {
789bd55e792SNaveen N. Rao 	mod->arch.tramp = stub_for_addr(sechdrs,
790bd55e792SNaveen N. Rao 					(unsigned long)ftrace_caller,
791bd55e792SNaveen N. Rao 					mod,
792bd55e792SNaveen N. Rao 					"ftrace_caller");
793ae30cc05SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
794bd55e792SNaveen N. Rao 	mod->arch.tramp_regs = stub_for_addr(sechdrs,
795bd55e792SNaveen N. Rao 					(unsigned long)ftrace_regs_caller,
796bd55e792SNaveen N. Rao 					mod,
797bd55e792SNaveen N. Rao 					"ftrace_regs_caller");
798ae30cc05SNaveen N. Rao 	if (!mod->arch.tramp_regs)
799ae30cc05SNaveen N. Rao 		return -ENOENT;
800ae30cc05SNaveen N. Rao #endif
801136cd345SMichael Ellerman 
802136cd345SMichael Ellerman 	if (!mod->arch.tramp)
803136cd345SMichael Ellerman 		return -ENOENT;
804f48cb8b4SSteven Rostedt 
8057568cb4eSPaul Mackerras 	return 0;
8067568cb4eSPaul Mackerras }
807136cd345SMichael Ellerman #endif
808