Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/sparc-next
[linux-2.6-microblaze.git] / arch / x86 / kernel / module.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*  Kernel module help for x86.
3     Copyright (C) 2001 Rusty Russell.
4
5 */
6
7 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
8
9 #include <linux/moduleloader.h>
10 #include <linux/elf.h>
11 #include <linux/vmalloc.h>
12 #include <linux/fs.h>
13 #include <linux/string.h>
14 #include <linux/kernel.h>
15 #include <linux/kasan.h>
16 #include <linux/bug.h>
17 #include <linux/mm.h>
18 #include <linux/gfp.h>
19 #include <linux/jump_label.h>
20 #include <linux/random.h>
21 #include <linux/memory.h>
22
23 #include <asm/text-patching.h>
24 #include <asm/page.h>
25 #include <asm/pgtable.h>
26 #include <asm/setup.h>
27 #include <asm/unwind.h>
28
29 #if 0
30 #define DEBUGP(fmt, ...)                                \
31         printk(KERN_DEBUG fmt, ##__VA_ARGS__)
32 #else
33 #define DEBUGP(fmt, ...)                                \
34 do {                                                    \
35         if (0)                                          \
36                 printk(KERN_DEBUG fmt, ##__VA_ARGS__);  \
37 } while (0)
38 #endif
39
40 #ifdef CONFIG_RANDOMIZE_BASE
41 static unsigned long module_load_offset;
42
43 /* Mutex protects the module_load_offset. */
44 static DEFINE_MUTEX(module_kaslr_mutex);
45
46 static unsigned long int get_module_load_offset(void)
47 {
48         if (kaslr_enabled()) {
49                 mutex_lock(&module_kaslr_mutex);
50                 /*
51                  * Calculate the module_load_offset the first time this
52                  * code is called. Once calculated it stays the same until
53                  * reboot.
54                  */
55                 if (module_load_offset == 0)
56                         module_load_offset =
57                                 (get_random_int() % 1024 + 1) * PAGE_SIZE;
58                 mutex_unlock(&module_kaslr_mutex);
59         }
60         return module_load_offset;
61 }
62 #else
63 static unsigned long int get_module_load_offset(void)
64 {
65         return 0;
66 }
67 #endif
68
69 void *module_alloc(unsigned long size)
70 {
71         void *p;
72
73         if (PAGE_ALIGN(size) > MODULES_LEN)
74                 return NULL;
75
76         p = __vmalloc_node_range(size, MODULE_ALIGN,
77                                     MODULES_VADDR + get_module_load_offset(),
78                                     MODULES_END, GFP_KERNEL,
79                                     PAGE_KERNEL, 0, NUMA_NO_NODE,
80                                     __builtin_return_address(0));
81         if (p && (kasan_module_alloc(p, size) < 0)) {
82                 vfree(p);
83                 return NULL;
84         }
85
86         return p;
87 }
88
89 #ifdef CONFIG_X86_32
90 int apply_relocate(Elf32_Shdr *sechdrs,
91                    const char *strtab,
92                    unsigned int symindex,
93                    unsigned int relsec,
94                    struct module *me)
95 {
96         unsigned int i;
97         Elf32_Rel *rel = (void *)sechdrs[relsec].sh_addr;
98         Elf32_Sym *sym;
99         uint32_t *location;
100
101         DEBUGP("Applying relocate section %u to %u\n",
102                relsec, sechdrs[relsec].sh_info);
103         for (i = 0; i < sechdrs[relsec].sh_size / sizeof(*rel); i++) {
104                 /* This is where to make the change */
105                 location = (void *)sechdrs[sechdrs[relsec].sh_info].sh_addr
106                         + rel[i].r_offset;
107                 /* This is the symbol it is referring to.  Note that all
108                    undefined symbols have been resolved.  */
109                 sym = (Elf32_Sym *)sechdrs[symindex].sh_addr
110                         + ELF32_R_SYM(rel[i].r_info);
111
112                 switch (ELF32_R_TYPE(rel[i].r_info)) {
113                 case R_386_32:
114                         /* We add the value into the location given */
115                         *location += sym->st_value;
116                         break;
117                 case R_386_PC32:
118                         /* Add the value, subtract its position */
119                         *location += sym->st_value - (uint32_t)location;
120                         break;
121                 default:
122                         pr_err("%s: Unknown relocation: %u\n",
123                                me->name, ELF32_R_TYPE(rel[i].r_info));
124                         return -ENOEXEC;
125                 }
126         }
127         return 0;
128 }
129 #else /*X86_64*/
130 static int __apply_relocate_add(Elf64_Shdr *sechdrs,
131                    const char *strtab,
132                    unsigned int symindex,
133                    unsigned int relsec,
134                    struct module *me,
135                    void *(*write)(void *dest, const void *src, size_t len))
136 {
137         unsigned int i;
138         Elf64_Rela *rel = (void *)sechdrs[relsec].sh_addr;
139         Elf64_Sym *sym;
140         void *loc;
141         u64 val;
142
143         DEBUGP("Applying relocate section %u to %u\n",
144                relsec, sechdrs[relsec].sh_info);
145         for (i = 0; i < sechdrs[relsec].sh_size / sizeof(*rel); i++) {
146                 /* This is where to make the change */
147                 loc = (void *)sechdrs[sechdrs[relsec].sh_info].sh_addr
148                         + rel[i].r_offset;
149
150                 /* This is the symbol it is referring to.  Note that all
151                    undefined symbols have been resolved.  */
152                 sym = (Elf64_Sym *)sechdrs[symindex].sh_addr
153                         + ELF64_R_SYM(rel[i].r_info);
154
155                 DEBUGP("type %d st_value %Lx r_addend %Lx loc %Lx\n",
156                        (int)ELF64_R_TYPE(rel[i].r_info),
157                        sym->st_value, rel[i].r_addend, (u64)loc);
158
159                 val = sym->st_value + rel[i].r_addend;
160
161                 switch (ELF64_R_TYPE(rel[i].r_info)) {
162                 case R_X86_64_NONE:
163                         break;
164                 case R_X86_64_64:
165                         if (*(u64 *)loc != 0)
166                                 goto invalid_relocation;
167                         write(loc, &val, 8);
168                         break;
169                 case R_X86_64_32:
170                         if (*(u32 *)loc != 0)
171                                 goto invalid_relocation;
172                         write(loc, &val, 4);
173                         if (val != *(u32 *)loc)
174                                 goto overflow;
175                         break;
176                 case R_X86_64_32S:
177                         if (*(s32 *)loc != 0)
178                                 goto invalid_relocation;
179                         write(loc, &val, 4);
180                         if ((s64)val != *(s32 *)loc)
181                                 goto overflow;
182                         break;
183                 case R_X86_64_PC32:
184                 case R_X86_64_PLT32:
185                         if (*(u32 *)loc != 0)
186                                 goto invalid_relocation;
187                         val -= (u64)loc;
188                         write(loc, &val, 4);
189 #if 0
190                         if ((s64)val != *(s32 *)loc)
191                                 goto overflow;
192 #endif
193                         break;
194                 case R_X86_64_PC64:
195                         if (*(u64 *)loc != 0)
196                                 goto invalid_relocation;
197                         val -= (u64)loc;
198                         write(loc, &val, 8);
199                         break;
200                 default:
201                         pr_err("%s: Unknown rela relocation: %llu\n",
202                                me->name, ELF64_R_TYPE(rel[i].r_info));
203                         return -ENOEXEC;
204                 }
205         }
206         return 0;
207
208 invalid_relocation:
209         pr_err("x86/modules: Skipping invalid relocation target, existing value is nonzero for type %d, loc %p, val %Lx\n",
210                (int)ELF64_R_TYPE(rel[i].r_info), loc, val);
211         return -ENOEXEC;
212
213 overflow:
214         pr_err("overflow in relocation type %d val %Lx\n",
215                (int)ELF64_R_TYPE(rel[i].r_info), val);
216         pr_err("`%s' likely not compiled with -mcmodel=kernel\n",
217                me->name);
218         return -ENOEXEC;
219 }
220
221 int apply_relocate_add(Elf64_Shdr *sechdrs,
222                    const char *strtab,
223                    unsigned int symindex,
224                    unsigned int relsec,
225                    struct module *me)
226 {
227         int ret;
228         bool early = me->state == MODULE_STATE_UNFORMED;
229         void *(*write)(void *, const void *, size_t) = memcpy;
230
231         if (!early) {
232                 write = text_poke;
233                 mutex_lock(&text_mutex);
234         }
235
236         ret = __apply_relocate_add(sechdrs, strtab, symindex, relsec, me,
237                                    write);
238
239         if (!early) {
240                 text_poke_sync();
241                 mutex_unlock(&text_mutex);
242         }
243
244         return ret;
245 }
246
247 #endif
248
249 int module_finalize(const Elf_Ehdr *hdr,
250                     const Elf_Shdr *sechdrs,
251                     struct module *me)
252 {
253         const Elf_Shdr *s, *text = NULL, *alt = NULL, *locks = NULL,
254                 *para = NULL, *orc = NULL, *orc_ip = NULL;
255         char *secstrings = (void *)hdr + sechdrs[hdr->e_shstrndx].sh_offset;
256
257         for (s = sechdrs; s < sechdrs + hdr->e_shnum; s++) {
258                 if (!strcmp(".text", secstrings + s->sh_name))
259                         text = s;
260                 if (!strcmp(".altinstructions", secstrings + s->sh_name))
261                         alt = s;
262                 if (!strcmp(".smp_locks", secstrings + s->sh_name))
263                         locks = s;
264                 if (!strcmp(".parainstructions", secstrings + s->sh_name))
265                         para = s;
266                 if (!strcmp(".orc_unwind", secstrings + s->sh_name))
267                         orc = s;
268                 if (!strcmp(".orc_unwind_ip", secstrings + s->sh_name))
269                         orc_ip = s;
270         }
271
272         if (alt) {
273                 /* patch .altinstructions */
274                 void *aseg = (void *)alt->sh_addr;
275                 apply_alternatives(aseg, aseg + alt->sh_size);
276         }
277         if (locks && text) {
278                 void *lseg = (void *)locks->sh_addr;
279                 void *tseg = (void *)text->sh_addr;
280                 alternatives_smp_module_add(me, me->name,
281                                             lseg, lseg + locks->sh_size,
282                                             tseg, tseg + text->sh_size);
283         }
284
285         if (para) {
286                 void *pseg = (void *)para->sh_addr;
287                 apply_paravirt(pseg, pseg + para->sh_size);
288         }
289
290         /* make jump label nops */
291         jump_label_apply_nops(me);
292
293         if (orc && orc_ip)
294                 unwind_module_init(me, (void *)orc_ip->sh_addr, orc_ip->sh_size,
295                                    (void *)orc->sh_addr, orc->sh_size);
296
297         return 0;
298 }
299
300 void module_arch_cleanup(struct module *mod)
301 {
302         alternatives_smp_module_del(mod);
303 }