760 lines
24 KiB
C
760 lines
24 KiB
C
|
/***************************************************************************
|
||
|
|
* __________ __ ___.
|
||
|
|
* Open \______ \ ____ ____ | | _\_ |__ _______ ___
|
||
|
|
* Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
|
||
|
|
* Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
|
||
|
|
* Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
|
||
|
|
* \/ \/ \/ \/ \/
|
||
|
|
* $Id$
|
||
|
|
*
|
||
|
|
* Copyright (C) 2021 by James Buren (libflate adaptations for RockBox)
|
||
|
|
*
|
||
|
|
* This program is free software; you can redistribute it and/or
|
||
|
|
* modify it under the terms of the GNU General Public License
|
||
|
|
* as published by the Free Software Foundation; either version 2
|
||
|
|
* of the License, or (at your option) any later version.
|
||
|
|
*
|
||
|
|
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
|
||
|
|
* KIND, either express or implied.
|
||
|
|
*
|
||
|
|
****************************************************************************/
|
||
|
|
|
||
|
|
/* Copyright 2021 Plan 9 Foundation
|
||
|
|
*
|
||
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
||
|
|
* a copy of this software and associated documentation files (the
|
||
|
|
* "Software"), to deal in the Software without restriction, including
|
||
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
||
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
||
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
||
|
|
* the following conditions:
|
||
|
|
*
|
||
|
|
* The above copyright notice and this permission notice shall be
|
||
|
|
* included in all copies or substantial portions of the Software.
|
||
|
|
*
|
||
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
||
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
|
||
|
|
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
|
||
|
|
* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
||
|
|
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
|
||
|
|
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "inflate.h"
|
||
|
|
#include <stdbool.h>
|
||
|
|
#include "adler32.h"
|
||
|
|
#include "crc32.h"
|
||
|
|
#include "system.h"
|
||
|
|
|
||
|
|
enum {
|
||
|
|
INFLATE_BUFFER_SIZE = 32768,
|
||
|
|
INFLATE_SYMBOL_MAX = 288,
|
||
|
|
INFLATE_OFFSET_MAX = 32,
|
||
|
|
INFLATE_CODELEN_MAX = 19,
|
||
|
|
INFLATE_HUFF_BITS = 17,
|
||
|
|
INFLATE_FLAT_BITS = 7,
|
||
|
|
INFLATE_SYMBOL_BITS = 7,
|
||
|
|
INFLATE_OFFSET_BITS = 6,
|
||
|
|
INFLATE_CODELEN_BITS = 6,
|
||
|
|
};
|
||
|
|
|
||
|
|
struct inflate_huff {
|
||
|
|
uint32_t max_bits;
|
||
|
|
uint32_t min_bits;
|
||
|
|
uint32_t flat_mask;
|
||
|
|
uint32_t flat[1 << INFLATE_FLAT_BITS];
|
||
|
|
uint32_t max_code[INFLATE_HUFF_BITS];
|
||
|
|
uint32_t last[INFLATE_HUFF_BITS];
|
||
|
|
uint32_t decode[INFLATE_SYMBOL_MAX];
|
||
|
|
};
|
||
|
|
|
||
|
|
struct inflate {
|
||
|
|
uint8_t in[INFLATE_BUFFER_SIZE];
|
||
|
|
uint8_t out[INFLATE_BUFFER_SIZE];
|
||
|
|
union {
|
||
|
|
struct {
|
||
|
|
uint8_t len[INFLATE_SYMBOL_MAX];
|
||
|
|
uint8_t off[INFLATE_OFFSET_MAX];
|
||
|
|
};
|
||
|
|
uint8_t combo[INFLATE_SYMBOL_MAX + INFLATE_OFFSET_MAX];
|
||
|
|
};
|
||
|
|
uint8_t clen[INFLATE_CODELEN_MAX];
|
||
|
|
struct inflate_huff lentab;
|
||
|
|
struct inflate_huff offtab;
|
||
|
|
struct inflate_huff clentab;
|
||
|
|
uint32_t bits[INFLATE_HUFF_BITS];
|
||
|
|
uint32_t codes[INFLATE_HUFF_BITS];
|
||
|
|
};
|
||
|
|
|
||
|
|
#define INFLATE_FILL(E) do { \
|
||
|
|
const uint32_t _size = read(is, sizeof(it->in), rctx); \
|
||
|
|
if (_size == 0) { \
|
||
|
|
rv = (E); \
|
||
|
|
goto bail; \
|
||
|
|
} \
|
||
|
|
ip = is; \
|
||
|
|
ie = is + _size; \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define INFLATE_FLUSH(E) do { \
|
||
|
|
const uint32_t _size = (op - os); \
|
||
|
|
if (write(os, _size, wctx) != _size) { \
|
||
|
|
rv = (E); \
|
||
|
|
goto bail; \
|
||
|
|
} \
|
||
|
|
flushed = true; \
|
||
|
|
if (st == INFLATE_ZLIB) \
|
||
|
|
chksum = adler_32(os, _size, chksum); \
|
||
|
|
else if (st == INFLATE_GZIP) \
|
||
|
|
chksum = crc_32r(os, _size, chksum); \
|
||
|
|
op = os; \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define INFLATE_GET_BYTE(E,C) do { \
|
||
|
|
if (ip == ie) \
|
||
|
|
INFLATE_FILL(E); \
|
||
|
|
(C) = ip[0]; \
|
||
|
|
++ip; \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define INFLATE_PUT_BYTE(E,B) do { \
|
||
|
|
if (op == oe) \
|
||
|
|
INFLATE_FLUSH(E); \
|
||
|
|
op[0] = (B); \
|
||
|
|
++op; \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define INFLATE_FILL_BITS(E,N) do { \
|
||
|
|
while ((N) > nbits) { \
|
||
|
|
uint8_t _byte; \
|
||
|
|
INFLATE_GET_BYTE(E, _byte); \
|
||
|
|
sreg |= (_byte << nbits); \
|
||
|
|
nbits += 8; \
|
||
|
|
} \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define INFLATE_CONSUME_BITS(N) do { \
|
||
|
|
sreg >>= (N); \
|
||
|
|
nbits -= (N); \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define INFLATE_EXTRACT_BITS(N,B) do { \
|
||
|
|
(B) = (sreg & ((1 << (N)) - 1)); \
|
||
|
|
INFLATE_CONSUME_BITS(N); \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define INFLATE_CHECK_LENGTH(E,N) do { \
|
||
|
|
if ((N) > (ie - ip)) { \
|
||
|
|
rv = (E); \
|
||
|
|
goto bail; \
|
||
|
|
} \
|
||
|
|
} while (0)
|
||
|
|
|
||
|
|
#define INFLATE_REVERSE(C,B) ({ \
|
||
|
|
uint32_t _c = (C); \
|
||
|
|
_c <<= (16 - (B)); \
|
||
|
|
((revtab[_c >> 8]) | (revtab[_c & 0xff] << 8)); \
|
||
|
|
})
|
||
|
|
|
||
|
|
#define INFLATE_DECODE(E,H) ({ \
|
||
|
|
__label__ _found; \
|
||
|
|
uint32_t _c = (H)->flat[sreg & (H)->flat_mask]; \
|
||
|
|
uint32_t _b = (_c & 0xff); \
|
||
|
|
uint32_t _code; \
|
||
|
|
if (_b == 0xff) { \
|
||
|
|
for (_b = (_c >> 8); _b <= (H)->max_bits; _b++) { \
|
||
|
|
_c = (revtab[sreg & 0xff] << 8); \
|
||
|
|
_c |= (revtab[(sreg >> 8) & 0xff]); \
|
||
|
|
_c >>= (16 - _b); \
|
||
|
|
if (_c <= (H)->max_code[_b]) { \
|
||
|
|
_code = (H)->decode[(H)->last[_b] - _c]; \
|
||
|
|
goto _found; \
|
||
|
|
} \
|
||
|
|
} \
|
||
|
|
rv = (E); \
|
||
|
|
goto bail; \
|
||
|
|
} \
|
||
|
|
_code = (_c >> 8); \
|
||
|
|
_found: \
|
||
|
|
INFLATE_CONSUME_BITS(_b); \
|
||
|
|
_code; \
|
||
|
|
})
|
||
|
|
|
||
|
|
static int inflate_blocks(struct inflate* it, int st, inflate_reader read, void* rctx, inflate_writer write, void* wctx) {
|
||
|
|
int rv = 0;
|
||
|
|
uint8_t* is = it->in;
|
||
|
|
uint8_t* ip;
|
||
|
|
uint8_t* ie;
|
||
|
|
uint8_t* os = it->out;
|
||
|
|
uint8_t* op = os;
|
||
|
|
uint8_t* oe = os + sizeof(it->out);
|
||
|
|
bool flushed = false;
|
||
|
|
uint32_t chksum;
|
||
|
|
uint32_t nbits = 0;
|
||
|
|
uint32_t sreg = 0;
|
||
|
|
uint32_t i;
|
||
|
|
uint32_t j;
|
||
|
|
bool final;
|
||
|
|
uint8_t type;
|
||
|
|
|
||
|
|
INFLATE_FILL(-1);
|
||
|
|
|
||
|
|
if (st == INFLATE_ZLIB) {
|
||
|
|
uint16_t header;
|
||
|
|
|
||
|
|
INFLATE_CHECK_LENGTH(-2, 2);
|
||
|
|
|
||
|
|
header = (ip[0] << 8 | ip[1]);
|
||
|
|
ip += 2;
|
||
|
|
|
||
|
|
if ((header % 31) != 0) {
|
||
|
|
rv = -3;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (((header & 0xf000) >> 12) > 7) {
|
||
|
|
rv = -4;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (((header & 0x0f00) >> 8) != 8) {
|
||
|
|
rv = -5;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (((header & 0x0020) >> 5) != 0) {
|
||
|
|
rv = -6;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
chksum = 1;
|
||
|
|
} else if (st == INFLATE_GZIP) {
|
||
|
|
uint8_t flg;
|
||
|
|
|
||
|
|
INFLATE_CHECK_LENGTH(-7, 10);
|
||
|
|
|
||
|
|
if (ip[0] != 0x1f || ip[1] != 0x8b) {
|
||
|
|
rv = -8;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (ip[2] != 8) {
|
||
|
|
rv = -9;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
flg = ip[3];
|
||
|
|
|
||
|
|
if ((flg & 0xe0) != 0) {
|
||
|
|
rv = -10;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
ip += 10;
|
||
|
|
chksum = 0xffffffff;
|
||
|
|
|
||
|
|
if ((flg & 0x04) != 0) {
|
||
|
|
uint16_t xlen;
|
||
|
|
|
||
|
|
INFLATE_CHECK_LENGTH(-11, 2);
|
||
|
|
|
||
|
|
xlen = (ip[0] | ip[1] << 8);
|
||
|
|
ip += 2;
|
||
|
|
|
||
|
|
while (xlen >= (ie - ip)) {
|
||
|
|
xlen -= (ie - ip);
|
||
|
|
|
||
|
|
if ((flg & 0x02) != 0)
|
||
|
|
chksum = crc_32r(is, ie - is, chksum);
|
||
|
|
|
||
|
|
INFLATE_FILL(-12);
|
||
|
|
}
|
||
|
|
|
||
|
|
ip += xlen;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ((flg & 0x08) != 0) {
|
||
|
|
while (ip++[0] != '\0') {
|
||
|
|
if (ip == ie) {
|
||
|
|
if ((flg & 0x02) != 0)
|
||
|
|
chksum = crc_32r(is, ie - is, chksum);
|
||
|
|
|
||
|
|
INFLATE_FILL(-13);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ((flg & 0x10) != 0) {
|
||
|
|
while (ip++[0] != '\0') {
|
||
|
|
if (ip == ie) {
|
||
|
|
if ((flg & 0x02) != 0)
|
||
|
|
chksum = crc_32r(is, ie - is, chksum);
|
||
|
|
|
||
|
|
INFLATE_FILL(-14);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ((flg & 0x02) != 0) {
|
||
|
|
uint16_t crc16;
|
||
|
|
|
||
|
|
INFLATE_CHECK_LENGTH(-15, 2);
|
||
|
|
|
||
|
|
crc16 = (ip[0] | ip[1] << 8);
|
||
|
|
chksum = crc_32r(is, ip - is, chksum);
|
||
|
|
chksum &= 0xffff;
|
||
|
|
chksum ^= 0xffff;
|
||
|
|
|
||
|
|
if (crc16 != chksum) {
|
||
|
|
rv = -16;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
ip += 2;
|
||
|
|
}
|
||
|
|
|
||
|
|
chksum = 0xffffffff;
|
||
|
|
} else {
|
||
|
|
chksum = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
do {
|
||
|
|
INFLATE_FILL_BITS(-17, 3);
|
||
|
|
final = (sreg & 0x01);
|
||
|
|
type = ((sreg & 0x06) >> 1);
|
||
|
|
INFLATE_CONSUME_BITS(3);
|
||
|
|
|
||
|
|
if (type == 0) {
|
||
|
|
uint8_t header[4];
|
||
|
|
uint32_t len;
|
||
|
|
uint32_t clen;
|
||
|
|
|
||
|
|
INFLATE_CONSUME_BITS(nbits & 0x07);
|
||
|
|
|
||
|
|
for (i = 0; i < 4; i++) {
|
||
|
|
if (nbits != 0)
|
||
|
|
INFLATE_EXTRACT_BITS(8, header[i]);
|
||
|
|
else
|
||
|
|
INFLATE_GET_BYTE(-18, header[i]);
|
||
|
|
}
|
||
|
|
|
||
|
|
len = (header[0] | (header[1] << 8));
|
||
|
|
clen = (header[2] | (header[3] << 8)) ^ 0xffff;
|
||
|
|
|
||
|
|
if (len != clen) {
|
||
|
|
rv = -19;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
while (len != 0) {
|
||
|
|
if (ip == ie)
|
||
|
|
INFLATE_FILL(-20);
|
||
|
|
|
||
|
|
if (op == oe)
|
||
|
|
INFLATE_FLUSH(-21);
|
||
|
|
|
||
|
|
j = MIN(len, MIN((uint32_t) (ie - ip), (uint32_t) (oe - op)));
|
||
|
|
for (i = 0; i < j; i++)
|
||
|
|
op[i] = ip[i];
|
||
|
|
|
||
|
|
len -= j;
|
||
|
|
ip += j;
|
||
|
|
op += j;
|
||
|
|
}
|
||
|
|
} else if (type == 3) {
|
||
|
|
rv = -22;
|
||
|
|
goto bail;
|
||
|
|
} else {
|
||
|
|
static const uint8_t revtab[256] =
|
||
|
|
{
|
||
|
|
0x00, 0x80, 0x40, 0xc0, 0x20, 0xa0, 0x60, 0xe0,
|
||
|
|
0x10, 0x90, 0x50, 0xd0, 0x30, 0xb0, 0x70, 0xf0,
|
||
|
|
0x08, 0x88, 0x48, 0xc8, 0x28, 0xa8, 0x68, 0xe8,
|
||
|
|
0x18, 0x98, 0x58, 0xd8, 0x38, 0xb8, 0x78, 0xf8,
|
||
|
|
0x04, 0x84, 0x44, 0xc4, 0x24, 0xa4, 0x64, 0xe4,
|
||
|
|
0x14, 0x94, 0x54, 0xd4, 0x34, 0xb4, 0x74, 0xf4,
|
||
|
|
0x0c, 0x8c, 0x4c, 0xcc, 0x2c, 0xac, 0x6c, 0xec,
|
||
|
|
0x1c, 0x9c, 0x5c, 0xdc, 0x3c, 0xbc, 0x7c, 0xfc,
|
||
|
|
0x02, 0x82, 0x42, 0xc2, 0x22, 0xa2, 0x62, 0xe2,
|
||
|
|
0x12, 0x92, 0x52, 0xd2, 0x32, 0xb2, 0x72, 0xf2,
|
||
|
|
0x0a, 0x8a, 0x4a, 0xca, 0x2a, 0xaa, 0x6a, 0xea,
|
||
|
|
0x1a, 0x9a, 0x5a, 0xda, 0x3a, 0xba, 0x7a, 0xfa,
|
||
|
|
0x06, 0x86, 0x46, 0xc6, 0x26, 0xa6, 0x66, 0xe6,
|
||
|
|
0x16, 0x96, 0x56, 0xd6, 0x36, 0xb6, 0x76, 0xf6,
|
||
|
|
0x0e, 0x8e, 0x4e, 0xce, 0x2e, 0xae, 0x6e, 0xee,
|
||
|
|
0x1e, 0x9e, 0x5e, 0xde, 0x3e, 0xbe, 0x7e, 0xfe,
|
||
|
|
0x01, 0x81, 0x41, 0xc1, 0x21, 0xa1, 0x61, 0xe1,
|
||
|
|
0x11, 0x91, 0x51, 0xd1, 0x31, 0xb1, 0x71, 0xf1,
|
||
|
|
0x09, 0x89, 0x49, 0xc9, 0x29, 0xa9, 0x69, 0xe9,
|
||
|
|
0x19, 0x99, 0x59, 0xd9, 0x39, 0xb9, 0x79, 0xf9,
|
||
|
|
0x05, 0x85, 0x45, 0xc5, 0x25, 0xa5, 0x65, 0xe5,
|
||
|
|
0x15, 0x95, 0x55, 0xd5, 0x35, 0xb5, 0x75, 0xf5,
|
||
|
|
0x0d, 0x8d, 0x4d, 0xcd, 0x2d, 0xad, 0x6d, 0xed,
|
||
|
|
0x1d, 0x9d, 0x5d, 0xdd, 0x3d, 0xbd, 0x7d, 0xfd,
|
||
|
|
0x03, 0x83, 0x43, 0xc3, 0x23, 0xa3, 0x63, 0xe3,
|
||
|
|
0x13, 0x93, 0x53, 0xd3, 0x33, 0xb3, 0x73, 0xf3,
|
||
|
|
0x0b, 0x8b, 0x4b, 0xcb, 0x2b, 0xab, 0x6b, 0xeb,
|
||
|
|
0x1b, 0x9b, 0x5b, 0xdb, 0x3b, 0xbb, 0x7b, 0xfb,
|
||
|
|
0x07, 0x87, 0x47, 0xc7, 0x27, 0xa7, 0x67, 0xe7,
|
||
|
|
0x17, 0x97, 0x57, 0xd7, 0x37, 0xb7, 0x77, 0xf7,
|
||
|
|
0x0f, 0x8f, 0x4f, 0xcf, 0x2f, 0xaf, 0x6f, 0xef,
|
||
|
|
0x1f, 0x9f, 0x5f, 0xdf, 0x3f, 0xbf, 0x7f, 0xff,
|
||
|
|
};
|
||
|
|
uint8_t* tab[3];
|
||
|
|
uint32_t tab_lens[3];
|
||
|
|
uint32_t tab_bits[3];
|
||
|
|
struct inflate_huff* tab_huffs[3] = { &it->lentab, &it->offtab, &it->clentab, };
|
||
|
|
uint32_t c;
|
||
|
|
uint32_t k;
|
||
|
|
|
||
|
|
if (type == 2) {
|
||
|
|
static const uint32_t min_tab_sizes[3] = { 257, 1, 4, };
|
||
|
|
static const uint32_t min_tab_bits[3] = { 5, 5, 4, };
|
||
|
|
static const uint32_t clen_order[INFLATE_CODELEN_MAX] = {
|
||
|
|
16, 17, 18, 0, 8,
|
||
|
|
7, 9, 6, 10, 5,
|
||
|
|
11, 4, 12, 3, 13,
|
||
|
|
2, 14, 1, 15,
|
||
|
|
};
|
||
|
|
|
||
|
|
INFLATE_FILL_BITS(-23, 14);
|
||
|
|
for (i = 0; i < 3; i++) {
|
||
|
|
INFLATE_EXTRACT_BITS(min_tab_bits[i], tab_lens[i]);
|
||
|
|
tab_lens[i] += min_tab_sizes[i];
|
||
|
|
}
|
||
|
|
|
||
|
|
tab[2] = it->clen;
|
||
|
|
for (i = 0; i < INFLATE_CODELEN_MAX; i++)
|
||
|
|
tab[2][i] = 0;
|
||
|
|
for (i = 0; i < tab_lens[2]; i++) {
|
||
|
|
INFLATE_FILL_BITS(-24, 3);
|
||
|
|
INFLATE_EXTRACT_BITS(3, tab[2][clen_order[i]]);
|
||
|
|
}
|
||
|
|
|
||
|
|
tab[0] = it->combo;
|
||
|
|
tab_bits[0] = INFLATE_SYMBOL_BITS;
|
||
|
|
|
||
|
|
tab[1] = tab[0] + tab_lens[0];
|
||
|
|
tab_bits[1] = INFLATE_OFFSET_BITS;
|
||
|
|
|
||
|
|
tab_lens[2] = INFLATE_CODELEN_MAX;
|
||
|
|
tab_bits[2] = INFLATE_CODELEN_BITS;
|
||
|
|
} else {
|
||
|
|
tab[0] = it->len;
|
||
|
|
for (i = 0; i < 144; i++)
|
||
|
|
tab[0][i] = 8;
|
||
|
|
for (; i < 256; i++)
|
||
|
|
tab[0][i] = 9;
|
||
|
|
for (; i < 280; i++)
|
||
|
|
tab[0][i] = 7;
|
||
|
|
for (; i < INFLATE_SYMBOL_MAX; i++)
|
||
|
|
tab[0][i] = 8;
|
||
|
|
tab_lens[0] = INFLATE_SYMBOL_MAX;
|
||
|
|
tab_bits[0] = INFLATE_FLAT_BITS;
|
||
|
|
|
||
|
|
tab[1] = it->off;
|
||
|
|
for (i = 0; i < INFLATE_OFFSET_MAX; i++)
|
||
|
|
tab[1][i] = 5;
|
||
|
|
tab_lens[1] = INFLATE_OFFSET_MAX;
|
||
|
|
tab_bits[1] = INFLATE_FLAT_BITS;
|
||
|
|
}
|
||
|
|
|
||
|
|
for(; type != 0xff; --type) {
|
||
|
|
const uint8_t* hb = tab[type];
|
||
|
|
uint32_t hb_len = tab_lens[type];
|
||
|
|
uint32_t flat_bits = tab_bits[type];
|
||
|
|
struct inflate_huff* h = tab_huffs[type];
|
||
|
|
uint32_t* bits = it->bits;
|
||
|
|
uint32_t* codes = it->codes;
|
||
|
|
uint32_t max_bits = 0;
|
||
|
|
uint32_t min_bits = INFLATE_HUFF_BITS + 1;
|
||
|
|
uint32_t min_code;
|
||
|
|
uint32_t b;
|
||
|
|
uint32_t code;
|
||
|
|
uint32_t fc;
|
||
|
|
uint32_t ec;
|
||
|
|
|
||
|
|
for (i = 0; i < INFLATE_HUFF_BITS; i++)
|
||
|
|
bits[i] = 0;
|
||
|
|
|
||
|
|
for (i = 0; i < hb_len; i++) {
|
||
|
|
b = hb[i];
|
||
|
|
|
||
|
|
if (b != 0) {
|
||
|
|
bits[b]++;
|
||
|
|
|
||
|
|
if (b < min_bits)
|
||
|
|
min_bits = b;
|
||
|
|
|
||
|
|
if (b > max_bits)
|
||
|
|
max_bits = b;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (max_bits == 0) {
|
||
|
|
h->flat_mask = h->min_bits = h->max_bits = 0;
|
||
|
|
goto table_done;
|
||
|
|
}
|
||
|
|
|
||
|
|
h->max_bits = max_bits;
|
||
|
|
|
||
|
|
for (b = c = code = 0; b <= max_bits; b++) {
|
||
|
|
h->last[b] = c;
|
||
|
|
c += bits[b];
|
||
|
|
|
||
|
|
min_code = (code << 1);
|
||
|
|
codes[b] = min_code;
|
||
|
|
code = (min_code + bits[b]);
|
||
|
|
|
||
|
|
if (code > (1U << b)) {
|
||
|
|
rv = -25;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
h->max_code[b] = (code - 1);
|
||
|
|
h->last[b] += (code - 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (flat_bits > max_bits)
|
||
|
|
flat_bits = max_bits;
|
||
|
|
|
||
|
|
h->flat_mask = ((1 << flat_bits) - 1);
|
||
|
|
|
||
|
|
if (min_bits > flat_bits)
|
||
|
|
min_bits = flat_bits;
|
||
|
|
|
||
|
|
h->min_bits = min_bits;
|
||
|
|
|
||
|
|
for (i = 0, b = (1 << flat_bits); i < b; i++)
|
||
|
|
h->flat[i] = 0xffffffff;
|
||
|
|
|
||
|
|
for (b = max_bits; b > flat_bits; b--) {
|
||
|
|
code = h->max_code[b];
|
||
|
|
|
||
|
|
if (code == 0xffffffff)
|
||
|
|
break;
|
||
|
|
|
||
|
|
min_code = ((code + 1) - bits[b]);
|
||
|
|
min_code >>= (b - flat_bits);
|
||
|
|
code >>= (b - flat_bits);
|
||
|
|
|
||
|
|
for (; min_code <= code; min_code++)
|
||
|
|
h->flat[INFLATE_REVERSE(min_code, flat_bits)] = ((b << 8) | 0xff);
|
||
|
|
}
|
||
|
|
|
||
|
|
for (i = 0; i < hb_len; i++) {
|
||
|
|
b = hb[i];
|
||
|
|
|
||
|
|
if (b == 0)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
c = codes[b]++;
|
||
|
|
|
||
|
|
if (b <= flat_bits) {
|
||
|
|
code = ((i << 8) | b);
|
||
|
|
ec = ((c + 1) << (flat_bits - b));
|
||
|
|
|
||
|
|
if (ec > (1U << flat_bits)) {
|
||
|
|
rv = -26;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (fc = (c << (flat_bits - b)); fc < ec; fc++)
|
||
|
|
h->flat[INFLATE_REVERSE(fc, flat_bits)] = code;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (b > min_bits) {
|
||
|
|
c = h->last[b] - c;
|
||
|
|
|
||
|
|
if (c >= hb_len) {
|
||
|
|
rv = -27;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
h->decode[c] = i;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
table_done:
|
||
|
|
|
||
|
|
if (type == 2) {
|
||
|
|
static const uint32_t bits[3] = { 2, 3, 7, };
|
||
|
|
static const uint32_t bases[3] = { 3, 3, 11, };
|
||
|
|
|
||
|
|
for (i = 0, j = tab_lens[0] + tab_lens[1]; i < j; ) {
|
||
|
|
uint32_t len;
|
||
|
|
uint8_t byte;
|
||
|
|
|
||
|
|
INFLATE_FILL_BITS(-28, h->max_bits);
|
||
|
|
|
||
|
|
c = INFLATE_DECODE(-29, h);
|
||
|
|
|
||
|
|
if (c < 16) {
|
||
|
|
tab[0][i++] = c;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
c -= 16;
|
||
|
|
|
||
|
|
if (c == 0 && i == 0) {
|
||
|
|
rv = -30;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
INFLATE_FILL_BITS(-31, bits[c]);
|
||
|
|
INFLATE_EXTRACT_BITS(bits[c], len);
|
||
|
|
len += bases[c];
|
||
|
|
|
||
|
|
if ((i + len) > j) {
|
||
|
|
rv = -32;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
byte = ((c == 0) ? tab[0][i - 1] : 0);
|
||
|
|
for (k = 0; k < len; k++)
|
||
|
|
tab[0][i + k] = byte;
|
||
|
|
|
||
|
|
i += len;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
static const uint32_t lenbase[INFLATE_SYMBOL_MAX - 257] = {
|
||
|
|
3, 4, 5, 6, 7, 8, 9, 10,
|
||
|
|
11, 13, 15, 17, 19, 23, 27, 31,
|
||
|
|
35, 43, 51, 59, 67, 83, 99, 115,
|
||
|
|
131, 163, 195, 227, 258, 0, 0,
|
||
|
|
};
|
||
|
|
static const uint32_t lenextra[INFLATE_SYMBOL_MAX - 257] = {
|
||
|
|
0, 0, 0, 0, 0, 0, 0, 0,
|
||
|
|
1, 1, 1, 1, 2, 2, 2, 2,
|
||
|
|
3, 3, 3, 3, 4, 4, 4, 4,
|
||
|
|
5, 5, 5, 5, 0, 0, 0,
|
||
|
|
};
|
||
|
|
static const uint32_t offbase[INFLATE_OFFSET_MAX] = {
|
||
|
|
1, 2, 3, 4, 5, 7, 9, 13,
|
||
|
|
17, 25, 33, 49, 65, 97, 129, 193,
|
||
|
|
257, 385, 513, 769, 1025, 1537, 2049, 3073,
|
||
|
|
4097, 6145, 8193, 12289, 16385, 24577, 0, 0,
|
||
|
|
};
|
||
|
|
static const uint32_t offextra[INFLATE_OFFSET_MAX] = {
|
||
|
|
0, 0, 0, 0, 1, 1, 2, 2,
|
||
|
|
3, 3, 4, 4, 5, 5, 6, 6,
|
||
|
|
7, 7, 8, 8, 9, 9, 10, 10,
|
||
|
|
11, 11, 12, 12, 13, 13, 0, 0,
|
||
|
|
};
|
||
|
|
uint32_t len;
|
||
|
|
uint32_t off;
|
||
|
|
uint8_t* cp;
|
||
|
|
|
||
|
|
INFLATE_FILL_BITS(-33, tab_huffs[0]->max_bits);
|
||
|
|
c = INFLATE_DECODE(-34, tab_huffs[0]);
|
||
|
|
|
||
|
|
if (c < 256) {
|
||
|
|
INFLATE_PUT_BYTE(-35, c);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (c == 256)
|
||
|
|
break;
|
||
|
|
|
||
|
|
if (c > 285) {
|
||
|
|
rv = -36;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
c -= 257;
|
||
|
|
INFLATE_FILL_BITS(-37, lenextra[c]);
|
||
|
|
INFLATE_EXTRACT_BITS(lenextra[c], len);
|
||
|
|
len += lenbase[c];
|
||
|
|
|
||
|
|
INFLATE_FILL_BITS(-38, tab_huffs[1]->max_bits);
|
||
|
|
c = INFLATE_DECODE(-39, tab_huffs[1]);
|
||
|
|
|
||
|
|
if (c > 29) {
|
||
|
|
rv = -40;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
INFLATE_FILL_BITS(-41, offextra[c]);
|
||
|
|
INFLATE_EXTRACT_BITS(offextra[c], off);
|
||
|
|
off += offbase[c];
|
||
|
|
|
||
|
|
cp = op - off;
|
||
|
|
if (cp < os) {
|
||
|
|
if (!flushed) {
|
||
|
|
rv = -42;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
|
||
|
|
cp += sizeof(it->out);
|
||
|
|
}
|
||
|
|
|
||
|
|
while (len != 0) {
|
||
|
|
if (op == oe)
|
||
|
|
INFLATE_FLUSH(-43);
|
||
|
|
|
||
|
|
if (cp == oe)
|
||
|
|
cp = os;
|
||
|
|
|
||
|
|
j = MIN(len, MIN((uint32_t) (oe - op), (uint32_t) (oe - cp)));
|
||
|
|
|
||
|
|
for (i = 0; i < j; i++)
|
||
|
|
op[i] = cp[i];
|
||
|
|
|
||
|
|
op += j;
|
||
|
|
cp += j;
|
||
|
|
len -= j;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} while (!final);
|
||
|
|
|
||
|
|
INFLATE_FLUSH(-44);
|
||
|
|
|
||
|
|
if (st != INFLATE_RAW) {
|
||
|
|
uint8_t header[4];
|
||
|
|
uint32_t chksum2;
|
||
|
|
|
||
|
|
INFLATE_CONSUME_BITS(nbits & 0x07);
|
||
|
|
|
||
|
|
for (i = 0; i < 4; i++) {
|
||
|
|
if (nbits != 0)
|
||
|
|
INFLATE_EXTRACT_BITS(8, header[i]);
|
||
|
|
else
|
||
|
|
INFLATE_GET_BYTE(-45, header[i]);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (st == INFLATE_ZLIB) {
|
||
|
|
chksum2 = ((header[0] << 24) | (header[1] << 16) | (header[2] << 8) | (header[3] << 0));
|
||
|
|
|
||
|
|
if (chksum != chksum2) {
|
||
|
|
rv = -46;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
} else if (st == INFLATE_GZIP) {
|
||
|
|
chksum2 = ((header[3] << 24) | (header[2] << 16) | (header[1] << 8) | (header[0] << 0));
|
||
|
|
|
||
|
|
if (~chksum != chksum2) {
|
||
|
|
rv = -47;
|
||
|
|
goto bail;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bail:
|
||
|
|
return rv;
|
||
|
|
}
|
||
|
|
|
||
|
|
const uint32_t inflate_size = sizeof(struct inflate);
|
||
|
|
const uint32_t inflate_align = _Alignof(struct inflate);
|
||
|
|
|
||
|
|
int inflate(struct inflate* it, int st, inflate_reader read, void* rctx, inflate_writer write, void* wctx) {
|
||
|
|
if (it == NULL || read == NULL || write == NULL || st < 0 || st > 2 || (((uintptr_t) it) & (_Alignof(struct inflate) - 1)) != 0)
|
||
|
|
return -48;
|
||
|
|
|
||
|
|
return inflate_blocks(it, st, read, rctx, write, wctx);
|
||
|
|
}
|