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
15/*
16 * gzip declarations
17 */
18#define STATIC static
19
20#undef memset
21#undef memcpy
22#undef memmove
23#define memmove memmove
24#define memzero(s, n) memset((s), 0, (n))
25
26#if defined(CONFIG_KERNEL_BZIP2)
27#define BOOT_HEAP_SIZE	0x400000
28#elif defined(CONFIG_KERNEL_ZSTD)
29#define BOOT_HEAP_SIZE	0x30000
30#else
31#define BOOT_HEAP_SIZE	0x10000
32#endif
33
34static unsigned long free_mem_ptr = (unsigned long) _end;
35static unsigned long free_mem_end_ptr = (unsigned long) _end + BOOT_HEAP_SIZE;
36
37#ifdef CONFIG_KERNEL_GZIP
38#include "../../../../lib/decompress_inflate.c"
39#endif
40
41#ifdef CONFIG_KERNEL_BZIP2
42#include "../../../../lib/decompress_bunzip2.c"
43#endif
44
45#ifdef CONFIG_KERNEL_LZ4
46#include "../../../../lib/decompress_unlz4.c"
47#endif
48
49#ifdef CONFIG_KERNEL_LZMA
50#include "../../../../lib/decompress_unlzma.c"
51#endif
52
53#ifdef CONFIG_KERNEL_LZO
54#include "../../../../lib/decompress_unlzo.c"
55#endif
56
57#ifdef CONFIG_KERNEL_XZ
58#include "../../../../lib/decompress_unxz.c"
59#endif
60
61#ifdef CONFIG_KERNEL_ZSTD
62#include "../../../../lib/decompress_unzstd.c"
63#endif
64
65#define decompress_offset ALIGN((unsigned long)_end + BOOT_HEAP_SIZE, PAGE_SIZE)
66
67unsigned long mem_safe_offset(void)
68{
69	/*
70	 * due to 4MB HEAD_SIZE for bzip2
71	 * 'decompress_offset + vmlinux.image_size' could be larger than
72	 * kernel at final position + its .bss, so take the larger of two
73	 */
74	return max(decompress_offset + vmlinux.image_size,
75		   vmlinux.default_lma + vmlinux.image_size + vmlinux.bss_size);
76}
77
78void *decompress_kernel(void)
79{
80	void *output = (void *)decompress_offset;
81
82	__decompress(_compressed_start, _compressed_end - _compressed_start,
83		     NULL, NULL, output, vmlinux.image_size, NULL, error);
84	return output;
85}

 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
35static unsigned long free_mem_ptr = (unsigned long) _end;
36static 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
68unsigned 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
79void *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}