diff --git a/libavcodec/sgidec.c b/libavcodec/sgidec.c index 91b68eb9f7..112a18159b 100644 --- a/libavcodec/sgidec.c +++ b/libavcodec/sgidec.c @@ -48,13 +48,14 @@ static int expand_rle_row(SgiState *s, uint8_t *out_buf, { unsigned char pixel, count; unsigned char *orig = out_buf; + uint8_t *out_end = out_buf + len; - while (1) { + while (out_buf < out_end) { if (bytestream2_get_bytes_left(&s->g) < 1) return AVERROR_INVALIDDATA; pixel = bytestream2_get_byteu(&s->g); if (!(count = (pixel & 0x7f))) { - return (out_buf - orig) / pixelstride; + break; } /* Check for buffer overflow. */ @@ -77,6 +78,7 @@ static int expand_rle_row(SgiState *s, uint8_t *out_buf, } } } + return (out_buf - orig) / pixelstride; } /**