1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Definitions and wrapper functions for kernel decompressor
4 *
5 * Copyright IBM Corp. 2010
6 *
7 * Author(s): Martin Schwidefsky <schwidefsky@de.ibm.com>
8 */
9
10 #include <linux/kernel.h>
11 #include <linux/string.h>
12 #include <asm/page.h>
13 #include "decompressor.h"
14 #include "boot.h"
15
16 /*
17 * gzip declarations
18 */
19 #define STATIC static
20
21 #undef memset
22 #undef memcpy
23 #undef memmove
24 #define memmove memmove
25 #define memzero(s, n) memset((s), 0, (n))
26
27 #if defined(CONFIG_KERNEL_BZIP2)
28 #define BOOT_HEAP_SIZE 0x400000
29 #elif defined(CONFIG_KERNEL_ZSTD)
30 #define BOOT_HEAP_SIZE 0x30000
31 #else
32 #define BOOT_HEAP_SIZE 0x10000
33 #endif
34
35 static unsigned long free_mem_ptr = (unsigned long) _end;
36 static unsigned long free_mem_end_ptr = (unsigned long) _end + BOOT_HEAP_SIZE;
37
38 #ifdef CONFIG_KERNEL_GZIP
39 #include "../../../../lib/decompress_inflate.c"
40 #endif
41
42 #ifdef CONFIG_KERNEL_BZIP2
43 #include "../../../../lib/decompress_bunzip2.c"
44 #endif
45
46 #ifdef CONFIG_KERNEL_LZ4
47 #include "../../../../lib/decompress_unlz4.c"
48 #endif
49
50 #ifdef CONFIG_KERNEL_LZMA
51 #include "../../../../lib/decompress_unlzma.c"
52 #endif
53
54 #ifdef CONFIG_KERNEL_LZO
55 #include "../../../../lib/decompress_unlzo.c"
56 #endif
57
58 #ifdef CONFIG_KERNEL_XZ
59 #include "../../../../lib/decompress_unxz.c"
60 #endif
61
62 #ifdef CONFIG_KERNEL_ZSTD
63 #include "../../../../lib/decompress_unzstd.c"
64 #endif
65
66 #define decompress_offset ALIGN((unsigned long)_end + BOOT_HEAP_SIZE, PAGE_SIZE)
67
mem_safe_offset(void)68 unsigned long mem_safe_offset(void)
69 {
70 /*
71 * due to 4MB HEAD_SIZE for bzip2
72 * 'decompress_offset + vmlinux.image_size' could be larger than
73 * kernel at final position + its .bss, so take the larger of two
74 */
75 return max(decompress_offset + vmlinux.image_size,
76 vmlinux.default_lma + vmlinux.image_size + vmlinux.bss_size);
77 }
78
decompress_kernel(void)79 void *decompress_kernel(void)
80 {
81 void *output = (void *)decompress_offset;
82
83 __decompress(_compressed_start, _compressed_end - _compressed_start,
84 NULL, NULL, output, vmlinux.image_size, NULL, error);
85 return output;
86 }
87