/*
* Apple HTTP Live Streaming segmenter
* Copyright (c) 2012, Luca Barbato
+ * Copyright (c) 2017 Akamai Technologies, Inc.
*
* This file is part of FFmpeg.
*
#include "avformat.h"
#include "avio_internal.h"
+#if CONFIG_HTTP_PROTOCOL
+#include "http.h"
+#endif
+#include "hlsplaylist.h"
#include "internal.h"
#include "os_support.h"
HLS_START_SEQUENCE_AS_FORMATTED_DATETIME = 2, // YYYYMMDDhhmmss
} StartSequenceSourceType;
+typedef enum {
+ CODEC_ATTRIBUTE_WRITTEN = 0,
+ CODEC_ATTRIBUTE_WILL_NOT_BE_WRITTEN,
+} CodecAttributeStatus;
+
#define KEYSIZE 16
#define LINE_BUFFER_SIZE 1024
#define HLS_MICROSECOND_UNIT 1000000
+#define POSTFIX_PATTERN "_%d"
typedef struct HLSSegment {
char filename[1024];
HLS_SECOND_LEVEL_SEGMENT_SIZE = (1 << 10), // include segment size (bytes) in segment filenames when use_localtime e.g.: %%014s
HLS_TEMP_FILE = (1 << 11),
HLS_PERIODIC_REKEY = (1 << 12),
+ HLS_INDEPENDENT_SEGMENTS = (1 << 13),
} HLSFlags;
typedef enum {
SEGMENT_TYPE_FMP4,
} SegmentType;
-typedef enum {
- PLAYLIST_TYPE_NONE,
- PLAYLIST_TYPE_EVENT,
- PLAYLIST_TYPE_VOD,
- PLAYLIST_TYPE_NB,
-} PlaylistType;
-
-typedef struct HLSContext {
- const AVClass *class; // Class for private options.
+typedef struct VariantStream {
unsigned number;
int64_t sequence;
- int64_t start_sequence;
- uint32_t start_sequence_source_type; // enum StartSequenceSourceType
AVOutputFormat *oformat;
AVOutputFormat *vtt_oformat;
+ AVIOContext *out;
+ int packets_written;
+ int init_range_length;
AVFormatContext *avf;
AVFormatContext *vtt_avf;
- float time; // Set by a private option.
- float init_time; // Set by a private option.
- int max_nb_segments; // Set by a private option.
-#if FF_API_HLS_WRAP
- int wrap; // Set by a private option.
-#endif
- uint32_t flags; // enum HLSFlags
- uint32_t pl_type; // enum PlaylistType
- char *segment_filename;
- char *fmp4_init_filename;
- int segment_type;
- int fmp4_init_mode;
-
- int use_localtime; ///< flag to expand filename with localtime
- int use_localtime_mkdir;///< flag to mkdir dirname in timebased filename
- int allowcache;
- int64_t recording_time;
int has_video;
int has_subtitle;
int new_start;
double duration; // last segment duration computed so far, in seconds
int64_t start_pos; // last segment starting position
int64_t size; // last segment size
- int64_t max_seg_size; // every segment file max size
int nb_entries;
int discontinuity_set;
int discontinuity;
HLSSegment *old_segments;
char *basename;
- char *base_output_dirname;
char *vtt_basename;
char *vtt_m3u8_name;
+ char *m3u8_name;
+
+ double initial_prog_date_time;
+ char current_segment_final_filename_fmt[1024]; // when renaming segments
+
+ char *fmp4_init_filename;
+ char *base_output_dirname;
+ int fmp4_init_mode;
+
+ AVStream **streams;
+ char codec_attr[128];
+ CodecAttributeStatus attr_status;
+ unsigned int nb_streams;
+ int m3u8_created; /* status of media play-list creation */
+ char *agroup; /* audio group name */
+ char *baseurl;
+} VariantStream;
+
+typedef struct HLSContext {
+ const AVClass *class; // Class for private options.
+ int64_t start_sequence;
+ uint32_t start_sequence_source_type; // enum StartSequenceSourceType
+
+ float time; // Set by a private option.
+ float init_time; // Set by a private option.
+ int max_nb_segments; // Set by a private option.
+#if FF_API_HLS_WRAP
+ int wrap; // Set by a private option.
+#endif
+ uint32_t flags; // enum HLSFlags
+ uint32_t pl_type; // enum PlaylistType
+ char *segment_filename;
+ char *fmp4_init_filename;
+ int segment_type;
+
+ int use_localtime; ///< flag to expand filename with localtime
+ int use_localtime_mkdir;///< flag to mkdir dirname in timebased filename
+ int allowcache;
+ int64_t recording_time;
+ int64_t max_seg_size; // every segment file max size
+
char *baseurl;
char *format_options_str;
char *vtt_format_options_str;
char *key_url;
char *iv;
char *key_basename;
+ int encrypt_started;
char *key_info_file;
char key_file[LINE_BUFFER_SIZE + 1];
AVDictionary *vtt_format_options;
char *method;
-
- double initial_prog_date_time;
- char current_segment_final_filename_fmt[1024]; // when renaming segments
char *user_agent;
-} HLSContext;
-static int get_int_from_double(double val)
-{
- return (int)((val - (int)val) >= 0.001) ? (int)(val + 1) : (int)val;
-}
+ VariantStream *var_streams;
+ unsigned int nb_varstreams;
+
+ int master_m3u8_created; /* status of master play-list creation */
+ char *master_m3u8_url; /* URL of the master m3u8 file */
+ int version; /* HLS version */
+ char *var_stream_map; /* user specified variant stream map string */
+ char *master_pl_name;
+ unsigned int master_publish_rate;
+ int http_persistent;
+ AVIOContext *m3u8_out;
+ AVIOContext *sub_m3u8_out;
+} HLSContext;
static int mkdir_p(const char *path) {
int ret = 0;
return ret;
}
+static int hlsenc_io_open(AVFormatContext *s, AVIOContext **pb, char *filename,
+ AVDictionary **options) {
+ HLSContext *hls = s->priv_data;
+ int http_base_proto = filename ? ff_is_http_proto(filename) : 0;
+ int err = AVERROR_MUXER_NOT_FOUND;
+ if (!*pb || !http_base_proto || !hls->http_persistent) {
+ err = s->io_open(s, pb, filename, AVIO_FLAG_WRITE, options);
+#if CONFIG_HTTP_PROTOCOL
+ } else {
+ URLContext *http_url_context = ffio_geturlcontext(*pb);
+ av_assert0(http_url_context);
+ err = ff_http_do_new_request(http_url_context, filename);
+#endif
+ }
+ return err;
+}
+
+static void hlsenc_io_close(AVFormatContext *s, AVIOContext **pb, char *filename) {
+ HLSContext *hls = s->priv_data;
+ int http_base_proto = filename ? ff_is_http_proto(filename) : 0;
+ if (!http_base_proto || !hls->http_persistent || hls->key_info_file || hls->encrypt) {
+ ff_format_io_close(s, pb);
+#if CONFIG_HTTP_PROTOCOL
+ } else {
+ URLContext *http_url_context = ffio_geturlcontext(*pb);
+ av_assert0(http_url_context);
+ avio_flush(*pb);
+ ffurl_shutdown(http_url_context, AVIO_FLAG_WRITE);
+#endif
+ }
+}
+
static void set_http_options(AVFormatContext *s, AVDictionary **options, HLSContext *c)
{
- const char *proto = avio_find_protocol_name(s->filename);
- int http_base_proto = proto ? (!av_strcasecmp(proto, "http") || !av_strcasecmp(proto, "https")) : 0;
+ int http_base_proto = ff_is_http_proto(s->filename);
if (c->method) {
av_dict_set(options, "method", c->method, 0);
}
if (c->user_agent)
av_dict_set(options, "user_agent", c->user_agent, 0);
+ if (c->http_persistent)
+ av_dict_set_int(options, "multiple_requests", 1, 0);
+
+}
+
+static void write_codec_attr(AVStream *st, VariantStream *vs) {
+ int codec_strlen = strlen(vs->codec_attr);
+ char attr[32];
+
+ if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE)
+ return;
+ if (vs->attr_status == CODEC_ATTRIBUTE_WILL_NOT_BE_WRITTEN)
+ return;
+
+ if (st->codecpar->codec_id == AV_CODEC_ID_H264) {
+ uint8_t *data = st->codecpar->extradata;
+ if (data && (data[0] | data[1] | data[2]) == 0 && data[3] == 1 && (data[4] & 0x1F) == 7) {
+ snprintf(attr, sizeof(attr),
+ "avc1.%02x%02x%02x", data[5], data[6], data[7]);
+ } else {
+ goto fail;
+ }
+ } else if (st->codecpar->codec_id == AV_CODEC_ID_MP2) {
+ snprintf(attr, sizeof(attr), "mp4a.40.33");
+ } else if (st->codecpar->codec_id == AV_CODEC_ID_MP3) {
+ snprintf(attr, sizeof(attr), "mp4a.40.34");
+ } else if (st->codecpar->codec_id == AV_CODEC_ID_AAC) {
+ /* TODO : For HE-AAC, HE-AACv2, the last digit needs to be set to 5 and 29 respectively */
+ snprintf(attr, sizeof(attr), "mp4a.40.2");
+ } else if (st->codecpar->codec_id == AV_CODEC_ID_AC3) {
+ snprintf(attr, sizeof(attr), "ac-3");
+ } else if (st->codecpar->codec_id == AV_CODEC_ID_EAC3) {
+ snprintf(attr, sizeof(attr), "ec-3");
+ } else {
+ goto fail;
+ }
+ // Don't write the same attribute multiple times
+ if (!av_stristr(vs->codec_attr, attr)) {
+ snprintf(vs->codec_attr + codec_strlen,
+ sizeof(vs->codec_attr) - codec_strlen,
+ "%s%s", codec_strlen ? "," : "", attr);
+ }
+ return;
+fail:
+ vs->codec_attr[0] = '\0';
+ vs->attr_status = CODEC_ATTRIBUTE_WILL_NOT_BE_WRITTEN;
+ return;
}
static int replace_int_data_in_filename(char *buf, int buf_size, const char *filename, char placeholder, int64_t number)
ffio_wfourcc(pb, "msix");
}
-static int hls_delete_old_segments(AVFormatContext *s, HLSContext *hls) {
+static int flush_dynbuf(VariantStream *vs, int *range_length)
+{
+ AVFormatContext *ctx = vs->avf;
+ uint8_t *buffer;
+
+ if (!ctx->pb) {
+ return AVERROR(EINVAL);
+ }
+
+ // flush
+ av_write_frame(ctx, NULL);
+ avio_flush(ctx->pb);
+
+ // write out to file
+ *range_length = avio_close_dyn_buf(ctx->pb, &buffer);
+ ctx->pb = NULL;
+ avio_write(vs->out, buffer, *range_length);
+ av_free(buffer);
+
+ // re-open buffer
+ return avio_open_dyn_buf(&ctx->pb);
+}
+
+static int hls_delete_old_segments(AVFormatContext *s, HLSContext *hls,
+ VariantStream *vs) {
HLSSegment *segment, *previous_segment = NULL;
float playlist_duration = 0.0f;
AVIOContext *out = NULL;
const char *proto = NULL;
- segment = hls->segments;
+ segment = vs->segments;
while (segment) {
playlist_duration += segment->duration;
segment = segment->next;
}
- segment = hls->old_segments;
+ segment = vs->old_segments;
while (segment) {
playlist_duration -= segment->duration;
previous_segment = segment;
if (hls->segment_filename) {
dirname = av_strdup(hls->segment_filename);
} else {
- dirname = av_strdup(hls->avf->filename);
+ dirname = av_strdup(vs->avf->filename);
}
if (!dirname) {
ret = AVERROR(ENOMEM);
proto = avio_find_protocol_name(s->filename);
if (hls->method || (proto && !av_strcasecmp(proto, "http"))) {
av_dict_set(&options, "method", "DELETE", 0);
- if ((ret = hls->avf->io_open(hls->avf, &out, path, AVIO_FLAG_WRITE, &options)) < 0)
+ if ((ret = vs->avf->io_open(vs->avf, &out, path, AVIO_FLAG_WRITE, &options)) < 0)
goto fail;
- ff_format_io_close(hls->avf, &out);
+ ff_format_io_close(vs->avf, &out);
} else if (unlink(path) < 0) {
av_log(hls, AV_LOG_ERROR, "failed to delete old segment %s: %s\n",
path, strerror(errno));
if (hls->method || (proto && !av_strcasecmp(proto, "http"))) {
av_dict_set(&options, "method", "DELETE", 0);
- if ((ret = hls->avf->io_open(hls->avf, &out, sub_path, AVIO_FLAG_WRITE, &options)) < 0) {
+ if ((ret = vs->avf->io_open(vs->avf, &out, sub_path, AVIO_FLAG_WRITE, &options)) < 0) {
av_free(sub_path);
goto fail;
}
- ff_format_io_close(hls->avf, &out);
+ ff_format_io_close(vs->avf, &out);
} else if (unlink(sub_path) < 0) {
av_log(hls, AV_LOG_ERROR, "failed to delete old segment %s: %s\n",
sub_path, strerror(errno));
return AVERROR(EINVAL);
}
-static int do_encrypt(AVFormatContext *s)
+static int do_encrypt(AVFormatContext *s, VariantStream *vs)
{
HLSContext *hls = s->priv_data;
int ret;
AVIOContext *pb;
uint8_t key[KEYSIZE];
- len = strlen(hls->basename) + 4 + 1;
+ len = strlen(s->filename) + 4 + 1;
hls->key_basename = av_mallocz(len);
if (!hls->key_basename)
return AVERROR(ENOMEM);
char buf[33];
if (!hls->iv) {
- AV_WB64(iv + 8, hls->sequence);
+ AV_WB64(iv + 8, vs->sequence);
} else {
memcpy(iv, hls->iv, sizeof(iv));
}
return len;
}
-static int hls_mux_init(AVFormatContext *s)
+static int hls_mux_init(AVFormatContext *s, VariantStream *vs)
{
AVDictionary *options = NULL;
HLSContext *hls = s->priv_data;
int byterange_mode = (hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size > 0);
int i, ret;
- ret = avformat_alloc_output_context2(&hls->avf, hls->oformat, NULL, NULL);
+ ret = avformat_alloc_output_context2(&vs->avf, vs->oformat, NULL, NULL);
if (ret < 0)
return ret;
- oc = hls->avf;
+ oc = vs->avf;
oc->filename[0] = '\0';
- oc->oformat = hls->oformat;
+ oc->oformat = vs->oformat;
oc->interrupt_callback = s->interrupt_callback;
oc->max_delay = s->max_delay;
oc->opaque = s->opaque;
oc->io_close = s->io_close;
av_dict_copy(&oc->metadata, s->metadata, 0);
- if(hls->vtt_oformat) {
- ret = avformat_alloc_output_context2(&hls->vtt_avf, hls->vtt_oformat, NULL, NULL);
+ if(vs->vtt_oformat) {
+ ret = avformat_alloc_output_context2(&vs->vtt_avf, vs->vtt_oformat, NULL, NULL);
if (ret < 0)
return ret;
- vtt_oc = hls->vtt_avf;
- vtt_oc->oformat = hls->vtt_oformat;
+ vtt_oc = vs->vtt_avf;
+ vtt_oc->oformat = vs->vtt_oformat;
av_dict_copy(&vtt_oc->metadata, s->metadata, 0);
}
- for (i = 0; i < s->nb_streams; i++) {
+ for (i = 0; i < vs->nb_streams; i++) {
AVStream *st;
AVFormatContext *loc;
- if (s->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE)
+ if (vs->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE)
loc = vtt_oc;
else
loc = oc;
if (!(st = avformat_new_stream(loc, NULL)))
return AVERROR(ENOMEM);
- avcodec_parameters_copy(st->codecpar, s->streams[i]->codecpar);
+ avcodec_parameters_copy(st->codecpar, vs->streams[i]->codecpar);
if (!oc->oformat->codec_tag ||
- av_codec_get_id (oc->oformat->codec_tag, s->streams[i]->codecpar->codec_tag) == st->codecpar->codec_id ||
- av_codec_get_tag(oc->oformat->codec_tag, s->streams[i]->codecpar->codec_id) <= 0) {
- st->codecpar->codec_tag = s->streams[i]->codecpar->codec_tag;
+ av_codec_get_id (oc->oformat->codec_tag, vs->streams[i]->codecpar->codec_tag) == st->codecpar->codec_id ||
+ av_codec_get_tag(oc->oformat->codec_tag, vs->streams[i]->codecpar->codec_id) <= 0) {
+ st->codecpar->codec_tag = vs->streams[i]->codecpar->codec_tag;
} else {
st->codecpar->codec_tag = 0;
}
- st->sample_aspect_ratio = s->streams[i]->sample_aspect_ratio;
- st->time_base = s->streams[i]->time_base;
- av_dict_copy(&st->metadata, s->streams[i]->metadata, 0);
+ st->sample_aspect_ratio = vs->streams[i]->sample_aspect_ratio;
+ st->time_base = vs->streams[i]->time_base;
+ av_dict_copy(&st->metadata, vs->streams[i]->metadata, 0);
}
- hls->start_pos = 0;
- hls->new_start = 1;
- hls->fmp4_init_mode = 0;
+
+ vs->packets_written = 1;
+ vs->start_pos = 0;
+ vs->new_start = 1;
+ vs->fmp4_init_mode = 0;
if (hls->segment_type == SEGMENT_TYPE_FMP4) {
if (hls->max_seg_size > 0) {
av_log(s, AV_LOG_WARNING, "Multi-file byterange mode is currently unsupported in the HLS muxer.\n");
return AVERROR_PATCHWELCOME;
}
- hls->fmp4_init_mode = !byterange_mode;
+
+ vs->packets_written = 0;
+ vs->init_range_length = 0;
+ vs->fmp4_init_mode = !byterange_mode;
set_http_options(s, &options, hls);
- if ((ret = s->io_open(s, &oc->pb, hls->base_output_dirname, AVIO_FLAG_WRITE, &options)) < 0) {
- av_log(s, AV_LOG_ERROR, "Failed to open segment '%s'\n", hls->fmp4_init_filename);
+ if ((ret = avio_open_dyn_buf(&oc->pb)) < 0)
+ return ret;
+
+ ret = hlsenc_io_open(s, &vs->out, vs->base_output_dirname, &options);
+ av_dict_free(&options);
+ if (ret < 0) {
+ av_log(s, AV_LOG_ERROR, "Failed to open segment '%s'\n", vs->fmp4_init_filename);
return ret;
}
av_dict_free(&options);
return AVERROR(EINVAL);
}
+ avio_flush(oc->pb);
av_dict_free(&options);
}
return 0;
return (HLSSegment *) NULL;
}
-static int sls_flags_filename_process(struct AVFormatContext *s, HLSContext *hls, HLSSegment *en, double duration,
- int64_t pos, int64_t size)
+static int sls_flags_filename_process(struct AVFormatContext *s, HLSContext *hls,
+ VariantStream *vs, HLSSegment *en,
+ double duration, int64_t pos, int64_t size)
{
if ((hls->flags & (HLS_SECOND_LEVEL_SEGMENT_SIZE | HLS_SECOND_LEVEL_SEGMENT_DURATION)) &&
- strlen(hls->current_segment_final_filename_fmt)) {
- av_strlcpy(hls->avf->filename, hls->current_segment_final_filename_fmt, sizeof(hls->avf->filename));
+ strlen(vs->current_segment_final_filename_fmt)) {
+ av_strlcpy(vs->avf->filename, vs->current_segment_final_filename_fmt, sizeof(vs->avf->filename));
if (hls->flags & HLS_SECOND_LEVEL_SEGMENT_SIZE) {
- char * filename = av_strdup(hls->avf->filename); // %%s will be %s after strftime
+ char * filename = av_strdup(vs->avf->filename); // %%s will be %s after strftime
if (!filename) {
av_free(en);
return AVERROR(ENOMEM);
}
- if (replace_int_data_in_filename(hls->avf->filename, sizeof(hls->avf->filename),
+ if (replace_int_data_in_filename(vs->avf->filename, sizeof(vs->avf->filename),
filename, 's', pos + size) < 1) {
av_log(hls, AV_LOG_ERROR,
"Invalid second level segment filename template '%s', "
av_free(filename);
}
if (hls->flags & HLS_SECOND_LEVEL_SEGMENT_DURATION) {
- char * filename = av_strdup(hls->avf->filename); // %%t will be %t after strftime
+ char * filename = av_strdup(vs->avf->filename); // %%t will be %t after strftime
if (!filename) {
av_free(en);
return AVERROR(ENOMEM);
}
- if (replace_int_data_in_filename(hls->avf->filename, sizeof(hls->avf->filename),
+ if (replace_int_data_in_filename(vs->avf->filename, sizeof(vs->avf->filename),
filename, 't', (int64_t)round(duration * HLS_MICROSECOND_UNIT)) < 1) {
av_log(hls, AV_LOG_ERROR,
"Invalid second level segment filename template '%s', "
return ret;
}
-static int sls_flag_check_duration_size(HLSContext *hls)
+static int sls_flag_check_duration_size(HLSContext *hls, VariantStream *vs)
{
- const char *proto = avio_find_protocol_name(hls->basename);
+ const char *proto = avio_find_protocol_name(vs->basename);
int segment_renaming_ok = proto && !strcmp(proto, "file");
int ret = 0;
return ret;
}
-static void sls_flag_file_rename(HLSContext *hls, char *old_filename) {
+static void sls_flag_file_rename(HLSContext *hls, VariantStream *vs, char *old_filename) {
if ((hls->flags & (HLS_SECOND_LEVEL_SEGMENT_SIZE | HLS_SECOND_LEVEL_SEGMENT_DURATION)) &&
- strlen(hls->current_segment_final_filename_fmt)) {
- ff_rename(old_filename, hls->avf->filename, hls);
+ strlen(vs->current_segment_final_filename_fmt)) {
+ ff_rename(old_filename, vs->avf->filename, hls);
}
}
-static int sls_flag_use_localtime_filename(AVFormatContext *oc, HLSContext *c)
+static int sls_flag_use_localtime_filename(AVFormatContext *oc, HLSContext *c, VariantStream *vs)
{
if (c->flags & HLS_SECOND_LEVEL_SEGMENT_INDEX) {
char * filename = av_strdup(oc->filename); // %%d will be %d after strftime
return AVERROR(ENOMEM);
if (replace_int_data_in_filename(oc->filename, sizeof(oc->filename),
#if FF_API_HLS_WRAP
- filename, 'd', c->wrap ? c->sequence % c->wrap : c->sequence) < 1) {
+ filename, 'd', c->wrap ? vs->sequence % c->wrap : vs->sequence) < 1) {
#else
- filename, 'd', c->sequence) < 1) {
+ filename, 'd', vs->sequence) < 1) {
#endif
av_log(c, AV_LOG_ERROR, "Invalid second level segment filename template '%s', "
"you can try to remove second_level_segment_index flag\n",
av_free(filename);
}
if (c->flags & (HLS_SECOND_LEVEL_SEGMENT_SIZE | HLS_SECOND_LEVEL_SEGMENT_DURATION)) {
- av_strlcpy(c->current_segment_final_filename_fmt, oc->filename,
- sizeof(c->current_segment_final_filename_fmt));
+ av_strlcpy(vs->current_segment_final_filename_fmt, oc->filename,
+ sizeof(vs->current_segment_final_filename_fmt));
if (c->flags & HLS_SECOND_LEVEL_SEGMENT_SIZE) {
char * filename = av_strdup(oc->filename); // %%s will be %s after strftime
if (!filename)
}
/* 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)
+static int hls_append_segment(struct AVFormatContext *s, HLSContext *hls,
+ VariantStream *vs, double duration, int64_t pos,
+ int64_t size)
{
HLSSegment *en = av_malloc(sizeof(*en));
const char *filename;
if (!en)
return AVERROR(ENOMEM);
- ret = sls_flags_filename_process(s, hls, en, duration, pos, size);
+ ret = sls_flags_filename_process(s, hls, vs, en, duration, pos, size);
if (ret < 0) {
return ret;
}
- filename = av_basename(hls->avf->filename);
+ filename = av_basename(vs->avf->filename);
if (hls->use_localtime_mkdir) {
- filename = hls->avf->filename;
+ filename = vs->avf->filename;
}
- if ((find_segment_by_filename(hls->segments, filename) || find_segment_by_filename(hls->old_segments, filename))
+ if ((find_segment_by_filename(vs->segments, filename) || find_segment_by_filename(vs->old_segments, filename))
&& !byterange_mode) {
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)
- av_strlcpy(en->sub_filename, av_basename(hls->vtt_avf->filename), sizeof(en->sub_filename));
+ if(vs->has_subtitle)
+ av_strlcpy(en->sub_filename, av_basename(vs->vtt_avf->filename), sizeof(en->sub_filename));
else
en->sub_filename[0] = '\0';
en->next = NULL;
en->discont = 0;
- if (hls->discontinuity) {
+ if (vs->discontinuity) {
en->discont = 1;
- hls->discontinuity = 0;
+ vs->discontinuity = 0;
}
if (hls->key_info_file || hls->encrypt) {
av_strlcpy(en->iv_string, hls->iv_string, sizeof(en->iv_string));
}
- if (!hls->segments)
- hls->segments = en;
+ if (!vs->segments)
+ vs->segments = en;
else
- hls->last_segment->next = en;
+ vs->last_segment->next = en;
- hls->last_segment = en;
+ vs->last_segment = en;
// EVENT or VOD playlists imply sliding window cannot be used
if (hls->pl_type != PLAYLIST_TYPE_NONE)
hls->max_nb_segments = 0;
- if (hls->max_nb_segments && hls->nb_entries >= hls->max_nb_segments) {
- en = hls->segments;
- hls->initial_prog_date_time += en->duration;
- hls->segments = en->next;
+ if (hls->max_nb_segments && vs->nb_entries >= hls->max_nb_segments) {
+ en = vs->segments;
+ vs->initial_prog_date_time += en->duration;
+ vs->segments = en->next;
if (en && hls->flags & HLS_DELETE_SEGMENTS &&
#if FF_API_HLS_WRAP
!(hls->flags & HLS_SINGLE_FILE || hls->wrap)) {
#else
!(hls->flags & HLS_SINGLE_FILE)) {
#endif
- en->next = hls->old_segments;
- hls->old_segments = en;
- if ((ret = hls_delete_old_segments(s, hls)) < 0)
+ en->next = vs->old_segments;
+ vs->old_segments = en;
+ if ((ret = hls_delete_old_segments(s, hls, vs)) < 0)
return ret;
} else
av_free(en);
} else
- hls->nb_entries++;
+ vs->nb_entries++;
if (hls->max_seg_size > 0) {
return 0;
}
- hls->sequence++;
+ vs->sequence++;
return 0;
}
-static int parse_playlist(AVFormatContext *s, const char *url)
+static int parse_playlist(AVFormatContext *s, const char *url, VariantStream *vs)
{
HLSContext *hls = s->priv_data;
AVIOContext *in;
goto fail;
}
- hls->discontinuity = 0;
+ vs->discontinuity = 0;
while (!avio_feof(in)) {
read_chomp_line(in, line, sizeof(line));
if (av_strstart(line, "#EXT-X-MEDIA-SEQUENCE:", &ptr)) {
int64_t tmp_sequence = strtoll(ptr, NULL, 10);
- if (tmp_sequence < hls->sequence)
+ if (tmp_sequence < vs->sequence)
av_log(hls, AV_LOG_VERBOSE,
"Found playlist sequence number was smaller """
"than specified start sequence number: %"PRId64" < %"PRId64", "
"omitting\n", tmp_sequence, hls->start_sequence);
else {
av_log(hls, AV_LOG_DEBUG, "Found playlist sequence number: %"PRId64"\n", tmp_sequence);
- hls->sequence = tmp_sequence;
+ vs->sequence = tmp_sequence;
}
} else if (av_strstart(line, "#EXT-X-DISCONTINUITY", &ptr)) {
is_segment = 1;
- hls->discontinuity = 1;
+ vs->discontinuity = 1;
} else if (av_strstart(line, "#EXTINF:", &ptr)) {
is_segment = 1;
- hls->duration = atof(ptr);
+ vs->duration = atof(ptr);
} else if (av_stristart(line, "#EXT-X-KEY:", &ptr)) {
ptr = av_stristr(line, "URI=\"");
if (ptr) {
} else if (line[0]) {
if (is_segment) {
is_segment = 0;
- new_start_pos = avio_tell(hls->avf->pb);
- hls->size = new_start_pos - hls->start_pos;
- av_strlcpy(hls->avf->filename, line, sizeof(line));
- ret = hls_append_segment(s, hls, hls->duration, hls->start_pos, hls->size);
+ new_start_pos = avio_tell(vs->avf->pb);
+ vs->size = new_start_pos - vs->start_pos;
+ av_strlcpy(vs->avf->filename, line, sizeof(line));
+ ret = hls_append_segment(s, hls, vs, vs->duration, vs->start_pos, vs->size);
if (ret < 0)
goto fail;
- hls->start_pos = new_start_pos;
+ vs->start_pos = new_start_pos;
}
}
}
}
}
-static void write_m3u8_head_block(HLSContext *hls, AVIOContext *out, int version,
- int target_duration, int64_t sequence)
-{
- avio_printf(out, "#EXTM3U\n");
- avio_printf(out, "#EXT-X-VERSION:%d\n", version);
- if (hls->allowcache == 0 || hls->allowcache == 1) {
- avio_printf(out, "#EXT-X-ALLOW-CACHE:%s\n", hls->allowcache == 0 ? "NO" : "YES");
- }
- avio_printf(out, "#EXT-X-TARGETDURATION:%d\n", target_duration);
- avio_printf(out, "#EXT-X-MEDIA-SEQUENCE:%"PRId64"\n", sequence);
- av_log(hls, AV_LOG_VERBOSE, "EXT-X-MEDIA-SEQUENCE:%"PRId64"\n", sequence);
-}
-
static void hls_rename_temp_file(AVFormatContext *s, AVFormatContext *oc)
{
size_t len = strlen(oc->filename);
oc->filename[len-4] = '\0';
}
-static int hls_window(AVFormatContext *s, int last)
+static int get_relative_url(const char *master_url, const char *media_url,
+ char *rel_url, int rel_url_buf_size)
+{
+ char *p = NULL;
+ int base_len = -1;
+ p = strrchr(master_url, '/') ? strrchr(master_url, '/') :\
+ strrchr(master_url, '\\');
+ if (p) {
+ base_len = FFABS(p - master_url);
+ if (av_strncasecmp(master_url, media_url, base_len)) {
+ av_log(NULL, AV_LOG_WARNING, "Unable to find relative url\n");
+ return AVERROR(EINVAL);
+ }
+ }
+ av_strlcpy(rel_url, &(media_url[base_len + 1]), rel_url_buf_size);
+ return 0;
+}
+
+static int create_master_playlist(AVFormatContext *s,
+ VariantStream * const input_vs)
+{
+ HLSContext *hls = s->priv_data;
+ VariantStream *vs, *temp_vs;
+ AVStream *vid_st, *aud_st;
+ AVDictionary *options = NULL;
+ unsigned int i, j;
+ int m3u8_name_size, ret, bandwidth;
+ char *m3u8_rel_name;
+
+ input_vs->m3u8_created = 1;
+ if (!hls->master_m3u8_created) {
+ /* For the first time, wait until all the media playlists are created */
+ for (i = 0; i < hls->nb_varstreams; i++)
+ if (!hls->var_streams[i].m3u8_created)
+ return 0;
+ } else {
+ /* Keep publishing the master playlist at the configured rate */
+ if (&hls->var_streams[0] != input_vs || !hls->master_publish_rate ||
+ input_vs->number % hls->master_publish_rate)
+ return 0;
+ }
+
+ set_http_options(s, &options, hls);
+
+ ret = hlsenc_io_open(s, &hls->m3u8_out, hls->master_m3u8_url, &options);
+ av_dict_free(&options);
+ if (ret < 0) {
+ av_log(NULL, AV_LOG_ERROR, "Failed to open master play list file '%s'\n",
+ hls->master_m3u8_url);
+ goto fail;
+ }
+
+ ff_hls_write_playlist_version(hls->m3u8_out, hls->version);
+
+ /* For audio only variant streams add #EXT-X-MEDIA tag with attributes*/
+ for (i = 0; i < hls->nb_varstreams; i++) {
+ vs = &(hls->var_streams[i]);
+
+ if (vs->has_video || vs->has_subtitle || !vs->agroup)
+ continue;
+
+ m3u8_name_size = strlen(vs->m3u8_name) + 1;
+ m3u8_rel_name = av_malloc(m3u8_name_size);
+ if (!m3u8_rel_name) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ av_strlcpy(m3u8_rel_name, vs->m3u8_name, m3u8_name_size);
+ ret = get_relative_url(hls->master_m3u8_url, vs->m3u8_name,
+ m3u8_rel_name, m3u8_name_size);
+ if (ret < 0) {
+ av_log(s, AV_LOG_ERROR, "Unable to find relative URL\n");
+ goto fail;
+ }
+
+ ff_hls_write_audio_rendition(hls->m3u8_out, vs->agroup, m3u8_rel_name, 0, 1);
+
+ av_freep(&m3u8_rel_name);
+ }
+
+ /* For variant streams with video add #EXT-X-STREAM-INF tag with attributes*/
+ for (i = 0; i < hls->nb_varstreams; i++) {
+ vs = &(hls->var_streams[i]);
+
+ m3u8_name_size = strlen(vs->m3u8_name) + 1;
+ m3u8_rel_name = av_malloc(m3u8_name_size);
+ if (!m3u8_rel_name) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ av_strlcpy(m3u8_rel_name, vs->m3u8_name, m3u8_name_size);
+ ret = get_relative_url(hls->master_m3u8_url, vs->m3u8_name,
+ m3u8_rel_name, m3u8_name_size);
+ if (ret < 0) {
+ av_log(NULL, AV_LOG_ERROR, "Unable to find relative URL\n");
+ goto fail;
+ }
+
+ vid_st = NULL;
+ aud_st = NULL;
+ for (j = 0; j < vs->nb_streams; j++) {
+ if (vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
+ vid_st = vs->streams[j];
+ else if (vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO)
+ aud_st = vs->streams[j];
+ }
+
+ if (!vid_st && !aud_st) {
+ av_log(NULL, AV_LOG_WARNING, "Media stream not found\n");
+ continue;
+ }
+
+ /**
+ * Traverse through the list of audio only rendition streams and find
+ * the rendition which has highest bitrate in the same audio group
+ */
+ if (vs->agroup) {
+ for (j = 0; j < hls->nb_varstreams; j++) {
+ temp_vs = &(hls->var_streams[j]);
+ if (!temp_vs->has_video && !temp_vs->has_subtitle &&
+ temp_vs->agroup &&
+ !av_strcasecmp(temp_vs->agroup, vs->agroup)) {
+ if (!aud_st)
+ aud_st = temp_vs->streams[0];
+ if (temp_vs->streams[0]->codecpar->bit_rate >
+ aud_st->codecpar->bit_rate)
+ aud_st = temp_vs->streams[0];
+ }
+ }
+ }
+
+ bandwidth = 0;
+ if (vid_st)
+ bandwidth += vid_st->codecpar->bit_rate;
+ if (aud_st)
+ bandwidth += aud_st->codecpar->bit_rate;
+ bandwidth += bandwidth / 10;
+
+ ff_hls_write_stream_info(vid_st, hls->m3u8_out, bandwidth, m3u8_rel_name,
+ aud_st ? vs->agroup : NULL, vs->codec_attr);
+
+ av_freep(&m3u8_rel_name);
+ }
+fail:
+ if(ret >=0)
+ hls->master_m3u8_created = 1;
+ av_freep(&m3u8_rel_name);
+ hlsenc_io_close(s, &hls->m3u8_out, hls->master_m3u8_url);
+ return ret;
+}
+
+static int hls_window(AVFormatContext *s, int last, VariantStream *vs)
{
HLSContext *hls = s->priv_data;
HLSSegment *en;
int target_duration = 0;
int ret = 0;
- AVIOContext *out = NULL;
- AVIOContext *sub_out = NULL;
char temp_filename[1024];
- int64_t sequence = FFMAX(hls->start_sequence, hls->sequence - hls->nb_entries);
- int version = 3;
+ int64_t sequence = FFMAX(hls->start_sequence, vs->sequence - vs->nb_entries);
const char *proto = avio_find_protocol_name(s->filename);
int use_rename = proto && !strcmp(proto, "file");
static unsigned warned_non_file;
char *key_uri = NULL;
char *iv_string = NULL;
AVDictionary *options = NULL;
- double prog_date_time = hls->initial_prog_date_time;
+ double prog_date_time = vs->initial_prog_date_time;
+ double *prog_date_time_p = (hls->flags & HLS_PROGRAM_DATE_TIME) ? &prog_date_time : NULL;
int byterange_mode = (hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size > 0);
+ hls->version = 3;
if (byterange_mode) {
- version = 4;
+ hls->version = 4;
sequence = 0;
}
+ if (hls->flags & HLS_INDEPENDENT_SEGMENTS) {
+ hls->version = 6;
+ }
+
if (hls->segment_type == SEGMENT_TYPE_FMP4) {
- version = 7;
+ hls->version = 7;
}
if (!use_rename && !warned_non_file++)
av_log(s, AV_LOG_ERROR, "Cannot use rename on non file protocol, this may lead to races and temporary partial files\n");
set_http_options(s, &options, hls);
- snprintf(temp_filename, sizeof(temp_filename), use_rename ? "%s.tmp" : "%s", s->filename);
- if ((ret = s->io_open(s, &out, temp_filename, AVIO_FLAG_WRITE, &options)) < 0)
+ snprintf(temp_filename, sizeof(temp_filename), use_rename ? "%s.tmp" : "%s", vs->m3u8_name);
+ if ((ret = hlsenc_io_open(s, &hls->m3u8_out, temp_filename, &options)) < 0)
goto fail;
- for (en = hls->segments; en; en = en->next) {
+ for (en = vs->segments; en; en = en->next) {
if (target_duration <= en->duration)
- target_duration = get_int_from_double(en->duration);
+ target_duration = lrint(en->duration);
}
- hls->discontinuity_set = 0;
- write_m3u8_head_block(hls, out, version, target_duration, sequence);
- if (hls->pl_type == PLAYLIST_TYPE_EVENT) {
- avio_printf(out, "#EXT-X-PLAYLIST-TYPE:EVENT\n");
- } else if (hls->pl_type == PLAYLIST_TYPE_VOD) {
- avio_printf(out, "#EXT-X-PLAYLIST-TYPE:VOD\n");
- }
+ vs->discontinuity_set = 0;
+ ff_hls_write_playlist_header(hls->m3u8_out, hls->version, hls->allowcache,
+ target_duration, sequence, hls->pl_type);
- if((hls->flags & HLS_DISCONT_START) && sequence==hls->start_sequence && hls->discontinuity_set==0 ){
- avio_printf(out, "#EXT-X-DISCONTINUITY\n");
- hls->discontinuity_set = 1;
+ if((hls->flags & HLS_DISCONT_START) && sequence==hls->start_sequence && vs->discontinuity_set==0 ){
+ avio_printf(hls->m3u8_out, "#EXT-X-DISCONTINUITY\n");
+ vs->discontinuity_set = 1;
}
- for (en = hls->segments; en; en = en->next) {
+ if (vs->has_video && (hls->flags & HLS_INDEPENDENT_SEGMENTS)) {
+ avio_printf(hls->m3u8_out, "#EXT-X-INDEPENDENT-SEGMENTS\n");
+ }
+ for (en = vs->segments; en; en = en->next) {
if ((hls->encrypt || hls->key_info_file) && (!key_uri || strcmp(en->key_uri, key_uri) ||
av_strcasecmp(en->iv_string, iv_string))) {
- avio_printf(out, "#EXT-X-KEY:METHOD=AES-128,URI=\"%s\"", en->key_uri);
+ avio_printf(hls->m3u8_out, "#EXT-X-KEY:METHOD=AES-128,URI=\"%s\"", en->key_uri);
if (*en->iv_string)
- avio_printf(out, ",IV=0x%s", en->iv_string);
- avio_printf(out, "\n");
+ avio_printf(hls->m3u8_out, ",IV=0x%s", en->iv_string);
+ avio_printf(hls->m3u8_out, "\n");
key_uri = en->key_uri;
iv_string = en->iv_string;
}
- if (en->discont) {
- avio_printf(out, "#EXT-X-DISCONTINUITY\n");
- }
-
- if ((hls->segment_type == SEGMENT_TYPE_FMP4) && (en == hls->segments)) {
- avio_printf(out, "#EXT-X-MAP:URI=\"%s\"", hls->fmp4_init_filename);
- if (hls->flags & HLS_SINGLE_FILE) {
- avio_printf(out, ",BYTERANGE=\"%"PRId64"@%"PRId64"\"", en->size, en->pos);
- }
- avio_printf(out, "\n");
+ if ((hls->segment_type == SEGMENT_TYPE_FMP4) && (en == vs->segments)) {
+ ff_hls_write_init_file(hls->m3u8_out, vs->fmp4_init_filename,
+ hls->flags & HLS_SINGLE_FILE, en->size, en->pos);
}
- if (hls->flags & HLS_ROUND_DURATIONS)
- avio_printf(out, "#EXTINF:%ld,\n", lrint(en->duration));
- else
- avio_printf(out, "#EXTINF:%f,\n", en->duration);
- if (byterange_mode)
- avio_printf(out, "#EXT-X-BYTERANGE:%"PRId64"@%"PRId64"\n",
- en->size, en->pos);
- if (hls->flags & HLS_PROGRAM_DATE_TIME) {
- time_t tt, wrongsecs;
- int milli;
- struct tm *tm, tmpbuf;
- char buf0[128], buf1[128];
- tt = (int64_t)prog_date_time;
- milli = av_clip(lrint(1000*(prog_date_time - tt)), 0, 999);
- tm = localtime_r(&tt, &tmpbuf);
- strftime(buf0, sizeof(buf0), "%Y-%m-%dT%H:%M:%S", tm);
- if (!strftime(buf1, sizeof(buf1), "%z", tm) || buf1[1]<'0' ||buf1[1]>'2') {
- int tz_min, dst = tm->tm_isdst;
- tm = gmtime_r(&tt, &tmpbuf);
- tm->tm_isdst = dst;
- wrongsecs = mktime(tm);
- tz_min = (abs(wrongsecs - tt) + 30) / 60;
- snprintf(buf1, sizeof(buf1),
- "%c%02d%02d",
- wrongsecs <= tt ? '+' : '-',
- tz_min / 60,
- tz_min % 60);
- }
- avio_printf(out, "#EXT-X-PROGRAM-DATE-TIME:%s.%03d%s\n", buf0, milli, buf1);
- prog_date_time += en->duration;
+ ret = ff_hls_write_file_entry(hls->m3u8_out, en->discont, byterange_mode,
+ en->duration, hls->flags & HLS_ROUND_DURATIONS,
+ en->size, en->pos, vs->baseurl,
+ en->filename, prog_date_time_p);
+ if (ret < 0) {
+ av_log(s, AV_LOG_WARNING, "ff_hls_write_file_entry get error\n");
}
- if (hls->baseurl)
- avio_printf(out, "%s", hls->baseurl);
- avio_printf(out, "%s\n", en->filename);
}
if (last && (hls->flags & HLS_OMIT_ENDLIST)==0)
- avio_printf(out, "#EXT-X-ENDLIST\n");
+ ff_hls_write_end_list(hls->m3u8_out);
- if( hls->vtt_m3u8_name ) {
- if ((ret = s->io_open(s, &sub_out, hls->vtt_m3u8_name, AVIO_FLAG_WRITE, &options)) < 0)
+ if( vs->vtt_m3u8_name ) {
+ if ((ret = hlsenc_io_open(s, &hls->sub_m3u8_out, vs->vtt_m3u8_name, &options)) < 0)
goto fail;
- write_m3u8_head_block(hls, sub_out, version, target_duration, sequence);
-
- for (en = hls->segments; en; en = en->next) {
- avio_printf(sub_out, "#EXTINF:%f,\n", en->duration);
- if (byterange_mode)
- avio_printf(sub_out, "#EXT-X-BYTERANGE:%"PRIi64"@%"PRIi64"\n",
- en->size, en->pos);
- if (hls->baseurl)
- avio_printf(sub_out, "%s", hls->baseurl);
- avio_printf(sub_out, "%s\n", en->sub_filename);
+ ff_hls_write_playlist_header(hls->sub_m3u8_out, hls->version, hls->allowcache,
+ target_duration, sequence, PLAYLIST_TYPE_NONE);
+ for (en = vs->segments; en; en = en->next) {
+ ret = ff_hls_write_file_entry(hls->sub_m3u8_out, 0, byterange_mode,
+ en->duration, 0, en->size, en->pos,
+ vs->baseurl, en->sub_filename, NULL);
+ if (ret < 0) {
+ av_log(s, AV_LOG_WARNING, "ff_hls_write_file_entry get error\n");
+ }
}
if (last)
- avio_printf(sub_out, "#EXT-X-ENDLIST\n");
+ ff_hls_write_end_list(hls->sub_m3u8_out);
}
fail:
av_dict_free(&options);
- ff_format_io_close(s, &out);
- ff_format_io_close(s, &sub_out);
+ hlsenc_io_close(s, &hls->m3u8_out, temp_filename);
+ hlsenc_io_close(s, &hls->sub_m3u8_out, vs->vtt_m3u8_name);
if (ret >= 0 && use_rename)
- ff_rename(temp_filename, s->filename, s);
+ ff_rename(temp_filename, vs->m3u8_name, s);
+
+ if (ret >= 0 && hls->master_pl_name)
+ if (create_master_playlist(s, vs) < 0)
+ av_log(s, AV_LOG_WARNING, "Master playlist creation failed\n");
+
return ret;
}
-static int hls_start(AVFormatContext *s)
+static int hls_start(AVFormatContext *s, VariantStream *vs)
{
HLSContext *c = s->priv_data;
- AVFormatContext *oc = c->avf;
- AVFormatContext *vtt_oc = c->vtt_avf;
+ AVFormatContext *oc = vs->avf;
+ AVFormatContext *vtt_oc = vs->vtt_avf;
AVDictionary *options = NULL;
char *filename, iv_string[KEYSIZE*2 + 1];
int err = 0;
if (c->flags & HLS_SINGLE_FILE) {
- av_strlcpy(oc->filename, c->basename,
+ av_strlcpy(oc->filename, vs->basename,
sizeof(oc->filename));
- if (c->vtt_basename)
- av_strlcpy(vtt_oc->filename, c->vtt_basename,
+ if (vs->vtt_basename)
+ av_strlcpy(vtt_oc->filename, vs->vtt_basename,
sizeof(vtt_oc->filename));
} else if (c->max_seg_size > 0) {
if (replace_int_data_in_filename(oc->filename, sizeof(oc->filename),
#if FF_API_HLS_WRAP
- c->basename, 'd', c->wrap ? c->sequence % c->wrap : c->sequence) < 1) {
+ vs->basename, 'd', c->wrap ? vs->sequence % c->wrap : vs->sequence) < 1) {
#else
- c->basename, 'd', c->sequence) < 1) {
+ vs->basename, 'd', vs->sequence) < 1) {
#endif
- av_log(oc, AV_LOG_ERROR, "Invalid segment filename template '%s', you can try to use -use_localtime 1 with it\n", c->basename);
+ av_log(oc, AV_LOG_ERROR, "Invalid segment filename template '%s', you can try to use -use_localtime 1 with it\n", vs->basename);
return AVERROR(EINVAL);
}
} else {
struct tm *tm, tmpbuf;
time(&now0);
tm = localtime_r(&now0, &tmpbuf);
- if (!strftime(oc->filename, sizeof(oc->filename), c->basename, tm)) {
+ if (!strftime(oc->filename, sizeof(oc->filename), vs->basename, tm)) {
av_log(oc, AV_LOG_ERROR, "Could not get segment filename with use_localtime\n");
return AVERROR(EINVAL);
}
- err = sls_flag_use_localtime_filename(oc, c);
+ err = sls_flag_use_localtime_filename(oc, c, vs);
if (err < 0) {
return AVERROR(ENOMEM);
}
}
} else if (replace_int_data_in_filename(oc->filename, sizeof(oc->filename),
#if FF_API_HLS_WRAP
- c->basename, 'd', c->wrap ? c->sequence % c->wrap : c->sequence) < 1) {
+ vs->basename, 'd', c->wrap ? vs->sequence % c->wrap : vs->sequence) < 1) {
#else
- c->basename, 'd', c->sequence) < 1) {
+ vs->basename, 'd', vs->sequence) < 1) {
#endif
- av_log(oc, AV_LOG_ERROR, "Invalid segment filename template '%s' you can try to use -use_localtime 1 with it\n", c->basename);
+ av_log(oc, AV_LOG_ERROR, "Invalid segment filename template '%s' you can try to use -use_localtime 1 with it\n", vs->basename);
return AVERROR(EINVAL);
}
- if( c->vtt_basename) {
+ if( vs->vtt_basename) {
if (replace_int_data_in_filename(vtt_oc->filename, sizeof(vtt_oc->filename),
#if FF_API_HLS_WRAP
- c->vtt_basename, 'd', c->wrap ? c->sequence % c->wrap : c->sequence) < 1) {
+ vs->vtt_basename, 'd', c->wrap ? vs->sequence % c->wrap : vs->sequence) < 1) {
#else
- c->vtt_basename, 'd', c->sequence) < 1) {
+ vs->vtt_basename, 'd', vs->sequence) < 1) {
#endif
- av_log(vtt_oc, AV_LOG_ERROR, "Invalid segment filename template '%s'\n", c->vtt_basename);
+ av_log(vtt_oc, AV_LOG_ERROR, "Invalid segment filename template '%s'\n", vs->vtt_basename);
return AVERROR(EINVAL);
}
}
}
- c->number++;
+ vs->number++;
set_http_options(s, &options, c);
" will use -hls_key_info_file priority\n");
}
- if (c->number <= 1 || (c->flags & HLS_PERIODIC_REKEY)) {
+ if (!c->encrypt_started || (c->flags & HLS_PERIODIC_REKEY)) {
if (c->key_info_file) {
if ((err = hls_encryption_start(s)) < 0)
goto fail;
} else {
- if ((err = do_encrypt(s)) < 0)
+ if ((err = do_encrypt(s, vs)) < 0)
goto fail;
}
+ c->encrypt_started = 1;
}
if ((err = av_dict_set(&options, "encryption_key", c->key_string, 0))
< 0)
goto fail;
err = av_strlcpy(iv_string, c->iv_string, sizeof(iv_string));
if (!err)
- snprintf(iv_string, sizeof(iv_string), "%032"PRIx64, c->sequence);
+ snprintf(iv_string, sizeof(iv_string), "%032"PRIx64, vs->sequence);
if ((err = av_dict_set(&options, "encryption_iv", iv_string, 0)) < 0)
goto fail;
err = AVERROR(ENOMEM);
goto fail;
}
- err = s->io_open(s, &oc->pb, filename, AVIO_FLAG_WRITE, &options);
+ err = hlsenc_io_open(s, &oc->pb, filename, &options);
av_free(filename);
av_dict_free(&options);
if (err < 0)
return err;
- } else
- if ((err = s->io_open(s, &oc->pb, oc->filename, AVIO_FLAG_WRITE, &options)) < 0)
+ } else if (c->segment_type != SEGMENT_TYPE_FMP4) {
+ if ((err = hlsenc_io_open(s, &oc->pb, oc->filename, &options)) < 0)
goto fail;
- if (c->vtt_basename) {
+ }
+ if (vs->vtt_basename) {
set_http_options(s, &options, c);
- if ((err = s->io_open(s, &vtt_oc->pb, vtt_oc->filename, AVIO_FLAG_WRITE, &options)) < 0)
+ if ((err = hlsenc_io_open(s, &vtt_oc->pb, vtt_oc->filename, &options)) < 0)
goto fail;
}
av_dict_free(&options);
- if (c->segment_type == SEGMENT_TYPE_FMP4 && !(c->flags & HLS_SINGLE_FILE)) {
- write_styp(oc->pb);
- } else {
+ if (c->segment_type != SEGMENT_TYPE_FMP4) {
/* We only require one PAT/PMT per segment. */
if (oc->oformat->priv_class && oc->priv_data) {
char period[21];
}
}
- if (c->vtt_basename) {
+ if (vs->vtt_basename) {
err = avformat_write_header(vtt_oc,NULL);
if (err < 0)
return err;
return (HAVE_LIBC_MSVCRT || !strftime(b, sizeof(b), "%s", p) || !strcmp(b, "%s")) ? "-%Y%m%d%H%M%S.ts" : "-%s.ts";
}
-static int hls_write_header(AVFormatContext *s)
+static int append_postfix(char *name, int name_buf_len, int i)
{
- HLSContext *hls = s->priv_data;
- int ret, i;
- char *p = NULL;
- const char *pattern = "%d.ts";
- const char *pattern_localtime_fmt = get_default_pattern_localtime_fmt(s);
- const char *vtt_pattern = "%d.vtt";
- AVDictionary *options = NULL;
- int basename_size = 0;
- int vtt_basename_size = 0;
- int fmp4_init_filename_len = strlen(hls->fmp4_init_filename) + 1;
+ char *p;
+ char extension[10] = {'\0'};
- if (hls->segment_type == SEGMENT_TYPE_FMP4) {
- pattern = "%d.m4s";
- }
- if ((hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH) ||
- (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_FORMATTED_DATETIME)) {
- time_t t = time(NULL); // we will need it in either case
- if (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH) {
- hls->start_sequence = (int64_t)t;
- } else if (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_FORMATTED_DATETIME) {
- char b[15];
- struct tm *p, tmbuf;
- if (!(p = localtime_r(&t, &tmbuf)))
- return AVERROR(ENOMEM);
- if (!strftime(b, sizeof(b), "%Y%m%d%H%M%S", p))
- return AVERROR(ENOMEM);
- hls->start_sequence = strtoll(b, NULL, 10);
- }
- av_log(hls, AV_LOG_DEBUG, "start_number evaluated to %"PRId64"\n", hls->start_sequence);
+ p = strrchr(name, '.');
+ if (p) {
+ av_strlcpy(extension, p, sizeof(extension));
+ *p = '\0';
}
- hls->sequence = hls->start_sequence;
- hls->recording_time = (hls->init_time ? hls->init_time : hls->time) * AV_TIME_BASE;
- hls->start_pts = AV_NOPTS_VALUE;
- hls->current_segment_final_filename_fmt[0] = '\0';
+ snprintf(name + strlen(name), name_buf_len - strlen(name), POSTFIX_PATTERN, i);
- if (hls->flags & HLS_PROGRAM_DATE_TIME) {
- time_t now0;
- time(&now0);
- hls->initial_prog_date_time = now0;
- }
+ if (strlen(extension))
+ av_strlcat(name, extension, name_buf_len);
- if (hls->format_options_str) {
- ret = av_dict_parse_string(&hls->format_options, hls->format_options_str, "=", ":", 0);
- if (ret < 0) {
- av_log(s, AV_LOG_ERROR, "Could not parse format options list '%s'\n", hls->format_options_str);
- goto fail;
- }
+ return 0;
+}
+
+static int validate_name(int nb_vs, const char *fn)
+{
+ const char *filename, *subdir_name;
+ char *fn_dup = NULL;
+ int ret = 0;
+
+ if (!fn) {
+ ret = AVERROR(EINVAL);
+ goto fail;
}
- for (i = 0; i < s->nb_streams; i++) {
- hls->has_video +=
- s->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO;
- hls->has_subtitle +=
- s->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE;
+ fn_dup = av_strdup(fn);
+ if (!fn_dup) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
}
- if (hls->has_video > 1)
- av_log(s, AV_LOG_WARNING,
- "More than a single video stream present, "
- "expect issues decoding it.\n");
+ filename = av_basename(fn);
+ subdir_name = av_dirname(fn_dup);
- if (hls->segment_type == SEGMENT_TYPE_FMP4) {
- hls->oformat = av_guess_format("mp4", NULL, NULL);
- } else {
- hls->oformat = av_guess_format("mpegts", NULL, NULL);
+ if (nb_vs > 1 && !av_stristr(filename, "%v") && !av_stristr(subdir_name, "%v")) {
+ av_log(NULL, AV_LOG_ERROR, "More than 1 variant streams are present, %%v is expected in the filename %s\n",
+ fn);
+ ret = AVERROR(EINVAL);
+ goto fail;
}
- if (!hls->oformat) {
- ret = AVERROR_MUXER_NOT_FOUND;
+ if (av_stristr(filename, "%v") && av_stristr(subdir_name, "%v")) {
+ av_log(NULL, AV_LOG_ERROR, "%%v is expected either in filename or in the sub-directory name of file %s\n",
+ fn);
+ ret = AVERROR(EINVAL);
goto fail;
}
- if(hls->has_subtitle) {
- hls->vtt_oformat = av_guess_format("webvtt", NULL, NULL);
- if (!hls->oformat) {
- ret = AVERROR_MUXER_NOT_FOUND;
- goto fail;
- }
+fail:
+ av_freep(&fn_dup);
+ return ret;
+}
+
+static int format_name(char *buf, int buf_len, int index)
+{
+ const char *proto, *dir;
+ char *orig_buf_dup = NULL, *mod_buf_dup = NULL;
+ int ret = 0;
+
+ if (!av_stristr(buf, "%v"))
+ return ret;
+
+ orig_buf_dup = av_strdup(buf);
+ if (!orig_buf_dup) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
}
- if (hls->segment_filename) {
- hls->basename = av_strdup(hls->segment_filename);
- if (!hls->basename) {
+ if (replace_int_data_in_filename(buf, buf_len, orig_buf_dup, 'v', index) < 1) {
+ ret = AVERROR(EINVAL);
+ goto fail;
+ }
+
+ proto = avio_find_protocol_name(orig_buf_dup);
+ dir = av_dirname(orig_buf_dup);
+
+ /* if %v is present in the file's directory, create sub-directory */
+ if (av_stristr(dir, "%v") && proto && !strcmp(proto, "file")) {
+ mod_buf_dup = av_strdup(buf);
+ if (!mod_buf_dup) {
ret = AVERROR(ENOMEM);
goto fail;
}
- } else {
- if (hls->flags & HLS_SINGLE_FILE) {
- if (hls->segment_type == SEGMENT_TYPE_FMP4) {
- pattern = ".m4s";
- } else {
- pattern = ".ts";
- }
- }
- if (hls->use_localtime) {
- basename_size = strlen(s->filename) + strlen(pattern_localtime_fmt) + 1;
- } else {
- basename_size = strlen(s->filename) + strlen(pattern) + 1;
- }
- hls->basename = av_malloc(basename_size);
- if (!hls->basename) {
- ret = AVERROR(ENOMEM);
+ dir = av_dirname(mod_buf_dup);
+ if (mkdir_p(dir) == -1 && errno != EEXIST) {
+ ret = AVERROR(errno);
goto fail;
}
+ }
- av_strlcpy(hls->basename, s->filename, basename_size);
+fail:
+ av_freep(&orig_buf_dup);
+ av_freep(&mod_buf_dup);
+ return ret;
+}
- p = strrchr(hls->basename, '.');
- if (p)
- *p = '\0';
- if (hls->use_localtime) {
- av_strlcat(hls->basename, pattern_localtime_fmt, basename_size);
- } else {
- av_strlcat(hls->basename, pattern, basename_size);
- }
+static int get_nth_codec_stream_index(AVFormatContext *s,
+ enum AVMediaType codec_type,
+ int stream_id)
+{
+ unsigned int stream_index, cnt;
+ if (stream_id < 0 || stream_id > s->nb_streams - 1)
+ return -1;
+ cnt = 0;
+ for (stream_index = 0; stream_index < s->nb_streams; stream_index++) {
+ if (s->streams[stream_index]->codecpar->codec_type != codec_type)
+ continue;
+ if (cnt == stream_id)
+ return stream_index;
+ cnt++;
}
+ return -1;
+}
- if (hls->segment_type == SEGMENT_TYPE_FMP4) {
- if (av_strcasecmp(hls->fmp4_init_filename, "init.mp4")) {
- hls->base_output_dirname = av_malloc(fmp4_init_filename_len);
- if (!hls->base_output_dirname) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- av_strlcpy(hls->base_output_dirname, hls->fmp4_init_filename, fmp4_init_filename_len);
- } else {
- if (basename_size > 0) {
- hls->base_output_dirname = av_malloc(basename_size);
+static int parse_variant_stream_mapstring(AVFormatContext *s)
+{
+ HLSContext *hls = s->priv_data;
+ VariantStream *vs;
+ int stream_index;
+ enum AVMediaType codec_type;
+ int nb_varstreams, nb_streams;
+ char *p, *q, *saveptr1, *saveptr2, *varstr, *keyval;
+ const char *val;
+
+ /**
+ * Expected format for var_stream_map string is as below:
+ * "a:0,v:0 a:1,v:1"
+ * "a:0,agroup:a0 a:1,agroup:a1 v:0,agroup:a0 v:1,agroup:a1"
+ * This string specifies how to group the audio, video and subtitle streams
+ * into different variant streams. The variant stream groups are separated
+ * by space.
+ *
+ * a:, v:, s: are keys to specify audio, video and subtitle streams
+ * respectively. Allowed values are 0 to 9 digits (limited just based on
+ * practical usage)
+ *
+ * agroup: is key to specify audio group. A string can be given as value.
+ */
+ p = av_strdup(hls->var_stream_map);
+ q = p;
+ while(av_strtok(q, " \t", &saveptr1)) {
+ q = NULL;
+ hls->nb_varstreams++;
+ }
+ av_freep(&p);
+
+ hls->var_streams = av_mallocz(sizeof(*hls->var_streams) * hls->nb_varstreams);
+ if (!hls->var_streams)
+ return AVERROR(ENOMEM);
+
+ p = hls->var_stream_map;
+ nb_varstreams = 0;
+ while (varstr = av_strtok(p, " \t", &saveptr1)) {
+ p = NULL;
+
+ if (nb_varstreams < hls->nb_varstreams)
+ vs = &(hls->var_streams[nb_varstreams++]);
+ else
+ return AVERROR(EINVAL);
+
+ q = varstr;
+ while (q < varstr + strlen(varstr)) {
+ if (!av_strncasecmp(q, "a:", 2) || !av_strncasecmp(q, "v:", 2) ||
+ !av_strncasecmp(q, "s:", 2))
+ vs->nb_streams++;
+ q++;
+ }
+ vs->streams = av_mallocz(sizeof(AVStream *) * vs->nb_streams);
+ if (!vs->streams)
+ return AVERROR(ENOMEM);
+
+ nb_streams = 0;
+ while (keyval = av_strtok(varstr, ",", &saveptr2)) {
+ varstr = NULL;
+
+ if (av_strstart(keyval, "agroup:", &val)) {
+ vs->agroup = av_strdup(val);
+ if (!vs->agroup)
+ return AVERROR(ENOMEM);
+ continue;
+ } else if (av_strstart(keyval, "v:", &val)) {
+ codec_type = AVMEDIA_TYPE_VIDEO;
+ } else if (av_strstart(keyval, "a:", &val)) {
+ codec_type = AVMEDIA_TYPE_AUDIO;
+ } else if (av_strstart(keyval, "s:", &val)) {
+ codec_type = AVMEDIA_TYPE_SUBTITLE;
} else {
- hls->base_output_dirname = av_malloc(strlen(hls->fmp4_init_filename));
- }
- if (!hls->base_output_dirname) {
- ret = AVERROR(ENOMEM);
- goto fail;
+ av_log(s, AV_LOG_ERROR, "Invalid keyval %s\n", keyval);
+ return AVERROR(EINVAL);
}
- if (basename_size > 0) {
- av_strlcpy(hls->base_output_dirname, s->filename, basename_size);
- p = strrchr(hls->base_output_dirname, '/');
- }
- if (p) {
- *(p + 1) = '\0';
- av_strlcat(hls->base_output_dirname, hls->fmp4_init_filename, basename_size);
+ stream_index = -1;
+ if (av_isdigit(*val))
+ stream_index = get_nth_codec_stream_index (s, codec_type,
+ atoi(val));
+
+ if (stream_index >= 0 && nb_streams < vs->nb_streams) {
+ vs->streams[nb_streams++] = s->streams[stream_index];
} else {
- av_strlcpy(hls->base_output_dirname, hls->fmp4_init_filename, fmp4_init_filename_len);
+ av_log(s, AV_LOG_ERROR, "Unable to map stream at %s\n", keyval);
+ return AVERROR(EINVAL);
}
}
}
+ av_log(s, AV_LOG_DEBUG, "Number of variant streams %d\n",
+ hls->nb_varstreams);
- if (!hls->use_localtime) {
- ret = sls_flag_check_duration_size_index(hls);
- if (ret < 0) {
- goto fail;
- }
+ return 0;
+}
+
+static int update_variant_stream_info(AVFormatContext *s) {
+ HLSContext *hls = s->priv_data;
+ unsigned int i;
+
+ if (hls->var_stream_map) {
+ return parse_variant_stream_mapstring(s);
} else {
- ret = sls_flag_check_duration_size(hls);
- if (ret < 0) {
- goto fail;
- }
+ //By default, a single variant stream with all the codec streams is created
+ hls->nb_varstreams = 1;
+ hls->var_streams = av_mallocz(sizeof(*hls->var_streams) *
+ hls->nb_varstreams);
+ if (!hls->var_streams)
+ return AVERROR(ENOMEM);
+
+ hls->var_streams[0].nb_streams = s->nb_streams;
+ hls->var_streams[0].streams = av_mallocz(sizeof(AVStream *) *
+ hls->var_streams[0].nb_streams);
+ if (!hls->var_streams[0].streams)
+ return AVERROR(ENOMEM);
+
+ for (i = 0; i < s->nb_streams; i++)
+ hls->var_streams[0].streams[i] = s->streams[i];
}
- if(hls->has_subtitle) {
+ return 0;
+}
- if (hls->flags & HLS_SINGLE_FILE)
- vtt_pattern = ".vtt";
- vtt_basename_size = strlen(s->filename) + strlen(vtt_pattern) + 1;
- hls->vtt_basename = av_malloc(vtt_basename_size);
- if (!hls->vtt_basename) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- hls->vtt_m3u8_name = av_malloc(vtt_basename_size);
- if (!hls->vtt_m3u8_name ) {
- ret = AVERROR(ENOMEM);
- goto fail;
- }
- av_strlcpy(hls->vtt_basename, s->filename, vtt_basename_size);
- p = strrchr(hls->vtt_basename, '.');
- if (p)
- *p = '\0';
+static int update_master_pl_info(AVFormatContext *s) {
+ HLSContext *hls = s->priv_data;
+ const char *dir;
+ char *fn1= NULL, *fn2 = NULL;
+ int ret = 0;
- if( hls->subtitle_filename ) {
- strcpy(hls->vtt_m3u8_name, hls->subtitle_filename);
- } else {
- strcpy(hls->vtt_m3u8_name, hls->vtt_basename);
- av_strlcat(hls->vtt_m3u8_name, "_vtt.m3u8", vtt_basename_size);
- }
- av_strlcat(hls->vtt_basename, vtt_pattern, vtt_basename_size);
+ fn1 = av_strdup(s->filename);
+ if (!fn1) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
}
- if ((hls->flags & HLS_SINGLE_FILE) && (hls->segment_type == SEGMENT_TYPE_FMP4)) {
- hls->fmp4_init_filename = av_strdup(hls->basename);
- if (!hls->fmp4_init_filename) {
+ dir = av_dirname(fn1);
+
+ /**
+ * if output file's directory has %v, variants are created in sub-directories
+ * then master is created at the sub-directories level
+ */
+ if (dir && av_stristr(av_basename(dir), "%v")) {
+ fn2 = av_strdup(dir);
+ if (!fn2) {
ret = AVERROR(ENOMEM);
goto fail;
}
+ dir = av_dirname(fn2);
}
- if ((ret = hls_mux_init(s)) < 0)
- goto fail;
+ if (dir && strcmp(dir, "."))
+ hls->master_m3u8_url = av_append_path_component(dir, hls->master_pl_name);
+ else
+ hls->master_m3u8_url = av_strdup(hls->master_pl_name);
- if (hls->flags & HLS_APPEND_LIST) {
- parse_playlist(s, s->filename);
- hls->discontinuity = 1;
- if (hls->init_time > 0) {
- av_log(s, AV_LOG_WARNING, "append_list mode does not support hls_init_time,"
- " hls_init_time value will have no effect\n");
- hls->init_time = 0;
- hls->recording_time = hls->time * AV_TIME_BASE;
- }
+ if (!hls->master_m3u8_url) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
}
- if (hls->segment_type != SEGMENT_TYPE_FMP4 || hls->flags & HLS_SINGLE_FILE) {
- if ((ret = hls_start(s)) < 0)
- goto fail;
- }
+fail:
+ av_freep(&fn1);
+ av_freep(&fn2);
- av_dict_copy(&options, hls->format_options, 0);
- ret = avformat_write_header(hls->avf, &options);
- if (av_dict_count(options)) {
- av_log(s, AV_LOG_ERROR, "Some of provided format options in '%s' are not recognized\n", hls->format_options_str);
- ret = AVERROR(EINVAL);
- goto fail;
- }
- //av_assert0(s->nb_streams == hls->avf->nb_streams);
- for (i = 0; i < s->nb_streams; i++) {
- AVStream *inner_st;
- AVStream *outer_st = s->streams[i];
+ return ret;
+}
- if (hls->max_seg_size > 0) {
- if ((outer_st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) &&
- (outer_st->codecpar->bit_rate > hls->max_seg_size)) {
- av_log(s, AV_LOG_WARNING, "Your video bitrate is bigger than hls_segment_size, "
- "(%"PRId64 " > %"PRId64 "), the result maybe not be what you want.",
- outer_st->codecpar->bit_rate, hls->max_seg_size);
- }
+static int hls_write_header(AVFormatContext *s)
+{
+ HLSContext *hls = s->priv_data;
+ int ret, i, j;
+ AVDictionary *options = NULL;
+ VariantStream *vs = NULL;
+
+ for (i = 0; i < hls->nb_varstreams; i++) {
+ vs = &hls->var_streams[i];
+
+ av_dict_copy(&options, hls->format_options, 0);
+ ret = avformat_write_header(vs->avf, &options);
+ if (av_dict_count(options)) {
+ av_log(s, AV_LOG_ERROR, "Some of provided format options in '%s' are not recognized\n", hls->format_options_str);
+ ret = AVERROR(EINVAL);
+ av_dict_free(&options);
+ goto fail;
}
+ av_dict_free(&options);
+ //av_assert0(s->nb_streams == hls->avf->nb_streams);
+ for (j = 0; j < vs->nb_streams; j++) {
+ AVStream *inner_st;
+ AVStream *outer_st = vs->streams[j];
+
+ if (hls->max_seg_size > 0) {
+ if ((outer_st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) &&
+ (outer_st->codecpar->bit_rate > hls->max_seg_size)) {
+ av_log(s, AV_LOG_WARNING, "Your video bitrate is bigger than hls_segment_size, "
+ "(%"PRId64 " > %"PRId64 "), the result maybe not be what you want.",
+ outer_st->codecpar->bit_rate, hls->max_seg_size);
+ }
+ }
+
+ if (outer_st->codecpar->codec_type != AVMEDIA_TYPE_SUBTITLE)
+ inner_st = vs->avf->streams[j];
+ else if (vs->vtt_avf)
+ inner_st = vs->vtt_avf->streams[0];
+ else {
+ /* We have a subtitle stream, when the user does not want one */
+ inner_st = NULL;
+ continue;
+ }
+ avpriv_set_pts_info(outer_st, inner_st->pts_wrap_bits, inner_st->time_base.num, inner_st->time_base.den);
+ write_codec_attr(outer_st, vs);
- if (outer_st->codecpar->codec_type != AVMEDIA_TYPE_SUBTITLE)
- inner_st = hls->avf->streams[i];
- else if (hls->vtt_avf)
- inner_st = hls->vtt_avf->streams[0];
- else {
- /* We have a subtitle stream, when the user does not want one */
- inner_st = NULL;
- continue;
}
- avpriv_set_pts_info(outer_st, inner_st->pts_wrap_bits, inner_st->time_base.num, inner_st->time_base.den);
+ /* Update the Codec Attr string for the mapped audio groups */
+ if (vs->has_video && vs->agroup) {
+ for (j = 0; j < hls->nb_varstreams; j++) {
+ VariantStream *vs_agroup = &(hls->var_streams[j]);
+ if (!vs_agroup->has_video && !vs_agroup->has_subtitle &&
+ vs_agroup->agroup &&
+ !av_strcasecmp(vs_agroup->agroup, vs->agroup)) {
+ write_codec_attr(vs_agroup->streams[0], vs);
+ }
+ }
+ }
}
fail:
- av_dict_free(&options);
- if (ret < 0) {
- av_freep(&hls->fmp4_init_filename);
- av_freep(&hls->basename);
- av_freep(&hls->vtt_basename);
- av_freep(&hls->key_basename);
- if (hls->avf)
- avformat_free_context(hls->avf);
- if (hls->vtt_avf)
- avformat_free_context(hls->vtt_avf);
-
- }
return ret;
}
HLSContext *hls = s->priv_data;
AVFormatContext *oc = NULL;
AVStream *st = s->streams[pkt->stream_index];
- int64_t end_pts = hls->recording_time * hls->number;
+ int64_t end_pts = 0;
int is_ref_pkt = 1;
- int ret = 0, can_split = 1;
+ int ret = 0, can_split = 1, i, j;
int stream_index = 0;
+ int range_length = 0;
+ uint8_t *buffer = NULL;
+ VariantStream *vs = NULL;
+
+ for (i = 0; i < hls->nb_varstreams; i++) {
+ vs = &hls->var_streams[i];
+ for (j = 0; j < vs->nb_streams; j++) {
+ if (vs->streams[j] == st) {
+ if( st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE ) {
+ oc = vs->vtt_avf;
+ stream_index = 0;
+ } else {
+ oc = vs->avf;
+ stream_index = j;
+ }
+ break;
+ }
+ }
+
+ if (oc)
+ break;
+ }
+
+ if (!oc) {
+ av_log(s, AV_LOG_ERROR, "Unable to find mapping variant stream\n");
+ return AVERROR(ENOMEM);
+ }
+
+ end_pts = hls->recording_time * vs->number;
- if (hls->sequence - hls->nb_entries > hls->start_sequence && hls->init_time > 0) {
+ if (vs->sequence - vs->nb_entries > hls->start_sequence && hls->init_time > 0) {
/* reset end_pts, hls->recording_time at end of the init hls list */
- int init_list_dur = hls->init_time * hls->nb_entries * AV_TIME_BASE;
- int after_init_list_dur = (hls->sequence - hls->nb_entries ) * hls->time * AV_TIME_BASE;
+ int init_list_dur = hls->init_time * vs->nb_entries * AV_TIME_BASE;
+ int after_init_list_dur = (vs->sequence - vs->nb_entries ) * hls->time * AV_TIME_BASE;
hls->recording_time = hls->time * AV_TIME_BASE;
end_pts = init_list_dur + after_init_list_dur ;
}
- if( st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE ) {
- oc = hls->vtt_avf;
- stream_index = 0;
- } else {
- oc = hls->avf;
- stream_index = pkt->stream_index;
- }
- if (hls->start_pts == AV_NOPTS_VALUE) {
- hls->start_pts = pkt->pts;
- hls->end_pts = pkt->pts;
+ if (vs->start_pts == AV_NOPTS_VALUE) {
+ vs->start_pts = pkt->pts;
}
- if (hls->has_video) {
+ if (vs->has_video) {
can_split = st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO &&
((pkt->flags & AV_PKT_FLAG_KEY) || (hls->flags & HLS_SPLIT_BY_TIME));
is_ref_pkt = st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO;
is_ref_pkt = can_split = 0;
if (is_ref_pkt) {
- if (hls->new_start) {
- hls->new_start = 0;
- hls->duration = (double)(pkt->pts - hls->end_pts)
+ if (vs->end_pts == AV_NOPTS_VALUE)
+ vs->end_pts = pkt->pts;
+ if (vs->new_start) {
+ vs->new_start = 0;
+ vs->duration = (double)(pkt->pts - vs->end_pts)
* st->time_base.num / st->time_base.den;
- hls->dpp = (double)(pkt->duration) * st->time_base.num / st->time_base.den;
+ vs->dpp = (double)(pkt->duration) * st->time_base.num / st->time_base.den;
} else {
if (pkt->duration) {
- hls->duration += (double)(pkt->duration) * st->time_base.num / st->time_base.den;
+ vs->duration += (double)(pkt->duration) * st->time_base.num / st->time_base.den;
} else {
av_log(s, AV_LOG_WARNING, "pkt->duration = 0, maybe the hls segment duration will not precise\n");
- hls->duration = (double)(pkt->pts - hls->end_pts) * st->time_base.num / st->time_base.den;
+ vs->duration = (double)(pkt->pts - vs->end_pts) * st->time_base.num / st->time_base.den;
}
}
}
- if (hls->fmp4_init_mode || can_split && av_compare_ts(pkt->pts - hls->start_pts, st->time_base,
+
+ if (vs->packets_written && can_split && av_compare_ts(pkt->pts - vs->start_pts, st->time_base,
end_pts, AV_TIME_BASE_Q) >= 0) {
int64_t new_start_pos;
- char *old_filename = av_strdup(hls->avf->filename);
+ char *old_filename = av_strdup(vs->avf->filename);
int byterange_mode = (hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size > 0);
if (!old_filename) {
return AVERROR(ENOMEM);
}
- av_write_frame(hls->avf, NULL); /* Flush any buffered data */
+ av_write_frame(vs->avf, NULL); /* Flush any buffered data */
- new_start_pos = avio_tell(hls->avf->pb);
- hls->size = new_start_pos - hls->start_pos;
+ new_start_pos = avio_tell(vs->avf->pb);
+ vs->size = new_start_pos - vs->start_pos;
if (!byterange_mode) {
- ff_format_io_close(s, &oc->pb);
- if (hls->vtt_avf) {
- ff_format_io_close(s, &hls->vtt_avf->pb);
+ if (hls->segment_type == SEGMENT_TYPE_FMP4) {
+ if (!vs->init_range_length) {
+ avio_flush(oc->pb);
+ range_length = avio_close_dyn_buf(oc->pb, &buffer);
+ avio_write(vs->out, buffer, range_length);
+ vs->init_range_length = range_length;
+ avio_open_dyn_buf(&oc->pb);
+ vs->packets_written = 0;
+ ff_format_io_close(s, &vs->out);
+ hlsenc_io_close(s, &vs->out, vs->base_output_dirname);
+ }
+ } else {
+ hlsenc_io_close(s, &oc->pb, oc->filename);
+ }
+ if (vs->vtt_avf) {
+ hlsenc_io_close(s, &vs->vtt_avf->pb, vs->vtt_avf->filename);
}
}
if ((hls->flags & HLS_TEMP_FILE) && oc->filename[0]) {
if (!(hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size <= 0))
- if ((hls->avf->oformat->priv_class && hls->avf->priv_data) && hls->segment_type != SEGMENT_TYPE_FMP4)
- av_opt_set(hls->avf->priv_data, "mpegts_flags", "resend_headers", 0);
+ if ((vs->avf->oformat->priv_class && vs->avf->priv_data) && hls->segment_type != SEGMENT_TYPE_FMP4)
+ av_opt_set(vs->avf->priv_data, "mpegts_flags", "resend_headers", 0);
hls_rename_temp_file(s, oc);
}
- if (hls->fmp4_init_mode) {
- hls->number--;
+ if (vs->fmp4_init_mode) {
+ vs->number--;
}
- if (!hls->fmp4_init_mode || byterange_mode)
- ret = hls_append_segment(s, hls, hls->duration, hls->start_pos, hls->size);
-
- hls->start_pos = new_start_pos;
+ if (hls->segment_type == SEGMENT_TYPE_FMP4) {
+ ret = hlsenc_io_open(s, &vs->out, vs->avf->filename, NULL);
+ if (ret < 0) {
+ av_log(NULL, AV_LOG_ERROR, "Failed to open file '%s'\n",
+ vs->avf->filename);
+ av_free(old_filename);
+ return ret;
+ }
+ write_styp(vs->out);
+ ret = flush_dynbuf(vs, &range_length);
+ if (ret < 0) {
+ av_free(old_filename);
+ return ret;
+ }
+ ff_format_io_close(s, &vs->out);
+ }
+ ret = hls_append_segment(s, hls, vs, vs->duration, vs->start_pos, vs->size);
+ vs->start_pos = new_start_pos;
if (ret < 0) {
av_free(old_filename);
return ret;
}
- hls->end_pts = pkt->pts;
- hls->duration = 0;
+ vs->end_pts = pkt->pts;
+ vs->duration = 0;
- hls->fmp4_init_mode = 0;
+ vs->fmp4_init_mode = 0;
if (hls->flags & HLS_SINGLE_FILE) {
- hls->number++;
+ vs->number++;
} else if (hls->max_seg_size > 0) {
- if (hls->start_pos >= hls->max_seg_size) {
- hls->sequence++;
- sls_flag_file_rename(hls, old_filename);
- ret = hls_start(s);
- hls->start_pos = 0;
+ if (vs->start_pos >= hls->max_seg_size) {
+ vs->sequence++;
+ sls_flag_file_rename(hls, vs, old_filename);
+ ret = hls_start(s, vs);
+ vs->start_pos = 0;
/* When split segment by byte, the duration is short than hls_time,
* so it is not enough one segment duration as hls_time, */
- hls->number--;
+ vs->number--;
}
- hls->number++;
+ vs->number++;
} else {
- sls_flag_file_rename(hls, old_filename);
- ret = hls_start(s);
+ sls_flag_file_rename(hls, vs, old_filename);
+ ret = hls_start(s, vs);
}
av_free(old_filename);
return ret;
}
- if (!hls->fmp4_init_mode || byterange_mode)
- if ((ret = hls_window(s, 0)) < 0) {
+ if (!vs->fmp4_init_mode || byterange_mode)
+ if ((ret = hls_window(s, 0, vs)) < 0) {
return ret;
}
}
+ vs->packets_written++;
ret = ff_write_chained(oc, stream_index, pkt, s, 0);
return ret;
static int hls_write_trailer(struct AVFormatContext *s)
{
HLSContext *hls = s->priv_data;
- AVFormatContext *oc = hls->avf;
- AVFormatContext *vtt_oc = hls->vtt_avf;
- char *old_filename = av_strdup(hls->avf->filename);
+ AVFormatContext *oc = NULL;
+ AVFormatContext *vtt_oc = NULL;
+ char *old_filename = NULL;
+ int i;
+ int ret = 0;
+ VariantStream *vs = NULL;
+
+ for (i = 0; i < hls->nb_varstreams; i++) {
+ vs = &hls->var_streams[i];
+
+ oc = vs->avf;
+ vtt_oc = vs->vtt_avf;
+ old_filename = av_strdup(vs->avf->filename);
if (!old_filename) {
return AVERROR(ENOMEM);
}
+ if ( hls->segment_type == SEGMENT_TYPE_FMP4) {
+ int range_length = 0;
+ ret = hlsenc_io_open(s, &vs->out, vs->avf->filename, NULL);
+ if (ret < 0) {
+ av_log(NULL, AV_LOG_ERROR, "Failed to open file '%s'\n", vs->avf->filename);
+ goto failed;
+ }
+ write_styp(vs->out);
+ ret = flush_dynbuf(vs, &range_length);
+ if (ret < 0) {
+ goto failed;
+ }
+ ff_format_io_close(s, &vs->out);
+ }
-
+failed:
av_write_trailer(oc);
if (oc->pb) {
- hls->size = avio_tell(hls->avf->pb) - hls->start_pos;
- ff_format_io_close(s, &oc->pb);
+ vs->size = avio_tell(vs->avf->pb) - vs->start_pos;
+ if (hls->segment_type != SEGMENT_TYPE_FMP4)
+ ff_format_io_close(s, &oc->pb);
if ((hls->flags & HLS_TEMP_FILE) && oc->filename[0]) {
hls_rename_temp_file(s, oc);
}
/* after av_write_trailer, then duration + 1 duration per packet */
- hls_append_segment(s, hls, hls->duration + hls->dpp, hls->start_pos, hls->size);
+ hls_append_segment(s, hls, vs, vs->duration + vs->dpp, vs->start_pos, vs->size);
}
- sls_flag_file_rename(hls, old_filename);
+ sls_flag_file_rename(hls, vs, old_filename);
if (vtt_oc) {
if (vtt_oc->pb)
av_write_trailer(vtt_oc);
- hls->size = avio_tell(hls->vtt_avf->pb) - hls->start_pos;
+ vs->size = avio_tell(vs->vtt_avf->pb) - vs->start_pos;
ff_format_io_close(s, &vtt_oc->pb);
}
- av_freep(&hls->basename);
- av_freep(&hls->base_output_dirname);
- av_freep(&hls->key_basename);
+ av_freep(&vs->basename);
+ av_freep(&vs->base_output_dirname);
avformat_free_context(oc);
- hls->avf = NULL;
- hls_window(s, 1);
+ vs->avf = NULL;
+ hls_window(s, 1, vs);
- av_freep(&hls->fmp4_init_filename);
+ av_freep(&vs->fmp4_init_filename);
if (vtt_oc) {
- av_freep(&hls->vtt_basename);
- av_freep(&hls->vtt_m3u8_name);
+ av_freep(&vs->vtt_basename);
+ av_freep(&vs->vtt_m3u8_name);
avformat_free_context(vtt_oc);
}
- hls_free_segments(hls->segments);
- hls_free_segments(hls->old_segments);
+ hls_free_segments(vs->segments);
+ hls_free_segments(vs->old_segments);
av_free(old_filename);
+ av_freep(&vs->m3u8_name);
+ av_freep(&vs->streams);
+ av_freep(&vs->agroup);
+ av_freep(&vs->baseurl);
+ }
+
+ ff_format_io_close(s, &hls->m3u8_out);
+ ff_format_io_close(s, &hls->sub_m3u8_out);
+ av_freep(&hls->key_basename);
+ av_freep(&hls->var_streams);
+ av_freep(&hls->master_m3u8_url);
return 0;
}
+
+static int hls_init(AVFormatContext *s)
+{
+ int ret = 0;
+ int i = 0;
+ int j = 0;
+ HLSContext *hls = s->priv_data;
+ const char *pattern = "%d.ts";
+ VariantStream *vs = NULL;
+ int basename_size = 0;
+ const char *pattern_localtime_fmt = get_default_pattern_localtime_fmt(s);
+ const char *vtt_pattern = "%d.vtt";
+ char *p = NULL;
+ int vtt_basename_size = 0;
+ int fmp4_init_filename_len = strlen(hls->fmp4_init_filename) + 1;
+
+ ret = update_variant_stream_info(s);
+ if (ret < 0) {
+ av_log(s, AV_LOG_ERROR, "Variant stream info update failed with status %x\n",
+ ret);
+ goto fail;
+ }
+ //TODO: Updates needed to encryption functionality with periodic re-key when more than one variant streams are present
+ if (hls->nb_varstreams > 1 && hls->flags & HLS_PERIODIC_REKEY) {
+ ret = AVERROR(EINVAL);
+ av_log(s, AV_LOG_ERROR, "Periodic re-key not supported when more than one variant streams are present\n");
+ goto fail;
+ }
+
+ ret = validate_name(hls->nb_varstreams, s->filename);
+ if (ret < 0)
+ goto fail;
+
+ if (hls->segment_filename) {
+ ret = validate_name(hls->nb_varstreams, hls->segment_filename);
+ if (ret < 0)
+ goto fail;
+ }
+
+ if (av_strcasecmp(hls->fmp4_init_filename, "init.mp4")) {
+ ret = validate_name(hls->nb_varstreams, hls->fmp4_init_filename);
+ if (ret < 0)
+ goto fail;
+ }
+
+ if (hls->subtitle_filename) {
+ ret = validate_name(hls->nb_varstreams, hls->subtitle_filename);
+ if (ret < 0)
+ goto fail;
+ }
+
+ if (hls->master_pl_name) {
+ ret = update_master_pl_info(s);
+ if (ret < 0) {
+ av_log(s, AV_LOG_ERROR, "Master stream info update failed with status %x\n",
+ ret);
+ goto fail;
+ }
+ }
+
+ if (hls->segment_type == SEGMENT_TYPE_FMP4) {
+ pattern = "%d.m4s";
+ }
+ if ((hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH) ||
+ (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_FORMATTED_DATETIME)) {
+ time_t t = time(NULL); // we will need it in either case
+ if (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH) {
+ hls->start_sequence = (int64_t)t;
+ } else if (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_FORMATTED_DATETIME) {
+ char b[15];
+ struct tm *p, tmbuf;
+ if (!(p = localtime_r(&t, &tmbuf)))
+ return AVERROR(ENOMEM);
+ if (!strftime(b, sizeof(b), "%Y%m%d%H%M%S", p))
+ return AVERROR(ENOMEM);
+ hls->start_sequence = strtoll(b, NULL, 10);
+ }
+ av_log(hls, AV_LOG_DEBUG, "start_number evaluated to %"PRId64"\n", hls->start_sequence);
+ }
+
+ hls->recording_time = (hls->init_time ? hls->init_time : hls->time) * AV_TIME_BASE;
+ for (i = 0; i < hls->nb_varstreams; i++) {
+ vs = &hls->var_streams[i];
+
+ vs->m3u8_name = av_strdup(s->filename);
+ if (!vs->m3u8_name ) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ ret = format_name(vs->m3u8_name, strlen(s->filename) + 1, i);
+ if (ret < 0)
+ goto fail;
+
+ vs->sequence = hls->start_sequence;
+ vs->start_pts = AV_NOPTS_VALUE;
+ vs->end_pts = AV_NOPTS_VALUE;
+ vs->current_segment_final_filename_fmt[0] = '\0';
+
+ if (hls->flags & HLS_SPLIT_BY_TIME && hls->flags & HLS_INDEPENDENT_SEGMENTS) {
+ // Independent segments cannot be guaranteed when splitting by time
+ hls->flags &= ~HLS_INDEPENDENT_SEGMENTS;
+ av_log(s, AV_LOG_WARNING,
+ "'split_by_time' and 'independent_segments' cannot be enabled together. "
+ "Disabling 'independent_segments' flag\n");
+ }
+
+ if (hls->flags & HLS_PROGRAM_DATE_TIME) {
+ time_t now0;
+ time(&now0);
+ vs->initial_prog_date_time = now0;
+ }
+ if (hls->format_options_str) {
+ ret = av_dict_parse_string(&hls->format_options, hls->format_options_str, "=", ":", 0);
+ if (ret < 0) {
+ av_log(s, AV_LOG_ERROR, "Could not parse format options list '%s'\n", hls->format_options_str);
+ goto fail;
+ }
+ }
+
+ for (j = 0; j < vs->nb_streams; j++) {
+ vs->has_video += vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO;
+ vs->has_subtitle += vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE;
+ }
+
+ if (vs->has_video > 1)
+ av_log(s, AV_LOG_WARNING, "More than a single video stream present, expect issues decoding it.\n");
+ if (hls->segment_type == SEGMENT_TYPE_FMP4) {
+ vs->oformat = av_guess_format("mp4", NULL, NULL);
+ } else {
+ vs->oformat = av_guess_format("mpegts", NULL, NULL);
+ }
+
+ if (!vs->oformat) {
+ ret = AVERROR_MUXER_NOT_FOUND;
+ goto fail;
+ }
+
+ if (vs->has_subtitle) {
+ vs->vtt_oformat = av_guess_format("webvtt", NULL, NULL);
+ if (!vs->oformat) {
+ ret = AVERROR_MUXER_NOT_FOUND;
+ goto fail;
+ }
+ }
+ if (hls->segment_filename) {
+ basename_size = strlen(hls->segment_filename) + 1;
+ vs->basename = av_malloc(basename_size);
+ if (!vs->basename) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+
+ av_strlcpy(vs->basename, hls->segment_filename, basename_size);
+ ret = format_name(vs->basename, basename_size, i);
+ if (ret < 0)
+ goto fail;
+ } else {
+ if (hls->flags & HLS_SINGLE_FILE) {
+ if (hls->segment_type == SEGMENT_TYPE_FMP4) {
+ pattern = ".m4s";
+ } else {
+ pattern = ".ts";
+ }
+ }
+
+ if (hls->use_localtime) {
+ basename_size = strlen(vs->m3u8_name) + strlen(pattern_localtime_fmt) + 1;
+ } else {
+ basename_size = strlen(vs->m3u8_name) + strlen(pattern) + 1;
+ }
+
+ vs->basename = av_malloc(basename_size);
+ if (!vs->basename) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+
+ av_strlcpy(vs->basename, vs->m3u8_name, basename_size);
+
+ p = strrchr(vs->basename, '.');
+ if (p)
+ *p = '\0';
+ if (hls->use_localtime) {
+ av_strlcat(vs->basename, pattern_localtime_fmt, basename_size);
+ } else {
+ av_strlcat(vs->basename, pattern, basename_size);
+ }
+ }
+
+ if (hls->segment_type == SEGMENT_TYPE_FMP4) {
+ if (hls->nb_varstreams > 1)
+ fmp4_init_filename_len += strlen(POSTFIX_PATTERN);
+ vs->fmp4_init_filename = av_malloc(fmp4_init_filename_len);
+ if (!vs->fmp4_init_filename ) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ av_strlcpy(vs->fmp4_init_filename, hls->fmp4_init_filename,
+ fmp4_init_filename_len);
+
+ if (av_strcasecmp(hls->fmp4_init_filename, "init.mp4")) {
+ ret = format_name(vs->fmp4_init_filename, fmp4_init_filename_len, i);
+ if (ret < 0)
+ goto fail;
+
+ fmp4_init_filename_len = strlen(vs->fmp4_init_filename) + 1;
+ vs->base_output_dirname = av_malloc(fmp4_init_filename_len);
+ if (!vs->base_output_dirname) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ av_strlcpy(vs->base_output_dirname, vs->fmp4_init_filename,
+ fmp4_init_filename_len);
+ } else {
+ if (hls->nb_varstreams > 1) {
+ ret = append_postfix(vs->fmp4_init_filename, fmp4_init_filename_len, i);
+ if (ret < 0)
+ goto fail;
+ }
+
+ fmp4_init_filename_len = strlen(vs->m3u8_name) +
+ strlen(vs->fmp4_init_filename) + 1;
+
+ vs->base_output_dirname = av_malloc(fmp4_init_filename_len);
+ if (!vs->base_output_dirname) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+
+ av_strlcpy(vs->base_output_dirname, vs->m3u8_name,
+ fmp4_init_filename_len);
+ p = strrchr(vs->base_output_dirname, '/');
+ if (p) {
+ *(p + 1) = '\0';
+ av_strlcat(vs->base_output_dirname, vs->fmp4_init_filename,
+ fmp4_init_filename_len);
+ } else {
+ av_strlcpy(vs->base_output_dirname, vs->fmp4_init_filename,
+ fmp4_init_filename_len);
+ }
+ }
+ }
+
+ if (!hls->use_localtime) {
+ ret = sls_flag_check_duration_size_index(hls);
+ if (ret < 0) {
+ goto fail;
+ }
+ } else {
+ ret = sls_flag_check_duration_size(hls, vs);
+ if (ret < 0) {
+ goto fail;
+ }
+ }
+ if (vs->has_subtitle) {
+
+ if (hls->flags & HLS_SINGLE_FILE)
+ vtt_pattern = ".vtt";
+ vtt_basename_size = strlen(vs->m3u8_name) + strlen(vtt_pattern) + 1;
+
+ vs->vtt_basename = av_malloc(vtt_basename_size);
+ if (!vs->vtt_basename) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ vs->vtt_m3u8_name = av_malloc(vtt_basename_size);
+ if (!vs->vtt_m3u8_name ) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ av_strlcpy(vs->vtt_basename, vs->m3u8_name, vtt_basename_size);
+ p = strrchr(vs->vtt_basename, '.');
+ if (p)
+ *p = '\0';
+
+ if ( hls->subtitle_filename ) {
+ strcpy(vs->vtt_m3u8_name, hls->subtitle_filename);
+ ret = format_name(vs->vtt_m3u8_name, vtt_basename_size, i);
+ if (ret < 0)
+ goto fail;
+ } else {
+ strcpy(vs->vtt_m3u8_name, vs->vtt_basename);
+ av_strlcat(vs->vtt_m3u8_name, "_vtt.m3u8", vtt_basename_size);
+ }
+ av_strlcat(vs->vtt_basename, vtt_pattern, vtt_basename_size);
+ }
+
+ if (hls->baseurl) {
+ vs->baseurl = av_strdup(hls->baseurl);
+ if (!vs->baseurl) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ }
+
+ if ((hls->flags & HLS_SINGLE_FILE) && (hls->segment_type == SEGMENT_TYPE_FMP4)) {
+ vs->fmp4_init_filename = av_strdup(vs->basename);
+ if (!vs->fmp4_init_filename) {
+ ret = AVERROR(ENOMEM);
+ goto fail;
+ }
+ }
+ if ((ret = hls_mux_init(s, vs)) < 0)
+ goto fail;
+
+ if (hls->flags & HLS_APPEND_LIST) {
+ parse_playlist(s, vs->m3u8_name, vs);
+ vs->discontinuity = 1;
+ if (hls->init_time > 0) {
+ av_log(s, AV_LOG_WARNING, "append_list mode does not support hls_init_time,"
+ " hls_init_time value will have no effect\n");
+ hls->init_time = 0;
+ hls->recording_time = hls->time * AV_TIME_BASE;
+ }
+ }
+
+ if ((ret = hls_start(s, vs)) < 0)
+ goto fail;
+ }
+
+fail:
+ if (ret < 0) {
+ av_freep(&hls->key_basename);
+ for (i = 0; i < hls->nb_varstreams && hls->var_streams; i++) {
+ vs = &hls->var_streams[i];
+ av_freep(&vs->basename);
+ av_freep(&vs->vtt_basename);
+ av_freep(&vs->fmp4_init_filename);
+ av_freep(&vs->m3u8_name);
+ av_freep(&vs->vtt_m3u8_name);
+ av_freep(&vs->streams);
+ av_freep(&vs->agroup);
+ av_freep(&vs->baseurl);
+ if (vs->avf)
+ avformat_free_context(vs->avf);
+ if (vs->vtt_avf)
+ avformat_free_context(vs->vtt_avf);
+ }
+ av_freep(&hls->var_streams);
+ av_freep(&hls->master_m3u8_url);
+ }
+
+ return ret;
+}
+
#define OFFSET(x) offsetof(HLSContext, x)
#define E AV_OPT_FLAG_ENCODING_PARAM
static const AVOption options[] = {
{"second_level_segment_duration", "include segment duration in segment filenames when use_localtime", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_SECOND_LEVEL_SEGMENT_DURATION }, 0, UINT_MAX, E, "flags"},
{"second_level_segment_size", "include segment size in segment filenames when use_localtime", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_SECOND_LEVEL_SEGMENT_SIZE }, 0, UINT_MAX, E, "flags"},
{"periodic_rekey", "reload keyinfo file periodically for re-keying", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_PERIODIC_REKEY }, 0, UINT_MAX, E, "flags"},
+ {"independent_segments", "add EXT-X-INDEPENDENT-SEGMENTS, whenever applicable", 0, AV_OPT_TYPE_CONST, { .i64 = HLS_INDEPENDENT_SEGMENTS }, 0, UINT_MAX, E, "flags"},
{"use_localtime", "set filename expansion with strftime at segment creation", OFFSET(use_localtime), AV_OPT_TYPE_BOOL, {.i64 = 0 }, 0, 1, E },
{"use_localtime_mkdir", "create last directory component in strftime-generated filename", OFFSET(use_localtime_mkdir), AV_OPT_TYPE_BOOL, {.i64 = 0 }, 0, 1, E },
{"hls_playlist_type", "set the HLS playlist type", OFFSET(pl_type), AV_OPT_TYPE_INT, {.i64 = PLAYLIST_TYPE_NONE }, 0, PLAYLIST_TYPE_NB-1, E, "pl_type" },
{"epoch", "seconds since epoch", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH }, INT_MIN, INT_MAX, E, "start_sequence_source_type" },
{"datetime", "current datetime as YYYYMMDDhhmmss", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_START_SEQUENCE_AS_FORMATTED_DATETIME }, INT_MIN, INT_MAX, E, "start_sequence_source_type" },
{"http_user_agent", "override User-Agent field in HTTP header", OFFSET(user_agent), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E},
+ {"var_stream_map", "Variant stream map string", OFFSET(var_stream_map), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E},
+ {"master_pl_name", "Create HLS master playlist with this name", OFFSET(master_pl_name), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E},
+ {"master_pl_publish_rate", "Publish master play list every after this many segment intervals", OFFSET(master_publish_rate), AV_OPT_TYPE_INT, {.i64 = 0}, 0, UINT_MAX, E},
+ {"http_persistent", "Use persistent HTTP connections", OFFSET(http_persistent), AV_OPT_TYPE_BOOL, {.i64 = 0 }, 0, 1, E },
{ NULL },
};
.video_codec = AV_CODEC_ID_H264,
.subtitle_codec = AV_CODEC_ID_WEBVTT,
.flags = AVFMT_NOFILE | AVFMT_GLOBALHEADER | AVFMT_ALLOW_FLUSH,
+ .init = hls_init,
.write_header = hls_write_header,
.write_packet = hls_write_packet,
.write_trailer = hls_write_trailer,