2 * This file is subject to the terms and conditions of the GNU General Public
3 * License. See the file "COPYING" in the main directory of this archive
6 * Support for Kernel relocation at boot time
8 * Copyright (C) 2015, Imagination Technologies Ltd.
9 * Authors: Matt Redfearn (matt.redfearn@imgtec.com)
11 #include <asm/bootinfo.h>
12 #include <asm/cacheflush.h>
13 #include <asm/fw/fw.h>
14 #include <asm/sections.h>
15 #include <asm/setup.h>
16 #include <asm/timex.h>
17 #include <linux/elf.h>
18 #include <linux/kernel.h>
19 #include <linux/libfdt.h>
20 #include <linux/of_fdt.h>
21 #include <linux/sched.h>
22 #include <linux/start_kernel.h>
23 #include <linux/string.h>
24 #include <linux/printk.h>
26 #define RELOCATED(x) ((void *)((long)x + offset))
28 extern u32 _relocation_start[]; /* End kernel image / start relocation table */
29 extern u32 _relocation_end[]; /* End relocation table */
31 extern long __start___ex_table; /* Start exception table */
32 extern long __stop___ex_table; /* End exception table */
34 static inline u32 __init get_synci_step(void)
38 __asm__("rdhwr %0, $1" : "=r" (res));
43 static void __init sync_icache(void *kbase, unsigned long kernel_length)
45 void *kend = kbase + kernel_length;
46 u32 step = get_synci_step();
55 } while (kbase < kend);
57 /* Completion barrier */
61 static int __init apply_r_mips_64_rel(u32 *loc_orig, u32 *loc_new, long offset)
63 *(u64 *)loc_new += offset;
68 static int __init apply_r_mips_32_rel(u32 *loc_orig, u32 *loc_new, long offset)
75 static int __init apply_r_mips_26_rel(u32 *loc_orig, u32 *loc_new, long offset)
77 unsigned long target_addr = (*loc_orig) & 0x03ffffff;
80 pr_err("Dangerous R_MIPS_26 REL relocation\n");
84 /* Original target address */
86 target_addr += (unsigned long)loc_orig & ~0x03ffffff;
88 /* Get the new target address */
89 target_addr += offset;
91 if ((target_addr & 0xf0000000) != ((unsigned long)loc_new & 0xf0000000)) {
92 pr_err("R_MIPS_26 REL relocation overflow\n");
96 target_addr -= (unsigned long)loc_new & ~0x03ffffff;
99 *loc_new = (*loc_new & ~0x03ffffff) | (target_addr & 0x03ffffff);
105 static int __init apply_r_mips_hi16_rel(u32 *loc_orig, u32 *loc_new, long offset)
107 unsigned long insn = *loc_orig;
108 unsigned long target = (insn & 0xffff) << 16; /* high 16bits of target */
112 *loc_new = (insn & ~0xffff) | ((target >> 16) & 0xffff);
116 static int (*reloc_handlers_rel[]) (u32 *, u32 *, long) __initdata = {
117 [R_MIPS_64] = apply_r_mips_64_rel,
118 [R_MIPS_32] = apply_r_mips_32_rel,
119 [R_MIPS_26] = apply_r_mips_26_rel,
120 [R_MIPS_HI16] = apply_r_mips_hi16_rel,
123 int __init do_relocations(void *kbase_old, void *kbase_new, long offset)
131 for (r = _relocation_start; r < _relocation_end; r++) {
132 /* Sentinel for last relocation */
136 type = (*r >> 24) & 0xff;
137 loc_orig = (void *)(kbase_old + ((*r & 0x00ffffff) << 2));
138 loc_new = RELOCATED(loc_orig);
140 if (reloc_handlers_rel[type] == NULL) {
141 /* Unsupported relocation */
142 pr_err("Unhandled relocation type %d at 0x%pK\n",
147 res = reloc_handlers_rel[type](loc_orig, loc_new, offset);
156 * The exception table is filled in by the relocs tool after vmlinux is linked.
157 * It must be relocated separately since there will not be any relocation
158 * information for it filled in by the linker.
160 static int __init relocate_exception_table(long offset)
162 unsigned long *etable_start, *etable_end, *e;
164 etable_start = RELOCATED(&__start___ex_table);
165 etable_end = RELOCATED(&__stop___ex_table);
167 for (e = etable_start; e < etable_end; e++)
173 #ifdef CONFIG_RANDOMIZE_BASE
175 static inline __init unsigned long rotate_xor(unsigned long hash,
176 const void *area, size_t size)
178 const typeof(hash) *ptr = PTR_ALIGN(area, sizeof(hash));
181 diff = (void *)ptr - area;
182 if (unlikely(size < diff + sizeof(hash)))
185 size = ALIGN_DOWN(size - diff, sizeof(hash));
187 for (i = 0; i < size / sizeof(hash); i++) {
188 /* Rotate by odd number of bits and XOR. */
189 hash = (hash << ((sizeof(hash) * 8) - 7)) | (hash >> 7);
196 static inline __init unsigned long get_random_boot(void)
198 unsigned long entropy = random_get_entropy();
199 unsigned long hash = 0;
201 /* Attempt to create a simple but unpredictable starting entropy. */
202 hash = rotate_xor(hash, linux_banner, strlen(linux_banner));
204 /* Add in any runtime entropy we can get */
205 hash = rotate_xor(hash, &entropy, sizeof(entropy));
207 #if defined(CONFIG_USE_OF)
208 /* Get any additional entropy passed in device tree */
209 if (initial_boot_params) {
213 node = fdt_path_offset(initial_boot_params, "/chosen");
215 prop = fdt_getprop_w(initial_boot_params, node,
217 if (prop && (len == sizeof(u64)))
218 hash = rotate_xor(hash, prop, sizeof(*prop));
221 #endif /* CONFIG_USE_OF */
226 static inline __init bool kaslr_disabled(void)
230 #if defined(CONFIG_CMDLINE_BOOL)
231 const char *builtin_cmdline = CONFIG_CMDLINE;
233 str = strstr(builtin_cmdline, "nokaslr");
234 if (str == builtin_cmdline ||
235 (str > builtin_cmdline && *(str - 1) == ' '))
238 str = strstr(arcs_cmdline, "nokaslr");
239 if (str == arcs_cmdline || (str > arcs_cmdline && *(str - 1) == ' '))
245 static inline void __init *determine_relocation_address(void)
247 /* Choose a new address for the kernel */
248 unsigned long kernel_length;
250 unsigned long offset;
252 if (kaslr_disabled())
255 kernel_length = (long)_end - (long)(&_text);
257 offset = get_random_boot() << 16;
258 offset &= (CONFIG_RANDOMIZE_BASE_MAX_OFFSET - 1);
259 if (offset < kernel_length)
260 offset += ALIGN(kernel_length, 0xffff);
262 return RELOCATED(dest);
267 static inline void __init *determine_relocation_address(void)
270 * Choose a new address for the kernel
271 * For now we'll hard code the destination
273 return (void *)0xffffffff81000000;
278 static inline int __init relocation_addr_valid(void *loc_new)
280 if ((unsigned long)loc_new & 0x0000ffff) {
281 /* Inappropriately aligned new location */
284 if ((unsigned long)loc_new < (unsigned long)&_end) {
285 /* New location overlaps original kernel */
291 void *__init relocate_kernel(void)
294 unsigned long kernel_length;
295 unsigned long bss_length;
298 /* Default to original kernel entry point */
299 void *kernel_entry = start_kernel;
301 /* Get the command line */
303 #if defined(CONFIG_USE_OF)
304 /* Deal with the device tree */
305 early_init_dt_scan(plat_get_fdt());
306 if (boot_command_line[0]) {
307 /* Boot command line was passed in device tree */
308 strlcpy(arcs_cmdline, boot_command_line, COMMAND_LINE_SIZE);
310 #endif /* CONFIG_USE_OF */
312 kernel_length = (long)(&_relocation_start) - (long)(&_text);
313 bss_length = (long)&__bss_stop - (long)&__bss_start;
315 loc_new = determine_relocation_address();
317 /* Sanity check relocation address */
318 if (relocation_addr_valid(loc_new))
319 offset = (unsigned long)loc_new - (unsigned long)(&_text);
321 /* Reset the command line now so we don't end up with a duplicate */
322 arcs_cmdline[0] = '\0';
325 /* Copy the kernel to it's new location */
326 memcpy(loc_new, &_text, kernel_length);
328 /* Perform relocations on the new kernel */
329 res = do_relocations(&_text, loc_new, offset);
333 /* Sync the caches ready for execution of new kernel */
334 sync_icache(loc_new, kernel_length);
336 res = relocate_exception_table(offset);
341 * The original .bss has already been cleared, and
342 * some variables such as command line parameters
343 * stored to it so make a copy in the new location.
345 memcpy(RELOCATED(&__bss_start), &__bss_start, bss_length);
347 /* The current thread is now within the relocated image */
348 __current_thread_info = RELOCATED(&init_thread_union);
350 /* Return the new kernel's entry point */
351 kernel_entry = RELOCATED(start_kernel);
358 * Show relocation information on panic.
360 void show_kernel_relocation(const char *level)
362 unsigned long offset;
364 offset = __pa_symbol(_text) - __pa_symbol(VMLINUX_LOAD_ADDRESS);
366 if (IS_ENABLED(CONFIG_RELOCATABLE) && offset > 0) {
368 pr_cont("Kernel relocated by 0x%pK\n", (void *)offset);
369 pr_cont(" .text @ 0x%pK\n", _text);
370 pr_cont(" .data @ 0x%pK\n", _sdata);
371 pr_cont(" .bss @ 0x%pK\n", __bss_start);
375 static int kernel_location_notifier_fn(struct notifier_block *self,
376 unsigned long v, void *p)
378 show_kernel_relocation(KERN_EMERG);
382 static struct notifier_block kernel_location_notifier = {
383 .notifier_call = kernel_location_notifier_fn
386 static int __init register_kernel_offset_dumper(void)
388 atomic_notifier_chain_register(&panic_notifier_list,
389 &kernel_location_notifier);
392 __initcall(register_kernel_offset_dumper);