17568cb4eSPaul Mackerras /* Kernel module help for PPC64. 27568cb4eSPaul Mackerras Copyright (C) 2001, 2003 Rusty Russell IBM Corporation. 37568cb4eSPaul Mackerras 47568cb4eSPaul Mackerras This program is free software; you can redistribute it and/or modify 57568cb4eSPaul Mackerras it under the terms of the GNU General Public License as published by 67568cb4eSPaul Mackerras the Free Software Foundation; either version 2 of the License, or 77568cb4eSPaul Mackerras (at your option) any later version. 87568cb4eSPaul Mackerras 97568cb4eSPaul Mackerras This program is distributed in the hope that it will be useful, 107568cb4eSPaul Mackerras but WITHOUT ANY WARRANTY; without even the implied warranty of 117568cb4eSPaul Mackerras MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 127568cb4eSPaul Mackerras GNU General Public License for more details. 137568cb4eSPaul Mackerras 147568cb4eSPaul Mackerras You should have received a copy of the GNU General Public License 157568cb4eSPaul Mackerras along with this program; if not, write to the Free Software 167568cb4eSPaul Mackerras Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 177568cb4eSPaul Mackerras */ 18c7d1f6afSAnton Blanchard 19c7d1f6afSAnton Blanchard #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 20c7d1f6afSAnton Blanchard 217568cb4eSPaul Mackerras #include <linux/module.h> 227568cb4eSPaul Mackerras #include <linux/elf.h> 237568cb4eSPaul Mackerras #include <linux/moduleloader.h> 247568cb4eSPaul Mackerras #include <linux/err.h> 257568cb4eSPaul Mackerras #include <linux/vmalloc.h> 26f48cb8b4SSteven Rostedt #include <linux/ftrace.h> 2773c9ceabSJeremy Fitzhardinge #include <linux/bug.h> 2883775b85SAnton Blanchard #include <linux/uaccess.h> 297568cb4eSPaul Mackerras #include <asm/module.h> 3021c4ff80SBenjamin Herrenschmidt #include <asm/firmware.h> 31b7bcda63SMichael Ellerman #include <asm/code-patching.h> 32eda09fbdSEmil Medve #include <linux/sort.h> 33b88c4767SRobert Jennings #include <asm/setup.h> 34336a7b5dSMichael Ellerman #include <asm/sections.h> 357568cb4eSPaul Mackerras 367568cb4eSPaul Mackerras /* FIXME: We don't do .init separately. To do this, we'd need to have 377568cb4eSPaul Mackerras a separate r2 value in the init and core section, and stub between 387568cb4eSPaul Mackerras them, too. 397568cb4eSPaul Mackerras 407568cb4eSPaul Mackerras Using a magic allocator which places modules within 32MB solves 417568cb4eSPaul Mackerras this, and makes other things simpler. Anton? 427568cb4eSPaul Mackerras --RR. */ 437568cb4eSPaul Mackerras 44f55d9665SMichael Ellerman #ifdef PPC64_ELF_ABI_v2 45008d7a91SRusty Russell 46008d7a91SRusty Russell /* An address is simply the address of the function. */ 47008d7a91SRusty Russell typedef unsigned long func_desc_t; 48008d7a91SRusty Russell 49008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr) 50008d7a91SRusty Russell { 51008d7a91SRusty Russell return addr; 52008d7a91SRusty Russell } 53008d7a91SRusty Russell static unsigned long func_addr(unsigned long addr) 54008d7a91SRusty Russell { 55008d7a91SRusty Russell return addr; 56008d7a91SRusty Russell } 57008d7a91SRusty Russell static unsigned long stub_func_addr(func_desc_t func) 58008d7a91SRusty Russell { 59008d7a91SRusty Russell return func; 60008d7a91SRusty Russell } 61008d7a91SRusty Russell 62008d7a91SRusty Russell /* PowerPC64 specific values for the Elf64_Sym st_other field. */ 63008d7a91SRusty Russell #define STO_PPC64_LOCAL_BIT 5 64008d7a91SRusty Russell #define STO_PPC64_LOCAL_MASK (7 << STO_PPC64_LOCAL_BIT) 65008d7a91SRusty Russell #define PPC64_LOCAL_ENTRY_OFFSET(other) \ 66008d7a91SRusty Russell (((1 << (((other) & STO_PPC64_LOCAL_MASK) >> STO_PPC64_LOCAL_BIT)) >> 2) << 2) 67008d7a91SRusty Russell 68008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym) 69008d7a91SRusty Russell { 70008d7a91SRusty Russell /* sym->st_other indicates offset to local entry point 71008d7a91SRusty Russell * (otherwise it will assume r12 is the address of the start 72008d7a91SRusty Russell * of function and try to derive r2 from it). */ 73008d7a91SRusty Russell return PPC64_LOCAL_ENTRY_OFFSET(sym->st_other); 74008d7a91SRusty Russell } 75d2fae548SRusty Russell #else 76008d7a91SRusty Russell 77008d7a91SRusty Russell /* An address is address of the OPD entry, which contains address of fn. */ 78008d7a91SRusty Russell typedef struct ppc64_opd_entry func_desc_t; 79008d7a91SRusty Russell 80008d7a91SRusty Russell static func_desc_t func_desc(unsigned long addr) 81008d7a91SRusty Russell { 82008d7a91SRusty Russell return *(struct ppc64_opd_entry *)addr; 83008d7a91SRusty Russell } 84008d7a91SRusty Russell static unsigned long func_addr(unsigned long addr) 85008d7a91SRusty Russell { 86008d7a91SRusty Russell return func_desc(addr).funcaddr; 87008d7a91SRusty Russell } 88008d7a91SRusty Russell static unsigned long stub_func_addr(func_desc_t func) 89008d7a91SRusty Russell { 90008d7a91SRusty Russell return func.funcaddr; 91008d7a91SRusty Russell } 92008d7a91SRusty Russell static unsigned int local_entry_offset(const Elf64_Sym *sym) 93008d7a91SRusty Russell { 94008d7a91SRusty Russell return 0; 95008d7a91SRusty Russell } 96d2fae548SRusty Russell #endif 97d2fae548SRusty Russell 98f17c4e01SMichael Ellerman #define STUB_MAGIC 0x73747562 /* stub */ 99f17c4e01SMichael Ellerman 1007568cb4eSPaul Mackerras /* Like PPC32, we need little trampolines to do > 24-bit jumps (into 1017568cb4eSPaul Mackerras the kernel itself). But on PPC64, these need to be used for every 1027568cb4eSPaul Mackerras jump, actually, to reset r2 (TOC+0x8000). */ 1037568cb4eSPaul Mackerras struct ppc64_stub_entry 1047568cb4eSPaul Mackerras { 10583775b85SAnton Blanchard /* 28 byte jump instruction sequence (7 instructions). We only 10683775b85SAnton Blanchard * need 6 instructions on ABIv2 but we always allocate 7 so 10783775b85SAnton Blanchard * so we don't have to modify the trampoline load instruction. */ 1080e60e46eSRusty Russell u32 jump[7]; 109f17c4e01SMichael Ellerman /* Used by ftrace to identify stubs */ 110f17c4e01SMichael Ellerman u32 magic; 1117568cb4eSPaul Mackerras /* Data for the above code */ 112008d7a91SRusty Russell func_desc_t funcdata; 1137568cb4eSPaul Mackerras }; 1147568cb4eSPaul Mackerras 1155c729a11SRusty Russell /* 1165c729a11SRusty Russell * PPC64 uses 24 bit jumps, but we need to jump into other modules or 1175c729a11SRusty Russell * the kernel which may be further. So we jump to a stub. 1185c729a11SRusty Russell * 1195c729a11SRusty Russell * For ELFv1 we need to use this to set up the new r2 value (aka TOC 1205c729a11SRusty Russell * pointer). For ELFv2 it's the callee's responsibility to set up the 1215c729a11SRusty Russell * new r2, but for both we need to save the old r2. 1225c729a11SRusty Russell * 1235c729a11SRusty Russell * We could simply patch the new r2 value and function pointer into 1245c729a11SRusty Russell * the stub, but it's significantly shorter to put these values at the 1255c729a11SRusty Russell * end of the stub code, and patch the stub address (32-bits relative 1265c729a11SRusty Russell * to the TOC ptr, r2) into the stub. 1275c729a11SRusty Russell */ 12883775b85SAnton Blanchard 12983775b85SAnton Blanchard static u32 ppc64_stub_insns[] = { 130b1ce369eSRusty Russell 0x3d620000, /* addis r11,r2, <high> */ 131b1ce369eSRusty Russell 0x396b0000, /* addi r11,r11, <low> */ 132fed8393eSEugene Surovegin /* Save current r2 value in magic place on the stack. */ 133d2fae548SRusty Russell 0xf8410000|R2_STACK_OFFSET, /* std r2,R2_STACK_OFFSET(r1) */ 134b1ce369eSRusty Russell 0xe98b0020, /* ld r12,32(r11) */ 135f55d9665SMichael Ellerman #ifdef PPC64_ELF_ABI_v1 1365c729a11SRusty Russell /* Set up new r2 from function descriptor */ 137872aa779SRusty Russell 0xe84b0028, /* ld r2,40(r11) */ 1385c729a11SRusty Russell #endif 139b1ce369eSRusty Russell 0x7d8903a6, /* mtctr r12 */ 1400e60e46eSRusty Russell 0x4e800420 /* bctr */ 14183775b85SAnton Blanchard }; 14283775b85SAnton Blanchard 14383775b85SAnton Blanchard #ifdef CONFIG_DYNAMIC_FTRACE 144f17c4e01SMichael Ellerman int module_trampoline_target(struct module *mod, unsigned long addr, 145dd9fa162SAnton Blanchard unsigned long *target) 146dd9fa162SAnton Blanchard { 147f17c4e01SMichael Ellerman struct ppc64_stub_entry *stub; 148f17c4e01SMichael Ellerman func_desc_t funcdata; 149f17c4e01SMichael Ellerman u32 magic; 150dd9fa162SAnton Blanchard 151f17c4e01SMichael Ellerman if (!within_module_core(addr, mod)) { 152f17c4e01SMichael Ellerman pr_err("%s: stub %lx not in module %s\n", __func__, addr, mod->name); 153dd9fa162SAnton Blanchard return -EFAULT; 154f17c4e01SMichael Ellerman } 155dd9fa162SAnton Blanchard 156f17c4e01SMichael Ellerman stub = (struct ppc64_stub_entry *)addr; 157dd9fa162SAnton Blanchard 158f17c4e01SMichael Ellerman if (probe_kernel_read(&magic, &stub->magic, sizeof(magic))) { 159f17c4e01SMichael Ellerman pr_err("%s: fault reading magic for stub %lx for %s\n", __func__, addr, mod->name); 160dd9fa162SAnton Blanchard return -EFAULT; 161f17c4e01SMichael Ellerman } 162f17c4e01SMichael Ellerman 163f17c4e01SMichael Ellerman if (magic != STUB_MAGIC) { 164f17c4e01SMichael Ellerman pr_err("%s: bad magic for stub %lx for %s\n", __func__, addr, mod->name); 165f17c4e01SMichael Ellerman return -EFAULT; 166f17c4e01SMichael Ellerman } 167f17c4e01SMichael Ellerman 168f17c4e01SMichael Ellerman if (probe_kernel_read(&funcdata, &stub->funcdata, sizeof(funcdata))) { 169f17c4e01SMichael Ellerman pr_err("%s: fault reading funcdata for stub %lx for %s\n", __func__, addr, mod->name); 170f17c4e01SMichael Ellerman return -EFAULT; 171f17c4e01SMichael Ellerman } 172f17c4e01SMichael Ellerman 173f17c4e01SMichael Ellerman *target = stub_func_addr(funcdata); 174dd9fa162SAnton Blanchard 175dd9fa162SAnton Blanchard return 0; 176dd9fa162SAnton Blanchard } 17783775b85SAnton Blanchard #endif 1787568cb4eSPaul Mackerras 1797568cb4eSPaul Mackerras /* Count how many different 24-bit relocations (different symbol, 1807568cb4eSPaul Mackerras different addend) */ 1817568cb4eSPaul Mackerras static unsigned int count_relocs(const Elf64_Rela *rela, unsigned int num) 1827568cb4eSPaul Mackerras { 183eda09fbdSEmil Medve unsigned int i, r_info, r_addend, _count_relocs; 1847568cb4eSPaul Mackerras 1857568cb4eSPaul Mackerras /* FIXME: Only count external ones --RR */ 186eda09fbdSEmil Medve _count_relocs = 0; 187eda09fbdSEmil Medve r_info = 0; 188eda09fbdSEmil Medve r_addend = 0; 189eda09fbdSEmil Medve for (i = 0; i < num; i++) 1907568cb4eSPaul Mackerras /* Only count 24-bit relocs, others don't need stubs */ 191eda09fbdSEmil Medve if (ELF64_R_TYPE(rela[i].r_info) == R_PPC_REL24 && 192eda09fbdSEmil Medve (r_info != ELF64_R_SYM(rela[i].r_info) || 193eda09fbdSEmil Medve r_addend != rela[i].r_addend)) { 194eda09fbdSEmil Medve _count_relocs++; 195eda09fbdSEmil Medve r_info = ELF64_R_SYM(rela[i].r_info); 196eda09fbdSEmil Medve r_addend = rela[i].r_addend; 1977568cb4eSPaul Mackerras } 198eda09fbdSEmil Medve 199eda09fbdSEmil Medve return _count_relocs; 2007568cb4eSPaul Mackerras } 2017568cb4eSPaul Mackerras 202eda09fbdSEmil Medve static int relacmp(const void *_x, const void *_y) 203eda09fbdSEmil Medve { 204eda09fbdSEmil Medve const Elf64_Rela *x, *y; 205eda09fbdSEmil Medve 206eda09fbdSEmil Medve y = (Elf64_Rela *)_x; 207eda09fbdSEmil Medve x = (Elf64_Rela *)_y; 208eda09fbdSEmil Medve 209eda09fbdSEmil Medve /* Compare the entire r_info (as opposed to ELF64_R_SYM(r_info) only) to 210eda09fbdSEmil Medve * make the comparison cheaper/faster. It won't affect the sorting or 211eda09fbdSEmil Medve * the counting algorithms' performance 212eda09fbdSEmil Medve */ 213eda09fbdSEmil Medve if (x->r_info < y->r_info) 214eda09fbdSEmil Medve return -1; 215eda09fbdSEmil Medve else if (x->r_info > y->r_info) 216eda09fbdSEmil Medve return 1; 217eda09fbdSEmil Medve else if (x->r_addend < y->r_addend) 218eda09fbdSEmil Medve return -1; 219eda09fbdSEmil Medve else if (x->r_addend > y->r_addend) 220eda09fbdSEmil Medve return 1; 221eda09fbdSEmil Medve else 222eda09fbdSEmil Medve return 0; 223eda09fbdSEmil Medve } 224eda09fbdSEmil Medve 225eda09fbdSEmil Medve static void relaswap(void *_x, void *_y, int size) 226eda09fbdSEmil Medve { 227eda09fbdSEmil Medve uint64_t *x, *y, tmp; 228eda09fbdSEmil Medve int i; 229eda09fbdSEmil Medve 230eda09fbdSEmil Medve y = (uint64_t *)_x; 231eda09fbdSEmil Medve x = (uint64_t *)_y; 232eda09fbdSEmil Medve 233eda09fbdSEmil Medve for (i = 0; i < sizeof(Elf64_Rela) / sizeof(uint64_t); i++) { 234eda09fbdSEmil Medve tmp = x[i]; 235eda09fbdSEmil Medve x[i] = y[i]; 236eda09fbdSEmil Medve y[i] = tmp; 237eda09fbdSEmil Medve } 238eda09fbdSEmil Medve } 239eda09fbdSEmil Medve 2407568cb4eSPaul Mackerras /* Get size of potential trampolines required. */ 2417568cb4eSPaul Mackerras static unsigned long get_stubs_size(const Elf64_Ehdr *hdr, 2427568cb4eSPaul Mackerras const Elf64_Shdr *sechdrs) 2437568cb4eSPaul Mackerras { 2447568cb4eSPaul Mackerras /* One extra reloc so it's always 0-funcaddr terminated */ 2457568cb4eSPaul Mackerras unsigned long relocs = 1; 2467568cb4eSPaul Mackerras unsigned i; 2477568cb4eSPaul Mackerras 2487568cb4eSPaul Mackerras /* Every relocated section... */ 2497568cb4eSPaul Mackerras for (i = 1; i < hdr->e_shnum; i++) { 2507568cb4eSPaul Mackerras if (sechdrs[i].sh_type == SHT_RELA) { 251c7d1f6afSAnton Blanchard pr_debug("Found relocations in section %u\n", i); 252c7d1f6afSAnton Blanchard pr_debug("Ptr: %p. Number: %Lu\n", 2537568cb4eSPaul Mackerras (void *)sechdrs[i].sh_addr, 2547568cb4eSPaul Mackerras sechdrs[i].sh_size / sizeof(Elf64_Rela)); 255eda09fbdSEmil Medve 256eda09fbdSEmil Medve /* Sort the relocation information based on a symbol and 257eda09fbdSEmil Medve * addend key. This is a stable O(n*log n) complexity 258eda09fbdSEmil Medve * alogrithm but it will reduce the complexity of 259eda09fbdSEmil Medve * count_relocs() to linear complexity O(n) 260eda09fbdSEmil Medve */ 261eda09fbdSEmil Medve sort((void *)sechdrs[i].sh_addr, 262eda09fbdSEmil Medve sechdrs[i].sh_size / sizeof(Elf64_Rela), 263eda09fbdSEmil Medve sizeof(Elf64_Rela), relacmp, relaswap); 264eda09fbdSEmil Medve 2657568cb4eSPaul Mackerras relocs += count_relocs((void *)sechdrs[i].sh_addr, 2667568cb4eSPaul Mackerras sechdrs[i].sh_size 2677568cb4eSPaul Mackerras / sizeof(Elf64_Rela)); 2687568cb4eSPaul Mackerras } 2697568cb4eSPaul Mackerras } 2707568cb4eSPaul Mackerras 271f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE 272f48cb8b4SSteven Rostedt /* make the trampoline to the ftrace_caller */ 273f48cb8b4SSteven Rostedt relocs++; 274f48cb8b4SSteven Rostedt #endif 275f48cb8b4SSteven Rostedt 276c7d1f6afSAnton Blanchard pr_debug("Looks like a total of %lu stubs, max\n", relocs); 2777568cb4eSPaul Mackerras return relocs * sizeof(struct ppc64_stub_entry); 2787568cb4eSPaul Mackerras } 2797568cb4eSPaul Mackerras 2805b12c5c6SRusty Russell /* Still needed for ELFv2, for .TOC. */ 2817568cb4eSPaul Mackerras static void dedotify_versions(struct modversion_info *vers, 2827568cb4eSPaul Mackerras unsigned long size) 2837568cb4eSPaul Mackerras { 2847568cb4eSPaul Mackerras struct modversion_info *end; 2857568cb4eSPaul Mackerras 2867568cb4eSPaul Mackerras for (end = (void *)vers + size; vers < end; vers++) 287c2cbcf53SLaurent Dufour if (vers->name[0] == '.') { 2887568cb4eSPaul Mackerras memmove(vers->name, vers->name+1, strlen(vers->name)); 289c2cbcf53SLaurent Dufour #ifdef ARCH_RELOCATES_KCRCTAB 290c2cbcf53SLaurent Dufour /* The TOC symbol has no CRC computed. To avoid CRC 291c2cbcf53SLaurent Dufour * check failing, we must force it to the expected 292c2cbcf53SLaurent Dufour * value (see CRC check in module.c). 293c2cbcf53SLaurent Dufour */ 294c2cbcf53SLaurent Dufour if (!strcmp(vers->name, "TOC.")) 295c2cbcf53SLaurent Dufour vers->crc = -(unsigned long)reloc_start; 296c2cbcf53SLaurent Dufour #endif 297c2cbcf53SLaurent Dufour } 2987568cb4eSPaul Mackerras } 2997568cb4eSPaul Mackerras 300c153693dSAlan Modra /* 301c153693dSAlan Modra * Undefined symbols which refer to .funcname, hack to funcname. Make .TOC. 302c153693dSAlan Modra * seem to be defined (value set later). 303c153693dSAlan Modra */ 3047568cb4eSPaul Mackerras static void dedotify(Elf64_Sym *syms, unsigned int numsyms, char *strtab) 3057568cb4eSPaul Mackerras { 3067568cb4eSPaul Mackerras unsigned int i; 3077568cb4eSPaul Mackerras 3087568cb4eSPaul Mackerras for (i = 1; i < numsyms; i++) { 3097568cb4eSPaul Mackerras if (syms[i].st_shndx == SHN_UNDEF) { 3107568cb4eSPaul Mackerras char *name = strtab + syms[i].st_name; 311c153693dSAlan Modra if (name[0] == '.') { 312c153693dSAlan Modra if (strcmp(name+1, "TOC.") == 0) 313c153693dSAlan Modra syms[i].st_shndx = SHN_ABS; 314f15838e9SAndreas Schwab syms[i].st_name++; 3157568cb4eSPaul Mackerras } 3167568cb4eSPaul Mackerras } 3177568cb4eSPaul Mackerras } 318c153693dSAlan Modra } 3197568cb4eSPaul Mackerras 3204edebbeaSRusty Russell static Elf64_Sym *find_dot_toc(Elf64_Shdr *sechdrs, 3214edebbeaSRusty Russell const char *strtab, 3224edebbeaSRusty Russell unsigned int symindex) 3234edebbeaSRusty Russell { 3244edebbeaSRusty Russell unsigned int i, numsyms; 3254edebbeaSRusty Russell Elf64_Sym *syms; 3264edebbeaSRusty Russell 3274edebbeaSRusty Russell syms = (Elf64_Sym *)sechdrs[symindex].sh_addr; 3284edebbeaSRusty Russell numsyms = sechdrs[symindex].sh_size / sizeof(Elf64_Sym); 3294edebbeaSRusty Russell 3304edebbeaSRusty Russell for (i = 1; i < numsyms; i++) { 331c153693dSAlan Modra if (syms[i].st_shndx == SHN_ABS 332008d7a91SRusty Russell && strcmp(strtab + syms[i].st_name, "TOC.") == 0) 3334edebbeaSRusty Russell return &syms[i]; 3344edebbeaSRusty Russell } 3354edebbeaSRusty Russell return NULL; 3364edebbeaSRusty Russell } 3374edebbeaSRusty Russell 3387568cb4eSPaul Mackerras int module_frob_arch_sections(Elf64_Ehdr *hdr, 3397568cb4eSPaul Mackerras Elf64_Shdr *sechdrs, 3407568cb4eSPaul Mackerras char *secstrings, 3417568cb4eSPaul Mackerras struct module *me) 3427568cb4eSPaul Mackerras { 3437568cb4eSPaul Mackerras unsigned int i; 3447568cb4eSPaul Mackerras 3457568cb4eSPaul Mackerras /* Find .toc and .stubs sections, symtab and strtab */ 3467568cb4eSPaul Mackerras for (i = 1; i < hdr->e_shnum; i++) { 3477568cb4eSPaul Mackerras char *p; 3487568cb4eSPaul Mackerras if (strcmp(secstrings + sechdrs[i].sh_name, ".stubs") == 0) 3497568cb4eSPaul Mackerras me->arch.stubs_section = i; 3507568cb4eSPaul Mackerras else if (strcmp(secstrings + sechdrs[i].sh_name, ".toc") == 0) 3517568cb4eSPaul Mackerras me->arch.toc_section = i; 3527568cb4eSPaul Mackerras else if (strcmp(secstrings+sechdrs[i].sh_name,"__versions")==0) 3537568cb4eSPaul Mackerras dedotify_versions((void *)hdr + sechdrs[i].sh_offset, 3547568cb4eSPaul Mackerras sechdrs[i].sh_size); 3557568cb4eSPaul Mackerras 3567568cb4eSPaul Mackerras /* We don't handle .init for the moment: rename to _init */ 3577568cb4eSPaul Mackerras while ((p = strstr(secstrings + sechdrs[i].sh_name, ".init"))) 3587568cb4eSPaul Mackerras p[0] = '_'; 3597568cb4eSPaul Mackerras 3607568cb4eSPaul Mackerras if (sechdrs[i].sh_type == SHT_SYMTAB) 3617568cb4eSPaul Mackerras dedotify((void *)hdr + sechdrs[i].sh_offset, 3627568cb4eSPaul Mackerras sechdrs[i].sh_size / sizeof(Elf64_Sym), 3637568cb4eSPaul Mackerras (void *)hdr 3647568cb4eSPaul Mackerras + sechdrs[sechdrs[i].sh_link].sh_offset); 3657568cb4eSPaul Mackerras } 366f749edaeSAlan Modra 367f749edaeSAlan Modra if (!me->arch.stubs_section) { 368c7d1f6afSAnton Blanchard pr_err("%s: doesn't contain .stubs.\n", me->name); 3697568cb4eSPaul Mackerras return -ENOEXEC; 3707568cb4eSPaul Mackerras } 3717568cb4eSPaul Mackerras 372f749edaeSAlan Modra /* If we don't have a .toc, just use .stubs. We need to set r2 373f749edaeSAlan Modra to some reasonable value in case the module calls out to 374f749edaeSAlan Modra other functions via a stub, or if a function pointer escapes 375f749edaeSAlan Modra the module by some means. */ 376f749edaeSAlan Modra if (!me->arch.toc_section) 377f749edaeSAlan Modra me->arch.toc_section = me->arch.stubs_section; 378f749edaeSAlan Modra 3797568cb4eSPaul Mackerras /* Override the stubs size */ 3807568cb4eSPaul Mackerras sechdrs[me->arch.stubs_section].sh_size = get_stubs_size(hdr, sechdrs); 3817568cb4eSPaul Mackerras return 0; 3827568cb4eSPaul Mackerras } 3837568cb4eSPaul Mackerras 3847568cb4eSPaul Mackerras /* r2 is the TOC pointer: it actually points 0x8000 into the TOC (this 3857568cb4eSPaul Mackerras gives the value maximum span in an instruction which uses a signed 3867568cb4eSPaul Mackerras offset) */ 387136cd345SMichael Ellerman static inline unsigned long my_r2(const Elf64_Shdr *sechdrs, struct module *me) 3887568cb4eSPaul Mackerras { 3897568cb4eSPaul Mackerras return sechdrs[me->arch.toc_section].sh_addr + 0x8000; 3907568cb4eSPaul Mackerras } 3917568cb4eSPaul Mackerras 3927568cb4eSPaul Mackerras /* Both low and high 16 bits are added as SIGNED additions, so if low 3937568cb4eSPaul Mackerras 16 bits has high bit set, high 16 bits must be adjusted. These 3947568cb4eSPaul Mackerras macros do that (stolen from binutils). */ 3957568cb4eSPaul Mackerras #define PPC_LO(v) ((v) & 0xffff) 3967568cb4eSPaul Mackerras #define PPC_HI(v) (((v) >> 16) & 0xffff) 3977568cb4eSPaul Mackerras #define PPC_HA(v) PPC_HI ((v) + 0x8000) 3987568cb4eSPaul Mackerras 3997568cb4eSPaul Mackerras /* Patch stub to reference function and correct r2 value. */ 400136cd345SMichael Ellerman static inline int create_stub(const Elf64_Shdr *sechdrs, 4017568cb4eSPaul Mackerras struct ppc64_stub_entry *entry, 402008d7a91SRusty Russell unsigned long addr, 4037568cb4eSPaul Mackerras struct module *me) 4047568cb4eSPaul Mackerras { 4057568cb4eSPaul Mackerras long reladdr; 4067568cb4eSPaul Mackerras 40783775b85SAnton Blanchard memcpy(entry->jump, ppc64_stub_insns, sizeof(ppc64_stub_insns)); 4087568cb4eSPaul Mackerras 4097568cb4eSPaul Mackerras /* Stub uses address relative to r2. */ 4107568cb4eSPaul Mackerras reladdr = (unsigned long)entry - my_r2(sechdrs, me); 4117568cb4eSPaul Mackerras if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) { 412c7d1f6afSAnton Blanchard pr_err("%s: Address %p of stub out of range of %p.\n", 4137568cb4eSPaul Mackerras me->name, (void *)reladdr, (void *)my_r2); 4147568cb4eSPaul Mackerras return 0; 4157568cb4eSPaul Mackerras } 416c7d1f6afSAnton Blanchard pr_debug("Stub %p get data from reladdr %li\n", entry, reladdr); 4177568cb4eSPaul Mackerras 4180e60e46eSRusty Russell entry->jump[0] |= PPC_HA(reladdr); 4190e60e46eSRusty Russell entry->jump[1] |= PPC_LO(reladdr); 420008d7a91SRusty Russell entry->funcdata = func_desc(addr); 421f17c4e01SMichael Ellerman entry->magic = STUB_MAGIC; 422f17c4e01SMichael Ellerman 4237568cb4eSPaul Mackerras return 1; 4247568cb4eSPaul Mackerras } 4257568cb4eSPaul Mackerras 426008d7a91SRusty Russell /* Create stub to jump to function described in this OPD/ptr: we need the 4277568cb4eSPaul Mackerras stub to set up the TOC ptr (r2) for the function. */ 428136cd345SMichael Ellerman static unsigned long stub_for_addr(const Elf64_Shdr *sechdrs, 429008d7a91SRusty Russell unsigned long addr, 4307568cb4eSPaul Mackerras struct module *me) 4317568cb4eSPaul Mackerras { 4327568cb4eSPaul Mackerras struct ppc64_stub_entry *stubs; 4337568cb4eSPaul Mackerras unsigned int i, num_stubs; 4347568cb4eSPaul Mackerras 4357568cb4eSPaul Mackerras num_stubs = sechdrs[me->arch.stubs_section].sh_size / sizeof(*stubs); 4367568cb4eSPaul Mackerras 4377568cb4eSPaul Mackerras /* Find this stub, or if that fails, the next avail. entry */ 4387568cb4eSPaul Mackerras stubs = (void *)sechdrs[me->arch.stubs_section].sh_addr; 439008d7a91SRusty Russell for (i = 0; stub_func_addr(stubs[i].funcdata); i++) { 4407568cb4eSPaul Mackerras BUG_ON(i >= num_stubs); 4417568cb4eSPaul Mackerras 442008d7a91SRusty Russell if (stub_func_addr(stubs[i].funcdata) == func_addr(addr)) 4437568cb4eSPaul Mackerras return (unsigned long)&stubs[i]; 4447568cb4eSPaul Mackerras } 4457568cb4eSPaul Mackerras 446008d7a91SRusty Russell if (!create_stub(sechdrs, &stubs[i], addr, me)) 4477568cb4eSPaul Mackerras return 0; 4487568cb4eSPaul Mackerras 4497568cb4eSPaul Mackerras return (unsigned long)&stubs[i]; 4507568cb4eSPaul Mackerras } 4517568cb4eSPaul Mackerras 45215308664STorsten Duwe #ifdef CC_USING_MPROFILE_KERNEL 45315308664STorsten Duwe static bool is_early_mcount_callsite(u32 *instruction) 45415308664STorsten Duwe { 45515308664STorsten Duwe /* 45615308664STorsten Duwe * Check if this is one of the -mprofile-kernel sequences. 45715308664STorsten Duwe */ 45815308664STorsten Duwe if (instruction[-1] == PPC_INST_STD_LR && 45915308664STorsten Duwe instruction[-2] == PPC_INST_MFLR) 46015308664STorsten Duwe return true; 46115308664STorsten Duwe 46215308664STorsten Duwe if (instruction[-1] == PPC_INST_MFLR) 46315308664STorsten Duwe return true; 46415308664STorsten Duwe 46515308664STorsten Duwe return false; 46615308664STorsten Duwe } 46715308664STorsten Duwe 46815308664STorsten Duwe /* 46915308664STorsten Duwe * In case of _mcount calls, do not save the current callee's TOC (in r2) into 47015308664STorsten Duwe * the original caller's stack frame. If we did we would clobber the saved TOC 47115308664STorsten Duwe * value of the original caller. 47215308664STorsten Duwe */ 47315308664STorsten Duwe static void squash_toc_save_inst(const char *name, unsigned long addr) 47415308664STorsten Duwe { 47515308664STorsten Duwe struct ppc64_stub_entry *stub = (struct ppc64_stub_entry *)addr; 47615308664STorsten Duwe 47715308664STorsten Duwe /* Only for calls to _mcount */ 47815308664STorsten Duwe if (strcmp("_mcount", name) != 0) 47915308664STorsten Duwe return; 48015308664STorsten Duwe 48115308664STorsten Duwe stub->jump[2] = PPC_INST_NOP; 48215308664STorsten Duwe } 48315308664STorsten Duwe #else 48415308664STorsten Duwe static void squash_toc_save_inst(const char *name, unsigned long addr) { } 48515308664STorsten Duwe 48615308664STorsten Duwe /* without -mprofile-kernel, mcount calls are never early */ 48715308664STorsten Duwe static bool is_early_mcount_callsite(u32 *instruction) 48815308664STorsten Duwe { 48915308664STorsten Duwe return false; 49015308664STorsten Duwe } 49115308664STorsten Duwe #endif 49215308664STorsten Duwe 4937568cb4eSPaul Mackerras /* We expect a noop next: if it is, replace it with instruction to 4947568cb4eSPaul Mackerras restore r2. */ 4957568cb4eSPaul Mackerras static int restore_r2(u32 *instruction, struct module *me) 4967568cb4eSPaul Mackerras { 49715308664STorsten Duwe if (is_early_mcount_callsite(instruction - 1)) 49815308664STorsten Duwe return 1; 49931278b17SMichael Ellerman 50031278b17SMichael Ellerman if (*instruction != PPC_INST_NOP) { 501c7d1f6afSAnton Blanchard pr_err("%s: Expect noop after relocate, got %08x\n", 5027568cb4eSPaul Mackerras me->name, *instruction); 5037568cb4eSPaul Mackerras return 0; 5047568cb4eSPaul Mackerras } 505d2fae548SRusty Russell /* ld r2,R2_STACK_OFFSET(r1) */ 50615308664STorsten Duwe *instruction = PPC_INST_LD_TOC; 5077568cb4eSPaul Mackerras return 1; 5087568cb4eSPaul Mackerras } 5097568cb4eSPaul Mackerras 5107568cb4eSPaul Mackerras int apply_relocate_add(Elf64_Shdr *sechdrs, 5117568cb4eSPaul Mackerras const char *strtab, 5127568cb4eSPaul Mackerras unsigned int symindex, 5137568cb4eSPaul Mackerras unsigned int relsec, 5147568cb4eSPaul Mackerras struct module *me) 5157568cb4eSPaul Mackerras { 5167568cb4eSPaul Mackerras unsigned int i; 5177568cb4eSPaul Mackerras Elf64_Rela *rela = (void *)sechdrs[relsec].sh_addr; 5187568cb4eSPaul Mackerras Elf64_Sym *sym; 5197568cb4eSPaul Mackerras unsigned long *location; 5207568cb4eSPaul Mackerras unsigned long value; 5217568cb4eSPaul Mackerras 522c7d1f6afSAnton Blanchard pr_debug("Applying ADD relocate section %u to %u\n", relsec, 5237568cb4eSPaul Mackerras sechdrs[relsec].sh_info); 5244edebbeaSRusty Russell 5254edebbeaSRusty Russell /* First time we're called, we can fix up .TOC. */ 5264edebbeaSRusty Russell if (!me->arch.toc_fixed) { 5274edebbeaSRusty Russell sym = find_dot_toc(sechdrs, strtab, symindex); 5284edebbeaSRusty Russell /* It's theoretically possible that a module doesn't want a 5294edebbeaSRusty Russell * .TOC. so don't fail it just for that. */ 5304edebbeaSRusty Russell if (sym) 5314edebbeaSRusty Russell sym->st_value = my_r2(sechdrs, me); 5324edebbeaSRusty Russell me->arch.toc_fixed = true; 5334edebbeaSRusty Russell } 5344edebbeaSRusty Russell 5357568cb4eSPaul Mackerras for (i = 0; i < sechdrs[relsec].sh_size / sizeof(*rela); i++) { 5367568cb4eSPaul Mackerras /* This is where to make the change */ 5377568cb4eSPaul Mackerras location = (void *)sechdrs[sechdrs[relsec].sh_info].sh_addr 5387568cb4eSPaul Mackerras + rela[i].r_offset; 5397568cb4eSPaul Mackerras /* This is the symbol it is referring to */ 5407568cb4eSPaul Mackerras sym = (Elf64_Sym *)sechdrs[symindex].sh_addr 5417568cb4eSPaul Mackerras + ELF64_R_SYM(rela[i].r_info); 5427568cb4eSPaul Mackerras 543c7d1f6afSAnton Blanchard pr_debug("RELOC at %p: %li-type as %s (0x%lx) + %li\n", 5447568cb4eSPaul Mackerras location, (long)ELF64_R_TYPE(rela[i].r_info), 5457568cb4eSPaul Mackerras strtab + sym->st_name, (unsigned long)sym->st_value, 5467568cb4eSPaul Mackerras (long)rela[i].r_addend); 5477568cb4eSPaul Mackerras 5487568cb4eSPaul Mackerras /* `Everything is relative'. */ 5497568cb4eSPaul Mackerras value = sym->st_value + rela[i].r_addend; 5507568cb4eSPaul Mackerras 5517568cb4eSPaul Mackerras switch (ELF64_R_TYPE(rela[i].r_info)) { 5527568cb4eSPaul Mackerras case R_PPC64_ADDR32: 5537568cb4eSPaul Mackerras /* Simply set it */ 5547568cb4eSPaul Mackerras *(u32 *)location = value; 5557568cb4eSPaul Mackerras break; 5567568cb4eSPaul Mackerras 5577568cb4eSPaul Mackerras case R_PPC64_ADDR64: 5587568cb4eSPaul Mackerras /* Simply set it */ 5597568cb4eSPaul Mackerras *(unsigned long *)location = value; 5607568cb4eSPaul Mackerras break; 5617568cb4eSPaul Mackerras 5627568cb4eSPaul Mackerras case R_PPC64_TOC: 5637568cb4eSPaul Mackerras *(unsigned long *)location = my_r2(sechdrs, me); 5647568cb4eSPaul Mackerras break; 5657568cb4eSPaul Mackerras 5667568cb4eSPaul Mackerras case R_PPC64_TOC16: 567f749edaeSAlan Modra /* Subtract TOC pointer */ 5687568cb4eSPaul Mackerras value -= my_r2(sechdrs, me); 5697568cb4eSPaul Mackerras if (value + 0x8000 > 0xffff) { 570c7d1f6afSAnton Blanchard pr_err("%s: bad TOC16 relocation (0x%lx)\n", 5717568cb4eSPaul Mackerras me->name, value); 5727568cb4eSPaul Mackerras return -ENOEXEC; 5737568cb4eSPaul Mackerras } 5747568cb4eSPaul Mackerras *((uint16_t *) location) 5757568cb4eSPaul Mackerras = (*((uint16_t *) location) & ~0xffff) 5767568cb4eSPaul Mackerras | (value & 0xffff); 5777568cb4eSPaul Mackerras break; 5787568cb4eSPaul Mackerras 5791fbe9cf2SAnton Blanchard case R_PPC64_TOC16_LO: 5801fbe9cf2SAnton Blanchard /* Subtract TOC pointer */ 5811fbe9cf2SAnton Blanchard value -= my_r2(sechdrs, me); 5821fbe9cf2SAnton Blanchard *((uint16_t *) location) 5831fbe9cf2SAnton Blanchard = (*((uint16_t *) location) & ~0xffff) 5841fbe9cf2SAnton Blanchard | (value & 0xffff); 5851fbe9cf2SAnton Blanchard break; 5861fbe9cf2SAnton Blanchard 5877568cb4eSPaul Mackerras case R_PPC64_TOC16_DS: 588f749edaeSAlan Modra /* Subtract TOC pointer */ 5897568cb4eSPaul Mackerras value -= my_r2(sechdrs, me); 5907568cb4eSPaul Mackerras if ((value & 3) != 0 || value + 0x8000 > 0xffff) { 591c7d1f6afSAnton Blanchard pr_err("%s: bad TOC16_DS relocation (0x%lx)\n", 5927568cb4eSPaul Mackerras me->name, value); 5937568cb4eSPaul Mackerras return -ENOEXEC; 5947568cb4eSPaul Mackerras } 5957568cb4eSPaul Mackerras *((uint16_t *) location) 5967568cb4eSPaul Mackerras = (*((uint16_t *) location) & ~0xfffc) 5977568cb4eSPaul Mackerras | (value & 0xfffc); 5987568cb4eSPaul Mackerras break; 5997568cb4eSPaul Mackerras 6001fbe9cf2SAnton Blanchard case R_PPC64_TOC16_LO_DS: 6011fbe9cf2SAnton Blanchard /* Subtract TOC pointer */ 6021fbe9cf2SAnton Blanchard value -= my_r2(sechdrs, me); 6031fbe9cf2SAnton Blanchard if ((value & 3) != 0) { 604c7d1f6afSAnton Blanchard pr_err("%s: bad TOC16_LO_DS relocation (0x%lx)\n", 6051fbe9cf2SAnton Blanchard me->name, value); 6061fbe9cf2SAnton Blanchard return -ENOEXEC; 6071fbe9cf2SAnton Blanchard } 6081fbe9cf2SAnton Blanchard *((uint16_t *) location) 6091fbe9cf2SAnton Blanchard = (*((uint16_t *) location) & ~0xfffc) 6101fbe9cf2SAnton Blanchard | (value & 0xfffc); 6111fbe9cf2SAnton Blanchard break; 6121fbe9cf2SAnton Blanchard 6131fbe9cf2SAnton Blanchard case R_PPC64_TOC16_HA: 6141fbe9cf2SAnton Blanchard /* Subtract TOC pointer */ 6151fbe9cf2SAnton Blanchard value -= my_r2(sechdrs, me); 6161fbe9cf2SAnton Blanchard value = ((value + 0x8000) >> 16); 6171fbe9cf2SAnton Blanchard *((uint16_t *) location) 6181fbe9cf2SAnton Blanchard = (*((uint16_t *) location) & ~0xffff) 6191fbe9cf2SAnton Blanchard | (value & 0xffff); 6201fbe9cf2SAnton Blanchard break; 6211fbe9cf2SAnton Blanchard 6227568cb4eSPaul Mackerras case R_PPC_REL24: 6237568cb4eSPaul Mackerras /* FIXME: Handle weak symbols here --RR */ 6247568cb4eSPaul Mackerras if (sym->st_shndx == SHN_UNDEF) { 6257568cb4eSPaul Mackerras /* External: go via stub */ 6267568cb4eSPaul Mackerras value = stub_for_addr(sechdrs, value, me); 6277568cb4eSPaul Mackerras if (!value) 6287568cb4eSPaul Mackerras return -ENOENT; 6297568cb4eSPaul Mackerras if (!restore_r2((u32 *)location + 1, me)) 6307568cb4eSPaul Mackerras return -ENOEXEC; 63115308664STorsten Duwe 63215308664STorsten Duwe squash_toc_save_inst(strtab + sym->st_name, value); 633008d7a91SRusty Russell } else 634008d7a91SRusty Russell value += local_entry_offset(sym); 6357568cb4eSPaul Mackerras 6367568cb4eSPaul Mackerras /* Convert value to relative */ 6377568cb4eSPaul Mackerras value -= (unsigned long)location; 6387568cb4eSPaul Mackerras if (value + 0x2000000 > 0x3ffffff || (value & 3) != 0){ 639c7d1f6afSAnton Blanchard pr_err("%s: REL24 %li out of range!\n", 6407568cb4eSPaul Mackerras me->name, (long int)value); 6417568cb4eSPaul Mackerras return -ENOEXEC; 6427568cb4eSPaul Mackerras } 6437568cb4eSPaul Mackerras 6447568cb4eSPaul Mackerras /* Only replace bits 2 through 26 */ 6457568cb4eSPaul Mackerras *(uint32_t *)location 6467568cb4eSPaul Mackerras = (*(uint32_t *)location & ~0x03fffffc) 6477568cb4eSPaul Mackerras | (value & 0x03fffffc); 6487568cb4eSPaul Mackerras break; 6497568cb4eSPaul Mackerras 65021c4ff80SBenjamin Herrenschmidt case R_PPC64_REL64: 65121c4ff80SBenjamin Herrenschmidt /* 64 bits relative (used by features fixups) */ 65221c4ff80SBenjamin Herrenschmidt *location = value - (unsigned long)location; 65321c4ff80SBenjamin Herrenschmidt break; 65421c4ff80SBenjamin Herrenschmidt 655*9f751b82SMichael Ellerman case R_PPC64_REL32: 656*9f751b82SMichael Ellerman /* 32 bits relative (used by relative exception tables) */ 657*9f751b82SMichael Ellerman *(u32 *)location = value - (unsigned long)location; 658*9f751b82SMichael Ellerman break; 659*9f751b82SMichael Ellerman 660d247da0aSRusty Russell case R_PPC64_TOCSAVE: 661d247da0aSRusty Russell /* 662d247da0aSRusty Russell * Marker reloc indicates we don't have to save r2. 663d247da0aSRusty Russell * That would only save us one instruction, so ignore 664d247da0aSRusty Russell * it. 665d247da0aSRusty Russell */ 666d247da0aSRusty Russell break; 667d247da0aSRusty Russell 668a61674bdSUlrich Weigand case R_PPC64_ENTRY: 669a61674bdSUlrich Weigand /* 670a61674bdSUlrich Weigand * Optimize ELFv2 large code model entry point if 671a61674bdSUlrich Weigand * the TOC is within 2GB range of current location. 672a61674bdSUlrich Weigand */ 673a61674bdSUlrich Weigand value = my_r2(sechdrs, me) - (unsigned long)location; 674a61674bdSUlrich Weigand if (value + 0x80008000 > 0xffffffff) 675a61674bdSUlrich Weigand break; 676a61674bdSUlrich Weigand /* 677a61674bdSUlrich Weigand * Check for the large code model prolog sequence: 678a61674bdSUlrich Weigand * ld r2, ...(r12) 679a61674bdSUlrich Weigand * add r2, r2, r12 680a61674bdSUlrich Weigand */ 681a61674bdSUlrich Weigand if ((((uint32_t *)location)[0] & ~0xfffc) 682a61674bdSUlrich Weigand != 0xe84c0000) 683a61674bdSUlrich Weigand break; 684a61674bdSUlrich Weigand if (((uint32_t *)location)[1] != 0x7c426214) 685a61674bdSUlrich Weigand break; 686a61674bdSUlrich Weigand /* 687a61674bdSUlrich Weigand * If found, replace it with: 688a61674bdSUlrich Weigand * addis r2, r12, (.TOC.-func)@ha 689a61674bdSUlrich Weigand * addi r2, r12, (.TOC.-func)@l 690a61674bdSUlrich Weigand */ 691a61674bdSUlrich Weigand ((uint32_t *)location)[0] = 0x3c4c0000 + PPC_HA(value); 692a61674bdSUlrich Weigand ((uint32_t *)location)[1] = 0x38420000 + PPC_LO(value); 693a61674bdSUlrich Weigand break; 694a61674bdSUlrich Weigand 6950906584aSRusty Russell case R_PPC64_REL16_HA: 6960906584aSRusty Russell /* Subtract location pointer */ 6970906584aSRusty Russell value -= (unsigned long)location; 6980906584aSRusty Russell value = ((value + 0x8000) >> 16); 6990906584aSRusty Russell *((uint16_t *) location) 7000906584aSRusty Russell = (*((uint16_t *) location) & ~0xffff) 7010906584aSRusty Russell | (value & 0xffff); 7020906584aSRusty Russell break; 7030906584aSRusty Russell 7040906584aSRusty Russell case R_PPC64_REL16_LO: 7050906584aSRusty Russell /* Subtract location pointer */ 7060906584aSRusty Russell value -= (unsigned long)location; 7070906584aSRusty Russell *((uint16_t *) location) 7080906584aSRusty Russell = (*((uint16_t *) location) & ~0xffff) 7090906584aSRusty Russell | (value & 0xffff); 7100906584aSRusty Russell break; 7110906584aSRusty Russell 7127568cb4eSPaul Mackerras default: 713c7d1f6afSAnton Blanchard pr_err("%s: Unknown ADD relocation: %lu\n", 7147568cb4eSPaul Mackerras me->name, 7157568cb4eSPaul Mackerras (unsigned long)ELF64_R_TYPE(rela[i].r_info)); 7167568cb4eSPaul Mackerras return -ENOEXEC; 7177568cb4eSPaul Mackerras } 7187568cb4eSPaul Mackerras } 7197568cb4eSPaul Mackerras 720136cd345SMichael Ellerman return 0; 721136cd345SMichael Ellerman } 722136cd345SMichael Ellerman 723f48cb8b4SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE 724336a7b5dSMichael Ellerman 725336a7b5dSMichael Ellerman #ifdef CC_USING_MPROFILE_KERNEL 726336a7b5dSMichael Ellerman 727336a7b5dSMichael Ellerman #define PACATOC offsetof(struct paca_struct, kernel_toc) 728336a7b5dSMichael Ellerman 729336a7b5dSMichael Ellerman /* 730336a7b5dSMichael Ellerman * For mprofile-kernel we use a special stub for ftrace_caller() because we 731336a7b5dSMichael Ellerman * can't rely on r2 containing this module's TOC when we enter the stub. 732336a7b5dSMichael Ellerman * 733336a7b5dSMichael Ellerman * That can happen if the function calling us didn't need to use the toc. In 734336a7b5dSMichael Ellerman * that case it won't have setup r2, and the r2 value will be either the 735336a7b5dSMichael Ellerman * kernel's toc, or possibly another modules toc. 736336a7b5dSMichael Ellerman * 737336a7b5dSMichael Ellerman * To deal with that this stub uses the kernel toc, which is always accessible 738336a7b5dSMichael Ellerman * via the paca (in r13). The target (ftrace_caller()) is responsible for 739336a7b5dSMichael Ellerman * saving and restoring the toc before returning. 740336a7b5dSMichael Ellerman */ 741336a7b5dSMichael Ellerman static unsigned long create_ftrace_stub(const Elf64_Shdr *sechdrs, struct module *me) 742336a7b5dSMichael Ellerman { 743336a7b5dSMichael Ellerman struct ppc64_stub_entry *entry; 744336a7b5dSMichael Ellerman unsigned int i, num_stubs; 745336a7b5dSMichael Ellerman static u32 stub_insns[] = { 746336a7b5dSMichael Ellerman 0xe98d0000 | PACATOC, /* ld r12,PACATOC(r13) */ 747336a7b5dSMichael Ellerman 0x3d8c0000, /* addis r12,r12,<high> */ 748336a7b5dSMichael Ellerman 0x398c0000, /* addi r12,r12,<low> */ 749336a7b5dSMichael Ellerman 0x7d8903a6, /* mtctr r12 */ 750336a7b5dSMichael Ellerman 0x4e800420, /* bctr */ 751336a7b5dSMichael Ellerman }; 752336a7b5dSMichael Ellerman long reladdr; 753336a7b5dSMichael Ellerman 754336a7b5dSMichael Ellerman num_stubs = sechdrs[me->arch.stubs_section].sh_size / sizeof(*entry); 755336a7b5dSMichael Ellerman 756336a7b5dSMichael Ellerman /* Find the next available stub entry */ 757336a7b5dSMichael Ellerman entry = (void *)sechdrs[me->arch.stubs_section].sh_addr; 758336a7b5dSMichael Ellerman for (i = 0; i < num_stubs && stub_func_addr(entry->funcdata); i++, entry++); 759336a7b5dSMichael Ellerman 760336a7b5dSMichael Ellerman if (i >= num_stubs) { 761336a7b5dSMichael Ellerman pr_err("%s: Unable to find a free slot for ftrace stub.\n", me->name); 762336a7b5dSMichael Ellerman return 0; 763336a7b5dSMichael Ellerman } 764336a7b5dSMichael Ellerman 765336a7b5dSMichael Ellerman memcpy(entry->jump, stub_insns, sizeof(stub_insns)); 766336a7b5dSMichael Ellerman 767336a7b5dSMichael Ellerman /* Stub uses address relative to kernel toc (from the paca) */ 768336a7b5dSMichael Ellerman reladdr = (unsigned long)ftrace_caller - kernel_toc_addr(); 769336a7b5dSMichael Ellerman if (reladdr > 0x7FFFFFFF || reladdr < -(0x80000000L)) { 770336a7b5dSMichael Ellerman pr_err("%s: Address of ftrace_caller out of range of kernel_toc.\n", me->name); 771336a7b5dSMichael Ellerman return 0; 772336a7b5dSMichael Ellerman } 773336a7b5dSMichael Ellerman 774336a7b5dSMichael Ellerman entry->jump[1] |= PPC_HA(reladdr); 775336a7b5dSMichael Ellerman entry->jump[2] |= PPC_LO(reladdr); 776336a7b5dSMichael Ellerman 777336a7b5dSMichael Ellerman /* Eventhough we don't use funcdata in the stub, it's needed elsewhere. */ 778336a7b5dSMichael Ellerman entry->funcdata = func_desc((unsigned long)ftrace_caller); 779336a7b5dSMichael Ellerman entry->magic = STUB_MAGIC; 780336a7b5dSMichael Ellerman 781336a7b5dSMichael Ellerman return (unsigned long)entry; 782336a7b5dSMichael Ellerman } 783336a7b5dSMichael Ellerman #else 784336a7b5dSMichael Ellerman static unsigned long create_ftrace_stub(const Elf64_Shdr *sechdrs, struct module *me) 785336a7b5dSMichael Ellerman { 786336a7b5dSMichael Ellerman return stub_for_addr(sechdrs, (unsigned long)ftrace_caller, me); 787336a7b5dSMichael Ellerman } 788f48cb8b4SSteven Rostedt #endif 789f48cb8b4SSteven Rostedt 790136cd345SMichael Ellerman int module_finalize_ftrace(struct module *mod, const Elf_Shdr *sechdrs) 791136cd345SMichael Ellerman { 792136cd345SMichael Ellerman mod->arch.toc = my_r2(sechdrs, mod); 793336a7b5dSMichael Ellerman mod->arch.tramp = create_ftrace_stub(sechdrs, mod); 794136cd345SMichael Ellerman 795136cd345SMichael Ellerman if (!mod->arch.tramp) 796136cd345SMichael Ellerman return -ENOENT; 7977568cb4eSPaul Mackerras 7987568cb4eSPaul Mackerras return 0; 7997568cb4eSPaul Mackerras } 800136cd345SMichael Ellerman #endif 801