openharmony/third_party_ffmpeg
1
0
Fork 0
mirror of https://gitee.com/openharmony/third_party_ffmpeg synced 2025-02-21 13:22:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

avformat/vobsub: do not create empty streams.

Browse Source 
Fixes NULL dereference in vobsub_read_packet() on vobsub->q[i]->subs.

Fixes ticket #3669.
This commit is contained in:
Clément Bœsch 2014-05-24 11:03:23 +02:00
parent 308188be34
commit cba92a2226
1 changed files with 35 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
libavformat/mpeg.c
View File

@ -661,6 +661,7 @@ AVInputFormat ff_mpegps_demuxer = {
#if CONFIG_VOBSUB_DEMUXER #if CONFIG_VOBSUB_DEMUXER
#define REF_STRING "# VobSub index file," #define REF_STRING "# VobSub index file,"
#define MAX_LINE_SIZE 2048
static int vobsub_probe(AVProbeData *p) static int vobsub_probe(AVProbeData *p)
{ {
@ -679,6 +680,9 @@ static int vobsub_read_header(AVFormatContext *s)
AVBPrint header; AVBPrint header;
int64_t delay = 0; int64_t delay = 0;
AVStream *st = NULL; AVStream *st = NULL;
int stream_id = -1;
char id[64] = {0};
char alt[MAX_LINE_SIZE] = {0};
sub_name = av_strdup(s->filename); sub_name = av_strdup(s->filename);
fname_len = strlen(sub_name); fname_len = strlen(sub_name);
@ -699,7 +703,7 @@ static int vobsub_read_header(AVFormatContext *s)
av_bprint_init(&header, 0, AV_BPRINT_SIZE_UNLIMITED); av_bprint_init(&header, 0, AV_BPRINT_SIZE_UNLIMITED);
while (!url_feof(s->pb)) { while (!url_feof(s->pb)) {
char line[2048]; char line[MAX_LINE_SIZE];
int len = ff_get_line(s->pb, line, sizeof(line)); int len = ff_get_line(s->pb, line, sizeof(line));
if (!len) if (!len)
@ -708,11 +712,7 @@ static int vobsub_read_header(AVFormatContext *s)
line[strcspn(line, "\r\n")] = 0; line[strcspn(line, "\r\n")] = 0;
if (!strncmp(line, "id:", 3)) { if (!strncmp(line, "id:", 3)) {
int n, stream_id = 0; if (sscanf(line, "id: %63[^,], index: %u", id, &stream_id) != 2) {
char id[64] = {0};
n = sscanf(line, "id: %63[^,], index: %u", id, &stream_id);
if (n != 2) {
av_log(s, AV_LOG_WARNING, "Unable to parse index line '%s', " av_log(s, AV_LOG_WARNING, "Unable to parse index line '%s', "
"assuming 'id: und, index: 0'\n", line); "assuming 'id: und, index: 0'\n", line);
strcpy(id, "und"); strcpy(id, "und");
@ -725,36 +725,46 @@ static int vobsub_read_header(AVFormatContext *s)
goto end; goto end;
} }
st = avformat_new_stream(s, NULL);
if (!st) {
ret = AVERROR(ENOMEM);
goto end;
}
st->id = stream_id;
st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
st->codec->codec_id = AV_CODEC_ID_DVD_SUBTITLE;
avpriv_set_pts_info(st, 64, 1, 1000);
av_dict_set(&st->metadata, "language", id, 0);
av_log(s, AV_LOG_DEBUG, "IDX stream[%d] id=%s\n", stream_id, id);
header_parsed = 1; header_parsed = 1;
alt[0] = '\0';
/* We do not create the stream immediately to avoid adding empty
* streams. See the following timestamp entry. */
} else if (st && !strncmp(line, "timestamp:", 10)) { av_log(s, AV_LOG_DEBUG, "IDX stream[%d] id=%s\n", stream_id, id);
} else if (!strncmp(line, "timestamp:", 10)) {
AVPacket *sub; AVPacket *sub;
int hh, mm, ss, ms; int hh, mm, ss, ms;
int64_t pos, timestamp; int64_t pos, timestamp;
const char *p = line + 10; const char *p = line + 10;
if (!s->nb_streams) { if (stream_id == -1) {
av_log(s, AV_LOG_ERROR, "Timestamp declared before any stream\n"); av_log(s, AV_LOG_ERROR, "Timestamp declared before any stream\n");
ret = AVERROR_INVALIDDATA; ret = AVERROR_INVALIDDATA;
goto end; goto end;
} }
if (!st || st->id != stream_id) {
st = avformat_new_stream(s, NULL);
if (!st) {
ret = AVERROR(ENOMEM);
goto end;
}
st->id = stream_id;
st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
st->codec->codec_id = AV_CODEC_ID_DVD_SUBTITLE;
avpriv_set_pts_info(st, 64, 1, 1000);
av_dict_set(&st->metadata, "language", id, 0);
if (alt[0])
av_dict_set(&st->metadata, "title", alt, 0);
}
if (sscanf(p, "%02d:%02d:%02d:%03d, filepos: %"SCNx64, if (sscanf(p, "%02d:%02d:%02d:%03d, filepos: %"SCNx64,
&hh, &mm, &ss, &ms, &pos) != 5) { &hh, &mm, &ss, &ms, &pos) != 5) {
av_log(s, AV_LOG_ERROR, "Unable to parse timestamp line '%s', " av_log(s, AV_LOG_ERROR, "Unable to parse timestamp line '%s', "
"abort parsing\n", line); "abort parsing\n", line);
break; ret = AVERROR_INVALIDDATA;
goto end;
} }
timestamp = (hh*3600LL + mm*60LL + ss) * 1000LL + ms + delay; timestamp = (hh*3600LL + mm*60LL + ss) * 1000LL + ms + delay;
timestamp = av_rescale_q(timestamp, av_make_q(1, 1000), st->time_base); timestamp = av_rescale_q(timestamp, av_make_q(1, 1000), st->time_base);
@ -768,13 +778,13 @@ static int vobsub_read_header(AVFormatContext *s)
sub->pts = timestamp; sub->pts = timestamp;
sub->stream_index = s->nb_streams - 1; sub->stream_index = s->nb_streams - 1;
} else if (st && !strncmp(line, "alt:", 4)) { } else if (!strncmp(line, "alt:", 4)) {
const char *p = line + 4; const char *p = line + 4;
while (*p == ' ') while (*p == ' ')
p++; p++;
av_dict_set(&st->metadata, "title", p, 0);
av_log(s, AV_LOG_DEBUG, "IDX stream[%d] name=%s\n", st->id, p); av_log(s, AV_LOG_DEBUG, "IDX stream[%d] name=%s\n", st->id, p);
av_strlcpy(alt, p, sizeof(alt));
header_parsed = 1; header_parsed = 1;
} else if (!strncmp(line, "delay:", 6)) { } else if (!strncmp(line, "delay:", 6)) {
@ -840,7 +850,9 @@ static int vobsub_read_packet(AVFormatContext *s, AVPacket *pkt)
int sid = 0; int sid = 0;
for (i = 0; i < s->nb_streams; i++) { for (i = 0; i < s->nb_streams; i++) {
FFDemuxSubtitlesQueue *tmpq = &vobsub->q[i]; FFDemuxSubtitlesQueue *tmpq = &vobsub->q[i];
int64_t ts = tmpq->subs[tmpq->current_sub_idx].pts; int64_t ts;
av_assert0(tmpq->nb_subs);
ts = tmpq->subs[tmpq->current_sub_idx].pts;
if (ts < min_ts) { if (ts < min_ts) {
min_ts = ts; min_ts = ts;
sid = i; sid = i;