1*600f14f4SXin Li /* libFLAC - Free Lossless Audio Codec library
2*600f14f4SXin Li * Copyright (C) 2000-2009 Josh Coalson
3*600f14f4SXin Li * Copyright (C) 2011-2023 Xiph.Org Foundation
4*600f14f4SXin Li *
5*600f14f4SXin Li * Redistribution and use in source and binary forms, with or without
6*600f14f4SXin Li * modification, are permitted provided that the following conditions
7*600f14f4SXin Li * are met:
8*600f14f4SXin Li *
9*600f14f4SXin Li * - Redistributions of source code must retain the above copyright
10*600f14f4SXin Li * notice, this list of conditions and the following disclaimer.
11*600f14f4SXin Li *
12*600f14f4SXin Li * - Redistributions in binary form must reproduce the above copyright
13*600f14f4SXin Li * notice, this list of conditions and the following disclaimer in the
14*600f14f4SXin Li * documentation and/or other materials provided with the distribution.
15*600f14f4SXin Li *
16*600f14f4SXin Li * - Neither the name of the Xiph.org Foundation nor the names of its
17*600f14f4SXin Li * contributors may be used to endorse or promote products derived from
18*600f14f4SXin Li * this software without specific prior written permission.
19*600f14f4SXin Li *
20*600f14f4SXin Li * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21*600f14f4SXin Li * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22*600f14f4SXin Li * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23*600f14f4SXin Li * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
24*600f14f4SXin Li * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
25*600f14f4SXin Li * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
26*600f14f4SXin Li * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
27*600f14f4SXin Li * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
28*600f14f4SXin Li * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
29*600f14f4SXin Li * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
30*600f14f4SXin Li * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31*600f14f4SXin Li */
32*600f14f4SXin Li
33*600f14f4SXin Li #ifdef HAVE_CONFIG_H
34*600f14f4SXin Li # include <config.h>
35*600f14f4SXin Li #endif
36*600f14f4SXin Li
37*600f14f4SXin Li #include <stdlib.h>
38*600f14f4SXin Li #include <string.h>
39*600f14f4SXin Li #include "private/bitmath.h"
40*600f14f4SXin Li #include "private/bitreader.h"
41*600f14f4SXin Li #include "private/crc.h"
42*600f14f4SXin Li #include "private/cpu.h"
43*600f14f4SXin Li #include "private/macros.h"
44*600f14f4SXin Li #include "FLAC/assert.h"
45*600f14f4SXin Li #include "share/compat.h"
46*600f14f4SXin Li #include "share/endswap.h"
47*600f14f4SXin Li
48*600f14f4SXin Li /* Things should be fastest when this matches the machine word size */
49*600f14f4SXin Li /* WATCHOUT: if you change this you must also change the following #defines down to COUNT_ZERO_MSBS2 below to match */
50*600f14f4SXin Li /* WATCHOUT: there are a few places where the code will not work unless brword is >= 32 bits wide */
51*600f14f4SXin Li /* also, some sections currently only have fast versions for 4 or 8 bytes per word */
52*600f14f4SXin Li
53*600f14f4SXin Li #if (ENABLE_64_BIT_WORDS == 0)
54*600f14f4SXin Li
55*600f14f4SXin Li typedef FLAC__uint32 brword;
56*600f14f4SXin Li #define FLAC__BYTES_PER_WORD 4 /* sizeof brword */
57*600f14f4SXin Li #define FLAC__BITS_PER_WORD 32
58*600f14f4SXin Li #define FLAC__WORD_ALL_ONES ((FLAC__uint32)0xffffffff)
59*600f14f4SXin Li /* SWAP_BE_WORD_TO_HOST swaps bytes in a brword (which is always big-endian) if necessary to match host byte order */
60*600f14f4SXin Li #if WORDS_BIGENDIAN
61*600f14f4SXin Li #define SWAP_BE_WORD_TO_HOST(x) (x)
62*600f14f4SXin Li #else
63*600f14f4SXin Li #define SWAP_BE_WORD_TO_HOST(x) ENDSWAP_32(x)
64*600f14f4SXin Li #endif
65*600f14f4SXin Li /* counts the # of zero MSBs in a word */
66*600f14f4SXin Li #define COUNT_ZERO_MSBS(word) FLAC__clz_uint32(word)
67*600f14f4SXin Li #define COUNT_ZERO_MSBS2(word) FLAC__clz2_uint32(word)
68*600f14f4SXin Li
69*600f14f4SXin Li #else
70*600f14f4SXin Li
71*600f14f4SXin Li typedef FLAC__uint64 brword;
72*600f14f4SXin Li #define FLAC__BYTES_PER_WORD 8 /* sizeof brword */
73*600f14f4SXin Li #define FLAC__BITS_PER_WORD 64
74*600f14f4SXin Li #define FLAC__WORD_ALL_ONES ((FLAC__uint64)FLAC__U64L(0xffffffffffffffff))
75*600f14f4SXin Li /* SWAP_BE_WORD_TO_HOST swaps bytes in a brword (which is always big-endian) if necessary to match host byte order */
76*600f14f4SXin Li #if WORDS_BIGENDIAN
77*600f14f4SXin Li #define SWAP_BE_WORD_TO_HOST(x) (x)
78*600f14f4SXin Li #else
79*600f14f4SXin Li #define SWAP_BE_WORD_TO_HOST(x) ENDSWAP_64(x)
80*600f14f4SXin Li #endif
81*600f14f4SXin Li /* counts the # of zero MSBs in a word */
82*600f14f4SXin Li #define COUNT_ZERO_MSBS(word) FLAC__clz_uint64(word)
83*600f14f4SXin Li #define COUNT_ZERO_MSBS2(word) FLAC__clz2_uint64(word)
84*600f14f4SXin Li
85*600f14f4SXin Li #endif
86*600f14f4SXin Li
87*600f14f4SXin Li /*
88*600f14f4SXin Li * This should be at least twice as large as the largest number of words
89*600f14f4SXin Li * required to represent any 'number' (in any encoding) you are going to
90*600f14f4SXin Li * read. With FLAC this is on the order of maybe a few hundred bits.
91*600f14f4SXin Li * If the buffer is smaller than that, the decoder won't be able to read
92*600f14f4SXin Li * in a whole number that is in a variable length encoding (e.g. Rice).
93*600f14f4SXin Li * But to be practical it should be at least 1K bytes.
94*600f14f4SXin Li *
95*600f14f4SXin Li * Increase this number to decrease the number of read callbacks, at the
96*600f14f4SXin Li * expense of using more memory. Or decrease for the reverse effect,
97*600f14f4SXin Li * keeping in mind the limit from the first paragraph. The optimal size
98*600f14f4SXin Li * also depends on the CPU cache size and other factors; some twiddling
99*600f14f4SXin Li * may be necessary to squeeze out the best performance.
100*600f14f4SXin Li */
101*600f14f4SXin Li static const uint32_t FLAC__BITREADER_DEFAULT_CAPACITY = 65536u / FLAC__BITS_PER_WORD; /* in words */
102*600f14f4SXin Li
103*600f14f4SXin Li struct FLAC__BitReader {
104*600f14f4SXin Li /* any partially-consumed word at the head will stay right-justified as bits are consumed from the left */
105*600f14f4SXin Li /* any incomplete word at the tail will be left-justified, and bytes from the read callback are added on the right */
106*600f14f4SXin Li brword *buffer;
107*600f14f4SXin Li uint32_t capacity; /* in words */
108*600f14f4SXin Li uint32_t words; /* # of completed words in buffer */
109*600f14f4SXin Li uint32_t bytes; /* # of bytes in incomplete word at buffer[words] */
110*600f14f4SXin Li uint32_t consumed_words; /* #words ... */
111*600f14f4SXin Li uint32_t consumed_bits; /* ... + (#bits of head word) already consumed from the front of buffer */
112*600f14f4SXin Li uint32_t read_crc16; /* the running frame CRC */
113*600f14f4SXin Li uint32_t crc16_offset; /* the number of words in the current buffer that should not be CRC'd */
114*600f14f4SXin Li uint32_t crc16_align; /* the number of bits in the current consumed word that should not be CRC'd */
115*600f14f4SXin Li FLAC__bool read_limit_set; /* whether reads are limited */
116*600f14f4SXin Li uint32_t read_limit; /* the remaining size of what can be read */
117*600f14f4SXin Li uint32_t last_seen_framesync; /* the location of the last seen framesync, if it is in the buffer, in bits from front of buffer */
118*600f14f4SXin Li FLAC__BitReaderReadCallback read_callback;
119*600f14f4SXin Li void *client_data;
120*600f14f4SXin Li };
121*600f14f4SXin Li
crc16_update_word_(FLAC__BitReader * br,brword word)122*600f14f4SXin Li static inline void crc16_update_word_(FLAC__BitReader *br, brword word)
123*600f14f4SXin Li {
124*600f14f4SXin Li register uint32_t crc = br->read_crc16;
125*600f14f4SXin Li
126*600f14f4SXin Li for ( ; br->crc16_align < FLAC__BITS_PER_WORD ; br->crc16_align += 8) {
127*600f14f4SXin Li uint32_t shift = FLAC__BITS_PER_WORD - 8 - br->crc16_align ;
128*600f14f4SXin Li crc = FLAC__CRC16_UPDATE ((uint32_t) (shift < FLAC__BITS_PER_WORD ? (word >> shift) & 0xff : 0), crc);
129*600f14f4SXin Li }
130*600f14f4SXin Li
131*600f14f4SXin Li br->read_crc16 = crc;
132*600f14f4SXin Li br->crc16_align = 0;
133*600f14f4SXin Li }
134*600f14f4SXin Li
crc16_update_block_(FLAC__BitReader * br)135*600f14f4SXin Li static inline void crc16_update_block_(FLAC__BitReader *br)
136*600f14f4SXin Li {
137*600f14f4SXin Li if(br->consumed_words > br->crc16_offset && br->crc16_align)
138*600f14f4SXin Li crc16_update_word_(br, br->buffer[br->crc16_offset++]);
139*600f14f4SXin Li
140*600f14f4SXin Li /* Prevent OOB read due to wrap-around. */
141*600f14f4SXin Li if (br->consumed_words > br->crc16_offset) {
142*600f14f4SXin Li #if FLAC__BYTES_PER_WORD == 4
143*600f14f4SXin Li br->read_crc16 = FLAC__crc16_update_words32(br->buffer + br->crc16_offset, br->consumed_words - br->crc16_offset, br->read_crc16);
144*600f14f4SXin Li #elif FLAC__BYTES_PER_WORD == 8
145*600f14f4SXin Li br->read_crc16 = FLAC__crc16_update_words64(br->buffer + br->crc16_offset, br->consumed_words - br->crc16_offset, br->read_crc16);
146*600f14f4SXin Li #else
147*600f14f4SXin Li unsigned i;
148*600f14f4SXin Li
149*600f14f4SXin Li for (i = br->crc16_offset; i < br->consumed_words; i++)
150*600f14f4SXin Li crc16_update_word_(br, br->buffer[i]);
151*600f14f4SXin Li #endif
152*600f14f4SXin Li }
153*600f14f4SXin Li
154*600f14f4SXin Li br->crc16_offset = 0;
155*600f14f4SXin Li }
156*600f14f4SXin Li
bitreader_read_from_client_(FLAC__BitReader * br)157*600f14f4SXin Li static FLAC__bool bitreader_read_from_client_(FLAC__BitReader *br)
158*600f14f4SXin Li {
159*600f14f4SXin Li uint32_t start, end;
160*600f14f4SXin Li size_t bytes;
161*600f14f4SXin Li FLAC__byte *target;
162*600f14f4SXin Li #if WORDS_BIGENDIAN
163*600f14f4SXin Li #else
164*600f14f4SXin Li brword preswap_backup;
165*600f14f4SXin Li #endif
166*600f14f4SXin Li
167*600f14f4SXin Li /* first shift the unconsumed buffer data toward the front as much as possible */
168*600f14f4SXin Li if(br->consumed_words > 0) {
169*600f14f4SXin Li /* invalidate last seen framesync */
170*600f14f4SXin Li br->last_seen_framesync = -1;
171*600f14f4SXin Li
172*600f14f4SXin Li crc16_update_block_(br); /* CRC consumed words */
173*600f14f4SXin Li
174*600f14f4SXin Li start = br->consumed_words;
175*600f14f4SXin Li end = br->words + (br->bytes? 1:0);
176*600f14f4SXin Li memmove(br->buffer, br->buffer+start, FLAC__BYTES_PER_WORD * (end - start));
177*600f14f4SXin Li
178*600f14f4SXin Li br->words -= start;
179*600f14f4SXin Li br->consumed_words = 0;
180*600f14f4SXin Li }
181*600f14f4SXin Li
182*600f14f4SXin Li /*
183*600f14f4SXin Li * set the target for reading, taking into account word alignment and endianness
184*600f14f4SXin Li */
185*600f14f4SXin Li bytes = (br->capacity - br->words) * FLAC__BYTES_PER_WORD - br->bytes;
186*600f14f4SXin Li if(bytes == 0)
187*600f14f4SXin Li return false; /* no space left, buffer is too small; see note for FLAC__BITREADER_DEFAULT_CAPACITY */
188*600f14f4SXin Li target = ((FLAC__byte*)(br->buffer+br->words)) + br->bytes;
189*600f14f4SXin Li
190*600f14f4SXin Li /* before reading, if the existing reader looks like this (say brword is 32 bits wide)
191*600f14f4SXin Li * bitstream : 11 22 33 44 55 br->words=1 br->bytes=1 (partial tail word is left-justified)
192*600f14f4SXin Li * buffer[BE]: 11 22 33 44 55 ?? ?? ?? (shown laid out as bytes sequentially in memory)
193*600f14f4SXin Li * buffer[LE]: 44 33 22 11 ?? ?? ?? 55 (?? being don't-care)
194*600f14f4SXin Li * ^^-------target, bytes=3
195*600f14f4SXin Li * on LE machines, have to byteswap the odd tail word so nothing is
196*600f14f4SXin Li * overwritten:
197*600f14f4SXin Li */
198*600f14f4SXin Li #if WORDS_BIGENDIAN
199*600f14f4SXin Li #else
200*600f14f4SXin Li preswap_backup = br->buffer[br->words];
201*600f14f4SXin Li if(br->bytes)
202*600f14f4SXin Li br->buffer[br->words] = SWAP_BE_WORD_TO_HOST(br->buffer[br->words]);
203*600f14f4SXin Li #endif
204*600f14f4SXin Li
205*600f14f4SXin Li /* now it looks like:
206*600f14f4SXin Li * bitstream : 11 22 33 44 55 br->words=1 br->bytes=1
207*600f14f4SXin Li * buffer[BE]: 11 22 33 44 55 ?? ?? ??
208*600f14f4SXin Li * buffer[LE]: 44 33 22 11 55 ?? ?? ??
209*600f14f4SXin Li * ^^-------target, bytes=3
210*600f14f4SXin Li */
211*600f14f4SXin Li
212*600f14f4SXin Li /* read in the data; note that the callback may return a smaller number of bytes */
213*600f14f4SXin Li if(!br->read_callback(target, &bytes, br->client_data)){
214*600f14f4SXin Li /* Despite the read callback failing, the data in the target
215*600f14f4SXin Li * might be used later, when the buffer is rewound. Therefore
216*600f14f4SXin Li * we revert the swap that was just done */
217*600f14f4SXin Li #if WORDS_BIGENDIAN
218*600f14f4SXin Li #else
219*600f14f4SXin Li br->buffer[br->words] = preswap_backup;
220*600f14f4SXin Li #endif
221*600f14f4SXin Li return false;
222*600f14f4SXin Li }
223*600f14f4SXin Li
224*600f14f4SXin Li /* after reading bytes 66 77 88 99 AA BB CC DD EE FF from the client:
225*600f14f4SXin Li * bitstream : 11 22 33 44 55 66 77 88 99 AA BB CC DD EE FF
226*600f14f4SXin Li * buffer[BE]: 11 22 33 44 55 66 77 88 99 AA BB CC DD EE FF ??
227*600f14f4SXin Li * buffer[LE]: 44 33 22 11 55 66 77 88 99 AA BB CC DD EE FF ??
228*600f14f4SXin Li * now have to byteswap on LE machines:
229*600f14f4SXin Li */
230*600f14f4SXin Li #if WORDS_BIGENDIAN
231*600f14f4SXin Li #else
232*600f14f4SXin Li end = (br->words*FLAC__BYTES_PER_WORD + br->bytes + (uint32_t)bytes + (FLAC__BYTES_PER_WORD-1)) / FLAC__BYTES_PER_WORD;
233*600f14f4SXin Li for(start = br->words; start < end; start++)
234*600f14f4SXin Li br->buffer[start] = SWAP_BE_WORD_TO_HOST(br->buffer[start]);
235*600f14f4SXin Li #endif
236*600f14f4SXin Li
237*600f14f4SXin Li /* now it looks like:
238*600f14f4SXin Li * bitstream : 11 22 33 44 55 66 77 88 99 AA BB CC DD EE FF
239*600f14f4SXin Li * buffer[BE]: 11 22 33 44 55 66 77 88 99 AA BB CC DD EE FF ??
240*600f14f4SXin Li * buffer[LE]: 44 33 22 11 88 77 66 55 CC BB AA 99 ?? FF EE DD
241*600f14f4SXin Li * finally we'll update the reader values:
242*600f14f4SXin Li */
243*600f14f4SXin Li end = br->words*FLAC__BYTES_PER_WORD + br->bytes + (uint32_t)bytes;
244*600f14f4SXin Li br->words = end / FLAC__BYTES_PER_WORD;
245*600f14f4SXin Li br->bytes = end % FLAC__BYTES_PER_WORD;
246*600f14f4SXin Li
247*600f14f4SXin Li return true;
248*600f14f4SXin Li }
249*600f14f4SXin Li
250*600f14f4SXin Li /***********************************************************************
251*600f14f4SXin Li *
252*600f14f4SXin Li * Class constructor/destructor
253*600f14f4SXin Li *
254*600f14f4SXin Li ***********************************************************************/
255*600f14f4SXin Li
FLAC__bitreader_new(void)256*600f14f4SXin Li FLAC__BitReader *FLAC__bitreader_new(void)
257*600f14f4SXin Li {
258*600f14f4SXin Li FLAC__BitReader *br = calloc(1, sizeof(FLAC__BitReader));
259*600f14f4SXin Li
260*600f14f4SXin Li /* calloc() implies:
261*600f14f4SXin Li memset(br, 0, sizeof(FLAC__BitReader));
262*600f14f4SXin Li br->buffer = 0;
263*600f14f4SXin Li br->capacity = 0;
264*600f14f4SXin Li br->words = br->bytes = 0;
265*600f14f4SXin Li br->consumed_words = br->consumed_bits = 0;
266*600f14f4SXin Li br->read_callback = 0;
267*600f14f4SXin Li br->client_data = 0;
268*600f14f4SXin Li */
269*600f14f4SXin Li return br;
270*600f14f4SXin Li }
271*600f14f4SXin Li
FLAC__bitreader_delete(FLAC__BitReader * br)272*600f14f4SXin Li void FLAC__bitreader_delete(FLAC__BitReader *br)
273*600f14f4SXin Li {
274*600f14f4SXin Li FLAC__ASSERT(0 != br);
275*600f14f4SXin Li
276*600f14f4SXin Li FLAC__bitreader_free(br);
277*600f14f4SXin Li free(br);
278*600f14f4SXin Li }
279*600f14f4SXin Li
280*600f14f4SXin Li /***********************************************************************
281*600f14f4SXin Li *
282*600f14f4SXin Li * Public class methods
283*600f14f4SXin Li *
284*600f14f4SXin Li ***********************************************************************/
285*600f14f4SXin Li
FLAC__bitreader_init(FLAC__BitReader * br,FLAC__BitReaderReadCallback rcb,void * cd)286*600f14f4SXin Li FLAC__bool FLAC__bitreader_init(FLAC__BitReader *br, FLAC__BitReaderReadCallback rcb, void *cd)
287*600f14f4SXin Li {
288*600f14f4SXin Li FLAC__ASSERT(0 != br);
289*600f14f4SXin Li
290*600f14f4SXin Li br->words = br->bytes = 0;
291*600f14f4SXin Li br->consumed_words = br->consumed_bits = 0;
292*600f14f4SXin Li br->capacity = FLAC__BITREADER_DEFAULT_CAPACITY;
293*600f14f4SXin Li br->buffer = malloc(sizeof(brword) * br->capacity);
294*600f14f4SXin Li if(br->buffer == 0)
295*600f14f4SXin Li return false;
296*600f14f4SXin Li br->read_callback = rcb;
297*600f14f4SXin Li br->client_data = cd;
298*600f14f4SXin Li br->read_limit_set = false;
299*600f14f4SXin Li br->read_limit = -1;
300*600f14f4SXin Li br->last_seen_framesync = -1;
301*600f14f4SXin Li
302*600f14f4SXin Li return true;
303*600f14f4SXin Li }
304*600f14f4SXin Li
FLAC__bitreader_free(FLAC__BitReader * br)305*600f14f4SXin Li void FLAC__bitreader_free(FLAC__BitReader *br)
306*600f14f4SXin Li {
307*600f14f4SXin Li FLAC__ASSERT(0 != br);
308*600f14f4SXin Li
309*600f14f4SXin Li if(0 != br->buffer)
310*600f14f4SXin Li free(br->buffer);
311*600f14f4SXin Li br->buffer = 0;
312*600f14f4SXin Li br->capacity = 0;
313*600f14f4SXin Li br->words = br->bytes = 0;
314*600f14f4SXin Li br->consumed_words = br->consumed_bits = 0;
315*600f14f4SXin Li br->read_callback = 0;
316*600f14f4SXin Li br->client_data = 0;
317*600f14f4SXin Li br->read_limit_set = false;
318*600f14f4SXin Li br->read_limit = -1;
319*600f14f4SXin Li br->last_seen_framesync = -1;
320*600f14f4SXin Li }
321*600f14f4SXin Li
FLAC__bitreader_clear(FLAC__BitReader * br)322*600f14f4SXin Li FLAC__bool FLAC__bitreader_clear(FLAC__BitReader *br)
323*600f14f4SXin Li {
324*600f14f4SXin Li br->words = br->bytes = 0;
325*600f14f4SXin Li br->consumed_words = br->consumed_bits = 0;
326*600f14f4SXin Li br->read_limit_set = false;
327*600f14f4SXin Li br->read_limit = -1;
328*600f14f4SXin Li br->last_seen_framesync = -1;
329*600f14f4SXin Li return true;
330*600f14f4SXin Li }
331*600f14f4SXin Li
FLAC__bitreader_set_framesync_location(FLAC__BitReader * br)332*600f14f4SXin Li void FLAC__bitreader_set_framesync_location(FLAC__BitReader *br)
333*600f14f4SXin Li {
334*600f14f4SXin Li br->last_seen_framesync = br->consumed_words * FLAC__BYTES_PER_WORD + br->consumed_bits / 8;
335*600f14f4SXin Li }
336*600f14f4SXin Li
FLAC__bitreader_rewind_to_after_last_seen_framesync(FLAC__BitReader * br)337*600f14f4SXin Li FLAC__bool FLAC__bitreader_rewind_to_after_last_seen_framesync(FLAC__BitReader *br)
338*600f14f4SXin Li {
339*600f14f4SXin Li if(br->last_seen_framesync == (uint32_t)-1) {
340*600f14f4SXin Li br->consumed_words = br->consumed_bits = 0;
341*600f14f4SXin Li return false;
342*600f14f4SXin Li }
343*600f14f4SXin Li else {
344*600f14f4SXin Li br->consumed_words = (br->last_seen_framesync + 1) / FLAC__BYTES_PER_WORD;
345*600f14f4SXin Li br->consumed_bits = ((br->last_seen_framesync + 1) % FLAC__BYTES_PER_WORD) * 8;
346*600f14f4SXin Li return true;
347*600f14f4SXin Li }
348*600f14f4SXin Li }
349*600f14f4SXin Li
FLAC__bitreader_reset_read_crc16(FLAC__BitReader * br,FLAC__uint16 seed)350*600f14f4SXin Li void FLAC__bitreader_reset_read_crc16(FLAC__BitReader *br, FLAC__uint16 seed)
351*600f14f4SXin Li {
352*600f14f4SXin Li FLAC__ASSERT(0 != br);
353*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
354*600f14f4SXin Li FLAC__ASSERT((br->consumed_bits & 7) == 0);
355*600f14f4SXin Li
356*600f14f4SXin Li br->read_crc16 = (uint32_t)seed;
357*600f14f4SXin Li br->crc16_offset = br->consumed_words;
358*600f14f4SXin Li br->crc16_align = br->consumed_bits;
359*600f14f4SXin Li }
360*600f14f4SXin Li
FLAC__bitreader_get_read_crc16(FLAC__BitReader * br)361*600f14f4SXin Li FLAC__uint16 FLAC__bitreader_get_read_crc16(FLAC__BitReader *br)
362*600f14f4SXin Li {
363*600f14f4SXin Li FLAC__ASSERT(0 != br);
364*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
365*600f14f4SXin Li
366*600f14f4SXin Li /* CRC consumed words up to here */
367*600f14f4SXin Li crc16_update_block_(br);
368*600f14f4SXin Li
369*600f14f4SXin Li FLAC__ASSERT((br->consumed_bits & 7) == 0);
370*600f14f4SXin Li FLAC__ASSERT(br->crc16_align <= br->consumed_bits);
371*600f14f4SXin Li
372*600f14f4SXin Li /* CRC any tail bytes in a partially-consumed word */
373*600f14f4SXin Li if(br->consumed_bits) {
374*600f14f4SXin Li const brword tail = br->buffer[br->consumed_words];
375*600f14f4SXin Li for( ; br->crc16_align < br->consumed_bits; br->crc16_align += 8)
376*600f14f4SXin Li br->read_crc16 = FLAC__CRC16_UPDATE((uint32_t)((tail >> (FLAC__BITS_PER_WORD-8-br->crc16_align)) & 0xff), br->read_crc16);
377*600f14f4SXin Li }
378*600f14f4SXin Li return br->read_crc16;
379*600f14f4SXin Li }
380*600f14f4SXin Li
FLAC__bitreader_is_consumed_byte_aligned(const FLAC__BitReader * br)381*600f14f4SXin Li inline FLAC__bool FLAC__bitreader_is_consumed_byte_aligned(const FLAC__BitReader *br)
382*600f14f4SXin Li {
383*600f14f4SXin Li return ((br->consumed_bits & 7) == 0);
384*600f14f4SXin Li }
385*600f14f4SXin Li
FLAC__bitreader_bits_left_for_byte_alignment(const FLAC__BitReader * br)386*600f14f4SXin Li inline uint32_t FLAC__bitreader_bits_left_for_byte_alignment(const FLAC__BitReader *br)
387*600f14f4SXin Li {
388*600f14f4SXin Li return 8 - (br->consumed_bits & 7);
389*600f14f4SXin Li }
390*600f14f4SXin Li
FLAC__bitreader_get_input_bits_unconsumed(const FLAC__BitReader * br)391*600f14f4SXin Li inline uint32_t FLAC__bitreader_get_input_bits_unconsumed(const FLAC__BitReader *br)
392*600f14f4SXin Li {
393*600f14f4SXin Li return (br->words-br->consumed_words)*FLAC__BITS_PER_WORD + br->bytes*8 - br->consumed_bits;
394*600f14f4SXin Li }
395*600f14f4SXin Li
FLAC__bitreader_set_limit(FLAC__BitReader * br,uint32_t limit)396*600f14f4SXin Li void FLAC__bitreader_set_limit(FLAC__BitReader *br, uint32_t limit)
397*600f14f4SXin Li {
398*600f14f4SXin Li br->read_limit = limit;
399*600f14f4SXin Li br->read_limit_set = true;
400*600f14f4SXin Li }
401*600f14f4SXin Li
FLAC__bitreader_remove_limit(FLAC__BitReader * br)402*600f14f4SXin Li void FLAC__bitreader_remove_limit(FLAC__BitReader *br)
403*600f14f4SXin Li {
404*600f14f4SXin Li br->read_limit_set = false;
405*600f14f4SXin Li br->read_limit = -1;
406*600f14f4SXin Li }
407*600f14f4SXin Li
FLAC__bitreader_limit_remaining(FLAC__BitReader * br)408*600f14f4SXin Li uint32_t FLAC__bitreader_limit_remaining(FLAC__BitReader *br)
409*600f14f4SXin Li {
410*600f14f4SXin Li FLAC__ASSERT(br->read_limit_set);
411*600f14f4SXin Li return br->read_limit;
412*600f14f4SXin Li }
FLAC__bitreader_limit_invalidate(FLAC__BitReader * br)413*600f14f4SXin Li void FLAC__bitreader_limit_invalidate(FLAC__BitReader *br)
414*600f14f4SXin Li {
415*600f14f4SXin Li br->read_limit = -1;
416*600f14f4SXin Li }
417*600f14f4SXin Li
FLAC__bitreader_read_raw_uint32(FLAC__BitReader * br,FLAC__uint32 * val,uint32_t bits)418*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_raw_uint32(FLAC__BitReader *br, FLAC__uint32 *val, uint32_t bits)
419*600f14f4SXin Li {
420*600f14f4SXin Li FLAC__ASSERT(0 != br);
421*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
422*600f14f4SXin Li
423*600f14f4SXin Li FLAC__ASSERT(bits <= 32);
424*600f14f4SXin Li FLAC__ASSERT((br->capacity*FLAC__BITS_PER_WORD) * 2 >= bits);
425*600f14f4SXin Li FLAC__ASSERT(br->consumed_words <= br->words);
426*600f14f4SXin Li
427*600f14f4SXin Li /* WATCHOUT: code does not work with <32bit words; we can make things much faster with this assertion */
428*600f14f4SXin Li FLAC__ASSERT(FLAC__BITS_PER_WORD >= 32);
429*600f14f4SXin Li
430*600f14f4SXin Li if(bits == 0) { /* OPT: investigate if this can ever happen, maybe change to assertion */
431*600f14f4SXin Li *val = 0;
432*600f14f4SXin Li return true;
433*600f14f4SXin Li }
434*600f14f4SXin Li
435*600f14f4SXin Li if(br->read_limit_set && br->read_limit < (uint32_t)-1){
436*600f14f4SXin Li if(br->read_limit < bits) {
437*600f14f4SXin Li br->read_limit = -1;
438*600f14f4SXin Li return false;
439*600f14f4SXin Li }
440*600f14f4SXin Li else
441*600f14f4SXin Li br->read_limit -= bits;
442*600f14f4SXin Li }
443*600f14f4SXin Li
444*600f14f4SXin Li while((br->words-br->consumed_words)*FLAC__BITS_PER_WORD + br->bytes*8 - br->consumed_bits < bits) {
445*600f14f4SXin Li if(!bitreader_read_from_client_(br))
446*600f14f4SXin Li return false;
447*600f14f4SXin Li }
448*600f14f4SXin Li if(br->consumed_words < br->words) { /* if we've not consumed up to a partial tail word... */
449*600f14f4SXin Li /* OPT: taking out the consumed_bits==0 "else" case below might make things faster if less code allows the compiler to inline this function */
450*600f14f4SXin Li if(br->consumed_bits) {
451*600f14f4SXin Li /* this also works when consumed_bits==0, it's just a little slower than necessary for that case */
452*600f14f4SXin Li const uint32_t n = FLAC__BITS_PER_WORD - br->consumed_bits;
453*600f14f4SXin Li const brword word = br->buffer[br->consumed_words];
454*600f14f4SXin Li const brword mask = br->consumed_bits < FLAC__BITS_PER_WORD ? FLAC__WORD_ALL_ONES >> br->consumed_bits : 0;
455*600f14f4SXin Li if(bits < n) {
456*600f14f4SXin Li uint32_t shift = n - bits;
457*600f14f4SXin Li *val = shift < FLAC__BITS_PER_WORD ? (FLAC__uint32)((word & mask) >> shift) : 0; /* The result has <= 32 non-zero bits */
458*600f14f4SXin Li br->consumed_bits += bits;
459*600f14f4SXin Li return true;
460*600f14f4SXin Li }
461*600f14f4SXin Li /* (FLAC__BITS_PER_WORD - br->consumed_bits <= bits) ==> (FLAC__WORD_ALL_ONES >> br->consumed_bits) has no more than 'bits' non-zero bits */
462*600f14f4SXin Li *val = (FLAC__uint32)(word & mask);
463*600f14f4SXin Li bits -= n;
464*600f14f4SXin Li br->consumed_words++;
465*600f14f4SXin Li br->consumed_bits = 0;
466*600f14f4SXin Li if(bits) { /* if there are still bits left to read, there have to be less than 32 so they will all be in the next word */
467*600f14f4SXin Li uint32_t shift = FLAC__BITS_PER_WORD - bits;
468*600f14f4SXin Li *val = bits < 32 ? *val << bits : 0;
469*600f14f4SXin Li *val |= shift < FLAC__BITS_PER_WORD ? (FLAC__uint32)(br->buffer[br->consumed_words] >> shift) : 0;
470*600f14f4SXin Li br->consumed_bits = bits;
471*600f14f4SXin Li }
472*600f14f4SXin Li return true;
473*600f14f4SXin Li }
474*600f14f4SXin Li else { /* br->consumed_bits == 0 */
475*600f14f4SXin Li const brword word = br->buffer[br->consumed_words];
476*600f14f4SXin Li if(bits < FLAC__BITS_PER_WORD) {
477*600f14f4SXin Li *val = (FLAC__uint32)(word >> (FLAC__BITS_PER_WORD-bits));
478*600f14f4SXin Li br->consumed_bits = bits;
479*600f14f4SXin Li return true;
480*600f14f4SXin Li }
481*600f14f4SXin Li /* at this point bits == FLAC__BITS_PER_WORD == 32; because of previous assertions, it can't be larger */
482*600f14f4SXin Li *val = (FLAC__uint32)word;
483*600f14f4SXin Li br->consumed_words++;
484*600f14f4SXin Li return true;
485*600f14f4SXin Li }
486*600f14f4SXin Li }
487*600f14f4SXin Li else {
488*600f14f4SXin Li /* in this case we're starting our read at a partial tail word;
489*600f14f4SXin Li * the reader has guaranteed that we have at least 'bits' bits
490*600f14f4SXin Li * available to read, which makes this case simpler.
491*600f14f4SXin Li */
492*600f14f4SXin Li /* OPT: taking out the consumed_bits==0 "else" case below might make things faster if less code allows the compiler to inline this function */
493*600f14f4SXin Li if(br->consumed_bits) {
494*600f14f4SXin Li /* this also works when consumed_bits==0, it's just a little slower than necessary for that case */
495*600f14f4SXin Li FLAC__ASSERT(br->consumed_bits + bits <= br->bytes*8);
496*600f14f4SXin Li *val = (FLAC__uint32)((br->buffer[br->consumed_words] & (FLAC__WORD_ALL_ONES >> br->consumed_bits)) >> (FLAC__BITS_PER_WORD-br->consumed_bits-bits));
497*600f14f4SXin Li br->consumed_bits += bits;
498*600f14f4SXin Li return true;
499*600f14f4SXin Li }
500*600f14f4SXin Li else {
501*600f14f4SXin Li *val = (FLAC__uint32)(br->buffer[br->consumed_words] >> (FLAC__BITS_PER_WORD-bits));
502*600f14f4SXin Li br->consumed_bits += bits;
503*600f14f4SXin Li return true;
504*600f14f4SXin Li }
505*600f14f4SXin Li }
506*600f14f4SXin Li }
507*600f14f4SXin Li
FLAC__bitreader_read_raw_int32(FLAC__BitReader * br,FLAC__int32 * val,uint32_t bits)508*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_raw_int32(FLAC__BitReader *br, FLAC__int32 *val, uint32_t bits)
509*600f14f4SXin Li {
510*600f14f4SXin Li FLAC__uint32 uval, mask;
511*600f14f4SXin Li /* OPT: inline raw uint32 code here, or make into a macro if possible in the .h file */
512*600f14f4SXin Li if (bits < 1 || ! FLAC__bitreader_read_raw_uint32(br, &uval, bits))
513*600f14f4SXin Li return false;
514*600f14f4SXin Li /* sign-extend *val assuming it is currently bits wide. */
515*600f14f4SXin Li /* From: https://graphics.stanford.edu/~seander/bithacks.html#FixedSignExtend */
516*600f14f4SXin Li mask = bits >= 33 ? 0 : 1lu << (bits - 1);
517*600f14f4SXin Li *val = (uval ^ mask) - mask;
518*600f14f4SXin Li return true;
519*600f14f4SXin Li }
520*600f14f4SXin Li
FLAC__bitreader_read_raw_uint64(FLAC__BitReader * br,FLAC__uint64 * val,uint32_t bits)521*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_raw_uint64(FLAC__BitReader *br, FLAC__uint64 *val, uint32_t bits)
522*600f14f4SXin Li {
523*600f14f4SXin Li FLAC__uint32 hi, lo;
524*600f14f4SXin Li
525*600f14f4SXin Li if(bits > 32) {
526*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &hi, bits-32))
527*600f14f4SXin Li return false;
528*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &lo, 32))
529*600f14f4SXin Li return false;
530*600f14f4SXin Li *val = hi;
531*600f14f4SXin Li *val <<= 32;
532*600f14f4SXin Li *val |= lo;
533*600f14f4SXin Li }
534*600f14f4SXin Li else {
535*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &lo, bits))
536*600f14f4SXin Li return false;
537*600f14f4SXin Li *val = lo;
538*600f14f4SXin Li }
539*600f14f4SXin Li return true;
540*600f14f4SXin Li }
541*600f14f4SXin Li
FLAC__bitreader_read_raw_int64(FLAC__BitReader * br,FLAC__int64 * val,uint32_t bits)542*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_raw_int64(FLAC__BitReader *br, FLAC__int64 *val, uint32_t bits)
543*600f14f4SXin Li {
544*600f14f4SXin Li FLAC__uint64 uval, mask;
545*600f14f4SXin Li /* OPT: inline raw uint64 code here, or make into a macro if possible in the .h file */
546*600f14f4SXin Li if (bits < 1 || ! FLAC__bitreader_read_raw_uint64(br, &uval, bits))
547*600f14f4SXin Li return false;
548*600f14f4SXin Li /* sign-extend *val assuming it is currently bits wide. */
549*600f14f4SXin Li /* From: https://graphics.stanford.edu/~seander/bithacks.html#FixedSignExtend */
550*600f14f4SXin Li mask = bits >= 65 ? 0 : 1llu << (bits - 1);
551*600f14f4SXin Li *val = (uval ^ mask) - mask;
552*600f14f4SXin Li return true;
553*600f14f4SXin Li }
554*600f14f4SXin Li
FLAC__bitreader_read_uint32_little_endian(FLAC__BitReader * br,FLAC__uint32 * val)555*600f14f4SXin Li inline FLAC__bool FLAC__bitreader_read_uint32_little_endian(FLAC__BitReader *br, FLAC__uint32 *val)
556*600f14f4SXin Li {
557*600f14f4SXin Li FLAC__uint32 x8, x32 = 0;
558*600f14f4SXin Li
559*600f14f4SXin Li /* this doesn't need to be that fast as currently it is only used for vorbis comments */
560*600f14f4SXin Li
561*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x32, 8))
562*600f14f4SXin Li return false;
563*600f14f4SXin Li
564*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x8, 8))
565*600f14f4SXin Li return false;
566*600f14f4SXin Li x32 |= (x8 << 8);
567*600f14f4SXin Li
568*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x8, 8))
569*600f14f4SXin Li return false;
570*600f14f4SXin Li x32 |= (x8 << 16);
571*600f14f4SXin Li
572*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x8, 8))
573*600f14f4SXin Li return false;
574*600f14f4SXin Li x32 |= (x8 << 24);
575*600f14f4SXin Li
576*600f14f4SXin Li *val = x32;
577*600f14f4SXin Li return true;
578*600f14f4SXin Li }
579*600f14f4SXin Li
FLAC__bitreader_skip_bits_no_crc(FLAC__BitReader * br,uint32_t bits)580*600f14f4SXin Li FLAC__bool FLAC__bitreader_skip_bits_no_crc(FLAC__BitReader *br, uint32_t bits)
581*600f14f4SXin Li {
582*600f14f4SXin Li /*
583*600f14f4SXin Li * OPT: a faster implementation is possible but probably not that useful
584*600f14f4SXin Li * since this is only called a couple of times in the metadata readers.
585*600f14f4SXin Li */
586*600f14f4SXin Li FLAC__ASSERT(0 != br);
587*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
588*600f14f4SXin Li
589*600f14f4SXin Li if(bits > 0) {
590*600f14f4SXin Li const uint32_t n = br->consumed_bits & 7;
591*600f14f4SXin Li uint32_t m;
592*600f14f4SXin Li FLAC__uint32 x;
593*600f14f4SXin Li
594*600f14f4SXin Li if(n != 0) {
595*600f14f4SXin Li m = flac_min(8-n, bits);
596*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, m))
597*600f14f4SXin Li return false;
598*600f14f4SXin Li bits -= m;
599*600f14f4SXin Li }
600*600f14f4SXin Li m = bits / 8;
601*600f14f4SXin Li if(m > 0) {
602*600f14f4SXin Li if(!FLAC__bitreader_skip_byte_block_aligned_no_crc(br, m))
603*600f14f4SXin Li return false;
604*600f14f4SXin Li bits %= 8;
605*600f14f4SXin Li }
606*600f14f4SXin Li if(bits > 0) {
607*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, bits))
608*600f14f4SXin Li return false;
609*600f14f4SXin Li }
610*600f14f4SXin Li }
611*600f14f4SXin Li
612*600f14f4SXin Li return true;
613*600f14f4SXin Li }
614*600f14f4SXin Li
FLAC__bitreader_skip_byte_block_aligned_no_crc(FLAC__BitReader * br,uint32_t nvals)615*600f14f4SXin Li FLAC__bool FLAC__bitreader_skip_byte_block_aligned_no_crc(FLAC__BitReader *br, uint32_t nvals)
616*600f14f4SXin Li {
617*600f14f4SXin Li FLAC__uint32 x;
618*600f14f4SXin Li
619*600f14f4SXin Li FLAC__ASSERT(0 != br);
620*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
621*600f14f4SXin Li FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(br));
622*600f14f4SXin Li
623*600f14f4SXin Li if(br->read_limit_set && br->read_limit < (uint32_t)-1){
624*600f14f4SXin Li if(br->read_limit < nvals*8){
625*600f14f4SXin Li br->read_limit = -1;
626*600f14f4SXin Li return false;
627*600f14f4SXin Li }
628*600f14f4SXin Li }
629*600f14f4SXin Li
630*600f14f4SXin Li /* step 1: skip over partial head word to get word aligned */
631*600f14f4SXin Li while(nvals && br->consumed_bits) { /* i.e. run until we read 'nvals' bytes or we hit the end of the head word */
632*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, 8))
633*600f14f4SXin Li return false;
634*600f14f4SXin Li nvals--;
635*600f14f4SXin Li }
636*600f14f4SXin Li if(0 == nvals)
637*600f14f4SXin Li return true;
638*600f14f4SXin Li
639*600f14f4SXin Li /* step 2: skip whole words in chunks */
640*600f14f4SXin Li while(nvals >= FLAC__BYTES_PER_WORD) {
641*600f14f4SXin Li if(br->consumed_words < br->words) {
642*600f14f4SXin Li br->consumed_words++;
643*600f14f4SXin Li nvals -= FLAC__BYTES_PER_WORD;
644*600f14f4SXin Li if(br->read_limit_set)
645*600f14f4SXin Li br->read_limit -= FLAC__BITS_PER_WORD;
646*600f14f4SXin Li }
647*600f14f4SXin Li else if(!bitreader_read_from_client_(br))
648*600f14f4SXin Li return false;
649*600f14f4SXin Li }
650*600f14f4SXin Li /* step 3: skip any remainder from partial tail bytes */
651*600f14f4SXin Li while(nvals) {
652*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, 8))
653*600f14f4SXin Li return false;
654*600f14f4SXin Li nvals--;
655*600f14f4SXin Li }
656*600f14f4SXin Li
657*600f14f4SXin Li return true;
658*600f14f4SXin Li }
659*600f14f4SXin Li
FLAC__bitreader_read_byte_block_aligned_no_crc(FLAC__BitReader * br,FLAC__byte * val,uint32_t nvals)660*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_byte_block_aligned_no_crc(FLAC__BitReader *br, FLAC__byte *val, uint32_t nvals)
661*600f14f4SXin Li {
662*600f14f4SXin Li FLAC__uint32 x;
663*600f14f4SXin Li
664*600f14f4SXin Li FLAC__ASSERT(0 != br);
665*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
666*600f14f4SXin Li FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(br));
667*600f14f4SXin Li
668*600f14f4SXin Li if(br->read_limit_set && br->read_limit < (uint32_t)-1){
669*600f14f4SXin Li if(br->read_limit < nvals*8){
670*600f14f4SXin Li br->read_limit = -1;
671*600f14f4SXin Li return false;
672*600f14f4SXin Li }
673*600f14f4SXin Li }
674*600f14f4SXin Li
675*600f14f4SXin Li /* step 1: read from partial head word to get word aligned */
676*600f14f4SXin Li while(nvals && br->consumed_bits) { /* i.e. run until we read 'nvals' bytes or we hit the end of the head word */
677*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, 8))
678*600f14f4SXin Li return false;
679*600f14f4SXin Li *val++ = (FLAC__byte)x;
680*600f14f4SXin Li nvals--;
681*600f14f4SXin Li }
682*600f14f4SXin Li if(0 == nvals)
683*600f14f4SXin Li return true;
684*600f14f4SXin Li /* step 2: read whole words in chunks */
685*600f14f4SXin Li while(nvals >= FLAC__BYTES_PER_WORD) {
686*600f14f4SXin Li if(br->consumed_words < br->words) {
687*600f14f4SXin Li const brword word = br->buffer[br->consumed_words++];
688*600f14f4SXin Li #if FLAC__BYTES_PER_WORD == 4
689*600f14f4SXin Li val[0] = (FLAC__byte)(word >> 24);
690*600f14f4SXin Li val[1] = (FLAC__byte)(word >> 16);
691*600f14f4SXin Li val[2] = (FLAC__byte)(word >> 8);
692*600f14f4SXin Li val[3] = (FLAC__byte)word;
693*600f14f4SXin Li #elif FLAC__BYTES_PER_WORD == 8
694*600f14f4SXin Li val[0] = (FLAC__byte)(word >> 56);
695*600f14f4SXin Li val[1] = (FLAC__byte)(word >> 48);
696*600f14f4SXin Li val[2] = (FLAC__byte)(word >> 40);
697*600f14f4SXin Li val[3] = (FLAC__byte)(word >> 32);
698*600f14f4SXin Li val[4] = (FLAC__byte)(word >> 24);
699*600f14f4SXin Li val[5] = (FLAC__byte)(word >> 16);
700*600f14f4SXin Li val[6] = (FLAC__byte)(word >> 8);
701*600f14f4SXin Li val[7] = (FLAC__byte)word;
702*600f14f4SXin Li #else
703*600f14f4SXin Li for(x = 0; x < FLAC__BYTES_PER_WORD; x++)
704*600f14f4SXin Li val[x] = (FLAC__byte)(word >> (8*(FLAC__BYTES_PER_WORD-x-1)));
705*600f14f4SXin Li #endif
706*600f14f4SXin Li val += FLAC__BYTES_PER_WORD;
707*600f14f4SXin Li nvals -= FLAC__BYTES_PER_WORD;
708*600f14f4SXin Li if(br->read_limit_set)
709*600f14f4SXin Li br->read_limit -= FLAC__BITS_PER_WORD;
710*600f14f4SXin Li }
711*600f14f4SXin Li else if(!bitreader_read_from_client_(br))
712*600f14f4SXin Li return false;
713*600f14f4SXin Li }
714*600f14f4SXin Li /* step 3: read any remainder from partial tail bytes */
715*600f14f4SXin Li while(nvals) {
716*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, 8))
717*600f14f4SXin Li return false;
718*600f14f4SXin Li *val++ = (FLAC__byte)x;
719*600f14f4SXin Li nvals--;
720*600f14f4SXin Li }
721*600f14f4SXin Li
722*600f14f4SXin Li return true;
723*600f14f4SXin Li }
724*600f14f4SXin Li
FLAC__bitreader_read_unary_unsigned(FLAC__BitReader * br,uint32_t * val)725*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_unary_unsigned(FLAC__BitReader *br, uint32_t *val)
726*600f14f4SXin Li #if 0 /* slow but readable version */
727*600f14f4SXin Li {
728*600f14f4SXin Li uint32_t bit;
729*600f14f4SXin Li
730*600f14f4SXin Li FLAC__ASSERT(0 != br);
731*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
732*600f14f4SXin Li
733*600f14f4SXin Li *val = 0;
734*600f14f4SXin Li while(1) {
735*600f14f4SXin Li if(!FLAC__bitreader_read_bit(br, &bit))
736*600f14f4SXin Li return false;
737*600f14f4SXin Li if(bit)
738*600f14f4SXin Li break;
739*600f14f4SXin Li else
740*600f14f4SXin Li *val++;
741*600f14f4SXin Li }
742*600f14f4SXin Li return true;
743*600f14f4SXin Li }
744*600f14f4SXin Li #else
745*600f14f4SXin Li {
746*600f14f4SXin Li uint32_t i;
747*600f14f4SXin Li
748*600f14f4SXin Li FLAC__ASSERT(0 != br);
749*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
750*600f14f4SXin Li
751*600f14f4SXin Li *val = 0;
752*600f14f4SXin Li while(1) {
753*600f14f4SXin Li while(br->consumed_words < br->words) { /* if we've not consumed up to a partial tail word... */
754*600f14f4SXin Li brword b = br->consumed_bits < FLAC__BITS_PER_WORD ? br->buffer[br->consumed_words] << br->consumed_bits : 0;
755*600f14f4SXin Li if(b) {
756*600f14f4SXin Li i = COUNT_ZERO_MSBS(b);
757*600f14f4SXin Li *val += i;
758*600f14f4SXin Li i++;
759*600f14f4SXin Li br->consumed_bits += i;
760*600f14f4SXin Li if(br->consumed_bits >= FLAC__BITS_PER_WORD) { /* faster way of testing if(br->consumed_bits == FLAC__BITS_PER_WORD) */
761*600f14f4SXin Li br->consumed_words++;
762*600f14f4SXin Li br->consumed_bits = 0;
763*600f14f4SXin Li }
764*600f14f4SXin Li return true;
765*600f14f4SXin Li }
766*600f14f4SXin Li else {
767*600f14f4SXin Li *val += FLAC__BITS_PER_WORD - br->consumed_bits;
768*600f14f4SXin Li br->consumed_words++;
769*600f14f4SXin Li br->consumed_bits = 0;
770*600f14f4SXin Li /* didn't find stop bit yet, have to keep going... */
771*600f14f4SXin Li }
772*600f14f4SXin Li }
773*600f14f4SXin Li /* at this point we've eaten up all the whole words; have to try
774*600f14f4SXin Li * reading through any tail bytes before calling the read callback.
775*600f14f4SXin Li * this is a repeat of the above logic adjusted for the fact we
776*600f14f4SXin Li * don't have a whole word. note though if the client is feeding
777*600f14f4SXin Li * us data a byte at a time (unlikely), br->consumed_bits may not
778*600f14f4SXin Li * be zero.
779*600f14f4SXin Li */
780*600f14f4SXin Li if(br->bytes*8 > br->consumed_bits) {
781*600f14f4SXin Li const uint32_t end = br->bytes * 8;
782*600f14f4SXin Li brword b = (br->buffer[br->consumed_words] & (FLAC__WORD_ALL_ONES << (FLAC__BITS_PER_WORD-end))) << br->consumed_bits;
783*600f14f4SXin Li if(b) {
784*600f14f4SXin Li i = COUNT_ZERO_MSBS(b);
785*600f14f4SXin Li *val += i;
786*600f14f4SXin Li i++;
787*600f14f4SXin Li br->consumed_bits += i;
788*600f14f4SXin Li FLAC__ASSERT(br->consumed_bits < FLAC__BITS_PER_WORD);
789*600f14f4SXin Li return true;
790*600f14f4SXin Li }
791*600f14f4SXin Li else {
792*600f14f4SXin Li *val += end - br->consumed_bits;
793*600f14f4SXin Li br->consumed_bits = end;
794*600f14f4SXin Li FLAC__ASSERT(br->consumed_bits < FLAC__BITS_PER_WORD);
795*600f14f4SXin Li /* didn't find stop bit yet, have to keep going... */
796*600f14f4SXin Li }
797*600f14f4SXin Li }
798*600f14f4SXin Li if(!bitreader_read_from_client_(br))
799*600f14f4SXin Li return false;
800*600f14f4SXin Li }
801*600f14f4SXin Li }
802*600f14f4SXin Li #endif
803*600f14f4SXin Li
804*600f14f4SXin Li #if 0 /* unused */
805*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_rice_signed(FLAC__BitReader *br, int *val, uint32_t parameter)
806*600f14f4SXin Li {
807*600f14f4SXin Li FLAC__uint32 lsbs = 0, msbs = 0;
808*600f14f4SXin Li uint32_t uval;
809*600f14f4SXin Li
810*600f14f4SXin Li FLAC__ASSERT(0 != br);
811*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
812*600f14f4SXin Li FLAC__ASSERT(parameter <= 31);
813*600f14f4SXin Li
814*600f14f4SXin Li /* read the unary MSBs and end bit */
815*600f14f4SXin Li if(!FLAC__bitreader_read_unary_unsigned(br, &msbs))
816*600f14f4SXin Li return false;
817*600f14f4SXin Li
818*600f14f4SXin Li /* read the binary LSBs */
819*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &lsbs, parameter))
820*600f14f4SXin Li return false;
821*600f14f4SXin Li
822*600f14f4SXin Li /* compose the value */
823*600f14f4SXin Li uval = (msbs << parameter) | lsbs;
824*600f14f4SXin Li if(uval & 1)
825*600f14f4SXin Li *val = -((int)(uval >> 1)) - 1;
826*600f14f4SXin Li else
827*600f14f4SXin Li *val = (int)(uval >> 1);
828*600f14f4SXin Li
829*600f14f4SXin Li return true;
830*600f14f4SXin Li }
831*600f14f4SXin Li #endif
832*600f14f4SXin Li
833*600f14f4SXin Li /* this is by far the most heavily used reader call. it ain't pretty but it's fast */
FLAC__bitreader_read_rice_signed_block(FLAC__BitReader * br,int vals[],uint32_t nvals,uint32_t parameter)834*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_rice_signed_block(FLAC__BitReader *br, int vals[], uint32_t nvals, uint32_t parameter)
835*600f14f4SXin Li #include "deduplication/bitreader_read_rice_signed_block.c"
836*600f14f4SXin Li
837*600f14f4SXin Li #ifdef FLAC__BMI2_SUPPORTED
838*600f14f4SXin Li FLAC__SSE_TARGET("bmi2")
839*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_rice_signed_block_bmi2(FLAC__BitReader *br, int vals[], uint32_t nvals, uint32_t parameter)
840*600f14f4SXin Li #include "deduplication/bitreader_read_rice_signed_block.c"
841*600f14f4SXin Li #endif
842*600f14f4SXin Li
843*600f14f4SXin Li #if 0 /* UNUSED */
844*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_golomb_signed(FLAC__BitReader *br, int *val, uint32_t parameter)
845*600f14f4SXin Li {
846*600f14f4SXin Li FLAC__uint32 lsbs = 0, msbs = 0;
847*600f14f4SXin Li uint32_t bit, uval, k;
848*600f14f4SXin Li
849*600f14f4SXin Li FLAC__ASSERT(0 != br);
850*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
851*600f14f4SXin Li
852*600f14f4SXin Li k = FLAC__bitmath_ilog2(parameter);
853*600f14f4SXin Li
854*600f14f4SXin Li /* read the unary MSBs and end bit */
855*600f14f4SXin Li if(!FLAC__bitreader_read_unary_unsigned(br, &msbs))
856*600f14f4SXin Li return false;
857*600f14f4SXin Li
858*600f14f4SXin Li /* read the binary LSBs */
859*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &lsbs, k))
860*600f14f4SXin Li return false;
861*600f14f4SXin Li
862*600f14f4SXin Li if(parameter == 1u<<k) {
863*600f14f4SXin Li /* compose the value */
864*600f14f4SXin Li uval = (msbs << k) | lsbs;
865*600f14f4SXin Li }
866*600f14f4SXin Li else {
867*600f14f4SXin Li uint32_t d = (1 << (k+1)) - parameter;
868*600f14f4SXin Li if(lsbs >= d) {
869*600f14f4SXin Li if(!FLAC__bitreader_read_bit(br, &bit))
870*600f14f4SXin Li return false;
871*600f14f4SXin Li lsbs <<= 1;
872*600f14f4SXin Li lsbs |= bit;
873*600f14f4SXin Li lsbs -= d;
874*600f14f4SXin Li }
875*600f14f4SXin Li /* compose the value */
876*600f14f4SXin Li uval = msbs * parameter + lsbs;
877*600f14f4SXin Li }
878*600f14f4SXin Li
879*600f14f4SXin Li /* unfold uint32_t to signed */
880*600f14f4SXin Li if(uval & 1)
881*600f14f4SXin Li *val = -((int)(uval >> 1)) - 1;
882*600f14f4SXin Li else
883*600f14f4SXin Li *val = (int)(uval >> 1);
884*600f14f4SXin Li
885*600f14f4SXin Li return true;
886*600f14f4SXin Li }
887*600f14f4SXin Li
888*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_golomb_unsigned(FLAC__BitReader *br, uint32_t *val, uint32_t parameter)
889*600f14f4SXin Li {
890*600f14f4SXin Li FLAC__uint32 lsbs, msbs = 0;
891*600f14f4SXin Li uint32_t bit, k;
892*600f14f4SXin Li
893*600f14f4SXin Li FLAC__ASSERT(0 != br);
894*600f14f4SXin Li FLAC__ASSERT(0 != br->buffer);
895*600f14f4SXin Li
896*600f14f4SXin Li k = FLAC__bitmath_ilog2(parameter);
897*600f14f4SXin Li
898*600f14f4SXin Li /* read the unary MSBs and end bit */
899*600f14f4SXin Li if(!FLAC__bitreader_read_unary_unsigned(br, &msbs))
900*600f14f4SXin Li return false;
901*600f14f4SXin Li
902*600f14f4SXin Li /* read the binary LSBs */
903*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &lsbs, k))
904*600f14f4SXin Li return false;
905*600f14f4SXin Li
906*600f14f4SXin Li if(parameter == 1u<<k) {
907*600f14f4SXin Li /* compose the value */
908*600f14f4SXin Li *val = (msbs << k) | lsbs;
909*600f14f4SXin Li }
910*600f14f4SXin Li else {
911*600f14f4SXin Li uint32_t d = (1 << (k+1)) - parameter;
912*600f14f4SXin Li if(lsbs >= d) {
913*600f14f4SXin Li if(!FLAC__bitreader_read_bit(br, &bit))
914*600f14f4SXin Li return false;
915*600f14f4SXin Li lsbs <<= 1;
916*600f14f4SXin Li lsbs |= bit;
917*600f14f4SXin Li lsbs -= d;
918*600f14f4SXin Li }
919*600f14f4SXin Li /* compose the value */
920*600f14f4SXin Li *val = msbs * parameter + lsbs;
921*600f14f4SXin Li }
922*600f14f4SXin Li
923*600f14f4SXin Li return true;
924*600f14f4SXin Li }
925*600f14f4SXin Li #endif /* UNUSED */
926*600f14f4SXin Li
927*600f14f4SXin Li /* on return, if *val == 0xffffffff then the utf-8 sequence was invalid, but the return value will be true */
928*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_utf8_uint32(FLAC__BitReader *br, FLAC__uint32 *val, FLAC__byte *raw, uint32_t *rawlen)
929*600f14f4SXin Li {
930*600f14f4SXin Li FLAC__uint32 v = 0;
931*600f14f4SXin Li FLAC__uint32 x;
932*600f14f4SXin Li uint32_t i;
933*600f14f4SXin Li
934*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, 8))
935*600f14f4SXin Li return false;
936*600f14f4SXin Li if(raw)
937*600f14f4SXin Li raw[(*rawlen)++] = (FLAC__byte)x;
938*600f14f4SXin Li if(!(x & 0x80)) { /* 0xxxxxxx */
939*600f14f4SXin Li v = x;
940*600f14f4SXin Li i = 0;
941*600f14f4SXin Li }
942*600f14f4SXin Li else if(x & 0xC0 && !(x & 0x20)) { /* 110xxxxx */
943*600f14f4SXin Li v = x & 0x1F;
944*600f14f4SXin Li i = 1;
945*600f14f4SXin Li }
946*600f14f4SXin Li else if(x & 0xE0 && !(x & 0x10)) { /* 1110xxxx */
947*600f14f4SXin Li v = x & 0x0F;
948*600f14f4SXin Li i = 2;
949*600f14f4SXin Li }
950*600f14f4SXin Li else if(x & 0xF0 && !(x & 0x08)) { /* 11110xxx */
951*600f14f4SXin Li v = x & 0x07;
952*600f14f4SXin Li i = 3;
953*600f14f4SXin Li }
954*600f14f4SXin Li else if(x & 0xF8 && !(x & 0x04)) { /* 111110xx */
955*600f14f4SXin Li v = x & 0x03;
956*600f14f4SXin Li i = 4;
957*600f14f4SXin Li }
958*600f14f4SXin Li else if(x & 0xFC && !(x & 0x02)) { /* 1111110x */
959*600f14f4SXin Li v = x & 0x01;
960*600f14f4SXin Li i = 5;
961*600f14f4SXin Li }
962*600f14f4SXin Li else {
963*600f14f4SXin Li *val = 0xffffffff;
964*600f14f4SXin Li return true;
965*600f14f4SXin Li }
966*600f14f4SXin Li for( ; i; i--) {
967*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, 8))
968*600f14f4SXin Li return false;
969*600f14f4SXin Li if(raw)
970*600f14f4SXin Li raw[(*rawlen)++] = (FLAC__byte)x;
971*600f14f4SXin Li if(!(x & 0x80) || (x & 0x40)) { /* 10xxxxxx */
972*600f14f4SXin Li *val = 0xffffffff;
973*600f14f4SXin Li return true;
974*600f14f4SXin Li }
975*600f14f4SXin Li v <<= 6;
976*600f14f4SXin Li v |= (x & 0x3F);
977*600f14f4SXin Li }
978*600f14f4SXin Li *val = v;
979*600f14f4SXin Li return true;
980*600f14f4SXin Li }
981*600f14f4SXin Li
982*600f14f4SXin Li /* on return, if *val == 0xffffffffffffffff then the utf-8 sequence was invalid, but the return value will be true */
FLAC__bitreader_read_utf8_uint64(FLAC__BitReader * br,FLAC__uint64 * val,FLAC__byte * raw,uint32_t * rawlen)983*600f14f4SXin Li FLAC__bool FLAC__bitreader_read_utf8_uint64(FLAC__BitReader *br, FLAC__uint64 *val, FLAC__byte *raw, uint32_t *rawlen)
984*600f14f4SXin Li {
985*600f14f4SXin Li FLAC__uint64 v = 0;
986*600f14f4SXin Li FLAC__uint32 x;
987*600f14f4SXin Li uint32_t i;
988*600f14f4SXin Li
989*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, 8))
990*600f14f4SXin Li return false;
991*600f14f4SXin Li if(raw)
992*600f14f4SXin Li raw[(*rawlen)++] = (FLAC__byte)x;
993*600f14f4SXin Li if(!(x & 0x80)) { /* 0xxxxxxx */
994*600f14f4SXin Li v = x;
995*600f14f4SXin Li i = 0;
996*600f14f4SXin Li }
997*600f14f4SXin Li else if(x & 0xC0 && !(x & 0x20)) { /* 110xxxxx */
998*600f14f4SXin Li v = x & 0x1F;
999*600f14f4SXin Li i = 1;
1000*600f14f4SXin Li }
1001*600f14f4SXin Li else if(x & 0xE0 && !(x & 0x10)) { /* 1110xxxx */
1002*600f14f4SXin Li v = x & 0x0F;
1003*600f14f4SXin Li i = 2;
1004*600f14f4SXin Li }
1005*600f14f4SXin Li else if(x & 0xF0 && !(x & 0x08)) { /* 11110xxx */
1006*600f14f4SXin Li v = x & 0x07;
1007*600f14f4SXin Li i = 3;
1008*600f14f4SXin Li }
1009*600f14f4SXin Li else if(x & 0xF8 && !(x & 0x04)) { /* 111110xx */
1010*600f14f4SXin Li v = x & 0x03;
1011*600f14f4SXin Li i = 4;
1012*600f14f4SXin Li }
1013*600f14f4SXin Li else if(x & 0xFC && !(x & 0x02)) { /* 1111110x */
1014*600f14f4SXin Li v = x & 0x01;
1015*600f14f4SXin Li i = 5;
1016*600f14f4SXin Li }
1017*600f14f4SXin Li else if(x & 0xFE && !(x & 0x01)) { /* 11111110 */
1018*600f14f4SXin Li v = 0;
1019*600f14f4SXin Li i = 6;
1020*600f14f4SXin Li }
1021*600f14f4SXin Li else {
1022*600f14f4SXin Li *val = FLAC__U64L(0xffffffffffffffff);
1023*600f14f4SXin Li return true;
1024*600f14f4SXin Li }
1025*600f14f4SXin Li for( ; i; i--) {
1026*600f14f4SXin Li if(!FLAC__bitreader_read_raw_uint32(br, &x, 8))
1027*600f14f4SXin Li return false;
1028*600f14f4SXin Li if(raw)
1029*600f14f4SXin Li raw[(*rawlen)++] = (FLAC__byte)x;
1030*600f14f4SXin Li if(!(x & 0x80) || (x & 0x40)) { /* 10xxxxxx */
1031*600f14f4SXin Li *val = FLAC__U64L(0xffffffffffffffff);
1032*600f14f4SXin Li return true;
1033*600f14f4SXin Li }
1034*600f14f4SXin Li v <<= 6;
1035*600f14f4SXin Li v |= (x & 0x3F);
1036*600f14f4SXin Li }
1037*600f14f4SXin Li *val = v;
1038*600f14f4SXin Li return true;
1039*600f14f4SXin Li }
1040*600f14f4SXin Li
1041*600f14f4SXin Li /* These functions are declared inline in this file but are also callable as
1042*600f14f4SXin Li * externs from elsewhere.
1043*600f14f4SXin Li * According to the C99 spec, section 6.7.4, simply providing a function
1044*600f14f4SXin Li * prototype in a header file without 'inline' and making the function inline
1045*600f14f4SXin Li * in this file should be sufficient.
1046*600f14f4SXin Li * Unfortunately, the Microsoft VS compiler doesn't pick them up externally. To
1047*600f14f4SXin Li * fix that we add extern declarations here.
1048*600f14f4SXin Li */
1049*600f14f4SXin Li extern FLAC__bool FLAC__bitreader_is_consumed_byte_aligned(const FLAC__BitReader *br);
1050*600f14f4SXin Li extern uint32_t FLAC__bitreader_bits_left_for_byte_alignment(const FLAC__BitReader *br);
1051*600f14f4SXin Li extern uint32_t FLAC__bitreader_get_input_bits_unconsumed(const FLAC__BitReader *br);
1052*600f14f4SXin Li extern FLAC__bool FLAC__bitreader_read_uint32_little_endian(FLAC__BitReader *br, FLAC__uint32 *val);
1053