60 lines
1.9 KiB
C
60 lines
1.9 KiB
C
|
// SPDX-License-Identifier: GPL-2.0-only
|
||
|
|
/*
|
||
|
|
* XIP kernel .data segment decompressor
|
||
|
|
*
|
||
|
|
* Created by: Nicolas Pitre, August 2017
|
||
|
|
* Copyright: (C) 2017 Linaro Limited
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <linux/init.h>
|
||
|
|
#include <linux/zutil.h>
|
||
|
|
|
||
|
|
/* for struct inflate_state */
|
||
|
|
#include "../../../lib/zlib_inflate/inftrees.h"
|
||
|
|
#include "../../../lib/zlib_inflate/inflate.h"
|
||
|
|
#include "../../../lib/zlib_inflate/infutil.h"
|
||
|
|
|
||
|
|
extern char __data_loc[];
|
||
|
|
extern char _edata_loc[];
|
||
|
|
extern char _sdata[];
|
||
|
|
|
||
|
|
/*
|
||
|
|
* This code is called very early during the boot process to decompress
|
||
|
|
* the .data segment stored compressed in ROM. Therefore none of the global
|
||
|
|
* variables are valid yet, hence no kernel services such as memory
|
||
|
|
* allocation is available. Everything must be allocated on the stack and
|
||
|
|
* we must avoid any global data access. We use a temporary stack located
|
||
|
|
* in the .bss area. The linker script makes sure the .bss is big enough
|
||
|
|
* to hold our stack frame plus some room for called functions.
|
||
|
|
*
|
||
|
|
* We mimic the code in lib/decompress_inflate.c to use the smallest work
|
||
|
|
* area possible. And because everything is statically allocated on the
|
||
|
|
* stack then there is no need to clean up before returning.
|
||
|
|
*/
|
||
|
|
|
||
|
|
int __init __inflate_kernel_data(void)
|
||
|
|
{
|
||
|
|
struct z_stream_s stream, *strm = &stream;
|
||
|
|
struct inflate_state state;
|
||
|
|
char *in = __data_loc;
|
||
|
|
int rc;
|
||
|
|
|
||
|
|
/* Check and skip gzip header (assume no filename) */
|
||
|
|
if (in[0] != 0x1f || in[1] != 0x8b || in[2] != 0x08 || in[3] & ~3)
|
||
|
|
return -1;
|
||
|
|
in += 10;
|
||
|
|
|
||
|
|
strm->workspace = &state;
|
||
|
|
strm->next_in = in;
|
||
|
|
strm->avail_in = _edata_loc - __data_loc; /* upper bound */
|
||
|
|
strm->next_out = _sdata;
|
||
|
|
strm->avail_out = _edata_loc - __data_loc;
|
||
|
|
zlib_inflateInit2(strm, -MAX_WBITS);
|
||
|
|
WS(strm)->inflate_state.wsize = 0;
|
||
|
|
WS(strm)->inflate_state.window = NULL;
|
||
|
|
rc = zlib_inflate(strm, Z_FINISH);
|
||
|
|
if (rc == Z_OK || rc == Z_STREAM_END)
|
||
|
|
rc = strm->avail_out; /* should be 0 */
|
||
|
|
return rc;
|
||
|
|
}
|