openharmony/third_party_ffmpeg
1
0
Fork 0
mirror of https://gitee.com/openharmony/third_party_ffmpeg synced 2024-11-23 19:30:05 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

avformat/hlsenc: Duplicated segment filenames and use_localtime_mkdir

Browse Source 
Current implementation of finding duplicate segment filenames may fail
if use_localtime_mkdir and use_localtime are in effect and
segment_filename option expression contains subdirectories with
date/time specifiers. This patch fixes this false behaviour.

Signed-off-by: Bela Bodecs <bodecsb@vivanet.hu>
Signed-off-by: Steven Liu <lq@chinaffmpeg.org>
This commit is contained in:
Bela Bodecs 2017-01-01 20:22:34 +08:00
parent 8fd3e02eee
commit 03a69bd897
1 changed files with 14 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
libavformat/hlsenc.c
View File

@ -367,6 +367,16 @@ static int hls_mux_init(AVFormatContext *s)
return 0;
}
static HLSSegment *find_segment_by_filename(HLSSegment *segment, const char *filename)
{
while (segment) {
if (!av_strcasecmp(segment->filename,filename))
return segment;
segment = segment->next;
}
return (HLSSegment *) NULL;
}
/* Create a new segment and append it to the segment list */
static int hls_append_segment(struct AVFormatContext *s, HLSContext *hls, double duration,
int64_t pos, int64_t size)
@ -383,6 +393,10 @@ static int hls_append_segment(struct AVFormatContext *s, HLSContext *hls, double
if (hls->use_localtime_mkdir) {
filename = hls->avf->filename;
}
if (find_segment_by_filename(hls->segments, filename)
|| find_segment_by_filename(hls->old_segments, en->filename)) {
av_log(hls, AV_LOG_WARNING, "Duplicated segment filename detected: %s\n", filename);
}
av_strlcpy(en->filename, filename, sizeof(en->filename));
if(hls->has_subtitle)
@ -659,38 +673,6 @@ fail:
return ret;
}
static HLSSegment *find_segment_by_filename(HLSSegment *segment, const char *filename)
{
/* filename may contain rel/abs path, but segments store only basename */
char *p = NULL, *dirname = NULL, *path = NULL;
int path_size;
HLSSegment *ret_segment = NULL;
dirname = av_strdup(filename);
if (!dirname)
return NULL;
p = (char *)av_basename(dirname); // av_dirname would return . in case of no dir
*p = '\0'; // maybe empty
while (segment) {
path_size = strlen(dirname) + strlen(segment->filename) + 1;
path = av_malloc(path_size);
if (!path)
goto end;
av_strlcpy(path, dirname, path_size);
av_strlcat(path, segment->filename, path_size);
if (!strcmp(path,filename)) {
ret_segment = segment;
av_free(path);
goto end;
}
av_free(path);
segment = segment->next;
}
end:
av_free(dirname);
return ret_segment;
}
static int hls_start(AVFormatContext *s)
{
HLSContext *c = s->priv_data;
@ -736,10 +718,6 @@ static int hls_start(AVFormatContext *s)
}
av_free(filename);
}
if (find_segment_by_filename(c->segments, oc->filename)
|| find_segment_by_filename(c->old_segments, oc->filename)) {
av_log(c, AV_LOG_WARNING, "Duplicated segment filename detected: %s\n", oc->filename);
}
if (c->use_localtime_mkdir) {
const char *dir;
char *fn_copy = av_strdup(oc->filename);