Browse Source

yuv4mpeg: return proper error codes.

Fixes Bug 373.

CC:libav-stable@libav.org
tags/n1.1
Anton Khirnov 13 years ago
parent
commit
d3a72becc6
1 changed files with 15 additions and 7 deletions
  1. +15
    -7
      libavformat/yuv4mpeg.c

+ 15
- 7
libavformat/yuv4mpeg.c View File

@@ -367,7 +367,7 @@ static int yuv4_read_packet(AVFormatContext *s, AVPacket *pkt)
{ {
int i; int i;
char header[MAX_FRAME_HEADER+1]; char header[MAX_FRAME_HEADER+1];
int packet_size, width, height;
int packet_size, width, height, ret;
AVStream *st = s->streams[0]; AVStream *st = s->streams[0];
struct frame_attributes *s1 = s->priv_data; struct frame_attributes *s1 = s->priv_data;


@@ -378,20 +378,28 @@ static int yuv4_read_packet(AVFormatContext *s, AVPacket *pkt)
break; break;
} }
} }
if (i == MAX_FRAME_HEADER)
return -1;
if (s->pb->error)
return s->pb->error;
else if (s->pb->eof_reached)
return AVERROR_EOF;
else if (i == MAX_FRAME_HEADER)
return AVERROR_INVALIDDATA;

if (strncmp(header, Y4M_FRAME_MAGIC, strlen(Y4M_FRAME_MAGIC))) if (strncmp(header, Y4M_FRAME_MAGIC, strlen(Y4M_FRAME_MAGIC)))
return -1;
return AVERROR_INVALIDDATA;


width = st->codec->width; width = st->codec->width;
height = st->codec->height; height = st->codec->height;


packet_size = avpicture_get_size(st->codec->pix_fmt, width, height); packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);
if (packet_size < 0) if (packet_size < 0)
return -1;
return packet_size;


if (av_get_packet(s->pb, pkt, packet_size) != packet_size)
return AVERROR(EIO);
ret = av_get_packet(s->pb, pkt, packet_size);
if (ret < 0)
return ret;
else if (ret != packet_size)
return s->pb->eof_reached ? AVERROR_EOF : AVERROR(EIO);


if (st->codec->coded_frame) { if (st->codec->coded_frame) {
st->codec->coded_frame->interlaced_frame = s1->interlaced_frame; st->codec->coded_frame->interlaced_frame = s1->interlaced_frame;


Loading…
Cancel
Save