Commit 45013b36 authored by James Almer's avatar James Almer Committed by Michael Niedermayer

lavc/flacdec: Propagate and return meaningful error codes

Signed-off-by: 's avatarJames Almer <jamrial@gmail.com>
Reviewed-by: 's avatarPaul B Mahol <onemda@gmail.com>
Signed-off-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parent f248b8a9
...@@ -101,7 +101,7 @@ static av_cold int flac_decode_init(AVCodecContext *avctx) ...@@ -101,7 +101,7 @@ static av_cold int flac_decode_init(AVCodecContext *avctx)
return 0; return 0;
if (!avpriv_flac_is_extradata_valid(avctx, &format, &streaminfo)) if (!avpriv_flac_is_extradata_valid(avctx, &format, &streaminfo))
return -1; return AVERROR_INVALIDDATA;
/* initialize based on the demuxer-supplied streamdata header */ /* initialize based on the demuxer-supplied streamdata header */
avpriv_flac_parse_streaminfo(avctx, (FLACStreaminfo *)s, streaminfo); avpriv_flac_parse_streaminfo(avctx, (FLACStreaminfo *)s, streaminfo);
...@@ -212,7 +212,7 @@ static int decode_residuals(FLACContext *s, int32_t *decoded, int pred_order) ...@@ -212,7 +212,7 @@ static int decode_residuals(FLACContext *s, int32_t *decoded, int pred_order)
if (method_type > 1) { if (method_type > 1) {
av_log(s->avctx, AV_LOG_ERROR, "illegal residual coding method %d\n", av_log(s->avctx, AV_LOG_ERROR, "illegal residual coding method %d\n",
method_type); method_type);
return -1; return AVERROR_INVALIDDATA;
} }
rice_order = get_bits(&s->gb, 4); rice_order = get_bits(&s->gb, 4);
...@@ -221,7 +221,7 @@ static int decode_residuals(FLACContext *s, int32_t *decoded, int pred_order) ...@@ -221,7 +221,7 @@ static int decode_residuals(FLACContext *s, int32_t *decoded, int pred_order)
if (pred_order > samples) { if (pred_order > samples) {
av_log(s->avctx, AV_LOG_ERROR, "invalid predictor order: %i > %i\n", av_log(s->avctx, AV_LOG_ERROR, "invalid predictor order: %i > %i\n",
pred_order, samples); pred_order, samples);
return -1; return AVERROR_INVALIDDATA;
} }
rice_bits = 4 + method_type; rice_bits = 4 + method_type;
...@@ -251,14 +251,15 @@ static int decode_subframe_fixed(FLACContext *s, int32_t *decoded, ...@@ -251,14 +251,15 @@ static int decode_subframe_fixed(FLACContext *s, int32_t *decoded,
{ {
const int blocksize = s->blocksize; const int blocksize = s->blocksize;
int av_uninit(a), av_uninit(b), av_uninit(c), av_uninit(d), i; int av_uninit(a), av_uninit(b), av_uninit(c), av_uninit(d), i;
int ret;
/* warm up samples */ /* warm up samples */
for (i = 0; i < pred_order; i++) { for (i = 0; i < pred_order; i++) {
decoded[i] = get_sbits_long(&s->gb, bps); decoded[i] = get_sbits_long(&s->gb, bps);
} }
if (decode_residuals(s, decoded, pred_order) < 0) if ((ret = decode_residuals(s, decoded, pred_order)) < 0)
return -1; return ret;
if (pred_order > 0) if (pred_order > 0)
a = decoded[pred_order-1]; a = decoded[pred_order-1];
...@@ -290,7 +291,7 @@ static int decode_subframe_fixed(FLACContext *s, int32_t *decoded, ...@@ -290,7 +291,7 @@ static int decode_subframe_fixed(FLACContext *s, int32_t *decoded,
break; break;
default: default:
av_log(s->avctx, AV_LOG_ERROR, "illegal pred order %d\n", pred_order); av_log(s->avctx, AV_LOG_ERROR, "illegal pred order %d\n", pred_order);
return -1; return AVERROR_INVALIDDATA;
} }
return 0; return 0;
...@@ -299,7 +300,7 @@ static int decode_subframe_fixed(FLACContext *s, int32_t *decoded, ...@@ -299,7 +300,7 @@ static int decode_subframe_fixed(FLACContext *s, int32_t *decoded,
static int decode_subframe_lpc(FLACContext *s, int32_t *decoded, int pred_order, static int decode_subframe_lpc(FLACContext *s, int32_t *decoded, int pred_order,
int bps) int bps)
{ {
int i; int i, ret;
int coeff_prec, qlevel; int coeff_prec, qlevel;
int coeffs[32]; int coeffs[32];
...@@ -311,21 +312,21 @@ static int decode_subframe_lpc(FLACContext *s, int32_t *decoded, int pred_order, ...@@ -311,21 +312,21 @@ static int decode_subframe_lpc(FLACContext *s, int32_t *decoded, int pred_order,
coeff_prec = get_bits(&s->gb, 4) + 1; coeff_prec = get_bits(&s->gb, 4) + 1;
if (coeff_prec == 16) { if (coeff_prec == 16) {
av_log(s->avctx, AV_LOG_ERROR, "invalid coeff precision\n"); av_log(s->avctx, AV_LOG_ERROR, "invalid coeff precision\n");
return -1; return AVERROR_INVALIDDATA;
} }
qlevel = get_sbits(&s->gb, 5); qlevel = get_sbits(&s->gb, 5);
if (qlevel < 0) { if (qlevel < 0) {
av_log(s->avctx, AV_LOG_ERROR, "qlevel %d not supported, maybe buggy stream\n", av_log(s->avctx, AV_LOG_ERROR, "qlevel %d not supported, maybe buggy stream\n",
qlevel); qlevel);
return -1; return AVERROR_INVALIDDATA;
} }
for (i = 0; i < pred_order; i++) { for (i = 0; i < pred_order; i++) {
coeffs[pred_order - i - 1] = get_sbits(&s->gb, coeff_prec); coeffs[pred_order - i - 1] = get_sbits(&s->gb, coeff_prec);
} }
if (decode_residuals(s, decoded, pred_order) < 0) if ((ret = decode_residuals(s, decoded, pred_order)) < 0)
return -1; return ret;
s->dsp.lpc(decoded, coeffs, pred_order, qlevel, s->blocksize); s->dsp.lpc(decoded, coeffs, pred_order, qlevel, s->blocksize);
...@@ -337,7 +338,7 @@ static inline int decode_subframe(FLACContext *s, int channel) ...@@ -337,7 +338,7 @@ static inline int decode_subframe(FLACContext *s, int channel)
int32_t *decoded = s->decoded[channel]; int32_t *decoded = s->decoded[channel];
int type, wasted = 0; int type, wasted = 0;
int bps = s->bps; int bps = s->bps;
int i, tmp; int i, tmp, ret;
if (channel == 0) { if (channel == 0) {
if (s->ch_mode == FLAC_CHMODE_RIGHT_SIDE) if (s->ch_mode == FLAC_CHMODE_RIGHT_SIDE)
...@@ -349,7 +350,7 @@ static inline int decode_subframe(FLACContext *s, int channel) ...@@ -349,7 +350,7 @@ static inline int decode_subframe(FLACContext *s, int channel)
if (get_bits1(&s->gb)) { if (get_bits1(&s->gb)) {
av_log(s->avctx, AV_LOG_ERROR, "invalid subframe padding\n"); av_log(s->avctx, AV_LOG_ERROR, "invalid subframe padding\n");
return -1; return AVERROR_INVALIDDATA;
} }
type = get_bits(&s->gb, 6); type = get_bits(&s->gb, 6);
...@@ -382,14 +383,14 @@ static inline int decode_subframe(FLACContext *s, int channel) ...@@ -382,14 +383,14 @@ static inline int decode_subframe(FLACContext *s, int channel)
for (i = 0; i < s->blocksize; i++) for (i = 0; i < s->blocksize; i++)
decoded[i] = get_sbits_long(&s->gb, bps); decoded[i] = get_sbits_long(&s->gb, bps);
} else if ((type >= 8) && (type <= 12)) { } else if ((type >= 8) && (type <= 12)) {
if (decode_subframe_fixed(s, decoded, type & ~0x8, bps) < 0) if ((ret = decode_subframe_fixed(s, decoded, type & ~0x8, bps)) < 0)
return -1; return ret;
} else if (type >= 32) { } else if (type >= 32) {
if (decode_subframe_lpc(s, decoded, (type & ~0x20)+1, bps) < 0) if ((ret = decode_subframe_lpc(s, decoded, (type & ~0x20)+1, bps)) < 0)
return -1; return ret;
} else { } else {
av_log(s->avctx, AV_LOG_ERROR, "invalid coding type\n"); av_log(s->avctx, AV_LOG_ERROR, "invalid coding type\n");
return -1; return AVERROR_INVALIDDATA;
} }
if (wasted) { if (wasted) {
...@@ -409,7 +410,7 @@ static int decode_frame(FLACContext *s) ...@@ -409,7 +410,7 @@ static int decode_frame(FLACContext *s)
if (ff_flac_decode_frame_header(s->avctx, gb, &fi, 0)) { if (ff_flac_decode_frame_header(s->avctx, gb, &fi, 0)) {
av_log(s->avctx, AV_LOG_ERROR, "invalid frame header\n"); av_log(s->avctx, AV_LOG_ERROR, "invalid frame header\n");
return -1; return AVERROR_INVALIDDATA;
} }
if (s->channels && fi.channels != s->channels && s->got_streaminfo) { if (s->channels && fi.channels != s->channels && s->got_streaminfo) {
...@@ -426,7 +427,7 @@ static int decode_frame(FLACContext *s) ...@@ -426,7 +427,7 @@ static int decode_frame(FLACContext *s)
if (!s->bps && !fi.bps) { if (!s->bps && !fi.bps) {
av_log(s->avctx, AV_LOG_ERROR, "bps not found in STREAMINFO or frame header\n"); av_log(s->avctx, AV_LOG_ERROR, "bps not found in STREAMINFO or frame header\n");
return -1; return AVERROR_INVALIDDATA;
} }
if (!fi.bps) { if (!fi.bps) {
fi.bps = s->bps; fi.bps = s->bps;
...@@ -446,14 +447,14 @@ static int decode_frame(FLACContext *s) ...@@ -446,14 +447,14 @@ static int decode_frame(FLACContext *s)
if (fi.blocksize > s->max_blocksize) { if (fi.blocksize > s->max_blocksize) {
av_log(s->avctx, AV_LOG_ERROR, "blocksize %d > %d\n", fi.blocksize, av_log(s->avctx, AV_LOG_ERROR, "blocksize %d > %d\n", fi.blocksize,
s->max_blocksize); s->max_blocksize);
return -1; return AVERROR_INVALIDDATA;
} }
s->blocksize = fi.blocksize; s->blocksize = fi.blocksize;
if (!s->samplerate && !fi.samplerate) { if (!s->samplerate && !fi.samplerate) {
av_log(s->avctx, AV_LOG_ERROR, "sample rate not found in STREAMINFO" av_log(s->avctx, AV_LOG_ERROR, "sample rate not found in STREAMINFO"
" or frame header\n"); " or frame header\n");
return -1; return AVERROR_INVALIDDATA;
} }
if (fi.samplerate == 0) if (fi.samplerate == 0)
fi.samplerate = s->samplerate; fi.samplerate = s->samplerate;
...@@ -472,8 +473,8 @@ static int decode_frame(FLACContext *s) ...@@ -472,8 +473,8 @@ static int decode_frame(FLACContext *s)
/* subframes */ /* subframes */
for (i = 0; i < s->channels; i++) { for (i = 0; i < s->channels; i++) {
if (decode_subframe(s, i) < 0) if ((ret = decode_subframe(s, i)) < 0)
return -1; return ret;
} }
align_get_bits(gb); align_get_bits(gb);
...@@ -522,16 +523,16 @@ static int flac_decode_frame(AVCodecContext *avctx, void *data, ...@@ -522,16 +523,16 @@ static int flac_decode_frame(AVCodecContext *avctx, void *data,
if (AV_RB32(buf) == MKBETAG('f','L','a','C')) { if (AV_RB32(buf) == MKBETAG('f','L','a','C')) {
if (!s->got_streaminfo && parse_streaminfo(s, buf, buf_size)) { if (!s->got_streaminfo && parse_streaminfo(s, buf, buf_size)) {
av_log(s->avctx, AV_LOG_ERROR, "invalid header\n"); av_log(s->avctx, AV_LOG_ERROR, "invalid header\n");
return -1; return AVERROR_INVALIDDATA;
} }
return get_metadata_size(buf, buf_size); return get_metadata_size(buf, buf_size);
} }
/* decode frame */ /* decode frame */
init_get_bits(&s->gb, buf, buf_size*8); init_get_bits(&s->gb, buf, buf_size*8);
if (decode_frame(s) < 0) { if ((ret = decode_frame(s)) < 0) {
av_log(s->avctx, AV_LOG_ERROR, "decode_frame() failed\n"); av_log(s->avctx, AV_LOG_ERROR, "decode_frame() failed\n");
return -1; return ret;
} }
bytes_read = get_bits_count(&s->gb)/8; bytes_read = get_bits_count(&s->gb)/8;
...@@ -553,7 +554,7 @@ static int flac_decode_frame(AVCodecContext *avctx, void *data, ...@@ -553,7 +554,7 @@ static int flac_decode_frame(AVCodecContext *avctx, void *data,
if (bytes_read > buf_size) { if (bytes_read > buf_size) {
av_log(s->avctx, AV_LOG_ERROR, "overread: %d\n", bytes_read - buf_size); av_log(s->avctx, AV_LOG_ERROR, "overread: %d\n", bytes_read - buf_size);
return -1; return AVERROR_INVALIDDATA;
} }
if (bytes_read < buf_size) { if (bytes_read < buf_size) {
av_log(s->avctx, AV_LOG_DEBUG, "underread: %d orig size: %d\n", av_log(s->avctx, AV_LOG_DEBUG, "underread: %d orig size: %d\n",
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment