diff options
| author | Michael Niedermayer <michaelni@gmx.at> | 2012-01-26 16:51:01 +0100 |
|---|---|---|
| committer | Michael Niedermayer <michaelni@gmx.at> | 2012-01-26 17:05:57 +0100 |
| commit | ddf0c1d86ad8e1df5ab3265206aef493a1bdc813 (patch) | |
| tree | 7afaed43423a399882ae40ca981a559bedcdfc71 /libavcodec/diracdec.c | |
| parent | e2291ea1534d17306f685b8c8abc8585bbed87bf (diff) | |
diracdec: Check num_refs.
Fixes: CVE-2011-3950
Found-by: Mateusz "j00ru" Jurczyk and Gynvael Coldwind
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
Diffstat (limited to 'libavcodec/diracdec.c')
| -rw-r--r-- | libavcodec/diracdec.c | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/libavcodec/diracdec.c b/libavcodec/diracdec.c index e6b33e2580..03e7d0abde 100644 --- a/libavcodec/diracdec.c +++ b/libavcodec/diracdec.c @@ -1722,6 +1722,7 @@ static int dirac_decode_data_unit(AVCodecContext *avctx, const uint8_t *buf, int DiracContext *s = avctx->priv_data; DiracFrame *pic = NULL; int i, parse_code = buf[4]; + unsigned tmp; if (size < DATA_UNIT_HEADER_SIZE) return -1; @@ -1772,7 +1773,12 @@ static int dirac_decode_data_unit(AVCodecContext *avctx, const uint8_t *buf, int avcodec_get_frame_defaults(&pic->avframe); /* [DIRAC_STD] Defined in 9.6.1 ... */ - s->num_refs = parse_code & 0x03; /* [DIRAC_STD] num_refs() */ + tmp = parse_code & 0x03; /* [DIRAC_STD] num_refs() */ + if (tmp > 2) { + av_log(avctx, AV_LOG_ERROR, "num_refs of 3\n"); + return -1; + } + s->num_refs = tmp; s->is_arith = (parse_code & 0x48) == 0x08; /* [DIRAC_STD] using_ac() */ s->low_delay = (parse_code & 0x88) == 0x88; /* [DIRAC_STD] is_low_delay() */ pic->avframe.reference = (parse_code & 0x0C) == 0x0C; /* [DIRAC_STD] is_reference() */ |