1 // SPDX-License-Identifier: GPL-2.0-only 2 /* 3 * XIP kernel .data segment decompressor 4 * 5 * Created by: Nicolas Pitre, August 2017 6 * Copyright: (C) 2017 Linaro Limited 7 */ 8 9 #include <linux/init.h> 10 #include <linux/zutil.h> 11 #include "head.h" 12 13 /* for struct inflate_state */ 14 #include "../../../lib/zlib_inflate/inftrees.h" 15 #include "../../../lib/zlib_inflate/inflate.h" 16 #include "../../../lib/zlib_inflate/infutil.h" 17 18 /* 19 * This code is called very early during the boot process to decompress 20 * the .data segment stored compressed in ROM. Therefore none of the global 21 * variables are valid yet, hence no kernel services such as memory 22 * allocation is available. Everything must be allocated on the stack and 23 * we must avoid any global data access. We use a temporary stack located 24 * in the .bss area. The linker script makes sure the .bss is big enough 25 * to hold our stack frame plus some room for called functions. 26 * 27 * We mimic the code in lib/decompress_inflate.c to use the smallest work 28 * area possible. And because everything is statically allocated on the 29 * stack then there is no need to clean up before returning. 30 */ 31 __inflate_kernel_data(void)32int __init __inflate_kernel_data(void) 33 { 34 struct z_stream_s stream, *strm = &stream; 35 struct inflate_state state; 36 char *in = __data_loc; 37 int rc; 38 39 /* Check and skip gzip header (assume no filename) */ 40 if (in[0] != 0x1f || in[1] != 0x8b || in[2] != 0x08 || in[3] & ~3) 41 return -1; 42 in += 10; 43 44 strm->workspace = &state; 45 strm->next_in = in; 46 strm->avail_in = _edata_loc - __data_loc; /* upper bound */ 47 strm->next_out = _sdata; 48 strm->avail_out = _edata_loc - __data_loc; 49 zlib_inflateInit2(strm, -MAX_WBITS); 50 WS(strm)->inflate_state.wsize = 0; 51 WS(strm)->inflate_state.window = NULL; 52 rc = zlib_inflate(strm, Z_FINISH); 53 if (rc == Z_OK || rc == Z_STREAM_END) 54 rc = strm->avail_out; /* should be 0 */ 55 return rc; 56 } 57