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