From 1cd89f276a0be52609904ea911a93cec78c92294 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Sun, 21 Aug 2005 15:44:59 +0000 Subject: segfault fix Originally committed as revision 4531 to svn://svn.ffmpeg.org/ffmpeg/trunk --- libavcodec/pnm.c | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'libavcodec') diff --git a/libavcodec/pnm.c b/libavcodec/pnm.c index 30a6570169..aae798bcaa 100644 --- a/libavcodec/pnm.c +++ b/libavcodec/pnm.c @@ -198,6 +198,8 @@ static int pnm_decode_frame(AVCodecContext *avctx, do_read: ptr = p->data[0]; linesize = p->linesize[0]; + if(s->bytestream + n*avctx->height > s->bytestream_end) + return -1; for(i = 0; i < avctx->height; i++) { memcpy(ptr, s->bytestream, n); s->bytestream += n; @@ -211,6 +213,8 @@ static int pnm_decode_frame(AVCodecContext *avctx, n = avctx->width; ptr = p->data[0]; linesize = p->linesize[0]; + if(s->bytestream + n*avctx->height*3/2 > s->bytestream_end) + return -1; for(i = 0; i < avctx->height; i++) { memcpy(ptr, s->bytestream, n); s->bytestream += n; @@ -233,6 +237,8 @@ static int pnm_decode_frame(AVCodecContext *avctx, case PIX_FMT_RGBA32: ptr = p->data[0]; linesize = p->linesize[0]; + if(s->bytestream + avctx->width*avctx->height*4 > s->bytestream_end) + return -1; for(i = 0; i < avctx->height; i++) { int j, r, g, b, a; -- cgit v1.2.3