/src/duckdb/third_party/brotli/dec/bit_reader.cpp
Line | Count | Source (jump to first uncovered line) |
1 | | /* Copyright 2013 Google Inc. All Rights Reserved. |
2 | | |
3 | | Distributed under MIT license. |
4 | | See file LICENSE for detail or copy at https://opensource.org/licenses/MIT |
5 | | */ |
6 | | |
7 | | /* Bit reading helpers */ |
8 | | |
9 | | #include "bit_reader.h" |
10 | | |
11 | | #include <brotli/types.h> |
12 | | |
13 | | #include "../common/brotli_platform.h" |
14 | | |
15 | | using namespace duckdb_brotli; |
16 | | |
17 | | const brotli_reg_t duckdb_brotli::kBrotliBitMask[33] = { 0x00000000, |
18 | | 0x00000001, 0x00000003, 0x00000007, 0x0000000F, |
19 | | 0x0000001F, 0x0000003F, 0x0000007F, 0x000000FF, |
20 | | 0x000001FF, 0x000003FF, 0x000007FF, 0x00000FFF, |
21 | | 0x00001FFF, 0x00003FFF, 0x00007FFF, 0x0000FFFF, |
22 | | 0x0001FFFF, 0x0003FFFF, 0x0007FFFF, 0x000FFFFF, |
23 | | 0x001FFFFF, 0x003FFFFF, 0x007FFFFF, 0x00FFFFFF, |
24 | | 0x01FFFFFF, 0x03FFFFFF, 0x07FFFFFF, 0x0FFFFFFF, |
25 | | 0x1FFFFFFF, 0x3FFFFFFF, 0x7FFFFFFF, 0xFFFFFFFF |
26 | | }; |
27 | | |
28 | 0 | void duckdb_brotli::BrotliInitBitReader(BrotliBitReader* const br) { |
29 | 0 | br->val_ = 0; |
30 | 0 | br->bit_pos_ = 0; |
31 | 0 | } |
32 | | |
33 | 0 | BROTLI_BOOL duckdb_brotli::BrotliWarmupBitReader(BrotliBitReader* const br) { |
34 | 0 | size_t aligned_read_mask = (sizeof(br->val_) >> 1) - 1; |
35 | | /* Fixing alignment after unaligned BrotliFillWindow would result accumulator |
36 | | overflow. If unalignment is caused by BrotliSafeReadBits, then there is |
37 | | enough space in accumulator to fix alignment. */ |
38 | 0 | if (BROTLI_UNALIGNED_READ_FAST) { |
39 | 0 | aligned_read_mask = 0; |
40 | 0 | } |
41 | 0 | if (BrotliGetAvailableBits(br) == 0) { |
42 | 0 | br->val_ = 0; |
43 | 0 | if (!BrotliPullByte(br)) { |
44 | 0 | return BROTLI_FALSE; |
45 | 0 | } |
46 | 0 | } |
47 | | |
48 | 0 | while ((((size_t)br->next_in) & aligned_read_mask) != 0) { |
49 | 0 | if (!BrotliPullByte(br)) { |
50 | | /* If we consumed all the input, we don't care about the alignment. */ |
51 | 0 | return BROTLI_TRUE; |
52 | 0 | } |
53 | 0 | } |
54 | 0 | return BROTLI_TRUE; |
55 | 0 | } |
56 | | |
57 | | BROTLI_BOOL duckdb_brotli::BrotliSafeReadBits32Slow(BrotliBitReader* const br, |
58 | 0 | brotli_reg_t n_bits, brotli_reg_t* val) { |
59 | 0 | brotli_reg_t low_val; |
60 | 0 | brotli_reg_t high_val; |
61 | 0 | BrotliBitReaderState memento; |
62 | 0 | BROTLI_DCHECK(n_bits <= 32); |
63 | 0 | BROTLI_DCHECK(n_bits > 24); |
64 | 0 | BrotliBitReaderSaveState(br, &memento); |
65 | 0 | if (!BrotliSafeReadBits(br, 16, &low_val) || |
66 | 0 | !BrotliSafeReadBits(br, n_bits - 16, &high_val)) { |
67 | 0 | BrotliBitReaderRestoreState(br, &memento); |
68 | 0 | return BROTLI_FALSE; |
69 | 0 | } |
70 | 0 | *val = low_val | (high_val << 16); |
71 | 0 | return BROTLI_TRUE; |
72 | 0 | } |
73 | | |
74 | | |