1 /* libFLAC - Free Lossless Audio Codec library
2  * Copyright (C) 2000-2009  Josh Coalson
3  * Copyright (C) 2011-2014  Xiph.Org Foundation
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * - Redistributions of source code must retain the above copyright
10  * notice, this list of conditions and the following disclaimer.
11  *
12  * - Redistributions in binary form must reproduce the above copyright
13  * notice, this list of conditions and the following disclaimer in the
14  * documentation and/or other materials provided with the distribution.
15  *
16  * - Neither the name of the Xiph.org Foundation nor the names of its
17  * contributors may be used to endorse or promote products derived from
18  * this software without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23  * A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
24  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
25  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
26  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
27  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
28  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
29  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
30  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31  */
32 
33 #ifdef HAVE_CONFIG_H
34 #  include <config.h>
35 #endif
36 
37 #include <stdio.h>
38 #include <stdlib.h> /* for malloc() */
39 #include <string.h> /* for memset/memcpy() */
40 #include <sys/stat.h> /* for stat() */
41 #include <sys/types.h> /* for off_t */
42 #include "share/compat.h"
43 #include "FLAC/assert.h"
44 #include "share/alloc.h"
45 #include "protected/stream_decoder.h"
46 #include "private/bitreader.h"
47 #include "private/bitmath.h"
48 #include "private/cpu.h"
49 #include "private/crc.h"
50 #include "private/fixed.h"
51 #include "private/format.h"
52 #include "private/lpc.h"
53 #include "private/md5.h"
54 #include "private/memory.h"
55 #include "private/macros.h"
56 
57 
58 /* technically this should be in an "export.c" but this is convenient enough */
59 FLAC_API int FLAC_API_SUPPORTS_OGG_FLAC =
60 #if FLAC__HAS_OGG
61 	1
62 #else
63 	0
64 #endif
65 ;
66 
67 
68 /***********************************************************************
69  *
70  * Private static data
71  *
72  ***********************************************************************/
73 
74 static const FLAC__byte ID3V2_TAG_[3] = { 'I', 'D', '3' };
75 
76 /***********************************************************************
77  *
78  * Private class method prototypes
79  *
80  ***********************************************************************/
81 
82 static void set_defaults_(FLAC__StreamDecoder *decoder);
83 static FILE *get_binary_stdin_(void);
84 static FLAC__bool allocate_output_(FLAC__StreamDecoder *decoder, unsigned size, unsigned channels);
85 static FLAC__bool has_id_filtered_(FLAC__StreamDecoder *decoder, FLAC__byte *id);
86 static FLAC__bool find_metadata_(FLAC__StreamDecoder *decoder);
87 static FLAC__bool read_metadata_(FLAC__StreamDecoder *decoder);
88 static FLAC__bool read_metadata_streaminfo_(FLAC__StreamDecoder *decoder, FLAC__bool is_last, unsigned length);
89 static FLAC__bool read_metadata_seektable_(FLAC__StreamDecoder *decoder, FLAC__bool is_last, unsigned length);
90 static FLAC__bool read_metadata_vorbiscomment_(FLAC__StreamDecoder *decoder, FLAC__StreamMetadata_VorbisComment *obj, unsigned length);
91 static FLAC__bool read_metadata_cuesheet_(FLAC__StreamDecoder *decoder, FLAC__StreamMetadata_CueSheet *obj);
92 static FLAC__bool read_metadata_picture_(FLAC__StreamDecoder *decoder, FLAC__StreamMetadata_Picture *obj);
93 static FLAC__bool skip_id3v2_tag_(FLAC__StreamDecoder *decoder);
94 static FLAC__bool frame_sync_(FLAC__StreamDecoder *decoder);
95 static FLAC__bool read_frame_(FLAC__StreamDecoder *decoder, FLAC__bool *got_a_frame, FLAC__bool do_full_decode);
96 static FLAC__bool read_frame_header_(FLAC__StreamDecoder *decoder);
97 static FLAC__bool read_subframe_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, FLAC__bool do_full_decode);
98 static FLAC__bool read_subframe_constant_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, FLAC__bool do_full_decode);
99 static FLAC__bool read_subframe_fixed_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, const unsigned order, FLAC__bool do_full_decode);
100 static FLAC__bool read_subframe_lpc_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, const unsigned order, FLAC__bool do_full_decode);
101 static FLAC__bool read_subframe_verbatim_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, FLAC__bool do_full_decode);
102 static FLAC__bool read_residual_partitioned_rice_(FLAC__StreamDecoder *decoder, unsigned predictor_order, unsigned partition_order, FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents, FLAC__int32 *residual, FLAC__bool is_extended);
103 static FLAC__bool read_zero_padding_(FLAC__StreamDecoder *decoder);
104 static FLAC__bool read_callback_(FLAC__byte buffer[], size_t *bytes, void *client_data);
105 #if FLAC__HAS_OGG
106 static FLAC__StreamDecoderReadStatus read_callback_ogg_aspect_(const FLAC__StreamDecoder *decoder, FLAC__byte buffer[], size_t *bytes);
107 static FLAC__OggDecoderAspectReadStatus read_callback_proxy_(const void *void_decoder, FLAC__byte buffer[], size_t *bytes, void *client_data);
108 #endif
109 static FLAC__StreamDecoderWriteStatus write_audio_frame_to_client_(FLAC__StreamDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[]);
110 static void send_error_to_client_(const FLAC__StreamDecoder *decoder, FLAC__StreamDecoderErrorStatus status);
111 static FLAC__bool seek_to_absolute_sample_(FLAC__StreamDecoder *decoder, FLAC__uint64 stream_length, FLAC__uint64 target_sample);
112 #if FLAC__HAS_OGG
113 static FLAC__bool seek_to_absolute_sample_ogg_(FLAC__StreamDecoder *decoder, FLAC__uint64 stream_length, FLAC__uint64 target_sample);
114 #endif
115 static FLAC__StreamDecoderReadStatus file_read_callback_(const FLAC__StreamDecoder *decoder, FLAC__byte buffer[], size_t *bytes, void *client_data);
116 static FLAC__StreamDecoderSeekStatus file_seek_callback_(const FLAC__StreamDecoder *decoder, FLAC__uint64 absolute_byte_offset, void *client_data);
117 static FLAC__StreamDecoderTellStatus file_tell_callback_(const FLAC__StreamDecoder *decoder, FLAC__uint64 *absolute_byte_offset, void *client_data);
118 static FLAC__StreamDecoderLengthStatus file_length_callback_(const FLAC__StreamDecoder *decoder, FLAC__uint64 *stream_length, void *client_data);
119 static FLAC__bool file_eof_callback_(const FLAC__StreamDecoder *decoder, void *client_data);
120 
121 /***********************************************************************
122  *
123  * Private class data
124  *
125  ***********************************************************************/
126 
127 typedef struct FLAC__StreamDecoderPrivate {
128 #if FLAC__HAS_OGG
129 	FLAC__bool is_ogg;
130 #endif
131 	FLAC__StreamDecoderReadCallback read_callback;
132 	FLAC__StreamDecoderSeekCallback seek_callback;
133 	FLAC__StreamDecoderTellCallback tell_callback;
134 	FLAC__StreamDecoderLengthCallback length_callback;
135 	FLAC__StreamDecoderEofCallback eof_callback;
136 	FLAC__StreamDecoderWriteCallback write_callback;
137 	FLAC__StreamDecoderMetadataCallback metadata_callback;
138 	FLAC__StreamDecoderErrorCallback error_callback;
139 	/* generic 32-bit datapath: */
140 	void (*local_lpc_restore_signal)(const FLAC__int32 residual[], unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 data[]);
141 	/* generic 64-bit datapath: */
142 	void (*local_lpc_restore_signal_64bit)(const FLAC__int32 residual[], unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 data[]);
143 	/* for use when the signal is <= 16 bits-per-sample, or <= 15 bits-per-sample on a side channel (which requires 1 extra bit): */
144 	void (*local_lpc_restore_signal_16bit)(const FLAC__int32 residual[], unsigned data_len, const FLAC__int32 qlp_coeff[], unsigned order, int lp_quantization, FLAC__int32 data[]);
145 	void *client_data;
146 	FILE *file; /* only used if FLAC__stream_decoder_init_file()/FLAC__stream_decoder_init_file() called, else NULL */
147 	FLAC__BitReader *input;
148 	FLAC__int32 *output[FLAC__MAX_CHANNELS];
149 	FLAC__int32 *residual[FLAC__MAX_CHANNELS]; /* WATCHOUT: these are the aligned pointers; the real pointers that should be free()'d are residual_unaligned[] below */
150 	FLAC__EntropyCodingMethod_PartitionedRiceContents partitioned_rice_contents[FLAC__MAX_CHANNELS];
151 	unsigned output_capacity, output_channels;
152 	FLAC__uint32 fixed_block_size, next_fixed_block_size;
153 	FLAC__uint64 samples_decoded;
154 	FLAC__bool has_stream_info, has_seek_table;
155 	FLAC__StreamMetadata stream_info;
156 	FLAC__StreamMetadata seek_table;
157 	FLAC__bool metadata_filter[128]; /* MAGIC number 128 == total number of metadata block types == 1 << 7 */
158 	FLAC__byte *metadata_filter_ids;
159 	size_t metadata_filter_ids_count, metadata_filter_ids_capacity; /* units for both are IDs, not bytes */
160 	FLAC__Frame frame;
161 	FLAC__bool cached; /* true if there is a byte in lookahead */
162 	FLAC__CPUInfo cpuinfo;
163 	FLAC__byte header_warmup[2]; /* contains the sync code and reserved bits */
164 	FLAC__byte lookahead; /* temp storage when we need to look ahead one byte in the stream */
165 	/* unaligned (original) pointers to allocated data */
166 	FLAC__int32 *residual_unaligned[FLAC__MAX_CHANNELS];
167 	FLAC__bool do_md5_checking; /* initially gets protected_->md5_checking but is turned off after a seek or if the metadata has a zero MD5 */
168 	FLAC__bool internal_reset_hack; /* used only during init() so we can call reset to set up the decoder without rewinding the input */
169 	FLAC__bool is_seeking;
170 	FLAC__MD5Context md5context;
171 	FLAC__byte computed_md5sum[16]; /* this is the sum we computed from the decoded data */
172 	/* (the rest of these are only used for seeking) */
173 	FLAC__Frame last_frame; /* holds the info of the last frame we seeked to */
174 	FLAC__uint64 first_frame_offset; /* hint to the seek routine of where in the stream the first audio frame starts */
175 	FLAC__uint64 target_sample;
176 	unsigned unparseable_frame_count; /* used to tell whether we're decoding a future version of FLAC or just got a bad sync */
177 #if FLAC__HAS_OGG
178 	FLAC__bool got_a_frame; /* hack needed in Ogg FLAC seek routine to check when process_single() actually writes a frame */
179 #endif
180 } FLAC__StreamDecoderPrivate;
181 
182 /***********************************************************************
183  *
184  * Public static class data
185  *
186  ***********************************************************************/
187 
188 FLAC_API const char * const FLAC__StreamDecoderStateString[] = {
189 	"FLAC__STREAM_DECODER_SEARCH_FOR_METADATA",
190 	"FLAC__STREAM_DECODER_READ_METADATA",
191 	"FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC",
192 	"FLAC__STREAM_DECODER_READ_FRAME",
193 	"FLAC__STREAM_DECODER_END_OF_STREAM",
194 	"FLAC__STREAM_DECODER_OGG_ERROR",
195 	"FLAC__STREAM_DECODER_SEEK_ERROR",
196 	"FLAC__STREAM_DECODER_ABORTED",
197 	"FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR",
198 	"FLAC__STREAM_DECODER_UNINITIALIZED"
199 };
200 
201 FLAC_API const char * const FLAC__StreamDecoderInitStatusString[] = {
202 	"FLAC__STREAM_DECODER_INIT_STATUS_OK",
203 	"FLAC__STREAM_DECODER_INIT_STATUS_UNSUPPORTED_CONTAINER",
204 	"FLAC__STREAM_DECODER_INIT_STATUS_INVALID_CALLBACKS",
205 	"FLAC__STREAM_DECODER_INIT_STATUS_MEMORY_ALLOCATION_ERROR",
206 	"FLAC__STREAM_DECODER_INIT_STATUS_ERROR_OPENING_FILE",
207 	"FLAC__STREAM_DECODER_INIT_STATUS_ALREADY_INITIALIZED"
208 };
209 
210 FLAC_API const char * const FLAC__StreamDecoderReadStatusString[] = {
211 	"FLAC__STREAM_DECODER_READ_STATUS_CONTINUE",
212 	"FLAC__STREAM_DECODER_READ_STATUS_END_OF_STREAM",
213 	"FLAC__STREAM_DECODER_READ_STATUS_ABORT"
214 };
215 
216 FLAC_API const char * const FLAC__StreamDecoderSeekStatusString[] = {
217 	"FLAC__STREAM_DECODER_SEEK_STATUS_OK",
218 	"FLAC__STREAM_DECODER_SEEK_STATUS_ERROR",
219 	"FLAC__STREAM_DECODER_SEEK_STATUS_UNSUPPORTED"
220 };
221 
222 FLAC_API const char * const FLAC__StreamDecoderTellStatusString[] = {
223 	"FLAC__STREAM_DECODER_TELL_STATUS_OK",
224 	"FLAC__STREAM_DECODER_TELL_STATUS_ERROR",
225 	"FLAC__STREAM_DECODER_TELL_STATUS_UNSUPPORTED"
226 };
227 
228 FLAC_API const char * const FLAC__StreamDecoderLengthStatusString[] = {
229 	"FLAC__STREAM_DECODER_LENGTH_STATUS_OK",
230 	"FLAC__STREAM_DECODER_LENGTH_STATUS_ERROR",
231 	"FLAC__STREAM_DECODER_LENGTH_STATUS_UNSUPPORTED"
232 };
233 
234 FLAC_API const char * const FLAC__StreamDecoderWriteStatusString[] = {
235 	"FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE",
236 	"FLAC__STREAM_DECODER_WRITE_STATUS_ABORT"
237 };
238 
239 FLAC_API const char * const FLAC__StreamDecoderErrorStatusString[] = {
240 	"FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC",
241 	"FLAC__STREAM_DECODER_ERROR_STATUS_BAD_HEADER",
242 	"FLAC__STREAM_DECODER_ERROR_STATUS_FRAME_CRC_MISMATCH",
243 	"FLAC__STREAM_DECODER_ERROR_STATUS_UNPARSEABLE_STREAM"
244 };
245 
246 /***********************************************************************
247  *
248  * Class constructor/destructor
249  *
250  ***********************************************************************/
FLAC__stream_decoder_new(void)251 FLAC_API FLAC__StreamDecoder *FLAC__stream_decoder_new(void)
252 {
253 	FLAC__StreamDecoder *decoder;
254 	unsigned i;
255 
256 	FLAC__ASSERT(sizeof(int) >= 4); /* we want to die right away if this is not true */
257 
258 	decoder = calloc(1, sizeof(FLAC__StreamDecoder));
259 	if(decoder == 0) {
260 		return 0;
261 	}
262 
263 	decoder->protected_ = calloc(1, sizeof(FLAC__StreamDecoderProtected));
264 	if(decoder->protected_ == 0) {
265 		free(decoder);
266 		return 0;
267 	}
268 
269 	decoder->private_ = calloc(1, sizeof(FLAC__StreamDecoderPrivate));
270 	if(decoder->private_ == 0) {
271 		free(decoder->protected_);
272 		free(decoder);
273 		return 0;
274 	}
275 
276 	decoder->private_->input = FLAC__bitreader_new();
277 	if(decoder->private_->input == 0) {
278 		free(decoder->private_);
279 		free(decoder->protected_);
280 		free(decoder);
281 		return 0;
282 	}
283 
284 	decoder->private_->metadata_filter_ids_capacity = 16;
285 	if(0 == (decoder->private_->metadata_filter_ids = malloc((FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8) * decoder->private_->metadata_filter_ids_capacity))) {
286 		FLAC__bitreader_delete(decoder->private_->input);
287 		free(decoder->private_);
288 		free(decoder->protected_);
289 		free(decoder);
290 		return 0;
291 	}
292 
293 	for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
294 		decoder->private_->output[i] = 0;
295 		decoder->private_->residual_unaligned[i] = decoder->private_->residual[i] = 0;
296 	}
297 
298 	decoder->private_->output_capacity = 0;
299 	decoder->private_->output_channels = 0;
300 	decoder->private_->has_seek_table = false;
301 
302 	for(i = 0; i < FLAC__MAX_CHANNELS; i++)
303 		FLAC__format_entropy_coding_method_partitioned_rice_contents_init(&decoder->private_->partitioned_rice_contents[i]);
304 
305 	decoder->private_->file = 0;
306 
307 	set_defaults_(decoder);
308 
309 	decoder->protected_->state = FLAC__STREAM_DECODER_UNINITIALIZED;
310 
311 	return decoder;
312 }
313 
FLAC__stream_decoder_delete(FLAC__StreamDecoder * decoder)314 FLAC_API void FLAC__stream_decoder_delete(FLAC__StreamDecoder *decoder)
315 {
316 	unsigned i;
317 
318 	if (decoder == NULL)
319 		return ;
320 
321 	FLAC__ASSERT(0 != decoder->protected_);
322 	FLAC__ASSERT(0 != decoder->private_);
323 	FLAC__ASSERT(0 != decoder->private_->input);
324 
325 	(void)FLAC__stream_decoder_finish(decoder);
326 
327 	if(0 != decoder->private_->metadata_filter_ids)
328 		free(decoder->private_->metadata_filter_ids);
329 
330 	FLAC__bitreader_delete(decoder->private_->input);
331 
332 	for(i = 0; i < FLAC__MAX_CHANNELS; i++)
333 		FLAC__format_entropy_coding_method_partitioned_rice_contents_clear(&decoder->private_->partitioned_rice_contents[i]);
334 
335 	free(decoder->private_);
336 	free(decoder->protected_);
337 	free(decoder);
338 }
339 
340 /***********************************************************************
341  *
342  * Public class methods
343  *
344  ***********************************************************************/
345 
init_stream_internal_(FLAC__StreamDecoder * decoder,FLAC__StreamDecoderReadCallback read_callback,FLAC__StreamDecoderSeekCallback seek_callback,FLAC__StreamDecoderTellCallback tell_callback,FLAC__StreamDecoderLengthCallback length_callback,FLAC__StreamDecoderEofCallback eof_callback,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data,FLAC__bool is_ogg)346 static FLAC__StreamDecoderInitStatus init_stream_internal_(
347 	FLAC__StreamDecoder *decoder,
348 	FLAC__StreamDecoderReadCallback read_callback,
349 	FLAC__StreamDecoderSeekCallback seek_callback,
350 	FLAC__StreamDecoderTellCallback tell_callback,
351 	FLAC__StreamDecoderLengthCallback length_callback,
352 	FLAC__StreamDecoderEofCallback eof_callback,
353 	FLAC__StreamDecoderWriteCallback write_callback,
354 	FLAC__StreamDecoderMetadataCallback metadata_callback,
355 	FLAC__StreamDecoderErrorCallback error_callback,
356 	void *client_data,
357 	FLAC__bool is_ogg
358 )
359 {
360 	FLAC__ASSERT(0 != decoder);
361 
362 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
363 		return FLAC__STREAM_DECODER_INIT_STATUS_ALREADY_INITIALIZED;
364 
365 #if !FLAC__HAS_OGG
366 	if(is_ogg)
367 		return FLAC__STREAM_DECODER_INIT_STATUS_UNSUPPORTED_CONTAINER;
368 #endif
369 
370 	if(
371 		0 == read_callback ||
372 		0 == write_callback ||
373 		0 == error_callback ||
374 		(seek_callback && (0 == tell_callback || 0 == length_callback || 0 == eof_callback))
375 	)
376 		return FLAC__STREAM_DECODER_INIT_STATUS_INVALID_CALLBACKS;
377 
378 #if FLAC__HAS_OGG
379 	decoder->private_->is_ogg = is_ogg;
380 	if(is_ogg && !FLAC__ogg_decoder_aspect_init(&decoder->protected_->ogg_decoder_aspect))
381 		return decoder->protected_->initstate = FLAC__STREAM_DECODER_INIT_STATUS_ERROR_OPENING_FILE;
382 #endif
383 
384 	/*
385 	 * get the CPU info and set the function pointers
386 	 */
387 	FLAC__cpu_info(&decoder->private_->cpuinfo);
388 	/* first default to the non-asm routines */
389 	decoder->private_->local_lpc_restore_signal = FLAC__lpc_restore_signal;
390 	decoder->private_->local_lpc_restore_signal_64bit = FLAC__lpc_restore_signal_wide;
391 	decoder->private_->local_lpc_restore_signal_16bit = FLAC__lpc_restore_signal;
392 	/* now override with asm where appropriate */
393 #ifndef FLAC__NO_ASM
394 	if(decoder->private_->cpuinfo.use_asm) {
395 #ifdef FLAC__CPU_IA32
396 		FLAC__ASSERT(decoder->private_->cpuinfo.type == FLAC__CPUINFO_TYPE_IA32);
397 #ifdef FLAC__HAS_NASM
398 		decoder->private_->local_lpc_restore_signal_64bit = FLAC__lpc_restore_signal_wide_asm_ia32; /* OPT_IA32: was really necessary for GCC < 4.9 */
399 		if(decoder->private_->cpuinfo.ia32.mmx) {
400 			decoder->private_->local_lpc_restore_signal = FLAC__lpc_restore_signal_asm_ia32;
401 			decoder->private_->local_lpc_restore_signal_16bit = FLAC__lpc_restore_signal_asm_ia32_mmx;
402 		}
403 		else {
404 			decoder->private_->local_lpc_restore_signal = FLAC__lpc_restore_signal_asm_ia32;
405 			decoder->private_->local_lpc_restore_signal_16bit = FLAC__lpc_restore_signal_asm_ia32;
406 		}
407 #endif
408 #ifdef FLAC__HAS_X86INTRIN
409 # if defined FLAC__SSE2_SUPPORTED && !defined FLAC__HAS_NASM /* OPT_SSE: not better than MMX asm */
410 		if(decoder->private_->cpuinfo.ia32.sse2) {
411 			decoder->private_->local_lpc_restore_signal_16bit = FLAC__lpc_restore_signal_16_intrin_sse2;
412 		}
413 # endif
414 # if defined FLAC__SSE4_1_SUPPORTED
415 		if(decoder->private_->cpuinfo.ia32.sse41) {
416 			decoder->private_->local_lpc_restore_signal_64bit = FLAC__lpc_restore_signal_wide_intrin_sse41;
417 		}
418 # endif
419 #endif
420 #elif defined FLAC__CPU_X86_64
421 		FLAC__ASSERT(decoder->private_->cpuinfo.type == FLAC__CPUINFO_TYPE_X86_64);
422 		/* No useful SSE optimizations yet */
423 #endif
424 	}
425 #endif
426 
427 	/* from here on, errors are fatal */
428 
429 	if(!FLAC__bitreader_init(decoder->private_->input, read_callback_, decoder)) {
430 		decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
431 		return FLAC__STREAM_DECODER_INIT_STATUS_MEMORY_ALLOCATION_ERROR;
432 	}
433 
434 	decoder->private_->read_callback = read_callback;
435 	decoder->private_->seek_callback = seek_callback;
436 	decoder->private_->tell_callback = tell_callback;
437 	decoder->private_->length_callback = length_callback;
438 	decoder->private_->eof_callback = eof_callback;
439 	decoder->private_->write_callback = write_callback;
440 	decoder->private_->metadata_callback = metadata_callback;
441 	decoder->private_->error_callback = error_callback;
442 	decoder->private_->client_data = client_data;
443 	decoder->private_->fixed_block_size = decoder->private_->next_fixed_block_size = 0;
444 	decoder->private_->samples_decoded = 0;
445 	decoder->private_->has_stream_info = false;
446 	decoder->private_->cached = false;
447 
448 	decoder->private_->do_md5_checking = decoder->protected_->md5_checking;
449 	decoder->private_->is_seeking = false;
450 
451 	decoder->private_->internal_reset_hack = true; /* so the following reset does not try to rewind the input */
452 	if(!FLAC__stream_decoder_reset(decoder)) {
453 		/* above call sets the state for us */
454 		return FLAC__STREAM_DECODER_INIT_STATUS_MEMORY_ALLOCATION_ERROR;
455 	}
456 
457 	return FLAC__STREAM_DECODER_INIT_STATUS_OK;
458 }
459 
FLAC__stream_decoder_init_stream(FLAC__StreamDecoder * decoder,FLAC__StreamDecoderReadCallback read_callback,FLAC__StreamDecoderSeekCallback seek_callback,FLAC__StreamDecoderTellCallback tell_callback,FLAC__StreamDecoderLengthCallback length_callback,FLAC__StreamDecoderEofCallback eof_callback,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data)460 FLAC_API FLAC__StreamDecoderInitStatus FLAC__stream_decoder_init_stream(
461 	FLAC__StreamDecoder *decoder,
462 	FLAC__StreamDecoderReadCallback read_callback,
463 	FLAC__StreamDecoderSeekCallback seek_callback,
464 	FLAC__StreamDecoderTellCallback tell_callback,
465 	FLAC__StreamDecoderLengthCallback length_callback,
466 	FLAC__StreamDecoderEofCallback eof_callback,
467 	FLAC__StreamDecoderWriteCallback write_callback,
468 	FLAC__StreamDecoderMetadataCallback metadata_callback,
469 	FLAC__StreamDecoderErrorCallback error_callback,
470 	void *client_data
471 )
472 {
473 	return init_stream_internal_(
474 		decoder,
475 		read_callback,
476 		seek_callback,
477 		tell_callback,
478 		length_callback,
479 		eof_callback,
480 		write_callback,
481 		metadata_callback,
482 		error_callback,
483 		client_data,
484 		/*is_ogg=*/false
485 	);
486 }
487 
FLAC__stream_decoder_init_ogg_stream(FLAC__StreamDecoder * decoder,FLAC__StreamDecoderReadCallback read_callback,FLAC__StreamDecoderSeekCallback seek_callback,FLAC__StreamDecoderTellCallback tell_callback,FLAC__StreamDecoderLengthCallback length_callback,FLAC__StreamDecoderEofCallback eof_callback,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data)488 FLAC_API FLAC__StreamDecoderInitStatus FLAC__stream_decoder_init_ogg_stream(
489 	FLAC__StreamDecoder *decoder,
490 	FLAC__StreamDecoderReadCallback read_callback,
491 	FLAC__StreamDecoderSeekCallback seek_callback,
492 	FLAC__StreamDecoderTellCallback tell_callback,
493 	FLAC__StreamDecoderLengthCallback length_callback,
494 	FLAC__StreamDecoderEofCallback eof_callback,
495 	FLAC__StreamDecoderWriteCallback write_callback,
496 	FLAC__StreamDecoderMetadataCallback metadata_callback,
497 	FLAC__StreamDecoderErrorCallback error_callback,
498 	void *client_data
499 )
500 {
501 	return init_stream_internal_(
502 		decoder,
503 		read_callback,
504 		seek_callback,
505 		tell_callback,
506 		length_callback,
507 		eof_callback,
508 		write_callback,
509 		metadata_callback,
510 		error_callback,
511 		client_data,
512 		/*is_ogg=*/true
513 	);
514 }
515 
init_FILE_internal_(FLAC__StreamDecoder * decoder,FILE * file,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data,FLAC__bool is_ogg)516 static FLAC__StreamDecoderInitStatus init_FILE_internal_(
517 	FLAC__StreamDecoder *decoder,
518 	FILE *file,
519 	FLAC__StreamDecoderWriteCallback write_callback,
520 	FLAC__StreamDecoderMetadataCallback metadata_callback,
521 	FLAC__StreamDecoderErrorCallback error_callback,
522 	void *client_data,
523 	FLAC__bool is_ogg
524 )
525 {
526 	FLAC__ASSERT(0 != decoder);
527 	FLAC__ASSERT(0 != file);
528 
529 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
530 		return decoder->protected_->initstate = FLAC__STREAM_DECODER_INIT_STATUS_ALREADY_INITIALIZED;
531 
532 	if(0 == write_callback || 0 == error_callback)
533 		return decoder->protected_->initstate = FLAC__STREAM_DECODER_INIT_STATUS_INVALID_CALLBACKS;
534 
535 	/*
536 	 * To make sure that our file does not go unclosed after an error, we
537 	 * must assign the FILE pointer before any further error can occur in
538 	 * this routine.
539 	 */
540 	if(file == stdin)
541 		file = get_binary_stdin_(); /* just to be safe */
542 
543 	decoder->private_->file = file;
544 
545 	return init_stream_internal_(
546 		decoder,
547 		file_read_callback_,
548 		decoder->private_->file == stdin? 0: file_seek_callback_,
549 		decoder->private_->file == stdin? 0: file_tell_callback_,
550 		decoder->private_->file == stdin? 0: file_length_callback_,
551 		file_eof_callback_,
552 		write_callback,
553 		metadata_callback,
554 		error_callback,
555 		client_data,
556 		is_ogg
557 	);
558 }
559 
FLAC__stream_decoder_init_FILE(FLAC__StreamDecoder * decoder,FILE * file,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data)560 FLAC_API FLAC__StreamDecoderInitStatus FLAC__stream_decoder_init_FILE(
561 	FLAC__StreamDecoder *decoder,
562 	FILE *file,
563 	FLAC__StreamDecoderWriteCallback write_callback,
564 	FLAC__StreamDecoderMetadataCallback metadata_callback,
565 	FLAC__StreamDecoderErrorCallback error_callback,
566 	void *client_data
567 )
568 {
569 	return init_FILE_internal_(decoder, file, write_callback, metadata_callback, error_callback, client_data, /*is_ogg=*/false);
570 }
571 
FLAC__stream_decoder_init_ogg_FILE(FLAC__StreamDecoder * decoder,FILE * file,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data)572 FLAC_API FLAC__StreamDecoderInitStatus FLAC__stream_decoder_init_ogg_FILE(
573 	FLAC__StreamDecoder *decoder,
574 	FILE *file,
575 	FLAC__StreamDecoderWriteCallback write_callback,
576 	FLAC__StreamDecoderMetadataCallback metadata_callback,
577 	FLAC__StreamDecoderErrorCallback error_callback,
578 	void *client_data
579 )
580 {
581 	return init_FILE_internal_(decoder, file, write_callback, metadata_callback, error_callback, client_data, /*is_ogg=*/true);
582 }
583 
init_file_internal_(FLAC__StreamDecoder * decoder,const char * filename,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data,FLAC__bool is_ogg)584 static FLAC__StreamDecoderInitStatus init_file_internal_(
585 	FLAC__StreamDecoder *decoder,
586 	const char *filename,
587 	FLAC__StreamDecoderWriteCallback write_callback,
588 	FLAC__StreamDecoderMetadataCallback metadata_callback,
589 	FLAC__StreamDecoderErrorCallback error_callback,
590 	void *client_data,
591 	FLAC__bool is_ogg
592 )
593 {
594 	FILE *file;
595 
596 	FLAC__ASSERT(0 != decoder);
597 
598 	/*
599 	 * To make sure that our file does not go unclosed after an error, we
600 	 * have to do the same entrance checks here that are later performed
601 	 * in FLAC__stream_decoder_init_FILE() before the FILE* is assigned.
602 	 */
603 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
604 		return decoder->protected_->initstate = FLAC__STREAM_DECODER_INIT_STATUS_ALREADY_INITIALIZED;
605 
606 	if(0 == write_callback || 0 == error_callback)
607 		return decoder->protected_->initstate = FLAC__STREAM_DECODER_INIT_STATUS_INVALID_CALLBACKS;
608 
609 	file = filename? flac_fopen(filename, "rb") : stdin;
610 
611 	if(0 == file)
612 		return FLAC__STREAM_DECODER_INIT_STATUS_ERROR_OPENING_FILE;
613 
614 	return init_FILE_internal_(decoder, file, write_callback, metadata_callback, error_callback, client_data, is_ogg);
615 }
616 
FLAC__stream_decoder_init_file(FLAC__StreamDecoder * decoder,const char * filename,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data)617 FLAC_API FLAC__StreamDecoderInitStatus FLAC__stream_decoder_init_file(
618 	FLAC__StreamDecoder *decoder,
619 	const char *filename,
620 	FLAC__StreamDecoderWriteCallback write_callback,
621 	FLAC__StreamDecoderMetadataCallback metadata_callback,
622 	FLAC__StreamDecoderErrorCallback error_callback,
623 	void *client_data
624 )
625 {
626 	return init_file_internal_(decoder, filename, write_callback, metadata_callback, error_callback, client_data, /*is_ogg=*/false);
627 }
628 
FLAC__stream_decoder_init_ogg_file(FLAC__StreamDecoder * decoder,const char * filename,FLAC__StreamDecoderWriteCallback write_callback,FLAC__StreamDecoderMetadataCallback metadata_callback,FLAC__StreamDecoderErrorCallback error_callback,void * client_data)629 FLAC_API FLAC__StreamDecoderInitStatus FLAC__stream_decoder_init_ogg_file(
630 	FLAC__StreamDecoder *decoder,
631 	const char *filename,
632 	FLAC__StreamDecoderWriteCallback write_callback,
633 	FLAC__StreamDecoderMetadataCallback metadata_callback,
634 	FLAC__StreamDecoderErrorCallback error_callback,
635 	void *client_data
636 )
637 {
638 	return init_file_internal_(decoder, filename, write_callback, metadata_callback, error_callback, client_data, /*is_ogg=*/true);
639 }
640 
FLAC__stream_decoder_finish(FLAC__StreamDecoder * decoder)641 FLAC_API FLAC__bool FLAC__stream_decoder_finish(FLAC__StreamDecoder *decoder)
642 {
643 	FLAC__bool md5_failed = false;
644 	unsigned i;
645 
646 	FLAC__ASSERT(0 != decoder);
647 	FLAC__ASSERT(0 != decoder->private_);
648 	FLAC__ASSERT(0 != decoder->protected_);
649 
650 	if(decoder->protected_->state == FLAC__STREAM_DECODER_UNINITIALIZED)
651 		return true;
652 
653 	/* see the comment in FLAC__stream_decoder_reset() as to why we
654 	 * always call FLAC__MD5Final()
655 	 */
656 	FLAC__MD5Final(decoder->private_->computed_md5sum, &decoder->private_->md5context);
657 
658 	if(decoder->private_->has_seek_table && 0 != decoder->private_->seek_table.data.seek_table.points) {
659 		free(decoder->private_->seek_table.data.seek_table.points);
660 		decoder->private_->seek_table.data.seek_table.points = 0;
661 		decoder->private_->has_seek_table = false;
662 	}
663 	FLAC__bitreader_free(decoder->private_->input);
664 	for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
665 		/* WATCHOUT:
666 		 * FLAC__lpc_restore_signal_asm_ia32_mmx() requires that the
667 		 * output arrays have a buffer of up to 3 zeroes in front
668 		 * (at negative indices) for alignment purposes; we use 4
669 		 * to keep the data well-aligned.
670 		 */
671 		if(0 != decoder->private_->output[i]) {
672 			free(decoder->private_->output[i]-4);
673 			decoder->private_->output[i] = 0;
674 		}
675 		if(0 != decoder->private_->residual_unaligned[i]) {
676 			free(decoder->private_->residual_unaligned[i]);
677 			decoder->private_->residual_unaligned[i] = decoder->private_->residual[i] = 0;
678 		}
679 	}
680 	decoder->private_->output_capacity = 0;
681 	decoder->private_->output_channels = 0;
682 
683 #if FLAC__HAS_OGG
684 	if(decoder->private_->is_ogg)
685 		FLAC__ogg_decoder_aspect_finish(&decoder->protected_->ogg_decoder_aspect);
686 #endif
687 
688 	if(0 != decoder->private_->file) {
689 		if(decoder->private_->file != stdin)
690 			fclose(decoder->private_->file);
691 		decoder->private_->file = 0;
692 	}
693 
694 	if(decoder->private_->do_md5_checking) {
695 		if(memcmp(decoder->private_->stream_info.data.stream_info.md5sum, decoder->private_->computed_md5sum, 16))
696 			md5_failed = true;
697 	}
698 	decoder->private_->is_seeking = false;
699 
700 	set_defaults_(decoder);
701 
702 	decoder->protected_->state = FLAC__STREAM_DECODER_UNINITIALIZED;
703 
704 	return !md5_failed;
705 }
706 
FLAC__stream_decoder_set_ogg_serial_number(FLAC__StreamDecoder * decoder,long value)707 FLAC_API FLAC__bool FLAC__stream_decoder_set_ogg_serial_number(FLAC__StreamDecoder *decoder, long value)
708 {
709 	FLAC__ASSERT(0 != decoder);
710 	FLAC__ASSERT(0 != decoder->private_);
711 	FLAC__ASSERT(0 != decoder->protected_);
712 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
713 		return false;
714 #if FLAC__HAS_OGG
715 	/* can't check decoder->private_->is_ogg since that's not set until init time */
716 	FLAC__ogg_decoder_aspect_set_serial_number(&decoder->protected_->ogg_decoder_aspect, value);
717 	return true;
718 #else
719 	(void)value;
720 	return false;
721 #endif
722 }
723 
FLAC__stream_decoder_set_md5_checking(FLAC__StreamDecoder * decoder,FLAC__bool value)724 FLAC_API FLAC__bool FLAC__stream_decoder_set_md5_checking(FLAC__StreamDecoder *decoder, FLAC__bool value)
725 {
726 	FLAC__ASSERT(0 != decoder);
727 	FLAC__ASSERT(0 != decoder->protected_);
728 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
729 		return false;
730 	decoder->protected_->md5_checking = value;
731 	return true;
732 }
733 
FLAC__stream_decoder_set_metadata_respond(FLAC__StreamDecoder * decoder,FLAC__MetadataType type)734 FLAC_API FLAC__bool FLAC__stream_decoder_set_metadata_respond(FLAC__StreamDecoder *decoder, FLAC__MetadataType type)
735 {
736 	FLAC__ASSERT(0 != decoder);
737 	FLAC__ASSERT(0 != decoder->private_);
738 	FLAC__ASSERT(0 != decoder->protected_);
739 	FLAC__ASSERT((unsigned)type <= FLAC__MAX_METADATA_TYPE_CODE);
740 	/* double protection */
741 	if((unsigned)type > FLAC__MAX_METADATA_TYPE_CODE)
742 		return false;
743 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
744 		return false;
745 	decoder->private_->metadata_filter[type] = true;
746 	if(type == FLAC__METADATA_TYPE_APPLICATION)
747 		decoder->private_->metadata_filter_ids_count = 0;
748 	return true;
749 }
750 
FLAC__stream_decoder_set_metadata_respond_application(FLAC__StreamDecoder * decoder,const FLAC__byte id[4])751 FLAC_API FLAC__bool FLAC__stream_decoder_set_metadata_respond_application(FLAC__StreamDecoder *decoder, const FLAC__byte id[4])
752 {
753 	FLAC__ASSERT(0 != decoder);
754 	FLAC__ASSERT(0 != decoder->private_);
755 	FLAC__ASSERT(0 != decoder->protected_);
756 	FLAC__ASSERT(0 != id);
757 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
758 		return false;
759 
760 	if(decoder->private_->metadata_filter[FLAC__METADATA_TYPE_APPLICATION])
761 		return true;
762 
763 	FLAC__ASSERT(0 != decoder->private_->metadata_filter_ids);
764 
765 	if(decoder->private_->metadata_filter_ids_count == decoder->private_->metadata_filter_ids_capacity) {
766 		if(0 == (decoder->private_->metadata_filter_ids = safe_realloc_mul_2op_(decoder->private_->metadata_filter_ids, decoder->private_->metadata_filter_ids_capacity, /*times*/2))) {
767 			decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
768 			return false;
769 		}
770 		decoder->private_->metadata_filter_ids_capacity *= 2;
771 	}
772 
773 	memcpy(decoder->private_->metadata_filter_ids + decoder->private_->metadata_filter_ids_count * (FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8), id, (FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8));
774 	decoder->private_->metadata_filter_ids_count++;
775 
776 	return true;
777 }
778 
FLAC__stream_decoder_set_metadata_respond_all(FLAC__StreamDecoder * decoder)779 FLAC_API FLAC__bool FLAC__stream_decoder_set_metadata_respond_all(FLAC__StreamDecoder *decoder)
780 {
781 	unsigned i;
782 	FLAC__ASSERT(0 != decoder);
783 	FLAC__ASSERT(0 != decoder->private_);
784 	FLAC__ASSERT(0 != decoder->protected_);
785 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
786 		return false;
787 	for(i = 0; i < sizeof(decoder->private_->metadata_filter) / sizeof(decoder->private_->metadata_filter[0]); i++)
788 		decoder->private_->metadata_filter[i] = true;
789 	decoder->private_->metadata_filter_ids_count = 0;
790 	return true;
791 }
792 
FLAC__stream_decoder_set_metadata_ignore(FLAC__StreamDecoder * decoder,FLAC__MetadataType type)793 FLAC_API FLAC__bool FLAC__stream_decoder_set_metadata_ignore(FLAC__StreamDecoder *decoder, FLAC__MetadataType type)
794 {
795 	FLAC__ASSERT(0 != decoder);
796 	FLAC__ASSERT(0 != decoder->private_);
797 	FLAC__ASSERT(0 != decoder->protected_);
798 	FLAC__ASSERT((unsigned)type <= FLAC__MAX_METADATA_TYPE_CODE);
799 	/* double protection */
800 	if((unsigned)type > FLAC__MAX_METADATA_TYPE_CODE)
801 		return false;
802 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
803 		return false;
804 	decoder->private_->metadata_filter[type] = false;
805 	if(type == FLAC__METADATA_TYPE_APPLICATION)
806 		decoder->private_->metadata_filter_ids_count = 0;
807 	return true;
808 }
809 
FLAC__stream_decoder_set_metadata_ignore_application(FLAC__StreamDecoder * decoder,const FLAC__byte id[4])810 FLAC_API FLAC__bool FLAC__stream_decoder_set_metadata_ignore_application(FLAC__StreamDecoder *decoder, const FLAC__byte id[4])
811 {
812 	FLAC__ASSERT(0 != decoder);
813 	FLAC__ASSERT(0 != decoder->private_);
814 	FLAC__ASSERT(0 != decoder->protected_);
815 	FLAC__ASSERT(0 != id);
816 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
817 		return false;
818 
819 	if(!decoder->private_->metadata_filter[FLAC__METADATA_TYPE_APPLICATION])
820 		return true;
821 
822 	FLAC__ASSERT(0 != decoder->private_->metadata_filter_ids);
823 
824 	if(decoder->private_->metadata_filter_ids_count == decoder->private_->metadata_filter_ids_capacity) {
825 		if(0 == (decoder->private_->metadata_filter_ids = safe_realloc_mul_2op_(decoder->private_->metadata_filter_ids, decoder->private_->metadata_filter_ids_capacity, /*times*/2))) {
826 			decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
827 			return false;
828 		}
829 		decoder->private_->metadata_filter_ids_capacity *= 2;
830 	}
831 
832 	memcpy(decoder->private_->metadata_filter_ids + decoder->private_->metadata_filter_ids_count * (FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8), id, (FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8));
833 	decoder->private_->metadata_filter_ids_count++;
834 
835 	return true;
836 }
837 
FLAC__stream_decoder_set_metadata_ignore_all(FLAC__StreamDecoder * decoder)838 FLAC_API FLAC__bool FLAC__stream_decoder_set_metadata_ignore_all(FLAC__StreamDecoder *decoder)
839 {
840 	FLAC__ASSERT(0 != decoder);
841 	FLAC__ASSERT(0 != decoder->private_);
842 	FLAC__ASSERT(0 != decoder->protected_);
843 	if(decoder->protected_->state != FLAC__STREAM_DECODER_UNINITIALIZED)
844 		return false;
845 	memset(decoder->private_->metadata_filter, 0, sizeof(decoder->private_->metadata_filter));
846 	decoder->private_->metadata_filter_ids_count = 0;
847 	return true;
848 }
849 
FLAC__stream_decoder_get_state(const FLAC__StreamDecoder * decoder)850 FLAC_API FLAC__StreamDecoderState FLAC__stream_decoder_get_state(const FLAC__StreamDecoder *decoder)
851 {
852 	FLAC__ASSERT(0 != decoder);
853 	FLAC__ASSERT(0 != decoder->protected_);
854 	return decoder->protected_->state;
855 }
856 
FLAC__stream_decoder_get_resolved_state_string(const FLAC__StreamDecoder * decoder)857 FLAC_API const char *FLAC__stream_decoder_get_resolved_state_string(const FLAC__StreamDecoder *decoder)
858 {
859 	return FLAC__StreamDecoderStateString[decoder->protected_->state];
860 }
861 
FLAC__stream_decoder_get_md5_checking(const FLAC__StreamDecoder * decoder)862 FLAC_API FLAC__bool FLAC__stream_decoder_get_md5_checking(const FLAC__StreamDecoder *decoder)
863 {
864 	FLAC__ASSERT(0 != decoder);
865 	FLAC__ASSERT(0 != decoder->protected_);
866 	return decoder->protected_->md5_checking;
867 }
868 
FLAC__stream_decoder_get_total_samples(const FLAC__StreamDecoder * decoder)869 FLAC_API FLAC__uint64 FLAC__stream_decoder_get_total_samples(const FLAC__StreamDecoder *decoder)
870 {
871 	FLAC__ASSERT(0 != decoder);
872 	FLAC__ASSERT(0 != decoder->protected_);
873 	return decoder->private_->has_stream_info? decoder->private_->stream_info.data.stream_info.total_samples : 0;
874 }
875 
FLAC__stream_decoder_get_channels(const FLAC__StreamDecoder * decoder)876 FLAC_API unsigned FLAC__stream_decoder_get_channels(const FLAC__StreamDecoder *decoder)
877 {
878 	FLAC__ASSERT(0 != decoder);
879 	FLAC__ASSERT(0 != decoder->protected_);
880 	return decoder->protected_->channels;
881 }
882 
FLAC__stream_decoder_get_channel_assignment(const FLAC__StreamDecoder * decoder)883 FLAC_API FLAC__ChannelAssignment FLAC__stream_decoder_get_channel_assignment(const FLAC__StreamDecoder *decoder)
884 {
885 	FLAC__ASSERT(0 != decoder);
886 	FLAC__ASSERT(0 != decoder->protected_);
887 	return decoder->protected_->channel_assignment;
888 }
889 
FLAC__stream_decoder_get_bits_per_sample(const FLAC__StreamDecoder * decoder)890 FLAC_API unsigned FLAC__stream_decoder_get_bits_per_sample(const FLAC__StreamDecoder *decoder)
891 {
892 	FLAC__ASSERT(0 != decoder);
893 	FLAC__ASSERT(0 != decoder->protected_);
894 	return decoder->protected_->bits_per_sample;
895 }
896 
FLAC__stream_decoder_get_sample_rate(const FLAC__StreamDecoder * decoder)897 FLAC_API unsigned FLAC__stream_decoder_get_sample_rate(const FLAC__StreamDecoder *decoder)
898 {
899 	FLAC__ASSERT(0 != decoder);
900 	FLAC__ASSERT(0 != decoder->protected_);
901 	return decoder->protected_->sample_rate;
902 }
903 
FLAC__stream_decoder_get_blocksize(const FLAC__StreamDecoder * decoder)904 FLAC_API unsigned FLAC__stream_decoder_get_blocksize(const FLAC__StreamDecoder *decoder)
905 {
906 	FLAC__ASSERT(0 != decoder);
907 	FLAC__ASSERT(0 != decoder->protected_);
908 	return decoder->protected_->blocksize;
909 }
910 
FLAC__stream_decoder_get_decode_position(const FLAC__StreamDecoder * decoder,FLAC__uint64 * position)911 FLAC_API FLAC__bool FLAC__stream_decoder_get_decode_position(const FLAC__StreamDecoder *decoder, FLAC__uint64 *position)
912 {
913 	FLAC__ASSERT(0 != decoder);
914 	FLAC__ASSERT(0 != decoder->private_);
915 	FLAC__ASSERT(0 != position);
916 
917 #if FLAC__HAS_OGG
918 	if(decoder->private_->is_ogg)
919 		return false;
920 #endif
921 	if(0 == decoder->private_->tell_callback)
922 		return false;
923 	if(decoder->private_->tell_callback(decoder, position, decoder->private_->client_data) != FLAC__STREAM_DECODER_TELL_STATUS_OK)
924 		return false;
925 	/* should never happen since all FLAC frames and metadata blocks are byte aligned, but check just in case */
926 	if(!FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input))
927 		return false;
928 	FLAC__ASSERT(*position >= FLAC__stream_decoder_get_input_bytes_unconsumed(decoder));
929 	*position -= FLAC__stream_decoder_get_input_bytes_unconsumed(decoder);
930 	return true;
931 }
932 
FLAC__stream_decoder_flush(FLAC__StreamDecoder * decoder)933 FLAC_API FLAC__bool FLAC__stream_decoder_flush(FLAC__StreamDecoder *decoder)
934 {
935 	FLAC__ASSERT(0 != decoder);
936 	FLAC__ASSERT(0 != decoder->private_);
937 	FLAC__ASSERT(0 != decoder->protected_);
938 
939 	decoder->private_->samples_decoded = 0;
940 	decoder->private_->do_md5_checking = false;
941 
942 #if FLAC__HAS_OGG
943 	if(decoder->private_->is_ogg)
944 		FLAC__ogg_decoder_aspect_flush(&decoder->protected_->ogg_decoder_aspect);
945 #endif
946 
947 	if(!FLAC__bitreader_clear(decoder->private_->input)) {
948 		decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
949 		return false;
950 	}
951 	decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
952 
953 	return true;
954 }
955 
FLAC__stream_decoder_reset(FLAC__StreamDecoder * decoder)956 FLAC_API FLAC__bool FLAC__stream_decoder_reset(FLAC__StreamDecoder *decoder)
957 {
958 	FLAC__ASSERT(0 != decoder);
959 	FLAC__ASSERT(0 != decoder->private_);
960 	FLAC__ASSERT(0 != decoder->protected_);
961 
962 	if(!FLAC__stream_decoder_flush(decoder)) {
963 		/* above call sets the state for us */
964 		return false;
965 	}
966 
967 #if FLAC__HAS_OGG
968 	/*@@@ could go in !internal_reset_hack block below */
969 	if(decoder->private_->is_ogg)
970 		FLAC__ogg_decoder_aspect_reset(&decoder->protected_->ogg_decoder_aspect);
971 #endif
972 
973 	/* Rewind if necessary.  If FLAC__stream_decoder_init() is calling us,
974 	 * (internal_reset_hack) don't try to rewind since we are already at
975 	 * the beginning of the stream and don't want to fail if the input is
976 	 * not seekable.
977 	 */
978 	if(!decoder->private_->internal_reset_hack) {
979 		if(decoder->private_->file == stdin)
980 			return false; /* can't rewind stdin, reset fails */
981 		if(decoder->private_->seek_callback && decoder->private_->seek_callback(decoder, 0, decoder->private_->client_data) == FLAC__STREAM_DECODER_SEEK_STATUS_ERROR)
982 			return false; /* seekable and seek fails, reset fails */
983 	}
984 	else
985 		decoder->private_->internal_reset_hack = false;
986 
987 	decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_METADATA;
988 
989 	decoder->private_->has_stream_info = false;
990 	if(decoder->private_->has_seek_table && 0 != decoder->private_->seek_table.data.seek_table.points) {
991 		free(decoder->private_->seek_table.data.seek_table.points);
992 		decoder->private_->seek_table.data.seek_table.points = 0;
993 		decoder->private_->has_seek_table = false;
994 	}
995 	decoder->private_->do_md5_checking = decoder->protected_->md5_checking;
996 	/*
997 	 * This goes in reset() and not flush() because according to the spec, a
998 	 * fixed-blocksize stream must stay that way through the whole stream.
999 	 */
1000 	decoder->private_->fixed_block_size = decoder->private_->next_fixed_block_size = 0;
1001 
1002 	/* We initialize the FLAC__MD5Context even though we may never use it.  This
1003 	 * is because md5 checking may be turned on to start and then turned off if
1004 	 * a seek occurs.  So we init the context here and finalize it in
1005 	 * FLAC__stream_decoder_finish() to make sure things are always cleaned up
1006 	 * properly.
1007 	 */
1008 	FLAC__MD5Init(&decoder->private_->md5context);
1009 
1010 	decoder->private_->first_frame_offset = 0;
1011 	decoder->private_->unparseable_frame_count = 0;
1012 
1013 	return true;
1014 }
1015 
FLAC__stream_decoder_process_single(FLAC__StreamDecoder * decoder)1016 FLAC_API FLAC__bool FLAC__stream_decoder_process_single(FLAC__StreamDecoder *decoder)
1017 {
1018 	FLAC__bool got_a_frame;
1019 	FLAC__ASSERT(0 != decoder);
1020 	FLAC__ASSERT(0 != decoder->protected_);
1021 
1022 	while(1) {
1023 		switch(decoder->protected_->state) {
1024 			case FLAC__STREAM_DECODER_SEARCH_FOR_METADATA:
1025 				if(!find_metadata_(decoder))
1026 					return false; /* above function sets the status for us */
1027 				break;
1028 			case FLAC__STREAM_DECODER_READ_METADATA:
1029 				if(!read_metadata_(decoder))
1030 					return false; /* above function sets the status for us */
1031 				else
1032 					return true;
1033 			case FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC:
1034 				if(!frame_sync_(decoder))
1035 					return true; /* above function sets the status for us */
1036 				break;
1037 			case FLAC__STREAM_DECODER_READ_FRAME:
1038 				if(!read_frame_(decoder, &got_a_frame, /*do_full_decode=*/true))
1039 					return false; /* above function sets the status for us */
1040 				if(got_a_frame)
1041 					return true; /* above function sets the status for us */
1042 				break;
1043 			case FLAC__STREAM_DECODER_END_OF_STREAM:
1044 			case FLAC__STREAM_DECODER_ABORTED:
1045 				return true;
1046 			default:
1047 				FLAC__ASSERT(0);
1048 				return false;
1049 		}
1050 	}
1051 }
1052 
FLAC__stream_decoder_process_until_end_of_metadata(FLAC__StreamDecoder * decoder)1053 FLAC_API FLAC__bool FLAC__stream_decoder_process_until_end_of_metadata(FLAC__StreamDecoder *decoder)
1054 {
1055 	FLAC__ASSERT(0 != decoder);
1056 	FLAC__ASSERT(0 != decoder->protected_);
1057 
1058 	while(1) {
1059 		switch(decoder->protected_->state) {
1060 			case FLAC__STREAM_DECODER_SEARCH_FOR_METADATA:
1061 				if(!find_metadata_(decoder))
1062 					return false; /* above function sets the status for us */
1063 				break;
1064 			case FLAC__STREAM_DECODER_READ_METADATA:
1065 				if(!read_metadata_(decoder))
1066 					return false; /* above function sets the status for us */
1067 				break;
1068 			case FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC:
1069 			case FLAC__STREAM_DECODER_READ_FRAME:
1070 			case FLAC__STREAM_DECODER_END_OF_STREAM:
1071 			case FLAC__STREAM_DECODER_ABORTED:
1072 				return true;
1073 			default:
1074 				FLAC__ASSERT(0);
1075 				return false;
1076 		}
1077 	}
1078 }
1079 
FLAC__stream_decoder_process_until_end_of_stream(FLAC__StreamDecoder * decoder)1080 FLAC_API FLAC__bool FLAC__stream_decoder_process_until_end_of_stream(FLAC__StreamDecoder *decoder)
1081 {
1082 	FLAC__bool dummy;
1083 	FLAC__ASSERT(0 != decoder);
1084 	FLAC__ASSERT(0 != decoder->protected_);
1085 
1086 	while(1) {
1087 		switch(decoder->protected_->state) {
1088 			case FLAC__STREAM_DECODER_SEARCH_FOR_METADATA:
1089 				if(!find_metadata_(decoder))
1090 					return false; /* above function sets the status for us */
1091 				break;
1092 			case FLAC__STREAM_DECODER_READ_METADATA:
1093 				if(!read_metadata_(decoder))
1094 					return false; /* above function sets the status for us */
1095 				break;
1096 			case FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC:
1097 				if(!frame_sync_(decoder))
1098 					return true; /* above function sets the status for us */
1099 				break;
1100 			case FLAC__STREAM_DECODER_READ_FRAME:
1101 				if(!read_frame_(decoder, &dummy, /*do_full_decode=*/true))
1102 					return false; /* above function sets the status for us */
1103 				break;
1104 			case FLAC__STREAM_DECODER_END_OF_STREAM:
1105 			case FLAC__STREAM_DECODER_ABORTED:
1106 				return true;
1107 			default:
1108 				FLAC__ASSERT(0);
1109 				return false;
1110 		}
1111 	}
1112 }
1113 
FLAC__stream_decoder_skip_single_frame(FLAC__StreamDecoder * decoder)1114 FLAC_API FLAC__bool FLAC__stream_decoder_skip_single_frame(FLAC__StreamDecoder *decoder)
1115 {
1116 	FLAC__bool got_a_frame;
1117 	FLAC__ASSERT(0 != decoder);
1118 	FLAC__ASSERT(0 != decoder->protected_);
1119 
1120 	while(1) {
1121 		switch(decoder->protected_->state) {
1122 			case FLAC__STREAM_DECODER_SEARCH_FOR_METADATA:
1123 			case FLAC__STREAM_DECODER_READ_METADATA:
1124 				return false; /* above function sets the status for us */
1125 			case FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC:
1126 				if(!frame_sync_(decoder))
1127 					return true; /* above function sets the status for us */
1128 				break;
1129 			case FLAC__STREAM_DECODER_READ_FRAME:
1130 				if(!read_frame_(decoder, &got_a_frame, /*do_full_decode=*/false))
1131 					return false; /* above function sets the status for us */
1132 				if(got_a_frame)
1133 					return true; /* above function sets the status for us */
1134 				break;
1135 			case FLAC__STREAM_DECODER_END_OF_STREAM:
1136 			case FLAC__STREAM_DECODER_ABORTED:
1137 				return true;
1138 			default:
1139 				FLAC__ASSERT(0);
1140 				return false;
1141 		}
1142 	}
1143 }
1144 
FLAC__stream_decoder_seek_absolute(FLAC__StreamDecoder * decoder,FLAC__uint64 sample)1145 FLAC_API FLAC__bool FLAC__stream_decoder_seek_absolute(FLAC__StreamDecoder *decoder, FLAC__uint64 sample)
1146 {
1147 	FLAC__uint64 length;
1148 
1149 	FLAC__ASSERT(0 != decoder);
1150 
1151 	if(
1152 		decoder->protected_->state != FLAC__STREAM_DECODER_SEARCH_FOR_METADATA &&
1153 		decoder->protected_->state != FLAC__STREAM_DECODER_READ_METADATA &&
1154 		decoder->protected_->state != FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC &&
1155 		decoder->protected_->state != FLAC__STREAM_DECODER_READ_FRAME &&
1156 		decoder->protected_->state != FLAC__STREAM_DECODER_END_OF_STREAM
1157 	)
1158 		return false;
1159 
1160 	if(0 == decoder->private_->seek_callback)
1161 		return false;
1162 
1163 	FLAC__ASSERT(decoder->private_->seek_callback);
1164 	FLAC__ASSERT(decoder->private_->tell_callback);
1165 	FLAC__ASSERT(decoder->private_->length_callback);
1166 	FLAC__ASSERT(decoder->private_->eof_callback);
1167 
1168 	if(FLAC__stream_decoder_get_total_samples(decoder) > 0 && sample >= FLAC__stream_decoder_get_total_samples(decoder))
1169 		return false;
1170 
1171 	decoder->private_->is_seeking = true;
1172 
1173 	/* turn off md5 checking if a seek is attempted */
1174 	decoder->private_->do_md5_checking = false;
1175 
1176 	/* get the file length (currently our algorithm needs to know the length so it's also an error to get FLAC__STREAM_DECODER_LENGTH_STATUS_UNSUPPORTED) */
1177 	if(decoder->private_->length_callback(decoder, &length, decoder->private_->client_data) != FLAC__STREAM_DECODER_LENGTH_STATUS_OK) {
1178 		decoder->private_->is_seeking = false;
1179 		return false;
1180 	}
1181 
1182 	/* if we haven't finished processing the metadata yet, do that so we have the STREAMINFO, SEEK_TABLE, and first_frame_offset */
1183 	if(
1184 		decoder->protected_->state == FLAC__STREAM_DECODER_SEARCH_FOR_METADATA ||
1185 		decoder->protected_->state == FLAC__STREAM_DECODER_READ_METADATA
1186 	) {
1187 		if(!FLAC__stream_decoder_process_until_end_of_metadata(decoder)) {
1188 			/* above call sets the state for us */
1189 			decoder->private_->is_seeking = false;
1190 			return false;
1191 		}
1192 		/* check this again in case we didn't know total_samples the first time */
1193 		if(FLAC__stream_decoder_get_total_samples(decoder) > 0 && sample >= FLAC__stream_decoder_get_total_samples(decoder)) {
1194 			decoder->private_->is_seeking = false;
1195 			return false;
1196 		}
1197 	}
1198 
1199 	{
1200 		const FLAC__bool ok =
1201 #if FLAC__HAS_OGG
1202 			decoder->private_->is_ogg?
1203 			seek_to_absolute_sample_ogg_(decoder, length, sample) :
1204 #endif
1205 			seek_to_absolute_sample_(decoder, length, sample)
1206 		;
1207 		decoder->private_->is_seeking = false;
1208 		return ok;
1209 	}
1210 }
1211 
1212 /***********************************************************************
1213  *
1214  * Protected class methods
1215  *
1216  ***********************************************************************/
1217 
FLAC__stream_decoder_get_input_bytes_unconsumed(const FLAC__StreamDecoder * decoder)1218 unsigned FLAC__stream_decoder_get_input_bytes_unconsumed(const FLAC__StreamDecoder *decoder)
1219 {
1220 	FLAC__ASSERT(0 != decoder);
1221 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
1222 	FLAC__ASSERT(!(FLAC__bitreader_get_input_bits_unconsumed(decoder->private_->input) & 7));
1223 	return FLAC__bitreader_get_input_bits_unconsumed(decoder->private_->input) / 8;
1224 }
1225 
1226 /***********************************************************************
1227  *
1228  * Private class methods
1229  *
1230  ***********************************************************************/
1231 
set_defaults_(FLAC__StreamDecoder * decoder)1232 void set_defaults_(FLAC__StreamDecoder *decoder)
1233 {
1234 #if FLAC__HAS_OGG
1235 	decoder->private_->is_ogg = false;
1236 #endif
1237 	decoder->private_->read_callback = 0;
1238 	decoder->private_->seek_callback = 0;
1239 	decoder->private_->tell_callback = 0;
1240 	decoder->private_->length_callback = 0;
1241 	decoder->private_->eof_callback = 0;
1242 	decoder->private_->write_callback = 0;
1243 	decoder->private_->metadata_callback = 0;
1244 	decoder->private_->error_callback = 0;
1245 	decoder->private_->client_data = 0;
1246 
1247 	memset(decoder->private_->metadata_filter, 0, sizeof(decoder->private_->metadata_filter));
1248 	decoder->private_->metadata_filter[FLAC__METADATA_TYPE_STREAMINFO] = true;
1249 	decoder->private_->metadata_filter_ids_count = 0;
1250 
1251 	decoder->protected_->md5_checking = false;
1252 
1253 #if FLAC__HAS_OGG
1254 	FLAC__ogg_decoder_aspect_set_defaults(&decoder->protected_->ogg_decoder_aspect);
1255 #endif
1256 }
1257 
1258 /*
1259  * This will forcibly set stdin to binary mode (for OSes that require it)
1260  */
get_binary_stdin_(void)1261 FILE *get_binary_stdin_(void)
1262 {
1263 	/* if something breaks here it is probably due to the presence or
1264 	 * absence of an underscore before the identifiers 'setmode',
1265 	 * 'fileno', and/or 'O_BINARY'; check your system header files.
1266 	 */
1267 #if defined _MSC_VER || defined __MINGW32__
1268 	_setmode(_fileno(stdin), _O_BINARY);
1269 #elif defined __CYGWIN__
1270 	/* almost certainly not needed for any modern Cygwin, but let's be safe... */
1271 	setmode(_fileno(stdin), _O_BINARY);
1272 #elif defined __EMX__
1273 	setmode(fileno(stdin), O_BINARY);
1274 #endif
1275 
1276 	return stdin;
1277 }
1278 
allocate_output_(FLAC__StreamDecoder * decoder,unsigned size,unsigned channels)1279 FLAC__bool allocate_output_(FLAC__StreamDecoder *decoder, unsigned size, unsigned channels)
1280 {
1281 	unsigned i;
1282 	FLAC__int32 *tmp;
1283 
1284 	/* Make sure size is some sensible minimum value. Plumb through predictor_order maybe? */
1285 	size = size < FLAC__MAX_LPC_ORDER ? FLAC__MAX_LPC_ORDER : size ;
1286 
1287 	if(size <= decoder->private_->output_capacity && channels <= decoder->private_->output_channels)
1288 		return true;
1289 
1290 	/* simply using realloc() is not practical because the number of channels may change mid-stream */
1291 
1292 	for(i = 0; i < FLAC__MAX_CHANNELS; i++) {
1293 		if(0 != decoder->private_->output[i]) {
1294 			free(decoder->private_->output[i]-4);
1295 			decoder->private_->output[i] = 0;
1296 		}
1297 		if(0 != decoder->private_->residual_unaligned[i]) {
1298 			free(decoder->private_->residual_unaligned[i]);
1299 			decoder->private_->residual_unaligned[i] = decoder->private_->residual[i] = 0;
1300 		}
1301 	}
1302 
1303 	for(i = 0; i < channels; i++) {
1304 		/* WATCHOUT:
1305 		 * FLAC__lpc_restore_signal_asm_ia32_mmx() requires that the
1306 		 * output arrays have a buffer of up to 3 zeroes in front
1307 		 * (at negative indices) for alignment purposes; we use 4
1308 		 * to keep the data well-aligned.
1309 		 */
1310 		tmp = safe_malloc_muladd2_(sizeof(FLAC__int32), /*times (*/size, /*+*/4/*)*/);
1311 		if(tmp == 0) {
1312 			decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1313 			return false;
1314 		}
1315 		memset(tmp, 0, sizeof(FLAC__int32)*4);
1316 		decoder->private_->output[i] = tmp + 4;
1317 
1318 		if(!FLAC__memory_alloc_aligned_int32_array(size, &decoder->private_->residual_unaligned[i], &decoder->private_->residual[i])) {
1319 			decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1320 			return false;
1321 		}
1322 	}
1323 
1324 	decoder->private_->output_capacity = size;
1325 	decoder->private_->output_channels = channels;
1326 
1327 	return true;
1328 }
1329 
has_id_filtered_(FLAC__StreamDecoder * decoder,FLAC__byte * id)1330 FLAC__bool has_id_filtered_(FLAC__StreamDecoder *decoder, FLAC__byte *id)
1331 {
1332 	size_t i;
1333 
1334 	FLAC__ASSERT(0 != decoder);
1335 	FLAC__ASSERT(0 != decoder->private_);
1336 
1337 	for(i = 0; i < decoder->private_->metadata_filter_ids_count; i++)
1338 		if(0 == memcmp(decoder->private_->metadata_filter_ids + i * (FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8), id, (FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8)))
1339 			return true;
1340 
1341 	return false;
1342 }
1343 
find_metadata_(FLAC__StreamDecoder * decoder)1344 FLAC__bool find_metadata_(FLAC__StreamDecoder *decoder)
1345 {
1346 	FLAC__uint32 x;
1347 	unsigned i, id;
1348 	FLAC__bool first = true;
1349 
1350 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
1351 
1352 	for(i = id = 0; i < 4; ) {
1353 		if(decoder->private_->cached) {
1354 			x = (FLAC__uint32)decoder->private_->lookahead;
1355 			decoder->private_->cached = false;
1356 		}
1357 		else {
1358 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
1359 				return false; /* read_callback_ sets the state for us */
1360 		}
1361 		if(x == FLAC__STREAM_SYNC_STRING[i]) {
1362 			first = true;
1363 			i++;
1364 			id = 0;
1365 			continue;
1366 		}
1367 
1368 		if(id >= 3)
1369 			return false;
1370 
1371 		if(x == ID3V2_TAG_[id]) {
1372 			id++;
1373 			i = 0;
1374 			if(id == 3) {
1375 				if(!skip_id3v2_tag_(decoder))
1376 					return false; /* skip_id3v2_tag_ sets the state for us */
1377 			}
1378 			continue;
1379 		}
1380 		id = 0;
1381 		if(x == 0xff) { /* MAGIC NUMBER for the first 8 frame sync bits */
1382 			decoder->private_->header_warmup[0] = (FLAC__byte)x;
1383 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
1384 				return false; /* read_callback_ sets the state for us */
1385 
1386 			/* we have to check if we just read two 0xff's in a row; the second may actually be the beginning of the sync code */
1387 			/* else we have to check if the second byte is the end of a sync code */
1388 			if(x == 0xff) { /* MAGIC NUMBER for the first 8 frame sync bits */
1389 				decoder->private_->lookahead = (FLAC__byte)x;
1390 				decoder->private_->cached = true;
1391 			}
1392 			else if(x >> 1 == 0x7c) { /* MAGIC NUMBER for the last 6 sync bits and reserved 7th bit */
1393 				decoder->private_->header_warmup[1] = (FLAC__byte)x;
1394 				decoder->protected_->state = FLAC__STREAM_DECODER_READ_FRAME;
1395 				return true;
1396 			}
1397 		}
1398 		i = 0;
1399 		if(first) {
1400 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC);
1401 			first = false;
1402 		}
1403 	}
1404 
1405 	decoder->protected_->state = FLAC__STREAM_DECODER_READ_METADATA;
1406 	return true;
1407 }
1408 
read_metadata_(FLAC__StreamDecoder * decoder)1409 FLAC__bool read_metadata_(FLAC__StreamDecoder *decoder)
1410 {
1411 	FLAC__bool is_last;
1412 	FLAC__uint32 i, x, type, length;
1413 
1414 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
1415 
1416 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_IS_LAST_LEN))
1417 		return false; /* read_callback_ sets the state for us */
1418 	is_last = x? true : false;
1419 
1420 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &type, FLAC__STREAM_METADATA_TYPE_LEN))
1421 		return false; /* read_callback_ sets the state for us */
1422 
1423 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &length, FLAC__STREAM_METADATA_LENGTH_LEN))
1424 		return false; /* read_callback_ sets the state for us */
1425 
1426 	if(type == FLAC__METADATA_TYPE_STREAMINFO) {
1427 		if(!read_metadata_streaminfo_(decoder, is_last, length))
1428 			return false;
1429 
1430 		decoder->private_->has_stream_info = true;
1431 		if(0 == memcmp(decoder->private_->stream_info.data.stream_info.md5sum, "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0", 16))
1432 			decoder->private_->do_md5_checking = false;
1433 		if(!decoder->private_->is_seeking && decoder->private_->metadata_filter[FLAC__METADATA_TYPE_STREAMINFO] && decoder->private_->metadata_callback)
1434 			decoder->private_->metadata_callback(decoder, &decoder->private_->stream_info, decoder->private_->client_data);
1435 	}
1436 	else if(type == FLAC__METADATA_TYPE_SEEKTABLE) {
1437 		if(!read_metadata_seektable_(decoder, is_last, length))
1438 			return false;
1439 
1440 		decoder->private_->has_seek_table = true;
1441 		if(!decoder->private_->is_seeking && decoder->private_->metadata_filter[FLAC__METADATA_TYPE_SEEKTABLE] && decoder->private_->metadata_callback)
1442 			decoder->private_->metadata_callback(decoder, &decoder->private_->seek_table, decoder->private_->client_data);
1443 	}
1444 	else {
1445 		FLAC__bool skip_it = !decoder->private_->metadata_filter[type];
1446 		unsigned real_length = length;
1447 		FLAC__StreamMetadata block;
1448 
1449 		memset(&block, 0, sizeof(block));
1450 		block.is_last = is_last;
1451 		block.type = (FLAC__MetadataType)type;
1452 		block.length = length;
1453 
1454 		if(type == FLAC__METADATA_TYPE_APPLICATION) {
1455 			if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, block.data.application.id, FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8))
1456 				return false; /* read_callback_ sets the state for us */
1457 
1458 			if(real_length < FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8) { /* underflow check */
1459 				decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;/*@@@@@@ maybe wrong error? need to resync?*/
1460 				return false;
1461 			}
1462 
1463 			real_length -= FLAC__STREAM_METADATA_APPLICATION_ID_LEN/8;
1464 
1465 			if(decoder->private_->metadata_filter_ids_count > 0 && has_id_filtered_(decoder, block.data.application.id))
1466 				skip_it = !skip_it;
1467 		}
1468 
1469 		if(skip_it) {
1470 			if(!FLAC__bitreader_skip_byte_block_aligned_no_crc(decoder->private_->input, real_length))
1471 				return false; /* read_callback_ sets the state for us */
1472 		}
1473 		else {
1474 			FLAC__bool ok = true;
1475 			switch(type) {
1476 				case FLAC__METADATA_TYPE_PADDING:
1477 					/* skip the padding bytes */
1478 					if(!FLAC__bitreader_skip_byte_block_aligned_no_crc(decoder->private_->input, real_length))
1479 						ok = false; /* read_callback_ sets the state for us */
1480 					break;
1481 				case FLAC__METADATA_TYPE_APPLICATION:
1482 					/* remember, we read the ID already */
1483 					if(real_length > 0) {
1484 						if(0 == (block.data.application.data = malloc(real_length))) {
1485 							decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1486 							ok = false;
1487 						}
1488 						else if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, block.data.application.data, real_length))
1489 							ok = false; /* read_callback_ sets the state for us */
1490 					}
1491 					else
1492 						block.data.application.data = 0;
1493 					break;
1494 				case FLAC__METADATA_TYPE_VORBIS_COMMENT:
1495 					if(!read_metadata_vorbiscomment_(decoder, &block.data.vorbis_comment, real_length))
1496 						ok = false;
1497 					break;
1498 				case FLAC__METADATA_TYPE_CUESHEET:
1499 					if(!read_metadata_cuesheet_(decoder, &block.data.cue_sheet))
1500 						ok = false;
1501 					break;
1502 				case FLAC__METADATA_TYPE_PICTURE:
1503 					if(!read_metadata_picture_(decoder, &block.data.picture))
1504 						ok = false;
1505 					break;
1506 				case FLAC__METADATA_TYPE_STREAMINFO:
1507 				case FLAC__METADATA_TYPE_SEEKTABLE:
1508 					FLAC__ASSERT(0);
1509 					break;
1510 				default:
1511 					if(real_length > 0) {
1512 						if(0 == (block.data.unknown.data = malloc(real_length))) {
1513 							decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1514 							ok = false;
1515 						}
1516 						else if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, block.data.unknown.data, real_length))
1517 							ok = false; /* read_callback_ sets the state for us */
1518 					}
1519 					else
1520 						block.data.unknown.data = 0;
1521 					break;
1522 			}
1523 			if(ok && !decoder->private_->is_seeking && decoder->private_->metadata_callback)
1524 				decoder->private_->metadata_callback(decoder, &block, decoder->private_->client_data);
1525 
1526 			/* now we have to free any malloc()ed data in the block */
1527 			switch(type) {
1528 				case FLAC__METADATA_TYPE_PADDING:
1529 					break;
1530 				case FLAC__METADATA_TYPE_APPLICATION:
1531 					if(0 != block.data.application.data)
1532 						free(block.data.application.data);
1533 					break;
1534 				case FLAC__METADATA_TYPE_VORBIS_COMMENT:
1535 					if(0 != block.data.vorbis_comment.vendor_string.entry)
1536 						free(block.data.vorbis_comment.vendor_string.entry);
1537 					if(block.data.vorbis_comment.num_comments > 0)
1538 						for(i = 0; i < block.data.vorbis_comment.num_comments; i++)
1539 							if(0 != block.data.vorbis_comment.comments[i].entry)
1540 								free(block.data.vorbis_comment.comments[i].entry);
1541 					if(0 != block.data.vorbis_comment.comments)
1542 						free(block.data.vorbis_comment.comments);
1543 					break;
1544 				case FLAC__METADATA_TYPE_CUESHEET:
1545 					if(block.data.cue_sheet.num_tracks > 0)
1546 						for(i = 0; i < block.data.cue_sheet.num_tracks; i++)
1547 							if(0 != block.data.cue_sheet.tracks[i].indices)
1548 								free(block.data.cue_sheet.tracks[i].indices);
1549 					if(0 != block.data.cue_sheet.tracks)
1550 						free(block.data.cue_sheet.tracks);
1551 					break;
1552 				case FLAC__METADATA_TYPE_PICTURE:
1553 					if(0 != block.data.picture.mime_type)
1554 						free(block.data.picture.mime_type);
1555 					if(0 != block.data.picture.description)
1556 						free(block.data.picture.description);
1557 					if(0 != block.data.picture.data)
1558 						free(block.data.picture.data);
1559 					break;
1560 				case FLAC__METADATA_TYPE_STREAMINFO:
1561 				case FLAC__METADATA_TYPE_SEEKTABLE:
1562 					FLAC__ASSERT(0);
1563 				default:
1564 					if(0 != block.data.unknown.data)
1565 						free(block.data.unknown.data);
1566 					break;
1567 			}
1568 
1569 			if(!ok) /* anything that unsets "ok" should also make sure decoder->protected_->state is updated */
1570 				return false;
1571 		}
1572 	}
1573 
1574 	if(is_last) {
1575 		/* if this fails, it's OK, it's just a hint for the seek routine */
1576 		if(!FLAC__stream_decoder_get_decode_position(decoder, &decoder->private_->first_frame_offset))
1577 			decoder->private_->first_frame_offset = 0;
1578 		decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
1579 	}
1580 
1581 	return true;
1582 }
1583 
read_metadata_streaminfo_(FLAC__StreamDecoder * decoder,FLAC__bool is_last,unsigned length)1584 FLAC__bool read_metadata_streaminfo_(FLAC__StreamDecoder *decoder, FLAC__bool is_last, unsigned length)
1585 {
1586 	FLAC__uint32 x;
1587 	unsigned bits, used_bits = 0;
1588 
1589 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
1590 
1591 	decoder->private_->stream_info.type = FLAC__METADATA_TYPE_STREAMINFO;
1592 	decoder->private_->stream_info.is_last = is_last;
1593 	decoder->private_->stream_info.length = length;
1594 
1595 	bits = FLAC__STREAM_METADATA_STREAMINFO_MIN_BLOCK_SIZE_LEN;
1596 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, bits))
1597 		return false; /* read_callback_ sets the state for us */
1598 	decoder->private_->stream_info.data.stream_info.min_blocksize = x;
1599 	used_bits += bits;
1600 
1601 	bits = FLAC__STREAM_METADATA_STREAMINFO_MAX_BLOCK_SIZE_LEN;
1602 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_STREAMINFO_MAX_BLOCK_SIZE_LEN))
1603 		return false; /* read_callback_ sets the state for us */
1604 	decoder->private_->stream_info.data.stream_info.max_blocksize = x;
1605 	used_bits += bits;
1606 
1607 	bits = FLAC__STREAM_METADATA_STREAMINFO_MIN_FRAME_SIZE_LEN;
1608 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_STREAMINFO_MIN_FRAME_SIZE_LEN))
1609 		return false; /* read_callback_ sets the state for us */
1610 	decoder->private_->stream_info.data.stream_info.min_framesize = x;
1611 	used_bits += bits;
1612 
1613 	bits = FLAC__STREAM_METADATA_STREAMINFO_MAX_FRAME_SIZE_LEN;
1614 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_STREAMINFO_MAX_FRAME_SIZE_LEN))
1615 		return false; /* read_callback_ sets the state for us */
1616 	decoder->private_->stream_info.data.stream_info.max_framesize = x;
1617 	used_bits += bits;
1618 
1619 	bits = FLAC__STREAM_METADATA_STREAMINFO_SAMPLE_RATE_LEN;
1620 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_STREAMINFO_SAMPLE_RATE_LEN))
1621 		return false; /* read_callback_ sets the state for us */
1622 	decoder->private_->stream_info.data.stream_info.sample_rate = x;
1623 	used_bits += bits;
1624 
1625 	bits = FLAC__STREAM_METADATA_STREAMINFO_CHANNELS_LEN;
1626 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_STREAMINFO_CHANNELS_LEN))
1627 		return false; /* read_callback_ sets the state for us */
1628 	decoder->private_->stream_info.data.stream_info.channels = x+1;
1629 	used_bits += bits;
1630 
1631 	bits = FLAC__STREAM_METADATA_STREAMINFO_BITS_PER_SAMPLE_LEN;
1632 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_STREAMINFO_BITS_PER_SAMPLE_LEN))
1633 		return false; /* read_callback_ sets the state for us */
1634 	decoder->private_->stream_info.data.stream_info.bits_per_sample = x+1;
1635 	used_bits += bits;
1636 
1637 	bits = FLAC__STREAM_METADATA_STREAMINFO_TOTAL_SAMPLES_LEN;
1638 	if(!FLAC__bitreader_read_raw_uint64(decoder->private_->input, &decoder->private_->stream_info.data.stream_info.total_samples, FLAC__STREAM_METADATA_STREAMINFO_TOTAL_SAMPLES_LEN))
1639 		return false; /* read_callback_ sets the state for us */
1640 	used_bits += bits;
1641 
1642 	if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, decoder->private_->stream_info.data.stream_info.md5sum, 16))
1643 		return false; /* read_callback_ sets the state for us */
1644 	used_bits += 16*8;
1645 
1646 	/* skip the rest of the block */
1647 	FLAC__ASSERT(used_bits % 8 == 0);
1648 	length -= (used_bits / 8);
1649 	if(!FLAC__bitreader_skip_byte_block_aligned_no_crc(decoder->private_->input, length))
1650 		return false; /* read_callback_ sets the state for us */
1651 
1652 	return true;
1653 }
1654 
read_metadata_seektable_(FLAC__StreamDecoder * decoder,FLAC__bool is_last,unsigned length)1655 FLAC__bool read_metadata_seektable_(FLAC__StreamDecoder *decoder, FLAC__bool is_last, unsigned length)
1656 {
1657 	FLAC__uint32 i, x;
1658 	FLAC__uint64 xx;
1659 
1660 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
1661 
1662 	decoder->private_->seek_table.type = FLAC__METADATA_TYPE_SEEKTABLE;
1663 	decoder->private_->seek_table.is_last = is_last;
1664 	decoder->private_->seek_table.length = length;
1665 
1666 	decoder->private_->seek_table.data.seek_table.num_points = length / FLAC__STREAM_METADATA_SEEKPOINT_LENGTH;
1667 
1668 	/* use realloc since we may pass through here several times (e.g. after seeking) */
1669 	if(0 == (decoder->private_->seek_table.data.seek_table.points = safe_realloc_mul_2op_(decoder->private_->seek_table.data.seek_table.points, decoder->private_->seek_table.data.seek_table.num_points, /*times*/sizeof(FLAC__StreamMetadata_SeekPoint)))) {
1670 		decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1671 		return false;
1672 	}
1673 	for(i = 0; i < decoder->private_->seek_table.data.seek_table.num_points; i++) {
1674 		if(!FLAC__bitreader_read_raw_uint64(decoder->private_->input, &xx, FLAC__STREAM_METADATA_SEEKPOINT_SAMPLE_NUMBER_LEN))
1675 			return false; /* read_callback_ sets the state for us */
1676 		decoder->private_->seek_table.data.seek_table.points[i].sample_number = xx;
1677 
1678 		if(!FLAC__bitreader_read_raw_uint64(decoder->private_->input, &xx, FLAC__STREAM_METADATA_SEEKPOINT_STREAM_OFFSET_LEN))
1679 			return false; /* read_callback_ sets the state for us */
1680 		decoder->private_->seek_table.data.seek_table.points[i].stream_offset = xx;
1681 
1682 		if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_SEEKPOINT_FRAME_SAMPLES_LEN))
1683 			return false; /* read_callback_ sets the state for us */
1684 		decoder->private_->seek_table.data.seek_table.points[i].frame_samples = x;
1685 	}
1686 	length -= (decoder->private_->seek_table.data.seek_table.num_points * FLAC__STREAM_METADATA_SEEKPOINT_LENGTH);
1687 	/* if there is a partial point left, skip over it */
1688 	if(length > 0) {
1689 		/*@@@ do a send_error_to_client_() here?  there's an argument for either way */
1690 		if(!FLAC__bitreader_skip_byte_block_aligned_no_crc(decoder->private_->input, length))
1691 			return false; /* read_callback_ sets the state for us */
1692 	}
1693 
1694 	return true;
1695 }
1696 
read_metadata_vorbiscomment_(FLAC__StreamDecoder * decoder,FLAC__StreamMetadata_VorbisComment * obj,unsigned length)1697 FLAC__bool read_metadata_vorbiscomment_(FLAC__StreamDecoder *decoder, FLAC__StreamMetadata_VorbisComment *obj, unsigned length)
1698 {
1699 	FLAC__uint32 i;
1700 
1701 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
1702 
1703 	/* read vendor string */
1704 	if (length >= 8) {
1705 		length -= 8; /* vendor string length + num comments entries alone take 8 bytes */
1706 		FLAC__ASSERT(FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN == 32);
1707 		if (!FLAC__bitreader_read_uint32_little_endian(decoder->private_->input, &obj->vendor_string.length))
1708 			return false; /* read_callback_ sets the state for us */
1709 		if (obj->vendor_string.length > 0) {
1710 			if (length < obj->vendor_string.length) {
1711 				obj->vendor_string.length = 0;
1712 				obj->vendor_string.entry = 0;
1713 				goto skip;
1714 			}
1715 			else
1716 				length -= obj->vendor_string.length;
1717 			if (0 == (obj->vendor_string.entry = safe_malloc_add_2op_(obj->vendor_string.length, /*+*/1))) {
1718 				decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1719 				return false;
1720 			}
1721 			if (!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, obj->vendor_string.entry, obj->vendor_string.length))
1722 				return false; /* read_callback_ sets the state for us */
1723 			obj->vendor_string.entry[obj->vendor_string.length] = '\0';
1724 		}
1725 		else
1726 			obj->vendor_string.entry = 0;
1727 
1728 		/* read num comments */
1729 		FLAC__ASSERT(FLAC__STREAM_METADATA_VORBIS_COMMENT_NUM_COMMENTS_LEN == 32);
1730 		if (!FLAC__bitreader_read_uint32_little_endian(decoder->private_->input, &obj->num_comments))
1731 			return false; /* read_callback_ sets the state for us */
1732 
1733 		/* read comments */
1734 		if (obj->num_comments > 100000) {
1735 			/* Possibly malicious file. */
1736 			obj->num_comments = 0;
1737 			return false;
1738 		}
1739 		if (obj->num_comments > 0) {
1740 			if (0 == (obj->comments = safe_malloc_mul_2op_p(obj->num_comments, /*times*/sizeof(FLAC__StreamMetadata_VorbisComment_Entry)))) {
1741 				decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1742 				return false;
1743 			}
1744 			for (i = 0; i < obj->num_comments; i++) {
1745 				/* Initialize here just to make sure. */
1746 				obj->comments[i].length = 0;
1747 				obj->comments[i].entry = 0;
1748 
1749 				FLAC__ASSERT(FLAC__STREAM_METADATA_VORBIS_COMMENT_ENTRY_LENGTH_LEN == 32);
1750 				if (length < 4) {
1751 					obj->num_comments = i;
1752 					goto skip;
1753 				}
1754 				else
1755 					length -= 4;
1756 				if (!FLAC__bitreader_read_uint32_little_endian(decoder->private_->input, &obj->comments[i].length))
1757 					return false; /* read_callback_ sets the state for us */
1758 				if (obj->comments[i].length > 0) {
1759 					if (length < obj->comments[i].length) {
1760 						obj->num_comments = i;
1761 						goto skip;
1762 					}
1763 					else
1764 						length -= obj->comments[i].length;
1765 					if (0 == (obj->comments[i].entry = safe_malloc_add_2op_(obj->comments[i].length, /*+*/1))) {
1766 						decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1767 						return false;
1768 					}
1769 					memset (obj->comments[i].entry, 0, obj->comments[i].length) ;
1770 					if (!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, obj->comments[i].entry, obj->comments[i].length)) {
1771 						obj->num_comments = i;
1772 						goto skip;
1773 					}
1774 					obj->comments[i].entry[obj->comments[i].length] = '\0';
1775 				}
1776 				else
1777 					obj->comments[i].entry = 0;
1778 			}
1779 		}
1780 		else
1781 			obj->comments = 0;
1782 	}
1783 
1784   skip:
1785 	if (length > 0) {
1786 		/* This will only happen on files with invalid data in comments */
1787 		if(!FLAC__bitreader_skip_byte_block_aligned_no_crc(decoder->private_->input, length))
1788 			return false; /* read_callback_ sets the state for us */
1789 	}
1790 
1791 	return true;
1792 }
1793 
read_metadata_cuesheet_(FLAC__StreamDecoder * decoder,FLAC__StreamMetadata_CueSheet * obj)1794 FLAC__bool read_metadata_cuesheet_(FLAC__StreamDecoder *decoder, FLAC__StreamMetadata_CueSheet *obj)
1795 {
1796 	FLAC__uint32 i, j, x;
1797 
1798 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
1799 
1800 	memset(obj, 0, sizeof(FLAC__StreamMetadata_CueSheet));
1801 
1802 	FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_MEDIA_CATALOG_NUMBER_LEN % 8 == 0);
1803 	if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, (FLAC__byte*)obj->media_catalog_number, FLAC__STREAM_METADATA_CUESHEET_MEDIA_CATALOG_NUMBER_LEN/8))
1804 		return false; /* read_callback_ sets the state for us */
1805 
1806 	if(!FLAC__bitreader_read_raw_uint64(decoder->private_->input, &obj->lead_in, FLAC__STREAM_METADATA_CUESHEET_LEAD_IN_LEN))
1807 		return false; /* read_callback_ sets the state for us */
1808 
1809 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_CUESHEET_IS_CD_LEN))
1810 		return false; /* read_callback_ sets the state for us */
1811 	obj->is_cd = x? true : false;
1812 
1813 	if(!FLAC__bitreader_skip_bits_no_crc(decoder->private_->input, FLAC__STREAM_METADATA_CUESHEET_RESERVED_LEN))
1814 		return false; /* read_callback_ sets the state for us */
1815 
1816 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_CUESHEET_NUM_TRACKS_LEN))
1817 		return false; /* read_callback_ sets the state for us */
1818 	obj->num_tracks = x;
1819 
1820 	if(obj->num_tracks > 0) {
1821 		if(0 == (obj->tracks = safe_calloc_(obj->num_tracks, sizeof(FLAC__StreamMetadata_CueSheet_Track)))) {
1822 			decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1823 			return false;
1824 		}
1825 		for(i = 0; i < obj->num_tracks; i++) {
1826 			FLAC__StreamMetadata_CueSheet_Track *track = &obj->tracks[i];
1827 			if(!FLAC__bitreader_read_raw_uint64(decoder->private_->input, &track->offset, FLAC__STREAM_METADATA_CUESHEET_TRACK_OFFSET_LEN))
1828 				return false; /* read_callback_ sets the state for us */
1829 
1830 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_CUESHEET_TRACK_NUMBER_LEN))
1831 				return false; /* read_callback_ sets the state for us */
1832 			track->number = (FLAC__byte)x;
1833 
1834 			FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_ISRC_LEN % 8 == 0);
1835 			if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, (FLAC__byte*)track->isrc, FLAC__STREAM_METADATA_CUESHEET_TRACK_ISRC_LEN/8))
1836 				return false; /* read_callback_ sets the state for us */
1837 
1838 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN))
1839 				return false; /* read_callback_ sets the state for us */
1840 			track->type = x;
1841 
1842 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN))
1843 				return false; /* read_callback_ sets the state for us */
1844 			track->pre_emphasis = x;
1845 
1846 			if(!FLAC__bitreader_skip_bits_no_crc(decoder->private_->input, FLAC__STREAM_METADATA_CUESHEET_TRACK_RESERVED_LEN))
1847 				return false; /* read_callback_ sets the state for us */
1848 
1849 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_CUESHEET_TRACK_NUM_INDICES_LEN))
1850 				return false; /* read_callback_ sets the state for us */
1851 			track->num_indices = (FLAC__byte)x;
1852 
1853 			if(track->num_indices > 0) {
1854 				if(0 == (track->indices = safe_calloc_(track->num_indices, sizeof(FLAC__StreamMetadata_CueSheet_Index)))) {
1855 					decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1856 					return false;
1857 				}
1858 				for(j = 0; j < track->num_indices; j++) {
1859 					FLAC__StreamMetadata_CueSheet_Index *indx = &track->indices[j];
1860 					if(!FLAC__bitreader_read_raw_uint64(decoder->private_->input, &indx->offset, FLAC__STREAM_METADATA_CUESHEET_INDEX_OFFSET_LEN))
1861 						return false; /* read_callback_ sets the state for us */
1862 
1863 					if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_CUESHEET_INDEX_NUMBER_LEN))
1864 						return false; /* read_callback_ sets the state for us */
1865 					indx->number = (FLAC__byte)x;
1866 
1867 					if(!FLAC__bitreader_skip_bits_no_crc(decoder->private_->input, FLAC__STREAM_METADATA_CUESHEET_INDEX_RESERVED_LEN))
1868 						return false; /* read_callback_ sets the state for us */
1869 				}
1870 			}
1871 		}
1872 	}
1873 
1874 	return true;
1875 }
1876 
read_metadata_picture_(FLAC__StreamDecoder * decoder,FLAC__StreamMetadata_Picture * obj)1877 FLAC__bool read_metadata_picture_(FLAC__StreamDecoder *decoder, FLAC__StreamMetadata_Picture *obj)
1878 {
1879 	FLAC__uint32 x;
1880 
1881 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
1882 
1883 	/* read type */
1884 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_PICTURE_TYPE_LEN))
1885 		return false; /* read_callback_ sets the state for us */
1886 	obj->type = x;
1887 
1888 	/* read MIME type */
1889 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_PICTURE_MIME_TYPE_LENGTH_LEN))
1890 		return false; /* read_callback_ sets the state for us */
1891 	if(0 == (obj->mime_type = safe_malloc_add_2op_(x, /*+*/1))) {
1892 		decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1893 		return false;
1894 	}
1895 	if(x > 0) {
1896 		if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, (FLAC__byte*)obj->mime_type, x))
1897 			return false; /* read_callback_ sets the state for us */
1898 	}
1899 	obj->mime_type[x] = '\0';
1900 
1901 	/* read description */
1902 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__STREAM_METADATA_PICTURE_DESCRIPTION_LENGTH_LEN))
1903 		return false; /* read_callback_ sets the state for us */
1904 	if(0 == (obj->description = safe_malloc_add_2op_(x, /*+*/1))) {
1905 		decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1906 		return false;
1907 	}
1908 	if(x > 0) {
1909 		if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, obj->description, x))
1910 			return false; /* read_callback_ sets the state for us */
1911 	}
1912 	obj->description[x] = '\0';
1913 
1914 	/* read width */
1915 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &obj->width, FLAC__STREAM_METADATA_PICTURE_WIDTH_LEN))
1916 		return false; /* read_callback_ sets the state for us */
1917 
1918 	/* read height */
1919 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &obj->height, FLAC__STREAM_METADATA_PICTURE_HEIGHT_LEN))
1920 		return false; /* read_callback_ sets the state for us */
1921 
1922 	/* read depth */
1923 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &obj->depth, FLAC__STREAM_METADATA_PICTURE_DEPTH_LEN))
1924 		return false; /* read_callback_ sets the state for us */
1925 
1926 	/* read colors */
1927 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &obj->colors, FLAC__STREAM_METADATA_PICTURE_COLORS_LEN))
1928 		return false; /* read_callback_ sets the state for us */
1929 
1930 	/* read data */
1931 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &(obj->data_length), FLAC__STREAM_METADATA_PICTURE_DATA_LENGTH_LEN))
1932 		return false; /* read_callback_ sets the state for us */
1933 	if(0 == (obj->data = safe_malloc_(obj->data_length))) {
1934 		decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
1935 		return false;
1936 	}
1937 	if(obj->data_length > 0) {
1938 		if(!FLAC__bitreader_read_byte_block_aligned_no_crc(decoder->private_->input, obj->data, obj->data_length))
1939 			return false; /* read_callback_ sets the state for us */
1940 	}
1941 
1942 	return true;
1943 }
1944 
skip_id3v2_tag_(FLAC__StreamDecoder * decoder)1945 FLAC__bool skip_id3v2_tag_(FLAC__StreamDecoder *decoder)
1946 {
1947 	FLAC__uint32 x;
1948 	unsigned i, skip;
1949 
1950 	/* skip the version and flags bytes */
1951 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 24))
1952 		return false; /* read_callback_ sets the state for us */
1953 	/* get the size (in bytes) to skip */
1954 	skip = 0;
1955 	for(i = 0; i < 4; i++) {
1956 		if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
1957 			return false; /* read_callback_ sets the state for us */
1958 		skip <<= 7;
1959 		skip |= (x & 0x7f);
1960 	}
1961 	/* skip the rest of the tag */
1962 	if(!FLAC__bitreader_skip_byte_block_aligned_no_crc(decoder->private_->input, skip))
1963 		return false; /* read_callback_ sets the state for us */
1964 	return true;
1965 }
1966 
frame_sync_(FLAC__StreamDecoder * decoder)1967 FLAC__bool frame_sync_(FLAC__StreamDecoder *decoder)
1968 {
1969 	FLAC__uint32 x;
1970 	FLAC__bool first = true;
1971 
1972 	/* If we know the total number of samples in the stream, stop if we've read that many. */
1973 	/* This will stop us, for example, from wasting time trying to sync on an ID3V1 tag. */
1974 	if(FLAC__stream_decoder_get_total_samples(decoder) > 0) {
1975 		if(decoder->private_->samples_decoded >= FLAC__stream_decoder_get_total_samples(decoder)) {
1976 			decoder->protected_->state = FLAC__STREAM_DECODER_END_OF_STREAM;
1977 			return true;
1978 		}
1979 	}
1980 
1981 	/* make sure we're byte aligned */
1982 	if(!FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input)) {
1983 		if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__bitreader_bits_left_for_byte_alignment(decoder->private_->input)))
1984 			return false; /* read_callback_ sets the state for us */
1985 	}
1986 
1987 	while(1) {
1988 		if(decoder->private_->cached) {
1989 			x = (FLAC__uint32)decoder->private_->lookahead;
1990 			decoder->private_->cached = false;
1991 		}
1992 		else {
1993 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
1994 				return false; /* read_callback_ sets the state for us */
1995 		}
1996 		if(x == 0xff) { /* MAGIC NUMBER for the first 8 frame sync bits */
1997 			decoder->private_->header_warmup[0] = (FLAC__byte)x;
1998 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
1999 				return false; /* read_callback_ sets the state for us */
2000 
2001 			/* we have to check if we just read two 0xff's in a row; the second may actually be the beginning of the sync code */
2002 			/* else we have to check if the second byte is the end of a sync code */
2003 			if(x == 0xff) { /* MAGIC NUMBER for the first 8 frame sync bits */
2004 				decoder->private_->lookahead = (FLAC__byte)x;
2005 				decoder->private_->cached = true;
2006 			}
2007 			else if(x >> 1 == 0x7c) { /* MAGIC NUMBER for the last 6 sync bits and reserved 7th bit */
2008 				decoder->private_->header_warmup[1] = (FLAC__byte)x;
2009 				decoder->protected_->state = FLAC__STREAM_DECODER_READ_FRAME;
2010 				return true;
2011 			}
2012 		}
2013 		if(first) {
2014 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC);
2015 			first = false;
2016 		}
2017 	}
2018 
2019 	return true;
2020 }
2021 
read_frame_(FLAC__StreamDecoder * decoder,FLAC__bool * got_a_frame,FLAC__bool do_full_decode)2022 FLAC__bool read_frame_(FLAC__StreamDecoder *decoder, FLAC__bool *got_a_frame, FLAC__bool do_full_decode)
2023 {
2024 	unsigned channel;
2025 	unsigned i;
2026 	FLAC__int32 mid, side;
2027 	unsigned frame_crc; /* the one we calculate from the input stream */
2028 	FLAC__uint32 x;
2029 
2030 	*got_a_frame = false;
2031 
2032 	/* init the CRC */
2033 	frame_crc = 0;
2034 	frame_crc = FLAC__CRC16_UPDATE(decoder->private_->header_warmup[0], frame_crc);
2035 	frame_crc = FLAC__CRC16_UPDATE(decoder->private_->header_warmup[1], frame_crc);
2036 	FLAC__bitreader_reset_read_crc16(decoder->private_->input, (FLAC__uint16)frame_crc);
2037 
2038 	if(!read_frame_header_(decoder))
2039 		return false;
2040 	if(decoder->protected_->state == FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC) /* means we didn't sync on a valid header */
2041 		return true;
2042 	if(!allocate_output_(decoder, decoder->private_->frame.header.blocksize, decoder->private_->frame.header.channels))
2043 		return false;
2044 	for(channel = 0; channel < decoder->private_->frame.header.channels; channel++) {
2045 		/*
2046 		 * first figure the correct bits-per-sample of the subframe
2047 		 */
2048 		unsigned bps = decoder->private_->frame.header.bits_per_sample;
2049 		switch(decoder->private_->frame.header.channel_assignment) {
2050 			case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
2051 				/* no adjustment needed */
2052 				break;
2053 			case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
2054 				FLAC__ASSERT(decoder->private_->frame.header.channels == 2);
2055 				if(channel == 1)
2056 					bps++;
2057 				break;
2058 			case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
2059 				FLAC__ASSERT(decoder->private_->frame.header.channels == 2);
2060 				if(channel == 0)
2061 					bps++;
2062 				break;
2063 			case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
2064 				FLAC__ASSERT(decoder->private_->frame.header.channels == 2);
2065 				if(channel == 1)
2066 					bps++;
2067 				break;
2068 			default:
2069 				FLAC__ASSERT(0);
2070 		}
2071 		/*
2072 		 * now read it
2073 		 */
2074 		if(!read_subframe_(decoder, channel, bps, do_full_decode))
2075 			return false;
2076 		if(decoder->protected_->state == FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC) /* means bad sync or got corruption */
2077 			return true;
2078 	}
2079 	if(!read_zero_padding_(decoder))
2080 		return false;
2081 	if(decoder->protected_->state == FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC) /* means bad sync or got corruption (i.e. "zero bits" were not all zeroes) */
2082 		return true;
2083 
2084 	/*
2085 	 * Read the frame CRC-16 from the footer and check
2086 	 */
2087 	frame_crc = FLAC__bitreader_get_read_crc16(decoder->private_->input);
2088 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, FLAC__FRAME_FOOTER_CRC_LEN))
2089 		return false; /* read_callback_ sets the state for us */
2090 	if(frame_crc == x) {
2091 		if(do_full_decode) {
2092 			/* Undo any special channel coding */
2093 			switch(decoder->private_->frame.header.channel_assignment) {
2094 				case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
2095 					/* do nothing */
2096 					break;
2097 				case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
2098 					FLAC__ASSERT(decoder->private_->frame.header.channels == 2);
2099 					for(i = 0; i < decoder->private_->frame.header.blocksize; i++)
2100 						decoder->private_->output[1][i] = decoder->private_->output[0][i] - decoder->private_->output[1][i];
2101 					break;
2102 				case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
2103 					FLAC__ASSERT(decoder->private_->frame.header.channels == 2);
2104 					for(i = 0; i < decoder->private_->frame.header.blocksize; i++)
2105 						decoder->private_->output[0][i] += decoder->private_->output[1][i];
2106 					break;
2107 				case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
2108 					FLAC__ASSERT(decoder->private_->frame.header.channels == 2);
2109 					for(i = 0; i < decoder->private_->frame.header.blocksize; i++) {
2110 #if 1
2111 						mid = decoder->private_->output[0][i];
2112 						side = decoder->private_->output[1][i];
2113 						mid <<= 1;
2114 						mid |= (side & 1); /* i.e. if 'side' is odd... */
2115 						decoder->private_->output[0][i] = (mid + side) >> 1;
2116 						decoder->private_->output[1][i] = (mid - side) >> 1;
2117 #else
2118 						/* OPT: without 'side' temp variable */
2119 						mid = (decoder->private_->output[0][i] << 1) | (decoder->private_->output[1][i] & 1); /* i.e. if 'side' is odd... */
2120 						decoder->private_->output[0][i] = (mid + decoder->private_->output[1][i]) >> 1;
2121 						decoder->private_->output[1][i] = (mid - decoder->private_->output[1][i]) >> 1;
2122 #endif
2123 					}
2124 					break;
2125 				default:
2126 					FLAC__ASSERT(0);
2127 					break;
2128 			}
2129 		}
2130 	}
2131 	else {
2132 		/* Bad frame, emit error and zero the output signal */
2133 		send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_FRAME_CRC_MISMATCH);
2134 		if(do_full_decode) {
2135 			for(channel = 0; channel < decoder->private_->frame.header.channels; channel++) {
2136 				memset(decoder->private_->output[channel], 0, sizeof(FLAC__int32) * decoder->private_->frame.header.blocksize);
2137 			}
2138 		}
2139 	}
2140 
2141 	*got_a_frame = true;
2142 
2143 	/* we wait to update fixed_block_size until here, when we're sure we've got a proper frame and hence a correct blocksize */
2144 	if(decoder->private_->next_fixed_block_size)
2145 		decoder->private_->fixed_block_size = decoder->private_->next_fixed_block_size;
2146 
2147 	/* put the latest values into the public section of the decoder instance */
2148 	decoder->protected_->channels = decoder->private_->frame.header.channels;
2149 	decoder->protected_->channel_assignment = decoder->private_->frame.header.channel_assignment;
2150 	decoder->protected_->bits_per_sample = decoder->private_->frame.header.bits_per_sample;
2151 	decoder->protected_->sample_rate = decoder->private_->frame.header.sample_rate;
2152 	decoder->protected_->blocksize = decoder->private_->frame.header.blocksize;
2153 
2154 	FLAC__ASSERT(decoder->private_->frame.header.number_type == FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER);
2155 	decoder->private_->samples_decoded = decoder->private_->frame.header.number.sample_number + decoder->private_->frame.header.blocksize;
2156 
2157 	/* write it */
2158 	if(do_full_decode) {
2159 		if(write_audio_frame_to_client_(decoder, &decoder->private_->frame, (const FLAC__int32 * const *)decoder->private_->output) != FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE)
2160 			return false;
2161 	}
2162 
2163 	decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2164 	return true;
2165 }
2166 
read_frame_header_(FLAC__StreamDecoder * decoder)2167 FLAC__bool read_frame_header_(FLAC__StreamDecoder *decoder)
2168 {
2169 	FLAC__uint32 x;
2170 	FLAC__uint64 xx;
2171 	unsigned i, blocksize_hint = 0, sample_rate_hint = 0;
2172 	FLAC__byte crc8, raw_header[16]; /* MAGIC NUMBER based on the maximum frame header size, including CRC */
2173 	unsigned raw_header_len;
2174 	FLAC__bool is_unparseable = false;
2175 
2176 	FLAC__ASSERT(FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input));
2177 
2178 	/* init the raw header with the saved bits from synchronization */
2179 	raw_header[0] = decoder->private_->header_warmup[0];
2180 	raw_header[1] = decoder->private_->header_warmup[1];
2181 	raw_header_len = 2;
2182 
2183 	/* check to make sure that reserved bit is 0 */
2184 	if(raw_header[1] & 0x02) /* MAGIC NUMBER */
2185 		is_unparseable = true;
2186 
2187 	/*
2188 	 * Note that along the way as we read the header, we look for a sync
2189 	 * code inside.  If we find one it would indicate that our original
2190 	 * sync was bad since there cannot be a sync code in a valid header.
2191 	 *
2192 	 * Three kinds of things can go wrong when reading the frame header:
2193 	 *  1) We may have sync'ed incorrectly and not landed on a frame header.
2194 	 *     If we don't find a sync code, it can end up looking like we read
2195 	 *     a valid but unparseable header, until getting to the frame header
2196 	 *     CRC.  Even then we could get a false positive on the CRC.
2197 	 *  2) We may have sync'ed correctly but on an unparseable frame (from a
2198 	 *     future encoder).
2199 	 *  3) We may be on a damaged frame which appears valid but unparseable.
2200 	 *
2201 	 * For all these reasons, we try and read a complete frame header as
2202 	 * long as it seems valid, even if unparseable, up until the frame
2203 	 * header CRC.
2204 	 */
2205 
2206 	/*
2207 	 * read in the raw header as bytes so we can CRC it, and parse it on the way
2208 	 */
2209 	for(i = 0; i < 2; i++) {
2210 		if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
2211 			return false; /* read_callback_ sets the state for us */
2212 		if(x == 0xff) { /* MAGIC NUMBER for the first 8 frame sync bits */
2213 			/* if we get here it means our original sync was erroneous since the sync code cannot appear in the header */
2214 			decoder->private_->lookahead = (FLAC__byte)x;
2215 			decoder->private_->cached = true;
2216 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_BAD_HEADER);
2217 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2218 			return true;
2219 		}
2220 		raw_header[raw_header_len++] = (FLAC__byte)x;
2221 	}
2222 
2223 	switch(x = raw_header[2] >> 4) {
2224 		case 0:
2225 			is_unparseable = true;
2226 			break;
2227 		case 1:
2228 			decoder->private_->frame.header.blocksize = 192;
2229 			break;
2230 		case 2:
2231 		case 3:
2232 		case 4:
2233 		case 5:
2234 			decoder->private_->frame.header.blocksize = 576 << (x-2);
2235 			break;
2236 		case 6:
2237 		case 7:
2238 			blocksize_hint = x;
2239 			break;
2240 		case 8:
2241 		case 9:
2242 		case 10:
2243 		case 11:
2244 		case 12:
2245 		case 13:
2246 		case 14:
2247 		case 15:
2248 			decoder->private_->frame.header.blocksize = 256 << (x-8);
2249 			break;
2250 		default:
2251 			FLAC__ASSERT(0);
2252 			break;
2253 	}
2254 
2255 	switch(x = raw_header[2] & 0x0f) {
2256 		case 0:
2257 			if(decoder->private_->has_stream_info)
2258 				decoder->private_->frame.header.sample_rate = decoder->private_->stream_info.data.stream_info.sample_rate;
2259 			else
2260 				is_unparseable = true;
2261 			break;
2262 		case 1:
2263 			decoder->private_->frame.header.sample_rate = 88200;
2264 			break;
2265 		case 2:
2266 			decoder->private_->frame.header.sample_rate = 176400;
2267 			break;
2268 		case 3:
2269 			decoder->private_->frame.header.sample_rate = 192000;
2270 			break;
2271 		case 4:
2272 			decoder->private_->frame.header.sample_rate = 8000;
2273 			break;
2274 		case 5:
2275 			decoder->private_->frame.header.sample_rate = 16000;
2276 			break;
2277 		case 6:
2278 			decoder->private_->frame.header.sample_rate = 22050;
2279 			break;
2280 		case 7:
2281 			decoder->private_->frame.header.sample_rate = 24000;
2282 			break;
2283 		case 8:
2284 			decoder->private_->frame.header.sample_rate = 32000;
2285 			break;
2286 		case 9:
2287 			decoder->private_->frame.header.sample_rate = 44100;
2288 			break;
2289 		case 10:
2290 			decoder->private_->frame.header.sample_rate = 48000;
2291 			break;
2292 		case 11:
2293 			decoder->private_->frame.header.sample_rate = 96000;
2294 			break;
2295 		case 12:
2296 		case 13:
2297 		case 14:
2298 			sample_rate_hint = x;
2299 			break;
2300 		case 15:
2301 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_BAD_HEADER);
2302 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2303 			return true;
2304 		default:
2305 			FLAC__ASSERT(0);
2306 	}
2307 
2308 	x = (unsigned)(raw_header[3] >> 4);
2309 	if(x & 8) {
2310 		decoder->private_->frame.header.channels = 2;
2311 		switch(x & 7) {
2312 			case 0:
2313 				decoder->private_->frame.header.channel_assignment = FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE;
2314 				break;
2315 			case 1:
2316 				decoder->private_->frame.header.channel_assignment = FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE;
2317 				break;
2318 			case 2:
2319 				decoder->private_->frame.header.channel_assignment = FLAC__CHANNEL_ASSIGNMENT_MID_SIDE;
2320 				break;
2321 			default:
2322 				is_unparseable = true;
2323 				break;
2324 		}
2325 	}
2326 	else {
2327 		decoder->private_->frame.header.channels = (unsigned)x + 1;
2328 		decoder->private_->frame.header.channel_assignment = FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT;
2329 	}
2330 
2331 	switch(x = (unsigned)(raw_header[3] & 0x0e) >> 1) {
2332 		case 0:
2333 			if(decoder->private_->has_stream_info)
2334 				decoder->private_->frame.header.bits_per_sample = decoder->private_->stream_info.data.stream_info.bits_per_sample;
2335 			else
2336 				is_unparseable = true;
2337 			break;
2338 		case 1:
2339 			decoder->private_->frame.header.bits_per_sample = 8;
2340 			break;
2341 		case 2:
2342 			decoder->private_->frame.header.bits_per_sample = 12;
2343 			break;
2344 		case 4:
2345 			decoder->private_->frame.header.bits_per_sample = 16;
2346 			break;
2347 		case 5:
2348 			decoder->private_->frame.header.bits_per_sample = 20;
2349 			break;
2350 		case 6:
2351 			decoder->private_->frame.header.bits_per_sample = 24;
2352 			break;
2353 		case 3:
2354 		case 7:
2355 			is_unparseable = true;
2356 			break;
2357 		default:
2358 			FLAC__ASSERT(0);
2359 			break;
2360 	}
2361 
2362 	/* check to make sure that reserved bit is 0 */
2363 	if(raw_header[3] & 0x01) /* MAGIC NUMBER */
2364 		is_unparseable = true;
2365 
2366 	/* read the frame's starting sample number (or frame number as the case may be) */
2367 	if(
2368 		raw_header[1] & 0x01 ||
2369 		/*@@@ this clause is a concession to the old way of doing variable blocksize; the only known implementation is flake and can probably be removed without inconveniencing anyone */
2370 		(decoder->private_->has_stream_info && decoder->private_->stream_info.data.stream_info.min_blocksize != decoder->private_->stream_info.data.stream_info.max_blocksize)
2371 	) { /* variable blocksize */
2372 		if(!FLAC__bitreader_read_utf8_uint64(decoder->private_->input, &xx, raw_header, &raw_header_len))
2373 			return false; /* read_callback_ sets the state for us */
2374 		if(xx == FLAC__U64L(0xffffffffffffffff)) { /* i.e. non-UTF8 code... */
2375 			decoder->private_->lookahead = raw_header[raw_header_len-1]; /* back up as much as we can */
2376 			decoder->private_->cached = true;
2377 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_BAD_HEADER);
2378 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2379 			return true;
2380 		}
2381 		decoder->private_->frame.header.number_type = FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER;
2382 		decoder->private_->frame.header.number.sample_number = xx;
2383 	}
2384 	else { /* fixed blocksize */
2385 		if(!FLAC__bitreader_read_utf8_uint32(decoder->private_->input, &x, raw_header, &raw_header_len))
2386 			return false; /* read_callback_ sets the state for us */
2387 		if(x == 0xffffffff) { /* i.e. non-UTF8 code... */
2388 			decoder->private_->lookahead = raw_header[raw_header_len-1]; /* back up as much as we can */
2389 			decoder->private_->cached = true;
2390 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_BAD_HEADER);
2391 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2392 			return true;
2393 		}
2394 		decoder->private_->frame.header.number_type = FLAC__FRAME_NUMBER_TYPE_FRAME_NUMBER;
2395 		decoder->private_->frame.header.number.frame_number = x;
2396 	}
2397 
2398 	if(blocksize_hint) {
2399 		if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
2400 			return false; /* read_callback_ sets the state for us */
2401 		raw_header[raw_header_len++] = (FLAC__byte)x;
2402 		if(blocksize_hint == 7) {
2403 			FLAC__uint32 _x;
2404 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &_x, 8))
2405 				return false; /* read_callback_ sets the state for us */
2406 			raw_header[raw_header_len++] = (FLAC__byte)_x;
2407 			x = (x << 8) | _x;
2408 		}
2409 		decoder->private_->frame.header.blocksize = x+1;
2410 	}
2411 
2412 	if(sample_rate_hint) {
2413 		if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
2414 			return false; /* read_callback_ sets the state for us */
2415 		raw_header[raw_header_len++] = (FLAC__byte)x;
2416 		if(sample_rate_hint != 12) {
2417 			FLAC__uint32 _x;
2418 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &_x, 8))
2419 				return false; /* read_callback_ sets the state for us */
2420 			raw_header[raw_header_len++] = (FLAC__byte)_x;
2421 			x = (x << 8) | _x;
2422 		}
2423 		if(sample_rate_hint == 12)
2424 			decoder->private_->frame.header.sample_rate = x*1000;
2425 		else if(sample_rate_hint == 13)
2426 			decoder->private_->frame.header.sample_rate = x;
2427 		else
2428 			decoder->private_->frame.header.sample_rate = x*10;
2429 	}
2430 
2431 	/* read the CRC-8 byte */
2432 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8))
2433 		return false; /* read_callback_ sets the state for us */
2434 	crc8 = (FLAC__byte)x;
2435 
2436 	if(FLAC__crc8(raw_header, raw_header_len) != crc8) {
2437 		send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_BAD_HEADER);
2438 		decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2439 		return true;
2440 	}
2441 
2442 	/* calculate the sample number from the frame number if needed */
2443 	decoder->private_->next_fixed_block_size = 0;
2444 	if(decoder->private_->frame.header.number_type == FLAC__FRAME_NUMBER_TYPE_FRAME_NUMBER) {
2445 		x = decoder->private_->frame.header.number.frame_number;
2446 		decoder->private_->frame.header.number_type = FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER;
2447 		if(decoder->private_->fixed_block_size)
2448 			decoder->private_->frame.header.number.sample_number = (FLAC__uint64)decoder->private_->fixed_block_size * (FLAC__uint64)x;
2449 		else if(decoder->private_->has_stream_info) {
2450 			if(decoder->private_->stream_info.data.stream_info.min_blocksize == decoder->private_->stream_info.data.stream_info.max_blocksize) {
2451 				decoder->private_->frame.header.number.sample_number = (FLAC__uint64)decoder->private_->stream_info.data.stream_info.min_blocksize * (FLAC__uint64)x;
2452 				decoder->private_->next_fixed_block_size = decoder->private_->stream_info.data.stream_info.max_blocksize;
2453 			}
2454 			else
2455 				is_unparseable = true;
2456 		}
2457 		else if(x == 0) {
2458 			decoder->private_->frame.header.number.sample_number = 0;
2459 			decoder->private_->next_fixed_block_size = decoder->private_->frame.header.blocksize;
2460 		}
2461 		else {
2462 			/* can only get here if the stream has invalid frame numbering and no STREAMINFO, so assume it's not the last (possibly short) frame */
2463 			decoder->private_->frame.header.number.sample_number = (FLAC__uint64)decoder->private_->frame.header.blocksize * (FLAC__uint64)x;
2464 		}
2465 	}
2466 
2467 	if(is_unparseable) {
2468 		send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_UNPARSEABLE_STREAM);
2469 		decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2470 		return true;
2471 	}
2472 
2473 	return true;
2474 }
2475 
read_subframe_(FLAC__StreamDecoder * decoder,unsigned channel,unsigned bps,FLAC__bool do_full_decode)2476 FLAC__bool read_subframe_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, FLAC__bool do_full_decode)
2477 {
2478 	FLAC__uint32 x;
2479 	FLAC__bool wasted_bits;
2480 	unsigned i;
2481 
2482 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &x, 8)) /* MAGIC NUMBER */
2483 		return false; /* read_callback_ sets the state for us */
2484 
2485 	wasted_bits = (x & 1);
2486 	x &= 0xfe;
2487 
2488 	if(wasted_bits) {
2489 		unsigned u;
2490 		if(!FLAC__bitreader_read_unary_unsigned(decoder->private_->input, &u))
2491 			return false; /* read_callback_ sets the state for us */
2492 		decoder->private_->frame.subframes[channel].wasted_bits = u+1;
2493 		bps -= decoder->private_->frame.subframes[channel].wasted_bits;
2494 	}
2495 	else
2496 		decoder->private_->frame.subframes[channel].wasted_bits = 0;
2497 
2498 	/*
2499 	 * Lots of magic numbers here
2500 	 */
2501 	if(x & 0x80) {
2502 		send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC);
2503 		decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2504 		return true;
2505 	}
2506 	else if(x == 0) {
2507 		if(!read_subframe_constant_(decoder, channel, bps, do_full_decode))
2508 			return false;
2509 	}
2510 	else if(x == 2) {
2511 		if(!read_subframe_verbatim_(decoder, channel, bps, do_full_decode))
2512 			return false;
2513 	}
2514 	else if(x < 16) {
2515 		send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_UNPARSEABLE_STREAM);
2516 		decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2517 		return true;
2518 	}
2519 	else if(x <= 24) {
2520 		if(!read_subframe_fixed_(decoder, channel, bps, (x>>1)&7, do_full_decode))
2521 			return false;
2522 		if(decoder->protected_->state == FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC) /* means bad sync or got corruption */
2523 			return true;
2524 	}
2525 	else if(x < 64) {
2526 		send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_UNPARSEABLE_STREAM);
2527 		decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2528 		return true;
2529 	}
2530 	else {
2531 		if(!read_subframe_lpc_(decoder, channel, bps, ((x>>1)&31)+1, do_full_decode))
2532 			return false;
2533 		if(decoder->protected_->state == FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC) /* means bad sync or got corruption */
2534 			return true;
2535 	}
2536 
2537 	if(wasted_bits && do_full_decode) {
2538 		x = decoder->private_->frame.subframes[channel].wasted_bits;
2539 		for(i = 0; i < decoder->private_->frame.header.blocksize; i++)
2540 			decoder->private_->output[channel][i] <<= x;
2541 	}
2542 
2543 	return true;
2544 }
2545 
read_subframe_constant_(FLAC__StreamDecoder * decoder,unsigned channel,unsigned bps,FLAC__bool do_full_decode)2546 FLAC__bool read_subframe_constant_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, FLAC__bool do_full_decode)
2547 {
2548 	FLAC__Subframe_Constant *subframe = &decoder->private_->frame.subframes[channel].data.constant;
2549 	FLAC__int32 x;
2550 	unsigned i;
2551 	FLAC__int32 *output = decoder->private_->output[channel];
2552 
2553 	decoder->private_->frame.subframes[channel].type = FLAC__SUBFRAME_TYPE_CONSTANT;
2554 
2555 	if(!FLAC__bitreader_read_raw_int32(decoder->private_->input, &x, bps))
2556 		return false; /* read_callback_ sets the state for us */
2557 
2558 	subframe->value = x;
2559 
2560 	/* decode the subframe */
2561 	if(do_full_decode) {
2562 		for(i = 0; i < decoder->private_->frame.header.blocksize; i++)
2563 			output[i] = x;
2564 	}
2565 
2566 	return true;
2567 }
2568 
read_subframe_fixed_(FLAC__StreamDecoder * decoder,unsigned channel,unsigned bps,const unsigned order,FLAC__bool do_full_decode)2569 FLAC__bool read_subframe_fixed_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, const unsigned order, FLAC__bool do_full_decode)
2570 {
2571 	FLAC__Subframe_Fixed *subframe = &decoder->private_->frame.subframes[channel].data.fixed;
2572 	FLAC__int32 i32;
2573 	FLAC__uint32 u32;
2574 	unsigned u;
2575 
2576 	decoder->private_->frame.subframes[channel].type = FLAC__SUBFRAME_TYPE_FIXED;
2577 
2578 	subframe->residual = decoder->private_->residual[channel];
2579 	subframe->order = order;
2580 
2581 	/* read warm-up samples */
2582 	for(u = 0; u < order; u++) {
2583 		if(!FLAC__bitreader_read_raw_int32(decoder->private_->input, &i32, bps))
2584 			return false; /* read_callback_ sets the state for us */
2585 		subframe->warmup[u] = i32;
2586 	}
2587 
2588 	/* read entropy coding method info */
2589 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &u32, FLAC__ENTROPY_CODING_METHOD_TYPE_LEN))
2590 		return false; /* read_callback_ sets the state for us */
2591 	subframe->entropy_coding_method.type = (FLAC__EntropyCodingMethodType)u32;
2592 	switch(subframe->entropy_coding_method.type) {
2593 		case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE:
2594 		case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE2:
2595 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &u32, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN))
2596 				return false; /* read_callback_ sets the state for us */
2597 			subframe->entropy_coding_method.data.partitioned_rice.order = u32;
2598 			subframe->entropy_coding_method.data.partitioned_rice.contents = &decoder->private_->partitioned_rice_contents[channel];
2599 			break;
2600 		default:
2601 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_UNPARSEABLE_STREAM);
2602 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2603 			return true;
2604 	}
2605 
2606 	/* read residual */
2607 	switch(subframe->entropy_coding_method.type) {
2608 		case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE:
2609 		case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE2:
2610 			if(!read_residual_partitioned_rice_(decoder, order, subframe->entropy_coding_method.data.partitioned_rice.order, &decoder->private_->partitioned_rice_contents[channel], decoder->private_->residual[channel], /*is_extended=*/subframe->entropy_coding_method.type == FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE2))
2611 				return false;
2612 			break;
2613 		default:
2614 			FLAC__ASSERT(0);
2615 	}
2616 
2617 	/* decode the subframe */
2618 	if(do_full_decode) {
2619 		memcpy(decoder->private_->output[channel], subframe->warmup, sizeof(FLAC__int32) * order);
2620 		FLAC__fixed_restore_signal(decoder->private_->residual[channel], decoder->private_->frame.header.blocksize-order, order, decoder->private_->output[channel]+order);
2621 	}
2622 
2623 	return true;
2624 }
2625 
read_subframe_lpc_(FLAC__StreamDecoder * decoder,unsigned channel,unsigned bps,const unsigned order,FLAC__bool do_full_decode)2626 FLAC__bool read_subframe_lpc_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, const unsigned order, FLAC__bool do_full_decode)
2627 {
2628 	FLAC__Subframe_LPC *subframe = &decoder->private_->frame.subframes[channel].data.lpc;
2629 	FLAC__int32 i32;
2630 	FLAC__uint32 u32;
2631 	unsigned u;
2632 
2633 	decoder->private_->frame.subframes[channel].type = FLAC__SUBFRAME_TYPE_LPC;
2634 
2635 	subframe->residual = decoder->private_->residual[channel];
2636 	subframe->order = order;
2637 
2638 	/* read warm-up samples */
2639 	for(u = 0; u < order; u++) {
2640 		if(!FLAC__bitreader_read_raw_int32(decoder->private_->input, &i32, bps))
2641 			return false; /* read_callback_ sets the state for us */
2642 		subframe->warmup[u] = i32;
2643 	}
2644 
2645 	/* read qlp coeff precision */
2646 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &u32, FLAC__SUBFRAME_LPC_QLP_COEFF_PRECISION_LEN))
2647 		return false; /* read_callback_ sets the state for us */
2648 	if(u32 == (1u << FLAC__SUBFRAME_LPC_QLP_COEFF_PRECISION_LEN) - 1) {
2649 		send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC);
2650 		decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2651 		return true;
2652 	}
2653 	subframe->qlp_coeff_precision = u32+1;
2654 
2655 	/* read qlp shift */
2656 	if(!FLAC__bitreader_read_raw_int32(decoder->private_->input, &i32, FLAC__SUBFRAME_LPC_QLP_SHIFT_LEN))
2657 		return false; /* read_callback_ sets the state for us */
2658 	subframe->quantization_level = i32;
2659 
2660 	/* read quantized lp coefficiencts */
2661 	for(u = 0; u < order; u++) {
2662 		if(!FLAC__bitreader_read_raw_int32(decoder->private_->input, &i32, subframe->qlp_coeff_precision))
2663 			return false; /* read_callback_ sets the state for us */
2664 		subframe->qlp_coeff[u] = i32;
2665 	}
2666 
2667 	/* read entropy coding method info */
2668 	if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &u32, FLAC__ENTROPY_CODING_METHOD_TYPE_LEN))
2669 		return false; /* read_callback_ sets the state for us */
2670 	subframe->entropy_coding_method.type = (FLAC__EntropyCodingMethodType)u32;
2671 	switch(subframe->entropy_coding_method.type) {
2672 		case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE:
2673 		case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE2:
2674 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &u32, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN))
2675 				return false; /* read_callback_ sets the state for us */
2676 			subframe->entropy_coding_method.data.partitioned_rice.order = u32;
2677 			subframe->entropy_coding_method.data.partitioned_rice.contents = &decoder->private_->partitioned_rice_contents[channel];
2678 			break;
2679 		default:
2680 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_UNPARSEABLE_STREAM);
2681 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2682 			return true;
2683 	}
2684 
2685 	/* read residual */
2686 	switch(subframe->entropy_coding_method.type) {
2687 		case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE:
2688 		case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE2:
2689 			if(!read_residual_partitioned_rice_(decoder, order, subframe->entropy_coding_method.data.partitioned_rice.order, &decoder->private_->partitioned_rice_contents[channel], decoder->private_->residual[channel], /*is_extended=*/subframe->entropy_coding_method.type == FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE2))
2690 				return false;
2691 			break;
2692 		default:
2693 			FLAC__ASSERT(0);
2694 	}
2695 
2696 	/* decode the subframe */
2697 	if(do_full_decode) {
2698 		memcpy(decoder->private_->output[channel], subframe->warmup, sizeof(FLAC__int32) * order);
2699 		/*@@@@@@ technically not pessimistic enough, should be more like
2700 		if( (FLAC__uint64)order * ((((FLAC__uint64)1)<<bps)-1) * ((1<<subframe->qlp_coeff_precision)-1) < (((FLAC__uint64)-1) << 32) )
2701 		*/
2702 		if(bps + subframe->qlp_coeff_precision + FLAC__bitmath_ilog2(order) <= 32)
2703 			if(bps <= 16 && subframe->qlp_coeff_precision <= 16)
2704 				decoder->private_->local_lpc_restore_signal_16bit(decoder->private_->residual[channel], decoder->private_->frame.header.blocksize-order, subframe->qlp_coeff, order, subframe->quantization_level, decoder->private_->output[channel]+order);
2705 			else
2706 				decoder->private_->local_lpc_restore_signal(decoder->private_->residual[channel], decoder->private_->frame.header.blocksize-order, subframe->qlp_coeff, order, subframe->quantization_level, decoder->private_->output[channel]+order);
2707 		else
2708 			decoder->private_->local_lpc_restore_signal_64bit(decoder->private_->residual[channel], decoder->private_->frame.header.blocksize-order, subframe->qlp_coeff, order, subframe->quantization_level, decoder->private_->output[channel]+order);
2709 	}
2710 
2711 	return true;
2712 }
2713 
read_subframe_verbatim_(FLAC__StreamDecoder * decoder,unsigned channel,unsigned bps,FLAC__bool do_full_decode)2714 FLAC__bool read_subframe_verbatim_(FLAC__StreamDecoder *decoder, unsigned channel, unsigned bps, FLAC__bool do_full_decode)
2715 {
2716 	FLAC__Subframe_Verbatim *subframe = &decoder->private_->frame.subframes[channel].data.verbatim;
2717 	FLAC__int32 x, *residual = decoder->private_->residual[channel];
2718 	unsigned i;
2719 
2720 	decoder->private_->frame.subframes[channel].type = FLAC__SUBFRAME_TYPE_VERBATIM;
2721 
2722 	subframe->data = residual;
2723 
2724 	for(i = 0; i < decoder->private_->frame.header.blocksize; i++) {
2725 		if(!FLAC__bitreader_read_raw_int32(decoder->private_->input, &x, bps))
2726 			return false; /* read_callback_ sets the state for us */
2727 		residual[i] = x;
2728 	}
2729 
2730 	/* decode the subframe */
2731 	if(do_full_decode)
2732 		memcpy(decoder->private_->output[channel], subframe->data, sizeof(FLAC__int32) * decoder->private_->frame.header.blocksize);
2733 
2734 	return true;
2735 }
2736 
read_residual_partitioned_rice_(FLAC__StreamDecoder * decoder,unsigned predictor_order,unsigned partition_order,FLAC__EntropyCodingMethod_PartitionedRiceContents * partitioned_rice_contents,FLAC__int32 * residual,FLAC__bool is_extended)2737 FLAC__bool read_residual_partitioned_rice_(FLAC__StreamDecoder *decoder, unsigned predictor_order, unsigned partition_order, FLAC__EntropyCodingMethod_PartitionedRiceContents *partitioned_rice_contents, FLAC__int32 *residual, FLAC__bool is_extended)
2738 {
2739 	FLAC__uint32 rice_parameter;
2740 	int i;
2741 	unsigned partition, sample, u;
2742 	const unsigned partitions = 1u << partition_order;
2743 	const unsigned partition_samples = partition_order > 0? decoder->private_->frame.header.blocksize >> partition_order : decoder->private_->frame.header.blocksize - predictor_order;
2744 	const unsigned plen = is_extended? FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE2_PARAMETER_LEN : FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN;
2745 	const unsigned pesc = is_extended? FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE2_ESCAPE_PARAMETER : FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER;
2746 
2747 	/* sanity checks */
2748 	if(partition_order == 0) {
2749 		if(decoder->private_->frame.header.blocksize < predictor_order) {
2750 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC);
2751 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2752 			return true;
2753 		}
2754 	}
2755 	else {
2756 		if(partition_samples < predictor_order) {
2757 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC);
2758 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2759 			return true;
2760 		}
2761 	}
2762 
2763 	if(!FLAC__format_entropy_coding_method_partitioned_rice_contents_ensure_size(partitioned_rice_contents, flac_max(6u, partition_order))) {
2764 		decoder->protected_->state = FLAC__STREAM_DECODER_MEMORY_ALLOCATION_ERROR;
2765 		return false;
2766 	}
2767 
2768 	sample = 0;
2769 	for(partition = 0; partition < partitions; partition++) {
2770 		if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &rice_parameter, plen))
2771 			return false; /* read_callback_ sets the state for us */
2772 		partitioned_rice_contents->parameters[partition] = rice_parameter;
2773 		if(rice_parameter < pesc) {
2774 			partitioned_rice_contents->raw_bits[partition] = 0;
2775 			u = (partition_order == 0 || partition > 0)? partition_samples : partition_samples - predictor_order;
2776 			if(!FLAC__bitreader_read_rice_signed_block(decoder->private_->input, residual + sample, u, rice_parameter))
2777 				return false; /* read_callback_ sets the state for us */
2778 			sample += u;
2779 		}
2780 		else {
2781 			if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &rice_parameter, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_RAW_LEN))
2782 				return false; /* read_callback_ sets the state for us */
2783 			partitioned_rice_contents->raw_bits[partition] = rice_parameter;
2784 			for(u = (partition_order == 0 || partition > 0)? 0 : predictor_order; u < partition_samples; u++, sample++) {
2785 				if(!FLAC__bitreader_read_raw_int32(decoder->private_->input, &i, rice_parameter))
2786 					return false; /* read_callback_ sets the state for us */
2787 				residual[sample] = i;
2788 			}
2789 		}
2790 	}
2791 
2792 	return true;
2793 }
2794 
read_zero_padding_(FLAC__StreamDecoder * decoder)2795 FLAC__bool read_zero_padding_(FLAC__StreamDecoder *decoder)
2796 {
2797 	if(!FLAC__bitreader_is_consumed_byte_aligned(decoder->private_->input)) {
2798 		FLAC__uint32 zero = 0;
2799 		if(!FLAC__bitreader_read_raw_uint32(decoder->private_->input, &zero, FLAC__bitreader_bits_left_for_byte_alignment(decoder->private_->input)))
2800 			return false; /* read_callback_ sets the state for us */
2801 		if(zero != 0) {
2802 			send_error_to_client_(decoder, FLAC__STREAM_DECODER_ERROR_STATUS_LOST_SYNC);
2803 			decoder->protected_->state = FLAC__STREAM_DECODER_SEARCH_FOR_FRAME_SYNC;
2804 		}
2805 	}
2806 	return true;
2807 }
2808 
read_callback_(FLAC__byte buffer[],size_t * bytes,void * client_data)2809 FLAC__bool read_callback_(FLAC__byte buffer[], size_t *bytes, void *client_data)
2810 {
2811 	FLAC__StreamDecoder *decoder = (FLAC__StreamDecoder *)client_data;
2812 
2813 	if(
2814 #if FLAC__HAS_OGG
2815 		/* see [1] HACK NOTE below for why we don't call the eof_callback when decoding Ogg FLAC */
2816 		!decoder->private_->is_ogg &&
2817 #endif
2818 		decoder->private_->eof_callback && decoder->private_->eof_callback(decoder, decoder->private_->client_data)
2819 	) {
2820 		*bytes = 0;
2821 		decoder->protected_->state = FLAC__STREAM_DECODER_END_OF_STREAM;
2822 		return false;
2823 	}
2824 	else if(*bytes > 0) {
2825 		/* While seeking, it is possible for our seek to land in the
2826 		 * middle of audio data that looks exactly like a frame header
2827 		 * from a future version of an encoder.  When that happens, our
2828 		 * error callback will get an
2829 		 * FLAC__STREAM_DECODER_UNPARSEABLE_STREAM and increment its
2830 		 * unparseable_frame_count.  But there is a remote possibility
2831 		 * that it is properly synced at such a "future-codec frame",
2832 		 * so to make sure, we wait to see many "unparseable" errors in
2833 		 * a row before bailing out.
2834 		 */
2835 		if(decoder->private_->is_seeking && decoder->private_->unparseable_frame_count > 20) {
2836 			decoder->protected_->state = FLAC__STREAM_DECODER_ABORTED;
2837 			return false;
2838 		}
2839 		else {
2840 			const FLAC__StreamDecoderReadStatus status =
2841 #if FLAC__HAS_OGG
2842 				decoder->private_->is_ogg?
2843 				read_callback_ogg_aspect_(decoder, buffer, bytes) :
2844 #endif
2845 				decoder->private_->read_callback(decoder, buffer, bytes, decoder->private_->client_data)
2846 			;
2847 			if(status == FLAC__STREAM_DECODER_READ_STATUS_ABORT) {
2848 				decoder->protected_->state = FLAC__STREAM_DECODER_ABORTED;
2849 				return false;
2850 			}
2851 			else if(*bytes == 0) {
2852 				if(
2853 					status == FLAC__STREAM_DECODER_READ_STATUS_END_OF_STREAM ||
2854 					(
2855 #if FLAC__HAS_OGG
2856 						/* see [1] HACK NOTE below for why we don't call the eof_callback when decoding Ogg FLAC */
2857 						!decoder->private_->is_ogg &&
2858 #endif
2859 						decoder->private_->eof_callback && decoder->private_->eof_callback(decoder, decoder->private_->client_data)
2860 					)
2861 				) {
2862 					decoder->protected_->state = FLAC__STREAM_DECODER_END_OF_STREAM;
2863 					return false;
2864 				}
2865 				else
2866 					return true;
2867 			}
2868 			else
2869 				return true;
2870 		}
2871 	}
2872 	else {
2873 		/* abort to avoid a deadlock */
2874 		decoder->protected_->state = FLAC__STREAM_DECODER_ABORTED;
2875 		return false;
2876 	}
2877 	/* [1] @@@ HACK NOTE: The end-of-stream checking has to be hacked around
2878 	 * for Ogg FLAC.  This is because the ogg decoder aspect can lose sync
2879 	 * and at the same time hit the end of the stream (for example, seeking
2880 	 * to a point that is after the beginning of the last Ogg page).  There
2881 	 * is no way to report an Ogg sync loss through the callbacks (see note
2882 	 * in read_callback_ogg_aspect_()) so it returns CONTINUE with *bytes==0.
2883 	 * So to keep the decoder from stopping at this point we gate the call
2884 	 * to the eof_callback and let the Ogg decoder aspect set the
2885 	 * end-of-stream state when it is needed.
2886 	 */
2887 }
2888 
2889 #if FLAC__HAS_OGG
read_callback_ogg_aspect_(const FLAC__StreamDecoder * decoder,FLAC__byte buffer[],size_t * bytes)2890 FLAC__StreamDecoderReadStatus read_callback_ogg_aspect_(const FLAC__StreamDecoder *decoder, FLAC__byte buffer[], size_t *bytes)
2891 {
2892 	switch(FLAC__ogg_decoder_aspect_read_callback_wrapper(&decoder->protected_->ogg_decoder_aspect, buffer, bytes, read_callback_proxy_, decoder, decoder->private_->client_data)) {
2893 		case FLAC__OGG_DECODER_ASPECT_READ_STATUS_OK:
2894 			return FLAC__STREAM_DECODER_READ_STATUS_CONTINUE;
2895 		/* we don't really have a way to handle lost sync via read
2896 		 * callback so we'll let it pass and let the underlying
2897 		 * FLAC decoder catch the error
2898 		 */
2899 		case FLAC__OGG_DECODER_ASPECT_READ_STATUS_LOST_SYNC:
2900 			return FLAC__STREAM_DECODER_READ_STATUS_CONTINUE;
2901 		case FLAC__OGG_DECODER_ASPECT_READ_STATUS_END_OF_STREAM:
2902 			return FLAC__STREAM_DECODER_READ_STATUS_END_OF_STREAM;
2903 		case FLAC__OGG_DECODER_ASPECT_READ_STATUS_NOT_FLAC:
2904 		case FLAC__OGG_DECODER_ASPECT_READ_STATUS_UNSUPPORTED_MAPPING_VERSION:
2905 		case FLAC__OGG_DECODER_ASPECT_READ_STATUS_ABORT:
2906 		case FLAC__OGG_DECODER_ASPECT_READ_STATUS_ERROR:
2907 		case FLAC__OGG_DECODER_ASPECT_READ_STATUS_MEMORY_ALLOCATION_ERROR:
2908 			return FLAC__STREAM_DECODER_READ_STATUS_ABORT;
2909 		default:
2910 			FLAC__ASSERT(0);
2911 			/* double protection */
2912 			return FLAC__STREAM_DECODER_READ_STATUS_ABORT;
2913 	}
2914 }
2915 
read_callback_proxy_(const void * void_decoder,FLAC__byte buffer[],size_t * bytes,void * client_data)2916 FLAC__OggDecoderAspectReadStatus read_callback_proxy_(const void *void_decoder, FLAC__byte buffer[], size_t *bytes, void *client_data)
2917 {
2918 	FLAC__StreamDecoder *decoder = (FLAC__StreamDecoder*)void_decoder;
2919 
2920 	switch(decoder->private_->read_callback(decoder, buffer, bytes, client_data)) {
2921 		case FLAC__STREAM_DECODER_READ_STATUS_CONTINUE:
2922 			return FLAC__OGG_DECODER_ASPECT_READ_STATUS_OK;
2923 		case FLAC__STREAM_DECODER_READ_STATUS_END_OF_STREAM:
2924 			return FLAC__OGG_DECODER_ASPECT_READ_STATUS_END_OF_STREAM;
2925 		case FLAC__STREAM_DECODER_READ_STATUS_ABORT:
2926 			return FLAC__OGG_DECODER_ASPECT_READ_STATUS_ABORT;
2927 		default:
2928 			/* double protection: */
2929 			FLAC__ASSERT(0);
2930 			return FLAC__OGG_DECODER_ASPECT_READ_STATUS_ABORT;
2931 	}
2932 }
2933 #endif
2934 
write_audio_frame_to_client_(FLAC__StreamDecoder * decoder,const FLAC__Frame * frame,const FLAC__int32 * const buffer[])2935 FLAC__StreamDecoderWriteStatus write_audio_frame_to_client_(FLAC__StreamDecoder *decoder, const FLAC__Frame *frame, const FLAC__int32 * const buffer[])
2936 {
2937 	if(decoder->private_->is_seeking) {
2938 		FLAC__uint64 this_frame_sample = frame->header.number.sample_number;
2939 		FLAC__uint64 next_frame_sample = this_frame_sample + (FLAC__uint64)frame->header.blocksize;
2940 		FLAC__uint64 target_sample = decoder->private_->target_sample;
2941 
2942 		FLAC__ASSERT(frame->header.number_type == FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER);
2943 
2944 #if FLAC__HAS_OGG
2945 		decoder->private_->got_a_frame = true;
2946 #endif
2947 		decoder->private_->last_frame = *frame; /* save the frame */
2948 		if(this_frame_sample <= target_sample && target_sample < next_frame_sample) { /* we hit our target frame */
2949 			unsigned delta = (unsigned)(target_sample - this_frame_sample);
2950 			/* kick out of seek mode */
2951 			decoder->private_->is_seeking = false;
2952 			/* shift out the samples before target_sample */
2953 			if(delta > 0) {
2954 				unsigned channel;
2955 				const FLAC__int32 *newbuffer[FLAC__MAX_CHANNELS];
2956 				for(channel = 0; channel < frame->header.channels; channel++)
2957 					newbuffer[channel] = buffer[channel] + delta;
2958 				decoder->private_->last_frame.header.blocksize -= delta;
2959 				decoder->private_->last_frame.header.number.sample_number += (FLAC__uint64)delta;
2960 				/* write the relevant samples */
2961 				return decoder->private_->write_callback(decoder, &decoder->private_->last_frame, newbuffer, decoder->private_->client_data);
2962 			}
2963 			else {
2964 				/* write the relevant samples */
2965 				return decoder->private_->write_callback(decoder, frame, buffer, decoder->private_->client_data);
2966 			}
2967 		}
2968 		else {
2969 			return FLAC__STREAM_DECODER_WRITE_STATUS_CONTINUE;
2970 		}
2971 	}
2972 	else {
2973 		/*
2974 		 * If we never got STREAMINFO, turn off MD5 checking to save
2975 		 * cycles since we don't have a sum to compare to anyway
2976 		 */
2977 		if(!decoder->private_->has_stream_info)
2978 			decoder->private_->do_md5_checking = false;
2979 		if(decoder->private_->do_md5_checking) {
2980 			if(!FLAC__MD5Accumulate(&decoder->private_->md5context, buffer, frame->header.channels, frame->header.blocksize, (frame->header.bits_per_sample+7) / 8))
2981 				return FLAC__STREAM_DECODER_WRITE_STATUS_ABORT;
2982 		}
2983 		return decoder->private_->write_callback(decoder, frame, buffer, decoder->private_->client_data);
2984 	}
2985 }
2986 
send_error_to_client_(const FLAC__StreamDecoder * decoder,FLAC__StreamDecoderErrorStatus status)2987 void send_error_to_client_(const FLAC__StreamDecoder *decoder, FLAC__StreamDecoderErrorStatus status)
2988 {
2989 	if(!decoder->private_->is_seeking)
2990 		decoder->private_->error_callback(decoder, status, decoder->private_->client_data);
2991 	else if(status == FLAC__STREAM_DECODER_ERROR_STATUS_UNPARSEABLE_STREAM)
2992 		decoder->private_->unparseable_frame_count++;
2993 }
2994 
seek_to_absolute_sample_(FLAC__StreamDecoder * decoder,FLAC__uint64 stream_length,FLAC__uint64 target_sample)2995 FLAC__bool seek_to_absolute_sample_(FLAC__StreamDecoder *decoder, FLAC__uint64 stream_length, FLAC__uint64 target_sample)
2996 {
2997 	FLAC__uint64 first_frame_offset = decoder->private_->first_frame_offset, lower_bound, upper_bound, lower_bound_sample, upper_bound_sample, this_frame_sample;
2998 	FLAC__int64 pos = -1;
2999 	int i;
3000 	unsigned approx_bytes_per_frame;
3001 	FLAC__bool first_seek = true;
3002 	const FLAC__uint64 total_samples = FLAC__stream_decoder_get_total_samples(decoder);
3003 	const unsigned min_blocksize = decoder->private_->stream_info.data.stream_info.min_blocksize;
3004 	const unsigned max_blocksize = decoder->private_->stream_info.data.stream_info.max_blocksize;
3005 	const unsigned max_framesize = decoder->private_->stream_info.data.stream_info.max_framesize;
3006 	const unsigned min_framesize = decoder->private_->stream_info.data.stream_info.min_framesize;
3007 	/* take these from the current frame in case they've changed mid-stream */
3008 	unsigned channels = FLAC__stream_decoder_get_channels(decoder);
3009 	unsigned bps = FLAC__stream_decoder_get_bits_per_sample(decoder);
3010 	const FLAC__StreamMetadata_SeekTable *seek_table = decoder->private_->has_seek_table? &decoder->private_->seek_table.data.seek_table : 0;
3011 
3012 	/* use values from stream info if we didn't decode a frame */
3013 	if(channels == 0)
3014 		channels = decoder->private_->stream_info.data.stream_info.channels;
3015 	if(bps == 0)
3016 		bps = decoder->private_->stream_info.data.stream_info.bits_per_sample;
3017 
3018 	/* we are just guessing here */
3019 	if(max_framesize > 0)
3020 		approx_bytes_per_frame = (max_framesize + min_framesize) / 2 + 1;
3021 	/*
3022 	 * Check if it's a known fixed-blocksize stream.  Note that though
3023 	 * the spec doesn't allow zeroes in the STREAMINFO block, we may
3024 	 * never get a STREAMINFO block when decoding so the value of
3025 	 * min_blocksize might be zero.
3026 	 */
3027 	else if(min_blocksize == max_blocksize && min_blocksize > 0) {
3028 		/* note there are no () around 'bps/8' to keep precision up since it's an integer calulation */
3029 		approx_bytes_per_frame = min_blocksize * channels * bps/8 + 64;
3030 	}
3031 	else
3032 		approx_bytes_per_frame = 4096 * channels * bps/8 + 64;
3033 
3034 	/*
3035 	 * First, we set an upper and lower bound on where in the
3036 	 * stream we will search.  For now we assume the worst case
3037 	 * scenario, which is our best guess at the beginning of
3038 	 * the first frame and end of the stream.
3039 	 */
3040 	lower_bound = first_frame_offset;
3041 	lower_bound_sample = 0;
3042 	upper_bound = stream_length;
3043 	upper_bound_sample = total_samples > 0 ? total_samples : target_sample /*estimate it*/;
3044 
3045 	/*
3046 	 * Now we refine the bounds if we have a seektable with
3047 	 * suitable points.  Note that according to the spec they
3048 	 * must be ordered by ascending sample number.
3049 	 *
3050 	 * Note: to protect against invalid seek tables we will ignore points
3051 	 * that have frame_samples==0 or sample_number>=total_samples
3052 	 */
3053 	if(seek_table) {
3054 		FLAC__uint64 new_lower_bound = lower_bound;
3055 		FLAC__uint64 new_upper_bound = upper_bound;
3056 		FLAC__uint64 new_lower_bound_sample = lower_bound_sample;
3057 		FLAC__uint64 new_upper_bound_sample = upper_bound_sample;
3058 
3059 		/* find the closest seek point <= target_sample, if it exists */
3060 		for(i = (int)seek_table->num_points - 1; i >= 0; i--) {
3061 			if(
3062 				seek_table->points[i].sample_number != FLAC__STREAM_METADATA_SEEKPOINT_PLACEHOLDER &&
3063 				seek_table->points[i].frame_samples > 0 && /* defense against bad seekpoints */
3064 				(total_samples <= 0 || seek_table->points[i].sample_number < total_samples) && /* defense against bad seekpoints */
3065 				seek_table->points[i].sample_number <= target_sample
3066 			)
3067 				break;
3068 		}
3069 		if(i >= 0) { /* i.e. we found a suitable seek point... */
3070 			new_lower_bound = first_frame_offset + seek_table->points[i].stream_offset;
3071 			new_lower_bound_sample = seek_table->points[i].sample_number;
3072 		}
3073 
3074 		/* find the closest seek point > target_sample, if it exists */
3075 		for(i = 0; i < (int)seek_table->num_points; i++) {
3076 			if(
3077 				seek_table->points[i].sample_number != FLAC__STREAM_METADATA_SEEKPOINT_PLACEHOLDER &&
3078 				seek_table->points[i].frame_samples > 0 && /* defense against bad seekpoints */
3079 				(total_samples <= 0 || seek_table->points[i].sample_number < total_samples) && /* defense against bad seekpoints */
3080 				seek_table->points[i].sample_number > target_sample
3081 			)
3082 				break;
3083 		}
3084 		if(i < (int)seek_table->num_points) { /* i.e. we found a suitable seek point... */
3085 			new_upper_bound = first_frame_offset + seek_table->points[i].stream_offset;
3086 			new_upper_bound_sample = seek_table->points[i].sample_number;
3087 		}
3088 		/* final protection against unsorted seek tables; keep original values if bogus */
3089 		if(new_upper_bound >= new_lower_bound) {
3090 			lower_bound = new_lower_bound;
3091 			upper_bound = new_upper_bound;
3092 			lower_bound_sample = new_lower_bound_sample;
3093 			upper_bound_sample = new_upper_bound_sample;
3094 		}
3095 	}
3096 
3097 	FLAC__ASSERT(upper_bound_sample >= lower_bound_sample);
3098 	/* there are 2 insidious ways that the following equality occurs, which
3099 	 * we need to fix:
3100 	 *  1) total_samples is 0 (unknown) and target_sample is 0
3101 	 *  2) total_samples is 0 (unknown) and target_sample happens to be
3102 	 *     exactly equal to the last seek point in the seek table; this
3103 	 *     means there is no seek point above it, and upper_bound_samples
3104 	 *     remains equal to the estimate (of target_samples) we made above
3105 	 * in either case it does not hurt to move upper_bound_sample up by 1
3106 	 */
3107 	if(upper_bound_sample == lower_bound_sample)
3108 		upper_bound_sample++;
3109 
3110 	decoder->private_->target_sample = target_sample;
3111 	while(1) {
3112 		/* check if the bounds are still ok */
3113 		if (lower_bound_sample >= upper_bound_sample || lower_bound > upper_bound) {
3114 			decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3115 			return false;
3116 		}
3117 #ifndef FLAC__INTEGER_ONLY_LIBRARY
3118 		pos = (FLAC__int64)lower_bound + (FLAC__int64)((FLAC__double)(target_sample - lower_bound_sample) / (FLAC__double)(upper_bound_sample - lower_bound_sample) * (FLAC__double)(upper_bound - lower_bound)) - approx_bytes_per_frame;
3119 #else
3120 		/* a little less accurate: */
3121 		if(upper_bound - lower_bound < 0xffffffff)
3122 			pos = (FLAC__int64)lower_bound + (FLAC__int64)(((target_sample - lower_bound_sample) * (upper_bound - lower_bound)) / (upper_bound_sample - lower_bound_sample)) - approx_bytes_per_frame;
3123 		else /* @@@ WATCHOUT, ~2TB limit */
3124 			pos = (FLAC__int64)lower_bound + (FLAC__int64)((((target_sample - lower_bound_sample)>>8) * ((upper_bound - lower_bound)>>8)) / ((upper_bound_sample - lower_bound_sample)>>16)) - approx_bytes_per_frame;
3125 #endif
3126 		if(pos >= (FLAC__int64)upper_bound)
3127 			pos = (FLAC__int64)upper_bound - 1;
3128 		if(pos < (FLAC__int64)lower_bound)
3129 			pos = (FLAC__int64)lower_bound;
3130 		if(decoder->private_->seek_callback(decoder, (FLAC__uint64)pos, decoder->private_->client_data) != FLAC__STREAM_DECODER_SEEK_STATUS_OK) {
3131 			decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3132 			return false;
3133 		}
3134 		if(!FLAC__stream_decoder_flush(decoder)) {
3135 			/* above call sets the state for us */
3136 			return false;
3137 		}
3138 		/* Now we need to get a frame.  First we need to reset our
3139 		 * unparseable_frame_count; if we get too many unparseable
3140 		 * frames in a row, the read callback will return
3141 		 * FLAC__STREAM_DECODER_READ_STATUS_ABORT, causing
3142 		 * FLAC__stream_decoder_process_single() to return false.
3143 		 */
3144 		decoder->private_->unparseable_frame_count = 0;
3145 		if(!FLAC__stream_decoder_process_single(decoder)) {
3146 			decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3147 			return false;
3148 		}
3149 		/* our write callback will change the state when it gets to the target frame */
3150 		/* actually, we could have got_a_frame if our decoder is at FLAC__STREAM_DECODER_END_OF_STREAM so we need to check for that also */
3151 #if 0
3152 		/*@@@@@@ used to be the following; not clear if the check for end of stream is needed anymore */
3153 		if(decoder->protected_->state != FLAC__SEEKABLE_STREAM_DECODER_SEEKING && decoder->protected_->state != FLAC__STREAM_DECODER_END_OF_STREAM)
3154 			break;
3155 #endif
3156 		if(!decoder->private_->is_seeking)
3157 			break;
3158 
3159 		FLAC__ASSERT(decoder->private_->last_frame.header.number_type == FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER);
3160 		this_frame_sample = decoder->private_->last_frame.header.number.sample_number;
3161 
3162 		if (0 == decoder->private_->samples_decoded || (this_frame_sample + decoder->private_->last_frame.header.blocksize >= upper_bound_sample && !first_seek)) {
3163 			if (pos == (FLAC__int64)lower_bound) {
3164 				/* can't move back any more than the first frame, something is fatally wrong */
3165 				decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3166 				return false;
3167 			}
3168 			/* our last move backwards wasn't big enough, try again */
3169 			approx_bytes_per_frame = approx_bytes_per_frame? approx_bytes_per_frame * 2 : 16;
3170 			continue;
3171 		}
3172 		/* allow one seek over upper bound, so we can get a correct upper_bound_sample for streams with unknown total_samples */
3173 		first_seek = false;
3174 
3175 		/* make sure we are not seeking in corrupted stream */
3176 		if (this_frame_sample < lower_bound_sample) {
3177 			decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3178 			return false;
3179 		}
3180 
3181 		/* we need to narrow the search */
3182 		if(target_sample < this_frame_sample) {
3183 			upper_bound_sample = this_frame_sample + decoder->private_->last_frame.header.blocksize;
3184 /*@@@@@@ what will decode position be if at end of stream? */
3185 			if(!FLAC__stream_decoder_get_decode_position(decoder, &upper_bound)) {
3186 				decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3187 				return false;
3188 			}
3189 			approx_bytes_per_frame = (unsigned)(2 * (upper_bound - pos) / 3 + 16);
3190 		}
3191 		else { /* target_sample >= this_frame_sample + this frame's blocksize */
3192 			lower_bound_sample = this_frame_sample + decoder->private_->last_frame.header.blocksize;
3193 			if(!FLAC__stream_decoder_get_decode_position(decoder, &lower_bound)) {
3194 				decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3195 				return false;
3196 			}
3197 			approx_bytes_per_frame = (unsigned)(2 * (lower_bound - pos) / 3 + 16);
3198 		}
3199 	}
3200 
3201 	return true;
3202 }
3203 
3204 #if FLAC__HAS_OGG
seek_to_absolute_sample_ogg_(FLAC__StreamDecoder * decoder,FLAC__uint64 stream_length,FLAC__uint64 target_sample)3205 FLAC__bool seek_to_absolute_sample_ogg_(FLAC__StreamDecoder *decoder, FLAC__uint64 stream_length, FLAC__uint64 target_sample)
3206 {
3207 	FLAC__uint64 left_pos = 0, right_pos = stream_length;
3208 	FLAC__uint64 left_sample = 0, right_sample = FLAC__stream_decoder_get_total_samples(decoder);
3209 	FLAC__uint64 this_frame_sample = (FLAC__uint64)0 - 1;
3210 	FLAC__uint64 pos = 0; /* only initialized to avoid compiler warning */
3211 	FLAC__bool did_a_seek;
3212 	unsigned iteration = 0;
3213 
3214 	/* In the first iterations, we will calculate the target byte position
3215 	 * by the distance from the target sample to left_sample and
3216 	 * right_sample (let's call it "proportional search").  After that, we
3217 	 * will switch to binary search.
3218 	 */
3219 	unsigned BINARY_SEARCH_AFTER_ITERATION = 2;
3220 
3221 	/* We will switch to a linear search once our current sample is less
3222 	 * than this number of samples ahead of the target sample
3223 	 */
3224 	static const FLAC__uint64 LINEAR_SEARCH_WITHIN_SAMPLES = FLAC__MAX_BLOCK_SIZE * 2;
3225 
3226 	/* If the total number of samples is unknown, use a large value, and
3227 	 * force binary search immediately.
3228 	 */
3229 	if(right_sample == 0) {
3230 		right_sample = (FLAC__uint64)(-1);
3231 		BINARY_SEARCH_AFTER_ITERATION = 0;
3232 	}
3233 
3234 	decoder->private_->target_sample = target_sample;
3235 	for( ; ; iteration++) {
3236 		if (iteration == 0 || this_frame_sample > target_sample || target_sample - this_frame_sample > LINEAR_SEARCH_WITHIN_SAMPLES) {
3237 			if (iteration >= BINARY_SEARCH_AFTER_ITERATION) {
3238 				pos = (right_pos + left_pos) / 2;
3239 			}
3240 			else {
3241 #ifndef FLAC__INTEGER_ONLY_LIBRARY
3242 				pos = (FLAC__uint64)((FLAC__double)(target_sample - left_sample) / (FLAC__double)(right_sample - left_sample) * (FLAC__double)(right_pos - left_pos));
3243 #else
3244 				/* a little less accurate: */
3245 				if ((target_sample-left_sample <= 0xffffffff) && (right_pos-left_pos <= 0xffffffff))
3246 					pos = (FLAC__int64)(((target_sample-left_sample) * (right_pos-left_pos)) / (right_sample-left_sample));
3247 				else /* @@@ WATCHOUT, ~2TB limit */
3248 					pos = (FLAC__int64)((((target_sample-left_sample)>>8) * ((right_pos-left_pos)>>8)) / ((right_sample-left_sample)>>16));
3249 #endif
3250 				/* @@@ TODO: might want to limit pos to some distance
3251 				 * before EOF, to make sure we land before the last frame,
3252 				 * thereby getting a this_frame_sample and so having a better
3253 				 * estimate.
3254 				 */
3255 			}
3256 
3257 			/* physical seek */
3258 			if(decoder->private_->seek_callback((FLAC__StreamDecoder*)decoder, (FLAC__uint64)pos, decoder->private_->client_data) != FLAC__STREAM_DECODER_SEEK_STATUS_OK) {
3259 				decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3260 				return false;
3261 			}
3262 			if(!FLAC__stream_decoder_flush(decoder)) {
3263 				/* above call sets the state for us */
3264 				return false;
3265 			}
3266 			did_a_seek = true;
3267 		}
3268 		else
3269 			did_a_seek = false;
3270 
3271 		decoder->private_->got_a_frame = false;
3272 		if(!FLAC__stream_decoder_process_single(decoder)) {
3273 			decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3274 			return false;
3275 		}
3276 		if(!decoder->private_->got_a_frame) {
3277 			if(did_a_seek) {
3278 				/* this can happen if we seek to a point after the last frame; we drop
3279 				 * to binary search right away in this case to avoid any wasted
3280 				 * iterations of proportional search.
3281 				 */
3282 				right_pos = pos;
3283 				BINARY_SEARCH_AFTER_ITERATION = 0;
3284 			}
3285 			else {
3286 				/* this can probably only happen if total_samples is unknown and the
3287 				 * target_sample is past the end of the stream
3288 				 */
3289 				decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3290 				return false;
3291 			}
3292 		}
3293 		/* our write callback will change the state when it gets to the target frame */
3294 		else if(!decoder->private_->is_seeking) {
3295 			break;
3296 		}
3297 		else {
3298 			this_frame_sample = decoder->private_->last_frame.header.number.sample_number;
3299 			FLAC__ASSERT(decoder->private_->last_frame.header.number_type == FLAC__FRAME_NUMBER_TYPE_SAMPLE_NUMBER);
3300 
3301 			if (did_a_seek) {
3302 				if (this_frame_sample <= target_sample) {
3303 					/* The 'equal' case should not happen, since
3304 					 * FLAC__stream_decoder_process_single()
3305 					 * should recognize that it has hit the
3306 					 * target sample and we would exit through
3307 					 * the 'break' above.
3308 					 */
3309 					FLAC__ASSERT(this_frame_sample != target_sample);
3310 
3311 					left_sample = this_frame_sample;
3312 					/* sanity check to avoid infinite loop */
3313 					if (left_pos == pos) {
3314 						decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3315 						return false;
3316 					}
3317 					left_pos = pos;
3318 				}
3319 				else if(this_frame_sample > target_sample) {
3320 					right_sample = this_frame_sample;
3321 					/* sanity check to avoid infinite loop */
3322 					if (right_pos == pos) {
3323 						decoder->protected_->state = FLAC__STREAM_DECODER_SEEK_ERROR;
3324 						return false;
3325 					}
3326 					right_pos = pos;
3327 				}
3328 			}
3329 		}
3330 	}
3331 
3332 	return true;
3333 }
3334 #endif
3335 
file_read_callback_(const FLAC__StreamDecoder * decoder,FLAC__byte buffer[],size_t * bytes,void * client_data)3336 FLAC__StreamDecoderReadStatus file_read_callback_(const FLAC__StreamDecoder *decoder, FLAC__byte buffer[], size_t *bytes, void *client_data)
3337 {
3338 	(void)client_data;
3339 
3340 	if(*bytes > 0) {
3341 		*bytes = fread(buffer, sizeof(FLAC__byte), *bytes, decoder->private_->file);
3342 		if(ferror(decoder->private_->file))
3343 			return FLAC__STREAM_DECODER_READ_STATUS_ABORT;
3344 		else if(*bytes == 0)
3345 			return FLAC__STREAM_DECODER_READ_STATUS_END_OF_STREAM;
3346 		else
3347 			return FLAC__STREAM_DECODER_READ_STATUS_CONTINUE;
3348 	}
3349 	else
3350 		return FLAC__STREAM_DECODER_READ_STATUS_ABORT; /* abort to avoid a deadlock */
3351 }
3352 
file_seek_callback_(const FLAC__StreamDecoder * decoder,FLAC__uint64 absolute_byte_offset,void * client_data)3353 FLAC__StreamDecoderSeekStatus file_seek_callback_(const FLAC__StreamDecoder *decoder, FLAC__uint64 absolute_byte_offset, void *client_data)
3354 {
3355 	(void)client_data;
3356 
3357 	if(decoder->private_->file == stdin)
3358 		return FLAC__STREAM_DECODER_SEEK_STATUS_UNSUPPORTED;
3359 	else if(fseeko(decoder->private_->file, (FLAC__off_t)absolute_byte_offset, SEEK_SET) < 0)
3360 		return FLAC__STREAM_DECODER_SEEK_STATUS_ERROR;
3361 	else
3362 		return FLAC__STREAM_DECODER_SEEK_STATUS_OK;
3363 }
3364 
file_tell_callback_(const FLAC__StreamDecoder * decoder,FLAC__uint64 * absolute_byte_offset,void * client_data)3365 FLAC__StreamDecoderTellStatus file_tell_callback_(const FLAC__StreamDecoder *decoder, FLAC__uint64 *absolute_byte_offset, void *client_data)
3366 {
3367 	FLAC__off_t pos;
3368 	(void)client_data;
3369 
3370 	if(decoder->private_->file == stdin)
3371 		return FLAC__STREAM_DECODER_TELL_STATUS_UNSUPPORTED;
3372 	else if((pos = ftello(decoder->private_->file)) < 0)
3373 		return FLAC__STREAM_DECODER_TELL_STATUS_ERROR;
3374 	else {
3375 		*absolute_byte_offset = (FLAC__uint64)pos;
3376 		return FLAC__STREAM_DECODER_TELL_STATUS_OK;
3377 	}
3378 }
3379 
file_length_callback_(const FLAC__StreamDecoder * decoder,FLAC__uint64 * stream_length,void * client_data)3380 FLAC__StreamDecoderLengthStatus file_length_callback_(const FLAC__StreamDecoder *decoder, FLAC__uint64 *stream_length, void *client_data)
3381 {
3382 	struct flac_stat_s filestats;
3383 	(void)client_data;
3384 
3385 	if(decoder->private_->file == stdin)
3386 		return FLAC__STREAM_DECODER_LENGTH_STATUS_UNSUPPORTED;
3387 	else if(flac_fstat(fileno(decoder->private_->file), &filestats) != 0)
3388 		return FLAC__STREAM_DECODER_LENGTH_STATUS_ERROR;
3389 	else {
3390 		*stream_length = (FLAC__uint64)filestats.st_size;
3391 		return FLAC__STREAM_DECODER_LENGTH_STATUS_OK;
3392 	}
3393 }
3394 
file_eof_callback_(const FLAC__StreamDecoder * decoder,void * client_data)3395 FLAC__bool file_eof_callback_(const FLAC__StreamDecoder *decoder, void *client_data)
3396 {
3397 	(void)client_data;
3398 
3399 	return feof(decoder->private_->file)? true : false;
3400 }
3401