From b08a742526d05bcf300af820da5c79d2a05f9959 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Mon, 19 Dec 2011 19:23:56 +0100 Subject: siff: simplify code by using ffio_limit() Signed-off-by: Michael Niedermayer --- libavformat/siff.c | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) (limited to 'libavformat/siff.c') diff --git a/libavformat/siff.c b/libavformat/siff.c index 51dc694e4f..1a3f0a4e47 100644 --- a/libavformat/siff.c +++ b/libavformat/siff.c @@ -22,6 +22,7 @@ #include "libavutil/intreadwrite.h" #include "avformat.h" #include "internal.h" +#include "avio_internal.h" enum SIFFTags{ TAG_SIFF = MKTAG('S', 'I', 'F', 'F'), @@ -201,18 +202,16 @@ static int siff_read_packet(AVFormatContext *s, AVPacket *pkt) } if (!c->curstrm){ - int64_t fsize= avio_size(s->pb); - size = c->pktsize - c->sndsize; - if(fsize>0) - size= FFMIN(size, fsize - avio_tell(s->pb) + c->gmcsize + 3); - if(size < 2 + c->gmcsize || c->pktsize < c->sndsize) + size = c->pktsize - c->sndsize - c->gmcsize - 2; + size = ffio_limit(s->pb, size); + if(size < 0 || c->pktsize < c->sndsize) return AVERROR_INVALIDDATA; - if (av_new_packet(pkt, size) < 0) + if (av_new_packet(pkt, size + c->gmcsize + 2) < 0) return AVERROR(ENOMEM); AV_WL16(pkt->data, c->flags); if (c->gmcsize) memcpy(pkt->data + 2, c->gmc, c->gmcsize); - avio_read(s->pb, pkt->data + 2 + c->gmcsize, size - c->gmcsize - 2); + avio_read(s->pb, pkt->data + 2 + c->gmcsize, size); pkt->stream_index = 0; c->curstrm = -1; }else{ -- cgit v1.2.3