diff options
author | Michael Niedermayer <michaelni@gmx.at> | 2012-01-09 00:32:38 +0100 |
---|---|---|
committer | Michael Niedermayer <michaelni@gmx.at> | 2012-01-09 00:32:38 +0100 |
commit | 9273efac1b9ef79ca0ea5673d8088c501e19d70f (patch) | |
tree | 844f8c736972c6f2e5669fc5810acb46c769f65f /libavcodec/bfi.c | |
parent | b1435626392bd05bd3e79a56d9f64c089654afda (diff) | |
parent | ccc27e2139336b66cdec3bb73a2cc7e60ef7e599 (diff) | |
download | ffmpeg-9273efac1b9ef79ca0ea5673d8088c501e19d70f.tar.gz |
Merge remote-tracking branch 'qatar/master'
* qatar/master:
bfi: Use bytestream2 functions to prevent buffer overreads.
dpcm: Fix invalid writes
utvideo: frame multithreading.
vorbis: An additional defense in the Vorbis codec.
vorbisdec: Fix decoding bug with channel handling
Conflicts:
libavcodec/dpcm.c
Merged-by: Michael Niedermayer <michaelni@gmx.at>
Diffstat (limited to 'libavcodec/bfi.c')
-rw-r--r-- | libavcodec/bfi.c | 26 |
1 files changed, 14 insertions, 12 deletions
diff --git a/libavcodec/bfi.c b/libavcodec/bfi.c index 2aa1a4cb65..0a7324a297 100644 --- a/libavcodec/bfi.c +++ b/libavcodec/bfi.c @@ -49,7 +49,7 @@ static av_cold int bfi_decode_init(AVCodecContext *avctx) static int bfi_decode_frame(AVCodecContext *avctx, void *data, int *data_size, AVPacket *avpkt) { - const uint8_t *buf = avpkt->data, *buf_end = avpkt->data + avpkt->size; + GetByteContext g; int buf_size = avpkt->size; BFIContext *bfi = avctx->priv_data; uint8_t *dst = bfi->dst; @@ -68,6 +68,8 @@ static int bfi_decode_frame(AVCodecContext *avctx, void *data, return -1; } + bytestream2_init(&g, avpkt->data, buf_size); + /* Set frame parameters and palette, if necessary */ if (!avctx->frame_number) { bfi->frame.pict_type = AV_PICTURE_TYPE_I; @@ -96,15 +98,15 @@ static int bfi_decode_frame(AVCodecContext *avctx, void *data, memcpy(bfi->frame.data[1], bfi->pal, sizeof(bfi->pal)); } - buf += 4; // Unpacked size, not required. + bytestream2_skip(&g, 4); // Unpacked size, not required. while (dst != frame_end) { static const uint8_t lentab[4] = { 0, 2, 0, 1 }; - unsigned int byte = *buf++, av_uninit(offset); + unsigned int byte = bytestream2_get_byte(&g), av_uninit(offset); unsigned int code = byte >> 6; unsigned int length = byte & ~0xC0; - if (buf >= buf_end) { + if (!bytestream2_get_bytes_left(&g)) { av_log(avctx, AV_LOG_ERROR, "Input resolution larger than actual frame.\n"); return -1; @@ -113,16 +115,16 @@ static int bfi_decode_frame(AVCodecContext *avctx, void *data, /* Get length and offset(if required) */ if (length == 0) { if (code == 1) { - length = bytestream_get_byte(&buf); - offset = bytestream_get_le16(&buf); + length = bytestream2_get_byte(&g); + offset = bytestream2_get_le16(&g); } else { - length = bytestream_get_le16(&buf); + length = bytestream2_get_le16(&g); if (code == 2 && length == 0) break; } } else { if (code == 1) - offset = bytestream_get_byte(&buf); + offset = bytestream2_get_byte(&g); } /* Do boundary check */ @@ -132,11 +134,11 @@ static int bfi_decode_frame(AVCodecContext *avctx, void *data, switch (code) { case 0: //Normal Chain - if (length >= buf_end - buf) { + if (length >= bytestream2_get_bytes_left(&g)) { av_log(avctx, AV_LOG_ERROR, "Frame larger than buffer.\n"); return -1; } - bytestream_get_buffer(&buf, dst, length); + bytestream2_get_buffer(&g, dst, length); dst += length; break; @@ -154,8 +156,8 @@ static int bfi_decode_frame(AVCodecContext *avctx, void *data, break; case 3: //Fill Chain - colour1 = bytestream_get_byte(&buf); - colour2 = bytestream_get_byte(&buf); + colour1 = bytestream2_get_byte(&g); + colour2 = bytestream2_get_byte(&g); while (length--) { *dst++ = colour1; *dst++ = colour2; |