diff options
author | Laurent Aimar <fenrir@videolan.org> | 2011-10-02 00:48:12 +0000 |
---|---|---|
committer | Janne Grunau <janne-libav@jannau.net> | 2011-10-10 21:37:15 +0200 |
commit | dc64f203a62083c3d5f81e8201018279c29581af (patch) | |
tree | 7160b67a9dcc06316829f7b092959fa05bd0dea5 | |
parent | 04a845caa7cdcdd1457f8c0dde52a7b2085ed92f (diff) | |
download | ffmpeg-dc64f203a62083c3d5f81e8201018279c29581af.tar.gz |
ptx: check for out of bound reads
Signed-off-by: Janne Grunau <janne-libav@jannau.net>
-rw-r--r-- | libavcodec/ptx.c | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/libavcodec/ptx.c b/libavcodec/ptx.c index 5dadaf6ece..eee0d58994 100644 --- a/libavcodec/ptx.c +++ b/libavcodec/ptx.c @@ -39,12 +39,15 @@ static av_cold int ptx_init(AVCodecContext *avctx) { static int ptx_decode_frame(AVCodecContext *avctx, void *data, int *data_size, AVPacket *avpkt) { const uint8_t *buf = avpkt->data; + const uint8_t *buf_end = avpkt->data + avpkt->size; PTXContext * const s = avctx->priv_data; AVFrame *picture = data; AVFrame * const p = &s->picture; unsigned int offset, w, h, y, stride, bytes_per_pixel; uint8_t *ptr; + if (buf_end - buf < 14) + return AVERROR_INVALIDDATA; offset = AV_RL16(buf); w = AV_RL16(buf+8); h = AV_RL16(buf+10); @@ -57,6 +60,8 @@ static int ptx_decode_frame(AVCodecContext *avctx, void *data, int *data_size, avctx->pix_fmt = PIX_FMT_RGB555; + if (buf_end - buf < offset) + return AVERROR_INVALIDDATA; if (offset != 0x2c) av_log_ask_for_sample(avctx, "offset != 0x2c\n"); @@ -80,6 +85,8 @@ static int ptx_decode_frame(AVCodecContext *avctx, void *data, int *data_size, stride = p->linesize[0]; for (y=0; y<h; y++) { + if (buf_end - buf < w * bytes_per_pixel) + break; #if HAVE_BIGENDIAN unsigned int x; for (x=0; x<w*bytes_per_pixel; x+=bytes_per_pixel) |