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