xref: /linux-6.15/arch/powerpc/kernel/module_64.c (revision 1fd02f66)
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>
178a0edc72SGuo 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 static func_desc_t func_desc(unsigned long addr)
37008d7a91SRusty Russell {
382fd98637SChristophe Leroy 	func_desc_t desc = {
392fd98637SChristophe Leroy 		.addr = addr,
402fd98637SChristophe Leroy 	};
412fd98637SChristophe Leroy 
422fd98637SChristophe Leroy 	return desc;
43008d7a91SRusty Russell }
44008d7a91SRusty Russell 
45008d7a91SRusty Russell /* PowerPC64 specific values for the Elf64_Sym st_other field.  */
46008d7a91SRusty Russell #define STO_PPC64_LOCAL_BIT	5
47008d7a91SRusty Russell #define STO_PPC64_LOCAL_MASK	(7 << STO_PPC64_LOCAL_BIT)
48008d7a91SRusty Russell #define PPC64_LOCAL_ENTRY_OFFSET(other)					\
49008d7a91SRusty Russell  (((1 << (((other) & STO_PPC64_LOCAL_MASK) >> STO_PPC64_LOCAL_BIT)) >> 2) << 2)
50008d7a91SRusty Russell 
51008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym)
52008d7a91SRusty Russell {
53008d7a91SRusty Russell 	/* sym->st_other indicates offset to local entry point
54008d7a91SRusty Russell 	 * (otherwise it will assume r12 is the address of the start
55008d7a91SRusty Russell 	 * of function and try to derive r2 from it). */
56008d7a91SRusty Russell 	return PPC64_LOCAL_ENTRY_OFFSET(sym->st_other);
57008d7a91SRusty Russell }
58d2fae548SRusty Russell #else
59008d7a91SRusty Russell 
60008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr)
61008d7a91SRusty Russell {
62d3e32b99SChristophe Leroy 	return *(struct func_desc *)addr;
63008d7a91SRusty Russell }
64008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym)
65008d7a91SRusty Russell {
66008d7a91SRusty Russell 	return 0;
67008d7a91SRusty Russell }
685633e85bSSergey Senozhatsky 
695633e85bSSergey Senozhatsky void *dereference_module_function_descriptor(struct module *mod, void *ptr)
705633e85bSSergey Senozhatsky {
715633e85bSSergey Senozhatsky 	if (ptr < (void *)mod->arch.start_opd ||
725633e85bSSergey Senozhatsky 			ptr >= (void *)mod->arch.end_opd)
735633e85bSSergey Senozhatsky 		return ptr;
745633e85bSSergey Senozhatsky 
755633e85bSSergey Senozhatsky 	return dereference_function_descriptor(ptr);
765633e85bSSergey Senozhatsky }
77d2fae548SRusty Russell #endif
78d2fae548SRusty Russell 
792fd98637SChristophe Leroy static unsigned long func_addr(unsigned long addr)
802fd98637SChristophe Leroy {
812fd98637SChristophe Leroy 	return func_desc(addr).addr;
822fd98637SChristophe Leroy }
832fd98637SChristophe Leroy 
842fd98637SChristophe Leroy static unsigned long stub_func_addr(func_desc_t func)
852fd98637SChristophe Leroy {
862fd98637SChristophe Leroy 	return func.addr;
872fd98637SChristophe Leroy }
882fd98637SChristophe Leroy 
89f17c4e01SMichael Ellerman #define STUB_MAGIC 0x73747562 /* stub */
90f17c4e01SMichael Ellerman 
917568cb4eSPaul Mackerras /* Like PPC32, we need little trampolines to do > 24-bit jumps (into
927568cb4eSPaul Mackerras    the kernel itself).  But on PPC64, these need to be used for every
937568cb4eSPaul Mackerras    jump, actually, to reset r2 (TOC+0x8000). */
947568cb4eSPaul Mackerras struct ppc64_stub_entry
957568cb4eSPaul Mackerras {
9683775b85SAnton Blanchard 	/* 28 byte jump instruction sequence (7 instructions). We only
9783775b85SAnton Blanchard 	 * need 6 instructions on ABIv2 but we always allocate 7 so
9883775b85SAnton Blanchard 	 * so we don't have to modify the trampoline load instruction. */
990e60e46eSRusty Russell 	u32 jump[7];
100f17c4e01SMichael Ellerman 	/* Used by ftrace to identify stubs */
101f17c4e01SMichael Ellerman 	u32 magic;
1027568cb4eSPaul Mackerras 	/* Data for the above code */
103008d7a91SRusty Russell 	func_desc_t funcdata;
1047568cb4eSPaul Mackerras };
1057568cb4eSPaul Mackerras 
1065c729a11SRusty Russell /*
1075c729a11SRusty Russell  * PPC64 uses 24 bit jumps, but we need to jump into other modules or
1085c729a11SRusty Russell  * the kernel which may be further.  So we jump to a stub.
1095c729a11SRusty Russell  *
1105c729a11SRusty Russell  * For ELFv1 we need to use this to set up the new r2 value (aka TOC
1115c729a11SRusty Russell  * pointer).  For ELFv2 it's the callee's responsibility to set up the
1125c729a11SRusty Russell  * new r2, but for both we need to save the old r2.
1135c729a11SRusty Russell  *
1145c729a11SRusty Russell  * We could simply patch the new r2 value and function pointer into
1155c729a11SRusty Russell  * the stub, but it's significantly shorter to put these values at the
1165c729a11SRusty Russell  * end of the stub code, and patch the stub address (32-bits relative
1175c729a11SRusty Russell  * to the TOC ptr, r2) into the stub.
1185c729a11SRusty Russell  */
11983775b85SAnton Blanchard static u32 ppc64_stub_insns[] = {
12047b04699SChristophe Leroy 	PPC_RAW_ADDIS(_R11, _R2, 0),
12147b04699SChristophe Leroy 	PPC_RAW_ADDI(_R11, _R11, 0),
122fed8393eSEugene Surovegin 	/* Save current r2 value in magic place on the stack. */
12347b04699SChristophe Leroy 	PPC_RAW_STD(_R2, _R1, R2_STACK_OFFSET),
12447b04699SChristophe Leroy 	PPC_RAW_LD(_R12, _R11, 32),
125f55d9665SMichael Ellerman #ifdef PPC64_ELF_ABI_v1
1265c729a11SRusty Russell 	/* Set up new r2 from function descriptor */
12747b04699SChristophe Leroy 	PPC_RAW_LD(_R2, _R11, 40),
1285c729a11SRusty Russell #endif
12947b04699SChristophe Leroy 	PPC_RAW_MTCTR(_R12),
13047b04699SChristophe Leroy 	PPC_RAW_BCTR(),
13183775b85SAnton Blanchard };
13283775b85SAnton Blanchard 
1337568cb4eSPaul Mackerras /* Count how many different 24-bit relocations (different symbol,
1347568cb4eSPaul Mackerras    different addend) */
1357568cb4eSPaul Mackerras static unsigned int count_relocs(const Elf64_Rela *rela, unsigned int num)
1367568cb4eSPaul Mackerras {
137eda09fbdSEmil Medve 	unsigned int i, r_info, r_addend, _count_relocs;
1387568cb4eSPaul Mackerras 
1397568cb4eSPaul Mackerras 	/* FIXME: Only count external ones --RR */
140eda09fbdSEmil Medve 	_count_relocs = 0;
141eda09fbdSEmil Medve 	r_info = 0;
142eda09fbdSEmil Medve 	r_addend = 0;
143eda09fbdSEmil Medve 	for (i = 0; i < num; i++)
1447568cb4eSPaul Mackerras 		/* Only count 24-bit relocs, others don't need stubs */
145eda09fbdSEmil Medve 		if (ELF64_R_TYPE(rela[i].r_info) == R_PPC_REL24 &&
146eda09fbdSEmil Medve 		    (r_info != ELF64_R_SYM(rela[i].r_info) ||
147eda09fbdSEmil Medve 		     r_addend != rela[i].r_addend)) {
148eda09fbdSEmil Medve 			_count_relocs++;
149eda09fbdSEmil Medve 			r_info = ELF64_R_SYM(rela[i].r_info);
150eda09fbdSEmil Medve 			r_addend = rela[i].r_addend;
1517568cb4eSPaul Mackerras 		}
152eda09fbdSEmil Medve 
153eda09fbdSEmil Medve 	return _count_relocs;
1547568cb4eSPaul Mackerras }
1557568cb4eSPaul Mackerras 
156eda09fbdSEmil Medve static int relacmp(const void *_x, const void *_y)
157eda09fbdSEmil Medve {
158eda09fbdSEmil Medve 	const Elf64_Rela *x, *y;
159eda09fbdSEmil Medve 
160eda09fbdSEmil Medve 	y = (Elf64_Rela *)_x;
161eda09fbdSEmil Medve 	x = (Elf64_Rela *)_y;
162eda09fbdSEmil Medve 
163eda09fbdSEmil Medve 	/* Compare the entire r_info (as opposed to ELF64_R_SYM(r_info) only) to
164eda09fbdSEmil Medve 	 * make the comparison cheaper/faster. It won't affect the sorting or
165eda09fbdSEmil Medve 	 * the counting algorithms' performance
166eda09fbdSEmil Medve 	 */
167eda09fbdSEmil Medve 	if (x->r_info < y->r_info)
168eda09fbdSEmil Medve 		return -1;
169eda09fbdSEmil Medve 	else if (x->r_info > y->r_info)
170eda09fbdSEmil Medve 		return 1;
171eda09fbdSEmil Medve 	else if (x->r_addend < y->r_addend)
172eda09fbdSEmil Medve 		return -1;
173eda09fbdSEmil Medve 	else if (x->r_addend > y->r_addend)
174eda09fbdSEmil Medve 		return 1;
175eda09fbdSEmil Medve 	else
176eda09fbdSEmil Medve 		return 0;
177eda09fbdSEmil Medve }
178eda09fbdSEmil Medve 
1797568cb4eSPaul Mackerras /* Get size of potential trampolines required. */
1807568cb4eSPaul Mackerras static unsigned long get_stubs_size(const Elf64_Ehdr *hdr,
1817568cb4eSPaul Mackerras 				    const Elf64_Shdr *sechdrs)
1827568cb4eSPaul Mackerras {
183d3e32b99SChristophe Leroy 	/* One extra reloc so it's always 0-addr terminated */
1847568cb4eSPaul Mackerras 	unsigned long relocs = 1;
1857568cb4eSPaul Mackerras 	unsigned i;
1867568cb4eSPaul Mackerras 
1877568cb4eSPaul Mackerras 	/* Every relocated section... */
1887568cb4eSPaul Mackerras 	for (i = 1; i < hdr->e_shnum; i++) {
1897568cb4eSPaul Mackerras 		if (sechdrs[i].sh_type == SHT_RELA) {
190c7d1f6afSAnton Blanchard 			pr_debug("Found relocations in section %u\n", i);
191c7d1f6afSAnton Blanchard 			pr_debug("Ptr: %p.  Number: %Lu\n",
1927568cb4eSPaul Mackerras 			       (void *)sechdrs[i].sh_addr,
1937568cb4eSPaul Mackerras 			       sechdrs[i].sh_size / sizeof(Elf64_Rela));
194eda09fbdSEmil Medve 
195eda09fbdSEmil Medve 			/* Sort the relocation information based on a symbol and
196eda09fbdSEmil Medve 			 * addend key. This is a stable O(n*log n) complexity
197*1fd02f66SJulia Lawall 			 * algorithm but it will reduce the complexity of
198eda09fbdSEmil Medve 			 * count_relocs() to linear complexity O(n)
199eda09fbdSEmil Medve 			 */
200eda09fbdSEmil Medve 			sort((void *)sechdrs[i].sh_addr,
201eda09fbdSEmil Medve 			     sechdrs[i].sh_size / sizeof(Elf64_Rela),
202bac7ca7bSAndrey Abramov 			     sizeof(Elf64_Rela), relacmp, NULL);
203eda09fbdSEmil Medve 
2047568cb4eSPaul Mackerras 			relocs += count_relocs((void *)sechdrs[i].sh_addr,
2057568cb4eSPaul Mackerras 					       sechdrs[i].sh_size
2067568cb4eSPaul Mackerras 					       / sizeof(Elf64_Rela));
2077568cb4eSPaul Mackerras 		}
2087568cb4eSPaul Mackerras 	}
2097568cb4eSPaul Mackerras 
210f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
211f48cb8b4SSteven Rostedt 	/* make the trampoline to the ftrace_caller */
212f48cb8b4SSteven Rostedt 	relocs++;
213ae30cc05SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
214ae30cc05SNaveen N. Rao 	/* an additional one for ftrace_regs_caller */
215ae30cc05SNaveen N. Rao 	relocs++;
216ae30cc05SNaveen N. Rao #endif
217f48cb8b4SSteven Rostedt #endif
218f48cb8b4SSteven Rostedt 
219c7d1f6afSAnton Blanchard 	pr_debug("Looks like a total of %lu stubs, max\n", relocs);
2207568cb4eSPaul Mackerras 	return relocs * sizeof(struct ppc64_stub_entry);
2217568cb4eSPaul Mackerras }
2227568cb4eSPaul Mackerras 
2235b12c5c6SRusty Russell /* Still needed for ELFv2, for .TOC. */
2247568cb4eSPaul Mackerras static void dedotify_versions(struct modversion_info *vers,
2257568cb4eSPaul Mackerras 			      unsigned long size)
2267568cb4eSPaul Mackerras {
2277568cb4eSPaul Mackerras 	struct modversion_info *end;
2287568cb4eSPaul Mackerras 
2297568cb4eSPaul Mackerras 	for (end = (void *)vers + size; vers < end; vers++)
230c2cbcf53SLaurent Dufour 		if (vers->name[0] == '.') {
2317568cb4eSPaul Mackerras 			memmove(vers->name, vers->name+1, strlen(vers->name));
232c2cbcf53SLaurent Dufour 		}
2337568cb4eSPaul Mackerras }
2347568cb4eSPaul Mackerras 
235c153693dSAlan Modra /*
236c153693dSAlan Modra  * Undefined symbols which refer to .funcname, hack to funcname. Make .TOC.
237c153693dSAlan Modra  * seem to be defined (value set later).
238c153693dSAlan Modra  */
2397568cb4eSPaul Mackerras static void dedotify(Elf64_Sym *syms, unsigned int numsyms, char *strtab)
2407568cb4eSPaul Mackerras {
2417568cb4eSPaul Mackerras 	unsigned int i;
2427568cb4eSPaul Mackerras 
2437568cb4eSPaul Mackerras 	for (i = 1; i < numsyms; i++) {
2447568cb4eSPaul Mackerras 		if (syms[i].st_shndx == SHN_UNDEF) {
2457568cb4eSPaul Mackerras 			char *name = strtab + syms[i].st_name;
246c153693dSAlan Modra 			if (name[0] == '.') {
247c153693dSAlan Modra 				if (strcmp(name+1, "TOC.") == 0)
248c153693dSAlan Modra 					syms[i].st_shndx = SHN_ABS;
249f15838e9SAndreas Schwab 				syms[i].st_name++;
2507568cb4eSPaul Mackerras 			}
2517568cb4eSPaul Mackerras 		}
2527568cb4eSPaul Mackerras 	}
253c153693dSAlan Modra }
2547568cb4eSPaul Mackerras 
2554edebbeaSRusty Russell static Elf64_Sym *find_dot_toc(Elf64_Shdr *sechdrs,
2564edebbeaSRusty Russell 			       const char *strtab,
2574edebbeaSRusty Russell 			       unsigned int symindex)
2584edebbeaSRusty Russell {
2594edebbeaSRusty Russell 	unsigned int i, numsyms;
2604edebbeaSRusty Russell 	Elf64_Sym *syms;
2614edebbeaSRusty Russell 
2624edebbeaSRusty Russell 	syms = (Elf64_Sym *)sechdrs[symindex].sh_addr;
2634edebbeaSRusty Russell 	numsyms = sechdrs[symindex].sh_size / sizeof(Elf64_Sym);
2644edebbeaSRusty Russell 
2654edebbeaSRusty Russell 	for (i = 1; i < numsyms; i++) {
266c153693dSAlan Modra 		if (syms[i].st_shndx == SHN_ABS
267008d7a91SRusty Russell 		    && strcmp(strtab + syms[i].st_name, "TOC.") == 0)
2684edebbeaSRusty Russell 			return &syms[i];
2694edebbeaSRusty Russell 	}
2704edebbeaSRusty Russell 	return NULL;
2714edebbeaSRusty Russell }
2724edebbeaSRusty Russell 
273d4be60feSWedson Almeida Filho bool module_init_section(const char *name)
274d4be60feSWedson Almeida Filho {
275d4be60feSWedson Almeida Filho 	/* We don't handle .init for the moment: always return false. */
276d4be60feSWedson Almeida Filho 	return false;
277d4be60feSWedson Almeida Filho }
278d4be60feSWedson Almeida Filho 
2797568cb4eSPaul Mackerras int module_frob_arch_sections(Elf64_Ehdr *hdr,
2807568cb4eSPaul Mackerras 			      Elf64_Shdr *sechdrs,
2817568cb4eSPaul Mackerras 			      char *secstrings,
2827568cb4eSPaul Mackerras 			      struct module *me)
2837568cb4eSPaul Mackerras {
2847568cb4eSPaul Mackerras 	unsigned int i;
2857568cb4eSPaul Mackerras 
2867568cb4eSPaul Mackerras 	/* Find .toc and .stubs sections, symtab and strtab */
2877568cb4eSPaul Mackerras 	for (i = 1; i < hdr->e_shnum; i++) {
2887568cb4eSPaul Mackerras 		if (strcmp(secstrings + sechdrs[i].sh_name, ".stubs") == 0)
2897568cb4eSPaul Mackerras 			me->arch.stubs_section = i;
2905c45b528SAlan Modra 		else if (strcmp(secstrings + sechdrs[i].sh_name, ".toc") == 0) {
2917568cb4eSPaul Mackerras 			me->arch.toc_section = i;
2925c45b528SAlan Modra 			if (sechdrs[i].sh_addralign < 8)
2935c45b528SAlan Modra 				sechdrs[i].sh_addralign = 8;
2945c45b528SAlan Modra 		}
2957568cb4eSPaul Mackerras 		else if (strcmp(secstrings+sechdrs[i].sh_name,"__versions")==0)
2967568cb4eSPaul Mackerras 			dedotify_versions((void *)hdr + sechdrs[i].sh_offset,
2977568cb4eSPaul Mackerras 					  sechdrs[i].sh_size);
2987568cb4eSPaul Mackerras 
2997568cb4eSPaul Mackerras 		if (sechdrs[i].sh_type == SHT_SYMTAB)
3007568cb4eSPaul Mackerras 			dedotify((void *)hdr + sechdrs[i].sh_offset,
3017568cb4eSPaul Mackerras 				 sechdrs[i].sh_size / sizeof(Elf64_Sym),
3027568cb4eSPaul Mackerras 				 (void *)hdr
3037568cb4eSPaul Mackerras 				 + sechdrs[sechdrs[i].sh_link].sh_offset);
3047568cb4eSPaul Mackerras 	}
305f749edaeSAlan Modra 
306f749edaeSAlan Modra 	if (!me->arch.stubs_section) {
307c7d1f6afSAnton Blanchard 		pr_err("%s: doesn't contain .stubs.\n", me->name);
3087568cb4eSPaul Mackerras 		return -ENOEXEC;
3097568cb4eSPaul Mackerras 	}
3107568cb4eSPaul Mackerras 
311f749edaeSAlan Modra 	/* If we don't have a .toc, just use .stubs.  We need to set r2
312f749edaeSAlan Modra 	   to some reasonable value in case the module calls out to
313f749edaeSAlan Modra 	   other functions via a stub, or if a function pointer escapes
314f749edaeSAlan Modra 	   the module by some means.  */
315f749edaeSAlan Modra 	if (!me->arch.toc_section)
316f749edaeSAlan Modra 		me->arch.toc_section = me->arch.stubs_section;
317f749edaeSAlan Modra 
3187568cb4eSPaul Mackerras 	/* Override the stubs size */
3197568cb4eSPaul Mackerras 	sechdrs[me->arch.stubs_section].sh_size = get_stubs_size(hdr, sechdrs);
3207568cb4eSPaul Mackerras 	return 0;
3217568cb4eSPaul Mackerras }
3227568cb4eSPaul Mackerras 
323bd55e792SNaveen N. Rao #ifdef CONFIG_MPROFILE_KERNEL
324bd55e792SNaveen N. Rao 
325bd55e792SNaveen N. Rao static u32 stub_insns[] = {
32647b04699SChristophe Leroy 	PPC_RAW_LD(_R12, _R13, offsetof(struct paca_struct, kernel_toc)),
32747b04699SChristophe Leroy 	PPC_RAW_ADDIS(_R12, _R12, 0),
32847b04699SChristophe Leroy 	PPC_RAW_ADDI(_R12, _R12, 0),
32947b04699SChristophe Leroy 	PPC_RAW_MTCTR(_R12),
33047b04699SChristophe Leroy 	PPC_RAW_BCTR(),
331bd55e792SNaveen N. Rao };
332bd55e792SNaveen N. Rao 
333bd55e792SNaveen N. Rao /*
334bd55e792SNaveen N. Rao  * For mprofile-kernel we use a special stub for ftrace_caller() because we
335bd55e792SNaveen N. Rao  * can't rely on r2 containing this module's TOC when we enter the stub.
336bd55e792SNaveen N. Rao  *
337bd55e792SNaveen N. Rao  * That can happen if the function calling us didn't need to use the toc. In
338bd55e792SNaveen N. Rao  * that case it won't have setup r2, and the r2 value will be either the
339bd55e792SNaveen N. Rao  * kernel's toc, or possibly another modules toc.
340bd55e792SNaveen N. Rao  *
341bd55e792SNaveen N. Rao  * To deal with that this stub uses the kernel toc, which is always accessible
342bd55e792SNaveen N. Rao  * via the paca (in r13). The target (ftrace_caller()) is responsible for
343bd55e792SNaveen N. Rao  * saving and restoring the toc before returning.
344bd55e792SNaveen N. Rao  */
345bd55e792SNaveen N. Rao static inline int create_ftrace_stub(struct ppc64_stub_entry *entry,
346bd55e792SNaveen N. Rao 					unsigned long addr,
347bd55e792SNaveen N. Rao 					struct module *me)
348bd55e792SNaveen N. Rao {
349bd55e792SNaveen N. Rao 	long reladdr;
350bd55e792SNaveen N. Rao 
351bd55e792SNaveen N. Rao 	memcpy(entry->jump, stub_insns, sizeof(stub_insns));
352bd55e792SNaveen N. Rao 
353bd55e792SNaveen N. Rao 	/* Stub uses address relative to kernel toc (from the paca) */
354bd55e792SNaveen N. Rao 	reladdr = addr - kernel_toc_addr();
355bd55e792SNaveen N. Rao 	if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) {
356bd55e792SNaveen N. Rao 		pr_err("%s: Address of %ps out of range of kernel_toc.\n",
357bd55e792SNaveen N. Rao 							me->name, (void *)addr);
358bd55e792SNaveen N. Rao 		return 0;
359bd55e792SNaveen N. Rao 	}
360bd55e792SNaveen N. Rao 
361bd55e792SNaveen N. Rao 	entry->jump[1] |= PPC_HA(reladdr);
362bd55e792SNaveen N. Rao 	entry->jump[2] |= PPC_LO(reladdr);
363bd55e792SNaveen N. Rao 
364bd55e792SNaveen N. Rao 	/* Even though we don't use funcdata in the stub, it's needed elsewhere. */
365bd55e792SNaveen N. Rao 	entry->funcdata = func_desc(addr);
366bd55e792SNaveen N. Rao 	entry->magic = STUB_MAGIC;
367bd55e792SNaveen N. Rao 
368bd55e792SNaveen N. Rao 	return 1;
369bd55e792SNaveen N. Rao }
370bd55e792SNaveen N. Rao 
371bd55e792SNaveen N. Rao static bool is_mprofile_ftrace_call(const char *name)
372bd55e792SNaveen N. Rao {
373bd55e792SNaveen N. Rao 	if (!strcmp("_mcount", name))
374bd55e792SNaveen N. Rao 		return true;
375bd55e792SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE
376bd55e792SNaveen N. Rao 	if (!strcmp("ftrace_caller", name))
377bd55e792SNaveen N. Rao 		return true;
378bd55e792SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
379bd55e792SNaveen N. Rao 	if (!strcmp("ftrace_regs_caller", name))
380bd55e792SNaveen N. Rao 		return true;
381bd55e792SNaveen N. Rao #endif
382bd55e792SNaveen N. Rao #endif
383bd55e792SNaveen N. Rao 
384bd55e792SNaveen N. Rao 	return false;
385bd55e792SNaveen N. Rao }
386bd55e792SNaveen N. Rao #else
387bd55e792SNaveen N. Rao static inline int create_ftrace_stub(struct ppc64_stub_entry *entry,
388bd55e792SNaveen N. Rao 					unsigned long addr,
389bd55e792SNaveen N. Rao 					struct module *me)
390bd55e792SNaveen N. Rao {
391bd55e792SNaveen N. Rao 	return 0;
392bd55e792SNaveen N. Rao }
393bd55e792SNaveen N. Rao 
394bd55e792SNaveen N. Rao static bool is_mprofile_ftrace_call(const char *name)
395bd55e792SNaveen N. Rao {
396bd55e792SNaveen N. Rao 	return false;
397bd55e792SNaveen N. Rao }
398bd55e792SNaveen N. Rao #endif
399bd55e792SNaveen N. Rao 
4005c45b528SAlan Modra /*
4015c45b528SAlan Modra  * r2 is the TOC pointer: it actually points 0x8000 into the TOC (this gives the
4025c45b528SAlan Modra  * value maximum span in an instruction which uses a signed offset). Round down
4035c45b528SAlan Modra  * to a 256 byte boundary for the odd case where we are setting up r2 without a
4045c45b528SAlan Modra  * .toc section.
4055c45b528SAlan Modra  */
406136cd345SMichael Ellerman static inline unsigned long my_r2(const Elf64_Shdr *sechdrs, struct module *me)
4077568cb4eSPaul Mackerras {
4085c45b528SAlan Modra 	return (sechdrs[me->arch.toc_section].sh_addr & ~0xfful) + 0x8000;
4097568cb4eSPaul Mackerras }
4107568cb4eSPaul Mackerras 
4117568cb4eSPaul Mackerras /* Patch stub to reference function and correct r2 value. */
412136cd345SMichael Ellerman static inline int create_stub(const Elf64_Shdr *sechdrs,
4137568cb4eSPaul Mackerras 			      struct ppc64_stub_entry *entry,
414008d7a91SRusty Russell 			      unsigned long addr,
415bd55e792SNaveen N. Rao 			      struct module *me,
416bd55e792SNaveen N. Rao 			      const char *name)
4177568cb4eSPaul Mackerras {
4187568cb4eSPaul Mackerras 	long reladdr;
4198734b41bSRussell Currey 	func_desc_t desc;
4208734b41bSRussell Currey 	int i;
4217568cb4eSPaul Mackerras 
422bd55e792SNaveen N. Rao 	if (is_mprofile_ftrace_call(name))
423bd55e792SNaveen N. Rao 		return create_ftrace_stub(entry, addr, me);
424bd55e792SNaveen N. Rao 
4258a0edc72SGuo Zhengkui 	for (i = 0; i < ARRAY_SIZE(ppc64_stub_insns); i++) {
4268734b41bSRussell Currey 		if (patch_instruction(&entry->jump[i],
4278734b41bSRussell Currey 				      ppc_inst(ppc64_stub_insns[i])))
4288734b41bSRussell Currey 			return 0;
4298734b41bSRussell Currey 	}
4307568cb4eSPaul Mackerras 
4317568cb4eSPaul Mackerras 	/* Stub uses address relative to r2. */
4327568cb4eSPaul Mackerras 	reladdr = (unsigned long)entry - my_r2(sechdrs, me);
4337568cb4eSPaul Mackerras 	if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) {
434c7d1f6afSAnton Blanchard 		pr_err("%s: Address %p of stub out of range of %p.\n",
4357568cb4eSPaul Mackerras 		       me->name, (void *)reladdr, (void *)my_r2);
4367568cb4eSPaul Mackerras 		return 0;
4377568cb4eSPaul Mackerras 	}
438c7d1f6afSAnton Blanchard 	pr_debug("Stub %p get data from reladdr %li\n", entry, reladdr);
4397568cb4eSPaul Mackerras 
4408734b41bSRussell Currey 	if (patch_instruction(&entry->jump[0],
4418734b41bSRussell Currey 			      ppc_inst(entry->jump[0] | PPC_HA(reladdr))))
4428734b41bSRussell Currey 		return 0;
4438734b41bSRussell Currey 
4448734b41bSRussell Currey 	if (patch_instruction(&entry->jump[1],
4458734b41bSRussell Currey 			  ppc_inst(entry->jump[1] | PPC_LO(reladdr))))
4468734b41bSRussell Currey 		return 0;
4478734b41bSRussell Currey 
4488734b41bSRussell Currey 	// func_desc_t is 8 bytes if ABIv2, else 16 bytes
4498734b41bSRussell Currey 	desc = func_desc(addr);
4508734b41bSRussell Currey 	for (i = 0; i < sizeof(func_desc_t) / sizeof(u32); i++) {
4518734b41bSRussell Currey 		if (patch_instruction(((u32 *)&entry->funcdata) + i,
4528734b41bSRussell Currey 				      ppc_inst(((u32 *)(&desc))[i])))
4538734b41bSRussell Currey 			return 0;
4548734b41bSRussell Currey 	}
4558734b41bSRussell Currey 
4568734b41bSRussell Currey 	if (patch_instruction(&entry->magic, ppc_inst(STUB_MAGIC)))
4578734b41bSRussell Currey 		return 0;
458f17c4e01SMichael Ellerman 
4597568cb4eSPaul Mackerras 	return 1;
4607568cb4eSPaul Mackerras }
4617568cb4eSPaul Mackerras 
462008d7a91SRusty Russell /* Create stub to jump to function described in this OPD/ptr: we need the
4637568cb4eSPaul Mackerras    stub to set up the TOC ptr (r2) for the function. */
464136cd345SMichael Ellerman static unsigned long stub_for_addr(const Elf64_Shdr *sechdrs,
465008d7a91SRusty Russell 				   unsigned long addr,
466bd55e792SNaveen N. Rao 				   struct module *me,
467bd55e792SNaveen N. Rao 				   const char *name)
4687568cb4eSPaul Mackerras {
4697568cb4eSPaul Mackerras 	struct ppc64_stub_entry *stubs;
4707568cb4eSPaul Mackerras 	unsigned int i, num_stubs;
4717568cb4eSPaul Mackerras 
4727568cb4eSPaul Mackerras 	num_stubs = sechdrs[me->arch.stubs_section].sh_size / sizeof(*stubs);
4737568cb4eSPaul Mackerras 
4747568cb4eSPaul Mackerras 	/* Find this stub, or if that fails, the next avail. entry */
4757568cb4eSPaul Mackerras 	stubs = (void *)sechdrs[me->arch.stubs_section].sh_addr;
476008d7a91SRusty Russell 	for (i = 0; stub_func_addr(stubs[i].funcdata); i++) {
4771c0437afSKamalesh Babulal 		if (WARN_ON(i >= num_stubs))
4781c0437afSKamalesh Babulal 			return 0;
4797568cb4eSPaul Mackerras 
480008d7a91SRusty Russell 		if (stub_func_addr(stubs[i].funcdata) == func_addr(addr))
4817568cb4eSPaul Mackerras 			return (unsigned long)&stubs[i];
4827568cb4eSPaul Mackerras 	}
4837568cb4eSPaul Mackerras 
484bd55e792SNaveen N. Rao 	if (!create_stub(sechdrs, &stubs[i], addr, me, name))
4857568cb4eSPaul Mackerras 		return 0;
4867568cb4eSPaul Mackerras 
4877568cb4eSPaul Mackerras 	return (unsigned long)&stubs[i];
4887568cb4eSPaul Mackerras }
4897568cb4eSPaul Mackerras 
4907568cb4eSPaul Mackerras /* We expect a noop next: if it is, replace it with instruction to
4917568cb4eSPaul Mackerras    restore r2. */
492250122baSNaveen N. Rao static int restore_r2(const char *name, u32 *instruction, struct module *me)
4937568cb4eSPaul Mackerras {
494b9eab08dSJosh Poimboeuf 	u32 *prev_insn = instruction - 1;
495b9eab08dSJosh Poimboeuf 
4961f2aaed2SNaveen N. Rao 	if (is_mprofile_ftrace_call(name))
497b9eab08dSJosh Poimboeuf 		return 1;
498b9eab08dSJosh Poimboeuf 
499b9eab08dSJosh Poimboeuf 	/*
500b9eab08dSJosh Poimboeuf 	 * Make sure the branch isn't a sibling call.  Sibling calls aren't
501b9eab08dSJosh Poimboeuf 	 * "link" branches and they don't return, so they don't need the r2
502b9eab08dSJosh Poimboeuf 	 * restore afterwards.
503b9eab08dSJosh Poimboeuf 	 */
50475346251SJordan Niethe 	if (!instr_is_relative_link_branch(ppc_inst(*prev_insn)))
50515308664STorsten Duwe 		return 1;
50631278b17SMichael Ellerman 
50747b04699SChristophe Leroy 	if (*instruction != PPC_RAW_NOP()) {
5081ea61ea2SJosh Poimboeuf 		pr_err("%s: Expected nop after call, got %08x at %pS\n",
5091ea61ea2SJosh Poimboeuf 			me->name, *instruction, instruction);
5107568cb4eSPaul Mackerras 		return 0;
5117568cb4eSPaul Mackerras 	}
5128734b41bSRussell Currey 
513d2fae548SRusty Russell 	/* ld r2,R2_STACK_OFFSET(r1) */
5148734b41bSRussell Currey 	if (patch_instruction(instruction, ppc_inst(PPC_INST_LD_TOC)))
5158734b41bSRussell Currey 		return 0;
5168734b41bSRussell Currey 
5177568cb4eSPaul Mackerras 	return 1;
5187568cb4eSPaul Mackerras }
5197568cb4eSPaul Mackerras 
5207568cb4eSPaul Mackerras int apply_relocate_add(Elf64_Shdr *sechdrs,
5217568cb4eSPaul Mackerras 		       const char *strtab,
5227568cb4eSPaul Mackerras 		       unsigned int symindex,
5237568cb4eSPaul Mackerras 		       unsigned int relsec,
5247568cb4eSPaul Mackerras 		       struct module *me)
5257568cb4eSPaul Mackerras {
5267568cb4eSPaul Mackerras 	unsigned int i;
5277568cb4eSPaul Mackerras 	Elf64_Rela *rela = (void *)sechdrs[relsec].sh_addr;
5287568cb4eSPaul Mackerras 	Elf64_Sym *sym;
5297568cb4eSPaul Mackerras 	unsigned long *location;
5307568cb4eSPaul Mackerras 	unsigned long value;
5317568cb4eSPaul Mackerras 
532c7d1f6afSAnton Blanchard 	pr_debug("Applying ADD relocate section %u to %u\n", relsec,
5337568cb4eSPaul Mackerras 	       sechdrs[relsec].sh_info);
5344edebbeaSRusty Russell 
5354edebbeaSRusty Russell 	/* First time we're called, we can fix up .TOC. */
5364edebbeaSRusty Russell 	if (!me->arch.toc_fixed) {
5374edebbeaSRusty Russell 		sym = find_dot_toc(sechdrs, strtab, symindex);
5384edebbeaSRusty Russell 		/* It's theoretically possible that a module doesn't want a
5394edebbeaSRusty Russell 		 * .TOC. so don't fail it just for that. */
5404edebbeaSRusty Russell 		if (sym)
5414edebbeaSRusty Russell 			sym->st_value = my_r2(sechdrs, me);
5424edebbeaSRusty Russell 		me->arch.toc_fixed = true;
5434edebbeaSRusty Russell 	}
5444edebbeaSRusty Russell 
5457568cb4eSPaul Mackerras 	for (i = 0; i < sechdrs[relsec].sh_size / sizeof(*rela); i++) {
5467568cb4eSPaul Mackerras 		/* This is where to make the change */
5477568cb4eSPaul Mackerras 		location = (void *)sechdrs[sechdrs[relsec].sh_info].sh_addr
5487568cb4eSPaul Mackerras 			+ rela[i].r_offset;
5497568cb4eSPaul Mackerras 		/* This is the symbol it is referring to */
5507568cb4eSPaul Mackerras 		sym = (Elf64_Sym *)sechdrs[symindex].sh_addr
5517568cb4eSPaul Mackerras 			+ ELF64_R_SYM(rela[i].r_info);
5527568cb4eSPaul Mackerras 
553c7d1f6afSAnton Blanchard 		pr_debug("RELOC at %p: %li-type as %s (0x%lx) + %li\n",
5547568cb4eSPaul Mackerras 		       location, (long)ELF64_R_TYPE(rela[i].r_info),
5557568cb4eSPaul Mackerras 		       strtab + sym->st_name, (unsigned long)sym->st_value,
5567568cb4eSPaul Mackerras 		       (long)rela[i].r_addend);
5577568cb4eSPaul Mackerras 
5587568cb4eSPaul Mackerras 		/* `Everything is relative'. */
5597568cb4eSPaul Mackerras 		value = sym->st_value + rela[i].r_addend;
5607568cb4eSPaul Mackerras 
5617568cb4eSPaul Mackerras 		switch (ELF64_R_TYPE(rela[i].r_info)) {
5627568cb4eSPaul Mackerras 		case R_PPC64_ADDR32:
5637568cb4eSPaul Mackerras 			/* Simply set it */
5647568cb4eSPaul Mackerras 			*(u32 *)location = value;
5657568cb4eSPaul Mackerras 			break;
5667568cb4eSPaul Mackerras 
5677568cb4eSPaul Mackerras 		case R_PPC64_ADDR64:
5687568cb4eSPaul Mackerras 			/* Simply set it */
5697568cb4eSPaul Mackerras 			*(unsigned long *)location = value;
5707568cb4eSPaul Mackerras 			break;
5717568cb4eSPaul Mackerras 
5727568cb4eSPaul Mackerras 		case R_PPC64_TOC:
5737568cb4eSPaul Mackerras 			*(unsigned long *)location = my_r2(sechdrs, me);
5747568cb4eSPaul Mackerras 			break;
5757568cb4eSPaul Mackerras 
5767568cb4eSPaul Mackerras 		case R_PPC64_TOC16:
577f749edaeSAlan Modra 			/* Subtract TOC pointer */
5787568cb4eSPaul Mackerras 			value -= my_r2(sechdrs, me);
5797568cb4eSPaul Mackerras 			if (value + 0x8000 > 0xffff) {
580c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16 relocation (0x%lx)\n",
5817568cb4eSPaul Mackerras 				       me->name, value);
5827568cb4eSPaul Mackerras 				return -ENOEXEC;
5837568cb4eSPaul Mackerras 			}
5847568cb4eSPaul Mackerras 			*((uint16_t *) location)
5857568cb4eSPaul Mackerras 				= (*((uint16_t *) location) & ~0xffff)
5867568cb4eSPaul Mackerras 				| (value & 0xffff);
5877568cb4eSPaul Mackerras 			break;
5887568cb4eSPaul Mackerras 
5891fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_LO:
5901fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
5911fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
5921fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
5931fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xffff)
5941fbe9cf2SAnton Blanchard 				| (value & 0xffff);
5951fbe9cf2SAnton Blanchard 			break;
5961fbe9cf2SAnton Blanchard 
5977568cb4eSPaul Mackerras 		case R_PPC64_TOC16_DS:
598f749edaeSAlan Modra 			/* Subtract TOC pointer */
5997568cb4eSPaul Mackerras 			value -= my_r2(sechdrs, me);
6007568cb4eSPaul Mackerras 			if ((value & 3) != 0 || value + 0x8000 > 0xffff) {
601c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16_DS relocation (0x%lx)\n",
6027568cb4eSPaul Mackerras 				       me->name, value);
6037568cb4eSPaul Mackerras 				return -ENOEXEC;
6047568cb4eSPaul Mackerras 			}
6057568cb4eSPaul Mackerras 			*((uint16_t *) location)
6067568cb4eSPaul Mackerras 				= (*((uint16_t *) location) & ~0xfffc)
6077568cb4eSPaul Mackerras 				| (value & 0xfffc);
6087568cb4eSPaul Mackerras 			break;
6097568cb4eSPaul Mackerras 
6101fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_LO_DS:
6111fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
6121fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6131fbe9cf2SAnton Blanchard 			if ((value & 3) != 0) {
614c7d1f6afSAnton Blanchard 				pr_err("%s: bad TOC16_LO_DS relocation (0x%lx)\n",
6151fbe9cf2SAnton Blanchard 				       me->name, value);
6161fbe9cf2SAnton Blanchard 				return -ENOEXEC;
6171fbe9cf2SAnton Blanchard 			}
6181fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6191fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xfffc)
6201fbe9cf2SAnton Blanchard 				| (value & 0xfffc);
6211fbe9cf2SAnton Blanchard 			break;
6221fbe9cf2SAnton Blanchard 
6231fbe9cf2SAnton Blanchard 		case R_PPC64_TOC16_HA:
6241fbe9cf2SAnton Blanchard 			/* Subtract TOC pointer */
6251fbe9cf2SAnton Blanchard 			value -= my_r2(sechdrs, me);
6261fbe9cf2SAnton Blanchard 			value = ((value + 0x8000) >> 16);
6271fbe9cf2SAnton Blanchard 			*((uint16_t *) location)
6281fbe9cf2SAnton Blanchard 				= (*((uint16_t *) location) & ~0xffff)
6291fbe9cf2SAnton Blanchard 				| (value & 0xffff);
6301fbe9cf2SAnton Blanchard 			break;
6311fbe9cf2SAnton Blanchard 
6327568cb4eSPaul Mackerras 		case R_PPC_REL24:
6337568cb4eSPaul Mackerras 			/* FIXME: Handle weak symbols here --RR */
634a443bf6eSKamalesh Babulal 			if (sym->st_shndx == SHN_UNDEF ||
635a443bf6eSKamalesh Babulal 			    sym->st_shndx == SHN_LIVEPATCH) {
6367568cb4eSPaul Mackerras 				/* External: go via stub */
637bd55e792SNaveen N. Rao 				value = stub_for_addr(sechdrs, value, me,
638bd55e792SNaveen N. Rao 						strtab + sym->st_name);
6397568cb4eSPaul Mackerras 				if (!value)
6407568cb4eSPaul Mackerras 					return -ENOENT;
641250122baSNaveen N. Rao 				if (!restore_r2(strtab + sym->st_name,
642250122baSNaveen N. Rao 							(u32 *)location + 1, me))
6437568cb4eSPaul Mackerras 					return -ENOEXEC;
644008d7a91SRusty Russell 			} else
645008d7a91SRusty Russell 				value += local_entry_offset(sym);
6467568cb4eSPaul Mackerras 
6477568cb4eSPaul Mackerras 			/* Convert value to relative */
6487568cb4eSPaul Mackerras 			value -= (unsigned long)location;
6497568cb4eSPaul Mackerras 			if (value + 0x2000000 > 0x3ffffff || (value & 3) != 0){
650c7d1f6afSAnton Blanchard 				pr_err("%s: REL24 %li out of range!\n",
6517568cb4eSPaul Mackerras 				       me->name, (long int)value);
6527568cb4eSPaul Mackerras 				return -ENOEXEC;
6537568cb4eSPaul Mackerras 			}
6547568cb4eSPaul Mackerras 
6557568cb4eSPaul Mackerras 			/* Only replace bits 2 through 26 */
6568734b41bSRussell Currey 			value = (*(uint32_t *)location & ~0x03fffffc)
6577568cb4eSPaul Mackerras 				| (value & 0x03fffffc);
6588734b41bSRussell Currey 
6598734b41bSRussell Currey 			if (patch_instruction((u32 *)location, ppc_inst(value)))
6608734b41bSRussell Currey 				return -EFAULT;
6618734b41bSRussell Currey 
6627568cb4eSPaul Mackerras 			break;
6637568cb4eSPaul Mackerras 
66421c4ff80SBenjamin Herrenschmidt 		case R_PPC64_REL64:
66521c4ff80SBenjamin Herrenschmidt 			/* 64 bits relative (used by features fixups) */
66621c4ff80SBenjamin Herrenschmidt 			*location = value - (unsigned long)location;
66721c4ff80SBenjamin Herrenschmidt 			break;
66821c4ff80SBenjamin Herrenschmidt 
6699f751b82SMichael Ellerman 		case R_PPC64_REL32:
6709f751b82SMichael Ellerman 			/* 32 bits relative (used by relative exception tables) */
671b851ba02SNicholas Piggin 			/* Convert value to relative */
672b851ba02SNicholas Piggin 			value -= (unsigned long)location;
673b851ba02SNicholas Piggin 			if (value + 0x80000000 > 0xffffffff) {
674b851ba02SNicholas Piggin 				pr_err("%s: REL32 %li out of range!\n",
675b851ba02SNicholas Piggin 				       me->name, (long int)value);
676b851ba02SNicholas Piggin 				return -ENOEXEC;
677b851ba02SNicholas Piggin 			}
678b851ba02SNicholas Piggin 			*(u32 *)location = value;
6799f751b82SMichael Ellerman 			break;
6809f751b82SMichael Ellerman 
681d247da0aSRusty Russell 		case R_PPC64_TOCSAVE:
682d247da0aSRusty Russell 			/*
683d247da0aSRusty Russell 			 * Marker reloc indicates we don't have to save r2.
684d247da0aSRusty Russell 			 * That would only save us one instruction, so ignore
685d247da0aSRusty Russell 			 * it.
686d247da0aSRusty Russell 			 */
687d247da0aSRusty Russell 			break;
688d247da0aSRusty Russell 
689a61674bdSUlrich Weigand 		case R_PPC64_ENTRY:
690a61674bdSUlrich Weigand 			/*
691a61674bdSUlrich Weigand 			 * Optimize ELFv2 large code model entry point if
692a61674bdSUlrich Weigand 			 * the TOC is within 2GB range of current location.
693a61674bdSUlrich Weigand 			 */
694a61674bdSUlrich Weigand 			value = my_r2(sechdrs, me) - (unsigned long)location;
695a61674bdSUlrich Weigand 			if (value + 0x80008000 > 0xffffffff)
696a61674bdSUlrich Weigand 				break;
697a61674bdSUlrich Weigand 			/*
698a61674bdSUlrich Weigand 			 * Check for the large code model prolog sequence:
699a61674bdSUlrich Weigand 		         *	ld r2, ...(r12)
700a61674bdSUlrich Weigand 			 *	add r2, r2, r12
701a61674bdSUlrich Weigand 			 */
70247b04699SChristophe Leroy 			if ((((uint32_t *)location)[0] & ~0xfffc) != PPC_RAW_LD(_R2, _R12, 0))
703a61674bdSUlrich Weigand 				break;
70447b04699SChristophe Leroy 			if (((uint32_t *)location)[1] != PPC_RAW_ADD(_R2, _R2, _R12))
705a61674bdSUlrich Weigand 				break;
706a61674bdSUlrich Weigand 			/*
707a61674bdSUlrich Weigand 			 * If found, replace it with:
708a61674bdSUlrich Weigand 			 *	addis r2, r12, (.TOC.-func)@ha
7092fb0a2c9SMichael Ellerman 			 *	addi  r2,  r2, (.TOC.-func)@l
710a61674bdSUlrich Weigand 			 */
71147b04699SChristophe Leroy 			((uint32_t *)location)[0] = PPC_RAW_ADDIS(_R2, _R12, PPC_HA(value));
71247b04699SChristophe Leroy 			((uint32_t *)location)[1] = PPC_RAW_ADDI(_R2, _R2, PPC_LO(value));
713a61674bdSUlrich Weigand 			break;
714a61674bdSUlrich Weigand 
7150906584aSRusty Russell 		case R_PPC64_REL16_HA:
7160906584aSRusty Russell 			/* Subtract location pointer */
7170906584aSRusty Russell 			value -= (unsigned long)location;
7180906584aSRusty Russell 			value = ((value + 0x8000) >> 16);
7190906584aSRusty Russell 			*((uint16_t *) location)
7200906584aSRusty Russell 				= (*((uint16_t *) location) & ~0xffff)
7210906584aSRusty Russell 				| (value & 0xffff);
7220906584aSRusty Russell 			break;
7230906584aSRusty Russell 
7240906584aSRusty Russell 		case R_PPC64_REL16_LO:
7250906584aSRusty Russell 			/* Subtract location pointer */
7260906584aSRusty Russell 			value -= (unsigned long)location;
7270906584aSRusty Russell 			*((uint16_t *) location)
7280906584aSRusty Russell 				= (*((uint16_t *) location) & ~0xffff)
7290906584aSRusty Russell 				| (value & 0xffff);
7300906584aSRusty Russell 			break;
7310906584aSRusty Russell 
7327568cb4eSPaul Mackerras 		default:
733c7d1f6afSAnton Blanchard 			pr_err("%s: Unknown ADD relocation: %lu\n",
7347568cb4eSPaul Mackerras 			       me->name,
7357568cb4eSPaul Mackerras 			       (unsigned long)ELF64_R_TYPE(rela[i].r_info));
7367568cb4eSPaul Mackerras 			return -ENOEXEC;
7377568cb4eSPaul Mackerras 		}
7387568cb4eSPaul Mackerras 	}
7397568cb4eSPaul Mackerras 
740136cd345SMichael Ellerman 	return 0;
741136cd345SMichael Ellerman }
742136cd345SMichael Ellerman 
743f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
74403b51416SNaveen N. Rao int module_trampoline_target(struct module *mod, unsigned long addr,
74503b51416SNaveen N. Rao 			     unsigned long *target)
74603b51416SNaveen N. Rao {
74703b51416SNaveen N. Rao 	struct ppc64_stub_entry *stub;
74803b51416SNaveen N. Rao 	func_desc_t funcdata;
74903b51416SNaveen N. Rao 	u32 magic;
75003b51416SNaveen N. Rao 
75103b51416SNaveen N. Rao 	if (!within_module_core(addr, mod)) {
75203b51416SNaveen N. Rao 		pr_err("%s: stub %lx not in module %s\n", __func__, addr, mod->name);
75303b51416SNaveen N. Rao 		return -EFAULT;
75403b51416SNaveen N. Rao 	}
75503b51416SNaveen N. Rao 
75603b51416SNaveen N. Rao 	stub = (struct ppc64_stub_entry *)addr;
75703b51416SNaveen N. Rao 
758fe557319SChristoph Hellwig 	if (copy_from_kernel_nofault(&magic, &stub->magic,
759fe557319SChristoph Hellwig 			sizeof(magic))) {
76003b51416SNaveen N. Rao 		pr_err("%s: fault reading magic for stub %lx for %s\n", __func__, addr, mod->name);
76103b51416SNaveen N. Rao 		return -EFAULT;
76203b51416SNaveen N. Rao 	}
76303b51416SNaveen N. Rao 
76403b51416SNaveen N. Rao 	if (magic != STUB_MAGIC) {
76503b51416SNaveen N. Rao 		pr_err("%s: bad magic for stub %lx for %s\n", __func__, addr, mod->name);
76603b51416SNaveen N. Rao 		return -EFAULT;
76703b51416SNaveen N. Rao 	}
76803b51416SNaveen N. Rao 
769fe557319SChristoph Hellwig 	if (copy_from_kernel_nofault(&funcdata, &stub->funcdata,
770fe557319SChristoph Hellwig 			sizeof(funcdata))) {
77103b51416SNaveen N. Rao 		pr_err("%s: fault reading funcdata for stub %lx for %s\n", __func__, addr, mod->name);
77203b51416SNaveen N. Rao                 return -EFAULT;
77303b51416SNaveen N. Rao 	}
77403b51416SNaveen N. Rao 
77503b51416SNaveen N. Rao 	*target = stub_func_addr(funcdata);
77603b51416SNaveen N. Rao 
77703b51416SNaveen N. Rao 	return 0;
77803b51416SNaveen N. Rao }
779336a7b5dSMichael Ellerman 
780136cd345SMichael Ellerman int module_finalize_ftrace(struct module *mod, const Elf_Shdr *sechdrs)
781136cd345SMichael Ellerman {
782bd55e792SNaveen N. Rao 	mod->arch.tramp = stub_for_addr(sechdrs,
783bd55e792SNaveen N. Rao 					(unsigned long)ftrace_caller,
784bd55e792SNaveen N. Rao 					mod,
785bd55e792SNaveen N. Rao 					"ftrace_caller");
786ae30cc05SNaveen N. Rao #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
787bd55e792SNaveen N. Rao 	mod->arch.tramp_regs = stub_for_addr(sechdrs,
788bd55e792SNaveen N. Rao 					(unsigned long)ftrace_regs_caller,
789bd55e792SNaveen N. Rao 					mod,
790bd55e792SNaveen N. Rao 					"ftrace_regs_caller");
791ae30cc05SNaveen N. Rao 	if (!mod->arch.tramp_regs)
792ae30cc05SNaveen N. Rao 		return -ENOENT;
793ae30cc05SNaveen N. Rao #endif
794136cd345SMichael Ellerman 
795136cd345SMichael Ellerman 	if (!mod->arch.tramp)
796136cd345SMichael Ellerman 		return -ENOENT;
797f48cb8b4SSteven Rostedt 
7987568cb4eSPaul Mackerras 	return 0;
7997568cb4eSPaul Mackerras }
800136cd345SMichael Ellerman #endif
801