arch/mips/boot/compressed/decompress.c

   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 }