[FFmpeg-cvslog] r21846 - trunk/libavcodec/iff.c

[pross]

Author: pross
Date: Tue Feb 16 10:33:59 2010
New Revision: 21846

Log:
Support <8-bit ILBM uncompressed bitmaps

Modified:
   trunk/libavcodec/iff.c

Modified: trunk/libavcodec/iff.c
==============================================================================
--- trunk/libavcodec/iff.c	Tue Feb 16 03:51:37 2010	(r21845)
+++ trunk/libavcodec/iff.c	Tue Feb 16 10:33:59 2010	(r21846)
@@ -114,6 +114,7 @@ static int decode_frame_ilbm(AVCodecCont
     IffContext *s = avctx->priv_data;
     const uint8_t *buf = avpkt->data;
     int buf_size = avpkt->size;
+    const uint8_t *buf_end = buf+buf_size;
     int y, plane;
 
     if (avctx->reget_buffer(avctx, &s->frame) < 0){
@@ -121,19 +122,14 @@ static int decode_frame_ilbm(AVCodecCont
         return -1;
     }
 
-    if (buf_size < avctx->width * avctx->height) {
-        av_log(avctx, AV_LOG_ERROR, "buffer underflow\n");
-        return -1;
-    }
-
     for(y = 0; y < avctx->height; y++ ) {
         uint8_t *row = &s->frame.data[0][ y*s->frame.linesize[0] ];
         memset(row, 0, avctx->pix_fmt == PIX_FMT_PAL8 ? avctx->width : (avctx->width * 4));
-        for (plane = 0; plane < avctx->bits_per_coded_sample; plane++) {
+        for (plane = 0; plane < avctx->bits_per_coded_sample && buf < buf_end; plane++) {
             if (avctx->pix_fmt == PIX_FMT_PAL8) {
-                decodeplane8(row, buf, s->planesize, avctx->bits_per_coded_sample, plane);
+                decodeplane8(row, buf, FFMIN(s->planesize, buf_end - buf), avctx->bits_per_coded_sample, plane);
             } else { // PIX_FMT_BGR32
-                decodeplane32(row, buf, s->planesize, avctx->bits_per_coded_sample, plane);
+                decodeplane32(row, buf, FFMIN(s->planesize, buf_end - buf), avctx->bits_per_coded_sample, plane);
             }
             buf += s->planesize;
         }