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 */
26unsigned long free_mem_ptr;
27unsigned long free_mem_end_ptr;
28
29/* The linker tells us where the image is. */
30extern unsigned char __image_begin[], __image_end[];
31
32/* debug interfaces  */
33#ifdef CONFIG_DEBUG_ZBOOT
34extern void puts(const char *s);
35extern 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
41extern char __appended_dtb[];
42
43void 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
84const unsigned long __stack_chk_guard = 0x000a0dff;
85
86void __stack_chk_fail(void)
87{
88	error("stack-protector: Kernel stack is corrupted\n");
89}
90
91void 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