RepoMirrors
/
ffmpeg
mirror of https://git.ffmpeg.org/ffmpeg.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

mpjpeg: probe should require same constraints as packet reader - both proper content-type and content-size must be present 

return AVPROBE_SCORE_MAX, rather than random positive number on success

Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
This commit is contained in:
Alex Agranovsky 2015-09-13 16:47:54 -04:00 committed by Michael Niedermayer
parent b65ffa316e
commit 0572bd1a62
1 changed files with 14 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
libavformat/mpjpegdec.c
View File

@ -77,6 +77,8 @@ static int check_content_type(char *line)
return 0;
}
static int parse_multipart_header(AVIOContext *pb, void *log_ctx);
static int mpjpeg_read_probe(AVProbeData *p)
{
AVIOContext *pb;
@ -90,17 +92,7 @@ static int mpjpeg_read_probe(AVProbeData *p)
if (!pb)
return AVERROR(ENOMEM);
while (!pb->eof_reached) {
ret = get_line(pb, line, sizeof(line));
if (ret < 0)
break;
ret = check_content_type(line);
if (!ret) {
ret = AVPROBE_SCORE_MAX;
break;
}
}
ret = (parse_multipart_header(pb, NULL)>0)?AVPROBE_SCORE_MAX:0;
av_free(pb);
@ -147,23 +139,23 @@ static int parse_content_length(const char *value)
return val;
}
static int parse_multipart_header(AVFormatContext *s)
static int parse_multipart_header(AVIOContext *pb, void *log_ctx)
{
char line[128];
int found_content_type = 0;
int ret, size = -1;
ret = get_line(s->pb, line, sizeof(line));
ret = get_line(pb, line, sizeof(line));
if (ret < 0)
return ret;
if (strncmp(line, "--", 2))
return AVERROR_INVALIDDATA;
while (!s->pb->eof_reached) {
while (!pb->eof_reached) {
char *tag, *value;
ret = get_line(s->pb, line, sizeof(line));
ret = get_line(pb, line, sizeof(line));
if (ret < 0) {
if (ret==AVERROR_EOF)
break;
@ -179,9 +171,12 @@ static int parse_multipart_header(AVFormatContext *s)
if (!av_strcasecmp(tag, "Content-type")) {
if (av_strcasecmp(value, "image/jpeg")) {
av_log(s, AV_LOG_ERROR,
if (log_ctx) {
av_log(log_ctx, AV_LOG_ERROR,
"Unexpected %s : %s\n",
tag, value);
}
return AVERROR_INVALIDDATA;
} else
found_content_type = 1;
@ -202,7 +197,7 @@ static int parse_multipart_header(AVFormatContext *s)
static int mpjpeg_read_packet(AVFormatContext *s, AVPacket *pkt)
{
int ret;
int size = parse_multipart_header(s);
int size = parse_multipart_header(s->pb, s);
if (size < 0)
return size;