1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3 * Copyright 2001 MontaVista Software Inc.
4 * Author: Matt Porter <mporter@mvista.com>
5 *
6 * Copyright (C) 2009 Lemote, Inc.
7 * Author: Wu Zhangjin <wuzhangjin@gmail.com>
8 */
9
10 #define DISABLE_BRANCH_PROFILING
11
12 #include <linux/types.h>
13 #include <linux/kernel.h>
14 #include <linux/string.h>
15 #include <linux/libfdt.h>
16
17 #include <asm/addrspace.h>
18 #include <asm/unaligned.h>
19 #include <asm-generic/vmlinux.lds.h>
20
21 /*
22 * These two variables specify the free mem region
23 * that can be used for temporary malloc area
24 */
25 unsigned long free_mem_ptr;
26 unsigned long free_mem_end_ptr;
27
28 /* The linker tells us where the image is. */
29 extern unsigned char __image_begin, __image_end;
30
31 /* debug interfaces */
32 #ifdef CONFIG_DEBUG_ZBOOT
33 extern void puts(const char *s);
34 extern void puthex(unsigned long long val);
35 #else
36 #define puts(s) do {} while (0)
37 #define puthex(val) do {} while (0)
38 #endif
39
40 extern char __appended_dtb[];
41
error(char * x)42 void error(char *x)
43 {
44 puts("\n\n");
45 puts(x);
46 puts("\n\n -- System halted");
47
48 while (1)
49 ; /* Halt */
50 }
51
52 /* activate the code for pre-boot environment */
53 #define STATIC static
54
55 #ifdef CONFIG_KERNEL_GZIP
56 #include "../../../../lib/decompress_inflate.c"
57 #endif
58
59 #ifdef CONFIG_KERNEL_BZIP2
60 #include "../../../../lib/decompress_bunzip2.c"
61 #endif
62
63 #ifdef CONFIG_KERNEL_LZ4
64 #include "../../../../lib/decompress_unlz4.c"
65 #endif
66
67 #ifdef CONFIG_KERNEL_LZMA
68 #include "../../../../lib/decompress_unlzma.c"
69 #endif
70
71 #ifdef CONFIG_KERNEL_LZO
72 #include "../../../../lib/decompress_unlzo.c"
73 #endif
74
75 #ifdef CONFIG_KERNEL_XZ
76 #include "../../../../lib/decompress_unxz.c"
77 #endif
78
79 #ifdef CONFIG_KERNEL_ZSTD
80 #include "../../../../lib/decompress_unzstd.c"
81 #endif
82
83 const unsigned long __stack_chk_guard = 0x000a0dff;
84
__stack_chk_fail(void)85 void __stack_chk_fail(void)
86 {
87 error("stack-protector: Kernel stack is corrupted\n");
88 }
89
decompress_kernel(unsigned long boot_heap_start)90 void decompress_kernel(unsigned long boot_heap_start)
91 {
92 unsigned long zimage_start, zimage_size;
93
94 zimage_start = (unsigned long)(&__image_begin);
95 zimage_size = (unsigned long)(&__image_end) -
96 (unsigned long)(&__image_begin);
97
98 puts("zimage at: ");
99 puthex(zimage_start);
100 puts(" ");
101 puthex(zimage_size + zimage_start);
102 puts("\n");
103
104 /* This area are prepared for mallocing when decompressing */
105 free_mem_ptr = boot_heap_start;
106 free_mem_end_ptr = boot_heap_start + BOOT_HEAP_SIZE;
107
108 /* Display standard Linux/MIPS boot prompt */
109 puts("Uncompressing Linux at load address ");
110 puthex(VMLINUX_LOAD_ADDRESS_ULL);
111 puts("\n");
112
113 /* Decompress the kernel with according algorithm */
114 __decompress((char *)zimage_start, zimage_size, 0, 0,
115 (void *)VMLINUX_LOAD_ADDRESS_ULL, 0, 0, error);
116
117 if (IS_ENABLED(CONFIG_MIPS_RAW_APPENDED_DTB) &&
118 fdt_magic((void *)&__appended_dtb) == FDT_MAGIC) {
119 unsigned int image_size, dtb_size;
120
121 dtb_size = fdt_totalsize((void *)&__appended_dtb);
122
123 /* last four bytes is always image size in little endian */
124 image_size = get_unaligned_le32((void *)&__image_end - 4);
125
126 /* The device tree's address must be properly aligned */
127 image_size = ALIGN(image_size, STRUCT_ALIGNMENT);
128
129 puts("Copy device tree to address ");
130 puthex(VMLINUX_LOAD_ADDRESS_ULL + image_size);
131 puts("\n");
132
133 /* copy dtb to where the booted kernel will expect it */
134 memcpy((void *)VMLINUX_LOAD_ADDRESS_ULL + image_size,
135 __appended_dtb, dtb_size);
136 }
137
138 /* FIXME: should we flush cache here? */
139 puts("Now, booting the kernel...\n");
140 }
141