mirror of
https://github.com/CloverHackyColor/CloverBootloader.git
synced 2024-12-11 14:28:08 +01:00
2501 lines
89 KiB
C
2501 lines
89 KiB
C
|
/* Copyright 2013 Google Inc. All Rights Reserved.
|
||
|
|
||
|
Distributed under MIT license.
|
||
|
See file LICENSE for detail or copy at https://opensource.org/licenses/MIT
|
||
|
*/
|
||
|
|
||
|
#include <brotli/decode.h>
|
||
|
|
||
|
#if defined(__ARM_NEON__)
|
||
|
#include <arm_neon.h>
|
||
|
#endif
|
||
|
|
||
|
//#include <stdlib.h> /* free, malloc */
|
||
|
//#include <string.h> /* memcpy, memset */
|
||
|
|
||
|
#include "../common/constants.h"
|
||
|
#include "../common/context.h"
|
||
|
#include "../common/dictionary.h"
|
||
|
#include "../common/platform.h"
|
||
|
#include "../common/transform.h"
|
||
|
#include "../common/version.h"
|
||
|
#include "./bit_reader.h"
|
||
|
#include "./huffman.h"
|
||
|
#include "./prefix.h"
|
||
|
#include "./state.h"
|
||
|
|
||
|
#if defined(__cplusplus) || defined(c_plusplus)
|
||
|
extern "C" {
|
||
|
#endif
|
||
|
|
||
|
#define BROTLI_FAILURE(CODE) (BROTLI_DUMP(), CODE)
|
||
|
|
||
|
#define BROTLI_LOG_UINT(name) \
|
||
|
BROTLI_LOG(("[%s] %s = %lu\n", __func__, #name, (unsigned long)(name)))
|
||
|
#define BROTLI_LOG_ARRAY_INDEX(array_name, idx) \
|
||
|
BROTLI_LOG(("[%s] %s[%lu] = %lu\n", __func__, #array_name, \
|
||
|
(unsigned long)(idx), (unsigned long)array_name[idx]))
|
||
|
|
||
|
#define HUFFMAN_TABLE_BITS 8U
|
||
|
#define HUFFMAN_TABLE_MASK 0xFF
|
||
|
|
||
|
/* We need the slack region for the following reasons:
|
||
|
- doing up to two 16-byte copies for fast backward copying
|
||
|
- inserting transformed dictionary word (5 prefix + 24 base + 8 suffix) */
|
||
|
static const uint32_t kRingBufferWriteAheadSlack = 42;
|
||
|
|
||
|
static const uint8_t kCodeLengthCodeOrder[BROTLI_CODE_LENGTH_CODES] = {
|
||
|
1, 2, 3, 4, 0, 5, 17, 6, 16, 7, 8, 9, 10, 11, 12, 13, 14, 15,
|
||
|
};
|
||
|
|
||
|
/* Static prefix code for the complex code length code lengths. */
|
||
|
static const uint8_t kCodeLengthPrefixLength[16] = {
|
||
|
2, 2, 2, 3, 2, 2, 2, 4, 2, 2, 2, 3, 2, 2, 2, 4,
|
||
|
};
|
||
|
|
||
|
static const uint8_t kCodeLengthPrefixValue[16] = {
|
||
|
0, 4, 3, 2, 0, 4, 3, 1, 0, 4, 3, 2, 0, 4, 3, 5,
|
||
|
};
|
||
|
|
||
|
BROTLI_BOOL BrotliDecoderSetParameter(
|
||
|
BrotliDecoderState* state, BrotliDecoderParameter p, uint32_t value) {
|
||
|
if (state->state != BROTLI_STATE_UNINITED) return BROTLI_FALSE;
|
||
|
switch (p) {
|
||
|
case BROTLI_DECODER_PARAM_DISABLE_RING_BUFFER_REALLOCATION:
|
||
|
state->canny_ringbuffer_allocation = !!value ? 0 : 1;
|
||
|
return BROTLI_TRUE;
|
||
|
|
||
|
case BROTLI_DECODER_PARAM_LARGE_WINDOW:
|
||
|
state->large_window = TO_BROTLI_BOOL(!!value);
|
||
|
return BROTLI_TRUE;
|
||
|
|
||
|
default: return BROTLI_FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
BrotliDecoderState* BrotliDecoderCreateInstance(
|
||
|
brotli_alloc_func alloc_func, brotli_free_func free_func, void* opaque) {
|
||
|
BrotliDecoderState* state = 0;
|
||
|
if (!alloc_func && !free_func) {
|
||
|
state = (BrotliDecoderState*)BrDummyMalloc(sizeof(BrotliDecoderState));
|
||
|
} else if (alloc_func && free_func) {
|
||
|
state = (BrotliDecoderState*)alloc_func(opaque, sizeof(BrotliDecoderState));
|
||
|
}
|
||
|
if (state == 0) {
|
||
|
BROTLI_DUMP();
|
||
|
return 0;
|
||
|
}
|
||
|
if (!BrotliDecoderStateInit(state, alloc_func, free_func, opaque)) {
|
||
|
BROTLI_DUMP();
|
||
|
if (!alloc_func && !free_func) {
|
||
|
BrDummyFree(state);
|
||
|
} else if (alloc_func && free_func) {
|
||
|
free_func(opaque, state);
|
||
|
}
|
||
|
return 0;
|
||
|
}
|
||
|
return state;
|
||
|
}
|
||
|
|
||
|
/* Deinitializes and frees BrotliDecoderState instance. */
|
||
|
void BrotliDecoderDestroyInstance(BrotliDecoderState* state) {
|
||
|
if (!state) {
|
||
|
return;
|
||
|
} else {
|
||
|
brotli_free_func free_func = state->free_func;
|
||
|
void* opaque = state->memory_manager_opaque;
|
||
|
BrotliDecoderStateCleanup(state);
|
||
|
free_func(opaque, state);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Saves error code and converts it to BrotliDecoderResult. */
|
||
|
static BROTLI_NOINLINE BrotliDecoderResult SaveErrorCode(
|
||
|
BrotliDecoderState* s, BrotliDecoderErrorCode e) {
|
||
|
s->error_code = (int)e;
|
||
|
switch (e) {
|
||
|
case BROTLI_DECODER_SUCCESS:
|
||
|
return BROTLI_DECODER_RESULT_SUCCESS;
|
||
|
|
||
|
case BROTLI_DECODER_NEEDS_MORE_INPUT:
|
||
|
return BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT;
|
||
|
|
||
|
case BROTLI_DECODER_NEEDS_MORE_OUTPUT:
|
||
|
return BROTLI_DECODER_RESULT_NEEDS_MORE_OUTPUT;
|
||
|
|
||
|
default:
|
||
|
return BROTLI_DECODER_RESULT_ERROR;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Decodes WBITS by reading 1 - 7 bits, or 0x11 for "Large Window Brotli".
|
||
|
Precondition: bit-reader accumulator has at least 8 bits. */
|
||
|
static BrotliDecoderErrorCode DecodeWindowBits(BrotliDecoderState* s,
|
||
|
BrotliBitReader* br) {
|
||
|
uint32_t n;
|
||
|
BROTLI_BOOL large_window = s->large_window;
|
||
|
s->large_window = BROTLI_FALSE;
|
||
|
BrotliTakeBits(br, 1, &n);
|
||
|
if (n == 0) {
|
||
|
s->window_bits = 16;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
BrotliTakeBits(br, 3, &n);
|
||
|
if (n != 0) {
|
||
|
s->window_bits = 17 + n;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
BrotliTakeBits(br, 3, &n);
|
||
|
if (n == 1) {
|
||
|
if (large_window) {
|
||
|
BrotliTakeBits(br, 1, &n);
|
||
|
if (n == 1) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_WINDOW_BITS);
|
||
|
}
|
||
|
s->large_window = BROTLI_TRUE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
} else {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_WINDOW_BITS);
|
||
|
}
|
||
|
}
|
||
|
if (n != 0) {
|
||
|
s->window_bits = 8 + n;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
s->window_bits = 17;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE void memmove16(uint8_t* dst, uint8_t* src) {
|
||
|
#if defined(__ARM_NEON__)
|
||
|
vst1q_u8(dst, vld1q_u8(src));
|
||
|
#else
|
||
|
uint32_t buffer[4];
|
||
|
memcpy(buffer, src, 16);
|
||
|
memcpy(dst, buffer, 16);
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
/* Decodes a number in the range [0..255], by reading 1 - 11 bits. */
|
||
|
static BROTLI_NOINLINE BrotliDecoderErrorCode DecodeVarLenUint8(
|
||
|
BrotliDecoderState* s, BrotliBitReader* br, uint32_t* value) {
|
||
|
uint32_t bits;
|
||
|
switch (s->substate_decode_uint8) {
|
||
|
case BROTLI_STATE_DECODE_UINT8_NONE:
|
||
|
if (BROTLI_PREDICT_FALSE(!BrotliSafeReadBits(br, 1, &bits))) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (bits == 0) {
|
||
|
*value = 0;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_DECODE_UINT8_SHORT:
|
||
|
if (BROTLI_PREDICT_FALSE(!BrotliSafeReadBits(br, 3, &bits))) {
|
||
|
s->substate_decode_uint8 = BROTLI_STATE_DECODE_UINT8_SHORT;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (bits == 0) {
|
||
|
*value = 1;
|
||
|
s->substate_decode_uint8 = BROTLI_STATE_DECODE_UINT8_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
/* Use output value as a temporary storage. It MUST be persisted. */
|
||
|
*value = bits;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_DECODE_UINT8_LONG:
|
||
|
if (BROTLI_PREDICT_FALSE(!BrotliSafeReadBits(br, *value, &bits))) {
|
||
|
s->substate_decode_uint8 = BROTLI_STATE_DECODE_UINT8_LONG;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
*value = (1U << *value) + bits;
|
||
|
s->substate_decode_uint8 = BROTLI_STATE_DECODE_UINT8_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
|
||
|
default:
|
||
|
return
|
||
|
BROTLI_FAILURE(BROTLI_DECODER_ERROR_UNREACHABLE);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Decodes a metablock length and flags by reading 2 - 31 bits. */
|
||
|
static BrotliDecoderErrorCode BROTLI_NOINLINE DecodeMetaBlockLength(
|
||
|
BrotliDecoderState* s, BrotliBitReader* br) {
|
||
|
uint32_t bits;
|
||
|
unsigned int i;
|
||
|
for (;;) {
|
||
|
switch (s->substate_metablock_header) {
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_NONE:
|
||
|
if (!BrotliSafeReadBits(br, 1, &bits)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
s->is_last_metablock = bits ? 1 : 0;
|
||
|
s->meta_block_remaining_len = 0;
|
||
|
s->is_uncompressed = 0;
|
||
|
s->is_metadata = 0;
|
||
|
if (!s->is_last_metablock) {
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_NIBBLES;
|
||
|
break;
|
||
|
}
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_EMPTY;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_EMPTY:
|
||
|
if (!BrotliSafeReadBits(br, 1, &bits)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (bits) {
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_NIBBLES;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_NIBBLES:
|
||
|
if (!BrotliSafeReadBits(br, 2, &bits)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
s->size_nibbles = (uint8_t)(bits + 4);
|
||
|
s->loop_counter = 0;
|
||
|
if (bits == 3) {
|
||
|
s->is_metadata = 1;
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_RESERVED;
|
||
|
break;
|
||
|
}
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_SIZE;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_SIZE:
|
||
|
i = s->loop_counter;
|
||
|
for (; i < (int)s->size_nibbles; ++i) {
|
||
|
if (!BrotliSafeReadBits(br, 4, &bits)) {
|
||
|
s->loop_counter = i;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (i + 1 == s->size_nibbles && s->size_nibbles > 4 && bits == 0) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_EXUBERANT_NIBBLE);
|
||
|
}
|
||
|
s->meta_block_remaining_len |= (int)(bits << (i * 4));
|
||
|
}
|
||
|
s->substate_metablock_header =
|
||
|
BROTLI_STATE_METABLOCK_HEADER_UNCOMPRESSED;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_UNCOMPRESSED:
|
||
|
if (!s->is_last_metablock) {
|
||
|
if (!BrotliSafeReadBits(br, 1, &bits)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
s->is_uncompressed = bits ? 1 : 0;
|
||
|
}
|
||
|
++s->meta_block_remaining_len;
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_RESERVED:
|
||
|
if (!BrotliSafeReadBits(br, 1, &bits)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (bits != 0) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_RESERVED);
|
||
|
}
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_BYTES;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_BYTES:
|
||
|
if (!BrotliSafeReadBits(br, 2, &bits)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (bits == 0) {
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
s->size_nibbles = (uint8_t)bits;
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_METADATA;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_METADATA:
|
||
|
i = s->loop_counter;
|
||
|
for (; i < (int)s->size_nibbles; ++i) {
|
||
|
if (!BrotliSafeReadBits(br, 8, &bits)) {
|
||
|
s->loop_counter = i;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (i + 1 == s->size_nibbles && s->size_nibbles > 1 && bits == 0) {
|
||
|
return BROTLI_FAILURE(
|
||
|
BROTLI_DECODER_ERROR_FORMAT_EXUBERANT_META_NIBBLE);
|
||
|
}
|
||
|
s->meta_block_remaining_len |= (int)(bits << (i * 8));
|
||
|
}
|
||
|
++s->meta_block_remaining_len;
|
||
|
s->substate_metablock_header = BROTLI_STATE_METABLOCK_HEADER_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
|
||
|
default:
|
||
|
return
|
||
|
BROTLI_FAILURE(BROTLI_DECODER_ERROR_UNREACHABLE);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Decodes the Huffman code.
|
||
|
This method doesn't read data from the bit reader, BUT drops the amount of
|
||
|
bits that correspond to the decoded symbol.
|
||
|
bits MUST contain at least 15 (BROTLI_HUFFMAN_MAX_CODE_LENGTH) valid bits. */
|
||
|
static BROTLI_INLINE uint32_t DecodeSymbol(uint32_t bits,
|
||
|
const HuffmanCode* table,
|
||
|
BrotliBitReader* br) {
|
||
|
table += bits & HUFFMAN_TABLE_MASK;
|
||
|
if (table->bits > HUFFMAN_TABLE_BITS) {
|
||
|
uint32_t nbits = table->bits - HUFFMAN_TABLE_BITS;
|
||
|
BrotliDropBits(br, HUFFMAN_TABLE_BITS);
|
||
|
table += table->value;
|
||
|
table += (bits >> HUFFMAN_TABLE_BITS) & BitMask(nbits);
|
||
|
}
|
||
|
BrotliDropBits(br, table->bits);
|
||
|
return table->value;
|
||
|
}
|
||
|
|
||
|
/* Reads and decodes the next Huffman code from bit-stream.
|
||
|
This method peeks 16 bits of input and drops 0 - 15 of them. */
|
||
|
static BROTLI_INLINE uint32_t ReadSymbol(const HuffmanCode* table,
|
||
|
BrotliBitReader* br) {
|
||
|
return DecodeSymbol(BrotliGet16BitsUnmasked(br), table, br);
|
||
|
}
|
||
|
|
||
|
/* Same as DecodeSymbol, but it is known that there is less than 15 bits of
|
||
|
input are currently available. */
|
||
|
static BROTLI_NOINLINE BROTLI_BOOL SafeDecodeSymbol(
|
||
|
const HuffmanCode* table, BrotliBitReader* br, uint32_t* result) {
|
||
|
uint32_t val;
|
||
|
uint32_t available_bits = BrotliGetAvailableBits(br);
|
||
|
if (available_bits == 0) {
|
||
|
if (table->bits == 0) {
|
||
|
*result = table->value;
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
return BROTLI_FALSE; /* No valid bits at all. */
|
||
|
}
|
||
|
val = (uint32_t)BrotliGetBitsUnmasked(br);
|
||
|
table += val & HUFFMAN_TABLE_MASK;
|
||
|
if (table->bits <= HUFFMAN_TABLE_BITS) {
|
||
|
if (table->bits <= available_bits) {
|
||
|
BrotliDropBits(br, table->bits);
|
||
|
*result = table->value;
|
||
|
return BROTLI_TRUE;
|
||
|
} else {
|
||
|
return BROTLI_FALSE; /* Not enough bits for the first level. */
|
||
|
}
|
||
|
}
|
||
|
if (available_bits <= HUFFMAN_TABLE_BITS) {
|
||
|
return BROTLI_FALSE; /* Not enough bits to move to the second level. */
|
||
|
}
|
||
|
|
||
|
/* Speculatively drop HUFFMAN_TABLE_BITS. */
|
||
|
val = (val & BitMask(table->bits)) >> HUFFMAN_TABLE_BITS;
|
||
|
available_bits -= HUFFMAN_TABLE_BITS;
|
||
|
table += table->value + val;
|
||
|
if (available_bits < table->bits) {
|
||
|
return BROTLI_FALSE; /* Not enough bits for the second level. */
|
||
|
}
|
||
|
|
||
|
BrotliDropBits(br, HUFFMAN_TABLE_BITS + table->bits);
|
||
|
*result = table->value;
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE BROTLI_BOOL SafeReadSymbol(
|
||
|
const HuffmanCode* table, BrotliBitReader* br, uint32_t* result) {
|
||
|
uint32_t val;
|
||
|
if (BROTLI_PREDICT_TRUE(BrotliSafeGetBits(br, 15, &val))) {
|
||
|
*result = DecodeSymbol(val, table, br);
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
return SafeDecodeSymbol(table, br, result);
|
||
|
}
|
||
|
|
||
|
/* Makes a look-up in first level Huffman table. Peeks 8 bits. */
|
||
|
static BROTLI_INLINE void PreloadSymbol(int safe,
|
||
|
const HuffmanCode* table,
|
||
|
BrotliBitReader* br,
|
||
|
uint32_t* bits,
|
||
|
uint32_t* value) {
|
||
|
if (safe) {
|
||
|
return;
|
||
|
}
|
||
|
table += BrotliGetBits(br, HUFFMAN_TABLE_BITS);
|
||
|
*bits = table->bits;
|
||
|
*value = table->value;
|
||
|
}
|
||
|
|
||
|
/* Decodes the next Huffman code using data prepared by PreloadSymbol.
|
||
|
Reads 0 - 15 bits. Also peeks 8 following bits. */
|
||
|
static BROTLI_INLINE uint32_t ReadPreloadedSymbol(const HuffmanCode* table,
|
||
|
BrotliBitReader* br,
|
||
|
uint32_t* bits,
|
||
|
uint32_t* value) {
|
||
|
uint32_t result = *value;
|
||
|
if (BROTLI_PREDICT_FALSE(*bits > HUFFMAN_TABLE_BITS)) {
|
||
|
uint32_t val = BrotliGet16BitsUnmasked(br);
|
||
|
const HuffmanCode* ext = table + (val & HUFFMAN_TABLE_MASK) + *value;
|
||
|
uint32_t mask = BitMask((*bits - HUFFMAN_TABLE_BITS));
|
||
|
BrotliDropBits(br, HUFFMAN_TABLE_BITS);
|
||
|
ext += (val >> HUFFMAN_TABLE_BITS) & mask;
|
||
|
BrotliDropBits(br, ext->bits);
|
||
|
result = ext->value;
|
||
|
} else {
|
||
|
BrotliDropBits(br, *bits);
|
||
|
}
|
||
|
PreloadSymbol(0, table, br, bits, value);
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE uint32_t Log2Floor(uint32_t x) {
|
||
|
uint32_t result = 0;
|
||
|
while (x) {
|
||
|
x >>= 1;
|
||
|
++result;
|
||
|
}
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
/* Reads (s->symbol + 1) symbols.
|
||
|
Totally 1..4 symbols are read, 1..11 bits each.
|
||
|
The list of symbols MUST NOT contain duplicates. */
|
||
|
static BrotliDecoderErrorCode ReadSimpleHuffmanSymbols(
|
||
|
uint32_t alphabet_size, uint32_t max_symbol, BrotliDecoderState* s) {
|
||
|
/* max_bits == 1..11; symbol == 0..3; 1..44 bits will be read. */
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
uint32_t max_bits = Log2Floor(alphabet_size - 1);
|
||
|
uint32_t i = s->sub_loop_counter;
|
||
|
uint32_t num_symbols = s->symbol;
|
||
|
while (i <= num_symbols) {
|
||
|
uint32_t v;
|
||
|
if (BROTLI_PREDICT_FALSE(!BrotliSafeReadBits(br, max_bits, &v))) {
|
||
|
s->sub_loop_counter = i;
|
||
|
s->substate_huffman = BROTLI_STATE_HUFFMAN_SIMPLE_READ;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (v >= max_symbol) {
|
||
|
return
|
||
|
BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_SIMPLE_HUFFMAN_ALPHABET);
|
||
|
}
|
||
|
s->symbols_lists_array[i] = (uint16_t)v;
|
||
|
BROTLI_LOG_UINT(s->symbols_lists_array[i]);
|
||
|
++i;
|
||
|
}
|
||
|
|
||
|
for (i = 0; i < num_symbols; ++i) {
|
||
|
uint32_t k = i + 1;
|
||
|
for (; k <= num_symbols; ++k) {
|
||
|
if (s->symbols_lists_array[i] == s->symbols_lists_array[k]) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_SIMPLE_HUFFMAN_SAME);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
/* Process single decoded symbol code length:
|
||
|
A) reset the repeat variable
|
||
|
B) remember code length (if it is not 0)
|
||
|
C) extend corresponding index-chain
|
||
|
D) reduce the Huffman space
|
||
|
E) update the histogram */
|
||
|
static BROTLI_INLINE void ProcessSingleCodeLength(uint32_t code_len,
|
||
|
uint32_t* symbol, uint32_t* repeat, uint32_t* space,
|
||
|
uint32_t* prev_code_len, uint16_t* symbol_lists,
|
||
|
uint16_t* code_length_histo, int* next_symbol) {
|
||
|
*repeat = 0;
|
||
|
if (code_len != 0) { /* code_len == 1..15 */
|
||
|
symbol_lists[next_symbol[code_len]] = (uint16_t)(*symbol);
|
||
|
next_symbol[code_len] = (int)(*symbol);
|
||
|
*prev_code_len = code_len;
|
||
|
*space -= 32768U >> code_len;
|
||
|
code_length_histo[code_len]++;
|
||
|
BROTLI_LOG(("[ReadHuffmanCode] code_length[%d] = %d\n",
|
||
|
(int)*symbol, (int)code_len));
|
||
|
}
|
||
|
(*symbol)++;
|
||
|
}
|
||
|
|
||
|
/* Process repeated symbol code length.
|
||
|
A) Check if it is the extension of previous repeat sequence; if the decoded
|
||
|
value is not BROTLI_REPEAT_PREVIOUS_CODE_LENGTH, then it is a new
|
||
|
symbol-skip
|
||
|
B) Update repeat variable
|
||
|
C) Check if operation is feasible (fits alphabet)
|
||
|
D) For each symbol do the same operations as in ProcessSingleCodeLength
|
||
|
|
||
|
PRECONDITION: code_len == BROTLI_REPEAT_PREVIOUS_CODE_LENGTH or
|
||
|
code_len == BROTLI_REPEAT_ZERO_CODE_LENGTH */
|
||
|
static BROTLI_INLINE void ProcessRepeatedCodeLength(uint32_t code_len,
|
||
|
uint32_t repeat_delta, uint32_t alphabet_size, uint32_t* symbol,
|
||
|
uint32_t* repeat, uint32_t* space, uint32_t* prev_code_len,
|
||
|
uint32_t* repeat_code_len, uint16_t* symbol_lists,
|
||
|
uint16_t* code_length_histo, int* next_symbol) {
|
||
|
uint32_t old_repeat;
|
||
|
uint32_t extra_bits = 3; /* for BROTLI_REPEAT_ZERO_CODE_LENGTH */
|
||
|
uint32_t new_len = 0; /* for BROTLI_REPEAT_ZERO_CODE_LENGTH */
|
||
|
if (code_len == BROTLI_REPEAT_PREVIOUS_CODE_LENGTH) {
|
||
|
new_len = *prev_code_len;
|
||
|
extra_bits = 2;
|
||
|
}
|
||
|
if (*repeat_code_len != new_len) {
|
||
|
*repeat = 0;
|
||
|
*repeat_code_len = new_len;
|
||
|
}
|
||
|
old_repeat = *repeat;
|
||
|
if (*repeat > 0) {
|
||
|
*repeat -= 2;
|
||
|
*repeat <<= extra_bits;
|
||
|
}
|
||
|
*repeat += repeat_delta + 3U;
|
||
|
repeat_delta = *repeat - old_repeat;
|
||
|
if (*symbol + repeat_delta > alphabet_size) {
|
||
|
BROTLI_DUMP();
|
||
|
*symbol = alphabet_size;
|
||
|
*space = 0xFFFFF;
|
||
|
return;
|
||
|
}
|
||
|
BROTLI_LOG(("[ReadHuffmanCode] code_length[%d..%d] = %d\n",
|
||
|
(int)*symbol, (int)(*symbol + repeat_delta - 1), (int)*repeat_code_len));
|
||
|
if (*repeat_code_len != 0) {
|
||
|
unsigned last = *symbol + repeat_delta;
|
||
|
int next = next_symbol[*repeat_code_len];
|
||
|
do {
|
||
|
symbol_lists[next] = (uint16_t)*symbol;
|
||
|
next = (int)*symbol;
|
||
|
} while (++(*symbol) != last);
|
||
|
next_symbol[*repeat_code_len] = next;
|
||
|
*space -= repeat_delta << (15 - *repeat_code_len);
|
||
|
code_length_histo[*repeat_code_len] =
|
||
|
(uint16_t)(code_length_histo[*repeat_code_len] + repeat_delta);
|
||
|
} else {
|
||
|
*symbol += repeat_delta;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Reads and decodes symbol codelengths. */
|
||
|
static BrotliDecoderErrorCode ReadSymbolCodeLengths(
|
||
|
uint32_t alphabet_size, BrotliDecoderState* s) {
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
uint32_t symbol = s->symbol;
|
||
|
uint32_t repeat = s->repeat;
|
||
|
uint32_t space = s->space;
|
||
|
uint32_t prev_code_len = s->prev_code_len;
|
||
|
uint32_t repeat_code_len = s->repeat_code_len;
|
||
|
uint16_t* symbol_lists = s->symbol_lists;
|
||
|
uint16_t* code_length_histo = s->code_length_histo;
|
||
|
int* next_symbol = s->next_symbol;
|
||
|
if (!BrotliWarmupBitReader(br)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
while (symbol < alphabet_size && space > 0) {
|
||
|
const HuffmanCode* p = s->table;
|
||
|
uint32_t code_len;
|
||
|
if (!BrotliCheckInputAmount(br, BROTLI_SHORT_FILL_BIT_WINDOW_READ)) {
|
||
|
s->symbol = symbol;
|
||
|
s->repeat = repeat;
|
||
|
s->prev_code_len = prev_code_len;
|
||
|
s->repeat_code_len = repeat_code_len;
|
||
|
s->space = space;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
BrotliFillBitWindow16(br);
|
||
|
p += BrotliGetBitsUnmasked(br) &
|
||
|
BitMask(BROTLI_HUFFMAN_MAX_CODE_LENGTH_CODE_LENGTH);
|
||
|
BrotliDropBits(br, p->bits); /* Use 1..5 bits. */
|
||
|
code_len = p->value; /* code_len == 0..17 */
|
||
|
if (code_len < BROTLI_REPEAT_PREVIOUS_CODE_LENGTH) {
|
||
|
ProcessSingleCodeLength(code_len, &symbol, &repeat, &space,
|
||
|
&prev_code_len, symbol_lists, code_length_histo, next_symbol);
|
||
|
} else { /* code_len == 16..17, extra_bits == 2..3 */
|
||
|
uint32_t extra_bits =
|
||
|
(code_len == BROTLI_REPEAT_PREVIOUS_CODE_LENGTH) ? 2 : 3;
|
||
|
uint32_t repeat_delta =
|
||
|
(uint32_t)BrotliGetBitsUnmasked(br) & BitMask(extra_bits);
|
||
|
BrotliDropBits(br, extra_bits);
|
||
|
ProcessRepeatedCodeLength(code_len, repeat_delta, alphabet_size,
|
||
|
&symbol, &repeat, &space, &prev_code_len, &repeat_code_len,
|
||
|
symbol_lists, code_length_histo, next_symbol);
|
||
|
}
|
||
|
}
|
||
|
s->space = space;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
static BrotliDecoderErrorCode SafeReadSymbolCodeLengths(
|
||
|
uint32_t alphabet_size, BrotliDecoderState* s) {
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
BROTLI_BOOL get_byte = BROTLI_FALSE;
|
||
|
while (s->symbol < alphabet_size && s->space > 0) {
|
||
|
const HuffmanCode* p = s->table;
|
||
|
uint32_t code_len;
|
||
|
uint32_t available_bits;
|
||
|
uint32_t bits = 0;
|
||
|
if (get_byte && !BrotliPullByte(br)) return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
get_byte = BROTLI_FALSE;
|
||
|
available_bits = BrotliGetAvailableBits(br);
|
||
|
if (available_bits != 0) {
|
||
|
bits = (uint32_t)BrotliGetBitsUnmasked(br);
|
||
|
}
|
||
|
p += bits & BitMask(BROTLI_HUFFMAN_MAX_CODE_LENGTH_CODE_LENGTH);
|
||
|
if (p->bits > available_bits) {
|
||
|
get_byte = BROTLI_TRUE;
|
||
|
continue;
|
||
|
}
|
||
|
code_len = p->value; /* code_len == 0..17 */
|
||
|
if (code_len < BROTLI_REPEAT_PREVIOUS_CODE_LENGTH) {
|
||
|
BrotliDropBits(br, p->bits);
|
||
|
ProcessSingleCodeLength(code_len, &s->symbol, &s->repeat, &s->space,
|
||
|
&s->prev_code_len, s->symbol_lists, s->code_length_histo,
|
||
|
s->next_symbol);
|
||
|
} else { /* code_len == 16..17, extra_bits == 2..3 */
|
||
|
uint32_t extra_bits = code_len - 14U;
|
||
|
uint32_t repeat_delta = (bits >> p->bits) & BitMask(extra_bits);
|
||
|
if (available_bits < p->bits + extra_bits) {
|
||
|
get_byte = BROTLI_TRUE;
|
||
|
continue;
|
||
|
}
|
||
|
BrotliDropBits(br, p->bits + extra_bits);
|
||
|
ProcessRepeatedCodeLength(code_len, repeat_delta, alphabet_size,
|
||
|
&s->symbol, &s->repeat, &s->space, &s->prev_code_len,
|
||
|
&s->repeat_code_len, s->symbol_lists, s->code_length_histo,
|
||
|
s->next_symbol);
|
||
|
}
|
||
|
}
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
/* Reads and decodes 15..18 codes using static prefix code.
|
||
|
Each code is 2..4 bits long. In total 30..72 bits are used. */
|
||
|
static BrotliDecoderErrorCode ReadCodeLengthCodeLengths(BrotliDecoderState* s) {
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
uint32_t num_codes = s->repeat;
|
||
|
unsigned space = s->space;
|
||
|
uint32_t i = s->sub_loop_counter;
|
||
|
for (; i < BROTLI_CODE_LENGTH_CODES; ++i) {
|
||
|
const uint8_t code_len_idx = kCodeLengthCodeOrder[i];
|
||
|
uint32_t ix;
|
||
|
uint32_t v;
|
||
|
if (BROTLI_PREDICT_FALSE(!BrotliSafeGetBits(br, 4, &ix))) {
|
||
|
uint32_t available_bits = BrotliGetAvailableBits(br);
|
||
|
if (available_bits != 0) {
|
||
|
ix = BrotliGetBitsUnmasked(br) & 0xF;
|
||
|
} else {
|
||
|
ix = 0;
|
||
|
}
|
||
|
if (kCodeLengthPrefixLength[ix] > available_bits) {
|
||
|
s->sub_loop_counter = i;
|
||
|
s->repeat = num_codes;
|
||
|
s->space = space;
|
||
|
s->substate_huffman = BROTLI_STATE_HUFFMAN_COMPLEX;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
}
|
||
|
v = kCodeLengthPrefixValue[ix];
|
||
|
BrotliDropBits(br, kCodeLengthPrefixLength[ix]);
|
||
|
s->code_length_code_lengths[code_len_idx] = (uint8_t)v;
|
||
|
BROTLI_LOG_ARRAY_INDEX(s->code_length_code_lengths, code_len_idx);
|
||
|
if (v != 0) {
|
||
|
space = space - (32U >> v);
|
||
|
++num_codes;
|
||
|
++s->code_length_histo[v];
|
||
|
if (space - 1U >= 32U) {
|
||
|
/* space is 0 or wrapped around. */
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if (!(num_codes == 1 || space == 0)) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_CL_SPACE);
|
||
|
}
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
/* Decodes the Huffman tables.
|
||
|
There are 2 scenarios:
|
||
|
A) Huffman code contains only few symbols (1..4). Those symbols are read
|
||
|
directly; their code lengths are defined by the number of symbols.
|
||
|
For this scenario 4 - 49 bits will be read.
|
||
|
|
||
|
B) 2-phase decoding:
|
||
|
B.1) Small Huffman table is decoded; it is specified with code lengths
|
||
|
encoded with predefined entropy code. 32 - 74 bits are used.
|
||
|
B.2) Decoded table is used to decode code lengths of symbols in resulting
|
||
|
Huffman table. In worst case 3520 bits are read. */
|
||
|
static BrotliDecoderErrorCode ReadHuffmanCode(uint32_t alphabet_size,
|
||
|
uint32_t max_symbol,
|
||
|
HuffmanCode* table,
|
||
|
uint32_t* opt_table_size,
|
||
|
BrotliDecoderState* s) {
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
/* Unnecessary masking, but might be good for safety. */
|
||
|
alphabet_size &= 0x7FF;
|
||
|
/* State machine. */
|
||
|
for (;;) {
|
||
|
switch (s->substate_huffman) {
|
||
|
case BROTLI_STATE_HUFFMAN_NONE:
|
||
|
if (!BrotliSafeReadBits(br, 2, &s->sub_loop_counter)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
BROTLI_LOG_UINT(s->sub_loop_counter);
|
||
|
/* The value is used as follows:
|
||
|
1 for simple code;
|
||
|
0 for no skipping, 2 skips 2 code lengths, 3 skips 3 code lengths */
|
||
|
if (s->sub_loop_counter != 1) {
|
||
|
s->space = 32;
|
||
|
s->repeat = 0; /* num_codes */
|
||
|
memset(&s->code_length_histo[0], 0, sizeof(s->code_length_histo[0]) *
|
||
|
(BROTLI_HUFFMAN_MAX_CODE_LENGTH_CODE_LENGTH + 1));
|
||
|
memset(&s->code_length_code_lengths[0], 0,
|
||
|
sizeof(s->code_length_code_lengths));
|
||
|
s->substate_huffman = BROTLI_STATE_HUFFMAN_COMPLEX;
|
||
|
continue;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_HUFFMAN_SIMPLE_SIZE:
|
||
|
/* Read symbols, codes & code lengths directly. */
|
||
|
if (!BrotliSafeReadBits(br, 2, &s->symbol)) { /* num_symbols */
|
||
|
s->substate_huffman = BROTLI_STATE_HUFFMAN_SIMPLE_SIZE;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
s->sub_loop_counter = 0;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_HUFFMAN_SIMPLE_READ: {
|
||
|
BrotliDecoderErrorCode result =
|
||
|
ReadSimpleHuffmanSymbols(alphabet_size, max_symbol, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
return result;
|
||
|
}
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_HUFFMAN_SIMPLE_BUILD: {
|
||
|
uint32_t table_size;
|
||
|
if (s->symbol == 3) {
|
||
|
uint32_t bits;
|
||
|
if (!BrotliSafeReadBits(br, 1, &bits)) {
|
||
|
s->substate_huffman = BROTLI_STATE_HUFFMAN_SIMPLE_BUILD;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
s->symbol += bits;
|
||
|
}
|
||
|
BROTLI_LOG_UINT(s->symbol);
|
||
|
table_size = BrotliBuildSimpleHuffmanTable(
|
||
|
table, HUFFMAN_TABLE_BITS, s->symbols_lists_array, s->symbol);
|
||
|
if (opt_table_size) {
|
||
|
*opt_table_size = table_size;
|
||
|
}
|
||
|
s->substate_huffman = BROTLI_STATE_HUFFMAN_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
/* Decode Huffman-coded code lengths. */
|
||
|
case BROTLI_STATE_HUFFMAN_COMPLEX: {
|
||
|
uint32_t i;
|
||
|
BrotliDecoderErrorCode result = ReadCodeLengthCodeLengths(s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
return result;
|
||
|
}
|
||
|
BrotliBuildCodeLengthsHuffmanTable(s->table,
|
||
|
s->code_length_code_lengths,
|
||
|
s->code_length_histo);
|
||
|
memset(&s->code_length_histo[0], 0, sizeof(s->code_length_histo));
|
||
|
for (i = 0; i <= BROTLI_HUFFMAN_MAX_CODE_LENGTH; ++i) {
|
||
|
s->next_symbol[i] = (int)i - (BROTLI_HUFFMAN_MAX_CODE_LENGTH + 1);
|
||
|
s->symbol_lists[s->next_symbol[i]] = 0xFFFF;
|
||
|
}
|
||
|
|
||
|
s->symbol = 0;
|
||
|
s->prev_code_len = BROTLI_INITIAL_REPEATED_CODE_LENGTH;
|
||
|
s->repeat = 0;
|
||
|
s->repeat_code_len = 0;
|
||
|
s->space = 32768;
|
||
|
s->substate_huffman = BROTLI_STATE_HUFFMAN_LENGTH_SYMBOLS;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_HUFFMAN_LENGTH_SYMBOLS: {
|
||
|
uint32_t table_size;
|
||
|
BrotliDecoderErrorCode result = ReadSymbolCodeLengths(max_symbol, s);
|
||
|
if (result == BROTLI_DECODER_NEEDS_MORE_INPUT) {
|
||
|
result = SafeReadSymbolCodeLengths(max_symbol, s);
|
||
|
}
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
if (s->space != 0) {
|
||
|
BROTLI_LOG(("[ReadHuffmanCode] space = %d\n", (int)s->space));
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_HUFFMAN_SPACE);
|
||
|
}
|
||
|
table_size = BrotliBuildHuffmanTable(
|
||
|
table, HUFFMAN_TABLE_BITS, s->symbol_lists, s->code_length_histo);
|
||
|
if (opt_table_size) {
|
||
|
*opt_table_size = table_size;
|
||
|
}
|
||
|
s->substate_huffman = BROTLI_STATE_HUFFMAN_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
default:
|
||
|
return
|
||
|
BROTLI_FAILURE(BROTLI_DECODER_ERROR_UNREACHABLE);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Decodes a block length by reading 3..39 bits. */
|
||
|
static BROTLI_INLINE uint32_t ReadBlockLength(const HuffmanCode* table,
|
||
|
BrotliBitReader* br) {
|
||
|
uint32_t code;
|
||
|
uint32_t nbits;
|
||
|
code = ReadSymbol(table, br);
|
||
|
ASSERT (code < BROTLI_NUM_BLOCK_LEN_SYMBOLS);
|
||
|
nbits = kBlockLengthPrefixCode[code].nbits; /* nbits == 2..24 */
|
||
|
return kBlockLengthPrefixCode[code].offset + BrotliReadBits(br, nbits);
|
||
|
}
|
||
|
|
||
|
/* WARNING: if state is not BROTLI_STATE_READ_BLOCK_LENGTH_NONE, then
|
||
|
reading can't be continued with ReadBlockLength. */
|
||
|
static BROTLI_INLINE BROTLI_BOOL SafeReadBlockLength(
|
||
|
BrotliDecoderState* s, uint32_t* result, const HuffmanCode* table,
|
||
|
BrotliBitReader* br) {
|
||
|
uint32_t index;
|
||
|
if (s->substate_read_block_length == BROTLI_STATE_READ_BLOCK_LENGTH_NONE) {
|
||
|
if (!SafeReadSymbol(table, br, &index)) {
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
} else {
|
||
|
index = s->block_length_index;
|
||
|
}
|
||
|
{
|
||
|
uint32_t bits;
|
||
|
uint32_t nbits = kBlockLengthPrefixCode[index].nbits; /* nbits == 2..24 */
|
||
|
if (!BrotliSafeReadBits(br, nbits, &bits)) {
|
||
|
s->block_length_index = index;
|
||
|
s->substate_read_block_length = BROTLI_STATE_READ_BLOCK_LENGTH_SUFFIX;
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
*result = kBlockLengthPrefixCode[index].offset + bits;
|
||
|
s->substate_read_block_length = BROTLI_STATE_READ_BLOCK_LENGTH_NONE;
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Transform:
|
||
|
1) initialize list L with values 0, 1,... 255
|
||
|
2) For each input element X:
|
||
|
2.1) let Y = L[X]
|
||
|
2.2) remove X-th element from L
|
||
|
2.3) prepend Y to L
|
||
|
2.4) append Y to output
|
||
|
|
||
|
In most cases max(Y) <= 7, so most of L remains intact.
|
||
|
To reduce the cost of initialization, we reuse L, remember the upper bound
|
||
|
of Y values, and reinitialize only first elements in L.
|
||
|
|
||
|
Most of input values are 0 and 1. To reduce number of branches, we replace
|
||
|
inner for loop with do-while. */
|
||
|
static BROTLI_NOINLINE void InverseMoveToFrontTransform(
|
||
|
uint8_t* v, uint32_t v_len, BrotliDecoderState* state) {
|
||
|
/* Reinitialize elements that could have been changed. */
|
||
|
uint32_t i = 1;
|
||
|
uint32_t upper_bound = state->mtf_upper_bound;
|
||
|
uint32_t* mtf = &state->mtf[1]; /* Make mtf[-1] addressable. */
|
||
|
uint8_t* mtf_u8 = (uint8_t*)mtf;
|
||
|
uint8_t* mtf_u8t = mtf_u8 - 1;
|
||
|
/* Load endian-aware constant. */
|
||
|
const uint8_t b0123[4] = {0, 1, 2, 3};
|
||
|
uint32_t pattern;
|
||
|
memcpy(&pattern, &b0123, 4);
|
||
|
|
||
|
/* Initialize list using 4 consequent values pattern. */
|
||
|
mtf[0] = pattern;
|
||
|
do {
|
||
|
pattern += 0x04040404; /* Advance all 4 values by 4. */
|
||
|
mtf[i] = pattern;
|
||
|
i++;
|
||
|
} while (i <= upper_bound);
|
||
|
|
||
|
/* Transform the input. */
|
||
|
upper_bound = 0;
|
||
|
for (i = 0; i < v_len; ++i) {
|
||
|
int index = v[i];
|
||
|
uint8_t value = mtf_u8[index];
|
||
|
upper_bound |= (uint32_t) v[i];
|
||
|
v[i] = value;
|
||
|
mtf_u8t[0] = value;
|
||
|
while (index >= 0) {
|
||
|
mtf_u8t[index + 1] = mtf_u8t[index];
|
||
|
index--;
|
||
|
}
|
||
|
}
|
||
|
/* Remember amount of elements to be reinitialized. */
|
||
|
state->mtf_upper_bound = upper_bound >> 2;
|
||
|
}
|
||
|
|
||
|
/* Decodes a series of Huffman table using ReadHuffmanCode function. */
|
||
|
static BrotliDecoderErrorCode HuffmanTreeGroupDecode(
|
||
|
HuffmanTreeGroup* group, BrotliDecoderState* s) {
|
||
|
if (s->substate_tree_group != BROTLI_STATE_TREE_GROUP_LOOP) {
|
||
|
s->next = group->codes;
|
||
|
s->htree_index = 0;
|
||
|
s->substate_tree_group = BROTLI_STATE_TREE_GROUP_LOOP;
|
||
|
}
|
||
|
while (s->htree_index < group->num_htrees) {
|
||
|
uint32_t table_size;
|
||
|
BrotliDecoderErrorCode result =
|
||
|
ReadHuffmanCode(group->alphabet_size, group->max_symbol,
|
||
|
s->next, &table_size, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) return result;
|
||
|
group->htrees[s->htree_index] = s->next;
|
||
|
s->next += table_size;
|
||
|
++s->htree_index;
|
||
|
}
|
||
|
s->substate_tree_group = BROTLI_STATE_TREE_GROUP_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
/* Decodes a context map.
|
||
|
Decoding is done in 4 phases:
|
||
|
1) Read auxiliary information (6..16 bits) and allocate memory.
|
||
|
In case of trivial context map, decoding is finished at this phase.
|
||
|
2) Decode Huffman table using ReadHuffmanCode function.
|
||
|
This table will be used for reading context map items.
|
||
|
3) Read context map items; "0" values could be run-length encoded.
|
||
|
4) Optionally, apply InverseMoveToFront transform to the resulting map. */
|
||
|
static BrotliDecoderErrorCode DecodeContextMap(uint32_t context_map_size,
|
||
|
uint32_t* num_htrees,
|
||
|
uint8_t** context_map_arg,
|
||
|
BrotliDecoderState* s) {
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
BrotliDecoderErrorCode result = BROTLI_DECODER_SUCCESS;
|
||
|
|
||
|
switch ((int)s->substate_context_map) {
|
||
|
case BROTLI_STATE_CONTEXT_MAP_NONE:
|
||
|
result = DecodeVarLenUint8(s, br, num_htrees);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
return result;
|
||
|
}
|
||
|
(*num_htrees)++;
|
||
|
s->context_index = 0;
|
||
|
BROTLI_LOG_UINT(context_map_size);
|
||
|
BROTLI_LOG_UINT(*num_htrees);
|
||
|
*context_map_arg =
|
||
|
(uint8_t*)BROTLI_DECODER_ALLOC(s, (size_t)context_map_size);
|
||
|
if (*context_map_arg == 0) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_ALLOC_CONTEXT_MAP);
|
||
|
}
|
||
|
if (*num_htrees <= 1) {
|
||
|
memset(*context_map_arg, 0, (size_t)context_map_size);
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
s->substate_context_map = BROTLI_STATE_CONTEXT_MAP_READ_PREFIX;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_CONTEXT_MAP_READ_PREFIX: {
|
||
|
uint32_t bits;
|
||
|
/* In next stage ReadHuffmanCode uses at least 4 bits, so it is safe
|
||
|
to peek 4 bits ahead. */
|
||
|
if (!BrotliSafeGetBits(br, 5, &bits)) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if ((bits & 1) != 0) { /* Use RLE for zeros. */
|
||
|
s->max_run_length_prefix = (bits >> 1) + 1;
|
||
|
BrotliDropBits(br, 5);
|
||
|
} else {
|
||
|
s->max_run_length_prefix = 0;
|
||
|
BrotliDropBits(br, 1);
|
||
|
}
|
||
|
BROTLI_LOG_UINT(s->max_run_length_prefix);
|
||
|
s->substate_context_map = BROTLI_STATE_CONTEXT_MAP_HUFFMAN;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_CONTEXT_MAP_HUFFMAN: {
|
||
|
uint32_t alphabet_size = *num_htrees + s->max_run_length_prefix;
|
||
|
result = ReadHuffmanCode(alphabet_size, alphabet_size,
|
||
|
s->context_map_table, NULL, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) return result;
|
||
|
s->code = 0xFFFF;
|
||
|
s->substate_context_map = BROTLI_STATE_CONTEXT_MAP_DECODE;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_CONTEXT_MAP_DECODE: {
|
||
|
uint32_t context_index = s->context_index;
|
||
|
uint32_t max_run_length_prefix = s->max_run_length_prefix;
|
||
|
uint8_t* context_map = *context_map_arg;
|
||
|
uint32_t code = s->code;
|
||
|
BROTLI_BOOL skip_preamble = (code != 0xFFFF);
|
||
|
while (context_index < context_map_size || skip_preamble) {
|
||
|
if (!skip_preamble) {
|
||
|
if (!SafeReadSymbol(s->context_map_table, br, &code)) {
|
||
|
s->code = 0xFFFF;
|
||
|
s->context_index = context_index;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
BROTLI_LOG_UINT(code);
|
||
|
|
||
|
if (code == 0) {
|
||
|
context_map[context_index++] = 0;
|
||
|
continue;
|
||
|
}
|
||
|
if (code > max_run_length_prefix) {
|
||
|
context_map[context_index++] =
|
||
|
(uint8_t)(code - max_run_length_prefix);
|
||
|
continue;
|
||
|
}
|
||
|
} else {
|
||
|
skip_preamble = BROTLI_FALSE;
|
||
|
}
|
||
|
/* RLE sub-stage. */
|
||
|
{
|
||
|
uint32_t reps;
|
||
|
if (!BrotliSafeReadBits(br, code, &reps)) {
|
||
|
s->code = code;
|
||
|
s->context_index = context_index;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
reps += 1U << code;
|
||
|
BROTLI_LOG_UINT(reps);
|
||
|
if (context_index + reps > context_map_size) {
|
||
|
return
|
||
|
BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_CONTEXT_MAP_REPEAT);
|
||
|
}
|
||
|
do {
|
||
|
context_map[context_index++] = 0;
|
||
|
} while (--reps);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_CONTEXT_MAP_TRANSFORM: {
|
||
|
uint32_t bits;
|
||
|
if (!BrotliSafeReadBits(br, 1, &bits)) {
|
||
|
s->substate_context_map = BROTLI_STATE_CONTEXT_MAP_TRANSFORM;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
if (bits != 0) {
|
||
|
InverseMoveToFrontTransform(*context_map_arg, context_map_size, s);
|
||
|
}
|
||
|
s->substate_context_map = BROTLI_STATE_CONTEXT_MAP_NONE;
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
default:
|
||
|
return
|
||
|
BROTLI_FAILURE(BROTLI_DECODER_ERROR_UNREACHABLE);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Decodes a command or literal and updates block type ring-buffer.
|
||
|
Reads 3..54 bits. */
|
||
|
static BROTLI_INLINE BROTLI_BOOL DecodeBlockTypeAndLength(
|
||
|
int safe, BrotliDecoderState* s, int tree_type) {
|
||
|
uint32_t max_block_type = s->num_block_types[tree_type];
|
||
|
const HuffmanCode* type_tree = &s->block_type_trees[
|
||
|
tree_type * BROTLI_HUFFMAN_MAX_SIZE_258];
|
||
|
const HuffmanCode* len_tree = &s->block_len_trees[
|
||
|
tree_type * BROTLI_HUFFMAN_MAX_SIZE_26];
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
uint32_t* ringbuffer = &s->block_type_rb[tree_type * 2];
|
||
|
uint32_t block_type;
|
||
|
if (max_block_type <= 1) {
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
|
||
|
/* Read 0..15 + 3..39 bits. */
|
||
|
if (!safe) {
|
||
|
block_type = ReadSymbol(type_tree, br);
|
||
|
s->block_length[tree_type] = ReadBlockLength(len_tree, br);
|
||
|
} else {
|
||
|
BrotliBitReaderState memento;
|
||
|
BrotliBitReaderSaveState(br, &memento);
|
||
|
if (!SafeReadSymbol(type_tree, br, &block_type)) return BROTLI_FALSE;
|
||
|
if (!SafeReadBlockLength(s, &s->block_length[tree_type], len_tree, br)) {
|
||
|
s->substate_read_block_length = BROTLI_STATE_READ_BLOCK_LENGTH_NONE;
|
||
|
BrotliBitReaderRestoreState(br, &memento);
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (block_type == 1) {
|
||
|
block_type = ringbuffer[1] + 1;
|
||
|
} else if (block_type == 0) {
|
||
|
block_type = ringbuffer[0];
|
||
|
} else {
|
||
|
block_type -= 2;
|
||
|
}
|
||
|
if (block_type >= max_block_type) {
|
||
|
block_type -= max_block_type;
|
||
|
}
|
||
|
ringbuffer[0] = ringbuffer[1];
|
||
|
ringbuffer[1] = block_type;
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE void DetectTrivialLiteralBlockTypes(
|
||
|
BrotliDecoderState* s) {
|
||
|
size_t i;
|
||
|
for (i = 0; i < 8; ++i) s->trivial_literal_contexts[i] = 0;
|
||
|
for (i = 0; i < s->num_block_types[0]; i++) {
|
||
|
size_t offset = i << BROTLI_LITERAL_CONTEXT_BITS;
|
||
|
size_t error = 0;
|
||
|
size_t sample = s->context_map[offset];
|
||
|
size_t j;
|
||
|
for (j = 0; j < (1u << BROTLI_LITERAL_CONTEXT_BITS);) {
|
||
|
BROTLI_REPEAT(4, error |= s->context_map[offset + j++] ^ sample;)
|
||
|
}
|
||
|
if (error == 0) {
|
||
|
s->trivial_literal_contexts[i >> 5] |= 1u << (i & 31);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE void PrepareLiteralDecoding(BrotliDecoderState* s) {
|
||
|
uint8_t context_mode;
|
||
|
size_t trivial;
|
||
|
uint32_t block_type = s->block_type_rb[1];
|
||
|
uint32_t context_offset = block_type << BROTLI_LITERAL_CONTEXT_BITS;
|
||
|
s->context_map_slice = s->context_map + context_offset;
|
||
|
trivial = s->trivial_literal_contexts[block_type >> 5];
|
||
|
s->trivial_literal_context = (trivial >> (block_type & 31)) & 1;
|
||
|
s->literal_htree = s->literal_hgroup.htrees[s->context_map_slice[0]];
|
||
|
context_mode = s->context_modes[block_type] & 3;
|
||
|
s->context_lookup = BROTLI_CONTEXT_LUT(context_mode);
|
||
|
}
|
||
|
|
||
|
/* Decodes the block type and updates the state for literal context.
|
||
|
Reads 3..54 bits. */
|
||
|
static BROTLI_INLINE BROTLI_BOOL DecodeLiteralBlockSwitchInternal(
|
||
|
int safe, BrotliDecoderState* s) {
|
||
|
if (!DecodeBlockTypeAndLength(safe, s, 0)) {
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
PrepareLiteralDecoding(s);
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
static void BROTLI_NOINLINE DecodeLiteralBlockSwitch(BrotliDecoderState* s) {
|
||
|
DecodeLiteralBlockSwitchInternal(0, s);
|
||
|
}
|
||
|
|
||
|
static BROTLI_BOOL BROTLI_NOINLINE SafeDecodeLiteralBlockSwitch(
|
||
|
BrotliDecoderState* s) {
|
||
|
return DecodeLiteralBlockSwitchInternal(1, s);
|
||
|
}
|
||
|
|
||
|
/* Block switch for insert/copy length.
|
||
|
Reads 3..54 bits. */
|
||
|
static BROTLI_INLINE BROTLI_BOOL DecodeCommandBlockSwitchInternal(
|
||
|
int safe, BrotliDecoderState* s) {
|
||
|
if (!DecodeBlockTypeAndLength(safe, s, 1)) {
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
s->htree_command = s->insert_copy_hgroup.htrees[s->block_type_rb[3]];
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
static void BROTLI_NOINLINE DecodeCommandBlockSwitch(BrotliDecoderState* s) {
|
||
|
DecodeCommandBlockSwitchInternal(0, s);
|
||
|
}
|
||
|
|
||
|
static BROTLI_BOOL BROTLI_NOINLINE SafeDecodeCommandBlockSwitch(
|
||
|
BrotliDecoderState* s) {
|
||
|
return DecodeCommandBlockSwitchInternal(1, s);
|
||
|
}
|
||
|
|
||
|
/* Block switch for distance codes.
|
||
|
Reads 3..54 bits. */
|
||
|
static BROTLI_INLINE BROTLI_BOOL DecodeDistanceBlockSwitchInternal(
|
||
|
int safe, BrotliDecoderState* s) {
|
||
|
if (!DecodeBlockTypeAndLength(safe, s, 2)) {
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
s->dist_context_map_slice = s->dist_context_map +
|
||
|
(s->block_type_rb[5] << BROTLI_DISTANCE_CONTEXT_BITS);
|
||
|
s->dist_htree_index = s->dist_context_map_slice[s->distance_context];
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
static void BROTLI_NOINLINE DecodeDistanceBlockSwitch(BrotliDecoderState* s) {
|
||
|
DecodeDistanceBlockSwitchInternal(0, s);
|
||
|
}
|
||
|
|
||
|
static BROTLI_BOOL BROTLI_NOINLINE SafeDecodeDistanceBlockSwitch(
|
||
|
BrotliDecoderState* s) {
|
||
|
return DecodeDistanceBlockSwitchInternal(1, s);
|
||
|
}
|
||
|
|
||
|
static size_t UnwrittenBytes(const BrotliDecoderState* s, BROTLI_BOOL wrap) {
|
||
|
size_t pos = wrap && s->pos > s->ringbuffer_size ?
|
||
|
(size_t)s->ringbuffer_size : (size_t)(s->pos);
|
||
|
size_t partial_pos_rb = (s->rb_roundtrips * (size_t)s->ringbuffer_size) + pos;
|
||
|
return partial_pos_rb - s->partial_pos_out;
|
||
|
}
|
||
|
|
||
|
/* Dumps output.
|
||
|
Returns BROTLI_DECODER_NEEDS_MORE_OUTPUT only if there is more output to push
|
||
|
and either ring-buffer is as big as window size, or |force| is true. */
|
||
|
static BrotliDecoderErrorCode BROTLI_NOINLINE WriteRingBuffer(
|
||
|
BrotliDecoderState* s, size_t* available_out, uint8_t** next_out,
|
||
|
size_t* total_out, BROTLI_BOOL force) {
|
||
|
uint8_t* start =
|
||
|
s->ringbuffer + (s->partial_pos_out & (size_t)s->ringbuffer_mask);
|
||
|
size_t to_write = UnwrittenBytes(s, BROTLI_TRUE);
|
||
|
size_t num_written = *available_out;
|
||
|
if (num_written > to_write) {
|
||
|
num_written = to_write;
|
||
|
}
|
||
|
if (s->meta_block_remaining_len < 0) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_BLOCK_LENGTH_1);
|
||
|
}
|
||
|
if (next_out && !*next_out) {
|
||
|
*next_out = start;
|
||
|
} else {
|
||
|
if (next_out) {
|
||
|
memcpy(*next_out, start, num_written);
|
||
|
*next_out += num_written;
|
||
|
}
|
||
|
}
|
||
|
*available_out -= num_written;
|
||
|
BROTLI_LOG_UINT(to_write);
|
||
|
BROTLI_LOG_UINT(num_written);
|
||
|
s->partial_pos_out += num_written;
|
||
|
if (total_out) {
|
||
|
*total_out = s->partial_pos_out;
|
||
|
}
|
||
|
if (num_written < to_write) {
|
||
|
if (s->ringbuffer_size == (1 << s->window_bits) || force) {
|
||
|
return BROTLI_DECODER_NEEDS_MORE_OUTPUT;
|
||
|
} else {
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
}
|
||
|
/* Wrap ring buffer only if it has reached its maximal size. */
|
||
|
if (s->ringbuffer_size == (1 << s->window_bits) &&
|
||
|
s->pos >= s->ringbuffer_size) {
|
||
|
s->pos -= s->ringbuffer_size;
|
||
|
s->rb_roundtrips++;
|
||
|
s->should_wrap_ringbuffer = (size_t)s->pos != 0 ? 1 : 0;
|
||
|
}
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
static void BROTLI_NOINLINE WrapRingBuffer(BrotliDecoderState* s) {
|
||
|
if (s->should_wrap_ringbuffer) {
|
||
|
memcpy(s->ringbuffer, s->ringbuffer_end, (size_t)s->pos);
|
||
|
s->should_wrap_ringbuffer = 0;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Allocates ring-buffer.
|
||
|
|
||
|
s->ringbuffer_size MUST be updated by BrotliCalculateRingBufferSize before
|
||
|
this function is called.
|
||
|
|
||
|
Last two bytes of ring-buffer are initialized to 0, so context calculation
|
||
|
could be done uniformly for the first two and all other positions. */
|
||
|
static BROTLI_BOOL BROTLI_NOINLINE BrotliEnsureRingBuffer(
|
||
|
BrotliDecoderState* s) {
|
||
|
uint8_t* old_ringbuffer = s->ringbuffer;
|
||
|
if (s->ringbuffer_size == s->new_ringbuffer_size) {
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
s->ringbuffer = (uint8_t*)BROTLI_DECODER_ALLOC(s,
|
||
|
(size_t)(s->new_ringbuffer_size) + kRingBufferWriteAheadSlack);
|
||
|
if (s->ringbuffer == 0) {
|
||
|
/* Restore previous value. */
|
||
|
s->ringbuffer = old_ringbuffer;
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
s->ringbuffer[s->new_ringbuffer_size - 2] = 0;
|
||
|
s->ringbuffer[s->new_ringbuffer_size - 1] = 0;
|
||
|
|
||
|
if (!!old_ringbuffer) {
|
||
|
memcpy(s->ringbuffer, old_ringbuffer, (size_t)s->pos);
|
||
|
BROTLI_DECODER_FREE(s, old_ringbuffer);
|
||
|
}
|
||
|
|
||
|
s->ringbuffer_size = s->new_ringbuffer_size;
|
||
|
s->ringbuffer_mask = s->new_ringbuffer_size - 1;
|
||
|
s->ringbuffer_end = s->ringbuffer + s->ringbuffer_size;
|
||
|
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
static BrotliDecoderErrorCode BROTLI_NOINLINE CopyUncompressedBlockToOutput(
|
||
|
size_t* available_out, uint8_t** next_out, size_t* total_out,
|
||
|
BrotliDecoderState* s) {
|
||
|
/* TODO: avoid allocation for single uncompressed block. */
|
||
|
if (!BrotliEnsureRingBuffer(s)) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_ALLOC_RING_BUFFER_1);
|
||
|
}
|
||
|
|
||
|
/* State machine */
|
||
|
for (;;) {
|
||
|
switch (s->substate_uncompressed) {
|
||
|
case BROTLI_STATE_UNCOMPRESSED_NONE: {
|
||
|
int nbytes = (int)BrotliGetRemainingBytes(&s->br);
|
||
|
if (nbytes > s->meta_block_remaining_len) {
|
||
|
nbytes = s->meta_block_remaining_len;
|
||
|
}
|
||
|
if (s->pos + nbytes > s->ringbuffer_size) {
|
||
|
nbytes = s->ringbuffer_size - s->pos;
|
||
|
}
|
||
|
/* Copy remaining bytes from s->br.buf_ to ring-buffer. */
|
||
|
BrotliCopyBytes(&s->ringbuffer[s->pos], &s->br, (size_t)nbytes);
|
||
|
s->pos += nbytes;
|
||
|
s->meta_block_remaining_len -= nbytes;
|
||
|
if (s->pos < 1 << s->window_bits) {
|
||
|
if (s->meta_block_remaining_len == 0) {
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
s->substate_uncompressed = BROTLI_STATE_UNCOMPRESSED_WRITE;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_UNCOMPRESSED_WRITE: {
|
||
|
BrotliDecoderErrorCode result;
|
||
|
result = WriteRingBuffer(
|
||
|
s, available_out, next_out, total_out, BROTLI_FALSE);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
return result;
|
||
|
}
|
||
|
if (s->ringbuffer_size == 1 << s->window_bits) {
|
||
|
s->max_distance = s->max_backward_distance;
|
||
|
}
|
||
|
s->substate_uncompressed = BROTLI_STATE_UNCOMPRESSED_NONE;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
BROTLI_DCHECK(0); /* Unreachable */
|
||
|
}
|
||
|
|
||
|
/* Calculates the smallest feasible ring buffer.
|
||
|
|
||
|
If we know the data size is small, do not allocate more ring buffer
|
||
|
size than needed to reduce memory usage.
|
||
|
|
||
|
When this method is called, metablock size and flags MUST be decoded. */
|
||
|
static void BROTLI_NOINLINE BrotliCalculateRingBufferSize(
|
||
|
BrotliDecoderState* s) {
|
||
|
int window_size = 1 << s->window_bits;
|
||
|
int new_ringbuffer_size = window_size;
|
||
|
/* We need at least 2 bytes of ring buffer size to get the last two
|
||
|
bytes for context from there */
|
||
|
int min_size = s->ringbuffer_size ? s->ringbuffer_size : 1024;
|
||
|
int output_size;
|
||
|
|
||
|
/* If maximum is already reached, no further extension is retired. */
|
||
|
if (s->ringbuffer_size == window_size) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
/* Metadata blocks does not touch ring buffer. */
|
||
|
if (s->is_metadata) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if (!s->ringbuffer) {
|
||
|
output_size = 0;
|
||
|
} else {
|
||
|
output_size = s->pos;
|
||
|
}
|
||
|
output_size += s->meta_block_remaining_len;
|
||
|
min_size = min_size < output_size ? output_size : min_size;
|
||
|
|
||
|
if (!!s->canny_ringbuffer_allocation) {
|
||
|
/* Reduce ring buffer size to save memory when server is unscrupulous.
|
||
|
In worst case memory usage might be 1.5x bigger for a short period of
|
||
|
ring buffer reallocation. */
|
||
|
while ((new_ringbuffer_size >> 1) >= min_size) {
|
||
|
new_ringbuffer_size >>= 1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
s->new_ringbuffer_size = new_ringbuffer_size;
|
||
|
}
|
||
|
|
||
|
/* Reads 1..256 2-bit context modes. */
|
||
|
static BrotliDecoderErrorCode ReadContextModes(BrotliDecoderState* s) {
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
int i = s->loop_counter;
|
||
|
|
||
|
while (i < (int)s->num_block_types[0]) {
|
||
|
uint32_t bits;
|
||
|
if (!BrotliSafeReadBits(br, 2, &bits)) {
|
||
|
s->loop_counter = i;
|
||
|
return BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
}
|
||
|
s->context_modes[i] = (uint8_t)bits;
|
||
|
BROTLI_LOG_ARRAY_INDEX(s->context_modes, i);
|
||
|
i++;
|
||
|
}
|
||
|
return BROTLI_DECODER_SUCCESS;
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE void TakeDistanceFromRingBuffer(BrotliDecoderState* s) {
|
||
|
if (s->distance_code == 0) {
|
||
|
--s->dist_rb_idx;
|
||
|
s->distance_code = s->dist_rb[s->dist_rb_idx & 3];
|
||
|
/* Compensate double distance-ring-buffer roll for dictionary items. */
|
||
|
s->distance_context = 1;
|
||
|
} else {
|
||
|
int distance_code = s->distance_code << 1;
|
||
|
/* kDistanceShortCodeIndexOffset has 2-bit values from LSB:
|
||
|
3, 2, 1, 0, 3, 3, 3, 3, 3, 3, 2, 2, 2, 2, 2, 2 */
|
||
|
const uint32_t kDistanceShortCodeIndexOffset = 0xAAAFFF1B;
|
||
|
/* kDistanceShortCodeValueOffset has 2-bit values from LSB:
|
||
|
-0, 0,-0, 0,-1, 1,-2, 2,-3, 3,-1, 1,-2, 2,-3, 3 */
|
||
|
const uint32_t kDistanceShortCodeValueOffset = 0xFA5FA500;
|
||
|
int v = (s->dist_rb_idx +
|
||
|
(int)(kDistanceShortCodeIndexOffset >> distance_code)) & 0x3;
|
||
|
s->distance_code = s->dist_rb[v];
|
||
|
v = (int)(kDistanceShortCodeValueOffset >> distance_code) & 0x3;
|
||
|
if ((distance_code & 0x3) != 0) {
|
||
|
s->distance_code += v;
|
||
|
} else {
|
||
|
s->distance_code -= v;
|
||
|
if (s->distance_code <= 0) {
|
||
|
/* A huge distance will cause a BROTLI_FAILURE() soon.
|
||
|
This is a little faster than failing here. */
|
||
|
s->distance_code = 0x7FFFFFFF;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE BROTLI_BOOL SafeReadBits(
|
||
|
BrotliBitReader* const br, uint32_t n_bits, uint32_t* val) {
|
||
|
if (n_bits != 0) {
|
||
|
return BrotliSafeReadBits(br, n_bits, val);
|
||
|
} else {
|
||
|
*val = 0;
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/* Precondition: s->distance_code < 0. */
|
||
|
static BROTLI_INLINE BROTLI_BOOL ReadDistanceInternal(
|
||
|
int safe, BrotliDecoderState* s, BrotliBitReader* br) {
|
||
|
int distval;
|
||
|
BrotliBitReaderState memento;
|
||
|
HuffmanCode* distance_tree = s->distance_hgroup.htrees[s->dist_htree_index];
|
||
|
if (!safe) {
|
||
|
s->distance_code = (int)ReadSymbol(distance_tree, br);
|
||
|
} else {
|
||
|
uint32_t code;
|
||
|
BrotliBitReaderSaveState(br, &memento);
|
||
|
if (!SafeReadSymbol(distance_tree, br, &code)) {
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
s->distance_code = (int)code;
|
||
|
}
|
||
|
/* Convert the distance code to the actual distance by possibly
|
||
|
looking up past distances from the s->ringbuffer. */
|
||
|
s->distance_context = 0;
|
||
|
if ((s->distance_code & ~0xF) == 0) {
|
||
|
TakeDistanceFromRingBuffer(s);
|
||
|
--s->block_length[2];
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
distval = s->distance_code - (int)s->num_direct_distance_codes;
|
||
|
if (distval >= 0) {
|
||
|
uint32_t nbits;
|
||
|
int postfix;
|
||
|
int offset;
|
||
|
if (!safe && (s->distance_postfix_bits == 0)) {
|
||
|
nbits = ((uint32_t)distval >> 1) + 1;
|
||
|
offset = ((2 + (distval & 1)) << nbits) - 4;
|
||
|
s->distance_code = (int)s->num_direct_distance_codes + offset +
|
||
|
(int)BrotliReadBits(br, nbits);
|
||
|
} else {
|
||
|
/* This branch also works well when s->distance_postfix_bits == 0. */
|
||
|
uint32_t bits;
|
||
|
postfix = distval & s->distance_postfix_mask;
|
||
|
distval >>= s->distance_postfix_bits;
|
||
|
nbits = ((uint32_t)distval >> 1) + 1;
|
||
|
if (safe) {
|
||
|
if (!SafeReadBits(br, nbits, &bits)) {
|
||
|
s->distance_code = -1; /* Restore precondition. */
|
||
|
BrotliBitReaderRestoreState(br, &memento);
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
} else {
|
||
|
bits = BrotliReadBits(br, nbits);
|
||
|
}
|
||
|
offset = ((2 + (distval & 1)) << nbits) - 4;
|
||
|
s->distance_code = (int)s->num_direct_distance_codes +
|
||
|
((offset + (int)bits) << s->distance_postfix_bits) + postfix;
|
||
|
}
|
||
|
}
|
||
|
s->distance_code = s->distance_code - BROTLI_NUM_DISTANCE_SHORT_CODES + 1;
|
||
|
--s->block_length[2];
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE void ReadDistance(
|
||
|
BrotliDecoderState* s, BrotliBitReader* br) {
|
||
|
ReadDistanceInternal(0, s, br);
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE BROTLI_BOOL SafeReadDistance(
|
||
|
BrotliDecoderState* s, BrotliBitReader* br) {
|
||
|
return ReadDistanceInternal(1, s, br);
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE BROTLI_BOOL ReadCommandInternal(
|
||
|
int safe, BrotliDecoderState* s, BrotliBitReader* br, int* insert_length) {
|
||
|
uint32_t cmd_code;
|
||
|
uint32_t insert_len_extra = 0;
|
||
|
uint32_t copy_length;
|
||
|
CmdLutElement v;
|
||
|
BrotliBitReaderState memento;
|
||
|
if (!safe) {
|
||
|
cmd_code = ReadSymbol(s->htree_command, br);
|
||
|
ASSERT (cmd_code < BROTLI_NUM_COMMAND_SYMBOLS);
|
||
|
} else {
|
||
|
BrotliBitReaderSaveState(br, &memento);
|
||
|
if (!SafeReadSymbol(s->htree_command, br, &cmd_code)) {
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
}
|
||
|
v = kCmdLut[cmd_code];
|
||
|
s->distance_code = v.distance_code;
|
||
|
s->distance_context = v.context;
|
||
|
s->dist_htree_index = s->dist_context_map_slice[s->distance_context];
|
||
|
*insert_length = v.insert_len_offset;
|
||
|
if (!safe) {
|
||
|
if (BROTLI_PREDICT_FALSE(v.insert_len_extra_bits != 0)) {
|
||
|
insert_len_extra = BrotliReadBits(br, v.insert_len_extra_bits);
|
||
|
}
|
||
|
copy_length = BrotliReadBits(br, v.copy_len_extra_bits);
|
||
|
} else {
|
||
|
if (!SafeReadBits(br, v.insert_len_extra_bits, &insert_len_extra) ||
|
||
|
!SafeReadBits(br, v.copy_len_extra_bits, ©_length)) {
|
||
|
BrotliBitReaderRestoreState(br, &memento);
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
}
|
||
|
s->copy_length = (int)copy_length + v.copy_len_offset;
|
||
|
--s->block_length[1];
|
||
|
*insert_length += (int)insert_len_extra;
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE void ReadCommand(
|
||
|
BrotliDecoderState* s, BrotliBitReader* br, int* insert_length) {
|
||
|
ReadCommandInternal(0, s, br, insert_length);
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE BROTLI_BOOL SafeReadCommand(
|
||
|
BrotliDecoderState* s, BrotliBitReader* br, int* insert_length) {
|
||
|
return ReadCommandInternal(1, s, br, insert_length);
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE BROTLI_BOOL CheckInputAmount(
|
||
|
int safe, BrotliBitReader* const br, size_t num) {
|
||
|
if (safe) {
|
||
|
return BROTLI_TRUE;
|
||
|
}
|
||
|
return BrotliCheckInputAmount(br, num);
|
||
|
}
|
||
|
|
||
|
#define BROTLI_SAFE(METHOD) \
|
||
|
{ \
|
||
|
if (safe) { \
|
||
|
if (!Safe##METHOD) { \
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT; \
|
||
|
goto saveStateAndReturn; \
|
||
|
} \
|
||
|
} else { \
|
||
|
METHOD; \
|
||
|
} \
|
||
|
}
|
||
|
|
||
|
static BROTLI_INLINE BrotliDecoderErrorCode ProcessCommandsInternal(
|
||
|
int safe, BrotliDecoderState* s) {
|
||
|
int pos = s->pos;
|
||
|
int i = s->loop_counter;
|
||
|
BrotliDecoderErrorCode result = BROTLI_DECODER_SUCCESS;
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
|
||
|
if (!CheckInputAmount(safe, br, 28)) {
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
if (!safe) {
|
||
|
BROTLI_UNUSED(BrotliWarmupBitReader(br));
|
||
|
}
|
||
|
|
||
|
/* Jump into state machine. */
|
||
|
if (s->state == BROTLI_STATE_COMMAND_BEGIN) {
|
||
|
goto CommandBegin;
|
||
|
} else if (s->state == BROTLI_STATE_COMMAND_INNER) {
|
||
|
goto CommandInner;
|
||
|
} else if (s->state == BROTLI_STATE_COMMAND_POST_DECODE_LITERALS) {
|
||
|
goto CommandPostDecodeLiterals;
|
||
|
} else if (s->state == BROTLI_STATE_COMMAND_POST_WRAP_COPY) {
|
||
|
goto CommandPostWrapCopy;
|
||
|
} else {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_UNREACHABLE);
|
||
|
}
|
||
|
|
||
|
CommandBegin:
|
||
|
if (safe) {
|
||
|
s->state = BROTLI_STATE_COMMAND_BEGIN;
|
||
|
}
|
||
|
if (!CheckInputAmount(safe, br, 28)) { /* 156 bits + 7 bytes */
|
||
|
s->state = BROTLI_STATE_COMMAND_BEGIN;
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
if (BROTLI_PREDICT_FALSE(s->block_length[1] == 0)) {
|
||
|
BROTLI_SAFE(DecodeCommandBlockSwitch(s));
|
||
|
goto CommandBegin;
|
||
|
}
|
||
|
/* Read the insert/copy length in the command. */
|
||
|
BROTLI_SAFE(ReadCommand(s, br, &i));
|
||
|
BROTLI_LOG(("[ProcessCommandsInternal] pos = %d insert = %d copy = %d\n",
|
||
|
pos, i, s->copy_length));
|
||
|
if (i == 0) {
|
||
|
goto CommandPostDecodeLiterals;
|
||
|
}
|
||
|
s->meta_block_remaining_len -= i;
|
||
|
|
||
|
CommandInner:
|
||
|
if (safe) {
|
||
|
s->state = BROTLI_STATE_COMMAND_INNER;
|
||
|
}
|
||
|
/* Read the literals in the command. */
|
||
|
if (s->trivial_literal_context) {
|
||
|
uint32_t bits;
|
||
|
uint32_t value;
|
||
|
PreloadSymbol(safe, s->literal_htree, br, &bits, &value);
|
||
|
do {
|
||
|
if (!CheckInputAmount(safe, br, 28)) { /* 162 bits + 7 bytes */
|
||
|
s->state = BROTLI_STATE_COMMAND_INNER;
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
if (BROTLI_PREDICT_FALSE(s->block_length[0] == 0)) {
|
||
|
BROTLI_SAFE(DecodeLiteralBlockSwitch(s));
|
||
|
PreloadSymbol(safe, s->literal_htree, br, &bits, &value);
|
||
|
if (!s->trivial_literal_context) goto CommandInner;
|
||
|
}
|
||
|
if (!safe) {
|
||
|
s->ringbuffer[pos] =
|
||
|
(uint8_t)ReadPreloadedSymbol(s->literal_htree, br, &bits, &value);
|
||
|
} else {
|
||
|
uint32_t literal;
|
||
|
if (!SafeReadSymbol(s->literal_htree, br, &literal)) {
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
s->ringbuffer[pos] = (uint8_t)literal;
|
||
|
}
|
||
|
--s->block_length[0];
|
||
|
BROTLI_LOG_ARRAY_INDEX(s->ringbuffer, pos);
|
||
|
++pos;
|
||
|
if (BROTLI_PREDICT_FALSE(pos == s->ringbuffer_size)) {
|
||
|
s->state = BROTLI_STATE_COMMAND_INNER_WRITE;
|
||
|
--i;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
} while (--i != 0);
|
||
|
} else {
|
||
|
uint8_t p1 = s->ringbuffer[(pos - 1) & s->ringbuffer_mask];
|
||
|
uint8_t p2 = s->ringbuffer[(pos - 2) & s->ringbuffer_mask];
|
||
|
do {
|
||
|
const HuffmanCode* hc;
|
||
|
uint8_t context;
|
||
|
if (!CheckInputAmount(safe, br, 28)) { /* 162 bits + 7 bytes */
|
||
|
s->state = BROTLI_STATE_COMMAND_INNER;
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
if (BROTLI_PREDICT_FALSE(s->block_length[0] == 0)) {
|
||
|
BROTLI_SAFE(DecodeLiteralBlockSwitch(s));
|
||
|
if (s->trivial_literal_context) goto CommandInner;
|
||
|
}
|
||
|
context = BROTLI_CONTEXT(p1, p2, s->context_lookup);
|
||
|
BROTLI_LOG_UINT(context);
|
||
|
hc = s->literal_hgroup.htrees[s->context_map_slice[context]];
|
||
|
p2 = p1;
|
||
|
if (!safe) {
|
||
|
p1 = (uint8_t)ReadSymbol(hc, br);
|
||
|
} else {
|
||
|
uint32_t literal;
|
||
|
if (!SafeReadSymbol(hc, br, &literal)) {
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
p1 = (uint8_t)literal;
|
||
|
}
|
||
|
s->ringbuffer[pos] = p1;
|
||
|
--s->block_length[0];
|
||
|
BROTLI_LOG_UINT(s->context_map_slice[context]);
|
||
|
BROTLI_LOG_ARRAY_INDEX(s->ringbuffer, pos & s->ringbuffer_mask);
|
||
|
++pos;
|
||
|
if (BROTLI_PREDICT_FALSE(pos == s->ringbuffer_size)) {
|
||
|
s->state = BROTLI_STATE_COMMAND_INNER_WRITE;
|
||
|
--i;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
} while (--i != 0);
|
||
|
}
|
||
|
BROTLI_LOG_UINT(s->meta_block_remaining_len);
|
||
|
if (BROTLI_PREDICT_FALSE(s->meta_block_remaining_len <= 0)) {
|
||
|
s->state = BROTLI_STATE_METABLOCK_DONE;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
|
||
|
CommandPostDecodeLiterals:
|
||
|
if (safe) {
|
||
|
s->state = BROTLI_STATE_COMMAND_POST_DECODE_LITERALS;
|
||
|
}
|
||
|
if (s->distance_code >= 0) {
|
||
|
/* Implicit distance case. */
|
||
|
s->distance_context = s->distance_code ? 0 : 1;
|
||
|
--s->dist_rb_idx;
|
||
|
s->distance_code = s->dist_rb[s->dist_rb_idx & 3];
|
||
|
} else {
|
||
|
/* Read distance code in the command, unless it was implicitly zero. */
|
||
|
if (BROTLI_PREDICT_FALSE(s->block_length[2] == 0)) {
|
||
|
BROTLI_SAFE(DecodeDistanceBlockSwitch(s));
|
||
|
}
|
||
|
BROTLI_SAFE(ReadDistance(s, br));
|
||
|
}
|
||
|
BROTLI_LOG(("[ProcessCommandsInternal] pos = %d distance = %d\n",
|
||
|
pos, s->distance_code));
|
||
|
if (s->max_distance != s->max_backward_distance) {
|
||
|
s->max_distance =
|
||
|
(pos < s->max_backward_distance) ? pos : s->max_backward_distance;
|
||
|
}
|
||
|
i = s->copy_length;
|
||
|
/* Apply copy of LZ77 back-reference, or static dictionary reference if
|
||
|
the distance is larger than the max LZ77 distance */
|
||
|
if (s->distance_code > s->max_distance) {
|
||
|
/* The maximum allowed distance is BROTLI_MAX_ALLOWED_DISTANCE = 0x7FFFFFFC.
|
||
|
With this choice, no signed overflow can occur after decoding
|
||
|
a special distance code (e.g., after adding 3 to the last distance). */
|
||
|
if (s->distance_code > BROTLI_MAX_ALLOWED_DISTANCE) {
|
||
|
BROTLI_LOG(("Invalid backward reference. pos: %d distance: %d "
|
||
|
"len: %d bytes left: %d\n",
|
||
|
pos, s->distance_code, i, s->meta_block_remaining_len));
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_DISTANCE);
|
||
|
}
|
||
|
if (i >= BROTLI_MIN_DICTIONARY_WORD_LENGTH &&
|
||
|
i <= BROTLI_MAX_DICTIONARY_WORD_LENGTH) {
|
||
|
int address = s->distance_code - s->max_distance - 1;
|
||
|
const BrotliDictionary* words = s->dictionary;
|
||
|
const BrotliTransforms* transforms = s->transforms;
|
||
|
int offset = (int)s->dictionary->offsets_by_length[i];
|
||
|
uint32_t shift = s->dictionary->size_bits_by_length[i];
|
||
|
|
||
|
int mask = (int)BitMask(shift);
|
||
|
int word_idx = address & mask;
|
||
|
int transform_idx = address >> shift;
|
||
|
/* Compensate double distance-ring-buffer roll. */
|
||
|
s->dist_rb_idx += s->distance_context;
|
||
|
offset += word_idx * i;
|
||
|
if (BROTLI_PREDICT_FALSE(!words->data)) {
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_DICTIONARY_NOT_SET);
|
||
|
}
|
||
|
if (transform_idx < (int)transforms->num_transforms) {
|
||
|
const uint8_t* word = &words->data[offset];
|
||
|
int len = i;
|
||
|
if (transform_idx == transforms->cutOffTransforms[0]) {
|
||
|
memcpy(&s->ringbuffer[pos], word, (size_t)len);
|
||
|
BROTLI_LOG(("[ProcessCommandsInternal] dictionary word: [%.*s]\n",
|
||
|
len, word));
|
||
|
} else {
|
||
|
len = BrotliTransformDictionaryWord(&s->ringbuffer[pos], word, len,
|
||
|
transforms, transform_idx);
|
||
|
BROTLI_LOG(("[ProcessCommandsInternal] dictionary word: [%.*s],"
|
||
|
" transform_idx = %d, transformed: [%.*s]\n",
|
||
|
i, word, transform_idx, len, &s->ringbuffer[pos]));
|
||
|
}
|
||
|
pos += len;
|
||
|
s->meta_block_remaining_len -= len;
|
||
|
if (pos >= s->ringbuffer_size) {
|
||
|
s->state = BROTLI_STATE_COMMAND_POST_WRITE_1;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
} else {
|
||
|
BROTLI_LOG(("Invalid backward reference. pos: %d distance: %d "
|
||
|
"len: %d bytes left: %d\n",
|
||
|
pos, s->distance_code, i, s->meta_block_remaining_len));
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_TRANSFORM);
|
||
|
}
|
||
|
} else {
|
||
|
BROTLI_LOG(("Invalid backward reference. pos: %d distance: %d "
|
||
|
"len: %d bytes left: %d\n",
|
||
|
pos, s->distance_code, i, s->meta_block_remaining_len));
|
||
|
return BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_DICTIONARY);
|
||
|
}
|
||
|
} else {
|
||
|
int src_start = (pos - s->distance_code) & s->ringbuffer_mask;
|
||
|
uint8_t* copy_dst = &s->ringbuffer[pos];
|
||
|
uint8_t* copy_src = &s->ringbuffer[src_start];
|
||
|
int dst_end = pos + i;
|
||
|
int src_end = src_start + i;
|
||
|
/* Update the recent distances cache. */
|
||
|
s->dist_rb[s->dist_rb_idx & 3] = s->distance_code;
|
||
|
++s->dist_rb_idx;
|
||
|
s->meta_block_remaining_len -= i;
|
||
|
/* There are 32+ bytes of slack in the ring-buffer allocation.
|
||
|
Also, we have 16 short codes, that make these 16 bytes irrelevant
|
||
|
in the ring-buffer. Let's copy over them as a first guess. */
|
||
|
memmove16(copy_dst, copy_src);
|
||
|
if (src_end > pos && dst_end > src_start) {
|
||
|
/* Regions intersect. */
|
||
|
goto CommandPostWrapCopy;
|
||
|
}
|
||
|
if (dst_end >= s->ringbuffer_size || src_end >= s->ringbuffer_size) {
|
||
|
/* At least one region wraps. */
|
||
|
goto CommandPostWrapCopy;
|
||
|
}
|
||
|
pos += i;
|
||
|
if (i > 16) {
|
||
|
if (i > 32) {
|
||
|
memcpy(copy_dst + 16, copy_src + 16, (size_t)(i - 16));
|
||
|
} else {
|
||
|
/* This branch covers about 45% cases.
|
||
|
Fixed size short copy allows more compiler optimizations. */
|
||
|
memmove16(copy_dst + 16, copy_src + 16);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
BROTLI_LOG_UINT(s->meta_block_remaining_len);
|
||
|
if (s->meta_block_remaining_len <= 0) {
|
||
|
/* Next metablock, if any. */
|
||
|
s->state = BROTLI_STATE_METABLOCK_DONE;
|
||
|
goto saveStateAndReturn;
|
||
|
} else {
|
||
|
goto CommandBegin;
|
||
|
}
|
||
|
CommandPostWrapCopy:
|
||
|
{
|
||
|
int wrap_guard = s->ringbuffer_size - pos;
|
||
|
while (--i >= 0) {
|
||
|
s->ringbuffer[pos] =
|
||
|
s->ringbuffer[(pos - s->distance_code) & s->ringbuffer_mask];
|
||
|
++pos;
|
||
|
if (BROTLI_PREDICT_FALSE(--wrap_guard == 0)) {
|
||
|
s->state = BROTLI_STATE_COMMAND_POST_WRITE_2;
|
||
|
goto saveStateAndReturn;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
if (s->meta_block_remaining_len <= 0) {
|
||
|
/* Next metablock, if any. */
|
||
|
s->state = BROTLI_STATE_METABLOCK_DONE;
|
||
|
goto saveStateAndReturn;
|
||
|
} else {
|
||
|
goto CommandBegin;
|
||
|
}
|
||
|
|
||
|
saveStateAndReturn:
|
||
|
s->pos = pos;
|
||
|
s->loop_counter = i;
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
#undef BROTLI_SAFE
|
||
|
|
||
|
static BROTLI_NOINLINE BrotliDecoderErrorCode ProcessCommands(
|
||
|
BrotliDecoderState* s) {
|
||
|
return ProcessCommandsInternal(0, s);
|
||
|
}
|
||
|
|
||
|
static BROTLI_NOINLINE BrotliDecoderErrorCode SafeProcessCommands(
|
||
|
BrotliDecoderState* s) {
|
||
|
return ProcessCommandsInternal(1, s);
|
||
|
}
|
||
|
|
||
|
/* Returns the maximum number of distance symbols which can only represent
|
||
|
distances not exceeding BROTLI_MAX_ALLOWED_DISTANCE. */
|
||
|
static uint32_t BrotliMaxDistanceSymbol(uint32_t ndirect, uint32_t npostfix) {
|
||
|
static const uint32_t bound[BROTLI_MAX_NPOSTFIX + 1] = {0, 4, 12, 28};
|
||
|
static const uint32_t diff[BROTLI_MAX_NPOSTFIX + 1] = {73, 126, 228, 424};
|
||
|
uint32_t postfix = 1U << npostfix;
|
||
|
if (ndirect < bound[npostfix]) {
|
||
|
return ndirect + diff[npostfix] + postfix;
|
||
|
} else if (ndirect > bound[npostfix] + postfix) {
|
||
|
return ndirect + diff[npostfix];
|
||
|
} else {
|
||
|
return bound[npostfix] + diff[npostfix] + postfix;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
BrotliDecoderResult BrotliDecoderDecompress(
|
||
|
size_t encoded_size, const uint8_t* encoded_buffer, size_t* decoded_size,
|
||
|
uint8_t* decoded_buffer) {
|
||
|
BrotliDecoderState s;
|
||
|
BrotliDecoderResult result;
|
||
|
size_t total_out = 0;
|
||
|
size_t available_in = encoded_size;
|
||
|
const uint8_t* next_in = encoded_buffer;
|
||
|
size_t available_out = *decoded_size;
|
||
|
uint8_t* next_out = decoded_buffer;
|
||
|
if (!BrotliDecoderStateInit(&s, 0, 0, 0)) {
|
||
|
return BROTLI_DECODER_RESULT_ERROR;
|
||
|
}
|
||
|
result = BrotliDecoderDecompressStream(
|
||
|
&s, &available_in, &next_in, &available_out, &next_out, &total_out);
|
||
|
*decoded_size = total_out;
|
||
|
BrotliDecoderStateCleanup(&s);
|
||
|
if (result != BROTLI_DECODER_RESULT_SUCCESS) {
|
||
|
result = BROTLI_DECODER_RESULT_ERROR;
|
||
|
}
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
/* Invariant: input stream is never overconsumed:
|
||
|
- invalid input implies that the whole stream is invalid -> any amount of
|
||
|
input could be read and discarded
|
||
|
- when result is "needs more input", then at least one more byte is REQUIRED
|
||
|
to complete decoding; all input data MUST be consumed by decoder, so
|
||
|
client could swap the input buffer
|
||
|
- when result is "needs more output" decoder MUST ensure that it doesn't
|
||
|
hold more than 7 bits in bit reader; this saves client from swapping input
|
||
|
buffer ahead of time
|
||
|
- when result is "success" decoder MUST return all unused data back to input
|
||
|
buffer; this is possible because the invariant is held on enter */
|
||
|
BrotliDecoderResult BrotliDecoderDecompressStream(
|
||
|
BrotliDecoderState* s, size_t* available_in, const uint8_t** next_in,
|
||
|
size_t* available_out, uint8_t** next_out, size_t* total_out) {
|
||
|
BrotliDecoderErrorCode result = BROTLI_DECODER_SUCCESS;
|
||
|
BrotliBitReader* br = &s->br;
|
||
|
/* Ensure that |total_out| is set, even if no data will ever be pushed out. */
|
||
|
if (total_out) {
|
||
|
*total_out = s->partial_pos_out;
|
||
|
}
|
||
|
/* Do not try to process further in a case of unrecoverable error. */
|
||
|
if ((int)s->error_code < 0) {
|
||
|
return BROTLI_DECODER_RESULT_ERROR;
|
||
|
}
|
||
|
if (*available_out && (!next_out || !*next_out)) {
|
||
|
return SaveErrorCode(
|
||
|
s, BROTLI_FAILURE(BROTLI_DECODER_ERROR_INVALID_ARGUMENTS));
|
||
|
}
|
||
|
if (!*available_out) next_out = 0;
|
||
|
if (s->buffer_length == 0) { /* Just connect bit reader to input stream. */
|
||
|
br->avail_in = *available_in;
|
||
|
br->next_in = *next_in;
|
||
|
} else {
|
||
|
/* At least one byte of input is required. More than one byte of input may
|
||
|
be required to complete the transaction -> reading more data must be
|
||
|
done in a loop -> do it in a main loop. */
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
br->next_in = &s->buffer.u8[0];
|
||
|
}
|
||
|
/* State machine */
|
||
|
for (;;) {
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
/* Error, needs more input/output. */
|
||
|
if (result == BROTLI_DECODER_NEEDS_MORE_INPUT) {
|
||
|
if (s->ringbuffer != 0) { /* Pro-actively push output. */
|
||
|
BrotliDecoderErrorCode intermediate_result = WriteRingBuffer(s,
|
||
|
available_out, next_out, total_out, BROTLI_TRUE);
|
||
|
/* WriteRingBuffer checks s->meta_block_remaining_len validity. */
|
||
|
if ((int)intermediate_result < 0) {
|
||
|
result = intermediate_result;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
if (s->buffer_length != 0) { /* Used with internal buffer. */
|
||
|
if (br->avail_in == 0) {
|
||
|
/* Successfully finished read transaction.
|
||
|
Accumulator contains less than 8 bits, because internal buffer
|
||
|
is expanded byte-by-byte until it is enough to complete read. */
|
||
|
s->buffer_length = 0;
|
||
|
/* Switch to input stream and restart. */
|
||
|
result = BROTLI_DECODER_SUCCESS;
|
||
|
br->avail_in = *available_in;
|
||
|
br->next_in = *next_in;
|
||
|
continue;
|
||
|
} else if (*available_in != 0) {
|
||
|
/* Not enough data in buffer, but can take one more byte from
|
||
|
input stream. */
|
||
|
result = BROTLI_DECODER_SUCCESS;
|
||
|
s->buffer.u8[s->buffer_length] = **next_in;
|
||
|
s->buffer_length++;
|
||
|
br->avail_in = s->buffer_length;
|
||
|
(*next_in)++;
|
||
|
(*available_in)--;
|
||
|
/* Retry with more data in buffer. */
|
||
|
continue;
|
||
|
}
|
||
|
/* Can't finish reading and no more input. */
|
||
|
break;
|
||
|
} else { /* Input stream doesn't contain enough input. */
|
||
|
/* Copy tail to internal buffer and return. */
|
||
|
*next_in = br->next_in;
|
||
|
*available_in = br->avail_in;
|
||
|
while (*available_in) {
|
||
|
s->buffer.u8[s->buffer_length] = **next_in;
|
||
|
s->buffer_length++;
|
||
|
(*next_in)++;
|
||
|
(*available_in)--;
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
/* Unreachable. */
|
||
|
}
|
||
|
|
||
|
/* Fail or needs more output. */
|
||
|
|
||
|
if (s->buffer_length != 0) {
|
||
|
/* Just consumed the buffered input and produced some output. Otherwise
|
||
|
it would result in "needs more input". Reset internal buffer. */
|
||
|
s->buffer_length = 0;
|
||
|
} else {
|
||
|
/* Using input stream in last iteration. When decoder switches to input
|
||
|
stream it has less than 8 bits in accumulator, so it is safe to
|
||
|
return unused accumulator bits there. */
|
||
|
BrotliBitReaderUnload(br);
|
||
|
*available_in = br->avail_in;
|
||
|
*next_in = br->next_in;
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
switch (s->state) {
|
||
|
case BROTLI_STATE_UNINITED:
|
||
|
/* Prepare to the first read. */
|
||
|
if (!BrotliWarmupBitReader(br)) {
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
break;
|
||
|
}
|
||
|
/* Decode window size. */
|
||
|
result = DecodeWindowBits(s, br); /* Reads 1..8 bits. */
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
if (s->large_window) {
|
||
|
s->state = BROTLI_STATE_LARGE_WINDOW_BITS;
|
||
|
break;
|
||
|
}
|
||
|
s->state = BROTLI_STATE_INITIALIZE;
|
||
|
break;
|
||
|
|
||
|
case BROTLI_STATE_LARGE_WINDOW_BITS:
|
||
|
if (!BrotliSafeReadBits(br, 6, &s->window_bits)) {
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
break;
|
||
|
}
|
||
|
if (s->window_bits < BROTLI_LARGE_MIN_WBITS ||
|
||
|
s->window_bits > BROTLI_LARGE_MAX_WBITS) {
|
||
|
result = BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_WINDOW_BITS);
|
||
|
break;
|
||
|
}
|
||
|
s->state = BROTLI_STATE_INITIALIZE;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_INITIALIZE:
|
||
|
BROTLI_LOG_UINT(s->window_bits);
|
||
|
/* Maximum distance, see section 9.1. of the spec. */
|
||
|
s->max_backward_distance = (1 << s->window_bits) - BROTLI_WINDOW_GAP;
|
||
|
|
||
|
/* Allocate memory for both block_type_trees and block_len_trees. */
|
||
|
s->block_type_trees = (HuffmanCode*)BROTLI_DECODER_ALLOC(s,
|
||
|
sizeof(HuffmanCode) * 3 *
|
||
|
(BROTLI_HUFFMAN_MAX_SIZE_258 + BROTLI_HUFFMAN_MAX_SIZE_26));
|
||
|
if (s->block_type_trees == 0) {
|
||
|
result = BROTLI_FAILURE(BROTLI_DECODER_ERROR_ALLOC_BLOCK_TYPE_TREES);
|
||
|
break;
|
||
|
}
|
||
|
s->block_len_trees =
|
||
|
s->block_type_trees + 3 * BROTLI_HUFFMAN_MAX_SIZE_258;
|
||
|
|
||
|
s->state = BROTLI_STATE_METABLOCK_BEGIN;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_BEGIN:
|
||
|
BrotliDecoderStateMetablockBegin(s);
|
||
|
BROTLI_LOG_UINT(s->pos);
|
||
|
s->state = BROTLI_STATE_METABLOCK_HEADER;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER:
|
||
|
result = DecodeMetaBlockLength(s, br); /* Reads 2 - 31 bits. */
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
BROTLI_LOG_UINT(s->is_last_metablock);
|
||
|
BROTLI_LOG_UINT(s->meta_block_remaining_len);
|
||
|
BROTLI_LOG_UINT(s->is_metadata);
|
||
|
BROTLI_LOG_UINT(s->is_uncompressed);
|
||
|
if (s->is_metadata || s->is_uncompressed) {
|
||
|
if (!BrotliJumpToByteBoundary(br)) {
|
||
|
result = BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_PADDING_1);
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
if (s->is_metadata) {
|
||
|
s->state = BROTLI_STATE_METADATA;
|
||
|
break;
|
||
|
}
|
||
|
if (s->meta_block_remaining_len == 0) {
|
||
|
s->state = BROTLI_STATE_METABLOCK_DONE;
|
||
|
break;
|
||
|
}
|
||
|
BrotliCalculateRingBufferSize(s);
|
||
|
if (s->is_uncompressed) {
|
||
|
s->state = BROTLI_STATE_UNCOMPRESSED;
|
||
|
break;
|
||
|
}
|
||
|
s->loop_counter = 0;
|
||
|
s->state = BROTLI_STATE_HUFFMAN_CODE_0;
|
||
|
break;
|
||
|
|
||
|
case BROTLI_STATE_UNCOMPRESSED: {
|
||
|
result = CopyUncompressedBlockToOutput(
|
||
|
available_out, next_out, total_out, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
s->state = BROTLI_STATE_METABLOCK_DONE;
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
case BROTLI_STATE_METADATA:
|
||
|
for (; s->meta_block_remaining_len > 0; --s->meta_block_remaining_len) {
|
||
|
uint32_t bits;
|
||
|
/* Read one byte and ignore it. */
|
||
|
if (!BrotliSafeReadBits(br, 8, &bits)) {
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
if (result == BROTLI_DECODER_SUCCESS) {
|
||
|
s->state = BROTLI_STATE_METABLOCK_DONE;
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case BROTLI_STATE_HUFFMAN_CODE_0:
|
||
|
if (s->loop_counter >= 3) {
|
||
|
s->state = BROTLI_STATE_METABLOCK_HEADER_2;
|
||
|
break;
|
||
|
}
|
||
|
/* Reads 1..11 bits. */
|
||
|
result = DecodeVarLenUint8(s, br, &s->num_block_types[s->loop_counter]);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
s->num_block_types[s->loop_counter]++;
|
||
|
BROTLI_LOG_UINT(s->num_block_types[s->loop_counter]);
|
||
|
if (s->num_block_types[s->loop_counter] < 2) {
|
||
|
s->loop_counter++;
|
||
|
break;
|
||
|
}
|
||
|
s->state = BROTLI_STATE_HUFFMAN_CODE_1;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_HUFFMAN_CODE_1: {
|
||
|
uint32_t alphabet_size = s->num_block_types[s->loop_counter] + 2;
|
||
|
int tree_offset = s->loop_counter * BROTLI_HUFFMAN_MAX_SIZE_258;
|
||
|
result = ReadHuffmanCode(alphabet_size, alphabet_size,
|
||
|
&s->block_type_trees[tree_offset], NULL, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) break;
|
||
|
s->state = BROTLI_STATE_HUFFMAN_CODE_2;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_HUFFMAN_CODE_2: {
|
||
|
uint32_t alphabet_size = BROTLI_NUM_BLOCK_LEN_SYMBOLS;
|
||
|
int tree_offset = s->loop_counter * BROTLI_HUFFMAN_MAX_SIZE_26;
|
||
|
result = ReadHuffmanCode(alphabet_size, alphabet_size,
|
||
|
&s->block_len_trees[tree_offset], NULL, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) break;
|
||
|
s->state = BROTLI_STATE_HUFFMAN_CODE_3;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_HUFFMAN_CODE_3: {
|
||
|
int tree_offset = s->loop_counter * BROTLI_HUFFMAN_MAX_SIZE_26;
|
||
|
if (!SafeReadBlockLength(s, &s->block_length[s->loop_counter],
|
||
|
&s->block_len_trees[tree_offset], br)) {
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
break;
|
||
|
}
|
||
|
BROTLI_LOG_UINT(s->block_length[s->loop_counter]);
|
||
|
s->loop_counter++;
|
||
|
s->state = BROTLI_STATE_HUFFMAN_CODE_0;
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_HEADER_2: {
|
||
|
uint32_t bits;
|
||
|
if (!BrotliSafeReadBits(br, 6, &bits)) {
|
||
|
result = BROTLI_DECODER_NEEDS_MORE_INPUT;
|
||
|
break;
|
||
|
}
|
||
|
s->distance_postfix_bits = bits & BitMask(2);
|
||
|
bits >>= 2;
|
||
|
s->num_direct_distance_codes = BROTLI_NUM_DISTANCE_SHORT_CODES +
|
||
|
(bits << s->distance_postfix_bits);
|
||
|
BROTLI_LOG_UINT(s->num_direct_distance_codes);
|
||
|
BROTLI_LOG_UINT(s->distance_postfix_bits);
|
||
|
s->distance_postfix_mask = (int)BitMask(s->distance_postfix_bits);
|
||
|
s->context_modes =
|
||
|
(uint8_t*)BROTLI_DECODER_ALLOC(s, (size_t)s->num_block_types[0]);
|
||
|
if (s->context_modes == 0) {
|
||
|
result = BROTLI_FAILURE(BROTLI_DECODER_ERROR_ALLOC_CONTEXT_MODES);
|
||
|
break;
|
||
|
}
|
||
|
s->loop_counter = 0;
|
||
|
s->state = BROTLI_STATE_CONTEXT_MODES;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_CONTEXT_MODES:
|
||
|
result = ReadContextModes(s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
s->state = BROTLI_STATE_CONTEXT_MAP_1;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_CONTEXT_MAP_1:
|
||
|
result = DecodeContextMap(
|
||
|
s->num_block_types[0] << BROTLI_LITERAL_CONTEXT_BITS,
|
||
|
&s->num_literal_htrees, &s->context_map, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
DetectTrivialLiteralBlockTypes(s);
|
||
|
s->state = BROTLI_STATE_CONTEXT_MAP_2;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_CONTEXT_MAP_2: {
|
||
|
uint32_t num_direct_codes =
|
||
|
s->num_direct_distance_codes - BROTLI_NUM_DISTANCE_SHORT_CODES;
|
||
|
uint32_t num_distance_codes = BROTLI_DISTANCE_ALPHABET_SIZE(
|
||
|
s->distance_postfix_bits, num_direct_codes,
|
||
|
(s->large_window ? BROTLI_LARGE_MAX_DISTANCE_BITS :
|
||
|
BROTLI_MAX_DISTANCE_BITS));
|
||
|
uint32_t max_distance_symbol = (s->large_window ?
|
||
|
BrotliMaxDistanceSymbol(
|
||
|
num_direct_codes, s->distance_postfix_bits) :
|
||
|
num_distance_codes);
|
||
|
BROTLI_BOOL allocation_success = BROTLI_TRUE;
|
||
|
result = DecodeContextMap(
|
||
|
s->num_block_types[2] << BROTLI_DISTANCE_CONTEXT_BITS,
|
||
|
&s->num_dist_htrees, &s->dist_context_map, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
allocation_success &= BrotliDecoderHuffmanTreeGroupInit(
|
||
|
s, &s->literal_hgroup, BROTLI_NUM_LITERAL_SYMBOLS,
|
||
|
BROTLI_NUM_LITERAL_SYMBOLS, s->num_literal_htrees);
|
||
|
allocation_success &= BrotliDecoderHuffmanTreeGroupInit(
|
||
|
s, &s->insert_copy_hgroup, BROTLI_NUM_COMMAND_SYMBOLS,
|
||
|
BROTLI_NUM_COMMAND_SYMBOLS, s->num_block_types[1]);
|
||
|
allocation_success &= BrotliDecoderHuffmanTreeGroupInit(
|
||
|
s, &s->distance_hgroup, num_distance_codes,
|
||
|
max_distance_symbol, s->num_dist_htrees);
|
||
|
if (!allocation_success) {
|
||
|
return SaveErrorCode(s,
|
||
|
BROTLI_FAILURE(BROTLI_DECODER_ERROR_ALLOC_TREE_GROUPS));
|
||
|
}
|
||
|
s->loop_counter = 0;
|
||
|
s->state = BROTLI_STATE_TREE_GROUP;
|
||
|
}
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_TREE_GROUP: {
|
||
|
HuffmanTreeGroup* hgroup = NULL;
|
||
|
switch (s->loop_counter) {
|
||
|
case 0: hgroup = &s->literal_hgroup; break;
|
||
|
case 1: hgroup = &s->insert_copy_hgroup; break;
|
||
|
case 2: hgroup = &s->distance_hgroup; break;
|
||
|
default: return SaveErrorCode(s, BROTLI_FAILURE(
|
||
|
BROTLI_DECODER_ERROR_UNREACHABLE));
|
||
|
}
|
||
|
result = HuffmanTreeGroupDecode(hgroup, s);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) break;
|
||
|
s->loop_counter++;
|
||
|
if (s->loop_counter >= 3) {
|
||
|
PrepareLiteralDecoding(s);
|
||
|
s->dist_context_map_slice = s->dist_context_map;
|
||
|
s->htree_command = s->insert_copy_hgroup.htrees[0];
|
||
|
if (!BrotliEnsureRingBuffer(s)) {
|
||
|
result = BROTLI_FAILURE(BROTLI_DECODER_ERROR_ALLOC_RING_BUFFER_2);
|
||
|
break;
|
||
|
}
|
||
|
s->state = BROTLI_STATE_COMMAND_BEGIN;
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
case BROTLI_STATE_COMMAND_BEGIN:
|
||
|
/* Fall through. */
|
||
|
case BROTLI_STATE_COMMAND_INNER:
|
||
|
/* Fall through. */
|
||
|
case BROTLI_STATE_COMMAND_POST_DECODE_LITERALS:
|
||
|
/* Fall through. */
|
||
|
case BROTLI_STATE_COMMAND_POST_WRAP_COPY:
|
||
|
result = ProcessCommands(s);
|
||
|
if (result == BROTLI_DECODER_NEEDS_MORE_INPUT) {
|
||
|
result = SafeProcessCommands(s);
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case BROTLI_STATE_COMMAND_INNER_WRITE:
|
||
|
/* Fall through. */
|
||
|
case BROTLI_STATE_COMMAND_POST_WRITE_1:
|
||
|
/* Fall through. */
|
||
|
case BROTLI_STATE_COMMAND_POST_WRITE_2:
|
||
|
result = WriteRingBuffer(
|
||
|
s, available_out, next_out, total_out, BROTLI_FALSE);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
WrapRingBuffer(s);
|
||
|
if (s->ringbuffer_size == 1 << s->window_bits) {
|
||
|
s->max_distance = s->max_backward_distance;
|
||
|
}
|
||
|
if (s->state == BROTLI_STATE_COMMAND_POST_WRITE_1) {
|
||
|
if (s->meta_block_remaining_len == 0) {
|
||
|
/* Next metablock, if any. */
|
||
|
s->state = BROTLI_STATE_METABLOCK_DONE;
|
||
|
} else {
|
||
|
s->state = BROTLI_STATE_COMMAND_BEGIN;
|
||
|
}
|
||
|
break;
|
||
|
} else if (s->state == BROTLI_STATE_COMMAND_POST_WRITE_2) {
|
||
|
s->state = BROTLI_STATE_COMMAND_POST_WRAP_COPY;
|
||
|
} else { /* BROTLI_STATE_COMMAND_INNER_WRITE */
|
||
|
if (s->loop_counter == 0) {
|
||
|
if (s->meta_block_remaining_len == 0) {
|
||
|
s->state = BROTLI_STATE_METABLOCK_DONE;
|
||
|
} else {
|
||
|
s->state = BROTLI_STATE_COMMAND_POST_DECODE_LITERALS;
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
s->state = BROTLI_STATE_COMMAND_INNER;
|
||
|
}
|
||
|
break;
|
||
|
|
||
|
case BROTLI_STATE_METABLOCK_DONE:
|
||
|
if (s->meta_block_remaining_len < 0) {
|
||
|
result = BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_BLOCK_LENGTH_2);
|
||
|
break;
|
||
|
}
|
||
|
BrotliDecoderStateCleanupAfterMetablock(s);
|
||
|
if (!s->is_last_metablock) {
|
||
|
s->state = BROTLI_STATE_METABLOCK_BEGIN;
|
||
|
break;
|
||
|
}
|
||
|
if (!BrotliJumpToByteBoundary(br)) {
|
||
|
result = BROTLI_FAILURE(BROTLI_DECODER_ERROR_FORMAT_PADDING_2);
|
||
|
break;
|
||
|
}
|
||
|
if (s->buffer_length == 0) {
|
||
|
BrotliBitReaderUnload(br);
|
||
|
*available_in = br->avail_in;
|
||
|
*next_in = br->next_in;
|
||
|
}
|
||
|
s->state = BROTLI_STATE_DONE;
|
||
|
/* Fall through. */
|
||
|
|
||
|
case BROTLI_STATE_DONE:
|
||
|
if (s->ringbuffer != 0) {
|
||
|
result = WriteRingBuffer(
|
||
|
s, available_out, next_out, total_out, BROTLI_TRUE);
|
||
|
if (result != BROTLI_DECODER_SUCCESS) {
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
return SaveErrorCode(s, result);
|
||
|
}
|
||
|
}
|
||
|
return SaveErrorCode(s, result);
|
||
|
}
|
||
|
|
||
|
BROTLI_BOOL BrotliDecoderHasMoreOutput(const BrotliDecoderState* s) {
|
||
|
/* After unrecoverable error remaining output is considered nonsensical. */
|
||
|
if ((int)s->error_code < 0) {
|
||
|
return BROTLI_FALSE;
|
||
|
}
|
||
|
return TO_BROTLI_BOOL(
|
||
|
s->ringbuffer != 0 && UnwrittenBytes(s, BROTLI_FALSE) != 0);
|
||
|
}
|
||
|
|
||
|
const uint8_t* BrotliDecoderTakeOutput(BrotliDecoderState* s, size_t* size) {
|
||
|
uint8_t* result = 0;
|
||
|
size_t available_out = *size ? *size : 1u << 24;
|
||
|
size_t requested_out = available_out;
|
||
|
BrotliDecoderErrorCode status;
|
||
|
if ((s->ringbuffer == 0) || ((int)s->error_code < 0)) {
|
||
|
*size = 0;
|
||
|
return 0;
|
||
|
}
|
||
|
WrapRingBuffer(s);
|
||
|
status = WriteRingBuffer(s, &available_out, &result, 0, BROTLI_TRUE);
|
||
|
/* Either WriteRingBuffer returns those "success" codes... */
|
||
|
if (status == BROTLI_DECODER_SUCCESS ||
|
||
|
status == BROTLI_DECODER_NEEDS_MORE_OUTPUT) {
|
||
|
*size = requested_out - available_out;
|
||
|
} else {
|
||
|
/* ... or stream is broken. Normally this should be caught by
|
||
|
BrotliDecoderDecompressStream, this is just a safeguard. */
|
||
|
if ((int)status < 0) SaveErrorCode(s, status);
|
||
|
*size = 0;
|
||
|
result = 0;
|
||
|
}
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
BROTLI_BOOL BrotliDecoderIsUsed(const BrotliDecoderState* s) {
|
||
|
return TO_BROTLI_BOOL(s->state != BROTLI_STATE_UNINITED ||
|
||
|
BrotliGetAvailableBits(&s->br) != 0);
|
||
|
}
|
||
|
|
||
|
BROTLI_BOOL BrotliDecoderIsFinished(const BrotliDecoderState* s) {
|
||
|
return TO_BROTLI_BOOL(s->state == BROTLI_STATE_DONE) &&
|
||
|
!BrotliDecoderHasMoreOutput(s);
|
||
|
}
|
||
|
|
||
|
BrotliDecoderErrorCode BrotliDecoderGetErrorCode(const BrotliDecoderState* s) {
|
||
|
return (BrotliDecoderErrorCode)s->error_code;
|
||
|
}
|
||
|
|
||
|
const char* BrotliDecoderErrorString(BrotliDecoderErrorCode c) {
|
||
|
switch (c) {
|
||
|
#define BROTLI_ERROR_CODE_CASE_(PREFIX, NAME, CODE) \
|
||
|
case BROTLI_DECODER ## PREFIX ## NAME: return #NAME;
|
||
|
#define BROTLI_NOTHING_
|
||
|
BROTLI_DECODER_ERROR_CODES_LIST(BROTLI_ERROR_CODE_CASE_, BROTLI_NOTHING_)
|
||
|
#undef BROTLI_ERROR_CODE_CASE_
|
||
|
#undef BROTLI_NOTHING_
|
||
|
default: return "INVALID";
|
||
|
}
|
||
|
}
|
||
|
|
||
|
uint32_t BrotliDecoderVersion() {
|
||
|
return BROTLI_VERSION;
|
||
|
}
|
||
|
|
||
|
#if defined(__cplusplus) || defined(c_plusplus)
|
||
|
} /* extern "C" */
|
||
|
#endif
|