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