| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | /* | ||
| 2 | * Apple HTTP Live Streaming segmenter | ||
| 3 | * Copyright (c) 2012, Luca Barbato | ||
| 4 | * Copyright (c) 2017 Akamai Technologies, Inc. | ||
| 5 | * | ||
| 6 | * This file is part of FFmpeg. | ||
| 7 | * | ||
| 8 | * FFmpeg is free software; you can redistribute it and/or | ||
| 9 | * modify it under the terms of the GNU Lesser General Public | ||
| 10 | * License as published by the Free Software Foundation; either | ||
| 11 | * version 2.1 of the License, or (at your option) any later version. | ||
| 12 | * | ||
| 13 | * FFmpeg is distributed in the hope that it will be useful, | ||
| 14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | ||
| 16 | * Lesser General Public License for more details. | ||
| 17 | * | ||
| 18 | * You should have received a copy of the GNU Lesser General Public | ||
| 19 | * License along with FFmpeg; if not, write to the Free Software | ||
| 20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | ||
| 21 | */ | ||
| 22 | |||
| 23 | #include "config.h" | ||
| 24 | #include "config_components.h" | ||
| 25 | #include <stdint.h> | ||
| 26 | #include <time.h> | ||
| 27 | #if HAVE_UNISTD_H | ||
| 28 | #include <unistd.h> | ||
| 29 | #endif | ||
| 30 | |||
| 31 | #include "libavutil/attributes_internal.h" | ||
| 32 | #include "libavutil/avassert.h" | ||
| 33 | #include "libavutil/mathematics.h" | ||
| 34 | #include "libavutil/avstring.h" | ||
| 35 | #include "libavutil/bprint.h" | ||
| 36 | #include "libavutil/intreadwrite.h" | ||
| 37 | #include "libavutil/mem.h" | ||
| 38 | #include "libavutil/opt.h" | ||
| 39 | #include "libavutil/log.h" | ||
| 40 | #include "libavutil/random_seed.h" | ||
| 41 | #include "libavutil/time.h" | ||
| 42 | #include "libavutil/time_internal.h" | ||
| 43 | |||
| 44 | #include "libavcodec/defs.h" | ||
| 45 | |||
| 46 | #include "avformat.h" | ||
| 47 | #include "avio_internal.h" | ||
| 48 | #if CONFIG_HTTP_PROTOCOL | ||
| 49 | #include "http.h" | ||
| 50 | #endif | ||
| 51 | #include "hlsplaylist.h" | ||
| 52 | #include "internal.h" | ||
| 53 | #include "mux.h" | ||
| 54 | #include "os_support.h" | ||
| 55 | #include "url.h" | ||
| 56 | |||
| 57 | typedef enum { | ||
| 58 | HLS_START_SEQUENCE_AS_START_NUMBER = 0, | ||
| 59 | HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH = 1, | ||
| 60 | HLS_START_SEQUENCE_AS_FORMATTED_DATETIME = 2, // YYYYMMDDhhmmss | ||
| 61 | HLS_START_SEQUENCE_AS_MICROSECONDS_SINCE_EPOCH = 3, | ||
| 62 | HLS_START_SEQUENCE_LAST, // unused | ||
| 63 | } StartSequenceSourceType; | ||
| 64 | |||
| 65 | typedef enum { | ||
| 66 | CODEC_ATTRIBUTE_WRITTEN = 0, | ||
| 67 | CODEC_ATTRIBUTE_WILL_NOT_BE_WRITTEN, | ||
| 68 | } CodecAttributeStatus; | ||
| 69 | |||
| 70 | #define KEYSIZE 16 | ||
| 71 | #define LINE_BUFFER_SIZE MAX_URL_SIZE | ||
| 72 | #define HLS_MICROSECOND_UNIT 1000000 | ||
| 73 | #define BUFSIZE (16 * 1024) | ||
| 74 | #define POSTFIX_PATTERN "_%d" | ||
| 75 | |||
| 76 | typedef struct HLSSegment { | ||
| 77 | char filename[MAX_URL_SIZE]; | ||
| 78 | char sub_filename[MAX_URL_SIZE]; | ||
| 79 | double duration; /* in seconds */ | ||
| 80 | int discont; | ||
| 81 | int64_t pos; | ||
| 82 | int64_t size; | ||
| 83 | int64_t keyframe_pos; | ||
| 84 | int64_t keyframe_size; | ||
| 85 | unsigned var_stream_idx; | ||
| 86 | |||
| 87 | char key_uri[LINE_BUFFER_SIZE + 1]; | ||
| 88 | char iv_string[KEYSIZE*2 + 1]; | ||
| 89 | |||
| 90 | struct HLSSegment *next; | ||
| 91 | double discont_program_date_time; | ||
| 92 | } HLSSegment; | ||
| 93 | |||
| 94 | typedef enum HLSFlags { | ||
| 95 | // Generate a single media file and use byte ranges in the playlist. | ||
| 96 | HLS_SINGLE_FILE = (1 << 0), | ||
| 97 | HLS_DELETE_SEGMENTS = (1 << 1), | ||
| 98 | HLS_ROUND_DURATIONS = (1 << 2), | ||
| 99 | HLS_DISCONT_START = (1 << 3), | ||
| 100 | HLS_OMIT_ENDLIST = (1 << 4), | ||
| 101 | HLS_SPLIT_BY_TIME = (1 << 5), | ||
| 102 | HLS_APPEND_LIST = (1 << 6), | ||
| 103 | HLS_PROGRAM_DATE_TIME = (1 << 7), | ||
| 104 | HLS_SECOND_LEVEL_SEGMENT_INDEX = (1 << 8), // include segment index in segment filenames when use_localtime e.g.: %%03d | ||
| 105 | HLS_SECOND_LEVEL_SEGMENT_DURATION = (1 << 9), // include segment duration (microsec) in segment filenames when use_localtime e.g.: %%09t | ||
| 106 | HLS_SECOND_LEVEL_SEGMENT_SIZE = (1 << 10), // include segment size (bytes) in segment filenames when use_localtime e.g.: %%014s | ||
| 107 | HLS_TEMP_FILE = (1 << 11), | ||
| 108 | HLS_PERIODIC_REKEY = (1 << 12), | ||
| 109 | HLS_INDEPENDENT_SEGMENTS = (1 << 13), | ||
| 110 | HLS_I_FRAMES_ONLY = (1 << 14), | ||
| 111 | } HLSFlags; | ||
| 112 | |||
| 113 | typedef enum { | ||
| 114 | SEGMENT_TYPE_MPEGTS, | ||
| 115 | SEGMENT_TYPE_FMP4, | ||
| 116 | } SegmentType; | ||
| 117 | |||
| 118 | typedef struct VariantStream { | ||
| 119 | unsigned var_stream_idx; | ||
| 120 | unsigned number; | ||
| 121 | int64_t sequence; | ||
| 122 | const AVOutputFormat *oformat; | ||
| 123 | const AVOutputFormat *vtt_oformat; | ||
| 124 | AVIOContext *out; | ||
| 125 | AVIOContext *out_single_file; | ||
| 126 | int packets_written; | ||
| 127 | int init_range_length; | ||
| 128 | uint8_t *temp_buffer; | ||
| 129 | uint8_t *init_buffer; | ||
| 130 | |||
| 131 | AVFormatContext *avf; | ||
| 132 | AVFormatContext *vtt_avf; | ||
| 133 | |||
| 134 | int has_video; | ||
| 135 | int has_subtitle; | ||
| 136 | int new_start; | ||
| 137 | int start_pts_from_audio; | ||
| 138 | double dpp; // duration per packet | ||
| 139 | int64_t start_pts; | ||
| 140 | int64_t end_pts; | ||
| 141 | int64_t video_lastpos; | ||
| 142 | int64_t video_keyframe_pos; | ||
| 143 | int64_t video_keyframe_size; | ||
| 144 | double duration; // last segment duration computed so far, in seconds | ||
| 145 | int64_t start_pos; // last segment starting position | ||
| 146 | int64_t size; // last segment size | ||
| 147 | int nb_entries; | ||
| 148 | int discontinuity_set; | ||
| 149 | int discontinuity; | ||
| 150 | int reference_stream_index; | ||
| 151 | |||
| 152 | int64_t total_size; | ||
| 153 | double total_duration; | ||
| 154 | int64_t avg_bitrate; | ||
| 155 | int64_t max_bitrate; | ||
| 156 | |||
| 157 | HLSSegment *segments; | ||
| 158 | HLSSegment *last_segment; | ||
| 159 | HLSSegment *old_segments; | ||
| 160 | |||
| 161 | char *basename_tmp; | ||
| 162 | char *basename; | ||
| 163 | char *vtt_basename; | ||
| 164 | char *vtt_m3u8_name; | ||
| 165 | char *m3u8_name; | ||
| 166 | |||
| 167 | double initial_prog_date_time; | ||
| 168 | char current_segment_final_filename_fmt[MAX_URL_SIZE]; // when renaming segments | ||
| 169 | |||
| 170 | char *fmp4_init_filename; | ||
| 171 | char *base_output_dirname; | ||
| 172 | |||
| 173 | int encrypt_started; | ||
| 174 | |||
| 175 | char key_file[LINE_BUFFER_SIZE + 1]; | ||
| 176 | char key_uri[LINE_BUFFER_SIZE + 1]; | ||
| 177 | char key_string[KEYSIZE*2 + 1]; | ||
| 178 | char iv_string[KEYSIZE*2 + 1]; | ||
| 179 | |||
| 180 | AVStream **streams; | ||
| 181 | char codec_attr[128]; | ||
| 182 | CodecAttributeStatus attr_status; | ||
| 183 | unsigned int nb_streams; | ||
| 184 | int m3u8_created; /* status of media play-list creation */ | ||
| 185 | int is_default; /* default status of audio group */ | ||
| 186 | const char *language; /* audio language name */ | ||
| 187 | const char *agroup; /* audio group name */ | ||
| 188 | const char *sgroup; /* subtitle group name */ | ||
| 189 | const char *ccgroup; /* closed caption group name */ | ||
| 190 | const char *varname; /* variant name */ | ||
| 191 | const char *subtitle_varname; /* subtitle variant name */ | ||
| 192 | } VariantStream; | ||
| 193 | |||
| 194 | typedef struct ClosedCaptionsStream { | ||
| 195 | const char *ccgroup; /* closed caption group name */ | ||
| 196 | const char *instreamid; /* closed captions INSTREAM-ID */ | ||
| 197 | const char *language; /* closed captions language */ | ||
| 198 | } ClosedCaptionsStream; | ||
| 199 | |||
| 200 | typedef struct HLSContext { | ||
| 201 | const AVClass *class; // Class for private options. | ||
| 202 | int64_t start_sequence; | ||
| 203 | uint32_t start_sequence_source_type; // enum StartSequenceSourceType | ||
| 204 | |||
| 205 | int64_t time; // Set by a private option. | ||
| 206 | int64_t init_time; // Set by a private option. | ||
| 207 | int max_nb_segments; // Set by a private option. | ||
| 208 | int hls_delete_threshold; // Set by a private option. | ||
| 209 | uint32_t flags; // enum HLSFlags | ||
| 210 | uint32_t pl_type; // enum PlaylistType | ||
| 211 | char *segment_filename; | ||
| 212 | char *fmp4_init_filename; | ||
| 213 | int segment_type; | ||
| 214 | int resend_init_file; ///< resend init file into disk after refresh m3u8 | ||
| 215 | |||
| 216 | int use_localtime; ///< flag to expand filename with localtime | ||
| 217 | int use_localtime_mkdir;///< flag to mkdir dirname in timebased filename | ||
| 218 | int allowcache; | ||
| 219 | int64_t recording_time; | ||
| 220 | int64_t max_seg_size; // every segment file max size | ||
| 221 | |||
| 222 | char *baseurl; | ||
| 223 | char *vtt_format_options_str; | ||
| 224 | char *subtitle_filename; | ||
| 225 | AVDictionary *format_options; | ||
| 226 | |||
| 227 | int encrypt; | ||
| 228 | char *key; | ||
| 229 | char *key_url; | ||
| 230 | char *iv; | ||
| 231 | char *key_basename; | ||
| 232 | int encrypt_started; | ||
| 233 | |||
| 234 | char *key_info_file; | ||
| 235 | char key_file[LINE_BUFFER_SIZE + 1]; | ||
| 236 | char key_uri[LINE_BUFFER_SIZE + 1]; | ||
| 237 | char key_string[KEYSIZE*2 + 1]; | ||
| 238 | char iv_string[KEYSIZE*2 + 1]; | ||
| 239 | AVDictionary *vtt_format_options; | ||
| 240 | |||
| 241 | char *method; | ||
| 242 | char *user_agent; | ||
| 243 | |||
| 244 | VariantStream *var_streams; | ||
| 245 | unsigned int nb_varstreams; | ||
| 246 | ClosedCaptionsStream *cc_streams; | ||
| 247 | unsigned int nb_ccstreams; | ||
| 248 | |||
| 249 | int master_m3u8_created; /* status of master play-list creation */ | ||
| 250 | char *master_m3u8_url; /* URL of the master m3u8 file */ | ||
| 251 | int version; /* HLS version */ | ||
| 252 | char *var_stream_map; /* user specified variant stream map string */ | ||
| 253 | char *cc_stream_map; /* user specified closed caption streams map string */ | ||
| 254 | char *master_pl_name; | ||
| 255 | unsigned int master_publish_rate; | ||
| 256 | int http_persistent; | ||
| 257 | AVIOContext *m3u8_out; | ||
| 258 | AVIOContext *sub_m3u8_out; | ||
| 259 | AVIOContext *http_delete; | ||
| 260 | int64_t timeout; | ||
| 261 | int ignore_io_errors; | ||
| 262 | char *headers; | ||
| 263 | int has_default_key; /* has DEFAULT field of var_stream_map */ | ||
| 264 | int has_video_m3u8; /* has video stream m3u8 list */ | ||
| 265 | } HLSContext; | ||
| 266 | |||
| 267 | ✗ | static int strftime_expand(const char *fmt, char **dest) | |
| 268 | { | ||
| 269 | ✗ | int r = 1; | |
| 270 | time_t now0; | ||
| 271 | struct tm *tm, tmpbuf; | ||
| 272 | char *buf; | ||
| 273 | |||
| 274 | ✗ | buf = av_mallocz(MAX_URL_SIZE); | |
| 275 | ✗ | if (!buf) | |
| 276 | ✗ | return AVERROR(ENOMEM); | |
| 277 | |||
| 278 | ✗ | time(&now0); | |
| 279 | ✗ | tm = localtime_r(&now0, &tmpbuf); | |
| 280 | ✗ | r = strftime(buf, MAX_URL_SIZE, fmt, tm); | |
| 281 | ✗ | if (!r) { | |
| 282 | ✗ | av_free(buf); | |
| 283 | ✗ | return AVERROR(EINVAL); | |
| 284 | } | ||
| 285 | ✗ | *dest = buf; | |
| 286 | |||
| 287 | ✗ | return r; | |
| 288 | } | ||
| 289 | |||
| 290 | 170 | static int hlsenc_io_open(AVFormatContext *s, AVIOContext **pb, const char *filename, | |
| 291 | AVDictionary **options) | ||
| 292 | { | ||
| 293 | 170 | HLSContext *hls = s->priv_data; | |
| 294 |
1/2✓ Branch 0 taken 170 times.
✗ Branch 1 not taken.
|
170 | int http_base_proto = filename ? ff_is_http_proto(filename) : 0; |
| 295 | 170 | int err = AVERROR_MUXER_NOT_FOUND; | |
| 296 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 170 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
170 | if (!*pb || !http_base_proto || !hls->http_persistent) { |
| 297 | 170 | err = s->io_open(s, pb, filename, AVIO_FLAG_WRITE, options); | |
| 298 | #if CONFIG_HTTP_PROTOCOL | ||
| 299 | } else { | ||
| 300 | ✗ | URLContext *http_url_context = ffio_geturlcontext(*pb); | |
| 301 | ✗ | av_assert0(http_url_context); | |
| 302 | ✗ | err = ff_http_do_new_request(http_url_context, filename); | |
| 303 | ✗ | if (err < 0) | |
| 304 | ✗ | ff_format_io_close(s, pb); | |
| 305 | |||
| 306 | #endif | ||
| 307 | } | ||
| 308 | 170 | return err; | |
| 309 | } | ||
| 310 | |||
| 311 | 260 | static int hlsenc_io_close(AVFormatContext *s, AVIOContext **pb, char *filename) | |
| 312 | { | ||
| 313 | 260 | HLSContext *hls = s->priv_data; | |
| 314 |
2/2✓ Branch 0 taken 170 times.
✓ Branch 1 taken 90 times.
|
260 | int http_base_proto = filename ? ff_is_http_proto(filename) : 0; |
| 315 | 260 | int ret = 0; | |
| 316 |
2/2✓ Branch 0 taken 91 times.
✓ Branch 1 taken 169 times.
|
260 | if (!*pb) |
| 317 | 91 | return ret; | |
| 318 |
1/8✗ Branch 0 not taken.
✓ Branch 1 taken 169 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
|
169 | if (!http_base_proto || !hls->http_persistent || hls->key_info_file || hls->encrypt) { |
| 319 | 169 | ff_format_io_close(s, pb); | |
| 320 | #if CONFIG_HTTP_PROTOCOL | ||
| 321 | } else { | ||
| 322 | ✗ | URLContext *http_url_context = ffio_geturlcontext(*pb); | |
| 323 | ✗ | av_assert0(http_url_context); | |
| 324 | ✗ | avio_flush(*pb); | |
| 325 | ✗ | ret = ffurl_shutdown(http_url_context, AVIO_FLAG_WRITE); | |
| 326 | #endif | ||
| 327 | } | ||
| 328 | 169 | return ret; | |
| 329 | } | ||
| 330 | |||
| 331 | 169 | static void set_http_options(AVFormatContext *s, AVDictionary **options, HLSContext *c) | |
| 332 | { | ||
| 333 | 169 | int http_base_proto = ff_is_http_proto(s->url); | |
| 334 | |||
| 335 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 169 times.
|
169 | if (c->method) { |
| 336 | ✗ | av_dict_set(options, "method", c->method, 0); | |
| 337 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 169 times.
|
169 | } else if (http_base_proto) { |
| 338 | ✗ | av_dict_set(options, "method", "PUT", 0); | |
| 339 | } | ||
| 340 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 169 times.
|
169 | if (c->user_agent) |
| 341 | ✗ | av_dict_set(options, "user_agent", c->user_agent, 0); | |
| 342 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 169 times.
|
169 | if (c->http_persistent) |
| 343 | ✗ | av_dict_set_int(options, "multiple_requests", 1, 0); | |
| 344 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 169 times.
|
169 | if (c->timeout >= 0) |
| 345 | ✗ | av_dict_set_int(options, "timeout", c->timeout, 0); | |
| 346 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 169 times.
|
169 | if (c->headers) |
| 347 | ✗ | av_dict_set(options, "headers", c->headers, 0); | |
| 348 | 169 | } | |
| 349 | |||
| 350 | 18 | static void write_codec_attr(AVStream *st, VariantStream *vs) | |
| 351 | { | ||
| 352 | 18 | int codec_strlen = strlen(vs->codec_attr); | |
| 353 | char attr[32]; | ||
| 354 | AVBPrint buffer; | ||
| 355 | |||
| 356 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE) |
| 357 | ✗ | return; | |
| 358 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (vs->attr_status == CODEC_ATTRIBUTE_WILL_NOT_BE_WRITTEN) |
| 359 | ✗ | return; | |
| 360 | |||
| 361 | 18 | av_bprint_init_for_buffer(&buffer, attr, sizeof(attr)); | |
| 362 |
2/2✓ Branch 1 taken 6 times.
✓ Branch 2 taken 12 times.
|
18 | if (ff_make_codec_str(vs->avf, st->codecpar, &st->avg_frame_rate, |
| 363 | &buffer) < 0) | ||
| 364 | 6 | goto fail; | |
| 365 | |||
| 366 | // Don't write the same attribute multiple times | ||
| 367 |
1/2✓ Branch 1 taken 12 times.
✗ Branch 2 not taken.
|
12 | if (!av_stristr(vs->codec_attr, attr)) { |
| 368 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 12 times.
|
12 | snprintf(vs->codec_attr + codec_strlen, |
| 369 | sizeof(vs->codec_attr) - codec_strlen, | ||
| 370 | "%s%s", codec_strlen ? "," : "", attr); | ||
| 371 | } | ||
| 372 | 12 | return; | |
| 373 | |||
| 374 | 6 | fail: | |
| 375 | 6 | vs->codec_attr[0] = '\0'; | |
| 376 | 6 | vs->attr_status = CODEC_ATTRIBUTE_WILL_NOT_BE_WRITTEN; | |
| 377 | 6 | return; | |
| 378 | } | ||
| 379 | |||
| 380 | ✗ | static int replace_str_data_in_filename(char **s, const char *filename, char placeholder, const char *datastring) | |
| 381 | { | ||
| 382 | const char *p; | ||
| 383 | char c; | ||
| 384 | int addchar_count; | ||
| 385 | ✗ | int found_count = 0; | |
| 386 | AVBPrint buf; | ||
| 387 | int ret; | ||
| 388 | |||
| 389 | ✗ | av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |
| 390 | |||
| 391 | ✗ | p = filename; | |
| 392 | for (;;) { | ||
| 393 | ✗ | c = *p; | |
| 394 | ✗ | if (c == '\0') | |
| 395 | ✗ | break; | |
| 396 | ✗ | if (c == '%' && *(p+1) == '%') // %% | |
| 397 | ✗ | addchar_count = 2; | |
| 398 | ✗ | else if (c == '%' && *(p+1) == placeholder) { | |
| 399 | ✗ | av_bprintf(&buf, "%s", datastring); | |
| 400 | ✗ | p += 2; | |
| 401 | ✗ | addchar_count = 0; | |
| 402 | ✗ | found_count ++; | |
| 403 | } else | ||
| 404 | ✗ | addchar_count = 1; | |
| 405 | |||
| 406 | ✗ | if (addchar_count > 0) { | |
| 407 | ✗ | av_bprint_append_data(&buf, p, addchar_count); | |
| 408 | ✗ | p += addchar_count; | |
| 409 | } | ||
| 410 | } | ||
| 411 | ✗ | if (!av_bprint_is_complete(&buf)) { | |
| 412 | ✗ | av_bprint_finalize(&buf, NULL); | |
| 413 | ✗ | return AVERROR(ENOMEM); | |
| 414 | } | ||
| 415 | ✗ | if ((ret = av_bprint_finalize(&buf, s)) < 0) | |
| 416 | ✗ | return ret; | |
| 417 | ✗ | return found_count; | |
| 418 | } | ||
| 419 | |||
| 420 | 76 | static int replace_int_data_in_filename(char **s, const char *filename, char placeholder, int64_t number) | |
| 421 | { | ||
| 422 | const char *p; | ||
| 423 | char c; | ||
| 424 | int nd, addchar_count; | ||
| 425 | 76 | int found_count = 0; | |
| 426 | AVBPrint buf; | ||
| 427 | int ret; | ||
| 428 | |||
| 429 | 76 | av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); | |
| 430 | |||
| 431 | 76 | p = filename; | |
| 432 | for (;;) { | ||
| 433 | 6709 | c = *p; | |
| 434 |
2/2✓ Branch 0 taken 76 times.
✓ Branch 1 taken 6633 times.
|
6709 | if (c == '\0') |
| 435 | 76 | break; | |
| 436 |
3/4✓ Branch 0 taken 76 times.
✓ Branch 1 taken 6557 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 76 times.
|
6633 | if (c == '%' && *(p+1) == '%') // %% |
| 437 | ✗ | addchar_count = 2; | |
| 438 |
5/6✓ Branch 0 taken 76 times.
✓ Branch 1 taken 6557 times.
✓ Branch 2 taken 53 times.
✓ Branch 3 taken 23 times.
✓ Branch 4 taken 53 times.
✗ Branch 5 not taken.
|
6633 | else if (c == '%' && (av_isdigit(*(p+1)) || *(p+1) == placeholder)) { |
| 439 | 76 | nd = 0; | |
| 440 | 76 | addchar_count = 1; | |
| 441 |
2/2✓ Branch 0 taken 46 times.
✓ Branch 1 taken 76 times.
|
122 | while (av_isdigit(*(p + addchar_count))) { |
| 442 | 46 | nd = nd * 10 + *(p + addchar_count) - '0'; | |
| 443 | 46 | addchar_count++; | |
| 444 | } | ||
| 445 | |||
| 446 |
1/2✓ Branch 0 taken 76 times.
✗ Branch 1 not taken.
|
76 | if (*(p + addchar_count) == placeholder) { |
| 447 |
1/2✓ Branch 0 taken 76 times.
✗ Branch 1 not taken.
|
76 | av_bprintf(&buf, "%0*"PRId64, (number < 0) ? nd : nd++, number); |
| 448 | 76 | p += (addchar_count + 1); | |
| 449 | 76 | addchar_count = 0; | |
| 450 | 76 | found_count++; | |
| 451 | } | ||
| 452 | |||
| 453 | } else | ||
| 454 | 6557 | addchar_count = 1; | |
| 455 | |||
| 456 | 6633 | av_bprint_append_data(&buf, p, addchar_count); | |
| 457 | 6633 | p += addchar_count; | |
| 458 | } | ||
| 459 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 76 times.
|
76 | if (!av_bprint_is_complete(&buf)) { |
| 460 | ✗ | av_bprint_finalize(&buf, NULL); | |
| 461 | ✗ | return AVERROR(ENOMEM); | |
| 462 | } | ||
| 463 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 76 times.
|
76 | if ((ret = av_bprint_finalize(&buf, s)) < 0) |
| 464 | ✗ | return ret; | |
| 465 | 76 | return found_count; | |
| 466 | } | ||
| 467 | |||
| 468 | 6 | static void write_styp(AVIOContext *pb) | |
| 469 | { | ||
| 470 | 6 | avio_wb32(pb, 24); | |
| 471 | 6 | ffio_wfourcc(pb, "styp"); | |
| 472 | 6 | ffio_wfourcc(pb, "msdh"); | |
| 473 | 6 | avio_wb32(pb, 0); /* minor */ | |
| 474 | 6 | ffio_wfourcc(pb, "msdh"); | |
| 475 | 6 | ffio_wfourcc(pb, "msix"); | |
| 476 | 6 | } | |
| 477 | |||
| 478 | 86 | static int flush_dynbuf(VariantStream *vs, int *range_length) | |
| 479 | { | ||
| 480 | 86 | AVFormatContext *ctx = vs->avf; | |
| 481 | |||
| 482 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 86 times.
|
86 | if (!ctx->pb) { |
| 483 | ✗ | return AVERROR(EINVAL); | |
| 484 | } | ||
| 485 | |||
| 486 | // flush | ||
| 487 | 86 | av_write_frame(ctx, NULL); | |
| 488 | |||
| 489 | // write out to file | ||
| 490 | 86 | *range_length = avio_close_dyn_buf(ctx->pb, &vs->temp_buffer); | |
| 491 | 86 | ctx->pb = NULL; | |
| 492 | 86 | avio_write(vs->out, vs->temp_buffer, *range_length); | |
| 493 | 86 | avio_flush(vs->out); | |
| 494 | |||
| 495 | // re-open buffer | ||
| 496 | 86 | return avio_open_dyn_buf(&ctx->pb); | |
| 497 | } | ||
| 498 | |||
| 499 | ✗ | static void reflush_dynbuf(VariantStream *vs, int *range_length) | |
| 500 | { | ||
| 501 | // re-open buffer | ||
| 502 | ✗ | avio_write(vs->out, vs->temp_buffer, *range_length); | |
| 503 | ✗ | } | |
| 504 | |||
| 505 | ✗ | static int hls_delete_file(HLSContext *hls, AVFormatContext *avf, | |
| 506 | char *path, const char *proto) | ||
| 507 | { | ||
| 508 | ✗ | if (hls->method || (proto && !av_strcasecmp(proto, "http"))) { | |
| 509 | ✗ | AVDictionary *opt = NULL; | |
| 510 | int ret; | ||
| 511 | |||
| 512 | ✗ | set_http_options(avf, &opt, hls); | |
| 513 | ✗ | av_dict_set(&opt, "method", "DELETE", 0); | |
| 514 | |||
| 515 | ✗ | ret = hlsenc_io_open(avf, &hls->http_delete, path, &opt); | |
| 516 | ✗ | av_dict_free(&opt); | |
| 517 | ✗ | if (ret < 0) | |
| 518 | ✗ | return hls->ignore_io_errors ? 1 : ret; | |
| 519 | |||
| 520 | //Nothing to write | ||
| 521 | ✗ | hlsenc_io_close(avf, &hls->http_delete, path); | |
| 522 | ✗ | } else if (unlink(path) < 0) { | |
| 523 | ✗ | av_log(hls, AV_LOG_ERROR, "failed to delete old segment %s: %s\n", | |
| 524 | ✗ | path, strerror(errno)); | |
| 525 | } | ||
| 526 | ✗ | return 0; | |
| 527 | } | ||
| 528 | |||
| 529 | ✗ | static int hls_delete_old_segments(AVFormatContext *s, HLSContext *hls, | |
| 530 | VariantStream *vs) | ||
| 531 | { | ||
| 532 | |||
| 533 | ✗ | HLSSegment *segment, *previous_segment = NULL; | |
| 534 | ✗ | float playlist_duration = 0.0f; | |
| 535 | ✗ | int ret = 0; | |
| 536 | ✗ | int segment_cnt = 0; | |
| 537 | AVBPrint path; | ||
| 538 | ✗ | const char *dirname = NULL; | |
| 539 | ✗ | char *dirname_r = NULL; | |
| 540 | ✗ | char *dirname_repl = NULL; | |
| 541 | ✗ | const char *vtt_dirname = NULL; | |
| 542 | ✗ | char *vtt_dirname_r = NULL; | |
| 543 | ✗ | const char *proto = NULL; | |
| 544 | |||
| 545 | ✗ | av_bprint_init(&path, 0, AV_BPRINT_SIZE_UNLIMITED); | |
| 546 | |||
| 547 | ✗ | segment = vs->segments; | |
| 548 | ✗ | while (segment) { | |
| 549 | ✗ | playlist_duration += segment->duration; | |
| 550 | ✗ | segment = segment->next; | |
| 551 | } | ||
| 552 | |||
| 553 | ✗ | segment = vs->old_segments; | |
| 554 | ✗ | segment_cnt = 0; | |
| 555 | ✗ | while (segment) { | |
| 556 | ✗ | playlist_duration -= segment->duration; | |
| 557 | ✗ | previous_segment = segment; | |
| 558 | ✗ | segment = previous_segment->next; | |
| 559 | ✗ | segment_cnt++; | |
| 560 | ✗ | if (playlist_duration <= -previous_segment->duration) { | |
| 561 | ✗ | previous_segment->next = NULL; | |
| 562 | ✗ | break; | |
| 563 | } | ||
| 564 | ✗ | if (segment_cnt >= hls->hls_delete_threshold) { | |
| 565 | ✗ | previous_segment->next = NULL; | |
| 566 | ✗ | break; | |
| 567 | } | ||
| 568 | } | ||
| 569 | |||
| 570 | ✗ | if (segment && !hls->use_localtime_mkdir) { | |
| 571 | ✗ | dirname_r = hls->segment_filename ? av_strdup(hls->segment_filename): av_strdup(vs->avf->url); | |
| 572 | ✗ | dirname = av_dirname(dirname_r); | |
| 573 | } | ||
| 574 | |||
| 575 | /* if %v is present in the file's directory | ||
| 576 | * all segment belongs to the same variant, so do it only once before the loop*/ | ||
| 577 | ✗ | if (dirname && av_stristr(dirname, "%v")) { | |
| 578 | ✗ | if (!vs->varname) { | |
| 579 | ✗ | if (replace_int_data_in_filename(&dirname_repl, dirname, 'v', segment->var_stream_idx) < 1) { | |
| 580 | ✗ | ret = AVERROR(EINVAL); | |
| 581 | ✗ | goto fail; | |
| 582 | } | ||
| 583 | } else { | ||
| 584 | ✗ | if (replace_str_data_in_filename(&dirname_repl, dirname, 'v', vs->varname) < 1) { | |
| 585 | ✗ | ret = AVERROR(EINVAL); | |
| 586 | ✗ | goto fail; | |
| 587 | } | ||
| 588 | } | ||
| 589 | |||
| 590 | ✗ | dirname = dirname_repl; | |
| 591 | } | ||
| 592 | |||
| 593 | ✗ | while (segment) { | |
| 594 | ✗ | av_log(hls, AV_LOG_DEBUG, "deleting old segment %s\n", | |
| 595 | ✗ | segment->filename); | |
| 596 | ✗ | if (!hls->use_localtime_mkdir) // segment->filename contains basename only | |
| 597 | ✗ | av_bprintf(&path, "%s/", dirname); | |
| 598 | ✗ | av_bprintf(&path, "%s", segment->filename); | |
| 599 | |||
| 600 | ✗ | if (!av_bprint_is_complete(&path)) { | |
| 601 | ✗ | ret = AVERROR(ENOMEM); | |
| 602 | ✗ | goto fail; | |
| 603 | } | ||
| 604 | |||
| 605 | ✗ | proto = avio_find_protocol_name(s->url); | |
| 606 | ✗ | if (ret = hls_delete_file(hls, s, path.str, proto)) | |
| 607 | ✗ | goto fail; | |
| 608 | |||
| 609 | ✗ | if ((segment->sub_filename[0] != '\0')) { | |
| 610 | ✗ | vtt_dirname_r = av_strdup(vs->vtt_avf->url); | |
| 611 | ✗ | vtt_dirname = av_dirname(vtt_dirname_r); | |
| 612 | |||
| 613 | ✗ | av_bprint_clear(&path); | |
| 614 | ✗ | av_bprintf(&path, "%s/%s", vtt_dirname, segment->sub_filename); | |
| 615 | ✗ | av_freep(&vtt_dirname_r); | |
| 616 | |||
| 617 | ✗ | if (!av_bprint_is_complete(&path)) { | |
| 618 | ✗ | ret = AVERROR(ENOMEM); | |
| 619 | ✗ | goto fail; | |
| 620 | } | ||
| 621 | |||
| 622 | ✗ | if (ret = hls_delete_file(hls, s, path.str, proto)) | |
| 623 | ✗ | goto fail; | |
| 624 | } | ||
| 625 | ✗ | av_bprint_clear(&path); | |
| 626 | ✗ | previous_segment = segment; | |
| 627 | ✗ | segment = previous_segment->next; | |
| 628 | ✗ | av_freep(&previous_segment); | |
| 629 | } | ||
| 630 | |||
| 631 | ✗ | fail: | |
| 632 | ✗ | av_bprint_finalize(&path, NULL); | |
| 633 | ✗ | av_freep(&dirname_r); | |
| 634 | ✗ | av_freep(&dirname_repl); | |
| 635 | |||
| 636 | ✗ | return ret; | |
| 637 | } | ||
| 638 | |||
| 639 | ✗ | static int do_encrypt(AVFormatContext *s, VariantStream *vs) | |
| 640 | { | ||
| 641 | ✗ | HLSContext *hls = s->priv_data; | |
| 642 | int ret; | ||
| 643 | int len; | ||
| 644 | AVIOContext *pb; | ||
| 645 | uint8_t key[KEYSIZE]; | ||
| 646 | ✗ | char * key_basename_source = (hls->master_m3u8_url) ? hls->master_m3u8_url : s->url; | |
| 647 | |||
| 648 | ✗ | len = strlen(key_basename_source) + 4 + 1; | |
| 649 | ✗ | hls->key_basename = av_mallocz(len); | |
| 650 | ✗ | if (!hls->key_basename) | |
| 651 | ✗ | return AVERROR(ENOMEM); | |
| 652 | |||
| 653 | ✗ | av_strlcpy(hls->key_basename, key_basename_source, len); | |
| 654 | ✗ | av_strlcat(hls->key_basename, ".key", len); | |
| 655 | |||
| 656 | ✗ | if (hls->key_url) { | |
| 657 | ✗ | av_strlcpy(hls->key_file, hls->key_url, sizeof(hls->key_file)); | |
| 658 | ✗ | av_strlcpy(hls->key_uri, hls->key_url, sizeof(hls->key_uri)); | |
| 659 | } else { | ||
| 660 | ✗ | av_strlcpy(hls->key_file, hls->key_basename, sizeof(hls->key_file)); | |
| 661 | ✗ | av_strlcpy(hls->key_uri, hls->key_basename, sizeof(hls->key_uri)); | |
| 662 | } | ||
| 663 | |||
| 664 | ✗ | if (!*hls->iv_string) { | |
| 665 | ✗ | uint8_t iv[16] = { 0 }; | |
| 666 | char buf[33]; | ||
| 667 | |||
| 668 | ✗ | if (!hls->iv) { | |
| 669 | ✗ | AV_WB64(iv + 8, vs->sequence); | |
| 670 | } else { | ||
| 671 | ✗ | memcpy(iv, hls->iv, sizeof(iv)); | |
| 672 | } | ||
| 673 | ✗ | ff_data_to_hex(buf, iv, sizeof(iv), 0); | |
| 674 | ✗ | memcpy(hls->iv_string, buf, sizeof(hls->iv_string)); | |
| 675 | } | ||
| 676 | |||
| 677 | ✗ | if (!*hls->key_uri) { | |
| 678 | ✗ | av_log(hls, AV_LOG_ERROR, "no key URI specified in key info file\n"); | |
| 679 | ✗ | return AVERROR(EINVAL); | |
| 680 | } | ||
| 681 | |||
| 682 | ✗ | if (!*hls->key_file) { | |
| 683 | ✗ | av_log(hls, AV_LOG_ERROR, "no key file specified in key info file\n"); | |
| 684 | ✗ | return AVERROR(EINVAL); | |
| 685 | } | ||
| 686 | |||
| 687 | ✗ | if (!*hls->key_string) { | |
| 688 | ✗ | AVDictionary *options = NULL; | |
| 689 | ✗ | if (!hls->key) { | |
| 690 | ✗ | if ((ret = av_random_bytes(key, sizeof(key))) < 0) { | |
| 691 | ✗ | av_log(s, AV_LOG_ERROR, "Cannot generate a strong random key\n"); | |
| 692 | ✗ | return ret; | |
| 693 | } | ||
| 694 | } else { | ||
| 695 | ✗ | memcpy(key, hls->key, sizeof(key)); | |
| 696 | } | ||
| 697 | |||
| 698 | ✗ | ff_data_to_hex(hls->key_string, key, sizeof(key), 0); | |
| 699 | ✗ | set_http_options(s, &options, hls); | |
| 700 | ✗ | ret = s->io_open(s, &pb, hls->key_file, AVIO_FLAG_WRITE, &options); | |
| 701 | ✗ | av_dict_free(&options); | |
| 702 | ✗ | if (ret < 0) | |
| 703 | ✗ | return ret; | |
| 704 | ✗ | avio_seek(pb, 0, SEEK_CUR); | |
| 705 | ✗ | avio_write(pb, key, KEYSIZE); | |
| 706 | ✗ | avio_close(pb); | |
| 707 | } | ||
| 708 | ✗ | return 0; | |
| 709 | } | ||
| 710 | |||
| 711 | |||
| 712 | ✗ | static int hls_encryption_start(AVFormatContext *s, VariantStream *vs) | |
| 713 | { | ||
| 714 | ✗ | HLSContext *hls = s->priv_data; | |
| 715 | int ret; | ||
| 716 | AVIOContext *pb; | ||
| 717 | uint8_t key[KEYSIZE]; | ||
| 718 | ✗ | AVDictionary *options = NULL; | |
| 719 | |||
| 720 | ✗ | set_http_options(s, &options, hls); | |
| 721 | ✗ | ret = s->io_open(s, &pb, hls->key_info_file, AVIO_FLAG_READ, &options); | |
| 722 | ✗ | av_dict_free(&options); | |
| 723 | ✗ | if (ret < 0) { | |
| 724 | ✗ | av_log(hls, AV_LOG_ERROR, | |
| 725 | "error opening key info file %s\n", hls->key_info_file); | ||
| 726 | ✗ | return ret; | |
| 727 | } | ||
| 728 | |||
| 729 | ✗ | ff_get_line(pb, vs->key_uri, sizeof(vs->key_uri)); | |
| 730 | ✗ | vs->key_uri[strcspn(vs->key_uri, "\r\n")] = '\0'; | |
| 731 | |||
| 732 | ✗ | ff_get_line(pb, vs->key_file, sizeof(vs->key_file)); | |
| 733 | ✗ | vs->key_file[strcspn(vs->key_file, "\r\n")] = '\0'; | |
| 734 | |||
| 735 | ✗ | ff_get_line(pb, vs->iv_string, sizeof(vs->iv_string)); | |
| 736 | ✗ | vs->iv_string[strcspn(vs->iv_string, "\r\n")] = '\0'; | |
| 737 | |||
| 738 | ✗ | ff_format_io_close(s, &pb); | |
| 739 | |||
| 740 | ✗ | if (!*vs->key_uri) { | |
| 741 | ✗ | av_log(hls, AV_LOG_ERROR, "no key URI specified in key info file\n"); | |
| 742 | ✗ | return AVERROR(EINVAL); | |
| 743 | } | ||
| 744 | |||
| 745 | ✗ | if (!*vs->key_file) { | |
| 746 | ✗ | av_log(hls, AV_LOG_ERROR, "no key file specified in key info file\n"); | |
| 747 | ✗ | return AVERROR(EINVAL); | |
| 748 | } | ||
| 749 | |||
| 750 | ✗ | set_http_options(s, &options, hls); | |
| 751 | ✗ | ret = s->io_open(s, &pb, vs->key_file, AVIO_FLAG_READ, &options); | |
| 752 | ✗ | av_dict_free(&options); | |
| 753 | ✗ | if (ret < 0) { | |
| 754 | ✗ | av_log(hls, AV_LOG_ERROR, "error opening key file %s\n", vs->key_file); | |
| 755 | ✗ | return ret; | |
| 756 | } | ||
| 757 | |||
| 758 | ✗ | ret = avio_read(pb, key, sizeof(key)); | |
| 759 | ✗ | ff_format_io_close(s, &pb); | |
| 760 | ✗ | if (ret != sizeof(key)) { | |
| 761 | ✗ | av_log(hls, AV_LOG_ERROR, "error reading key file %s\n", vs->key_file); | |
| 762 | ✗ | if (ret >= 0 || ret == AVERROR_EOF) | |
| 763 | ✗ | ret = AVERROR(EINVAL); | |
| 764 | ✗ | return ret; | |
| 765 | } | ||
| 766 | ✗ | ff_data_to_hex(vs->key_string, key, sizeof(key), 0); | |
| 767 | |||
| 768 | ✗ | return 0; | |
| 769 | } | ||
| 770 | |||
| 771 | 18 | static int hls_mux_init(AVFormatContext *s, VariantStream *vs) | |
| 772 | { | ||
| 773 | 18 | AVDictionary *options = NULL; | |
| 774 | 18 | HLSContext *hls = s->priv_data; | |
| 775 | AVFormatContext *oc; | ||
| 776 | 18 | AVFormatContext *vtt_oc = NULL; | |
| 777 |
4/4✓ Branch 0 taken 17 times.
✓ Branch 1 taken 1 times.
✓ Branch 2 taken 1 times.
✓ Branch 3 taken 16 times.
|
18 | int byterange_mode = (hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size > 0); |
| 778 | int remaining_options; | ||
| 779 | int i, ret; | ||
| 780 | |||
| 781 | 18 | ret = avformat_alloc_output_context2(&vs->avf, vs->oformat, NULL, NULL); | |
| 782 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 783 | ✗ | return ret; | |
| 784 | 18 | oc = vs->avf; | |
| 785 | |||
| 786 | 18 | oc->url = av_strdup(""); | |
| 787 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (!oc->url) |
| 788 | ✗ | return AVERROR(ENOMEM); | |
| 789 | |||
| 790 | 18 | oc->interrupt_callback = s->interrupt_callback; | |
| 791 | 18 | oc->max_delay = s->max_delay; | |
| 792 | 18 | oc->opaque = s->opaque; | |
| 793 | 18 | oc->io_open = s->io_open; | |
| 794 | 18 | oc->io_close2 = s->io_close2; | |
| 795 | 18 | oc->strict_std_compliance = s->strict_std_compliance; | |
| 796 | 18 | av_dict_copy(&oc->metadata, s->metadata, 0); | |
| 797 | |||
| 798 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (vs->vtt_oformat) { |
| 799 | ✗ | ret = avformat_alloc_output_context2(&vs->vtt_avf, vs->vtt_oformat, NULL, NULL); | |
| 800 | ✗ | if (ret < 0) | |
| 801 | ✗ | return ret; | |
| 802 | ✗ | vtt_oc = vs->vtt_avf; | |
| 803 | ✗ | av_dict_copy(&vtt_oc->metadata, s->metadata, 0); | |
| 804 | } | ||
| 805 | |||
| 806 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 18 times.
|
36 | for (i = 0; i < vs->nb_streams; i++) { |
| 807 | AVStream *st; | ||
| 808 | AVFormatContext *loc; | ||
| 809 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (vs->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE) |
| 810 | ✗ | loc = vtt_oc; | |
| 811 | else | ||
| 812 | 18 | loc = oc; | |
| 813 | |||
| 814 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 18 times.
|
18 | if (!(st = avformat_new_stream(loc, NULL))) |
| 815 | ✗ | return AVERROR(ENOMEM); | |
| 816 | 18 | ret = avcodec_parameters_copy(st->codecpar, vs->streams[i]->codecpar); | |
| 817 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 818 | ✗ | return ret; | |
| 819 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 16 times.
|
18 | if (!oc->oformat->codec_tag || |
| 820 |
3/4✓ Branch 1 taken 1 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 1 times.
|
3 | av_codec_get_id (oc->oformat->codec_tag, vs->streams[i]->codecpar->codec_tag) == st->codecpar->codec_id || |
| 821 | 1 | av_codec_get_tag(oc->oformat->codec_tag, vs->streams[i]->codecpar->codec_id) <= 0) { | |
| 822 | 17 | st->codecpar->codec_tag = vs->streams[i]->codecpar->codec_tag; | |
| 823 | } else { | ||
| 824 | 1 | st->codecpar->codec_tag = 0; | |
| 825 | } | ||
| 826 | |||
| 827 | 18 | st->sample_aspect_ratio = vs->streams[i]->sample_aspect_ratio; | |
| 828 | 18 | st->time_base = vs->streams[i]->time_base; | |
| 829 | 18 | av_dict_copy(&st->metadata, vs->streams[i]->metadata, 0); | |
| 830 | 18 | st->id = vs->streams[i]->id; | |
| 831 | } | ||
| 832 | |||
| 833 | 18 | vs->start_pos = 0; | |
| 834 | 18 | vs->new_start = 1; | |
| 835 | |||
| 836 |
3/4✓ Branch 0 taken 2 times.
✓ Branch 1 taken 16 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 2 times.
|
18 | if (hls->segment_type == SEGMENT_TYPE_FMP4 && hls->max_seg_size > 0) { |
| 837 | ✗ | if (hls->http_persistent > 0) { | |
| 838 | //TODO: Support fragment fmp4 for http persistent in HLS muxer. | ||
| 839 | ✗ | av_log(s, AV_LOG_WARNING, "http persistent mode is currently unsupported for fragment mp4 in the HLS muxer.\n"); | |
| 840 | } | ||
| 841 | ✗ | if (hls->max_seg_size > 0) { | |
| 842 | ✗ | av_log(s, AV_LOG_WARNING, "Multi-file byterange mode is currently unsupported in the HLS muxer.\n"); | |
| 843 | ✗ | return AVERROR_PATCHWELCOME; | |
| 844 | } | ||
| 845 | } | ||
| 846 | |||
| 847 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 18 times.
|
18 | if ((ret = avio_open_dyn_buf(&oc->pb)) < 0) |
| 848 | ✗ | return ret; | |
| 849 | |||
| 850 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 16 times.
|
18 | if (hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 851 | 2 | set_http_options(s, &options, hls); | |
| 852 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (byterange_mode) { |
| 853 | ✗ | ret = hlsenc_io_open(s, &vs->out, vs->basename, &options); | |
| 854 | } else { | ||
| 855 | 2 | ret = hlsenc_io_open(s, &vs->out, vs->base_output_dirname, &options); | |
| 856 | } | ||
| 857 | 2 | av_dict_free(&options); | |
| 858 | } | ||
| 859 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) { |
| 860 | ✗ | av_log(s, AV_LOG_ERROR, "Failed to open segment '%s'\n", vs->fmp4_init_filename); | |
| 861 | ✗ | return ret; | |
| 862 | } | ||
| 863 | |||
| 864 | 18 | av_dict_copy(&options, hls->format_options, 0); | |
| 865 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 16 times.
|
18 | if (hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 866 | 2 | av_dict_set(&options, "fflags", "-autobsf", 0); | |
| 867 | 2 | av_dict_set(&options, "movflags", "+frag_custom+dash+delay_moov", AV_DICT_APPEND); | |
| 868 | } else { | ||
| 869 | /* We only require one PAT/PMT per segment. */ | ||
| 870 | char period[21]; | ||
| 871 | 16 | snprintf(period, sizeof(period), "%d", (INT_MAX / 2) - 1); | |
| 872 | 16 | av_dict_set(&options, "sdt_period", period, AV_DICT_DONT_OVERWRITE); | |
| 873 | 16 | av_dict_set(&options, "pat_period", period, AV_DICT_DONT_OVERWRITE); | |
| 874 | } | ||
| 875 | 18 | ret = avformat_init_output(oc, &options); | |
| 876 | 18 | remaining_options = av_dict_count(options); | |
| 877 | 18 | av_dict_free(&options); | |
| 878 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 879 | ✗ | return ret; | |
| 880 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (remaining_options) { |
| 881 | ✗ | av_log(s, AV_LOG_ERROR, "Some of the provided format options are not recognized\n"); | |
| 882 | ✗ | return AVERROR(EINVAL); | |
| 883 | } | ||
| 884 | 18 | avio_flush(oc->pb); | |
| 885 | 18 | return 0; | |
| 886 | } | ||
| 887 | |||
| 888 | 173 | static HLSSegment *find_segment_by_filename(HLSSegment *segment, const char *filename) | |
| 889 | { | ||
| 890 |
2/2✓ Branch 0 taken 212 times.
✓ Branch 1 taken 158 times.
|
370 | while (segment) { |
| 891 |
2/2✓ Branch 1 taken 15 times.
✓ Branch 2 taken 197 times.
|
212 | if (!av_strcasecmp(segment->filename,filename)) |
| 892 | 15 | return segment; | |
| 893 | 197 | segment = segment->next; | |
| 894 | } | ||
| 895 | 158 | return (HLSSegment *) NULL; | |
| 896 | } | ||
| 897 | |||
| 898 | 94 | static int sls_flags_filename_process(struct AVFormatContext *s, HLSContext *hls, | |
| 899 | VariantStream *vs, HLSSegment *en, | ||
| 900 | double duration, int64_t pos, int64_t size) | ||
| 901 | { | ||
| 902 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 94 times.
|
94 | if ((hls->flags & (HLS_SECOND_LEVEL_SEGMENT_SIZE | HLS_SECOND_LEVEL_SEGMENT_DURATION)) && |
| 903 | ✗ | strlen(vs->current_segment_final_filename_fmt)) { | |
| 904 | ✗ | char * new_url = av_strdup(vs->current_segment_final_filename_fmt); | |
| 905 | ✗ | if (!new_url) { | |
| 906 | ✗ | return AVERROR(ENOMEM); | |
| 907 | } | ||
| 908 | ✗ | ff_format_set_url(vs->avf, new_url); | |
| 909 | ✗ | if (hls->flags & HLS_SECOND_LEVEL_SEGMENT_SIZE) { | |
| 910 | ✗ | char *filename = NULL; | |
| 911 | ✗ | if (replace_int_data_in_filename(&filename, vs->avf->url, 's', pos + size) < 1) { | |
| 912 | ✗ | av_log(hls, AV_LOG_ERROR, | |
| 913 | "Invalid second level segment filename template '%s', " | ||
| 914 | "you can try to remove second_level_segment_size flag\n", | ||
| 915 | ✗ | vs->avf->url); | |
| 916 | ✗ | av_freep(&filename); | |
| 917 | ✗ | return AVERROR(EINVAL); | |
| 918 | } | ||
| 919 | ✗ | ff_format_set_url(vs->avf, filename); | |
| 920 | } | ||
| 921 | ✗ | if (hls->flags & HLS_SECOND_LEVEL_SEGMENT_DURATION) { | |
| 922 | ✗ | char *filename = NULL; | |
| 923 | ✗ | if (replace_int_data_in_filename(&filename, vs->avf->url, | |
| 924 | ✗ | 't', (int64_t)round(duration * HLS_MICROSECOND_UNIT)) < 1) { | |
| 925 | ✗ | av_log(hls, AV_LOG_ERROR, | |
| 926 | "Invalid second level segment filename template '%s', " | ||
| 927 | "you can try to remove second_level_segment_duration flag\n", | ||
| 928 | ✗ | vs->avf->url); | |
| 929 | ✗ | av_freep(&filename); | |
| 930 | ✗ | return AVERROR(EINVAL); | |
| 931 | } | ||
| 932 | ✗ | ff_format_set_url(vs->avf, filename); | |
| 933 | } | ||
| 934 | } | ||
| 935 | 94 | return 0; | |
| 936 | } | ||
| 937 | |||
| 938 | 18 | static int sls_flag_check_duration_size_index(HLSContext *hls) | |
| 939 | { | ||
| 940 | 18 | int ret = 0; | |
| 941 | |||
| 942 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->flags & HLS_SECOND_LEVEL_SEGMENT_DURATION) { |
| 943 | ✗ | av_log(hls, AV_LOG_ERROR, | |
| 944 | "second_level_segment_duration hls_flag requires strftime to be true\n"); | ||
| 945 | ✗ | ret = AVERROR(EINVAL); | |
| 946 | } | ||
| 947 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->flags & HLS_SECOND_LEVEL_SEGMENT_SIZE) { |
| 948 | ✗ | av_log(hls, AV_LOG_ERROR, | |
| 949 | "second_level_segment_size hls_flag requires strfime to be true\n"); | ||
| 950 | ✗ | ret = AVERROR(EINVAL); | |
| 951 | } | ||
| 952 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->flags & HLS_SECOND_LEVEL_SEGMENT_INDEX) { |
| 953 | ✗ | av_log(hls, AV_LOG_ERROR, | |
| 954 | "second_level_segment_index hls_flag requires strftime to be true\n"); | ||
| 955 | ✗ | ret = AVERROR(EINVAL); | |
| 956 | } | ||
| 957 | |||
| 958 | 18 | return ret; | |
| 959 | } | ||
| 960 | |||
| 961 | ✗ | static int sls_flag_check_duration_size(HLSContext *hls, VariantStream *vs) | |
| 962 | { | ||
| 963 | ✗ | const char *proto = avio_find_protocol_name(vs->basename); | |
| 964 | ✗ | int segment_renaming_ok = proto && !strcmp(proto, "file"); | |
| 965 | ✗ | int ret = 0; | |
| 966 | |||
| 967 | ✗ | if ((hls->flags & HLS_SECOND_LEVEL_SEGMENT_DURATION) && !segment_renaming_ok) { | |
| 968 | ✗ | av_log(hls, AV_LOG_ERROR, | |
| 969 | "second_level_segment_duration hls_flag works only with file protocol segment names\n"); | ||
| 970 | ✗ | ret = AVERROR(EINVAL); | |
| 971 | } | ||
| 972 | ✗ | if ((hls->flags & HLS_SECOND_LEVEL_SEGMENT_SIZE) && !segment_renaming_ok) { | |
| 973 | ✗ | av_log(hls, AV_LOG_ERROR, | |
| 974 | "second_level_segment_size hls_flag works only with file protocol segment names\n"); | ||
| 975 | ✗ | ret = AVERROR(EINVAL); | |
| 976 | } | ||
| 977 | |||
| 978 | ✗ | return ret; | |
| 979 | } | ||
| 980 | |||
| 981 | 77 | static void sls_flag_file_rename(HLSContext *hls, VariantStream *vs, char *old_filename) { | |
| 982 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 77 times.
|
77 | if ((hls->flags & (HLS_SECOND_LEVEL_SEGMENT_SIZE | HLS_SECOND_LEVEL_SEGMENT_DURATION)) && |
| 983 | ✗ | strlen(vs->current_segment_final_filename_fmt)) { | |
| 984 | ✗ | ff_rename(old_filename, vs->avf->url, hls); | |
| 985 | } | ||
| 986 | 77 | } | |
| 987 | |||
| 988 | ✗ | static int sls_flag_use_localtime_filename(AVFormatContext *oc, HLSContext *c, VariantStream *vs) | |
| 989 | { | ||
| 990 | ✗ | if (c->flags & HLS_SECOND_LEVEL_SEGMENT_INDEX) { | |
| 991 | ✗ | char *filename = NULL; | |
| 992 | ✗ | if (replace_int_data_in_filename(&filename, | |
| 993 | ✗ | oc->url, 'd', vs->sequence) < 1) { | |
| 994 | ✗ | av_log(c, AV_LOG_ERROR, "Invalid second level segment filename template '%s', " | |
| 995 | "you can try to remove second_level_segment_index flag\n", | ||
| 996 | oc->url); | ||
| 997 | ✗ | av_freep(&filename); | |
| 998 | ✗ | return AVERROR(EINVAL); | |
| 999 | } | ||
| 1000 | ✗ | ff_format_set_url(oc, filename); | |
| 1001 | } | ||
| 1002 | ✗ | if (c->flags & (HLS_SECOND_LEVEL_SEGMENT_SIZE | HLS_SECOND_LEVEL_SEGMENT_DURATION)) { | |
| 1003 | ✗ | av_strlcpy(vs->current_segment_final_filename_fmt, oc->url, | |
| 1004 | sizeof(vs->current_segment_final_filename_fmt)); | ||
| 1005 | ✗ | if (c->flags & HLS_SECOND_LEVEL_SEGMENT_SIZE) { | |
| 1006 | ✗ | char *filename = NULL; | |
| 1007 | ✗ | if (replace_int_data_in_filename(&filename, oc->url, 's', 0) < 1) { | |
| 1008 | ✗ | av_log(c, AV_LOG_ERROR, "Invalid second level segment filename template '%s', " | |
| 1009 | "you can try to remove second_level_segment_size flag\n", | ||
| 1010 | oc->url); | ||
| 1011 | ✗ | av_freep(&filename); | |
| 1012 | ✗ | return AVERROR(EINVAL); | |
| 1013 | } | ||
| 1014 | ✗ | ff_format_set_url(oc, filename); | |
| 1015 | } | ||
| 1016 | ✗ | if (c->flags & HLS_SECOND_LEVEL_SEGMENT_DURATION) { | |
| 1017 | ✗ | char *filename = NULL; | |
| 1018 | ✗ | if (replace_int_data_in_filename(&filename, oc->url, 't', 0) < 1) { | |
| 1019 | ✗ | av_log(c, AV_LOG_ERROR, "Invalid second level segment filename template '%s', " | |
| 1020 | "you can try to remove second_level_segment_duration flag\n", | ||
| 1021 | oc->url); | ||
| 1022 | ✗ | av_freep(&filename); | |
| 1023 | ✗ | return AVERROR(EINVAL); | |
| 1024 | } | ||
| 1025 | ✗ | ff_format_set_url(oc, filename); | |
| 1026 | } | ||
| 1027 | } | ||
| 1028 | ✗ | return 0; | |
| 1029 | } | ||
| 1030 | |||
| 1031 | /* Create a new segment and append it to the segment list */ | ||
| 1032 | 94 | static int hls_append_segment(struct AVFormatContext *s, HLSContext *hls, | |
| 1033 | VariantStream *vs, double duration, int64_t pos, | ||
| 1034 | int64_t size) | ||
| 1035 | { | ||
| 1036 | 94 | HLSSegment *en = av_malloc(sizeof(*en)); | |
| 1037 | const char *filename; | ||
| 1038 |
4/4✓ Branch 0 taken 84 times.
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 10 times.
✓ Branch 3 taken 74 times.
|
94 | int byterange_mode = (hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size > 0); |
| 1039 | int ret; | ||
| 1040 | |||
| 1041 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 94 times.
|
94 | if (!en) |
| 1042 | ✗ | return AVERROR(ENOMEM); | |
| 1043 | |||
| 1044 | 94 | vs->total_size += size; | |
| 1045 | 94 | vs->total_duration += duration; | |
| 1046 |
1/2✓ Branch 0 taken 94 times.
✗ Branch 1 not taken.
|
94 | if (duration > 0.5) { |
| 1047 | // Don't include the final, possibly very short segment in the | ||
| 1048 | // calculation of the max bitrate. | ||
| 1049 | 94 | int cur_bitrate = (int)(8 * size / duration); | |
| 1050 |
2/2✓ Branch 0 taken 49 times.
✓ Branch 1 taken 45 times.
|
94 | if (cur_bitrate > vs->max_bitrate) |
| 1051 | 49 | vs->max_bitrate = cur_bitrate; | |
| 1052 | } | ||
| 1053 |
1/2✓ Branch 0 taken 94 times.
✗ Branch 1 not taken.
|
94 | if (vs->total_duration > 0) |
| 1054 | 94 | vs->avg_bitrate = (int)(8 * vs->total_size / vs->total_duration); | |
| 1055 | |||
| 1056 | 94 | en->var_stream_idx = vs->var_stream_idx; | |
| 1057 | 94 | ret = sls_flags_filename_process(s, hls, vs, en, duration, pos, size); | |
| 1058 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 94 times.
|
94 | if (ret < 0) { |
| 1059 | ✗ | av_freep(&en); | |
| 1060 | ✗ | return ret; | |
| 1061 | } | ||
| 1062 | |||
| 1063 | 94 | filename = av_basename(vs->avf->url); | |
| 1064 | |||
| 1065 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 94 times.
|
94 | if (hls->use_localtime_mkdir) { |
| 1066 | ✗ | filename = vs->avf->url; | |
| 1067 | } | ||
| 1068 |
4/6✓ Branch 0 taken 94 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 79 times.
✓ Branch 4 taken 15 times.
✗ Branch 6 not taken.
✓ Branch 7 taken 79 times.
|
94 | if (vs->nb_entries <= 5000 && (find_segment_by_filename(vs->segments, filename) || find_segment_by_filename(vs->old_segments, filename)) |
| 1069 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
|
15 | && !byterange_mode) { |
| 1070 | ✗ | av_log(hls, AV_LOG_WARNING, "Duplicated segment filename detected: %s\n", filename); | |
| 1071 | } | ||
| 1072 | 94 | av_strlcpy(en->filename, filename, sizeof(en->filename)); | |
| 1073 | |||
| 1074 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 94 times.
|
94 | if (vs->has_subtitle) |
| 1075 | ✗ | av_strlcpy(en->sub_filename, av_basename(vs->vtt_avf->url), sizeof(en->sub_filename)); | |
| 1076 | else | ||
| 1077 | 94 | en->sub_filename[0] = '\0'; | |
| 1078 | |||
| 1079 | 94 | en->duration = duration; | |
| 1080 | 94 | en->pos = pos; | |
| 1081 | 94 | en->size = size; | |
| 1082 | 94 | en->keyframe_pos = vs->video_keyframe_pos; | |
| 1083 | 94 | en->keyframe_size = vs->video_keyframe_size; | |
| 1084 | 94 | en->next = NULL; | |
| 1085 | 94 | en->discont = 0; | |
| 1086 | 94 | en->discont_program_date_time = 0; | |
| 1087 | |||
| 1088 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 93 times.
|
94 | if (vs->discontinuity) { |
| 1089 | 1 | en->discont = 1; | |
| 1090 | 1 | vs->discontinuity = 0; | |
| 1091 | } | ||
| 1092 | |||
| 1093 |
2/4✓ Branch 0 taken 94 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 94 times.
|
94 | if (hls->key_info_file || hls->encrypt) { |
| 1094 | ✗ | av_strlcpy(en->key_uri, vs->key_uri, sizeof(en->key_uri)); | |
| 1095 | ✗ | av_strlcpy(en->iv_string, vs->iv_string, sizeof(en->iv_string)); | |
| 1096 | } | ||
| 1097 | |||
| 1098 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 76 times.
|
94 | if (!vs->segments) |
| 1099 | 18 | vs->segments = en; | |
| 1100 | else | ||
| 1101 | 76 | vs->last_segment->next = en; | |
| 1102 | |||
| 1103 | 94 | vs->last_segment = en; | |
| 1104 | |||
| 1105 | // EVENT or VOD playlists imply sliding window cannot be used | ||
| 1106 |
2/2✓ Branch 0 taken 13 times.
✓ Branch 1 taken 81 times.
|
94 | if (hls->pl_type != PLAYLIST_TYPE_NONE) |
| 1107 | 13 | hls->max_nb_segments = 0; | |
| 1108 | |||
| 1109 |
4/4✓ Branch 0 taken 31 times.
✓ Branch 1 taken 63 times.
✓ Branch 2 taken 8 times.
✓ Branch 3 taken 23 times.
|
94 | if (hls->max_nb_segments && vs->nb_entries >= hls->max_nb_segments) { |
| 1110 | 8 | en = vs->segments; | |
| 1111 |
2/4✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 8 times.
✗ Branch 3 not taken.
|
8 | if (!en->next->discont_program_date_time && !en->discont_program_date_time) |
| 1112 | 8 | vs->initial_prog_date_time += en->duration; | |
| 1113 | 8 | vs->segments = en->next; | |
| 1114 |
2/4✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 8 times.
|
8 | if (en && hls->flags & HLS_DELETE_SEGMENTS && |
| 1115 | ✗ | !(hls->flags & HLS_SINGLE_FILE)) { | |
| 1116 | ✗ | en->next = vs->old_segments; | |
| 1117 | ✗ | vs->old_segments = en; | |
| 1118 | ✗ | if ((ret = hls_delete_old_segments(s, hls, vs)) < 0) | |
| 1119 | ✗ | return ret; | |
| 1120 | } else | ||
| 1121 | 8 | av_freep(&en); | |
| 1122 | } else | ||
| 1123 | 86 | vs->nb_entries++; | |
| 1124 | |||
| 1125 |
2/2✓ Branch 0 taken 10 times.
✓ Branch 1 taken 84 times.
|
94 | if (hls->max_seg_size > 0) { |
| 1126 | 10 | return 0; | |
| 1127 | } | ||
| 1128 | 84 | vs->sequence++; | |
| 1129 | |||
| 1130 | 84 | return 0; | |
| 1131 | } | ||
| 1132 | |||
| 1133 | ✗ | static int extract_segment_number(const char *filename) { | |
| 1134 | ✗ | const char *dot = strrchr(filename, '.'); | |
| 1135 | ✗ | const char *num_start = dot - 1; | |
| 1136 | |||
| 1137 | ✗ | while (num_start > filename && *num_start >= '0' && *num_start <= '9') { | |
| 1138 | ✗ | num_start--; | |
| 1139 | } | ||
| 1140 | |||
| 1141 | ✗ | num_start++; | |
| 1142 | |||
| 1143 | ✗ | if (num_start == dot) | |
| 1144 | ✗ | return -1; | |
| 1145 | |||
| 1146 | ✗ | return atoi(num_start); | |
| 1147 | } | ||
| 1148 | |||
| 1149 | 1 | static int parse_playlist(AVFormatContext *s, const char *url, VariantStream *vs) | |
| 1150 | { | ||
| 1151 | 1 | HLSContext *hls = s->priv_data; | |
| 1152 | AVIOContext *in; | ||
| 1153 | 1 | int ret = 0, is_segment = 0; | |
| 1154 | int64_t new_start_pos; | ||
| 1155 | char line[MAX_URL_SIZE]; | ||
| 1156 | const char *ptr; | ||
| 1157 | const char *end; | ||
| 1158 | 1 | double discont_program_date_time = 0; | |
| 1159 | |||
| 1160 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if ((ret = ffio_open_whitelist(&in, url, AVIO_FLAG_READ, |
| 1161 | 1 | &s->interrupt_callback, NULL, | |
| 1162 | 1 | s->protocol_whitelist, s->protocol_blacklist)) < 0) | |
| 1163 | ✗ | return ret; | |
| 1164 | |||
| 1165 | 1 | ff_get_chomp_line(in, line, sizeof(line)); | |
| 1166 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (strcmp(line, "#EXTM3U")) { |
| 1167 | ✗ | ret = AVERROR_INVALIDDATA; | |
| 1168 | ✗ | goto fail; | |
| 1169 | } | ||
| 1170 | |||
| 1171 | 1 | vs->discontinuity = 0; | |
| 1172 |
2/2✓ Branch 1 taken 10 times.
✓ Branch 2 taken 1 times.
|
11 | while (!avio_feof(in)) { |
| 1173 | 10 | ff_get_chomp_line(in, line, sizeof(line)); | |
| 1174 |
2/2✓ Branch 1 taken 1 times.
✓ Branch 2 taken 9 times.
|
10 | if (av_strstart(line, "#EXT-X-MEDIA-SEQUENCE:", &ptr)) { |
| 1175 | 1 | int64_t tmp_sequence = strtoll(ptr, NULL, 10); | |
| 1176 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (tmp_sequence < vs->sequence) |
| 1177 | ✗ | av_log(hls, AV_LOG_VERBOSE, | |
| 1178 | "Found playlist sequence number was smaller """ | ||
| 1179 | "than specified start sequence number: %"PRId64" < %"PRId64", " | ||
| 1180 | "omitting\n", tmp_sequence, hls->start_sequence); | ||
| 1181 | else { | ||
| 1182 | 1 | av_log(hls, AV_LOG_DEBUG, "Found playlist sequence number: %"PRId64"\n", tmp_sequence); | |
| 1183 | 1 | vs->sequence = tmp_sequence; | |
| 1184 | } | ||
| 1185 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 9 times.
|
9 | } else if (av_strstart(line, "#EXT-X-DISCONTINUITY", &ptr)) { |
| 1186 | ✗ | is_segment = 1; | |
| 1187 | ✗ | vs->discontinuity = 1; | |
| 1188 |
2/2✓ Branch 1 taken 2 times.
✓ Branch 2 taken 7 times.
|
9 | } else if (av_strstart(line, "#EXTINF:", &ptr)) { |
| 1189 | 2 | is_segment = 1; | |
| 1190 | 2 | vs->duration = atof(ptr); | |
| 1191 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 7 times.
|
7 | } else if (av_stristart(line, "#EXT-X-KEY:", &ptr)) { |
| 1192 | ✗ | ptr = av_stristr(line, "URI=\""); | |
| 1193 | ✗ | if (ptr) { | |
| 1194 | ✗ | ptr += strlen("URI=\""); | |
| 1195 | ✗ | end = av_stristr(ptr, ","); | |
| 1196 | ✗ | if (end) { | |
| 1197 | ✗ | av_strlcpy(vs->key_uri, ptr, end - ptr); | |
| 1198 | } else { | ||
| 1199 | ✗ | av_strlcpy(vs->key_uri, ptr, sizeof(vs->key_uri)); | |
| 1200 | } | ||
| 1201 | } | ||
| 1202 | |||
| 1203 | ✗ | ptr = av_stristr(line, "IV=0x"); | |
| 1204 | ✗ | if (ptr) { | |
| 1205 | ✗ | ptr += strlen("IV=0x"); | |
| 1206 | ✗ | end = av_stristr(ptr, ","); | |
| 1207 | ✗ | if (end) { | |
| 1208 | ✗ | av_strlcpy(vs->iv_string, ptr, end - ptr); | |
| 1209 | } else { | ||
| 1210 | ✗ | av_strlcpy(vs->iv_string, ptr, sizeof(vs->iv_string)); | |
| 1211 | } | ||
| 1212 | } | ||
| 1213 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 7 times.
|
7 | } else if (av_strstart(line, "#EXT-X-PROGRAM-DATE-TIME:", &ptr)) { |
| 1214 | struct tm program_date_time; | ||
| 1215 | int y,M,d,h,m,sec; | ||
| 1216 | double ms; | ||
| 1217 | ✗ | if (sscanf(ptr, "%d-%d-%dT%d:%d:%d.%lf", &y, &M, &d, &h, &m, &sec, &ms) != 7) { | |
| 1218 | ✗ | ret = AVERROR_INVALIDDATA; | |
| 1219 | ✗ | goto fail; | |
| 1220 | } | ||
| 1221 | |||
| 1222 | ✗ | program_date_time.tm_year = y - 1900; | |
| 1223 | ✗ | program_date_time.tm_mon = M - 1; | |
| 1224 | ✗ | program_date_time.tm_mday = d; | |
| 1225 | ✗ | program_date_time.tm_hour = h; | |
| 1226 | ✗ | program_date_time.tm_min = m; | |
| 1227 | ✗ | program_date_time.tm_sec = sec; | |
| 1228 | ✗ | program_date_time.tm_isdst = -1; | |
| 1229 | |||
| 1230 | ✗ | discont_program_date_time = mktime(&program_date_time); | |
| 1231 | ✗ | discont_program_date_time += (double)(ms / 1000); | |
| 1232 |
2/2✓ Branch 1 taken 4 times.
✓ Branch 2 taken 3 times.
|
7 | } else if (av_strstart(line, "#", NULL)) { |
| 1233 | 4 | continue; | |
| 1234 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
|
3 | } else if (line[0]) { |
| 1235 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | if (is_segment) { |
| 1236 | 2 | char *new_file = av_strdup(line); | |
| 1237 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (!new_file) { |
| 1238 | ✗ | ret = AVERROR(ENOMEM); | |
| 1239 | ✗ | goto fail; | |
| 1240 | } | ||
| 1241 | 2 | ff_format_set_url(vs->avf, new_file); | |
| 1242 | |||
| 1243 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (vs->has_subtitle) { |
| 1244 | ✗ | int vtt_index = extract_segment_number(line); | |
| 1245 | ✗ | const char *vtt_basename = av_basename(vs->vtt_basename); | |
| 1246 | ✗ | char *vtt_file = NULL; | |
| 1247 | ✗ | ret = replace_int_data_in_filename(&vtt_file, vtt_basename, 'd', vtt_index); | |
| 1248 | ✗ | if (ret < 0 || !vtt_file) { | |
| 1249 | ✗ | ret = AVERROR(ENOMEM); | |
| 1250 | ✗ | goto fail; | |
| 1251 | } | ||
| 1252 | |||
| 1253 | ✗ | ff_format_set_url(vs->vtt_avf, vtt_file); | |
| 1254 | } | ||
| 1255 | |||
| 1256 | 2 | is_segment = 0; | |
| 1257 | 2 | new_start_pos = avio_tell(vs->avf->pb); | |
| 1258 | 2 | vs->size = new_start_pos - vs->start_pos; | |
| 1259 | 2 | ret = hls_append_segment(s, hls, vs, vs->duration, vs->start_pos, vs->size); | |
| 1260 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (discont_program_date_time) { |
| 1261 | ✗ | vs->last_segment->discont_program_date_time = discont_program_date_time; | |
| 1262 | ✗ | discont_program_date_time += vs->duration; | |
| 1263 | } | ||
| 1264 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (ret < 0) |
| 1265 | ✗ | goto fail; | |
| 1266 | 2 | vs->start_pos = new_start_pos; | |
| 1267 | } | ||
| 1268 | } | ||
| 1269 | } | ||
| 1270 | |||
| 1271 | 1 | fail: | |
| 1272 | 1 | avio_close(in); | |
| 1273 | 1 | return ret; | |
| 1274 | } | ||
| 1275 | |||
| 1276 | 36 | static void hls_free_segments(HLSSegment *p) | |
| 1277 | { | ||
| 1278 | HLSSegment *en; | ||
| 1279 | |||
| 1280 |
2/2✓ Branch 0 taken 86 times.
✓ Branch 1 taken 36 times.
|
122 | while (p) { |
| 1281 | 86 | en = p; | |
| 1282 | 86 | p = p->next; | |
| 1283 | 86 | av_freep(&en); | |
| 1284 | } | ||
| 1285 | 36 | } | |
| 1286 | |||
| 1287 | ✗ | static int hls_rename_temp_file(AVFormatContext *s, AVFormatContext *oc) | |
| 1288 | { | ||
| 1289 | ✗ | size_t len = strlen(oc->url); | |
| 1290 | ✗ | char *final_filename = av_strdup(oc->url); | |
| 1291 | int ret; | ||
| 1292 | |||
| 1293 | ✗ | if (!final_filename) | |
| 1294 | ✗ | return AVERROR(ENOMEM); | |
| 1295 | ✗ | final_filename[len-4] = '\0'; | |
| 1296 | ✗ | ret = ff_rename(oc->url, final_filename, s); | |
| 1297 | ✗ | oc->url[len-4] = '\0'; | |
| 1298 | ✗ | av_freep(&final_filename); | |
| 1299 | ✗ | return ret; | |
| 1300 | } | ||
| 1301 | |||
| 1302 | ✗ | static const char* get_relative_url(const char *master_url, const char *media_url) | |
| 1303 | { | ||
| 1304 | ✗ | const char *p = strrchr(master_url, '/'); | |
| 1305 | ✗ | size_t base_len = 0; | |
| 1306 | |||
| 1307 | ✗ | if (!p) p = strrchr(master_url, '\\'); | |
| 1308 | |||
| 1309 | ✗ | if (p) { | |
| 1310 | ✗ | base_len = p - master_url; | |
| 1311 | ✗ | if (av_strncasecmp(master_url, media_url, base_len)) { | |
| 1312 | ✗ | av_log(NULL, AV_LOG_WARNING, "Unable to find relative url\n"); | |
| 1313 | ✗ | return NULL; | |
| 1314 | } | ||
| 1315 | } else { | ||
| 1316 | ✗ | return media_url; | |
| 1317 | } | ||
| 1318 | |||
| 1319 | ✗ | return media_url + base_len + 1; | |
| 1320 | } | ||
| 1321 | |||
| 1322 | ✗ | static int64_t get_stream_bit_rate(AVStream *stream) | |
| 1323 | { | ||
| 1324 | ✗ | const AVPacketSideData *sd = av_packet_side_data_get( | |
| 1325 | ✗ | stream->codecpar->coded_side_data, stream->codecpar->nb_coded_side_data, | |
| 1326 | AV_PKT_DATA_CPB_PROPERTIES | ||
| 1327 | ); | ||
| 1328 | |||
| 1329 | ✗ | if (stream->codecpar->bit_rate) | |
| 1330 | ✗ | return stream->codecpar->bit_rate; | |
| 1331 | ✗ | else if (sd) { | |
| 1332 | ✗ | AVCPBProperties *props = (AVCPBProperties*)sd->data; | |
| 1333 | ✗ | return props->max_bitrate; | |
| 1334 | } | ||
| 1335 | |||
| 1336 | ✗ | return 0; | |
| 1337 | } | ||
| 1338 | |||
| 1339 | ✗ | static int create_master_playlist(AVFormatContext *s, | |
| 1340 | VariantStream * const input_vs, | ||
| 1341 | int final) | ||
| 1342 | { | ||
| 1343 | ✗ | HLSContext *hls = s->priv_data; | |
| 1344 | VariantStream *vs, *temp_vs; | ||
| 1345 | AVStream *vid_st, *aud_st; | ||
| 1346 | ✗ | AVDictionary *options = NULL; | |
| 1347 | unsigned int i, j; | ||
| 1348 | int ret, bandwidth, avg_bandwidth; | ||
| 1349 | ✗ | const char *m3u8_rel_name = NULL; | |
| 1350 | ✗ | const char *vtt_m3u8_rel_name = NULL; | |
| 1351 | const char *ccgroup; | ||
| 1352 | ✗ | const char *sgroup = NULL; | |
| 1353 | ClosedCaptionsStream *ccs; | ||
| 1354 | ✗ | const char *proto = avio_find_protocol_name(hls->master_m3u8_url); | |
| 1355 | ✗ | int is_file_proto = proto && !strcmp(proto, "file"); | |
| 1356 | ✗ | int use_temp_file = is_file_proto && ((hls->flags & HLS_TEMP_FILE) || hls->master_publish_rate); | |
| 1357 | char temp_filename[MAX_URL_SIZE]; | ||
| 1358 | int nb_channels; | ||
| 1359 | |||
| 1360 | ✗ | input_vs->m3u8_created = 1; | |
| 1361 | ✗ | if (!hls->master_m3u8_created) { | |
| 1362 | /* For the first time, wait until all the media playlists are created */ | ||
| 1363 | ✗ | for (i = 0; i < hls->nb_varstreams; i++) | |
| 1364 | ✗ | if (!hls->var_streams[i].m3u8_created) | |
| 1365 | ✗ | return 0; | |
| 1366 | } else { | ||
| 1367 | /* Keep publishing the master playlist at the configured rate */ | ||
| 1368 | ✗ | if ((&hls->var_streams[0] != input_vs || !hls->master_publish_rate || | |
| 1369 | ✗ | input_vs->number % hls->master_publish_rate) && !final) | |
| 1370 | ✗ | return 0; | |
| 1371 | } | ||
| 1372 | |||
| 1373 | ✗ | set_http_options(s, &options, hls); | |
| 1374 | ✗ | snprintf(temp_filename, sizeof(temp_filename), use_temp_file ? "%s.tmp" : "%s", hls->master_m3u8_url); | |
| 1375 | ✗ | ret = hlsenc_io_open(s, &hls->m3u8_out, temp_filename, &options); | |
| 1376 | ✗ | av_dict_free(&options); | |
| 1377 | ✗ | if (ret < 0) { | |
| 1378 | ✗ | av_log(s, AV_LOG_ERROR, "Failed to open master play list file '%s'\n", | |
| 1379 | temp_filename); | ||
| 1380 | ✗ | goto fail; | |
| 1381 | } | ||
| 1382 | |||
| 1383 | ✗ | ff_hls_write_playlist_version(hls->m3u8_out, hls->version); | |
| 1384 | |||
| 1385 | ✗ | for (i = 0; i < hls->nb_ccstreams; i++) { | |
| 1386 | ✗ | ccs = &(hls->cc_streams[i]); | |
| 1387 | ✗ | avio_printf(hls->m3u8_out, "#EXT-X-MEDIA:TYPE=CLOSED-CAPTIONS"); | |
| 1388 | ✗ | avio_printf(hls->m3u8_out, ",GROUP-ID=\"%s\"", ccs->ccgroup); | |
| 1389 | ✗ | avio_printf(hls->m3u8_out, ",NAME=\"%s\"", ccs->instreamid); | |
| 1390 | ✗ | if (ccs->language) | |
| 1391 | ✗ | avio_printf(hls->m3u8_out, ",LANGUAGE=\"%s\"", ccs->language); | |
| 1392 | ✗ | avio_printf(hls->m3u8_out, ",INSTREAM-ID=\"%s\"\n", ccs->instreamid); | |
| 1393 | } | ||
| 1394 | |||
| 1395 | /* For audio only variant streams add #EXT-X-MEDIA tag with attributes*/ | ||
| 1396 | ✗ | for (i = 0; i < hls->nb_varstreams; i++) { | |
| 1397 | ✗ | vs = &(hls->var_streams[i]); | |
| 1398 | |||
| 1399 | ✗ | if (vs->has_video || vs->has_subtitle || !vs->agroup) | |
| 1400 | ✗ | continue; | |
| 1401 | |||
| 1402 | ✗ | m3u8_rel_name = get_relative_url(hls->master_m3u8_url, vs->m3u8_name); | |
| 1403 | ✗ | if (!m3u8_rel_name) { | |
| 1404 | ✗ | av_log(s, AV_LOG_ERROR, "Unable to find relative URL\n"); | |
| 1405 | ✗ | goto fail; | |
| 1406 | } | ||
| 1407 | ✗ | nb_channels = 0; | |
| 1408 | ✗ | for (j = 0; j < vs->nb_streams; j++) | |
| 1409 | ✗ | if (vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) | |
| 1410 | ✗ | if (vs->streams[j]->codecpar->ch_layout.nb_channels > nb_channels) | |
| 1411 | ✗ | nb_channels = vs->streams[j]->codecpar->ch_layout.nb_channels; | |
| 1412 | |||
| 1413 | ✗ | ff_hls_write_audio_rendition(hls->m3u8_out, vs->agroup, m3u8_rel_name, vs->language, i, hls->has_default_key ? vs->is_default : 1, nb_channels); | |
| 1414 | } | ||
| 1415 | |||
| 1416 | /* For variant streams with video add #EXT-X-STREAM-INF tag with attributes*/ | ||
| 1417 | ✗ | for (i = 0; i < hls->nb_varstreams; i++) { | |
| 1418 | ✗ | vs = &(hls->var_streams[i]); | |
| 1419 | |||
| 1420 | ✗ | m3u8_rel_name = get_relative_url(hls->master_m3u8_url, vs->m3u8_name); | |
| 1421 | ✗ | if (!m3u8_rel_name) { | |
| 1422 | ✗ | av_log(s, AV_LOG_ERROR, "Unable to find relative URL\n"); | |
| 1423 | ✗ | goto fail; | |
| 1424 | } | ||
| 1425 | |||
| 1426 | ✗ | vid_st = NULL; | |
| 1427 | ✗ | aud_st = NULL; | |
| 1428 | ✗ | for (j = 0; j < vs->nb_streams; j++) { | |
| 1429 | ✗ | if (vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) | |
| 1430 | ✗ | vid_st = vs->streams[j]; | |
| 1431 | ✗ | else if (vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) | |
| 1432 | ✗ | aud_st = vs->streams[j]; | |
| 1433 | } | ||
| 1434 | |||
| 1435 | ✗ | if (!vid_st && !aud_st) { | |
| 1436 | ✗ | av_log(s, AV_LOG_WARNING, "Media stream not found\n"); | |
| 1437 | ✗ | continue; | |
| 1438 | } | ||
| 1439 | |||
| 1440 | /** | ||
| 1441 | * Traverse through the list of audio only rendition streams and find | ||
| 1442 | * the rendition which has highest bitrate in the same audio group | ||
| 1443 | */ | ||
| 1444 | ✗ | if (vs->agroup) { | |
| 1445 | ✗ | for (j = 0; j < hls->nb_varstreams; j++) { | |
| 1446 | ✗ | temp_vs = &(hls->var_streams[j]); | |
| 1447 | ✗ | if (!temp_vs->has_video && !temp_vs->has_subtitle && | |
| 1448 | ✗ | temp_vs->agroup && | |
| 1449 | ✗ | !av_strcasecmp(temp_vs->agroup, vs->agroup)) { | |
| 1450 | ✗ | if (!aud_st) | |
| 1451 | ✗ | aud_st = temp_vs->streams[0]; | |
| 1452 | ✗ | if (temp_vs->streams[0]->codecpar->bit_rate > | |
| 1453 | ✗ | aud_st->codecpar->bit_rate) | |
| 1454 | ✗ | aud_st = temp_vs->streams[0]; | |
| 1455 | } | ||
| 1456 | } | ||
| 1457 | } | ||
| 1458 | |||
| 1459 | ✗ | if (final) { | |
| 1460 | ✗ | bandwidth = vs->max_bitrate; | |
| 1461 | ✗ | avg_bandwidth = vs->avg_bitrate; | |
| 1462 | } else { | ||
| 1463 | ✗ | bandwidth = 0; | |
| 1464 | ✗ | if (vid_st) | |
| 1465 | ✗ | bandwidth += get_stream_bit_rate(vid_st); | |
| 1466 | ✗ | if (aud_st) | |
| 1467 | ✗ | bandwidth += get_stream_bit_rate(aud_st); | |
| 1468 | ✗ | bandwidth += bandwidth / 10; | |
| 1469 | } | ||
| 1470 | |||
| 1471 | ✗ | ccgroup = NULL; | |
| 1472 | ✗ | if (vid_st && vs->ccgroup) { | |
| 1473 | /* check if this group name is available in the cc map string */ | ||
| 1474 | ✗ | for (j = 0; j < hls->nb_ccstreams; j++) { | |
| 1475 | ✗ | ccs = &(hls->cc_streams[j]); | |
| 1476 | ✗ | if (!av_strcasecmp(ccs->ccgroup, vs->ccgroup)) { | |
| 1477 | ✗ | ccgroup = vs->ccgroup; | |
| 1478 | ✗ | break; | |
| 1479 | } | ||
| 1480 | } | ||
| 1481 | ✗ | if (j == hls->nb_ccstreams) | |
| 1482 | ✗ | av_log(s, AV_LOG_WARNING, "mapping ccgroup %s not found\n", | |
| 1483 | vs->ccgroup); | ||
| 1484 | } | ||
| 1485 | |||
| 1486 | ✗ | if (vid_st && vs->sgroup) { | |
| 1487 | ✗ | sgroup = vs->sgroup; | |
| 1488 | ✗ | vtt_m3u8_rel_name = get_relative_url(hls->master_m3u8_url, vs->vtt_m3u8_name); | |
| 1489 | ✗ | if (!vtt_m3u8_rel_name) { | |
| 1490 | ✗ | av_log(s, AV_LOG_WARNING, "Unable to find relative subtitle URL\n"); | |
| 1491 | ✗ | break; | |
| 1492 | } | ||
| 1493 | |||
| 1494 | ✗ | ff_hls_write_subtitle_rendition(hls->m3u8_out, sgroup, vtt_m3u8_rel_name, vs->language, | |
| 1495 | ✗ | vs->subtitle_varname, i, hls->has_default_key ? vs->is_default : 1); | |
| 1496 | } | ||
| 1497 | |||
| 1498 | ✗ | if (!hls->has_default_key || !hls->has_video_m3u8) { | |
| 1499 | ✗ | ff_hls_write_stream_info(vid_st, hls->m3u8_out, bandwidth, avg_bandwidth, m3u8_rel_name, | |
| 1500 | ✗ | aud_st ? vs->agroup : NULL, vs->codec_attr, ccgroup, sgroup); | |
| 1501 | } else { | ||
| 1502 | ✗ | if (vid_st) { | |
| 1503 | ✗ | ff_hls_write_stream_info(vid_st, hls->m3u8_out, bandwidth, avg_bandwidth, m3u8_rel_name, | |
| 1504 | ✗ | aud_st ? vs->agroup : NULL, vs->codec_attr, ccgroup, sgroup); | |
| 1505 | } | ||
| 1506 | } | ||
| 1507 | } | ||
| 1508 | ✗ | fail: | |
| 1509 | ✗ | if (ret >=0) | |
| 1510 | ✗ | hls->master_m3u8_created = 1; | |
| 1511 | ✗ | hlsenc_io_close(s, &hls->m3u8_out, temp_filename); | |
| 1512 | ✗ | if (use_temp_file) | |
| 1513 | ✗ | ff_rename(temp_filename, hls->master_m3u8_url, s); | |
| 1514 | |||
| 1515 | ✗ | return ret; | |
| 1516 | } | ||
| 1517 | |||
| 1518 | 90 | static int hls_window(AVFormatContext *s, int last, VariantStream *vs) | |
| 1519 | { | ||
| 1520 | 90 | HLSContext *hls = s->priv_data; | |
| 1521 | HLSSegment *en; | ||
| 1522 | 90 | int target_duration = 0; | |
| 1523 | 90 | int ret = 0; | |
| 1524 | char temp_filename[MAX_URL_SIZE]; | ||
| 1525 | char temp_vtt_filename[MAX_URL_SIZE]; | ||
| 1526 | 90 | int64_t sequence = FFMAX(hls->start_sequence, vs->sequence - vs->nb_entries); | |
| 1527 | 90 | const char *proto = avio_find_protocol_name(vs->m3u8_name); | |
| 1528 |
2/4✓ Branch 0 taken 90 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 90 times.
✗ Branch 3 not taken.
|
90 | int is_file_proto = proto && !strcmp(proto, "file"); |
| 1529 |
4/6✓ Branch 0 taken 90 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 90 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 89 times.
✓ Branch 5 taken 1 times.
|
90 | int use_temp_file = is_file_proto && ((hls->flags & HLS_TEMP_FILE) || !(hls->pl_type == PLAYLIST_TYPE_VOD)); |
| 1530 | static unsigned warned_non_file; | ||
| 1531 | 90 | char *key_uri = NULL; | |
| 1532 | 90 | char *iv_string = NULL; | |
| 1533 | 90 | AVDictionary *options = NULL; | |
| 1534 | 90 | double prog_date_time = vs->initial_prog_date_time; | |
| 1535 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 90 times.
|
90 | double *prog_date_time_p = (hls->flags & HLS_PROGRAM_DATE_TIME) ? &prog_date_time : NULL; |
| 1536 |
4/4✓ Branch 0 taken 80 times.
✓ Branch 1 taken 10 times.
✓ Branch 2 taken 10 times.
✓ Branch 3 taken 70 times.
|
90 | int byterange_mode = (hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size > 0); |
| 1537 | |||
| 1538 | 90 | hls->version = 2; | |
| 1539 |
2/2✓ Branch 0 taken 87 times.
✓ Branch 1 taken 3 times.
|
90 | if (!(hls->flags & HLS_ROUND_DURATIONS)) { |
| 1540 | 87 | hls->version = 3; | |
| 1541 | } | ||
| 1542 | |||
| 1543 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 70 times.
|
90 | if (byterange_mode) { |
| 1544 | 20 | hls->version = 4; | |
| 1545 | 20 | sequence = 0; | |
| 1546 | } | ||
| 1547 | |||
| 1548 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 90 times.
|
90 | if (hls->flags & HLS_I_FRAMES_ONLY) { |
| 1549 | ✗ | hls->version = 4; | |
| 1550 | } | ||
| 1551 | |||
| 1552 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 87 times.
|
90 | if (hls->flags & HLS_INDEPENDENT_SEGMENTS) { |
| 1553 | 3 | hls->version = 6; | |
| 1554 | } | ||
| 1555 | |||
| 1556 |
2/2✓ Branch 0 taken 6 times.
✓ Branch 1 taken 84 times.
|
90 | if (hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 1557 | 6 | hls->version = 7; | |
| 1558 | } | ||
| 1559 | |||
| 1560 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 90 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
90 | if (!is_file_proto && (hls->flags & HLS_TEMP_FILE) && !warned_non_file++) |
| 1561 | ✗ | av_log(s, AV_LOG_ERROR, "Cannot use rename on non file protocol, this may lead to races and temporary partial files\n"); | |
| 1562 | |||
| 1563 | 90 | set_http_options(s, &options, hls); | |
| 1564 |
2/2✓ Branch 0 taken 89 times.
✓ Branch 1 taken 1 times.
|
90 | snprintf(temp_filename, sizeof(temp_filename), use_temp_file ? "%s.tmp" : "%s", vs->m3u8_name); |
| 1565 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 70 times.
|
90 | ret = hlsenc_io_open(s, byterange_mode ? &hls->m3u8_out : &vs->out, temp_filename, &options); |
| 1566 | 90 | av_dict_free(&options); | |
| 1567 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 90 times.
|
90 | if (ret < 0) { |
| 1568 | ✗ | goto fail; | |
| 1569 | } | ||
| 1570 | |||
| 1571 |
2/2✓ Branch 0 taken 331 times.
✓ Branch 1 taken 90 times.
|
421 | for (en = vs->segments; en; en = en->next) { |
| 1572 |
2/2✓ Branch 0 taken 247 times.
✓ Branch 1 taken 84 times.
|
331 | if (target_duration <= en->duration) |
| 1573 | 247 | target_duration = lrint(en->duration); | |
| 1574 | } | ||
| 1575 | |||
| 1576 | 90 | vs->discontinuity_set = 0; | |
| 1577 | 90 | ff_hls_write_playlist_header(byterange_mode ? hls->m3u8_out : vs->out, hls->version, hls->allowcache, | |
| 1578 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 70 times.
|
90 | target_duration, sequence, hls->pl_type, hls->flags & HLS_I_FRAMES_ONLY); |
| 1579 | |||
| 1580 |
4/6✓ Branch 0 taken 3 times.
✓ Branch 1 taken 87 times.
✓ Branch 2 taken 3 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 3 times.
✗ Branch 5 not taken.
|
90 | if ((hls->flags & HLS_DISCONT_START) && sequence==hls->start_sequence && vs->discontinuity_set==0) { |
| 1581 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | avio_printf(byterange_mode ? hls->m3u8_out : vs->out, "#EXT-X-DISCONTINUITY\n"); |
| 1582 | 3 | vs->discontinuity_set = 1; | |
| 1583 | } | ||
| 1584 |
4/4✓ Branch 0 taken 16 times.
✓ Branch 1 taken 74 times.
✓ Branch 2 taken 3 times.
✓ Branch 3 taken 13 times.
|
90 | if (vs->has_video && (hls->flags & HLS_INDEPENDENT_SEGMENTS)) { |
| 1585 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | avio_printf(byterange_mode ? hls->m3u8_out : vs->out, "#EXT-X-INDEPENDENT-SEGMENTS\n"); |
| 1586 | } | ||
| 1587 |
2/2✓ Branch 0 taken 331 times.
✓ Branch 1 taken 90 times.
|
421 | for (en = vs->segments; en; en = en->next) { |
| 1588 |
2/10✓ Branch 0 taken 331 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 331 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
✗ Branch 8 not taken.
✗ Branch 9 not taken.
|
331 | if ((hls->encrypt || hls->key_info_file) && (!key_uri || strcmp(en->key_uri, key_uri) || |
| 1589 | ✗ | av_strcasecmp(en->iv_string, iv_string))) { | |
| 1590 | ✗ | avio_printf(byterange_mode ? hls->m3u8_out : vs->out, "#EXT-X-KEY:METHOD=AES-128,URI=\"%s\"", en->key_uri); | |
| 1591 | ✗ | if (*en->iv_string) | |
| 1592 | ✗ | avio_printf(byterange_mode ? hls->m3u8_out : vs->out, ",IV=0x%s", en->iv_string); | |
| 1593 | ✗ | avio_printf(byterange_mode ? hls->m3u8_out : vs->out, "\n"); | |
| 1594 | ✗ | key_uri = en->key_uri; | |
| 1595 | ✗ | iv_string = en->iv_string; | |
| 1596 | } | ||
| 1597 | |||
| 1598 |
4/4✓ Branch 0 taken 16 times.
✓ Branch 1 taken 315 times.
✓ Branch 2 taken 6 times.
✓ Branch 3 taken 10 times.
|
331 | if ((hls->segment_type == SEGMENT_TYPE_FMP4) && (en == vs->segments)) { |
| 1599 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | ff_hls_write_init_file(byterange_mode ? hls->m3u8_out : vs->out, (hls->flags & HLS_SINGLE_FILE) ? en->filename : vs->fmp4_init_filename, |
| 1600 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
|
6 | hls->flags & HLS_SINGLE_FILE, vs->init_range_length, 0); |
| 1601 | } | ||
| 1602 | |||
| 1603 | 662 | ret = ff_hls_write_file_entry(byterange_mode ? hls->m3u8_out : vs->out, en->discont, byterange_mode, | |
| 1604 | 331 | en->duration, hls->flags & HLS_ROUND_DURATIONS, | |
| 1605 | 331 | en->size, en->pos, hls->baseurl, | |
| 1606 |
2/2✓ Branch 0 taken 110 times.
✓ Branch 1 taken 221 times.
|
331 | en->filename, |
| 1607 | 331 | en->discont_program_date_time ? &en->discont_program_date_time : prog_date_time_p, | |
| 1608 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 331 times.
|
331 | en->keyframe_size, en->keyframe_pos, hls->flags & HLS_I_FRAMES_ONLY); |
| 1609 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 331 times.
|
331 | if (en->discont_program_date_time) |
| 1610 | ✗ | en->discont_program_date_time -= en->duration; | |
| 1611 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 331 times.
|
331 | if (ret < 0) { |
| 1612 | ✗ | av_log(s, AV_LOG_WARNING, "ff_hls_write_file_entry get error\n"); | |
| 1613 | } | ||
| 1614 | } | ||
| 1615 | |||
| 1616 |
4/4✓ Branch 0 taken 18 times.
✓ Branch 1 taken 72 times.
✓ Branch 2 taken 16 times.
✓ Branch 3 taken 2 times.
|
90 | if (last && (hls->flags & HLS_OMIT_ENDLIST)==0) |
| 1617 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 14 times.
|
16 | ff_hls_write_end_list(byterange_mode ? hls->m3u8_out : vs->out); |
| 1618 | |||
| 1619 |
1/2✓ Branch 0 taken 90 times.
✗ Branch 1 not taken.
|
90 | if (vs->vtt_m3u8_name) { |
| 1620 | ✗ | set_http_options(vs->vtt_avf, &options, hls); | |
| 1621 | ✗ | snprintf(temp_vtt_filename, sizeof(temp_vtt_filename), use_temp_file ? "%s.tmp" : "%s", vs->vtt_m3u8_name); | |
| 1622 | ✗ | ret = hlsenc_io_open(s, &hls->sub_m3u8_out, temp_vtt_filename, &options); | |
| 1623 | ✗ | av_dict_free(&options); | |
| 1624 | ✗ | if (ret < 0) { | |
| 1625 | ✗ | goto fail; | |
| 1626 | } | ||
| 1627 | ✗ | ff_hls_write_playlist_header(hls->sub_m3u8_out, hls->version, hls->allowcache, | |
| 1628 | target_duration, sequence, PLAYLIST_TYPE_NONE, 0); | ||
| 1629 | ✗ | for (en = vs->segments; en; en = en->next) { | |
| 1630 | ✗ | ret = ff_hls_write_file_entry(hls->sub_m3u8_out, en->discont, byterange_mode, | |
| 1631 | en->duration, 0, en->size, en->pos, | ||
| 1632 | ✗ | hls->baseurl, en->sub_filename, NULL, 0, 0, 0); | |
| 1633 | ✗ | if (ret < 0) { | |
| 1634 | ✗ | av_log(s, AV_LOG_WARNING, "ff_hls_write_file_entry get error\n"); | |
| 1635 | } | ||
| 1636 | } | ||
| 1637 | |||
| 1638 | ✗ | if (last && !(hls->flags & HLS_OMIT_ENDLIST)) | |
| 1639 | ✗ | ff_hls_write_end_list(hls->sub_m3u8_out); | |
| 1640 | |||
| 1641 | } | ||
| 1642 | |||
| 1643 | 90 | fail: | |
| 1644 | 90 | av_dict_free(&options); | |
| 1645 |
2/2✓ Branch 0 taken 20 times.
✓ Branch 1 taken 70 times.
|
90 | ret = hlsenc_io_close(s, byterange_mode ? &hls->m3u8_out : &vs->out, temp_filename); |
| 1646 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 90 times.
|
90 | if (ret < 0) { |
| 1647 | ✗ | return ret; | |
| 1648 | } | ||
| 1649 | 90 | hlsenc_io_close(s, &hls->sub_m3u8_out, vs->vtt_m3u8_name); | |
| 1650 |
2/2✓ Branch 0 taken 89 times.
✓ Branch 1 taken 1 times.
|
90 | if (use_temp_file) { |
| 1651 | 89 | ff_rename(temp_filename, vs->m3u8_name, s); | |
| 1652 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 89 times.
|
89 | if (vs->vtt_m3u8_name) |
| 1653 | ✗ | ff_rename(temp_vtt_filename, vs->vtt_m3u8_name, s); | |
| 1654 | } | ||
| 1655 |
2/4✓ Branch 0 taken 90 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 90 times.
|
90 | if (ret >= 0 && hls->master_pl_name) |
| 1656 | ✗ | if (create_master_playlist(s, vs, last) < 0) | |
| 1657 | ✗ | av_log(s, AV_LOG_WARNING, "Master playlist creation failed\n"); | |
| 1658 | |||
| 1659 | 90 | return ret; | |
| 1660 | } | ||
| 1661 | |||
| 1662 | 77 | static int hls_start(AVFormatContext *s, VariantStream *vs) | |
| 1663 | { | ||
| 1664 | 77 | HLSContext *c = s->priv_data; | |
| 1665 | 77 | AVFormatContext *oc = vs->avf; | |
| 1666 | 77 | AVFormatContext *vtt_oc = vs->vtt_avf; | |
| 1667 | 77 | AVDictionary *options = NULL; | |
| 1668 | 77 | const char *proto = NULL; | |
| 1669 | 77 | int use_temp_file = 0; | |
| 1670 | char iv_string[KEYSIZE*2 + 1]; | ||
| 1671 | 77 | int err = 0; | |
| 1672 | |||
| 1673 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 76 times.
|
77 | if (c->flags & HLS_SINGLE_FILE) { |
| 1674 | 1 | char *new_name = av_strdup(vs->basename); | |
| 1675 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (!new_name) |
| 1676 | ✗ | return AVERROR(ENOMEM); | |
| 1677 | 1 | ff_format_set_url(oc, new_name); | |
| 1678 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (vs->vtt_basename) { |
| 1679 | ✗ | new_name = av_strdup(vs->vtt_basename); | |
| 1680 | ✗ | if (!new_name) | |
| 1681 | ✗ | return AVERROR(ENOMEM); | |
| 1682 | ✗ | ff_format_set_url(vtt_oc, new_name); | |
| 1683 | } | ||
| 1684 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 72 times.
|
76 | } else if (c->max_seg_size > 0) { |
| 1685 | 4 | char *filename = NULL; | |
| 1686 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
|
4 | if (replace_int_data_in_filename(&filename, |
| 1687 | 4 | vs->basename, 'd', vs->sequence) < 1) { | |
| 1688 | ✗ | av_freep(&filename); | |
| 1689 | ✗ | av_log(oc, AV_LOG_ERROR, "Invalid segment filename template '%s', you can try to use -strftime 1 with it\n", vs->basename); | |
| 1690 | ✗ | return AVERROR(EINVAL); | |
| 1691 | } | ||
| 1692 | 4 | ff_format_set_url(oc, filename); | |
| 1693 | } else { | ||
| 1694 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 72 times.
|
72 | if (c->use_localtime) { |
| 1695 | int r; | ||
| 1696 | ✗ | char *expanded = NULL; | |
| 1697 | |||
| 1698 | ✗ | r = strftime_expand(vs->basename, &expanded); | |
| 1699 | ✗ | if (r < 0) { | |
| 1700 | ✗ | av_log(oc, AV_LOG_ERROR, "Could not get segment filename with strftime\n"); | |
| 1701 | ✗ | return r; | |
| 1702 | } | ||
| 1703 | ✗ | ff_format_set_url(oc, expanded); | |
| 1704 | |||
| 1705 | ✗ | err = sls_flag_use_localtime_filename(oc, c, vs); | |
| 1706 | ✗ | if (err < 0) { | |
| 1707 | ✗ | return AVERROR(ENOMEM); | |
| 1708 | } | ||
| 1709 | |||
| 1710 | ✗ | if (c->use_localtime_mkdir) { | |
| 1711 | const char *dir; | ||
| 1712 | ✗ | char *fn_copy = av_strdup(oc->url); | |
| 1713 | ✗ | if (!fn_copy) | |
| 1714 | ✗ | return AVERROR(ENOMEM); | |
| 1715 | ✗ | dir = av_dirname(fn_copy); | |
| 1716 | ✗ | if (ff_mkdir_p(dir) == -1 && errno != EEXIST) { | |
| 1717 | ✗ | av_log(oc, AV_LOG_ERROR, "Could not create directory %s with use_localtime_mkdir\n", dir); | |
| 1718 | ✗ | av_freep(&fn_copy); | |
| 1719 | ✗ | return AVERROR(errno); | |
| 1720 | } | ||
| 1721 | ✗ | av_freep(&fn_copy); | |
| 1722 | } | ||
| 1723 | } else { | ||
| 1724 | 72 | char *filename = NULL; | |
| 1725 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 72 times.
|
72 | if (replace_int_data_in_filename(&filename, |
| 1726 | 72 | vs->basename, 'd', vs->sequence) < 1) { | |
| 1727 | ✗ | av_freep(&filename); | |
| 1728 | ✗ | av_log(oc, AV_LOG_ERROR, "Invalid segment filename template '%s' you can try to use -strftime 1 with it\n", vs->basename); | |
| 1729 | ✗ | return AVERROR(EINVAL); | |
| 1730 | } | ||
| 1731 | 72 | ff_format_set_url(oc, filename); | |
| 1732 | } | ||
| 1733 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 72 times.
|
72 | if (vs->vtt_basename) { |
| 1734 | ✗ | char *filename = NULL; | |
| 1735 | ✗ | if (replace_int_data_in_filename(&filename, | |
| 1736 | ✗ | vs->vtt_basename, 'd', vs->sequence) < 1) { | |
| 1737 | ✗ | av_freep(&filename); | |
| 1738 | ✗ | av_log(vtt_oc, AV_LOG_ERROR, "Invalid segment filename template '%s'\n", vs->vtt_basename); | |
| 1739 | ✗ | return AVERROR(EINVAL); | |
| 1740 | } | ||
| 1741 | ✗ | ff_format_set_url(vtt_oc, filename); | |
| 1742 | } | ||
| 1743 | } | ||
| 1744 | |||
| 1745 | 77 | proto = avio_find_protocol_name(oc->url); | |
| 1746 |
3/6✓ Branch 0 taken 77 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 77 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 77 times.
|
77 | use_temp_file = proto && !strcmp(proto, "file") && (c->flags & HLS_TEMP_FILE); |
| 1747 | |||
| 1748 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 77 times.
|
77 | if (use_temp_file) { |
| 1749 | ✗ | char *new_name = av_asprintf("%s.tmp", oc->url); | |
| 1750 | ✗ | if (!new_name) | |
| 1751 | ✗ | return AVERROR(ENOMEM); | |
| 1752 | ✗ | ff_format_set_url(oc, new_name); | |
| 1753 | } | ||
| 1754 | |||
| 1755 |
2/4✓ Branch 0 taken 77 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 77 times.
|
77 | if (c->key_info_file || c->encrypt) { |
| 1756 | ✗ | if (c->segment_type == SEGMENT_TYPE_FMP4) { | |
| 1757 | ✗ | av_log(s, AV_LOG_ERROR, "Encrypted fmp4 not yet supported\n"); | |
| 1758 | ✗ | return AVERROR_PATCHWELCOME; | |
| 1759 | } | ||
| 1760 | |||
| 1761 | ✗ | if (c->key_info_file && c->encrypt) { | |
| 1762 | ✗ | av_log(s, AV_LOG_WARNING, "Cannot use both -hls_key_info_file and -hls_enc," | |
| 1763 | " ignoring -hls_enc\n"); | ||
| 1764 | } | ||
| 1765 | |||
| 1766 | ✗ | if (!vs->encrypt_started || (c->flags & HLS_PERIODIC_REKEY)) { | |
| 1767 | ✗ | if (c->key_info_file) { | |
| 1768 | ✗ | if ((err = hls_encryption_start(s, vs)) < 0) | |
| 1769 | ✗ | goto fail; | |
| 1770 | } else { | ||
| 1771 | ✗ | if (!c->encrypt_started) { | |
| 1772 | ✗ | if ((err = do_encrypt(s, vs)) < 0) | |
| 1773 | ✗ | goto fail; | |
| 1774 | ✗ | c->encrypt_started = 1; | |
| 1775 | } | ||
| 1776 | ✗ | av_strlcpy(vs->key_uri, c->key_uri, sizeof(vs->key_uri)); | |
| 1777 | ✗ | av_strlcpy(vs->key_string, c->key_string, sizeof(vs->key_string)); | |
| 1778 | ✗ | av_strlcpy(vs->iv_string, c->iv_string, sizeof(vs->iv_string)); | |
| 1779 | } | ||
| 1780 | ✗ | vs->encrypt_started = 1; | |
| 1781 | } | ||
| 1782 | ✗ | err = av_strlcpy(iv_string, vs->iv_string, sizeof(iv_string)); | |
| 1783 | ✗ | if (!err) { | |
| 1784 | ✗ | snprintf(iv_string, sizeof(iv_string), "%032"PRIx64, vs->sequence); | |
| 1785 | ✗ | memset(vs->iv_string, 0, sizeof(vs->iv_string)); | |
| 1786 | ✗ | memcpy(vs->iv_string, iv_string, sizeof(iv_string)); | |
| 1787 | } | ||
| 1788 | } | ||
| 1789 |
2/2✓ Branch 0 taken 71 times.
✓ Branch 1 taken 6 times.
|
77 | if (c->segment_type != SEGMENT_TYPE_FMP4) { |
| 1790 |
2/4✓ Branch 0 taken 71 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 71 times.
✗ Branch 3 not taken.
|
71 | if (oc->oformat->priv_class && oc->priv_data) { |
| 1791 | 71 | av_opt_set(oc->priv_data, "mpegts_flags", "resend_headers", 0); | |
| 1792 | } | ||
| 1793 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 70 times.
|
71 | if (c->flags & HLS_SINGLE_FILE) { |
| 1794 |
2/4✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
|
1 | if (c->key_info_file || c->encrypt) { |
| 1795 | ✗ | av_dict_set(&options, "encryption_key", vs->key_string, 0); | |
| 1796 | ✗ | av_dict_set(&options, "encryption_iv", vs->iv_string, 0); | |
| 1797 | |||
| 1798 | /* Write temp file with cryption content */ | ||
| 1799 | ✗ | av_freep(&vs->basename_tmp); | |
| 1800 | ✗ | vs->basename_tmp = av_asprintf("crypto:%s.tmp", oc->url); | |
| 1801 | |||
| 1802 | /* append temp file content into single file */ | ||
| 1803 | ✗ | av_freep(&vs->basename); | |
| 1804 | ✗ | vs->basename = av_asprintf("%s", oc->url); | |
| 1805 | } else { | ||
| 1806 | 1 | vs->basename_tmp = vs->basename; | |
| 1807 | } | ||
| 1808 | 1 | set_http_options(s, &options, c); | |
| 1809 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (!vs->out_single_file) |
| 1810 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
|
1 | if ((err = hlsenc_io_open(s, &vs->out_single_file, vs->basename, &options)) < 0) { |
| 1811 | ✗ | if (c->ignore_io_errors) | |
| 1812 | ✗ | err = 0; | |
| 1813 | ✗ | goto fail; | |
| 1814 | } | ||
| 1815 | |||
| 1816 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 1 times.
|
1 | if ((err = hlsenc_io_open(s, &vs->out, vs->basename_tmp, &options)) < 0) { |
| 1817 | ✗ | if (c->ignore_io_errors) | |
| 1818 | ✗ | err = 0; | |
| 1819 | ✗ | goto fail; | |
| 1820 | } | ||
| 1821 | |||
| 1822 | } | ||
| 1823 | } | ||
| 1824 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 77 times.
|
77 | if (vs->vtt_basename) { |
| 1825 | ✗ | set_http_options(s, &options, c); | |
| 1826 | ✗ | if ((err = hlsenc_io_open(s, &vtt_oc->pb, vtt_oc->url, &options)) < 0) { | |
| 1827 | ✗ | if (c->ignore_io_errors) | |
| 1828 | ✗ | err = 0; | |
| 1829 | ✗ | goto fail; | |
| 1830 | } | ||
| 1831 | } | ||
| 1832 | 77 | av_dict_free(&options); | |
| 1833 | |||
| 1834 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 77 times.
|
77 | if (vs->vtt_basename) { |
| 1835 | ✗ | err = avformat_write_header(vtt_oc,NULL); | |
| 1836 | ✗ | if (err < 0) | |
| 1837 | ✗ | return err; | |
| 1838 | } | ||
| 1839 | |||
| 1840 | 77 | return 0; | |
| 1841 | ✗ | fail: | |
| 1842 | ✗ | av_dict_free(&options); | |
| 1843 | |||
| 1844 | ✗ | return err; | |
| 1845 | } | ||
| 1846 | |||
| 1847 | ✗ | static const char * get_default_pattern_localtime_fmt(AVFormatContext *s) | |
| 1848 | { | ||
| 1849 | ✗ | HLSContext *hls = s->priv_data; | |
| 1850 | #if HAVE_LIBC_MSVCRT | ||
| 1851 | // no %s support on MSVC, which invokes the invalid parameter handler | ||
| 1852 | // on unsupported format strings, instead of returning an error | ||
| 1853 | int strftime_s_supported = 0; | ||
| 1854 | #else | ||
| 1855 | char b[21]; | ||
| 1856 | ✗ | time_t t = time(NULL); | |
| 1857 | ✗ | struct tm tmbuf, *p = localtime_r(&t, &tmbuf); | |
| 1858 | // no %s support when strftime returned error or left format string unchanged | ||
| 1859 | ✗ | int strftime_s_supported = strftime(b, sizeof(b), "%s", p) && strcmp(b, "%s"); | |
| 1860 | #endif | ||
| 1861 | |||
| 1862 | ✗ | if (hls->segment_type == SEGMENT_TYPE_FMP4) { | |
| 1863 | ✗ | return strftime_s_supported ? "-%s.m4s" : "-%Y%m%d%H%M%S.m4s"; | |
| 1864 | } | ||
| 1865 | ✗ | return strftime_s_supported ? "-%s.ts" : "-%Y%m%d%H%M%S.ts"; | |
| 1866 | } | ||
| 1867 | |||
| 1868 | ✗ | static int append_postfix(char *name, int name_buf_len, int i) | |
| 1869 | { | ||
| 1870 | char *p; | ||
| 1871 | ✗ | char extension[10] = {'\0'}; | |
| 1872 | |||
| 1873 | ✗ | p = strrchr(name, '.'); | |
| 1874 | ✗ | if (p) { | |
| 1875 | ✗ | av_strlcpy(extension, p, sizeof(extension)); | |
| 1876 | ✗ | *p = '\0'; | |
| 1877 | } | ||
| 1878 | |||
| 1879 | ✗ | snprintf(name + strlen(name), name_buf_len - strlen(name), POSTFIX_PATTERN, i); | |
| 1880 | |||
| 1881 | ✗ | if (strlen(extension)) | |
| 1882 | ✗ | av_strlcat(name, extension, name_buf_len); | |
| 1883 | |||
| 1884 | ✗ | return 0; | |
| 1885 | } | ||
| 1886 | |||
| 1887 | 39 | static int validate_name(int nb_vs, const char *fn) | |
| 1888 | { | ||
| 1889 | const char *filename, *subdir_name; | ||
| 1890 | 39 | char *fn_dup = NULL; | |
| 1891 | 39 | int ret = 0; | |
| 1892 | |||
| 1893 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 39 times.
|
39 | if (!fn) |
| 1894 | ✗ | return AVERROR(EINVAL); | |
| 1895 | |||
| 1896 | 39 | fn_dup = av_strdup(fn); | |
| 1897 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 39 times.
|
39 | if (!fn_dup) |
| 1898 | ✗ | return AVERROR(ENOMEM); | |
| 1899 | 39 | filename = av_basename(fn); | |
| 1900 | 39 | subdir_name = av_dirname(fn_dup); | |
| 1901 | |||
| 1902 |
1/6✗ Branch 0 not taken.
✓ Branch 1 taken 39 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 6 not taken.
✗ Branch 7 not taken.
|
39 | if (nb_vs > 1 && !av_stristr(filename, "%v") && !av_stristr(subdir_name, "%v")) { |
| 1903 | ✗ | av_log(NULL, AV_LOG_ERROR, "More than 1 variant streams are present, %%v is expected " | |
| 1904 | "either in the filename or in the sub-directory name of file %s\n", fn); | ||
| 1905 | ✗ | ret = AVERROR(EINVAL); | |
| 1906 | ✗ | goto fail; | |
| 1907 | } | ||
| 1908 | |||
| 1909 |
1/4✓ Branch 1 taken 39 times.
✗ Branch 2 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
39 | if (av_stristr(filename, "%v") && av_stristr(subdir_name, "%v")) { |
| 1910 | ✗ | av_log(NULL, AV_LOG_ERROR, "%%v is expected either in the filename or " | |
| 1911 | "in the sub-directory name of file %s, but only in one of them\n", fn); | ||
| 1912 | ✗ | ret = AVERROR(EINVAL); | |
| 1913 | ✗ | goto fail; | |
| 1914 | } | ||
| 1915 | |||
| 1916 | 39 | fail: | |
| 1917 | 39 | av_freep(&fn_dup); | |
| 1918 | 39 | return ret; | |
| 1919 | } | ||
| 1920 | |||
| 1921 | 36 | static int format_name(const char *buf, char **s, int index, const char *varname) | |
| 1922 | { | ||
| 1923 | const char *proto, *dir; | ||
| 1924 | 36 | char *orig_buf_dup = NULL, *mod_buf_dup = NULL; | |
| 1925 | 36 | int ret = 0; | |
| 1926 | |||
| 1927 | 36 | orig_buf_dup = av_strdup(buf); | |
| 1928 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 36 times.
|
36 | if (!orig_buf_dup) |
| 1929 | ✗ | return AVERROR(ENOMEM); | |
| 1930 | |||
| 1931 |
1/2✓ Branch 1 taken 36 times.
✗ Branch 2 not taken.
|
36 | if (!av_stristr(buf, "%v")) { |
| 1932 | 36 | *s = orig_buf_dup; | |
| 1933 | 36 | return 0; | |
| 1934 | } | ||
| 1935 | |||
| 1936 | ✗ | if (!varname) { | |
| 1937 | ✗ | if (replace_int_data_in_filename(s, orig_buf_dup, 'v', index) < 1) { | |
| 1938 | ✗ | ret = AVERROR(EINVAL); | |
| 1939 | ✗ | goto fail; | |
| 1940 | } | ||
| 1941 | } else { | ||
| 1942 | ✗ | if (replace_str_data_in_filename(s, orig_buf_dup, 'v', varname) < 1) { | |
| 1943 | ✗ | ret = AVERROR(EINVAL); | |
| 1944 | ✗ | goto fail; | |
| 1945 | } | ||
| 1946 | } | ||
| 1947 | |||
| 1948 | ✗ | proto = avio_find_protocol_name(orig_buf_dup); | |
| 1949 | ✗ | dir = av_dirname(orig_buf_dup); | |
| 1950 | |||
| 1951 | /* if %v is present in the file's directory, create sub-directory */ | ||
| 1952 | ✗ | if (av_stristr(dir, "%v") && proto && !strcmp(proto, "file")) { | |
| 1953 | ✗ | mod_buf_dup = av_strdup(*s); | |
| 1954 | ✗ | dir = av_dirname(mod_buf_dup); | |
| 1955 | ✗ | if (ff_mkdir_p(dir) == -1 && errno != EEXIST) { | |
| 1956 | ✗ | ret = AVERROR(errno); | |
| 1957 | ✗ | goto fail; | |
| 1958 | } | ||
| 1959 | } | ||
| 1960 | |||
| 1961 | ✗ | fail: | |
| 1962 | ✗ | av_freep(&orig_buf_dup); | |
| 1963 | ✗ | av_freep(&mod_buf_dup); | |
| 1964 | ✗ | return ret; | |
| 1965 | } | ||
| 1966 | |||
| 1967 | ✗ | static int get_nth_codec_stream_index(AVFormatContext *s, | |
| 1968 | enum AVMediaType codec_type, | ||
| 1969 | int64_t stream_id) | ||
| 1970 | { | ||
| 1971 | unsigned int stream_index, cnt; | ||
| 1972 | ✗ | if (stream_id < 0 || stream_id > s->nb_streams - 1) | |
| 1973 | ✗ | return -1; | |
| 1974 | ✗ | cnt = 0; | |
| 1975 | ✗ | for (stream_index = 0; stream_index < s->nb_streams; stream_index++) { | |
| 1976 | ✗ | if (s->streams[stream_index]->codecpar->codec_type != codec_type) | |
| 1977 | ✗ | continue; | |
| 1978 | ✗ | if (cnt == stream_id) | |
| 1979 | ✗ | return stream_index; | |
| 1980 | ✗ | cnt++; | |
| 1981 | } | ||
| 1982 | ✗ | return -1; | |
| 1983 | } | ||
| 1984 | |||
| 1985 | ✗ | static int parse_variant_stream_mapstring(AVFormatContext *s) | |
| 1986 | { | ||
| 1987 | ✗ | HLSContext *hls = s->priv_data; | |
| 1988 | VariantStream *vs; | ||
| 1989 | int stream_index, i, j; | ||
| 1990 | enum AVMediaType codec_type; | ||
| 1991 | ✗ | int nb_varstreams = 0, nb_streams; | |
| 1992 | char *p, *q, *saveptr1, *saveptr2, *varstr, *keyval; | ||
| 1993 | const char *val; | ||
| 1994 | |||
| 1995 | /** | ||
| 1996 | * Expected format for var_stream_map string is as below: | ||
| 1997 | * "a:0,v:0 a:1,v:1" | ||
| 1998 | * "a:0,agroup:a0,default:1,language:ENG a:1,agroup:a1,default:0 v:0,agroup:a0 v:1,agroup:a1" | ||
| 1999 | * This string specifies how to group the audio, video and subtitle streams | ||
| 2000 | * into different variant streams. The variant stream groups are separated | ||
| 2001 | * by space. | ||
| 2002 | * | ||
| 2003 | * a:, v:, s: are keys to specify audio, video and subtitle streams | ||
| 2004 | * respectively. Allowed values are 0 to 9 digits (limited just based on | ||
| 2005 | * practical usage) | ||
| 2006 | * | ||
| 2007 | * agroup: is key to specify audio group. A string can be given as value. | ||
| 2008 | * sgroup: is key to specify subtitle group. A string can be given as value. | ||
| 2009 | */ | ||
| 2010 | ✗ | p = av_strdup(hls->var_stream_map); | |
| 2011 | ✗ | if (!p) | |
| 2012 | ✗ | return AVERROR(ENOMEM); | |
| 2013 | |||
| 2014 | ✗ | q = p; | |
| 2015 | ✗ | while (av_strtok(q, " \t", &saveptr1)) { | |
| 2016 | ✗ | q = NULL; | |
| 2017 | ✗ | nb_varstreams++; | |
| 2018 | } | ||
| 2019 | ✗ | av_freep(&p); | |
| 2020 | |||
| 2021 | ✗ | hls->var_streams = av_mallocz(sizeof(*hls->var_streams) * nb_varstreams); | |
| 2022 | ✗ | if (!hls->var_streams) | |
| 2023 | ✗ | return AVERROR(ENOMEM); | |
| 2024 | ✗ | hls->nb_varstreams = nb_varstreams; | |
| 2025 | |||
| 2026 | ✗ | p = hls->var_stream_map; | |
| 2027 | ✗ | nb_varstreams = 0; | |
| 2028 | ✗ | while (varstr = av_strtok(p, " \t", &saveptr1)) { | |
| 2029 | ✗ | p = NULL; | |
| 2030 | |||
| 2031 | ✗ | if (nb_varstreams < hls->nb_varstreams) { | |
| 2032 | ✗ | vs = &(hls->var_streams[nb_varstreams]); | |
| 2033 | ✗ | vs->var_stream_idx = nb_varstreams; | |
| 2034 | ✗ | vs->is_default = 0; | |
| 2035 | ✗ | nb_varstreams++; | |
| 2036 | } else | ||
| 2037 | ✗ | return AVERROR(EINVAL); | |
| 2038 | |||
| 2039 | ✗ | q = varstr; | |
| 2040 | while (1) { | ||
| 2041 | ✗ | if (!av_strncasecmp(q, "a:", 2) || !av_strncasecmp(q, "v:", 2) || | |
| 2042 | ✗ | !av_strncasecmp(q, "s:", 2)) | |
| 2043 | ✗ | vs->nb_streams++; | |
| 2044 | ✗ | q = strchr(q, ','); | |
| 2045 | ✗ | if (!q) | |
| 2046 | ✗ | break; | |
| 2047 | ✗ | q++; | |
| 2048 | } | ||
| 2049 | ✗ | vs->streams = av_mallocz(sizeof(AVStream *) * vs->nb_streams); | |
| 2050 | ✗ | if (!vs->streams) | |
| 2051 | ✗ | return AVERROR(ENOMEM); | |
| 2052 | |||
| 2053 | ✗ | nb_streams = 0; | |
| 2054 | ✗ | while (keyval = av_strtok(varstr, ",", &saveptr2)) { | |
| 2055 | int64_t num; | ||
| 2056 | char *end; | ||
| 2057 | ✗ | varstr = NULL; | |
| 2058 | ✗ | if (av_strstart(keyval, "language:", &val)) { | |
| 2059 | ✗ | vs->language = val; | |
| 2060 | ✗ | continue; | |
| 2061 | ✗ | } else if (av_strstart(keyval, "default:", &val)) { | |
| 2062 | ✗ | vs->is_default = (!av_strncasecmp(val, "YES", strlen("YES")) || | |
| 2063 | ✗ | (!av_strncasecmp(val, "1", strlen("1")))); | |
| 2064 | ✗ | hls->has_default_key = 1; | |
| 2065 | ✗ | continue; | |
| 2066 | ✗ | } else if (av_strstart(keyval, "name:", &val)) { | |
| 2067 | ✗ | vs->varname = val; | |
| 2068 | ✗ | continue; | |
| 2069 | ✗ | } else if (av_strstart(keyval, "sname:", &val)) { | |
| 2070 | ✗ | vs->subtitle_varname = val; | |
| 2071 | ✗ | continue; | |
| 2072 | ✗ | } else if (av_strstart(keyval, "agroup:", &val)) { | |
| 2073 | ✗ | vs->agroup = val; | |
| 2074 | ✗ | continue; | |
| 2075 | ✗ | } else if (av_strstart(keyval, "sgroup:", &val)) { | |
| 2076 | ✗ | vs->sgroup = val; | |
| 2077 | ✗ | continue; | |
| 2078 | ✗ | } else if (av_strstart(keyval, "ccgroup:", &val)) { | |
| 2079 | ✗ | vs->ccgroup = val; | |
| 2080 | ✗ | continue; | |
| 2081 | ✗ | } else if (av_strstart(keyval, "v:", &val)) { | |
| 2082 | ✗ | codec_type = AVMEDIA_TYPE_VIDEO; | |
| 2083 | ✗ | hls->has_video_m3u8 = 1; | |
| 2084 | ✗ | } else if (av_strstart(keyval, "a:", &val)) { | |
| 2085 | ✗ | codec_type = AVMEDIA_TYPE_AUDIO; | |
| 2086 | ✗ | } else if (av_strstart(keyval, "s:", &val)) { | |
| 2087 | ✗ | codec_type = AVMEDIA_TYPE_SUBTITLE; | |
| 2088 | } else { | ||
| 2089 | ✗ | av_log(s, AV_LOG_ERROR, "Invalid keyval %s\n", keyval); | |
| 2090 | ✗ | return AVERROR(EINVAL); | |
| 2091 | } | ||
| 2092 | |||
| 2093 | ✗ | num = strtoll(val, &end, 10); | |
| 2094 | ✗ | if (!av_isdigit(*val) || *end != '\0') { | |
| 2095 | ✗ | av_log(s, AV_LOG_ERROR, "Invalid stream number: '%s'\n", val); | |
| 2096 | ✗ | return AVERROR(EINVAL); | |
| 2097 | } | ||
| 2098 | ✗ | stream_index = get_nth_codec_stream_index(s, codec_type, num); | |
| 2099 | |||
| 2100 | ✗ | if (stream_index >= 0 && nb_streams < vs->nb_streams) { | |
| 2101 | ✗ | for (i = 0; nb_streams > 0 && i < nb_streams; i++) { | |
| 2102 | ✗ | if (vs->streams[i] == s->streams[stream_index]) { | |
| 2103 | ✗ | av_log(s, AV_LOG_ERROR, "Same elementary stream found more than once inside " | |
| 2104 | "variant definition #%d\n", nb_varstreams - 1); | ||
| 2105 | ✗ | return AVERROR(EINVAL); | |
| 2106 | } | ||
| 2107 | } | ||
| 2108 | ✗ | for (j = 0; nb_varstreams > 1 && j < nb_varstreams - 1; j++) { | |
| 2109 | ✗ | for (i = 0; i < hls->var_streams[j].nb_streams; i++) { | |
| 2110 | ✗ | if (hls->var_streams[j].streams[i] == s->streams[stream_index]) { | |
| 2111 | ✗ | av_log(s, AV_LOG_ERROR, "Same elementary stream found more than once " | |
| 2112 | "in two different variant definitions #%d and #%d\n", | ||
| 2113 | j, nb_varstreams - 1); | ||
| 2114 | ✗ | return AVERROR(EINVAL); | |
| 2115 | } | ||
| 2116 | } | ||
| 2117 | } | ||
| 2118 | ✗ | vs->streams[nb_streams++] = s->streams[stream_index]; | |
| 2119 | } else { | ||
| 2120 | ✗ | av_log(s, AV_LOG_ERROR, "Unable to map stream at %s\n", keyval); | |
| 2121 | ✗ | return AVERROR(EINVAL); | |
| 2122 | } | ||
| 2123 | } | ||
| 2124 | } | ||
| 2125 | ✗ | av_log(s, AV_LOG_DEBUG, "Number of variant streams %d\n", | |
| 2126 | hls->nb_varstreams); | ||
| 2127 | |||
| 2128 | ✗ | return 0; | |
| 2129 | } | ||
| 2130 | |||
| 2131 | ✗ | static int parse_cc_stream_mapstring(AVFormatContext *s) | |
| 2132 | { | ||
| 2133 | ✗ | HLSContext *hls = s->priv_data; | |
| 2134 | ✗ | int nb_ccstreams = 0; | |
| 2135 | char *p, *q, *ccstr, *keyval; | ||
| 2136 | ✗ | char *saveptr1 = NULL, *saveptr2 = NULL; | |
| 2137 | const char *val; | ||
| 2138 | ClosedCaptionsStream *ccs; | ||
| 2139 | |||
| 2140 | ✗ | p = av_strdup(hls->cc_stream_map); | |
| 2141 | ✗ | if(!p) | |
| 2142 | ✗ | return AVERROR(ENOMEM); | |
| 2143 | |||
| 2144 | ✗ | q = p; | |
| 2145 | ✗ | while (av_strtok(q, " \t", &saveptr1)) { | |
| 2146 | ✗ | q = NULL; | |
| 2147 | ✗ | nb_ccstreams++; | |
| 2148 | } | ||
| 2149 | ✗ | av_freep(&p); | |
| 2150 | |||
| 2151 | ✗ | hls->cc_streams = av_mallocz(sizeof(*hls->cc_streams) * nb_ccstreams); | |
| 2152 | ✗ | if (!hls->cc_streams) | |
| 2153 | ✗ | return AVERROR(ENOMEM); | |
| 2154 | ✗ | hls->nb_ccstreams = nb_ccstreams; | |
| 2155 | |||
| 2156 | ✗ | p = hls->cc_stream_map; | |
| 2157 | ✗ | nb_ccstreams = 0; | |
| 2158 | ✗ | while (ccstr = av_strtok(p, " \t", &saveptr1)) { | |
| 2159 | ✗ | p = NULL; | |
| 2160 | |||
| 2161 | ✗ | if (nb_ccstreams < hls->nb_ccstreams) | |
| 2162 | ✗ | ccs = &(hls->cc_streams[nb_ccstreams++]); | |
| 2163 | else | ||
| 2164 | ✗ | return AVERROR(EINVAL); | |
| 2165 | |||
| 2166 | ✗ | while (keyval = av_strtok(ccstr, ",", &saveptr2)) { | |
| 2167 | ✗ | ccstr = NULL; | |
| 2168 | |||
| 2169 | ✗ | if (av_strstart(keyval, "ccgroup:", &val)) { | |
| 2170 | ✗ | ccs->ccgroup = val; | |
| 2171 | ✗ | } else if (av_strstart(keyval, "instreamid:", &val)) { | |
| 2172 | ✗ | ccs->instreamid = val; | |
| 2173 | ✗ | } else if (av_strstart(keyval, "language:", &val)) { | |
| 2174 | ✗ | ccs->language = val; | |
| 2175 | } else { | ||
| 2176 | ✗ | av_log(s, AV_LOG_ERROR, "Invalid keyval %s\n", keyval); | |
| 2177 | ✗ | return AVERROR(EINVAL); | |
| 2178 | } | ||
| 2179 | } | ||
| 2180 | |||
| 2181 | ✗ | if (!ccs->ccgroup || !ccs->instreamid) { | |
| 2182 | ✗ | av_log(s, AV_LOG_ERROR, "Insufficient parameters in cc stream map string\n"); | |
| 2183 | ✗ | return AVERROR(EINVAL); | |
| 2184 | } | ||
| 2185 | |||
| 2186 | ✗ | if (av_strstart(ccs->instreamid, "CC", &val)) { | |
| 2187 | ✗ | if (atoi(val) < 1 || atoi(val) > 4) { | |
| 2188 | ✗ | av_log(s, AV_LOG_ERROR, "Invalid instream ID CC index %d in %s, range 1-4\n", | |
| 2189 | atoi(val), ccs->instreamid); | ||
| 2190 | ✗ | return AVERROR(EINVAL); | |
| 2191 | } | ||
| 2192 | ✗ | } else if (av_strstart(ccs->instreamid, "SERVICE", &val)) { | |
| 2193 | ✗ | if (atoi(val) < 1 || atoi(val) > 63) { | |
| 2194 | ✗ | av_log(s, AV_LOG_ERROR, "Invalid instream ID SERVICE index %d in %s, range 1-63 \n", | |
| 2195 | atoi(val), ccs->instreamid); | ||
| 2196 | ✗ | return AVERROR(EINVAL); | |
| 2197 | } | ||
| 2198 | } else { | ||
| 2199 | ✗ | av_log(s, AV_LOG_ERROR, "Invalid instream ID %s, supported are CCn or SERVICEn\n", | |
| 2200 | ccs->instreamid); | ||
| 2201 | ✗ | return AVERROR(EINVAL); | |
| 2202 | } | ||
| 2203 | } | ||
| 2204 | |||
| 2205 | ✗ | return 0; | |
| 2206 | } | ||
| 2207 | |||
| 2208 | 18 | static int update_variant_stream_info(AVFormatContext *s) | |
| 2209 | { | ||
| 2210 | 18 | HLSContext *hls = s->priv_data; | |
| 2211 | unsigned int i; | ||
| 2212 | 18 | int ret = 0; | |
| 2213 | |||
| 2214 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->cc_stream_map) { |
| 2215 | ✗ | ret = parse_cc_stream_mapstring(s); | |
| 2216 | ✗ | if (ret < 0) | |
| 2217 | ✗ | return ret; | |
| 2218 | } | ||
| 2219 | |||
| 2220 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->var_stream_map) { |
| 2221 | ✗ | return parse_variant_stream_mapstring(s); | |
| 2222 | } else { | ||
| 2223 | //By default, a single variant stream with all the codec streams is created | ||
| 2224 | 18 | hls->var_streams = av_mallocz(sizeof(*hls->var_streams)); | |
| 2225 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (!hls->var_streams) |
| 2226 | ✗ | return AVERROR(ENOMEM); | |
| 2227 | 18 | hls->nb_varstreams = 1; | |
| 2228 | |||
| 2229 | 18 | hls->var_streams[0].var_stream_idx = 0; | |
| 2230 | 18 | hls->var_streams[0].nb_streams = s->nb_streams; | |
| 2231 | 36 | hls->var_streams[0].streams = av_mallocz(sizeof(AVStream *) * | |
| 2232 | 18 | hls->var_streams[0].nb_streams); | |
| 2233 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (!hls->var_streams[0].streams) |
| 2234 | ✗ | return AVERROR(ENOMEM); | |
| 2235 | |||
| 2236 | //by default, the first available ccgroup is mapped to the variant stream | ||
| 2237 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->nb_ccstreams) |
| 2238 | ✗ | hls->var_streams[0].ccgroup = hls->cc_streams[0].ccgroup; | |
| 2239 | |||
| 2240 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 18 times.
|
36 | for (i = 0; i < s->nb_streams; i++) |
| 2241 | 18 | hls->var_streams[0].streams[i] = s->streams[i]; | |
| 2242 | } | ||
| 2243 | 18 | return 0; | |
| 2244 | } | ||
| 2245 | |||
| 2246 | ✗ | static int update_master_pl_info(AVFormatContext *s) | |
| 2247 | { | ||
| 2248 | ✗ | HLSContext *hls = s->priv_data; | |
| 2249 | const char *dir; | ||
| 2250 | ✗ | char *fn1= NULL, *fn2 = NULL; | |
| 2251 | ✗ | int ret = 0; | |
| 2252 | |||
| 2253 | ✗ | fn1 = av_strdup(s->url); | |
| 2254 | ✗ | if (!fn1) | |
| 2255 | ✗ | return AVERROR(ENOMEM); | |
| 2256 | ✗ | dir = av_dirname(fn1); | |
| 2257 | |||
| 2258 | /** | ||
| 2259 | * if output file's directory has %v, variants are created in sub-directories | ||
| 2260 | * then master is created at the sub-directories level | ||
| 2261 | */ | ||
| 2262 | ✗ | if (dir && av_stristr(av_basename(dir), "%v")) { | |
| 2263 | ✗ | fn2 = av_strdup(dir); | |
| 2264 | ✗ | if (!fn2) { | |
| 2265 | ✗ | ret = AVERROR(ENOMEM); | |
| 2266 | ✗ | goto fail; | |
| 2267 | } | ||
| 2268 | ✗ | dir = av_dirname(fn2); | |
| 2269 | } | ||
| 2270 | |||
| 2271 | ✗ | if (dir && strcmp(dir, ".")) | |
| 2272 | ✗ | hls->master_m3u8_url = av_append_path_component(dir, hls->master_pl_name); | |
| 2273 | else | ||
| 2274 | ✗ | hls->master_m3u8_url = av_strdup(hls->master_pl_name); | |
| 2275 | |||
| 2276 | ✗ | if (!hls->master_m3u8_url) { | |
| 2277 | ✗ | ret = AVERROR(ENOMEM); | |
| 2278 | ✗ | goto fail; | |
| 2279 | } | ||
| 2280 | |||
| 2281 | ✗ | fail: | |
| 2282 | ✗ | av_freep(&fn1); | |
| 2283 | ✗ | av_freep(&fn2); | |
| 2284 | |||
| 2285 | ✗ | return ret; | |
| 2286 | } | ||
| 2287 | |||
| 2288 | 18 | static int hls_write_header(AVFormatContext *s) | |
| 2289 | { | ||
| 2290 | 18 | HLSContext *hls = s->priv_data; | |
| 2291 | int ret, i, j; | ||
| 2292 | 18 | VariantStream *vs = NULL; | |
| 2293 | |||
| 2294 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 18 times.
|
36 | for (i = 0; i < hls->nb_varstreams; i++) { |
| 2295 | 18 | int subtitle_streams = 0; | |
| 2296 | 18 | vs = &hls->var_streams[i]; | |
| 2297 | |||
| 2298 | 18 | ret = avformat_write_header(vs->avf, NULL); | |
| 2299 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 2300 | ✗ | return ret; | |
| 2301 | //av_assert0(s->nb_streams == hls->avf->nb_streams); | ||
| 2302 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 18 times.
|
36 | for (j = 0; j < vs->nb_streams; j++) { |
| 2303 | AVStream *inner_st; | ||
| 2304 | 18 | AVStream *outer_st = vs->streams[j]; | |
| 2305 | |||
| 2306 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 17 times.
|
18 | if (hls->max_seg_size > 0) { |
| 2307 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if ((outer_st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) && |
| 2308 | ✗ | (outer_st->codecpar->bit_rate > hls->max_seg_size)) { | |
| 2309 | ✗ | av_log(s, AV_LOG_WARNING, "Your video bitrate is bigger than hls_segment_size, " | |
| 2310 | "(%"PRId64 " > %"PRId64 "), the result maybe not be what you want.", | ||
| 2311 | ✗ | outer_st->codecpar->bit_rate, hls->max_seg_size); | |
| 2312 | } | ||
| 2313 | } | ||
| 2314 | |||
| 2315 |
1/2✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
|
18 | if (outer_st->codecpar->codec_type != AVMEDIA_TYPE_SUBTITLE) |
| 2316 | 18 | inner_st = vs->avf->streams[j - subtitle_streams]; | |
| 2317 | ✗ | else if (vs->vtt_avf) { | |
| 2318 | ✗ | inner_st = vs->vtt_avf->streams[0]; | |
| 2319 | ✗ | subtitle_streams++; | |
| 2320 | } else { | ||
| 2321 | /* We have a subtitle stream, when the user does not want one */ | ||
| 2322 | ✗ | inner_st = NULL; | |
| 2323 | ✗ | continue; | |
| 2324 | } | ||
| 2325 | 18 | avpriv_set_pts_info(outer_st, inner_st->pts_wrap_bits, inner_st->time_base.num, inner_st->time_base.den); | |
| 2326 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (outer_st->codecpar->codec_id == AV_CODEC_ID_HEVC && |
| 2327 | ✗ | outer_st->codecpar->codec_tag != MKTAG('h','v','c','1')) { | |
| 2328 | ✗ | av_log(s, AV_LOG_WARNING, "Stream HEVC is not hvc1, you should use tag:v hvc1 to set it.\n"); | |
| 2329 | } | ||
| 2330 | 18 | write_codec_attr(outer_st, vs); | |
| 2331 | |||
| 2332 | } | ||
| 2333 | /* Update the Codec Attr string for the mapped audio groups */ | ||
| 2334 |
3/4✓ Branch 0 taken 6 times.
✓ Branch 1 taken 12 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 6 times.
|
18 | if (vs->has_video && vs->agroup) { |
| 2335 | ✗ | for (j = 0; j < hls->nb_varstreams; j++) { | |
| 2336 | ✗ | VariantStream *vs_agroup = &(hls->var_streams[j]); | |
| 2337 | ✗ | if (!vs_agroup->has_video && !vs_agroup->has_subtitle && | |
| 2338 | ✗ | vs_agroup->agroup && | |
| 2339 | ✗ | !av_strcasecmp(vs_agroup->agroup, vs->agroup)) { | |
| 2340 | ✗ | write_codec_attr(vs_agroup->streams[0], vs); | |
| 2341 | } | ||
| 2342 | } | ||
| 2343 | } | ||
| 2344 | } | ||
| 2345 | |||
| 2346 | 18 | return 0; | |
| 2347 | } | ||
| 2348 | |||
| 2349 | ✗ | static int hls_init_file_resend(AVFormatContext *s, VariantStream *vs) | |
| 2350 | { | ||
| 2351 | ✗ | HLSContext *hls = s->priv_data; | |
| 2352 | ✗ | AVDictionary *options = NULL; | |
| 2353 | ✗ | int ret = 0; | |
| 2354 | |||
| 2355 | ✗ | set_http_options(s, &options, hls); | |
| 2356 | ✗ | ret = hlsenc_io_open(s, &vs->out, vs->base_output_dirname, &options); | |
| 2357 | ✗ | av_dict_free(&options); | |
| 2358 | ✗ | if (ret < 0) | |
| 2359 | ✗ | return ret; | |
| 2360 | ✗ | avio_write(vs->out, vs->init_buffer, vs->init_range_length); | |
| 2361 | ✗ | hlsenc_io_close(s, &vs->out, hls->fmp4_init_filename); | |
| 2362 | |||
| 2363 | ✗ | return ret; | |
| 2364 | } | ||
| 2365 | |||
| 2366 | ✗ | static int64_t append_single_file(AVFormatContext *s, VariantStream *vs) | |
| 2367 | { | ||
| 2368 | ✗ | int64_t ret = 0; | |
| 2369 | ✗ | int64_t read_byte = 0; | |
| 2370 | ✗ | int64_t total_size = 0; | |
| 2371 | ✗ | char *filename = NULL; | |
| 2372 | char buf[BUFSIZE]; | ||
| 2373 | ✗ | AVFormatContext *oc = vs->avf; | |
| 2374 | |||
| 2375 | ✗ | hlsenc_io_close(s, &vs->out, vs->basename_tmp); | |
| 2376 | ✗ | filename = av_asprintf("%s.tmp", oc->url); | |
| 2377 | ✗ | ret = s->io_open(s, &vs->out, filename, AVIO_FLAG_READ, NULL); | |
| 2378 | ✗ | if (ret < 0) { | |
| 2379 | ✗ | av_free(filename); | |
| 2380 | ✗ | return ret; | |
| 2381 | } | ||
| 2382 | |||
| 2383 | do { | ||
| 2384 | ✗ | read_byte = avio_read(vs->out, buf, BUFSIZE); | |
| 2385 | ✗ | if (read_byte > 0) { | |
| 2386 | ✗ | avio_write(vs->out_single_file, buf, read_byte); | |
| 2387 | ✗ | total_size += read_byte; | |
| 2388 | ✗ | ret = total_size; | |
| 2389 | } | ||
| 2390 | ✗ | } while (read_byte > 0); | |
| 2391 | |||
| 2392 | ✗ | hlsenc_io_close(s, &vs->out, filename); | |
| 2393 | ✗ | av_free(filename); | |
| 2394 | |||
| 2395 | ✗ | return ret; | |
| 2396 | } | ||
| 2397 | 7084 | static int hls_write_packet(AVFormatContext *s, AVPacket *pkt) | |
| 2398 | { | ||
| 2399 | 7084 | HLSContext *hls = s->priv_data; | |
| 2400 | 7084 | AVFormatContext *oc = NULL; | |
| 2401 | 7084 | AVStream *st = s->streams[pkt->stream_index]; | |
| 2402 | 7084 | int64_t end_pts = 0; | |
| 2403 | 7084 | int is_ref_pkt = 1; | |
| 2404 | 7084 | int ret = 0, can_split = 1, i, j; | |
| 2405 | 7084 | int stream_index = 0; | |
| 2406 | 7084 | int range_length = 0; | |
| 2407 | 7084 | const char *proto = NULL; | |
| 2408 | 7084 | int use_temp_file = 0; | |
| 2409 | 7084 | VariantStream *vs = NULL; | |
| 2410 | 7084 | char *old_filename = NULL; | |
| 2411 | |||
| 2412 |
1/2✓ Branch 0 taken 7084 times.
✗ Branch 1 not taken.
|
7084 | for (i = 0; i < hls->nb_varstreams; i++) { |
| 2413 | 7084 | int subtitle_streams = 0; | |
| 2414 | 7084 | vs = &hls->var_streams[i]; | |
| 2415 |
1/2✓ Branch 0 taken 7084 times.
✗ Branch 1 not taken.
|
7084 | for (j = 0; j < vs->nb_streams; j++) { |
| 2416 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7084 times.
|
7084 | if (vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE) { |
| 2417 | ✗ | subtitle_streams++; | |
| 2418 | } | ||
| 2419 |
1/2✓ Branch 0 taken 7084 times.
✗ Branch 1 not taken.
|
7084 | if (vs->streams[j] == st) { |
| 2420 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7084 times.
|
7084 | if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE) { |
| 2421 | ✗ | oc = vs->vtt_avf; | |
| 2422 | ✗ | stream_index = 0; | |
| 2423 | } else { | ||
| 2424 | 7084 | oc = vs->avf; | |
| 2425 | 7084 | stream_index = j - subtitle_streams; | |
| 2426 | } | ||
| 2427 | 7084 | break; | |
| 2428 | } | ||
| 2429 | } | ||
| 2430 | |||
| 2431 |
1/2✓ Branch 0 taken 7084 times.
✗ Branch 1 not taken.
|
7084 | if (oc) |
| 2432 | 7084 | break; | |
| 2433 | } | ||
| 2434 | |||
| 2435 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7084 times.
|
7084 | if (!oc) { |
| 2436 | ✗ | av_log(s, AV_LOG_ERROR, "Unable to find mapping variant stream\n"); | |
| 2437 | ✗ | return AVERROR(ENOMEM); | |
| 2438 | } | ||
| 2439 | |||
| 2440 | 7084 | end_pts = hls->recording_time * vs->number; | |
| 2441 | |||
| 2442 |
4/4✓ Branch 0 taken 341 times.
✓ Branch 1 taken 6743 times.
✓ Branch 2 taken 191 times.
✓ Branch 3 taken 150 times.
|
7084 | if (vs->sequence - vs->nb_entries > hls->start_sequence && hls->init_time > 0) { |
| 2443 | /* reset end_pts, hls->recording_time at end of the init hls list */ | ||
| 2444 | 191 | int64_t init_list_dur = hls->init_time * vs->nb_entries; | |
| 2445 | 191 | int64_t after_init_list_dur = (vs->sequence - hls->start_sequence - vs->nb_entries) * hls->time; | |
| 2446 | 191 | hls->recording_time = hls->time; | |
| 2447 | 191 | end_pts = init_list_dur + after_init_list_dur ; | |
| 2448 | } | ||
| 2449 | |||
| 2450 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 7066 times.
|
7084 | if (vs->start_pts == AV_NOPTS_VALUE) { |
| 2451 | 18 | vs->start_pts = pkt->pts; | |
| 2452 |
2/2✓ Branch 0 taken 12 times.
✓ Branch 1 taken 6 times.
|
18 | if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) |
| 2453 | 12 | vs->start_pts_from_audio = 1; | |
| 2454 | } | ||
| 2455 |
3/6✓ Branch 0 taken 7066 times.
✓ Branch 1 taken 18 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 7066 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
|
7084 | if (vs->start_pts_from_audio && st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && vs->start_pts > pkt->pts) { |
| 2456 | ✗ | vs->start_pts = pkt->pts; | |
| 2457 | ✗ | vs->start_pts_from_audio = 0; | |
| 2458 | } | ||
| 2459 | |||
| 2460 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 7066 times.
|
7084 | if (vs->has_video) { |
| 2461 |
1/2✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
|
36 | can_split = st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && |
| 2462 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
18 | ((pkt->flags & AV_PKT_FLAG_KEY) || (hls->flags & HLS_SPLIT_BY_TIME)); |
| 2463 |
2/4✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 18 times.
✗ Branch 3 not taken.
|
18 | is_ref_pkt = (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) && (pkt->stream_index == vs->reference_stream_index); |
| 2464 | } | ||
| 2465 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7084 times.
|
7084 | if (pkt->pts == AV_NOPTS_VALUE) |
| 2466 | ✗ | is_ref_pkt = can_split = 0; | |
| 2467 | |||
| 2468 |
1/2✓ Branch 0 taken 7084 times.
✗ Branch 1 not taken.
|
7084 | if (is_ref_pkt) { |
| 2469 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 7066 times.
|
7084 | if (vs->end_pts == AV_NOPTS_VALUE) |
| 2470 | 18 | vs->end_pts = pkt->pts; | |
| 2471 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 7066 times.
|
7084 | if (vs->new_start) { |
| 2472 | 18 | vs->new_start = 0; | |
| 2473 | 18 | vs->duration = (double)(pkt->pts - vs->end_pts) | |
| 2474 | 18 | * st->time_base.num / st->time_base.den; | |
| 2475 | 18 | vs->dpp = (double)(pkt->duration) * st->time_base.num / st->time_base.den; | |
| 2476 | } else { | ||
| 2477 |
1/2✓ Branch 0 taken 7066 times.
✗ Branch 1 not taken.
|
7066 | if (pkt->duration) { |
| 2478 | 7066 | vs->duration += (double)(pkt->duration) * st->time_base.num / st->time_base.den; | |
| 2479 | } else { | ||
| 2480 | ✗ | av_log(s, AV_LOG_WARNING, "Stream %d packet with pts %" PRId64 " has duration 0. The segment duration may not be precise.\n", | |
| 2481 | pkt->stream_index, pkt->pts); | ||
| 2482 | ✗ | vs->duration = (double)(pkt->pts - vs->end_pts) * st->time_base.num / st->time_base.den; | |
| 2483 | } | ||
| 2484 | } | ||
| 2485 | } | ||
| 2486 | |||
| 2487 |
3/4✓ Branch 0 taken 7084 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 7066 times.
✓ Branch 3 taken 18 times.
|
7084 | can_split = can_split && (pkt->pts - vs->end_pts > 0); |
| 2488 |
5/6✓ Branch 0 taken 7066 times.
✓ Branch 1 taken 18 times.
✓ Branch 2 taken 7066 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 74 times.
✓ Branch 5 taken 6992 times.
|
14150 | if (vs->packets_written && can_split && av_compare_ts(pkt->pts - vs->start_pts, st->time_base, |
| 2489 | 7066 | end_pts, AV_TIME_BASE_Q) >= 0) { | |
| 2490 | int64_t new_start_pos; | ||
| 2491 |
4/4✓ Branch 0 taken 65 times.
✓ Branch 1 taken 9 times.
✓ Branch 2 taken 9 times.
✓ Branch 3 taken 56 times.
|
74 | int byterange_mode = (hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size > 0); |
| 2492 | double cur_duration; | ||
| 2493 | |||
| 2494 | 74 | av_write_frame(oc, NULL); /* Flush any buffered data */ | |
| 2495 | 74 | new_start_pos = avio_tell(oc->pb); | |
| 2496 | 74 | vs->size = new_start_pos - vs->start_pos; | |
| 2497 | 74 | avio_flush(oc->pb); | |
| 2498 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 70 times.
|
74 | if (hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 2499 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 3 times.
|
4 | if (!vs->init_range_length) { |
| 2500 | 1 | range_length = avio_close_dyn_buf(oc->pb, &vs->init_buffer); | |
| 2501 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (range_length <= 0) |
| 2502 | ✗ | return AVERROR(EINVAL); | |
| 2503 | 1 | avio_write(vs->out, vs->init_buffer, range_length); | |
| 2504 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (!hls->resend_init_file) |
| 2505 | 1 | av_freep(&vs->init_buffer); | |
| 2506 | 1 | vs->init_range_length = range_length; | |
| 2507 | 1 | avio_open_dyn_buf(&oc->pb); | |
| 2508 | 1 | vs->packets_written = 0; | |
| 2509 | 1 | vs->start_pos = range_length; | |
| 2510 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (!byterange_mode) { |
| 2511 | 1 | hlsenc_io_close(s, &vs->out, vs->base_output_dirname); | |
| 2512 | } | ||
| 2513 | } | ||
| 2514 | } | ||
| 2515 |
2/2✓ Branch 0 taken 56 times.
✓ Branch 1 taken 18 times.
|
74 | if (!byterange_mode) { |
| 2516 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
|
56 | if (vs->vtt_avf) { |
| 2517 | ✗ | hlsenc_io_close(s, &vs->vtt_avf->pb, vs->vtt_avf->url); | |
| 2518 | } | ||
| 2519 | } | ||
| 2520 | |||
| 2521 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 65 times.
|
74 | if (hls->flags & HLS_SINGLE_FILE) { |
| 2522 | 9 | ret = flush_dynbuf(vs, &range_length); | |
| 2523 | 9 | av_freep(&vs->temp_buffer); | |
| 2524 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
|
9 | if (ret < 0) { |
| 2525 | ✗ | return ret; | |
| 2526 | } | ||
| 2527 | 9 | vs->size = range_length; | |
| 2528 |
2/4✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 9 times.
|
9 | if (hls->key_info_file || hls->encrypt) |
| 2529 | ✗ | vs->size = append_single_file(s, vs); | |
| 2530 | } else { | ||
| 2531 |
1/2✓ Branch 0 taken 65 times.
✗ Branch 1 not taken.
|
65 | if (oc->url[0]) { |
| 2532 | 65 | proto = avio_find_protocol_name(oc->url); | |
| 2533 |
1/2✓ Branch 0 taken 65 times.
✗ Branch 1 not taken.
|
130 | use_temp_file = proto && !strcmp(proto, "file") |
| 2534 |
2/4✓ Branch 0 taken 65 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 65 times.
|
130 | && (hls->flags & HLS_TEMP_FILE); |
| 2535 | } | ||
| 2536 | |||
| 2537 |
6/6✓ Branch 0 taken 9 times.
✓ Branch 1 taken 56 times.
✓ Branch 2 taken 6 times.
✓ Branch 3 taken 3 times.
✓ Branch 4 taken 56 times.
✓ Branch 5 taken 6 times.
|
65 | if ((hls->max_seg_size > 0 && (vs->size + vs->start_pos >= hls->max_seg_size)) || !byterange_mode) { |
| 2538 | 59 | AVDictionary *options = NULL; | |
| 2539 | 59 | char *filename = NULL; | |
| 2540 |
2/4✓ Branch 0 taken 59 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 59 times.
|
59 | if (hls->key_info_file || hls->encrypt) { |
| 2541 | ✗ | av_dict_set(&options, "encryption_key", vs->key_string, 0); | |
| 2542 | ✗ | av_dict_set(&options, "encryption_iv", vs->iv_string, 0); | |
| 2543 | ✗ | filename = av_asprintf("crypto:%s", oc->url); | |
| 2544 | } else { | ||
| 2545 | 59 | filename = av_asprintf("%s", oc->url); | |
| 2546 | } | ||
| 2547 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 59 times.
|
59 | if (!filename) { |
| 2548 | ✗ | av_dict_free(&options); | |
| 2549 | ✗ | return AVERROR(ENOMEM); | |
| 2550 | } | ||
| 2551 | |||
| 2552 | // look to rename the asset name | ||
| 2553 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 59 times.
|
59 | if (use_temp_file) |
| 2554 | ✗ | av_dict_set(&options, "mpegts_flags", "resend_headers", 0); | |
| 2555 | |||
| 2556 | 59 | set_http_options(s, &options, hls); | |
| 2557 | |||
| 2558 | 59 | ret = hlsenc_io_open(s, &vs->out, filename, &options); | |
| 2559 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 59 times.
|
59 | if (ret < 0) { |
| 2560 | ✗ | av_log(s, hls->ignore_io_errors ? AV_LOG_WARNING : AV_LOG_ERROR, | |
| 2561 | "Failed to open file '%s'\n", filename); | ||
| 2562 | ✗ | av_freep(&filename); | |
| 2563 | ✗ | av_dict_free(&options); | |
| 2564 | ✗ | return hls->ignore_io_errors ? 0 : ret; | |
| 2565 | } | ||
| 2566 |
2/2✓ Branch 0 taken 4 times.
✓ Branch 1 taken 55 times.
|
59 | if (hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 2567 | 4 | write_styp(vs->out); | |
| 2568 | } | ||
| 2569 | 59 | ret = flush_dynbuf(vs, &range_length); | |
| 2570 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 59 times.
|
59 | if (ret < 0) { |
| 2571 | ✗ | av_freep(&filename); | |
| 2572 | ✗ | av_dict_free(&options); | |
| 2573 | ✗ | return ret; | |
| 2574 | } | ||
| 2575 | 59 | vs->size = range_length; | |
| 2576 | 59 | ret = hlsenc_io_close(s, &vs->out, filename); | |
| 2577 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 59 times.
|
59 | if (ret < 0) { |
| 2578 | ✗ | av_log(s, AV_LOG_WARNING, "upload segment failed," | |
| 2579 | " will retry with a new http session.\n"); | ||
| 2580 | ✗ | ff_format_io_close(s, &vs->out); | |
| 2581 | ✗ | ret = hlsenc_io_open(s, &vs->out, filename, &options); | |
| 2582 | ✗ | if (ret >= 0) { | |
| 2583 | ✗ | reflush_dynbuf(vs, &range_length); | |
| 2584 | ✗ | ret = hlsenc_io_close(s, &vs->out, filename); | |
| 2585 | } | ||
| 2586 | } | ||
| 2587 | 59 | av_dict_free(&options); | |
| 2588 | 59 | av_freep(&vs->temp_buffer); | |
| 2589 | 59 | av_freep(&filename); | |
| 2590 | } | ||
| 2591 | |||
| 2592 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 65 times.
|
65 | if (use_temp_file) |
| 2593 | ✗ | hls_rename_temp_file(s, oc); | |
| 2594 | } | ||
| 2595 | |||
| 2596 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 74 times.
|
74 | if (ret < 0) |
| 2597 | ✗ | return ret; | |
| 2598 | |||
| 2599 | 74 | old_filename = av_strdup(oc->url); | |
| 2600 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 74 times.
|
74 | if (!old_filename) { |
| 2601 | ✗ | return AVERROR(ENOMEM); | |
| 2602 | } | ||
| 2603 | |||
| 2604 | 74 | cur_duration = (double)(pkt->pts - vs->end_pts) * st->time_base.num / st->time_base.den; | |
| 2605 | 74 | ret = hls_append_segment(s, hls, vs, cur_duration, vs->start_pos, vs->size); | |
| 2606 | 74 | vs->end_pts = pkt->pts; | |
| 2607 | 74 | vs->duration = 0; | |
| 2608 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 74 times.
|
74 | if (ret < 0) { |
| 2609 | ✗ | av_freep(&old_filename); | |
| 2610 | ✗ | return ret; | |
| 2611 | } | ||
| 2612 | |||
| 2613 | // if we're building a VOD playlist, skip writing the manifest multiple times, and just wait until the end | ||
| 2614 |
2/2✓ Branch 0 taken 72 times.
✓ Branch 1 taken 2 times.
|
74 | if (hls->pl_type != PLAYLIST_TYPE_VOD) { |
| 2615 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 72 times.
|
72 | if ((ret = hls_window(s, 0, vs)) < 0) { |
| 2616 | ✗ | av_log(s, AV_LOG_WARNING, "upload playlist failed, will retry with a new http session.\n"); | |
| 2617 | ✗ | ff_format_io_close(s, &vs->out); | |
| 2618 | ✗ | if ((ret = hls_window(s, 0, vs)) < 0) { | |
| 2619 | ✗ | av_freep(&old_filename); | |
| 2620 | ✗ | return ret; | |
| 2621 | } | ||
| 2622 | } | ||
| 2623 | } | ||
| 2624 | |||
| 2625 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 74 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
74 | if (hls->resend_init_file && hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 2626 | ✗ | ret = hls_init_file_resend(s, vs); | |
| 2627 | ✗ | if (ret < 0) { | |
| 2628 | ✗ | av_freep(&old_filename); | |
| 2629 | ✗ | return ret; | |
| 2630 | } | ||
| 2631 | } | ||
| 2632 | |||
| 2633 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 65 times.
|
74 | if (hls->flags & HLS_SINGLE_FILE) { |
| 2634 | 9 | vs->start_pos += vs->size; | |
| 2635 |
2/4✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 9 times.
|
9 | if (hls->key_info_file || hls->encrypt) |
| 2636 | ✗ | ret = hls_start(s, vs); | |
| 2637 |
3/6✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 9 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 9 times.
✗ Branch 5 not taken.
|
9 | if (hls->segment_type == SEGMENT_TYPE_MPEGTS && oc->oformat->priv_class && oc->priv_data) { |
| 2638 | 9 | av_opt_set(oc->priv_data, "mpegts_flags", "resend_headers", 0); | |
| 2639 | } | ||
| 2640 |
2/2✓ Branch 0 taken 9 times.
✓ Branch 1 taken 56 times.
|
65 | } else if (hls->max_seg_size > 0) { |
| 2641 |
2/2✓ Branch 0 taken 3 times.
✓ Branch 1 taken 6 times.
|
9 | if (vs->size + vs->start_pos >= hls->max_seg_size) { |
| 2642 | 3 | vs->sequence++; | |
| 2643 | 3 | sls_flag_file_rename(hls, vs, old_filename); | |
| 2644 | 3 | ret = hls_start(s, vs); | |
| 2645 | 3 | vs->start_pos = 0; | |
| 2646 | /* When split segment by byte, the duration is short than hls_time, | ||
| 2647 | * so it is not enough one segment duration as hls_time, */ | ||
| 2648 | } else { | ||
| 2649 | 6 | vs->start_pos = new_start_pos; | |
| 2650 | } | ||
| 2651 | } else { | ||
| 2652 | 56 | vs->start_pos = 0; | |
| 2653 | 56 | sls_flag_file_rename(hls, vs, old_filename); | |
| 2654 | 56 | ret = hls_start(s, vs); | |
| 2655 | } | ||
| 2656 | 74 | vs->number++; | |
| 2657 | 74 | av_freep(&old_filename); | |
| 2658 | |||
| 2659 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 74 times.
|
74 | if (ret < 0) { |
| 2660 | ✗ | return ret; | |
| 2661 | } | ||
| 2662 | } | ||
| 2663 | |||
| 2664 | 7084 | vs->packets_written++; | |
| 2665 |
1/2✓ Branch 0 taken 7084 times.
✗ Branch 1 not taken.
|
7084 | if (oc->pb) { |
| 2666 | 7084 | ret = ff_write_chained(oc, stream_index, pkt, s, 0); | |
| 2667 | 7084 | vs->video_keyframe_size += pkt->size; | |
| 2668 |
3/4✓ Branch 0 taken 18 times.
✓ Branch 1 taken 7066 times.
✓ Branch 2 taken 18 times.
✗ Branch 3 not taken.
|
7084 | if ((st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) && (pkt->flags & AV_PKT_FLAG_KEY)) { |
| 2669 | 18 | vs->video_keyframe_size = avio_tell(oc->pb); | |
| 2670 | } else { | ||
| 2671 | 7066 | vs->video_keyframe_pos = avio_tell(vs->out); | |
| 2672 | } | ||
| 2673 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 7084 times.
|
7084 | if (hls->ignore_io_errors) |
| 2674 | ✗ | ret = 0; | |
| 2675 | } | ||
| 2676 | |||
| 2677 | 7084 | return ret; | |
| 2678 | } | ||
| 2679 | |||
| 2680 | 18 | static void hls_deinit(AVFormatContext *s) | |
| 2681 | { | ||
| 2682 | 18 | HLSContext *hls = s->priv_data; | |
| 2683 | 18 | int i = 0; | |
| 2684 | 18 | VariantStream *vs = NULL; | |
| 2685 | |||
| 2686 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 18 times.
|
36 | for (i = 0; i < hls->nb_varstreams; i++) { |
| 2687 | 18 | vs = &hls->var_streams[i]; | |
| 2688 | |||
| 2689 | 18 | av_freep(&vs->basename); | |
| 2690 | 18 | av_freep(&vs->base_output_dirname); | |
| 2691 | 18 | av_freep(&vs->fmp4_init_filename); | |
| 2692 | 18 | av_freep(&vs->vtt_basename); | |
| 2693 | 18 | av_freep(&vs->vtt_m3u8_name); | |
| 2694 | |||
| 2695 | 18 | avformat_free_context(vs->vtt_avf); | |
| 2696 | 18 | avformat_free_context(vs->avf); | |
| 2697 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->resend_init_file) |
| 2698 | ✗ | av_freep(&vs->init_buffer); | |
| 2699 | 18 | hls_free_segments(vs->segments); | |
| 2700 | 18 | hls_free_segments(vs->old_segments); | |
| 2701 | 18 | av_freep(&vs->m3u8_name); | |
| 2702 | 18 | av_freep(&vs->streams); | |
| 2703 | } | ||
| 2704 | |||
| 2705 | 18 | ff_format_io_close(s, &hls->m3u8_out); | |
| 2706 | 18 | ff_format_io_close(s, &hls->sub_m3u8_out); | |
| 2707 | 18 | ff_format_io_close(s, &hls->http_delete); | |
| 2708 | 18 | av_freep(&hls->key_basename); | |
| 2709 | 18 | av_freep(&hls->var_streams); | |
| 2710 | 18 | av_freep(&hls->cc_streams); | |
| 2711 | 18 | av_freep(&hls->master_m3u8_url); | |
| 2712 | 18 | } | |
| 2713 | |||
| 2714 | 18 | static int hls_write_trailer(struct AVFormatContext *s) | |
| 2715 | { | ||
| 2716 | 18 | HLSContext *hls = s->priv_data; | |
| 2717 | 18 | AVFormatContext *oc = NULL; | |
| 2718 | 18 | AVFormatContext *vtt_oc = NULL; | |
| 2719 | 18 | char *old_filename = NULL; | |
| 2720 | 18 | const char *proto = NULL; | |
| 2721 | 18 | int use_temp_file = 0; | |
| 2722 | int i; | ||
| 2723 | 18 | int ret = 0; | |
| 2724 | 18 | VariantStream *vs = NULL; | |
| 2725 | 18 | AVDictionary *options = NULL; | |
| 2726 | int range_length, byterange_mode; | ||
| 2727 | |||
| 2728 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 18 times.
|
36 | for (i = 0; i < hls->nb_varstreams; i++) { |
| 2729 | 18 | char *filename = NULL; | |
| 2730 | 18 | vs = &hls->var_streams[i]; | |
| 2731 | 18 | oc = vs->avf; | |
| 2732 | 18 | vtt_oc = vs->vtt_avf; | |
| 2733 | 18 | old_filename = av_strdup(oc->url); | |
| 2734 | 18 | use_temp_file = 0; | |
| 2735 | |||
| 2736 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (!old_filename) { |
| 2737 | ✗ | return AVERROR(ENOMEM); | |
| 2738 | } | ||
| 2739 |
2/4✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 18 times.
|
18 | if (hls->key_info_file || hls->encrypt) { |
| 2740 | ✗ | av_dict_set(&options, "encryption_key", vs->key_string, 0); | |
| 2741 | ✗ | av_dict_set(&options, "encryption_iv", vs->iv_string, 0); | |
| 2742 | ✗ | filename = av_asprintf("crypto:%s", oc->url); | |
| 2743 | } else { | ||
| 2744 | 18 | filename = av_asprintf("%s", oc->url); | |
| 2745 | } | ||
| 2746 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (!filename) { |
| 2747 | ✗ | av_dict_free(&options); | |
| 2748 | ✗ | av_freep(&old_filename); | |
| 2749 | ✗ | return AVERROR(ENOMEM); | |
| 2750 | } | ||
| 2751 | |||
| 2752 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 16 times.
|
18 | if (hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 2753 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
|
2 | if (!vs->init_range_length) { |
| 2754 | 1 | uint8_t *buffer = NULL; | |
| 2755 | 1 | av_write_frame(oc, NULL); /* Flush any buffered data */ | |
| 2756 | |||
| 2757 | 1 | int init_range_length = avio_close_dyn_buf(oc->pb, &buffer); | |
| 2758 | 1 | avio_write(vs->out, buffer, init_range_length); | |
| 2759 | 1 | av_freep(&buffer); | |
| 2760 | 1 | vs->init_range_length = init_range_length; | |
| 2761 | 1 | avio_open_dyn_buf(&oc->pb); | |
| 2762 | 1 | vs->packets_written = 0; | |
| 2763 | 1 | vs->start_pos = init_range_length; | |
| 2764 |
2/4✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
|
1 | byterange_mode = (hls->flags & HLS_SINGLE_FILE) || (hls->max_seg_size > 0); |
| 2765 |
1/2✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
|
1 | if (!byterange_mode) { |
| 2766 | 1 | ff_format_io_close(s, &vs->out); | |
| 2767 | 1 | hlsenc_io_close(s, &vs->out, vs->base_output_dirname); | |
| 2768 | } | ||
| 2769 | } | ||
| 2770 | } | ||
| 2771 |
2/2✓ Branch 0 taken 17 times.
✓ Branch 1 taken 1 times.
|
18 | if (!(hls->flags & HLS_SINGLE_FILE)) { |
| 2772 | 17 | set_http_options(s, &options, hls); | |
| 2773 | 17 | ret = hlsenc_io_open(s, &vs->out, filename, &options); | |
| 2774 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 17 times.
|
17 | if (ret < 0) { |
| 2775 | ✗ | av_log(s, AV_LOG_ERROR, "Failed to open file '%s'\n", oc->url); | |
| 2776 | ✗ | goto failed; | |
| 2777 | } | ||
| 2778 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 15 times.
|
17 | if (hls->segment_type == SEGMENT_TYPE_FMP4) |
| 2779 | 2 | write_styp(vs->out); | |
| 2780 | } | ||
| 2781 | 18 | ret = flush_dynbuf(vs, &range_length); | |
| 2782 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 2783 | ✗ | goto failed; | |
| 2784 | |||
| 2785 | 18 | vs->size = range_length; | |
| 2786 | 18 | ret = hlsenc_io_close(s, &vs->out, filename); | |
| 2787 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) { |
| 2788 | ✗ | av_log(s, AV_LOG_WARNING, "upload segment failed, will retry with a new http session.\n"); | |
| 2789 | ✗ | ff_format_io_close(s, &vs->out); | |
| 2790 | ✗ | ret = hlsenc_io_open(s, &vs->out, filename, &options); | |
| 2791 | ✗ | if (ret < 0) { | |
| 2792 | ✗ | av_log(s, AV_LOG_ERROR, "Failed to open file '%s'\n", oc->url); | |
| 2793 | ✗ | goto failed; | |
| 2794 | } | ||
| 2795 | ✗ | reflush_dynbuf(vs, &range_length); | |
| 2796 | ✗ | ret = hlsenc_io_close(s, &vs->out, filename); | |
| 2797 | ✗ | if (ret < 0) | |
| 2798 | ✗ | av_log(s, AV_LOG_WARNING, "Failed to upload file '%s' at the end.\n", oc->url); | |
| 2799 | } | ||
| 2800 |
2/2✓ Branch 0 taken 17 times.
✓ Branch 1 taken 1 times.
|
18 | if (hls->flags & HLS_SINGLE_FILE) { |
| 2801 |
2/4✓ Branch 0 taken 1 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 1 times.
|
1 | if (hls->key_info_file || hls->encrypt) { |
| 2802 | ✗ | vs->size = append_single_file(s, vs); | |
| 2803 | } | ||
| 2804 | 1 | hlsenc_io_close(s, &vs->out_single_file, vs->basename); | |
| 2805 | } | ||
| 2806 | 17 | failed: | |
| 2807 | 18 | av_freep(&vs->temp_buffer); | |
| 2808 | 18 | av_dict_free(&options); | |
| 2809 | 18 | av_freep(&filename); | |
| 2810 | 18 | av_write_trailer(oc); | |
| 2811 |
1/2✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
|
18 | if (oc->url[0]) { |
| 2812 | 18 | proto = avio_find_protocol_name(oc->url); | |
| 2813 |
3/6✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 18 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 18 times.
|
18 | use_temp_file = proto && !strcmp(proto, "file") && (hls->flags & HLS_TEMP_FILE); |
| 2814 | } | ||
| 2815 | |||
| 2816 | // rename that segment from .tmp to the real one | ||
| 2817 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
18 | if (use_temp_file && !(hls->flags & HLS_SINGLE_FILE)) { |
| 2818 | ✗ | hls_rename_temp_file(s, oc); | |
| 2819 | ✗ | av_freep(&old_filename); | |
| 2820 | ✗ | old_filename = av_strdup(oc->url); | |
| 2821 | |||
| 2822 | ✗ | if (!old_filename) { | |
| 2823 | ✗ | return AVERROR(ENOMEM); | |
| 2824 | } | ||
| 2825 | } | ||
| 2826 | |||
| 2827 | /* after av_write_trailer, then duration + 1 duration per packet */ | ||
| 2828 | 18 | hls_append_segment(s, hls, vs, vs->duration + vs->dpp, vs->start_pos, vs->size); | |
| 2829 | |||
| 2830 | 18 | sls_flag_file_rename(hls, vs, old_filename); | |
| 2831 | |||
| 2832 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (vtt_oc) { |
| 2833 | ✗ | if (vtt_oc->pb) | |
| 2834 | ✗ | av_write_trailer(vtt_oc); | |
| 2835 | ✗ | vs->size = avio_tell(vs->vtt_avf->pb) - vs->start_pos; | |
| 2836 | ✗ | ff_format_io_close(s, &vtt_oc->pb); | |
| 2837 | } | ||
| 2838 | 18 | ret = hls_window(s, 1, vs); | |
| 2839 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) { |
| 2840 | ✗ | av_log(s, AV_LOG_WARNING, "upload playlist failed, will retry with a new http session.\n"); | |
| 2841 | ✗ | ff_format_io_close(s, &vs->out); | |
| 2842 | ✗ | hls_window(s, 1, vs); | |
| 2843 | } | ||
| 2844 | 18 | ffio_free_dyn_buf(&oc->pb); | |
| 2845 | |||
| 2846 | 18 | av_free(old_filename); | |
| 2847 | } | ||
| 2848 | |||
| 2849 | 18 | return 0; | |
| 2850 | } | ||
| 2851 | |||
| 2852 | |||
| 2853 | 18 | static int hls_init(AVFormatContext *s) | |
| 2854 | { | ||
| 2855 | 18 | int ret = 0; | |
| 2856 | 18 | int i = 0; | |
| 2857 | 18 | int j = 0; | |
| 2858 | 18 | HLSContext *hls = s->priv_data; | |
| 2859 | const char *pattern; | ||
| 2860 | 18 | VariantStream *vs = NULL; | |
| 2861 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 17 times.
|
18 | const char *vtt_pattern = hls->flags & HLS_SINGLE_FILE ? ".vtt" : "%d.vtt"; |
| 2862 | 18 | int http_base_proto = ff_is_http_proto(s->url); | |
| 2863 | 18 | int fmp4_init_filename_len = strlen(hls->fmp4_init_filename) + 1; | |
| 2864 | 18 | double initial_program_date_time = av_gettime() / 1000000.0; | |
| 2865 | |||
| 2866 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->use_localtime) { |
| 2867 | ✗ | pattern = get_default_pattern_localtime_fmt(s); | |
| 2868 | } else { | ||
| 2869 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 16 times.
|
18 | pattern = hls->segment_type == SEGMENT_TYPE_FMP4 ? "%d.m4s" : "%d.ts"; |
| 2870 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 17 times.
|
18 | if (hls->flags & HLS_SINGLE_FILE) |
| 2871 | 1 | pattern += 2; | |
| 2872 | } | ||
| 2873 | |||
| 2874 | 18 | hls->has_default_key = 0; | |
| 2875 | 18 | hls->has_video_m3u8 = 0; | |
| 2876 | 18 | ret = update_variant_stream_info(s); | |
| 2877 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) { |
| 2878 | ✗ | av_log(s, AV_LOG_ERROR, "Variant stream info update failed with status %x\n", | |
| 2879 | ret); | ||
| 2880 | ✗ | return ret; | |
| 2881 | } | ||
| 2882 | |||
| 2883 |
2/4✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 18 times.
|
18 | if (!hls->method && http_base_proto) { |
| 2884 | ✗ | av_log(hls, AV_LOG_WARNING, "No HTTP method set, hls muxer defaulting to method PUT.\n"); | |
| 2885 | } | ||
| 2886 | |||
| 2887 | 18 | ret = validate_name(hls->nb_varstreams, s->url); | |
| 2888 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 2889 | ✗ | return ret; | |
| 2890 | |||
| 2891 |
1/2✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
|
18 | if (hls->segment_filename) { |
| 2892 | 18 | ret = validate_name(hls->nb_varstreams, hls->segment_filename); | |
| 2893 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 2894 | ✗ | return ret; | |
| 2895 | } | ||
| 2896 | |||
| 2897 |
2/2✓ Branch 1 taken 3 times.
✓ Branch 2 taken 15 times.
|
18 | if (av_strcasecmp(hls->fmp4_init_filename, "init.mp4")) { |
| 2898 | 3 | ret = validate_name(hls->nb_varstreams, hls->fmp4_init_filename); | |
| 2899 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
|
3 | if (ret < 0) |
| 2900 | ✗ | return ret; | |
| 2901 | } | ||
| 2902 | |||
| 2903 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->subtitle_filename) { |
| 2904 | ✗ | ret = validate_name(hls->nb_varstreams, hls->subtitle_filename); | |
| 2905 | ✗ | if (ret < 0) | |
| 2906 | ✗ | return ret; | |
| 2907 | } | ||
| 2908 | |||
| 2909 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (hls->master_pl_name) { |
| 2910 | ✗ | ret = update_master_pl_info(s); | |
| 2911 | ✗ | if (ret < 0) { | |
| 2912 | ✗ | av_log(s, AV_LOG_ERROR, "Master stream info update failed with status %x\n", | |
| 2913 | ret); | ||
| 2914 | ✗ | return ret; | |
| 2915 | } | ||
| 2916 | } | ||
| 2917 | |||
| 2918 |
1/2✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
|
18 | if ((hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH) || |
| 2919 |
1/2✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
|
18 | (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_MICROSECONDS_SINCE_EPOCH) || |
| 2920 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_FORMATTED_DATETIME)) { |
| 2921 | ✗ | time_t t = time(NULL); | |
| 2922 | ✗ | if (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_MICROSECONDS_SINCE_EPOCH) { | |
| 2923 | ✗ | hls->start_sequence = av_gettime(); | |
| 2924 | ✗ | } else if (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH) { | |
| 2925 | ✗ | hls->start_sequence = (int64_t)t; | |
| 2926 | ✗ | } else if (hls->start_sequence_source_type == HLS_START_SEQUENCE_AS_FORMATTED_DATETIME) { | |
| 2927 | char b[15]; | ||
| 2928 | struct tm *p, tmbuf; | ||
| 2929 | ✗ | if (!(p = localtime_r(&t, &tmbuf))) | |
| 2930 | ✗ | return AVERROR(errno); | |
| 2931 | ✗ | if (!strftime(b, sizeof(b), "%Y%m%d%H%M%S", p)) | |
| 2932 | ✗ | return AVERROR(ENOMEM); | |
| 2933 | ✗ | hls->start_sequence = strtoll(b, NULL, 10); | |
| 2934 | } | ||
| 2935 | ✗ | av_log(hls, AV_LOG_DEBUG, "start_number evaluated to %"PRId64"\n", hls->start_sequence); | |
| 2936 | } | ||
| 2937 | |||
| 2938 |
3/4✓ Branch 0 taken 1 times.
✓ Branch 1 taken 17 times.
✓ Branch 2 taken 1 times.
✗ Branch 3 not taken.
|
18 | hls->recording_time = hls->init_time && hls->max_nb_segments > 0 ? hls->init_time : hls->time; |
| 2939 | |||
| 2940 |
1/4✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
|
18 | if (hls->flags & HLS_SPLIT_BY_TIME && hls->flags & HLS_INDEPENDENT_SEGMENTS) { |
| 2941 | // Independent segments cannot be guaranteed when splitting by time | ||
| 2942 | ✗ | hls->flags &= ~HLS_INDEPENDENT_SEGMENTS; | |
| 2943 | ✗ | av_log(s, AV_LOG_WARNING, | |
| 2944 | "'split_by_time' and 'independent_segments' cannot be " | ||
| 2945 | "enabled together. Disabling 'independent_segments' flag\n"); | ||
| 2946 | } | ||
| 2947 | |||
| 2948 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 18 times.
|
36 | for (i = 0; i < hls->nb_varstreams; i++) { |
| 2949 | 18 | vs = &hls->var_streams[i]; | |
| 2950 | |||
| 2951 | 18 | ret = format_name(s->url, &vs->m3u8_name, i, vs->varname); | |
| 2952 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 2953 | ✗ | return ret; | |
| 2954 | |||
| 2955 | 18 | vs->sequence = hls->start_sequence; | |
| 2956 | 18 | vs->start_pts = AV_NOPTS_VALUE; | |
| 2957 | 18 | vs->end_pts = AV_NOPTS_VALUE; | |
| 2958 | 18 | vs->current_segment_final_filename_fmt[0] = '\0'; | |
| 2959 | 18 | vs->initial_prog_date_time = initial_program_date_time; | |
| 2960 | |||
| 2961 |
2/2✓ Branch 0 taken 18 times.
✓ Branch 1 taken 18 times.
|
36 | for (j = 0; j < vs->nb_streams; j++) { |
| 2962 | 18 | vs->has_video += vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO; | |
| 2963 | /* Get one video stream to reference for split segments | ||
| 2964 | * so use the first video stream index. */ | ||
| 2965 |
3/4✓ Branch 0 taken 6 times.
✓ Branch 1 taken 12 times.
✓ Branch 2 taken 6 times.
✗ Branch 3 not taken.
|
18 | if ((vs->has_video == 1) && (vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)) { |
| 2966 | 6 | vs->reference_stream_index = vs->streams[j]->index; | |
| 2967 | } | ||
| 2968 | 18 | vs->has_subtitle += vs->streams[j]->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE; | |
| 2969 | } | ||
| 2970 | |||
| 2971 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (vs->has_video > 1) |
| 2972 | ✗ | av_log(s, AV_LOG_WARNING, "More than a single video stream present, expect issues decoding it.\n"); | |
| 2973 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 16 times.
|
18 | if (hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 2974 | #if CONFIG_MP4_MUXER | ||
| 2975 | EXTERN const FFOutputFormat ff_mp4_muxer; | ||
| 2976 | 2 | vs->oformat = &ff_mp4_muxer.p; | |
| 2977 | #else | ||
| 2978 | return AVERROR_MUXER_NOT_FOUND; | ||
| 2979 | #endif | ||
| 2980 | } else { | ||
| 2981 | EXTERN const FFOutputFormat ff_mpegts_muxer; | ||
| 2982 | 16 | vs->oformat = &ff_mpegts_muxer.p; | |
| 2983 | } | ||
| 2984 | |||
| 2985 |
1/2✓ Branch 0 taken 18 times.
✗ Branch 1 not taken.
|
18 | if (hls->segment_filename) { |
| 2986 | 18 | ret = format_name(hls->segment_filename, &vs->basename, i, vs->varname); | |
| 2987 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 2988 | ✗ | return ret; | |
| 2989 | } else { | ||
| 2990 | ✗ | char *p = strrchr(vs->m3u8_name, '.'); | |
| 2991 | ✗ | if (p) | |
| 2992 | ✗ | *p = '\0'; | |
| 2993 | |||
| 2994 | ✗ | vs->basename = av_asprintf("%s%s", vs->m3u8_name, pattern); | |
| 2995 | ✗ | if (!vs->basename) | |
| 2996 | ✗ | return AVERROR(ENOMEM); | |
| 2997 | |||
| 2998 | ✗ | if (p) | |
| 2999 | ✗ | *p = '.'; | |
| 3000 | } | ||
| 3001 | |||
| 3002 |
2/2✓ Branch 0 taken 2 times.
✓ Branch 1 taken 16 times.
|
18 | if (hls->segment_type == SEGMENT_TYPE_FMP4) { |
| 3003 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (hls->nb_varstreams > 1) |
| 3004 | ✗ | fmp4_init_filename_len += strlen(POSTFIX_PATTERN); | |
| 3005 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (hls->flags & HLS_SINGLE_FILE) { |
| 3006 | ✗ | vs->fmp4_init_filename = av_strdup(vs->basename); | |
| 3007 | ✗ | if (!vs->fmp4_init_filename) | |
| 3008 | ✗ | return AVERROR(ENOMEM); | |
| 3009 | } else { | ||
| 3010 | 2 | vs->fmp4_init_filename = av_malloc(fmp4_init_filename_len); | |
| 3011 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (!vs->fmp4_init_filename) |
| 3012 | ✗ | return AVERROR(ENOMEM); | |
| 3013 | 2 | av_strlcpy(vs->fmp4_init_filename, hls->fmp4_init_filename, | |
| 3014 | fmp4_init_filename_len); | ||
| 3015 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (hls->nb_varstreams > 1) { |
| 3016 | ✗ | if (av_stristr(vs->fmp4_init_filename, "%v")) { | |
| 3017 | ✗ | av_freep(&vs->fmp4_init_filename); | |
| 3018 | ✗ | ret = format_name(hls->fmp4_init_filename, | |
| 3019 | &vs->fmp4_init_filename, i, vs->varname); | ||
| 3020 | } else { | ||
| 3021 | ✗ | ret = append_postfix(vs->fmp4_init_filename, fmp4_init_filename_len, i); | |
| 3022 | } | ||
| 3023 | ✗ | if (ret < 0) | |
| 3024 | ✗ | return ret; | |
| 3025 | } | ||
| 3026 | |||
| 3027 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (hls->use_localtime) { |
| 3028 | int r; | ||
| 3029 | ✗ | char *expanded = NULL; | |
| 3030 | |||
| 3031 | ✗ | r = strftime_expand(vs->fmp4_init_filename, &expanded); | |
| 3032 | ✗ | if (r < 0) { | |
| 3033 | ✗ | av_log(s, AV_LOG_ERROR, "Could not get segment filename with strftime\n"); | |
| 3034 | ✗ | return r; | |
| 3035 | } | ||
| 3036 | ✗ | av_free(vs->fmp4_init_filename); | |
| 3037 | ✗ | vs->fmp4_init_filename = expanded; | |
| 3038 | } | ||
| 3039 | |||
| 3040 | 2 | char *p = strrchr(vs->m3u8_name, '/'); | |
| 3041 |
1/2✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
|
2 | if (p) { |
| 3042 | 2 | char tmp = *(++p); | |
| 3043 | 2 | *p = '\0'; | |
| 3044 | 2 | vs->base_output_dirname = av_asprintf("%s%s", vs->m3u8_name, | |
| 3045 | vs->fmp4_init_filename); | ||
| 3046 | 2 | *p = tmp; | |
| 3047 | } else { | ||
| 3048 | ✗ | vs->base_output_dirname = av_strdup(vs->fmp4_init_filename); | |
| 3049 | } | ||
| 3050 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
|
2 | if (!vs->base_output_dirname) |
| 3051 | ✗ | return AVERROR(ENOMEM); | |
| 3052 | } | ||
| 3053 | } | ||
| 3054 | |||
| 3055 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | ret = hls->use_localtime ? sls_flag_check_duration_size(hls, vs) : sls_flag_check_duration_size_index(hls); |
| 3056 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (ret < 0) |
| 3057 | ✗ | return ret; | |
| 3058 | |||
| 3059 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 18 times.
|
18 | if (vs->has_subtitle) { |
| 3060 | EXTERN const FFOutputFormat ff_webvtt_muxer; | ||
| 3061 | ✗ | vs->vtt_oformat = &ff_webvtt_muxer.p; | |
| 3062 | |||
| 3063 | ✗ | char *p = strrchr(vs->m3u8_name, '.'); | |
| 3064 | ✗ | if (p) | |
| 3065 | ✗ | *p = '\0'; | |
| 3066 | |||
| 3067 | ✗ | vs->vtt_basename = av_asprintf("%s%s", vs->m3u8_name, vtt_pattern); | |
| 3068 | ✗ | if (!vs->vtt_basename) | |
| 3069 | ✗ | return AVERROR(ENOMEM); | |
| 3070 | |||
| 3071 | ✗ | if (hls->subtitle_filename) { | |
| 3072 | ✗ | ret = format_name(hls->subtitle_filename, &vs->vtt_m3u8_name, i, vs->varname); | |
| 3073 | ✗ | if (ret < 0) | |
| 3074 | ✗ | return ret; | |
| 3075 | } else { | ||
| 3076 | ✗ | vs->vtt_m3u8_name = av_asprintf("%s_vtt.m3u8", vs->m3u8_name); | |
| 3077 | ✗ | if (!vs->vtt_m3u8_name) | |
| 3078 | ✗ | return AVERROR(ENOMEM); | |
| 3079 | } | ||
| 3080 | ✗ | if (p) | |
| 3081 | ✗ | *p = '.'; | |
| 3082 | } | ||
| 3083 | |||
| 3084 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 18 times.
|
18 | if ((ret = hls_mux_init(s, vs)) < 0) |
| 3085 | ✗ | return ret; | |
| 3086 | |||
| 3087 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 17 times.
|
18 | if (hls->flags & HLS_APPEND_LIST) { |
| 3088 | 1 | parse_playlist(s, vs->m3u8_name, vs); | |
| 3089 | 1 | vs->discontinuity = 1; | |
| 3090 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
|
1 | if (hls->init_time > 0) { |
| 3091 | ✗ | av_log(s, AV_LOG_WARNING, "append_list mode does not support hls_init_time," | |
| 3092 | " hls_init_time value will have no effect\n"); | ||
| 3093 | ✗ | hls->init_time = 0; | |
| 3094 | ✗ | hls->recording_time = hls->time; | |
| 3095 | } | ||
| 3096 | } | ||
| 3097 | |||
| 3098 |
1/2✗ Branch 1 not taken.
✓ Branch 2 taken 18 times.
|
18 | if ((ret = hls_start(s, vs)) < 0) |
| 3099 | ✗ | return ret; | |
| 3100 | 18 | vs->number++; | |
| 3101 | } | ||
| 3102 | |||
| 3103 | 18 | return ret; | |
| 3104 | } | ||
| 3105 | |||
| 3106 | #define OFFSET(x) offsetof(HLSContext, x) | ||
| 3107 | #define E AV_OPT_FLAG_ENCODING_PARAM | ||
| 3108 | static const AVOption options[] = { | ||
| 3109 | {"start_number", "set first number in the sequence", OFFSET(start_sequence),AV_OPT_TYPE_INT64, {.i64 = 0}, 0, INT64_MAX, E}, | ||
| 3110 | {"hls_time", "set segment length", OFFSET(time), AV_OPT_TYPE_DURATION, {.i64 = 2000000}, 0, INT64_MAX, E}, | ||
| 3111 | {"hls_init_time", "set segment length at init list", OFFSET(init_time), AV_OPT_TYPE_DURATION, {.i64 = 0}, 0, INT64_MAX, E}, | ||
| 3112 | {"hls_list_size", "set maximum number of playlist entries", OFFSET(max_nb_segments), AV_OPT_TYPE_INT, {.i64 = 5}, 0, INT_MAX, E}, | ||
| 3113 | {"hls_delete_threshold", "set number of unreferenced segments to keep before deleting", OFFSET(hls_delete_threshold), AV_OPT_TYPE_INT, {.i64 = 1}, 1, INT_MAX, E}, | ||
| 3114 | {"hls_vtt_options","set hls vtt list of options for the container format used for hls", OFFSET(vtt_format_options_str), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3115 | {"hls_allow_cache", "explicitly set whether the client MAY (1) or MUST NOT (0) cache media segments", OFFSET(allowcache), AV_OPT_TYPE_INT, {.i64 = -1}, INT_MIN, INT_MAX, E}, | ||
| 3116 | {"hls_base_url", "url to prepend to each playlist entry", OFFSET(baseurl), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3117 | {"hls_segment_filename", "filename template for segment files", OFFSET(segment_filename), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3118 | {"hls_segment_options","set segments files format options of hls", OFFSET(format_options), AV_OPT_TYPE_DICT, {.str = NULL}, 0, 0, E}, | ||
| 3119 | {"hls_segment_size", "maximum size per segment file, (in bytes)", OFFSET(max_seg_size), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, E}, | ||
| 3120 | {"hls_key_info_file", "file with key URI and key file path", OFFSET(key_info_file), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3121 | {"hls_enc", "enable AES128 encryption support", OFFSET(encrypt), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, E}, | ||
| 3122 | {"hls_enc_key", "hex-coded 16 byte key to encrypt the segments", OFFSET(key), AV_OPT_TYPE_STRING, .flags = E}, | ||
| 3123 | {"hls_enc_key_url", "url to access the key to decrypt the segments", OFFSET(key_url), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3124 | {"hls_enc_iv", "hex-coded 16 byte initialization vector", OFFSET(iv), AV_OPT_TYPE_STRING, .flags = E}, | ||
| 3125 | {"hls_subtitle_path", "set path of hls subtitles", OFFSET(subtitle_filename), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3126 | {"hls_segment_type", "set hls segment files type", OFFSET(segment_type), AV_OPT_TYPE_INT, {.i64 = SEGMENT_TYPE_MPEGTS }, 0, SEGMENT_TYPE_FMP4, E, .unit = "segment_type"}, | ||
| 3127 | {"mpegts", "make segment file to mpegts files in m3u8", 0, AV_OPT_TYPE_CONST, {.i64 = SEGMENT_TYPE_MPEGTS }, 0, UINT_MAX, E, .unit = "segment_type"}, | ||
| 3128 | {"fmp4", "make segment file to fragment mp4 files in m3u8", 0, AV_OPT_TYPE_CONST, {.i64 = SEGMENT_TYPE_FMP4 }, 0, UINT_MAX, E, .unit = "segment_type"}, | ||
| 3129 | {"hls_fmp4_init_filename", "set fragment mp4 file init filename", OFFSET(fmp4_init_filename), AV_OPT_TYPE_STRING, {.str = "init.mp4"}, 0, 0, E}, | ||
| 3130 | {"hls_fmp4_init_resend", "resend fragment mp4 init file after refresh m3u8 every time", OFFSET(resend_init_file), AV_OPT_TYPE_BOOL, {.i64 = 0 }, 0, 1, E }, | ||
| 3131 | {"hls_flags", "set flags affecting HLS playlist and media file generation", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64 = 0 }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3132 | {"single_file", "generate a single media file indexed with byte ranges", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_SINGLE_FILE }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3133 | {"temp_file", "write segment and playlist to temporary file and rename when complete", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_TEMP_FILE }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3134 | {"delete_segments", "delete segment files that are no longer part of the playlist", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_DELETE_SEGMENTS }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3135 | {"round_durations", "round durations in m3u8 to whole numbers", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_ROUND_DURATIONS }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3136 | {"discont_start", "start the playlist with a discontinuity tag", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_DISCONT_START }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3137 | {"omit_endlist", "Do not append an endlist when ending stream", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_OMIT_ENDLIST }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3138 | {"split_by_time", "split the hls segment by time which user set by hls_time", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_SPLIT_BY_TIME }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3139 | {"append_list", "append the new segments into old hls segment list", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_APPEND_LIST }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3140 | {"program_date_time", "add EXT-X-PROGRAM-DATE-TIME", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_PROGRAM_DATE_TIME }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3141 | {"second_level_segment_index", "include segment index in segment filenames when use_localtime", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_SECOND_LEVEL_SEGMENT_INDEX }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3142 | {"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, .unit = "flags"}, | ||
| 3143 | {"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, .unit = "flags"}, | ||
| 3144 | {"periodic_rekey", "reload keyinfo file periodically for re-keying", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_PERIODIC_REKEY }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3145 | {"independent_segments", "add EXT-X-INDEPENDENT-SEGMENTS, whenever applicable", 0, AV_OPT_TYPE_CONST, { .i64 = HLS_INDEPENDENT_SEGMENTS }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3146 | {"iframes_only", "add EXT-X-I-FRAMES-ONLY, whenever applicable", 0, AV_OPT_TYPE_CONST, { .i64 = HLS_I_FRAMES_ONLY }, 0, UINT_MAX, E, .unit = "flags"}, | ||
| 3147 | {"strftime", "set filename expansion with strftime at segment creation", OFFSET(use_localtime), AV_OPT_TYPE_BOOL, {.i64 = 0 }, 0, 1, E }, | ||
| 3148 | {"strftime_mkdir", "create last directory component in strftime-generated filename", OFFSET(use_localtime_mkdir), AV_OPT_TYPE_BOOL, {.i64 = 0 }, 0, 1, E }, | ||
| 3149 | {"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, .unit = "pl_type" }, | ||
| 3150 | {"event", "EVENT playlist", 0, AV_OPT_TYPE_CONST, {.i64 = PLAYLIST_TYPE_EVENT }, INT_MIN, INT_MAX, E, .unit = "pl_type" }, | ||
| 3151 | {"vod", "VOD playlist", 0, AV_OPT_TYPE_CONST, {.i64 = PLAYLIST_TYPE_VOD }, INT_MIN, INT_MAX, E, .unit = "pl_type" }, | ||
| 3152 | {"method", "set the HTTP method(default: PUT)", OFFSET(method), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3153 | {"hls_start_number_source", "set source of first number in sequence", OFFSET(start_sequence_source_type), AV_OPT_TYPE_INT, {.i64 = HLS_START_SEQUENCE_AS_START_NUMBER }, 0, HLS_START_SEQUENCE_LAST-1, E, .unit = "start_sequence_source_type" }, | ||
| 3154 | {"generic", "start_number value (default)", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_START_SEQUENCE_AS_START_NUMBER }, INT_MIN, INT_MAX, E, .unit = "start_sequence_source_type" }, | ||
| 3155 | {"epoch", "seconds since epoch", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_START_SEQUENCE_AS_SECONDS_SINCE_EPOCH }, INT_MIN, INT_MAX, E, .unit = "start_sequence_source_type" }, | ||
| 3156 | {"epoch_us", "microseconds since epoch", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_START_SEQUENCE_AS_MICROSECONDS_SINCE_EPOCH }, INT_MIN, INT_MAX, E, .unit = "start_sequence_source_type" }, | ||
| 3157 | {"datetime", "current datetime as YYYYMMDDhhmmss", 0, AV_OPT_TYPE_CONST, {.i64 = HLS_START_SEQUENCE_AS_FORMATTED_DATETIME }, INT_MIN, INT_MAX, E, .unit = "start_sequence_source_type" }, | ||
| 3158 | {"http_user_agent", "override User-Agent field in HTTP header", OFFSET(user_agent), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3159 | {"var_stream_map", "Variant stream map string", OFFSET(var_stream_map), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3160 | {"cc_stream_map", "Closed captions stream map string", OFFSET(cc_stream_map), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3161 | {"master_pl_name", "Create HLS master playlist with this name", OFFSET(master_pl_name), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E}, | ||
| 3162 | {"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}, | ||
| 3163 | {"http_persistent", "Use persistent HTTP connections", OFFSET(http_persistent), AV_OPT_TYPE_BOOL, {.i64 = 0 }, 0, 1, E }, | ||
| 3164 | {"timeout", "set timeout for socket I/O operations", OFFSET(timeout), AV_OPT_TYPE_DURATION, { .i64 = -1 }, -1, INT_MAX, .flags = E }, | ||
| 3165 | {"ignore_io_errors", "Ignore IO errors for stable long-duration runs with network output", OFFSET(ignore_io_errors), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, E }, | ||
| 3166 | {"headers", "set custom HTTP headers, can override built in default headers", OFFSET(headers), AV_OPT_TYPE_STRING, { .str = NULL }, 0, 0, E }, | ||
| 3167 | { NULL }, | ||
| 3168 | }; | ||
| 3169 | |||
| 3170 | static const AVClass hls_class = { | ||
| 3171 | .class_name = "hls muxer", | ||
| 3172 | .item_name = av_default_item_name, | ||
| 3173 | .option = options, | ||
| 3174 | .version = LIBAVUTIL_VERSION_INT, | ||
| 3175 | }; | ||
| 3176 | |||
| 3177 | |||
| 3178 | const FFOutputFormat ff_hls_muxer = { | ||
| 3179 | .p.name = "hls", | ||
| 3180 | .p.long_name = NULL_IF_CONFIG_SMALL("Apple HTTP Live Streaming"), | ||
| 3181 | .p.extensions = "m3u8", | ||
| 3182 | .p.audio_codec = AV_CODEC_ID_AAC, | ||
| 3183 | .p.video_codec = AV_CODEC_ID_H264, | ||
| 3184 | .p.subtitle_codec = AV_CODEC_ID_WEBVTT, | ||
| 3185 | .p.flags = AVFMT_NOFILE | AVFMT_GLOBALHEADER | AVFMT_NODIMENSIONS, | ||
| 3186 | .p.priv_class = &hls_class, | ||
| 3187 | .flags_internal = FF_OFMT_FLAG_ALLOW_FLUSH, | ||
| 3188 | .priv_data_size = sizeof(HLSContext), | ||
| 3189 | .init = hls_init, | ||
| 3190 | .write_header = hls_write_header, | ||
| 3191 | .write_packet = hls_write_packet, | ||
| 3192 | .write_trailer = hls_write_trailer, | ||
| 3193 | .deinit = hls_deinit, | ||
| 3194 | }; | ||
| 3195 |