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