123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302 |
- #ifdef STATIC
- #define PREBOOT
- #include "lzo/lzo1x_decompress_safe.c"
- #else
- #include <linux/decompress/unlzo.h>
- #endif
- #include <linux/types.h>
- #include <linux/lzo.h>
- #include <linux/decompress/mm.h>
- #include <linux/compiler.h>
- #include <asm/unaligned.h>
- static const unsigned char lzop_magic[] = {
- 0x89, 0x4c, 0x5a, 0x4f, 0x00, 0x0d, 0x0a, 0x1a, 0x0a };
- #define LZO_BLOCK_SIZE (256*1024l)
- #define HEADER_HAS_FILTER 0x00000800L
- #define HEADER_SIZE_MIN (9 + 7 + 4 + 8 + 1 + 4)
- #define HEADER_SIZE_MAX (9 + 7 + 1 + 8 + 8 + 4 + 1 + 255 + 4)
- STATIC inline long INIT parse_header(u8 *input, long *skip, long in_len)
- {
- int l;
- u8 *parse = input;
- u8 *end = input + in_len;
- u8 level = 0;
- u16 version;
-
- if (in_len < HEADER_SIZE_MIN)
- return 0;
-
- for (l = 0; l < 9; l++) {
- if (*parse++ != lzop_magic[l])
- return 0;
- }
-
- version = get_unaligned_be16(parse);
- parse += 7;
- if (version >= 0x0940)
- level = *parse++;
- if (get_unaligned_be32(parse) & HEADER_HAS_FILTER)
- parse += 8;
- else
- parse += 4;
-
- if (end - parse < 8 + 1 + 4)
- return 0;
-
- parse += 8;
- if (version >= 0x0940)
- parse += 4;
- l = *parse++;
-
- if (end - parse < l + 4)
- return 0;
- parse += l + 4;
- *skip = parse - input;
- return 1;
- }
- STATIC int INIT unlzo(u8 *input, long in_len,
- long (*fill)(void *, unsigned long),
- long (*flush)(void *, unsigned long),
- u8 *output, long *posp,
- void (*error) (char *x))
- {
- u8 r = 0;
- long skip = 0;
- u32 src_len, dst_len;
- size_t tmp;
- u8 *in_buf, *in_buf_save, *out_buf;
- int ret = -1;
- if (output) {
- out_buf = output;
- } else if (!flush) {
- error("NULL output pointer and no flush function provided");
- goto exit;
- } else {
- out_buf = malloc(LZO_BLOCK_SIZE);
- if (!out_buf) {
- error("Could not allocate output buffer");
- goto exit;
- }
- }
- if (input && fill) {
- error("Both input pointer and fill function provided, don't know what to do");
- goto exit_1;
- } else if (input) {
- in_buf = input;
- } else if (!fill) {
- error("NULL input pointer and missing fill function");
- goto exit_1;
- } else {
- in_buf = malloc(lzo1x_worst_compress(LZO_BLOCK_SIZE));
- if (!in_buf) {
- error("Could not allocate input buffer");
- goto exit_1;
- }
- }
- in_buf_save = in_buf;
- if (posp)
- *posp = 0;
- if (fill) {
-
- in_buf += HEADER_SIZE_MAX;
- in_len = fill(in_buf, HEADER_SIZE_MAX);
- }
- if (!parse_header(in_buf, &skip, in_len)) {
- error("invalid header");
- goto exit_2;
- }
- in_buf += skip;
- in_len -= skip;
- if (fill) {
-
- memcpy(in_buf_save, in_buf, in_len);
- in_buf = in_buf_save;
- }
- if (posp)
- *posp = skip;
- for (;;) {
-
- if (fill && in_len < 4) {
- skip = fill(in_buf + in_len, 4 - in_len);
- if (skip > 0)
- in_len += skip;
- }
- if (in_len < 4) {
- error("file corrupted");
- goto exit_2;
- }
- dst_len = get_unaligned_be32(in_buf);
- in_buf += 4;
- in_len -= 4;
-
- if (dst_len == 0) {
- if (posp)
- *posp += 4;
- break;
- }
- if (dst_len > LZO_BLOCK_SIZE) {
- error("dest len longer than block size");
- goto exit_2;
- }
-
- if (fill && in_len < 8) {
- skip = fill(in_buf + in_len, 8 - in_len);
- if (skip > 0)
- in_len += skip;
- }
- if (in_len < 8) {
- error("file corrupted");
- goto exit_2;
- }
- src_len = get_unaligned_be32(in_buf);
- in_buf += 8;
- in_len -= 8;
- if (src_len <= 0 || src_len > dst_len) {
- error("file corrupted");
- goto exit_2;
- }
-
- if (fill && in_len < src_len) {
- skip = fill(in_buf + in_len, src_len - in_len);
- if (skip > 0)
- in_len += skip;
- }
- if (in_len < src_len) {
- error("file corrupted");
- goto exit_2;
- }
- tmp = dst_len;
-
- if (unlikely(dst_len == src_len))
- memcpy(out_buf, in_buf, src_len);
- else {
- r = lzo1x_decompress_safe((u8 *) in_buf, src_len,
- out_buf, &tmp);
- if (r != LZO_E_OK || dst_len != tmp) {
- error("Compressed data violation");
- goto exit_2;
- }
- }
- if (flush && flush(out_buf, dst_len) != dst_len)
- goto exit_2;
- if (output)
- out_buf += dst_len;
- if (posp)
- *posp += src_len + 12;
- in_buf += src_len;
- in_len -= src_len;
- if (fill) {
-
- if (in_len > 0)
- for (skip = 0; skip < in_len; ++skip)
- in_buf_save[skip] = in_buf[skip];
- in_buf = in_buf_save;
- }
- }
- ret = 0;
- exit_2:
- if (!input)
- free(in_buf_save);
- exit_1:
- if (!output)
- free(out_buf);
- exit:
- return ret;
- }
- #ifdef PREBOOT
- STATIC int INIT __decompress(unsigned char *buf, long len,
- long (*fill)(void*, unsigned long),
- long (*flush)(void*, unsigned long),
- unsigned char *out_buf, long olen,
- long *pos,
- void (*error)(char *x))
- {
- return unlzo(buf, len, fill, flush, out_buf, pos, error);
- }
- #endif
|