cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

decompress.c (3420B)

      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}