xref: /linux-6.15/arch/powerpc/kernel/module_64.c (revision 2fd98637)
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. */
36*2fd98637SChristophe Leroy typedef struct {
37*2fd98637SChristophe Leroy 	unsigned long addr;
38*2fd98637SChristophe Leroy } func_desc_t;
39008d7a91SRusty Russell 
40008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr)
41008d7a91SRusty Russell {
42*2fd98637SChristophe Leroy 	func_desc_t desc = {
43*2fd98637SChristophe Leroy 		.addr = addr,
44*2fd98637SChristophe Leroy 	};
45*2fd98637SChristophe Leroy 
46*2fd98637SChristophe Leroy 	return desc;
47008d7a91SRusty Russell }
48008d7a91SRusty Russell 
49008d7a91SRusty Russell /* PowerPC64 specific values for the Elf64_Sym st_other field.  */
50008d7a91SRusty Russell #define STO_PPC64_LOCAL_BIT	5
51008d7a91SRusty Russell #define STO_PPC64_LOCAL_MASK	(7 << STO_PPC64_LOCAL_BIT)
52008d7a91SRusty Russell #define PPC64_LOCAL_ENTRY_OFFSET(other)					\
53008d7a91SRusty Russell  (((1 << (((other) & STO_PPC64_LOCAL_MASK) >> STO_PPC64_LOCAL_BIT)) >> 2) << 2)
54008d7a91SRusty Russell 
55008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym)
56008d7a91SRusty Russell {
57008d7a91SRusty Russell 	/* sym->st_other indicates offset to local entry point
58008d7a91SRusty Russell 	 * (otherwise it will assume r12 is the address of the start
59008d7a91SRusty Russell 	 * of function and try to derive r2 from it). */
60008d7a91SRusty Russell 	return PPC64_LOCAL_ENTRY_OFFSET(sym->st_other);
61008d7a91SRusty Russell }
62d2fae548SRusty Russell #else
63008d7a91SRusty Russell 
64008d7a91SRusty Russell /* An address is address of the OPD entry, which contains address of fn. */
65d3e32b99SChristophe Leroy typedef struct func_desc func_desc_t;
66008d7a91SRusty Russell 
67008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr)
68008d7a91SRusty Russell {
69d3e32b99SChristophe Leroy 	return *(struct func_desc *)addr;
70008d7a91SRusty Russell }
71008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym)
72008d7a91SRusty Russell {
73008d7a91SRusty Russell 	return 0;
74008d7a91SRusty Russell }
755633e85bSSergey Senozhatsky 
765633e85bSSergey Senozhatsky void *dereference_module_function_descriptor(struct module *mod, void *ptr)
775633e85bSSergey Senozhatsky {
785633e85bSSergey Senozhatsky 	if (ptr < (void *)mod->arch.start_opd ||
795633e85bSSergey Senozhatsky 			ptr >= (void *)mod->arch.end_opd)
805633e85bSSergey Senozhatsky 		return ptr;
815633e85bSSergey Senozhatsky 
825633e85bSSergey Senozhatsky 	return dereference_function_descriptor(ptr);
835633e85bSSergey Senozhatsky }
84d2fae548SRusty Russell #endif
85d2fae548SRusty Russell 
86*2fd98637SChristophe Leroy static unsigned long func_addr(unsigned long addr)
87*2fd98637SChristophe Leroy {
88*2fd98637SChristophe Leroy 	return func_desc(addr).addr;
89*2fd98637SChristophe Leroy }
90*2fd98637SChristophe Leroy 
91*2fd98637SChristophe Leroy static unsigned long stub_func_addr(func_desc_t func)
92*2fd98637SChristophe Leroy {
93*2fd98637SChristophe Leroy 	return func.addr;
94*2fd98637SChristophe Leroy }
95*2fd98637SChristophe Leroy 
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 {
190d3e32b99SChristophe Leroy 	/* One extra reloc so it's always 0-addr 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 
2807568cb4eSPaul Mackerras int module_frob_arch_sections(Elf64_Ehdr *hdr,
2817568cb4eSPaul Mackerras 			      Elf64_Shdr *sechdrs,
2827568cb4eSPaul Mackerras 			      char *secstrings,
2837568cb4eSPaul Mackerras 			      struct module *me)
2847568cb4eSPaul Mackerras {
2857568cb4eSPaul Mackerras 	unsigned int i;
2867568cb4eSPaul Mackerras 
2877568cb4eSPaul Mackerras 	/* Find .toc and .stubs sections, symtab and strtab */
2887568cb4eSPaul Mackerras 	for (i = 1; i < hdr->e_shnum; i++) {
2897568cb4eSPaul Mackerras 		char *p;
2907568cb4eSPaul Mackerras 		if (strcmp(secstrings + sechdrs[i].sh_name, ".stubs") == 0)
2917568cb4eSPaul Mackerras 			me->arch.stubs_section = i;
2925c45b528SAlan Modra 		else if (strcmp(secstrings + sechdrs[i].sh_name, ".toc") == 0) {
2937568cb4eSPaul Mackerras 			me->arch.toc_section = i;
2945c45b528SAlan Modra 			if (sechdrs[i].sh_addralign < 8)
2955c45b528SAlan Modra 				sechdrs[i].sh_addralign = 8;
2965c45b528SAlan Modra 		}
2977568cb4eSPaul Mackerras 		else if (strcmp(secstrings+sechdrs[i].sh_name,"__versions")==0)
2987568cb4eSPaul Mackerras 			dedotify_versions((void *)hdr + sechdrs[i].sh_offset,
2997568cb4eSPaul Mackerras 					  sechdrs[i].sh_size);
3007568cb4eSPaul Mackerras 
3017568cb4eSPaul Mackerras 		/* We don't handle .init for the moment: rename to _init */
3027568cb4eSPaul Mackerras 		while ((p = strstr(secstrings + sechdrs[i].sh_name, ".init")))
3037568cb4eSPaul Mackerras 			p[0] = '_';
3047568cb4eSPaul Mackerras 
3057568cb4eSPaul Mackerras 		if (sechdrs[i].sh_type == SHT_SYMTAB)
3067568cb4eSPaul Mackerras 			dedotify((void *)hdr + sechdrs[i].sh_offset,
3077568cb4eSPaul Mackerras 				 sechdrs[i].sh_size / sizeof(Elf64_Sym),
3087568cb4eSPaul Mackerras 				 (void *)hdr
3097568cb4eSPaul Mackerras 				 + sechdrs[sechdrs[i].sh_link].sh_offset);
3107568cb4eSPaul Mackerras 	}
311f749edaeSAlan Modra 
312f749edaeSAlan Modra 	if (!me->arch.stubs_section) {
313c7d1f6afSAnton Blanchard 		pr_err("%s: doesn't contain .stubs.\n", me->name);
3147568cb4eSPaul Mackerras 		return -ENOEXEC;
3157568cb4eSPaul Mackerras 	}
3167568cb4eSPaul Mackerras 
317f749edaeSAlan Modra 	/* If we don't have a .toc, just use .stubs.  We need to set r2
318f749edaeSAlan Modra 	   to some reasonable value in case the module calls out to
319f749edaeSAlan Modra 	   other functions via a stub, or if a function pointer escapes
320f749edaeSAlan Modra 	   the module by some means.  */
321f749edaeSAlan Modra 	if (!me->arch.toc_section)
322f749edaeSAlan Modra 		me->arch.toc_section = me->arch.stubs_section;
323f749edaeSAlan Modra 
3247568cb4eSPaul Mackerras 	/* Override the stubs size */
3257568cb4eSPaul Mackerras 	sechdrs[me->arch.stubs_section].sh_size = get_stubs_size(hdr, sechdrs);
3267568cb4eSPaul Mackerras 	return 0;
3277568cb4eSPaul Mackerras }
3287568cb4eSPaul Mackerras 
329bd55e792SNaveen N. Rao #ifdef CONFIG_MPROFILE_KERNEL
330bd55e792SNaveen N. Rao 
331bd55e792SNaveen N. Rao static u32 stub_insns[] = {
33247b04699SChristophe Leroy 	PPC_RAW_LD(_R12, _R13, offsetof(struct paca_struct, kernel_toc)),
33347b04699SChristophe Leroy 	PPC_RAW_ADDIS(_R12, _R12, 0),
33447b04699SChristophe Leroy 	PPC_RAW_ADDI(_R12, _R12, 0),
33547b04699SChristophe Leroy 	PPC_RAW_MTCTR(_R12),
33647b04699SChristophe Leroy 	PPC_RAW_BCTR(),
337bd55e792SNaveen N. Rao };
338bd55e792SNaveen N. Rao 
339bd55e792SNaveen N. Rao /*
340bd55e792SNaveen N. Rao  * For mprofile-kernel we use a special stub for ftrace_caller() because we
341bd55e792SNaveen N. Rao  * can't rely on r2 containing this module's TOC when we enter the stub.
342bd55e792SNaveen N. Rao  *
343bd55e792SNaveen N. Rao  * That can happen if the function calling us didn't need to use the toc. In
344bd55e792SNaveen N. Rao  * that case it won't have setup r2, and the r2 value will be either the
345bd55e792SNaveen N. Rao  * kernel's toc, or possibly another modules toc.
346bd55e792SNaveen N. Rao  *
347bd55e792SNaveen N. Rao  * To deal with that this stub uses the kernel toc, which is always accessible
348bd55e792SNaveen N. Rao  * via the paca (in r13). The target (ftrace_caller()) is responsible for
349bd55e792SNaveen N. Rao  * saving and restoring the toc before returning.
350bd55e792SNaveen N. Rao  */
351bd55e792SNaveen N. Rao static inline int create_ftrace_stub(struct ppc64_stub_entry *entry,
352bd55e792SNaveen N. Rao 					unsigned long addr,
353bd55e792SNaveen N. Rao 					struct module *me)
354bd55e792SNaveen N. Rao {
355bd55e792SNaveen N. Rao 	long reladdr;
356bd55e792SNaveen N. Rao 
357bd55e792SNaveen N. Rao 	memcpy(entry->jump, stub_insns, sizeof(stub_insns));
358bd55e792SNaveen N. Rao 
359bd55e792SNaveen N. Rao 	/* Stub uses address relative to kernel toc (from the paca) */
360bd55e792SNaveen N. Rao 	reladdr = addr - kernel_toc_addr();
361bd55e792SNaveen N. Rao 	if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) {
362bd55e792SNaveen N. Rao 		pr_err("%s: Address of %ps out of range of kernel_toc.\n",
363bd55e792SNaveen N. Rao 							me->name, (void *)addr);
364bd55e792SNaveen N. Rao 		return 0;
365bd55e792SNaveen N. Rao 	}
366bd55e792SNaveen N. Rao 
367bd55e792SNaveen N. Rao 	entry->jump[1] |= PPC_HA(reladdr);
368bd55e792SNaveen N. Rao 	entry->jump[2] |= PPC_LO(reladdr);
369bd55e792SNaveen N. Rao 
370bd55e792SNaveen N. Rao 	/* Eventhough we don't use funcdata in the stub, it's needed elsewhere. */
371bd55e792SNaveen N. Rao 	entry->funcdata = func_desc(addr);
372bd55e792SNaveen N. Rao 	entry->magic = STUB_MAGIC;
373bd55e792SNaveen N. Rao 
374bd55e792SNaveen N. Rao 	return 1;
375bd55e792SNaveen N. Rao }
376bd55e792SNaveen N. Rao 
377bd55e792SNaveen N. Rao static bool is_mprofile_ftrace_call(const char *name)
378bd55e792SNaveen N. Rao {
379bd55e792SNaveen N. Rao 	if (!strcmp("_mcount", name))
380bd55e792SNaveen N. Rao 		return true;
381bd55e792SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE
382bd55e792SNaveen N. Rao 	if (!strcmp("ftrace_caller", name))
383bd55e792SNaveen N. Rao 		return true;
384bd55e792SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
385bd55e792SNaveen N. Rao 	if (!strcmp("ftrace_regs_caller", name))
386bd55e792SNaveen N. Rao 		return true;
387bd55e792SNaveen N. Rao #endif
388bd55e792SNaveen N. Rao #endif
389bd55e792SNaveen N. Rao 
390bd55e792SNaveen N. Rao 	return false;
391bd55e792SNaveen N. Rao }
392bd55e792SNaveen N. Rao #else
393bd55e792SNaveen N. Rao static inline int create_ftrace_stub(struct ppc64_stub_entry *entry,
394bd55e792SNaveen N. Rao 					unsigned long addr,
395bd55e792SNaveen N. Rao 					struct module *me)
396bd55e792SNaveen N. Rao {
397bd55e792SNaveen N. Rao 	return 0;
398bd55e792SNaveen N. Rao }
399bd55e792SNaveen N. Rao 
400bd55e792SNaveen N. Rao static bool is_mprofile_ftrace_call(const char *name)
401bd55e792SNaveen N. Rao {
402bd55e792SNaveen N. Rao 	return false;
403bd55e792SNaveen N. Rao }
404bd55e792SNaveen N. Rao #endif
405bd55e792SNaveen N. Rao 
4065c45b528SAlan Modra /*
4075c45b528SAlan Modra  * r2 is the TOC pointer: it actually points 0x8000 into the TOC (this gives the
4085c45b528SAlan Modra  * value maximum span in an instruction which uses a signed offset). Round down
4095c45b528SAlan Modra  * to a 256 byte boundary for the odd case where we are setting up r2 without a
4105c45b528SAlan Modra  * .toc section.
4115c45b528SAlan Modra  */
412136cd345SMichael Ellerman static inline unsigned long my_r2(const Elf64_Shdr *sechdrs, struct module *me)
4137568cb4eSPaul Mackerras {
4145c45b528SAlan Modra 	return (sechdrs[me->arch.toc_section].sh_addr & ~0xfful) + 0x8000;
4157568cb4eSPaul Mackerras }
4167568cb4eSPaul Mackerras 
4177568cb4eSPaul Mackerras /* Patch stub to reference function and correct r2 value. */
418136cd345SMichael Ellerman static inline int create_stub(const Elf64_Shdr *sechdrs,
4197568cb4eSPaul Mackerras 			      struct ppc64_stub_entry *entry,
420008d7a91SRusty Russell 			      unsigned long addr,
421bd55e792SNaveen N. Rao 			      struct module *me,
422bd55e792SNaveen N. Rao 			      const char *name)
4237568cb4eSPaul Mackerras {
4247568cb4eSPaul Mackerras 	long reladdr;
4258734b41bSRussell Currey 	func_desc_t desc;
4268734b41bSRussell Currey 	int i;
4277568cb4eSPaul Mackerras 
428bd55e792SNaveen N. Rao 	if (is_mprofile_ftrace_call(name))
429bd55e792SNaveen N. Rao 		return create_ftrace_stub(entry, addr, me);
430bd55e792SNaveen N. Rao 
4318734b41bSRussell Currey 	for (i = 0; i < sizeof(ppc64_stub_insns) / sizeof(u32); i++) {
4328734b41bSRussell Currey 		if (patch_instruction(&entry->jump[i],
4338734b41bSRussell Currey 				      ppc_inst(ppc64_stub_insns[i])))
4348734b41bSRussell Currey 			return 0;
4358734b41bSRussell Currey 	}
4367568cb4eSPaul Mackerras 
4377568cb4eSPaul Mackerras 	/* Stub uses address relative to r2. */
4387568cb4eSPaul Mackerras 	reladdr = (unsigned long)entry - my_r2(sechdrs, me);
4397568cb4eSPaul Mackerras 	if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) {
440c7d1f6afSAnton Blanchard 		pr_err("%s: Address %p of stub out of range of %p.\n",
4417568cb4eSPaul Mackerras 		       me->name, (void *)reladdr, (void *)my_r2);
4427568cb4eSPaul Mackerras 		return 0;
4437568cb4eSPaul Mackerras 	}
444c7d1f6afSAnton Blanchard 	pr_debug("Stub %p get data from reladdr %li\n", entry, reladdr);
4457568cb4eSPaul Mackerras 
4468734b41bSRussell Currey 	if (patch_instruction(&entry->jump[0],
4478734b41bSRussell Currey 			      ppc_inst(entry->jump[0] | PPC_HA(reladdr))))
4488734b41bSRussell Currey 		return 0;
4498734b41bSRussell Currey 
4508734b41bSRussell Currey 	if (patch_instruction(&entry->jump[1],
4518734b41bSRussell Currey 			  ppc_inst(entry->jump[1] | PPC_LO(reladdr))))
4528734b41bSRussell Currey 		return 0;
4538734b41bSRussell Currey 
4548734b41bSRussell Currey 	// func_desc_t is 8 bytes if ABIv2, else 16 bytes
4558734b41bSRussell Currey 	desc = func_desc(addr);
4568734b41bSRussell Currey 	for (i = 0; i < sizeof(func_desc_t) / sizeof(u32); i++) {
4578734b41bSRussell Currey 		if (patch_instruction(((u32 *)&entry->funcdata) + i,
4588734b41bSRussell Currey 				      ppc_inst(((u32 *)(&desc))[i])))
4598734b41bSRussell Currey 			return 0;
4608734b41bSRussell Currey 	}
4618734b41bSRussell Currey 
4628734b41bSRussell Currey 	if (patch_instruction(&entry->magic, ppc_inst(STUB_MAGIC)))
4638734b41bSRussell Currey 		return 0;
464f17c4e01SMichael Ellerman 
4657568cb4eSPaul Mackerras 	return 1;
4667568cb4eSPaul Mackerras }
4677568cb4eSPaul Mackerras 
468008d7a91SRusty Russell /* Create stub to jump to function described in this OPD/ptr: we need the
4697568cb4eSPaul Mackerras    stub to set up the TOC ptr (r2) for the function. */
470136cd345SMichael Ellerman static unsigned long stub_for_addr(const Elf64_Shdr *sechdrs,
471008d7a91SRusty Russell 				   unsigned long addr,
472bd55e792SNaveen N. Rao 				   struct module *me,
473bd55e792SNaveen N. Rao 				   const char *name)
4747568cb4eSPaul Mackerras {
4757568cb4eSPaul Mackerras 	struct ppc64_stub_entry *stubs;
4767568cb4eSPaul Mackerras 	unsigned int i, num_stubs;
4777568cb4eSPaul Mackerras 
4787568cb4eSPaul Mackerras 	num_stubs = sechdrs[me->arch.stubs_section].sh_size / sizeof(*stubs);
4797568cb4eSPaul Mackerras 
4807568cb4eSPaul Mackerras 	/* Find this stub, or if that fails, the next avail. entry */
4817568cb4eSPaul Mackerras 	stubs = (void *)sechdrs[me->arch.stubs_section].sh_addr;
482008d7a91SRusty Russell 	for (i = 0; stub_func_addr(stubs[i].funcdata); i++) {
4831c0437afSKamalesh Babulal 		if (WARN_ON(i >= num_stubs))
4841c0437afSKamalesh Babulal 			return 0;
4857568cb4eSPaul Mackerras 
486008d7a91SRusty Russell 		if (stub_func_addr(stubs[i].funcdata) == func_addr(addr))
4877568cb4eSPaul Mackerras 			return (unsigned long)&stubs[i];
4887568cb4eSPaul Mackerras 	}
4897568cb4eSPaul Mackerras 
490bd55e792SNaveen N. Rao 	if (!create_stub(sechdrs, &stubs[i], addr, me, name))
4917568cb4eSPaul Mackerras 		return 0;
4927568cb4eSPaul Mackerras 
4937568cb4eSPaul Mackerras 	return (unsigned long)&stubs[i];
4947568cb4eSPaul Mackerras }
4957568cb4eSPaul Mackerras 
4967568cb4eSPaul Mackerras /* We expect a noop next: if it is, replace it with instruction to
4977568cb4eSPaul Mackerras    restore r2. */
498250122baSNaveen N. Rao static int restore_r2(const char *name, u32 *instruction, struct module *me)
4997568cb4eSPaul Mackerras {
500b9eab08dSJosh Poimboeuf 	u32 *prev_insn = instruction - 1;
501b9eab08dSJosh Poimboeuf 
5021f2aaed2SNaveen N. Rao 	if (is_mprofile_ftrace_call(name))
503b9eab08dSJosh Poimboeuf 		return 1;
504b9eab08dSJosh Poimboeuf 
505b9eab08dSJosh Poimboeuf 	/*
506b9eab08dSJosh Poimboeuf 	 * Make sure the branch isn't a sibling call.  Sibling calls aren't
507b9eab08dSJosh Poimboeuf 	 * "link" branches and they don't return, so they don't need the r2
508b9eab08dSJosh Poimboeuf 	 * restore afterwards.
509b9eab08dSJosh Poimboeuf 	 */
51075346251SJordan Niethe 	if (!instr_is_relative_link_branch(ppc_inst(*prev_insn)))
51115308664STorsten Duwe 		return 1;
51231278b17SMichael Ellerman 
51347b04699SChristophe Leroy 	if (*instruction != PPC_RAW_NOP()) {
5141ea61ea2SJosh Poimboeuf 		pr_err("%s: Expected nop after call, got %08x at %pS\n",
5151ea61ea2SJosh Poimboeuf 			me->name, *instruction, instruction);
5167568cb4eSPaul Mackerras 		return 0;
5177568cb4eSPaul Mackerras 	}
5188734b41bSRussell Currey 
519d2fae548SRusty Russell 	/* ld r2,R2_STACK_OFFSET(r1) */
5208734b41bSRussell Currey 	if (patch_instruction(instruction, ppc_inst(PPC_INST_LD_TOC)))
5218734b41bSRussell Currey 		return 0;
5228734b41bSRussell Currey 
5237568cb4eSPaul Mackerras 	return 1;
5247568cb4eSPaul Mackerras }
5257568cb4eSPaul Mackerras 
5267568cb4eSPaul Mackerras int apply_relocate_add(Elf64_Shdr *sechdrs,
5277568cb4eSPaul Mackerras 		       const char *strtab,
5287568cb4eSPaul Mackerras 		       unsigned int symindex,
5297568cb4eSPaul Mackerras 		       unsigned int relsec,
5307568cb4eSPaul Mackerras 		       struct module *me)
5317568cb4eSPaul Mackerras {
5327568cb4eSPaul Mackerras 	unsigned int i;
5337568cb4eSPaul Mackerras 	Elf64_Rela *rela = (void *)sechdrs[relsec].sh_addr;
5347568cb4eSPaul Mackerras 	Elf64_Sym *sym;
5357568cb4eSPaul Mackerras 	unsigned long *location;
5367568cb4eSPaul Mackerras 	unsigned long value;
5377568cb4eSPaul Mackerras 
538c7d1f6afSAnton Blanchard 	pr_debug("Applying ADD relocate section %u to %u\n", relsec,
5397568cb4eSPaul Mackerras 	       sechdrs[relsec].sh_info);
5404edebbeaSRusty Russell 
5414edebbeaSRusty Russell 	/* First time we're called, we can fix up .TOC. */
5424edebbeaSRusty Russell 	if (!me->arch.toc_fixed) {
5434edebbeaSRusty Russell 		sym = find_dot_toc(sechdrs, strtab, symindex);
5444edebbeaSRusty Russell 		/* It's theoretically possible that a module doesn't want a
5454edebbeaSRusty Russell 		 * .TOC. so don't fail it just for that. */
5464edebbeaSRusty Russell 		if (sym)
5474edebbeaSRusty Russell 			sym->st_value = my_r2(sechdrs, me);
5484edebbeaSRusty Russell 		me->arch.toc_fixed = true;
5494edebbeaSRusty Russell 	}
5504edebbeaSRusty Russell 
5517568cb4eSPaul Mackerras 	for (i = 0; i < sechdrs[relsec].sh_size / sizeof(*rela); i++) {
5527568cb4eSPaul Mackerras 		/* This is where to make the change */
5537568cb4eSPaul Mackerras 		location = (void *)sechdrs[sechdrs[relsec].sh_info].sh_addr
5547568cb4eSPaul Mackerras 			+ rela[i].r_offset;
5557568cb4eSPaul Mackerras 		/* This is the symbol it is referring to */
5567568cb4eSPaul Mackerras 		sym = (Elf64_Sym *)sechdrs[symindex].sh_addr
5577568cb4eSPaul Mackerras 			+ ELF64_R_SYM(rela[i].r_info);
5587568cb4eSPaul Mackerras 
559c7d1f6afSAnton Blanchard 		pr_debug("RELOC at %p: %li-type as %s (0x%lx) + %li\n",
5607568cb4eSPaul Mackerras 		       location, (long)ELF64_R_TYPE(rela[i].r_info),
5617568cb4eSPaul Mackerras 		       strtab + sym->st_name, (unsigned long)sym->st_value,
5627568cb4eSPaul Mackerras 		       (long)rela[i].r_addend);
5637568cb4eSPaul Mackerras 
5647568cb4eSPaul Mackerras 		/* `Everything is relative'. */
5657568cb4eSPaul Mackerras 		value = sym->st_value + rela[i].r_addend;
5667568cb4eSPaul Mackerras 
5677568cb4eSPaul Mackerras 		switch (ELF64_R_TYPE(rela[i].r_info)) {
5687568cb4eSPaul Mackerras 		case R_PPC64_ADDR32:
5697568cb4eSPaul Mackerras 			/* Simply set it */
5707568cb4eSPaul Mackerras 			*(u32 *)location = value;
5717568cb4eSPaul Mackerras 			break;
5727568cb4eSPaul Mackerras 
5737568cb4eSPaul Mackerras 		case R_PPC64_ADDR64:
5747568cb4eSPaul Mackerras 			/* Simply set it */
5757568cb4eSPaul Mackerras 			*(unsigned long *)location = value;
5767568cb4eSPaul Mackerras 			break;
5777568cb4eSPaul Mackerras 
5787568cb4eSPaul Mackerras 		case R_PPC64_TOC:
5797568cb4eSPaul Mackerras 			*(unsigned long *)location = my_r2(sechdrs, me);
5807568cb4eSPaul Mackerras 			break;
5817568cb4eSPaul Mackerras 
5827568cb4eSPaul Mackerras 		case R_PPC64_TOC16:
583f749edaeSAlan Modra 			/* Subtract TOC pointer */
5847568cb4eSPaul Mackerras 			value -= my_r2(sechdrs, me);
5857568cb4eSPaul Mackerras 			if (value + 0x8000 > 0xffff) {
586c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16 relocation (0x%lx)\n",
5877568cb4eSPaul Mackerras 				       me->name, value);
5887568cb4eSPaul Mackerras 				return -ENOEXEC;
5897568cb4eSPaul Mackerras 			}
5907568cb4eSPaul Mackerras 			*((uint16_t *) location)
5917568cb4eSPaul Mackerras 				= (*((uint16_t *) location) & ~0xffff)
5927568cb4eSPaul Mackerras 				| (value & 0xffff);
5937568cb4eSPaul Mackerras 			break;
5947568cb4eSPaul Mackerras 
5951fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_LO:
5961fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
5971fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
5981fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
5991fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xffff)
6001fbe9cf2SAnton Blanchard 				| (value & 0xffff);
6011fbe9cf2SAnton Blanchard 			break;
6021fbe9cf2SAnton Blanchard 
6037568cb4eSPaul Mackerras 		case R_PPC64_TOC16_DS:
604f749edaeSAlan Modra 			/* Subtract TOC pointer */
6057568cb4eSPaul Mackerras 			value -= my_r2(sechdrs, me);
6067568cb4eSPaul Mackerras 			if ((value & 3) != 0 || value + 0x8000 > 0xffff) {
607c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16_DS relocation (0x%lx)\n",
6087568cb4eSPaul Mackerras 				       me->name, value);
6097568cb4eSPaul Mackerras 				return -ENOEXEC;
6107568cb4eSPaul Mackerras 			}
6117568cb4eSPaul Mackerras 			*((uint16_t *) location)
6127568cb4eSPaul Mackerras 				= (*((uint16_t *) location) & ~0xfffc)
6137568cb4eSPaul Mackerras 				| (value & 0xfffc);
6147568cb4eSPaul Mackerras 			break;
6157568cb4eSPaul Mackerras 
6161fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_LO_DS:
6171fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
6181fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6191fbe9cf2SAnton Blanchard 			if ((value & 3) != 0) {
620c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16_LO_DS relocation (0x%lx)\n",
6211fbe9cf2SAnton Blanchard 				       me->name, value);
6221fbe9cf2SAnton Blanchard 				return -ENOEXEC;
6231fbe9cf2SAnton Blanchard 			}
6241fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6251fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xfffc)
6261fbe9cf2SAnton Blanchard 				| (value & 0xfffc);
6271fbe9cf2SAnton Blanchard 			break;
6281fbe9cf2SAnton Blanchard 
6291fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_HA:
6301fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
6311fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6321fbe9cf2SAnton Blanchard 			value = ((value + 0x8000) >> 16);
6331fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6341fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xffff)
6351fbe9cf2SAnton Blanchard 				| (value & 0xffff);
6361fbe9cf2SAnton Blanchard 			break;
6371fbe9cf2SAnton Blanchard 
6387568cb4eSPaul Mackerras 		case R_PPC_REL24:
6397568cb4eSPaul Mackerras 			/* FIXME: Handle weak symbols here --RR */
640a443bf6eSKamalesh Babulal 			if (sym->st_shndx == SHN_UNDEF ||
641a443bf6eSKamalesh Babulal 			    sym->st_shndx == SHN_LIVEPATCH) {
6427568cb4eSPaul Mackerras 				/* External: go via stub */
643bd55e792SNaveen N. Rao 				value = stub_for_addr(sechdrs, value, me,
644bd55e792SNaveen N. Rao 						strtab + sym->st_name);
6457568cb4eSPaul Mackerras 				if (!value)
6467568cb4eSPaul Mackerras 					return -ENOENT;
647250122baSNaveen N. Rao 				if (!restore_r2(strtab + sym->st_name,
648250122baSNaveen N. Rao 							(u32 *)location + 1, me))
6497568cb4eSPaul Mackerras 					return -ENOEXEC;
650008d7a91SRusty Russell 			} else
651008d7a91SRusty Russell 				value += local_entry_offset(sym);
6527568cb4eSPaul Mackerras 
6537568cb4eSPaul Mackerras 			/* Convert value to relative */
6547568cb4eSPaul Mackerras 			value -= (unsigned long)location;
6557568cb4eSPaul Mackerras 			if (value + 0x2000000 > 0x3ffffff || (value & 3) != 0){
656c7d1f6afSAnton Blanchard 				pr_err("%s: REL24 %li out of range!\n",
6577568cb4eSPaul Mackerras 				       me->name, (long int)value);
6587568cb4eSPaul Mackerras 				return -ENOEXEC;
6597568cb4eSPaul Mackerras 			}
6607568cb4eSPaul Mackerras 
6617568cb4eSPaul Mackerras 			/* Only replace bits 2 through 26 */
6628734b41bSRussell Currey 			value = (*(uint32_t *)location & ~0x03fffffc)
6637568cb4eSPaul Mackerras 				| (value & 0x03fffffc);
6648734b41bSRussell Currey 
6658734b41bSRussell Currey 			if (patch_instruction((u32 *)location, ppc_inst(value)))
6668734b41bSRussell Currey 				return -EFAULT;
6678734b41bSRussell Currey 
6687568cb4eSPaul Mackerras 			break;
6697568cb4eSPaul Mackerras 
67021c4ff80SBenjamin Herrenschmidt 		case R_PPC64_REL64:
67121c4ff80SBenjamin Herrenschmidt 			/* 64 bits relative (used by features fixups) */
67221c4ff80SBenjamin Herrenschmidt 			*location = value - (unsigned long)location;
67321c4ff80SBenjamin Herrenschmidt 			break;
67421c4ff80SBenjamin Herrenschmidt 
6759f751b82SMichael Ellerman 		case R_PPC64_REL32:
6769f751b82SMichael Ellerman 			/* 32 bits relative (used by relative exception tables) */
677b851ba02SNicholas Piggin 			/* Convert value to relative */
678b851ba02SNicholas Piggin 			value -= (unsigned long)location;
679b851ba02SNicholas Piggin 			if (value + 0x80000000 > 0xffffffff) {
680b851ba02SNicholas Piggin 				pr_err("%s: REL32 %li out of range!\n",
681b851ba02SNicholas Piggin 				       me->name, (long int)value);
682b851ba02SNicholas Piggin 				return -ENOEXEC;
683b851ba02SNicholas Piggin 			}
684b851ba02SNicholas Piggin 			*(u32 *)location = value;
6859f751b82SMichael Ellerman 			break;
6869f751b82SMichael Ellerman 
687d247da0aSRusty Russell 		case R_PPC64_TOCSAVE:
688d247da0aSRusty Russell 			/*
689d247da0aSRusty Russell 			 * Marker reloc indicates we don't have to save r2.
690d247da0aSRusty Russell 			 * That would only save us one instruction, so ignore
691d247da0aSRusty Russell 			 * it.
692d247da0aSRusty Russell 			 */
693d247da0aSRusty Russell 			break;
694d247da0aSRusty Russell 
695a61674bdSUlrich Weigand 		case R_PPC64_ENTRY:
696a61674bdSUlrich Weigand 			/*
697a61674bdSUlrich Weigand 			 * Optimize ELFv2 large code model entry point if
698a61674bdSUlrich Weigand 			 * the TOC is within 2GB range of current location.
699a61674bdSUlrich Weigand 			 */
700a61674bdSUlrich Weigand 			value = my_r2(sechdrs, me) - (unsigned long)location;
701a61674bdSUlrich Weigand 			if (value + 0x80008000 > 0xffffffff)
702a61674bdSUlrich Weigand 				break;
703a61674bdSUlrich Weigand 			/*
704a61674bdSUlrich Weigand 			 * Check for the large code model prolog sequence:
705a61674bdSUlrich Weigand 		         *	ld r2, ...(r12)
706a61674bdSUlrich Weigand 			 *	add r2, r2, r12
707a61674bdSUlrich Weigand 			 */
70847b04699SChristophe Leroy 			if ((((uint32_t *)location)[0] & ~0xfffc) != PPC_RAW_LD(_R2, _R12, 0))
709a61674bdSUlrich Weigand 				break;
71047b04699SChristophe Leroy 			if (((uint32_t *)location)[1] != PPC_RAW_ADD(_R2, _R2, _R12))
711a61674bdSUlrich Weigand 				break;
712a61674bdSUlrich Weigand 			/*
713a61674bdSUlrich Weigand 			 * If found, replace it with:
714a61674bdSUlrich Weigand 			 *	addis r2, r12, (.TOC.-func)@ha
7152fb0a2c9SMichael Ellerman 			 *	addi  r2,  r2, (.TOC.-func)@l
716a61674bdSUlrich Weigand 			 */
71747b04699SChristophe Leroy 			((uint32_t *)location)[0] = PPC_RAW_ADDIS(_R2, _R12, PPC_HA(value));
71847b04699SChristophe Leroy 			((uint32_t *)location)[1] = PPC_RAW_ADDI(_R2, _R2, PPC_LO(value));
719a61674bdSUlrich Weigand 			break;
720a61674bdSUlrich Weigand 
7210906584aSRusty Russell 		case R_PPC64_REL16_HA:
7220906584aSRusty Russell 			/* Subtract location pointer */
7230906584aSRusty Russell 			value -= (unsigned long)location;
7240906584aSRusty Russell 			value = ((value + 0x8000) >> 16);
7250906584aSRusty Russell 			*((uint16_t *) location)
7260906584aSRusty Russell 				= (*((uint16_t *) location) & ~0xffff)
7270906584aSRusty Russell 				| (value & 0xffff);
7280906584aSRusty Russell 			break;
7290906584aSRusty Russell 
7300906584aSRusty Russell 		case R_PPC64_REL16_LO:
7310906584aSRusty Russell 			/* Subtract location pointer */
7320906584aSRusty Russell 			value -= (unsigned long)location;
7330906584aSRusty Russell 			*((uint16_t *) location)
7340906584aSRusty Russell 				= (*((uint16_t *) location) & ~0xffff)
7350906584aSRusty Russell 				| (value & 0xffff);
7360906584aSRusty Russell 			break;
7370906584aSRusty Russell 
7387568cb4eSPaul Mackerras 		default:
739c7d1f6afSAnton Blanchard 			pr_err("%s: Unknown ADD relocation: %lu\n",
7407568cb4eSPaul Mackerras 			       me->name,
7417568cb4eSPaul Mackerras 			       (unsigned long)ELF64_R_TYPE(rela[i].r_info));
7427568cb4eSPaul Mackerras 			return -ENOEXEC;
7437568cb4eSPaul Mackerras 		}
7447568cb4eSPaul Mackerras 	}
7457568cb4eSPaul Mackerras 
746136cd345SMichael Ellerman 	return 0;
747136cd345SMichael Ellerman }
748136cd345SMichael Ellerman 
749f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
75003b51416SNaveen N. Rao int module_trampoline_target(struct module *mod, unsigned long addr,
75103b51416SNaveen N. Rao 			     unsigned long *target)
75203b51416SNaveen N. Rao {
75303b51416SNaveen N. Rao 	struct ppc64_stub_entry *stub;
75403b51416SNaveen N. Rao 	func_desc_t funcdata;
75503b51416SNaveen N. Rao 	u32 magic;
75603b51416SNaveen N. Rao 
75703b51416SNaveen N. Rao 	if (!within_module_core(addr, mod)) {
75803b51416SNaveen N. Rao 		pr_err("%s: stub %lx not in module %s\n", __func__, addr, mod->name);
75903b51416SNaveen N. Rao 		return -EFAULT;
76003b51416SNaveen N. Rao 	}
76103b51416SNaveen N. Rao 
76203b51416SNaveen N. Rao 	stub = (struct ppc64_stub_entry *)addr;
76303b51416SNaveen N. Rao 
764fe557319SChristoph Hellwig 	if (copy_from_kernel_nofault(&magic, &stub->magic,
765fe557319SChristoph Hellwig 			sizeof(magic))) {
76603b51416SNaveen N. Rao 		pr_err("%s: fault reading magic for stub %lx for %s\n", __func__, addr, mod->name);
76703b51416SNaveen N. Rao 		return -EFAULT;
76803b51416SNaveen N. Rao 	}
76903b51416SNaveen N. Rao 
77003b51416SNaveen N. Rao 	if (magic != STUB_MAGIC) {
77103b51416SNaveen N. Rao 		pr_err("%s: bad magic for stub %lx for %s\n", __func__, addr, mod->name);
77203b51416SNaveen N. Rao 		return -EFAULT;
77303b51416SNaveen N. Rao 	}
77403b51416SNaveen N. Rao 
775fe557319SChristoph Hellwig 	if (copy_from_kernel_nofault(&funcdata, &stub->funcdata,
776fe557319SChristoph Hellwig 			sizeof(funcdata))) {
77703b51416SNaveen N. Rao 		pr_err("%s: fault reading funcdata for stub %lx for %s\n", __func__, addr, mod->name);
77803b51416SNaveen N. Rao                 return -EFAULT;
77903b51416SNaveen N. Rao 	}
78003b51416SNaveen N. Rao 
78103b51416SNaveen N. Rao 	*target = stub_func_addr(funcdata);
78203b51416SNaveen N. Rao 
78303b51416SNaveen N. Rao 	return 0;
78403b51416SNaveen N. Rao }
785336a7b5dSMichael Ellerman 
786136cd345SMichael Ellerman int module_finalize_ftrace(struct module *mod, const Elf_Shdr *sechdrs)
787136cd345SMichael Ellerman {
788bd55e792SNaveen N. Rao 	mod->arch.tramp = stub_for_addr(sechdrs,
789bd55e792SNaveen N. Rao 					(unsigned long)ftrace_caller,
790bd55e792SNaveen N. Rao 					mod,
791bd55e792SNaveen N. Rao 					"ftrace_caller");
792ae30cc05SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
793bd55e792SNaveen N. Rao 	mod->arch.tramp_regs = stub_for_addr(sechdrs,
794bd55e792SNaveen N. Rao 					(unsigned long)ftrace_regs_caller,
795bd55e792SNaveen N. Rao 					mod,
796bd55e792SNaveen N. Rao 					"ftrace_regs_caller");
797ae30cc05SNaveen N. Rao 	if (!mod->arch.tramp_regs)
798ae30cc05SNaveen N. Rao 		return -ENOENT;
799ae30cc05SNaveen N. Rao #endif
800136cd345SMichael Ellerman 
801136cd345SMichael Ellerman 	if (!mod->arch.tramp)
802136cd345SMichael Ellerman 		return -ENOENT;
803f48cb8b4SSteven Rostedt 
8047568cb4eSPaul Mackerras 	return 0;
8057568cb4eSPaul Mackerras }
806136cd345SMichael Ellerman #endif
807