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