Coverage Report

Created: 2025-08-28 07:12

/src/ffmpeg/libavformat/hls.c
Line
Count
Source (jump to first uncovered line)
1
/*
2
 * Apple HTTP Live Streaming demuxer
3
 * Copyright (c) 2010 Martin Storsjo
4
 * Copyright (c) 2013 Anssi Hannula
5
 * Copyright (c) 2021 Nachiket Tarate
6
 *
7
 * This file is part of FFmpeg.
8
 *
9
 * FFmpeg is free software; you can redistribute it and/or
10
 * modify it under the terms of the GNU Lesser General Public
11
 * License as published by the Free Software Foundation; either
12
 * version 2.1 of the License, or (at your option) any later version.
13
 *
14
 * FFmpeg is distributed in the hope that it will be useful,
15
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17
 * Lesser General Public License for more details.
18
 *
19
 * You should have received a copy of the GNU Lesser General Public
20
 * License along with FFmpeg; if not, write to the Free Software
21
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22
 */
23
24
/**
25
 * @file
26
 * Apple HTTP Live Streaming demuxer
27
 * https://www.rfc-editor.org/rfc/rfc8216.txt
28
 */
29
30
#include "config_components.h"
31
32
#include "libavformat/http.h"
33
#include "libavutil/aes.h"
34
#include "libavutil/avstring.h"
35
#include "libavutil/avassert.h"
36
#include "libavutil/intreadwrite.h"
37
#include "libavutil/mathematics.h"
38
#include "libavutil/mem.h"
39
#include "libavutil/opt.h"
40
#include "libavutil/dict.h"
41
#include "libavutil/time.h"
42
#include "avformat.h"
43
#include "demux.h"
44
#include "internal.h"
45
#include "avio_internal.h"
46
#include "id3v2.h"
47
#include "url.h"
48
49
#include "hls_sample_encryption.h"
50
51
830
#define INITIAL_BUFFER_SIZE 32768
52
53
#define MAX_FIELD_LEN 64
54
#define MAX_CHARACTERISTICS_LEN 512
55
56
0
#define MPEG_TIME_BASE 90000
57
0
#define MPEG_TIME_BASE_Q (AVRational){1, MPEG_TIME_BASE}
58
59
/*
60
 * An apple http stream consists of a playlist with media segment files,
61
 * played sequentially. There may be several playlists with the same
62
 * video content, in different bandwidth variants, that are played in
63
 * parallel (preferably only one bandwidth variant at a time). In this case,
64
 * the user supplied the url to a main playlist that only lists the variant
65
 * playlists.
66
 *
67
 * If the main playlist doesn't point at any variants, we still create
68
 * one anonymous toplevel variant for this, to maintain the structure.
69
 */
70
71
enum KeyType {
72
    KEY_NONE,
73
    KEY_AES_128,
74
    KEY_SAMPLE_AES
75
};
76
77
struct segment {
78
    int64_t duration;
79
    int64_t url_offset;
80
    int64_t size;
81
    char *url;
82
    char *key;
83
    enum KeyType key_type;
84
    uint8_t iv[16];
85
    /* associated Media Initialization Section, treated as a segment */
86
    struct segment *init_section;
87
};
88
89
struct rendition;
90
91
enum PlaylistType {
92
    PLS_TYPE_UNSPECIFIED,
93
    PLS_TYPE_EVENT,
94
    PLS_TYPE_VOD
95
};
96
97
/*
98
 * Each playlist has its own demuxer. If it currently is active,
99
 * it has an open AVIOContext too, and potentially an AVPacket
100
 * containing the next packet from this stream.
101
 */
102
struct playlist {
103
    char url[MAX_URL_SIZE];
104
    FFIOContext pb;
105
    uint8_t* read_buffer;
106
    AVIOContext *input;
107
    int input_read_done;
108
    AVIOContext *input_next;
109
    int input_next_requested;
110
    AVFormatContext *parent;
111
    int index;
112
    AVFormatContext *ctx;
113
    AVPacket *pkt;
114
    int has_noheader_flag;
115
116
    /* main demuxer streams associated with this playlist
117
     * indexed by the subdemuxer stream indexes */
118
    AVStream **main_streams;
119
    int n_main_streams;
120
121
    int finished;
122
    enum PlaylistType type;
123
    int64_t target_duration;
124
    int64_t start_seq_no;
125
    int time_offset_flag;
126
    int64_t start_time_offset;
127
    int n_segments;
128
    struct segment **segments;
129
    int needed;
130
    int broken;
131
    int64_t cur_seq_no;
132
    int64_t last_seq_no;
133
    int m3u8_hold_counters;
134
    int64_t cur_seg_offset;
135
    int64_t last_load_time;
136
137
    /* Currently active Media Initialization Section */
138
    struct segment *cur_init_section;
139
    uint8_t *init_sec_buf;
140
    unsigned int init_sec_buf_size;
141
    unsigned int init_sec_data_len;
142
    unsigned int init_sec_buf_read_offset;
143
144
    char key_url[MAX_URL_SIZE];
145
    uint8_t key[16];
146
147
    /* ID3 timestamp handling (elementary audio streams have ID3 timestamps
148
     * (and possibly other ID3 tags) in the beginning of each segment) */
149
    int is_id3_timestamped; /* -1: not yet known */
150
    int64_t id3_mpegts_timestamp; /* in mpegts tb */
151
    int64_t id3_offset; /* in stream original tb */
152
    uint8_t* id3_buf; /* temp buffer for id3 parsing */
153
    unsigned int id3_buf_size;
154
    AVDictionary *id3_initial; /* data from first id3 tag */
155
    int id3_found; /* ID3 tag found at some point */
156
    int id3_changed; /* ID3 tag data has changed at some point */
157
    ID3v2ExtraMeta *id3_deferred_extra; /* stored here until subdemuxer is opened */
158
159
    HLSAudioSetupInfo audio_setup_info;
160
161
    int64_t seek_timestamp;
162
    int seek_flags;
163
    int seek_stream_index; /* into subdemuxer stream array */
164
165
    /* Renditions associated with this playlist, if any.
166
     * Alternative rendition playlists have a single rendition associated
167
     * with them, and variant main Media Playlists may have
168
     * multiple (playlist-less) renditions associated with them. */
169
    int n_renditions;
170
    struct rendition **renditions;
171
172
    /* Media Initialization Sections (EXT-X-MAP) associated with this
173
     * playlist, if any. */
174
    int n_init_sections;
175
    struct segment **init_sections;
176
    int is_subtitle; /* Indicates if it's a subtitle playlist */
177
};
178
179
/*
180
 * Renditions are e.g. alternative subtitle or audio streams.
181
 * The rendition may either be an external playlist or it may be
182
 * contained in the main Media Playlist of the variant (in which case
183
 * playlist is NULL).
184
 */
185
struct rendition {
186
    enum AVMediaType type;
187
    struct playlist *playlist;
188
    char group_id[MAX_FIELD_LEN];
189
    char language[MAX_FIELD_LEN];
190
    char name[MAX_FIELD_LEN];
191
    int disposition;
192
};
193
194
struct variant {
195
    int bandwidth;
196
197
    /* every variant contains at least the main Media Playlist in index 0 */
198
    int n_playlists;
199
    struct playlist **playlists;
200
201
    char audio_group[MAX_FIELD_LEN];
202
    char video_group[MAX_FIELD_LEN];
203
    char subtitles_group[MAX_FIELD_LEN];
204
};
205
206
typedef struct HLSContext {
207
    AVClass *class;
208
    AVFormatContext *ctx;
209
    int n_variants;
210
    struct variant **variants;
211
    int n_playlists;
212
    struct playlist **playlists;
213
    int n_renditions;
214
    struct rendition **renditions;
215
216
    int64_t cur_seq_no;
217
    int m3u8_hold_counters;
218
    int live_start_index;
219
    int prefer_x_start;
220
    int first_packet;
221
    int64_t first_timestamp;
222
    int64_t cur_timestamp;
223
    AVIOInterruptCB *interrupt_callback;
224
    AVDictionary *avio_opts;
225
    AVDictionary *seg_format_opts;
226
    char *allowed_extensions;
227
    char *allowed_segment_extensions;
228
    int extension_picky;
229
    int max_reload;
230
    int http_persistent;
231
    int http_multiple;
232
    int http_seekable;
233
    int seg_max_retry;
234
    AVIOContext *playlist_pb;
235
    HLSCryptoContext  crypto_ctx;
236
} HLSContext;
237
238
static void free_segment_dynarray(struct segment **segments, int n_segments)
239
139k
{
240
139k
    int i;
241
151k
    for (i = 0; i < n_segments; i++) {
242
12.0k
        av_freep(&segments[i]->key);
243
12.0k
        av_freep(&segments[i]->url);
244
12.0k
        av_freep(&segments[i]);
245
12.0k
    }
246
139k
}
247
248
static void free_segment_list(struct playlist *pls)
249
139k
{
250
139k
    free_segment_dynarray(pls->segments, pls->n_segments);
251
139k
    av_freep(&pls->segments);
252
139k
    pls->n_segments = 0;
253
139k
}
254
255
static void free_init_section_list(struct playlist *pls)
256
139k
{
257
139k
    int i;
258
143k
    for (i = 0; i < pls->n_init_sections; i++) {
259
3.42k
        av_freep(&pls->init_sections[i]->key);
260
3.42k
        av_freep(&pls->init_sections[i]->url);
261
3.42k
        av_freep(&pls->init_sections[i]);
262
3.42k
    }
263
139k
    av_freep(&pls->init_sections);
264
139k
    pls->n_init_sections = 0;
265
139k
}
266
267
static void free_playlist_list(HLSContext *c)
268
8.76k
{
269
8.76k
    int i;
270
148k
    for (i = 0; i < c->n_playlists; i++) {
271
139k
        struct playlist *pls = c->playlists[i];
272
139k
        free_segment_list(pls);
273
139k
        free_init_section_list(pls);
274
139k
        av_freep(&pls->main_streams);
275
139k
        av_freep(&pls->renditions);
276
139k
        av_freep(&pls->id3_buf);
277
139k
        av_dict_free(&pls->id3_initial);
278
139k
        ff_id3v2_free_extra_meta(&pls->id3_deferred_extra);
279
139k
        av_freep(&pls->init_sec_buf);
280
139k
        av_packet_free(&pls->pkt);
281
139k
        av_freep(&pls->pb.pub.buffer);
282
139k
        ff_format_io_close(c->ctx, &pls->input);
283
139k
        pls->input_read_done = 0;
284
139k
        ff_format_io_close(c->ctx, &pls->input_next);
285
139k
        pls->input_next_requested = 0;
286
139k
        if (pls->ctx) {
287
120k
            pls->ctx->pb = NULL;
288
120k
            avformat_close_input(&pls->ctx);
289
120k
        }
290
139k
        av_free(pls);
291
139k
    }
292
8.76k
    av_freep(&c->playlists);
293
8.76k
    c->n_playlists = 0;
294
8.76k
}
295
296
static void free_variant_list(HLSContext *c)
297
8.76k
{
298
8.76k
    int i;
299
145k
    for (i = 0; i < c->n_variants; i++) {
300
136k
        struct variant *var = c->variants[i];
301
136k
        av_freep(&var->playlists);
302
136k
        av_free(var);
303
136k
    }
304
8.76k
    av_freep(&c->variants);
305
8.76k
    c->n_variants = 0;
306
8.76k
}
307
308
static void free_rendition_list(HLSContext *c)
309
8.76k
{
310
8.76k
    int i;
311
25.9k
    for (i = 0; i < c->n_renditions; i++)
312
17.2k
        av_freep(&c->renditions[i]);
313
8.76k
    av_freep(&c->renditions);
314
8.76k
    c->n_renditions = 0;
315
8.76k
}
316
317
static struct playlist *new_playlist(HLSContext *c, const char *url,
318
                                     const char *base)
319
139k
{
320
139k
    struct playlist *pls = av_mallocz(sizeof(struct playlist));
321
139k
    if (!pls)
322
0
        return NULL;
323
139k
    pls->pkt = av_packet_alloc();
324
139k
    if (!pls->pkt) {
325
0
        av_free(pls);
326
0
        return NULL;
327
0
    }
328
139k
    ff_make_absolute_url(pls->url, sizeof(pls->url), base, url);
329
139k
    if (!pls->url[0]) {
330
33
        av_packet_free(&pls->pkt);
331
33
        av_free(pls);
332
33
        return NULL;
333
33
    }
334
139k
    pls->seek_timestamp = AV_NOPTS_VALUE;
335
336
139k
    pls->is_id3_timestamped = -1;
337
139k
    pls->id3_mpegts_timestamp = AV_NOPTS_VALUE;
338
339
139k
    dynarray_add(&c->playlists, &c->n_playlists, pls);
340
139k
    return pls;
341
139k
}
342
343
struct variant_info {
344
    char bandwidth[20];
345
    /* variant group ids: */
346
    char audio[MAX_FIELD_LEN];
347
    char video[MAX_FIELD_LEN];
348
    char subtitles[MAX_FIELD_LEN];
349
};
350
351
static struct variant *new_variant(HLSContext *c, struct variant_info *info,
352
                                   const char *url, const char *base)
353
136k
{
354
136k
    struct variant *var;
355
136k
    struct playlist *pls;
356
357
136k
    pls = new_playlist(c, url, base);
358
136k
    if (!pls)
359
33
        return NULL;
360
361
136k
    var = av_mallocz(sizeof(struct variant));
362
136k
    if (!var)
363
0
        return NULL;
364
365
136k
    if (info) {
366
136k
        var->bandwidth = atoi(info->bandwidth);
367
136k
        strcpy(var->audio_group, info->audio);
368
136k
        strcpy(var->video_group, info->video);
369
136k
        strcpy(var->subtitles_group, info->subtitles);
370
136k
    }
371
372
136k
    dynarray_add(&c->variants, &c->n_variants, var);
373
136k
    dynarray_add(&var->playlists, &var->n_playlists, pls);
374
136k
    return var;
375
136k
}
376
377
static void handle_variant_args(struct variant_info *info, const char *key,
378
                                int key_len, char **dest, int *dest_len)
379
10.1k
{
380
10.1k
    if (!strncmp(key, "BANDWIDTH=", key_len)) {
381
1.43k
        *dest     =        info->bandwidth;
382
1.43k
        *dest_len = sizeof(info->bandwidth);
383
8.71k
    } else if (!strncmp(key, "AUDIO=", key_len)) {
384
1.33k
        *dest     =        info->audio;
385
1.33k
        *dest_len = sizeof(info->audio);
386
7.37k
    } else if (!strncmp(key, "VIDEO=", key_len)) {
387
2.59k
        *dest     =        info->video;
388
2.59k
        *dest_len = sizeof(info->video);
389
4.78k
    } else if (!strncmp(key, "SUBTITLES=", key_len)) {
390
1.41k
        *dest     =        info->subtitles;
391
1.41k
        *dest_len = sizeof(info->subtitles);
392
1.41k
    }
393
10.1k
}
394
395
struct key_info {
396
     char uri[MAX_URL_SIZE];
397
     char method[11];
398
     char iv[35];
399
};
400
401
static void handle_key_args(struct key_info *info, const char *key,
402
                            int key_len, char **dest, int *dest_len)
403
24.0k
{
404
24.0k
    if (!strncmp(key, "METHOD=", key_len)) {
405
4.70k
        *dest     =        info->method;
406
4.70k
        *dest_len = sizeof(info->method);
407
19.3k
    } else if (!strncmp(key, "URI=", key_len)) {
408
9.06k
        *dest     =        info->uri;
409
9.06k
        *dest_len = sizeof(info->uri);
410
10.2k
    } else if (!strncmp(key, "IV=", key_len)) {
411
3.16k
        *dest     =        info->iv;
412
3.16k
        *dest_len = sizeof(info->iv);
413
3.16k
    }
414
24.0k
}
415
416
struct init_section_info {
417
    char uri[MAX_URL_SIZE];
418
    char byterange[32];
419
};
420
421
static struct segment *new_init_section(struct playlist *pls,
422
                                        struct init_section_info *info,
423
                                        const char *url_base)
424
3.48k
{
425
3.48k
    struct segment *sec;
426
3.48k
    char tmp_str[MAX_URL_SIZE], *ptr = tmp_str;
427
428
3.48k
    if (!info->uri[0])
429
61
        return NULL;
430
431
3.42k
    sec = av_mallocz(sizeof(*sec));
432
3.42k
    if (!sec)
433
0
        return NULL;
434
435
3.42k
    if (!av_strncasecmp(info->uri, "data:", 5)) {
436
479
        ptr = info->uri;
437
2.94k
    } else {
438
2.94k
        ff_make_absolute_url(tmp_str, sizeof(tmp_str), url_base, info->uri);
439
2.94k
        if (!tmp_str[0]) {
440
0
            av_free(sec);
441
0
            return NULL;
442
0
        }
443
2.94k
    }
444
3.42k
    sec->url = av_strdup(ptr);
445
3.42k
    if (!sec->url) {
446
0
        av_free(sec);
447
0
        return NULL;
448
0
    }
449
450
3.42k
    if (info->byterange[0]) {
451
677
        sec->size = strtoll(info->byterange, NULL, 10);
452
677
        ptr = strchr(info->byterange, '@');
453
677
        if (ptr)
454
444
            sec->url_offset = strtoll(ptr+1, NULL, 10);
455
2.74k
    } else {
456
        /* the entire file is the init section */
457
2.74k
        sec->size = -1;
458
2.74k
    }
459
460
3.42k
    dynarray_add(&pls->init_sections, &pls->n_init_sections, sec);
461
462
3.42k
    return sec;
463
3.42k
}
464
465
static void handle_init_section_args(void *context, const char *key,
466
                                     int key_len, char **dest, int *dest_len)
467
7.96k
{
468
7.96k
    struct init_section_info *info = context;
469
7.96k
    if (!strncmp(key, "URI=", key_len)) {
470
5.06k
        *dest     =        info->uri;
471
5.06k
        *dest_len = sizeof(info->uri);
472
5.06k
    } else if (!strncmp(key, "BYTERANGE=", key_len)) {
473
690
        *dest     =        info->byterange;
474
690
        *dest_len = sizeof(info->byterange);
475
690
    }
476
7.96k
}
477
478
struct rendition_info {
479
    char type[16];
480
    char uri[MAX_URL_SIZE];
481
    char group_id[MAX_FIELD_LEN];
482
    char language[MAX_FIELD_LEN];
483
    char assoc_language[MAX_FIELD_LEN];
484
    char name[MAX_FIELD_LEN];
485
    char defaultr[4];
486
    char forced[4];
487
    char characteristics[MAX_CHARACTERISTICS_LEN];
488
};
489
490
static struct rendition *new_rendition(HLSContext *c, struct rendition_info *info,
491
                                      const char *url_base)
492
25.5k
{
493
25.5k
    struct rendition *rend;
494
25.5k
    enum AVMediaType type = AVMEDIA_TYPE_UNKNOWN;
495
25.5k
    char *characteristic;
496
25.5k
    char *chr_ptr;
497
25.5k
    char *saveptr;
498
499
25.5k
    if (!strcmp(info->type, "AUDIO"))
500
11.9k
        type = AVMEDIA_TYPE_AUDIO;
501
13.6k
    else if (!strcmp(info->type, "VIDEO"))
502
4.48k
        type = AVMEDIA_TYPE_VIDEO;
503
9.20k
    else if (!strcmp(info->type, "SUBTITLES"))
504
1.20k
        type = AVMEDIA_TYPE_SUBTITLE;
505
8.00k
    else if (!strcmp(info->type, "CLOSED-CAPTIONS"))
506
        /* CLOSED-CAPTIONS is ignored since we do not support CEA-608 CC in
507
         * AVC SEI RBSP anyway */
508
1.57k
        return NULL;
509
510
24.0k
    if (type == AVMEDIA_TYPE_UNKNOWN) {
511
6.42k
        av_log(c->ctx, AV_LOG_WARNING, "Can't support the type: %s\n", info->type);
512
6.42k
        return NULL;
513
6.42k
    }
514
515
    /* URI is mandatory for subtitles as per spec */
516
17.5k
    if (type == AVMEDIA_TYPE_SUBTITLE && !info->uri[0]) {
517
352
        av_log(c->ctx, AV_LOG_ERROR, "The URI tag is REQUIRED for subtitle.\n");
518
352
        return NULL;
519
352
    }
520
521
17.2k
    rend = av_mallocz(sizeof(struct rendition));
522
17.2k
    if (!rend)
523
0
        return NULL;
524
525
17.2k
    dynarray_add(&c->renditions, &c->n_renditions, rend);
526
527
17.2k
    rend->type = type;
528
17.2k
    strcpy(rend->group_id, info->group_id);
529
17.2k
    strcpy(rend->language, info->language);
530
17.2k
    strcpy(rend->name, info->name);
531
532
    /* add the playlist if this is an external rendition */
533
17.2k
    if (info->uri[0]) {
534
3.01k
        rend->playlist = new_playlist(c, info->uri, url_base);
535
3.01k
        if (rend->playlist) {
536
3.01k
            if (type == AVMEDIA_TYPE_SUBTITLE) {
537
849
                rend->playlist->is_subtitle = 1;
538
849
                rend->playlist->is_id3_timestamped = 0;
539
849
            }
540
3.01k
            dynarray_add(&rend->playlist->renditions,
541
3.01k
                         &rend->playlist->n_renditions, rend);
542
3.01k
        }
543
3.01k
    }
544
545
17.2k
    if (info->assoc_language[0]) {
546
796
        size_t langlen = strlen(rend->language);
547
796
        if (langlen < sizeof(rend->language) - 3) {
548
517
            size_t assoc_len;
549
517
            rend->language[langlen] = ',';
550
517
            assoc_len = av_strlcpy(rend->language + langlen + 1,
551
517
                                   info->assoc_language,
552
517
                                   sizeof(rend->language) - langlen - 1);
553
517
            if (langlen + assoc_len + 2 > sizeof(rend->language)) // truncation occurred
554
211
                av_log(c->ctx, AV_LOG_WARNING, "Truncated rendition language: %s\n",
555
211
                       info->assoc_language);
556
517
        }
557
796
    }
558
559
17.2k
    if (!strcmp(info->defaultr, "YES"))
560
214
        rend->disposition |= AV_DISPOSITION_DEFAULT;
561
17.2k
    if (!strcmp(info->forced, "YES"))
562
481
        rend->disposition |= AV_DISPOSITION_FORCED;
563
564
17.2k
    chr_ptr = info->characteristics;
565
19.3k
    while ((characteristic = av_strtok(chr_ptr, ",", &saveptr))) {
566
2.13k
        if (!strcmp(characteristic, "public.accessibility.describes-music-and-sound"))
567
302
            rend->disposition |= AV_DISPOSITION_HEARING_IMPAIRED;
568
1.82k
        else if (!strcmp(characteristic, "public.accessibility.describes-video"))
569
1.11k
            rend->disposition |= AV_DISPOSITION_VISUAL_IMPAIRED;
570
571
2.13k
        chr_ptr = NULL;
572
2.13k
    }
573
574
17.2k
    return rend;
575
17.2k
}
576
577
static void handle_rendition_args(struct rendition_info *info, const char *key,
578
                                  int key_len, char **dest, int *dest_len)
579
38.3k
{
580
38.3k
    if (!strncmp(key, "TYPE=", key_len)) {
581
22.4k
        *dest     =        info->type;
582
22.4k
        *dest_len = sizeof(info->type);
583
22.4k
    } else if (!strncmp(key, "URI=", key_len)) {
584
3.69k
        *dest     =        info->uri;
585
3.69k
        *dest_len = sizeof(info->uri);
586
12.1k
    } else if (!strncmp(key, "GROUP-ID=", key_len)) {
587
1.23k
        *dest     =        info->group_id;
588
1.23k
        *dest_len = sizeof(info->group_id);
589
10.8k
    } else if (!strncmp(key, "LANGUAGE=", key_len)) {
590
671
        *dest     =        info->language;
591
671
        *dest_len = sizeof(info->language);
592
10.2k
    } else if (!strncmp(key, "ASSOC-LANGUAGE=", key_len)) {
593
1.22k
        *dest     =        info->assoc_language;
594
1.22k
        *dest_len = sizeof(info->assoc_language);
595
8.99k
    } else if (!strncmp(key, "NAME=", key_len)) {
596
228
        *dest     =        info->name;
597
228
        *dest_len = sizeof(info->name);
598
8.77k
    } else if (!strncmp(key, "DEFAULT=", key_len)) {
599
652
        *dest     =        info->defaultr;
600
652
        *dest_len = sizeof(info->defaultr);
601
8.11k
    } else if (!strncmp(key, "FORCED=", key_len)) {
602
777
        *dest     =        info->forced;
603
777
        *dest_len = sizeof(info->forced);
604
7.34k
    } else if (!strncmp(key, "CHARACTERISTICS=", key_len)) {
605
2.80k
        *dest     =        info->characteristics;
606
2.80k
        *dest_len = sizeof(info->characteristics);
607
2.80k
    }
608
    /*
609
     * ignored:
610
     * - AUTOSELECT: client may autoselect based on e.g. system language
611
     * - INSTREAM-ID: EIA-608 closed caption number ("CC1".."CC4")
612
     */
613
38.3k
}
614
615
/* used by parse_playlist to allocate a new variant+playlist when the
616
 * playlist is detected to be a Media Playlist (not Master Playlist)
617
 * and we have no parent Master Playlist (parsing of which would have
618
 * allocated the variant and playlist already)
619
 * *pls == NULL  => Master Playlist or parentless Media Playlist
620
 * *pls != NULL => parented Media Playlist, playlist+variant allocated */
621
static int ensure_playlist(HLSContext *c, struct playlist **pls, const char *url)
622
17.6k
{
623
17.6k
    if (*pls)
624
16.7k
        return 0;
625
900
    if (!new_variant(c, NULL, url, NULL))
626
33
        return AVERROR(ENOMEM);
627
867
    *pls = c->playlists[c->n_playlists - 1];
628
867
    return 0;
629
900
}
630
631
static int open_url_keepalive(AVFormatContext *s, AVIOContext **pb,
632
                              const char *url, AVDictionary **options)
633
0
{
634
#if !CONFIG_HTTP_PROTOCOL
635
0
    return AVERROR_PROTOCOL_NOT_FOUND;
636
#else
637
    int ret;
638
    URLContext *uc = ffio_geturlcontext(*pb);
639
0
    av_assert0(uc);
640
0
    (*pb)->eof_reached = 0;
641
0
    ret = ff_http_do_new_request2(uc, url, options);
642
0
    if (ret < 0) {
643
0
        ff_format_io_close(s, pb);
644
0
    }
645
0
    return ret;
646
#endif
647
0
}
Unexecuted instantiation: hls.c:open_url_keepalive
Unexecuted instantiation: hls.c:open_url_keepalive
648
649
static int open_url(AVFormatContext *s, AVIOContext **pb, const char *url,
650
                    AVDictionary **opts, AVDictionary *opts2, int *is_http_out)
651
578
{
652
578
    HLSContext *c = s->priv_data;
653
578
    AVDictionary *tmp = NULL;
654
578
    const char *proto_name = NULL;
655
578
    int ret;
656
578
    int is_http = 0;
657
658
578
    if (av_strstart(url, "crypto", NULL)) {
659
18
        if (url[6] == '+' || url[6] == ':')
660
12
            proto_name = avio_find_protocol_name(url + 7);
661
560
    } else if (av_strstart(url, "data", NULL)) {
662
22
        if (url[4] == '+' || url[4] == ':')
663
9
            proto_name = avio_find_protocol_name(url + 5);
664
22
    }
665
666
578
    if (!proto_name)
667
578
        proto_name = avio_find_protocol_name(url);
668
669
578
    if (!proto_name)
670
578
        return AVERROR_INVALIDDATA;
671
672
    // only http(s) & file are allowed
673
0
    if (av_strstart(proto_name, "file", NULL)) {
674
0
        if (strcmp(c->allowed_extensions, "ALL") && !av_match_ext(url, c->allowed_extensions)) {
675
0
            av_log(s, AV_LOG_ERROR,
676
0
                "Filename extension of \'%s\' is not a common multimedia extension, blocked for security reasons.\n"
677
0
                "If you wish to override this adjust allowed_extensions, you can set it to \'ALL\' to allow all\n",
678
0
                url);
679
0
            return AVERROR_INVALIDDATA;
680
0
        }
681
0
    } else if (av_strstart(proto_name, "http", NULL)) {
682
0
        is_http = 1;
683
0
    } else if (av_strstart(proto_name, "data", NULL)) {
684
0
        ;
685
0
    } else
686
0
        return AVERROR_INVALIDDATA;
687
688
0
    if (!strncmp(proto_name, url, strlen(proto_name)) && url[strlen(proto_name)] == ':')
689
0
        ;
690
0
    else if (av_strstart(url, "crypto", NULL) && !strncmp(proto_name, url + 7, strlen(proto_name)) && url[7 + strlen(proto_name)] == ':')
691
0
        ;
692
0
    else if (av_strstart(url, "data", NULL) && !strncmp(proto_name, url + 5, strlen(proto_name)) && url[5 + strlen(proto_name)] == ':')
693
0
        ;
694
0
    else if (strcmp(proto_name, "file") || !strncmp(url, "file,", 5))
695
0
        return AVERROR_INVALIDDATA;
696
697
0
    av_dict_copy(&tmp, *opts, 0);
698
0
    av_dict_copy(&tmp, opts2, 0);
699
700
0
    if (is_http && c->http_persistent && *pb) {
701
0
        ret = open_url_keepalive(c->ctx, pb, url, &tmp);
702
0
        if (ret == AVERROR_EXIT) {
703
0
            av_dict_free(&tmp);
704
0
            return ret;
705
0
        } else if (ret < 0) {
706
0
            if (ret != AVERROR_EOF)
707
0
                av_log(s, AV_LOG_WARNING,
708
0
                    "keepalive request failed for '%s' with error: '%s' when opening url, retrying with new connection\n",
709
0
                    url, av_err2str(ret));
710
0
            av_dict_copy(&tmp, *opts, 0);
711
0
            av_dict_copy(&tmp, opts2, 0);
712
0
            ret = s->io_open(s, pb, url, AVIO_FLAG_READ, &tmp);
713
0
        }
714
0
    } else {
715
0
        ret = s->io_open(s, pb, url, AVIO_FLAG_READ, &tmp);
716
0
    }
717
0
    if (ret >= 0) {
718
        // update cookies on http response with setcookies.
719
0
        char *new_cookies = NULL;
720
721
0
        if (!(s->flags & AVFMT_FLAG_CUSTOM_IO))
722
0
            av_opt_get(*pb, "cookies", AV_OPT_SEARCH_CHILDREN, (uint8_t**)&new_cookies);
723
724
0
        if (new_cookies)
725
0
            av_dict_set(opts, "cookies", new_cookies, AV_DICT_DONT_STRDUP_VAL);
726
0
    }
727
728
0
    av_dict_free(&tmp);
729
730
0
    if (is_http_out)
731
0
        *is_http_out = is_http;
732
733
0
    return ret;
734
0
}
735
736
static int test_segment(AVFormatContext *s, const AVInputFormat *in_fmt, struct playlist *pls, struct segment *seg)
737
12.1k
{
738
12.1k
    HLSContext *c = s->priv_data;
739
12.1k
    int matchA = 3;
740
12.1k
    int matchF = 0;
741
742
12.1k
    if (!c->extension_picky)
743
0
        return 0;
744
745
12.1k
    if (strcmp(c->allowed_segment_extensions, "ALL"))
746
12.1k
        matchA =      av_match_ext    (seg->url, c->allowed_segment_extensions)
747
12.1k
                 + 2*(ff_match_url_ext(seg->url, c->allowed_segment_extensions) > 0);
748
749
12.1k
    if (!matchA) {
750
111
        av_log(s, AV_LOG_ERROR, "URL %s is not in allowed_segment_extensions, consider updating hls.c and submitting a patch to ffmpeg-devel, if this should be added\n", seg->url);
751
111
        return AVERROR_INVALIDDATA;
752
111
    }
753
754
12.0k
    if (in_fmt) {
755
0
        if (in_fmt->extensions) {
756
0
            matchF =      av_match_ext(    seg->url, in_fmt->extensions)
757
0
                     + 2*(ff_match_url_ext(seg->url, in_fmt->extensions) > 0);
758
            // Youtube uses aac files with .ts extension
759
0
            if(av_match_name("mp4", in_fmt->name) || av_match_name("aac", in_fmt->name)) {
760
0
                matchF |=      av_match_ext(    seg->url, "ts,m2t,m2ts,mts,mpg,m4s,mpeg,mpegts,cmfv,cmfa")
761
0
                          + 2*(ff_match_url_ext(seg->url, "ts,m2t,m2ts,mts,mpg,m4s,mpeg,mpegts,cmfv,cmfa") > 0);
762
0
            }
763
0
        } else if (!strcmp(in_fmt->name, "mpegts")) {
764
0
            const char *str = "ts,m2t,m2ts,mts,mpg,m4s,mpeg,mpegts"
765
0
                              ",html" // https://flash1.bogulus.cfd/
766
0
                            ;
767
0
            matchF =      av_match_ext(    seg->url, str)
768
0
                     + 2*(ff_match_url_ext(seg->url, str) > 0);
769
0
        } else if (!strcmp(in_fmt->name, "webvtt")) {
770
0
            matchF =      av_match_ext(    seg->url, "vtt,webvtt")
771
0
                     + 2*(ff_match_url_ext(seg->url, "vtt,webvtt") > 0);
772
0
        }
773
774
0
        if (!(matchA & matchF)) {
775
0
            av_log(s, AV_LOG_ERROR, "detected format %s extension %s mismatches allowed extensions in url %s\n", in_fmt->name, in_fmt->extensions ? in_fmt->extensions : "none", seg->url);
776
0
            return AVERROR_INVALIDDATA;
777
0
        }
778
0
    }
779
780
12.0k
    return 0;
781
12.0k
}
782
783
static int parse_playlist(HLSContext *c, const char *url,
784
                          struct playlist *pls, AVIOContext *in)
785
146k
{
786
146k
    int ret = 0, is_segment = 0, is_variant = 0;
787
146k
    int64_t duration = 0;
788
146k
    enum KeyType key_type = KEY_NONE;
789
146k
    uint8_t iv[16] = "";
790
146k
    int has_iv = 0;
791
146k
    char key[MAX_URL_SIZE] = "";
792
146k
    char line[MAX_URL_SIZE];
793
146k
    const char *ptr;
794
146k
    int close_in = 0;
795
146k
    int64_t seg_offset = 0;
796
146k
    int64_t seg_size = -1;
797
146k
    uint8_t *new_url = NULL;
798
146k
    struct variant_info variant_info;
799
146k
    char tmp_str[MAX_URL_SIZE];
800
146k
    struct segment *cur_init_section = NULL;
801
146k
    int is_http = av_strstart(url, "http", NULL);
802
146k
    struct segment **prev_segments = NULL;
803
146k
    int prev_n_segments = 0;
804
146k
    int64_t prev_start_seq_no = -1;
805
806
146k
    if (is_http && !in && c->http_persistent && c->playlist_pb) {
807
0
        in = c->playlist_pb;
808
0
        ret = open_url_keepalive(c->ctx, &c->playlist_pb, url, NULL);
809
0
        if (ret == AVERROR_EXIT) {
810
0
            return ret;
811
0
        } else if (ret < 0) {
812
0
            if (ret != AVERROR_EOF)
813
0
                av_log(c->ctx, AV_LOG_WARNING,
814
0
                    "keepalive request failed for '%s' with error: '%s' when parsing playlist\n",
815
0
                    url, av_err2str(ret));
816
0
            in = NULL;
817
0
        }
818
0
    }
819
820
146k
    if (!in) {
821
137k
        AVDictionary *opts = NULL;
822
137k
        av_dict_copy(&opts, c->avio_opts, 0);
823
824
137k
        if (c->http_persistent)
825
137k
            av_dict_set(&opts, "multiple_requests", "1", 0);
826
827
137k
        ret = c->ctx->io_open(c->ctx, &in, url, AVIO_FLAG_READ, &opts);
828
137k
        av_dict_free(&opts);
829
137k
        if (ret < 0)
830
137k
            return ret;
831
832
0
        if (is_http && c->http_persistent)
833
0
            c->playlist_pb = in;
834
0
        else
835
0
            close_in = 1;
836
0
    }
837
838
8.76k
    if (av_opt_get(in, "location", AV_OPT_SEARCH_CHILDREN, &new_url) >= 0)
839
0
        url = new_url;
840
841
8.76k
    ff_get_chomp_line(in, line, sizeof(line));
842
8.76k
    if (strcmp(line, "#EXTM3U")) {
843
5.21k
        ret = AVERROR_INVALIDDATA;
844
5.21k
        goto fail;
845
5.21k
    }
846
847
3.54k
    if (pls) {
848
0
        prev_start_seq_no = pls->start_seq_no;
849
0
        prev_segments = pls->segments;
850
0
        prev_n_segments = pls->n_segments;
851
0
        pls->segments = NULL;
852
0
        pls->n_segments = 0;
853
854
0
        pls->finished = 0;
855
0
        pls->type = PLS_TYPE_UNSPECIFIED;
856
0
    }
857
1.41M
    while (!avio_feof(in)) {
858
1.41M
        ff_get_chomp_line(in, line, sizeof(line));
859
1.41M
        if (av_strstart(line, "#EXT-X-STREAM-INF:", &ptr)) {
860
136k
            is_variant = 1;
861
136k
            memset(&variant_info, 0, sizeof(variant_info));
862
136k
            ff_parse_key_value(ptr, (ff_parse_key_val_cb) handle_variant_args,
863
136k
                               &variant_info);
864
1.27M
        } else if (av_strstart(line, "#EXT-X-KEY:", &ptr)) {
865
8.24k
            struct key_info info = {{0}};
866
8.24k
            ff_parse_key_value(ptr, (ff_parse_key_val_cb) handle_key_args,
867
8.24k
                               &info);
868
8.24k
            key_type = KEY_NONE;
869
8.24k
            has_iv = 0;
870
8.24k
            if (!strcmp(info.method, "AES-128"))
871
217
                key_type = KEY_AES_128;
872
8.24k
            if (!strcmp(info.method, "SAMPLE-AES"))
873
2.61k
                key_type = KEY_SAMPLE_AES;
874
8.24k
            if (!av_strncasecmp(info.iv, "0x", 2)) {
875
2.67k
                ff_hex_to_data(iv, info.iv + 2);
876
2.67k
                has_iv = 1;
877
2.67k
            }
878
8.24k
            av_strlcpy(key, info.uri, sizeof(key));
879
1.26M
        } else if (av_strstart(line, "#EXT-X-MEDIA:", &ptr)) {
880
25.5k
            struct rendition_info info = {{0}};
881
25.5k
            ff_parse_key_value(ptr, (ff_parse_key_val_cb) handle_rendition_args,
882
25.5k
                               &info);
883
25.5k
            new_rendition(c, &info, url);
884
1.24M
        } else if (av_strstart(line, "#EXT-X-TARGETDURATION:", &ptr)) {
885
358
            int64_t t;
886
358
            ret = ensure_playlist(c, &pls, url);
887
358
            if (ret < 0)
888
1
                goto fail;
889
357
            t = strtoll(ptr, NULL, 10);
890
357
            if (t < 0 || t >= INT64_MAX / AV_TIME_BASE) {
891
67
                ret = AVERROR_INVALIDDATA;
892
67
                goto fail;
893
67
            }
894
290
            pls->target_duration = t * AV_TIME_BASE;
895
1.24M
        } else if (av_strstart(line, "#EXT-X-MEDIA-SEQUENCE:", &ptr)) {
896
1.38k
            uint64_t seq_no;
897
1.38k
            ret = ensure_playlist(c, &pls, url);
898
1.38k
            if (ret < 0)
899
2
                goto fail;
900
1.38k
            seq_no = strtoull(ptr, NULL, 10);
901
1.38k
            if (seq_no > INT64_MAX/2) {
902
847
                av_log(c->ctx, AV_LOG_DEBUG, "MEDIA-SEQUENCE higher than "
903
847
                        "INT64_MAX/2, mask out the highest bit\n");
904
847
                seq_no &= INT64_MAX/2;
905
847
            }
906
1.38k
            pls->start_seq_no = seq_no;
907
1.24M
        } else if (av_strstart(line, "#EXT-X-PLAYLIST-TYPE:", &ptr)) {
908
0
            ret = ensure_playlist(c, &pls, url);
909
0
            if (ret < 0)
910
0
                goto fail;
911
0
            if (!strcmp(ptr, "EVENT"))
912
0
                pls->type = PLS_TYPE_EVENT;
913
0
            else if (!strcmp(ptr, "VOD"))
914
0
                pls->type = PLS_TYPE_VOD;
915
1.24M
        } else if (av_strstart(line, "#EXT-X-MAP:", &ptr)) {
916
3.49k
            struct init_section_info info = {{0}};
917
3.49k
            ret = ensure_playlist(c, &pls, url);
918
3.49k
            if (ret < 0)
919
7
                goto fail;
920
3.48k
            ff_parse_key_value(ptr, handle_init_section_args, &info);
921
3.48k
            cur_init_section = new_init_section(pls, &info, url);
922
3.48k
            if (!cur_init_section) {
923
61
                ret = AVERROR(ENOMEM);
924
61
                goto fail;
925
61
            }
926
3.42k
            cur_init_section->key_type = key_type;
927
3.42k
            if (has_iv) {
928
389
                memcpy(cur_init_section->iv, iv, sizeof(iv));
929
3.03k
            } else {
930
3.03k
                int64_t seq = pls->start_seq_no + pls->n_segments;
931
3.03k
                memset(cur_init_section->iv, 0, sizeof(cur_init_section->iv));
932
3.03k
                AV_WB64(cur_init_section->iv + 8, seq);
933
3.03k
            }
934
935
3.42k
            if (key_type != KEY_NONE) {
936
962
                ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, key);
937
962
                if (!tmp_str[0]) {
938
0
                    av_free(cur_init_section);
939
0
                    ret = AVERROR_INVALIDDATA;
940
0
                    goto fail;
941
0
                }
942
962
                cur_init_section->key = av_strdup(tmp_str);
943
962
                if (!cur_init_section->key) {
944
0
                    av_free(cur_init_section);
945
0
                    ret = AVERROR(ENOMEM);
946
0
                    goto fail;
947
0
                }
948
2.46k
            } else {
949
2.46k
                cur_init_section->key = NULL;
950
2.46k
            }
951
952
1.23M
        } else if (av_strstart(line, "#EXT-X-START:", &ptr)) {
953
272
            const char *time_offset_value = NULL;
954
272
            ret = ensure_playlist(c, &pls, url);
955
272
            if (ret < 0) {
956
1
                goto fail;
957
1
            }
958
271
            if (av_strstart(ptr, "TIME-OFFSET=", &time_offset_value)) {
959
0
                float offset = strtof(time_offset_value, NULL);
960
0
                pls->start_time_offset = offset * AV_TIME_BASE;
961
0
                pls->time_offset_flag = 1;
962
271
            } else {
963
271
                av_log(c->ctx, AV_LOG_WARNING, "#EXT-X-START value is"
964
271
                                                "invalid, it will be ignored");
965
271
                continue;
966
271
            }
967
1.23M
        } else if (av_strstart(line, "#EXT-X-ENDLIST", &ptr)) {
968
0
            if (pls)
969
0
                pls->finished = 1;
970
1.23M
        } else if (av_strstart(line, "#EXTINF:", &ptr)) {
971
12.8k
            is_segment = 1;
972
12.8k
            duration   = atof(ptr) * AV_TIME_BASE;
973
1.22M
        } else if (av_strstart(line, "#EXT-X-BYTERANGE:", &ptr)) {
974
1.69k
            seg_size = strtoll(ptr, NULL, 10);
975
1.69k
            ptr = strchr(ptr, '@');
976
1.69k
            if (ptr)
977
549
                seg_offset = strtoll(ptr+1, NULL, 10);
978
1.22M
        } else if (av_strstart(line, "#", NULL)) {
979
56.2k
            av_log(c->ctx, AV_LOG_VERBOSE, "Skip ('%s')\n", line);
980
56.2k
            continue;
981
1.16M
        } else if (line[0]) {
982
357k
            if (is_variant) {
983
136k
                if (!new_variant(c, &variant_info, line, url)) {
984
0
                    ret = AVERROR(ENOMEM);
985
0
                    goto fail;
986
0
                }
987
136k
                is_variant = 0;
988
136k
            }
989
357k
            if (is_segment) {
990
12.1k
                struct segment *seg;
991
12.1k
                ret = ensure_playlist(c, &pls, url);
992
12.1k
                if (ret < 0)
993
22
                    goto fail;
994
12.1k
                seg = av_malloc(sizeof(struct segment));
995
12.1k
                if (!seg) {
996
0
                    ret = AVERROR(ENOMEM);
997
0
                    goto fail;
998
0
                }
999
12.1k
                if (has_iv) {
1000
2.13k
                    memcpy(seg->iv, iv, sizeof(iv));
1001
10.0k
                } else {
1002
10.0k
                    uint64_t seq = pls->start_seq_no + (uint64_t)pls->n_segments;
1003
10.0k
                    memset(seg->iv, 0, sizeof(seg->iv));
1004
10.0k
                    AV_WB64(seg->iv + 8, seq);
1005
10.0k
                }
1006
1007
12.1k
                if (key_type != KEY_NONE) {
1008
3.60k
                    ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, key);
1009
3.60k
                    if (!tmp_str[0]) {
1010
0
                        ret = AVERROR_INVALIDDATA;
1011
0
                        av_free(seg);
1012
0
                        goto fail;
1013
0
                    }
1014
3.60k
                    seg->key = av_strdup(tmp_str);
1015
3.60k
                    if (!seg->key) {
1016
0
                        av_free(seg);
1017
0
                        ret = AVERROR(ENOMEM);
1018
0
                        goto fail;
1019
0
                    }
1020
8.53k
                } else {
1021
8.53k
                    seg->key = NULL;
1022
8.53k
                }
1023
1024
12.1k
                ff_make_absolute_url(tmp_str, sizeof(tmp_str), url, line);
1025
12.1k
                if (!tmp_str[0]) {
1026
0
                    ret = AVERROR_INVALIDDATA;
1027
0
                    if (seg->key)
1028
0
                        av_free(seg->key);
1029
0
                    av_free(seg);
1030
0
                    goto fail;
1031
0
                }
1032
12.1k
                seg->url = av_strdup(tmp_str);
1033
12.1k
                if (!seg->url) {
1034
0
                    av_free(seg->key);
1035
0
                    av_free(seg);
1036
0
                    ret = AVERROR(ENOMEM);
1037
0
                    goto fail;
1038
0
                }
1039
1040
12.1k
                ret = test_segment(c->ctx, pls->ctx ? pls->ctx->iformat : NULL, pls, seg);
1041
12.1k
                if (ret < 0) {
1042
111
                    av_free(seg->url);
1043
111
                    av_free(seg->key);
1044
111
                    av_free(seg);
1045
111
                    goto fail;
1046
111
                }
1047
1048
12.0k
                if (duration < 0.001 * AV_TIME_BASE) {
1049
11.5k
                    av_log(c->ctx, AV_LOG_WARNING, "Cannot get correct #EXTINF value of segment %s,"
1050
11.5k
                                    " set to default value to 1ms.\n", seg->url);
1051
11.5k
                    duration = 0.001 * AV_TIME_BASE;
1052
11.5k
                }
1053
12.0k
                seg->duration = duration;
1054
12.0k
                seg->key_type = key_type;
1055
12.0k
                dynarray_add(&pls->segments, &pls->n_segments, seg);
1056
12.0k
                is_segment = 0;
1057
1058
12.0k
                seg->size = seg_size;
1059
12.0k
                if (seg_size >= 0) {
1060
529
                    seg->url_offset = seg_offset;
1061
529
                    seg_offset += seg_size;
1062
529
                    seg_size = -1;
1063
11.5k
                } else {
1064
11.5k
                    seg->url_offset = 0;
1065
11.5k
                    seg_offset = 0;
1066
11.5k
                }
1067
1068
12.0k
                seg->init_section = cur_init_section;
1069
12.0k
            }
1070
357k
        }
1071
1.41M
    }
1072
3.27k
    if (prev_segments) {
1073
0
        if (pls->start_seq_no > prev_start_seq_no && c->first_timestamp != AV_NOPTS_VALUE) {
1074
0
            int64_t prev_timestamp = c->first_timestamp;
1075
0
            int i;
1076
0
            int64_t diff = pls->start_seq_no - prev_start_seq_no;
1077
0
            for (i = 0; i < prev_n_segments && i < diff; i++) {
1078
0
                c->first_timestamp += prev_segments[i]->duration;
1079
0
            }
1080
0
            av_log(c->ctx, AV_LOG_DEBUG, "Media sequence change (%"PRId64" -> %"PRId64")"
1081
0
                   " reflected in first_timestamp: %"PRId64" -> %"PRId64"\n",
1082
0
                   prev_start_seq_no, pls->start_seq_no,
1083
0
                   prev_timestamp, c->first_timestamp);
1084
0
        } else if (pls->start_seq_no < prev_start_seq_no) {
1085
0
            av_log(c->ctx, AV_LOG_WARNING, "Media sequence changed unexpectedly: %"PRId64" -> %"PRId64"\n",
1086
0
                   prev_start_seq_no, pls->start_seq_no);
1087
0
        }
1088
0
        free_segment_dynarray(prev_segments, prev_n_segments);
1089
0
        av_freep(&prev_segments);
1090
0
    }
1091
3.27k
    if (pls)
1092
628
        pls->last_load_time = av_gettime_relative();
1093
1094
8.76k
fail:
1095
8.76k
    av_free(new_url);
1096
8.76k
    if (close_in)
1097
0
        ff_format_io_close(c->ctx, &in);
1098
8.76k
    c->ctx->ctx_flags = c->ctx->ctx_flags & ~(unsigned)AVFMTCTX_UNSEEKABLE;
1099
8.76k
    if (!c->n_variants || !c->variants[0]->n_playlists ||
1100
8.76k
        !(c->variants[0]->playlists[0]->finished ||
1101
1.92k
          c->variants[0]->playlists[0]->type == PLS_TYPE_EVENT))
1102
8.76k
        c->ctx->ctx_flags |= AVFMTCTX_UNSEEKABLE;
1103
8.76k
    return ret;
1104
3.27k
}
1105
1106
static struct segment *current_segment(struct playlist *pls)
1107
1.40k
{
1108
1.40k
    int64_t n = pls->cur_seq_no - pls->start_seq_no;
1109
1.40k
    if (n >= pls->n_segments)
1110
0
        return NULL;
1111
1.40k
    return pls->segments[n];
1112
1.40k
}
1113
1114
static struct segment *next_segment(struct playlist *pls)
1115
0
{
1116
0
    int64_t n = pls->cur_seq_no - pls->start_seq_no + 1;
1117
0
    if (n >= pls->n_segments)
1118
0
        return NULL;
1119
0
    return pls->segments[n];
1120
0
}
1121
1122
static int read_from_url(struct playlist *pls, struct segment *seg,
1123
                         uint8_t *buf, int buf_size)
1124
0
{
1125
0
    int ret;
1126
1127
     /* limit read if the segment was only a part of a file */
1128
0
    if (seg->size >= 0)
1129
0
        buf_size = FFMIN(buf_size, seg->size - pls->cur_seg_offset);
1130
1131
0
    ret = avio_read(pls->input, buf, buf_size);
1132
0
    if (ret > 0)
1133
0
        pls->cur_seg_offset += ret;
1134
1135
0
    return ret;
1136
0
}
1137
1138
/* Parse the raw ID3 data and pass contents to caller */
1139
static void parse_id3(AVFormatContext *s, AVIOContext *pb,
1140
                      AVDictionary **metadata, int64_t *dts, HLSAudioSetupInfo *audio_setup_info,
1141
                      ID3v2ExtraMetaAPIC **apic, ID3v2ExtraMeta **extra_meta)
1142
0
{
1143
0
    static const char id3_priv_owner_ts[] = "com.apple.streaming.transportStreamTimestamp";
1144
0
    static const char id3_priv_owner_audio_setup[] = "com.apple.streaming.audioDescription";
1145
0
    ID3v2ExtraMeta *meta;
1146
1147
0
    ff_id3v2_read_dict(pb, metadata, ID3v2_DEFAULT_MAGIC, extra_meta);
1148
0
    for (meta = *extra_meta; meta; meta = meta->next) {
1149
0
        if (!strcmp(meta->tag, "PRIV")) {
1150
0
            ID3v2ExtraMetaPRIV *priv = &meta->data.priv;
1151
0
            if (priv->datasize == 8 && !av_strncasecmp(priv->owner, id3_priv_owner_ts, 44)) {
1152
                /* 33-bit MPEG timestamp */
1153
0
                int64_t ts = AV_RB64(priv->data);
1154
0
                av_log(s, AV_LOG_DEBUG, "HLS ID3 audio timestamp %"PRId64"\n", ts);
1155
0
                if ((ts & ~((1ULL << 33) - 1)) == 0)
1156
0
                    *dts = ts;
1157
0
                else
1158
0
                    av_log(s, AV_LOG_ERROR, "Invalid HLS ID3 audio timestamp %"PRId64"\n", ts);
1159
0
            } else if (priv->datasize >= 8 && !av_strncasecmp(priv->owner, id3_priv_owner_audio_setup, 36)) {
1160
0
                ff_hls_senc_read_audio_setup_info(audio_setup_info, priv->data, priv->datasize);
1161
0
            }
1162
0
        } else if (!strcmp(meta->tag, "APIC") && apic)
1163
0
            *apic = &meta->data.apic;
1164
0
    }
1165
0
}
1166
1167
/* Check if the ID3 metadata contents have changed */
1168
static int id3_has_changed_values(struct playlist *pls, AVDictionary *metadata,
1169
                                  ID3v2ExtraMetaAPIC *apic)
1170
0
{
1171
0
    const AVDictionaryEntry *entry = NULL;
1172
0
    const AVDictionaryEntry *oldentry;
1173
    /* check that no keys have changed values */
1174
0
    while ((entry = av_dict_iterate(metadata, entry))) {
1175
0
        oldentry = av_dict_get(pls->id3_initial, entry->key, NULL, AV_DICT_MATCH_CASE);
1176
0
        if (!oldentry || strcmp(oldentry->value, entry->value) != 0)
1177
0
            return 1;
1178
0
    }
1179
1180
    /* check if apic appeared */
1181
0
    if (apic && (pls->ctx->nb_streams != 2 || !pls->ctx->streams[1]->attached_pic.data))
1182
0
        return 1;
1183
1184
0
    if (apic) {
1185
0
        int size = pls->ctx->streams[1]->attached_pic.size;
1186
0
        if (size != apic->buf->size - AV_INPUT_BUFFER_PADDING_SIZE)
1187
0
            return 1;
1188
1189
0
        if (memcmp(apic->buf->data, pls->ctx->streams[1]->attached_pic.data, size) != 0)
1190
0
            return 1;
1191
0
    }
1192
1193
0
    return 0;
1194
0
}
1195
1196
/* Parse ID3 data and handle the found data */
1197
static void handle_id3(AVIOContext *pb, struct playlist *pls)
1198
0
{
1199
0
    AVDictionary *metadata = NULL;
1200
0
    ID3v2ExtraMetaAPIC *apic = NULL;
1201
0
    ID3v2ExtraMeta *extra_meta = NULL;
1202
0
    int64_t timestamp = AV_NOPTS_VALUE;
1203
1204
0
    parse_id3(pls->ctx, pb, &metadata, &timestamp, &pls->audio_setup_info, &apic, &extra_meta);
1205
1206
0
    if (timestamp != AV_NOPTS_VALUE) {
1207
0
        pls->id3_mpegts_timestamp = timestamp;
1208
0
        pls->id3_offset = 0;
1209
0
    }
1210
1211
0
    if (!pls->id3_found) {
1212
        /* initial ID3 tags */
1213
0
        av_assert0(!pls->id3_deferred_extra);
1214
0
        pls->id3_found = 1;
1215
1216
        /* get picture attachment and set text metadata */
1217
0
        if (pls->ctx->nb_streams)
1218
0
            ff_id3v2_parse_apic(pls->ctx, extra_meta);
1219
0
        else
1220
            /* demuxer not yet opened, defer picture attachment */
1221
0
            pls->id3_deferred_extra = extra_meta;
1222
1223
0
        ff_id3v2_parse_priv_dict(&metadata, extra_meta);
1224
0
        av_dict_copy(&pls->ctx->metadata, metadata, 0);
1225
0
        pls->id3_initial = metadata;
1226
1227
0
    } else {
1228
0
        if (!pls->id3_changed && id3_has_changed_values(pls, metadata, apic)) {
1229
0
            avpriv_report_missing_feature(pls->parent, "Changing ID3 metadata in HLS audio elementary stream");
1230
0
            pls->id3_changed = 1;
1231
0
        }
1232
0
        av_dict_free(&metadata);
1233
0
    }
1234
1235
0
    if (!pls->id3_deferred_extra)
1236
0
        ff_id3v2_free_extra_meta(&extra_meta);
1237
0
}
1238
1239
static void intercept_id3(struct playlist *pls, uint8_t *buf,
1240
                         int buf_size, int *len)
1241
0
{
1242
    /* intercept id3 tags, we do not want to pass them to the raw
1243
     * demuxer on all segment switches */
1244
0
    int bytes;
1245
0
    int id3_buf_pos = 0;
1246
0
    int fill_buf = 0;
1247
0
    struct segment *seg = current_segment(pls);
1248
1249
    /* gather all the id3 tags */
1250
0
    while (1) {
1251
        /* see if we can retrieve enough data for ID3 header */
1252
0
        if (*len < ID3v2_HEADER_SIZE && buf_size >= ID3v2_HEADER_SIZE) {
1253
0
            bytes = read_from_url(pls, seg, buf + *len, ID3v2_HEADER_SIZE - *len);
1254
0
            if (bytes > 0) {
1255
1256
0
                if (bytes == ID3v2_HEADER_SIZE - *len)
1257
                    /* no EOF yet, so fill the caller buffer again after
1258
                     * we have stripped the ID3 tags */
1259
0
                    fill_buf = 1;
1260
1261
0
                *len += bytes;
1262
1263
0
            } else if (*len <= 0) {
1264
                /* error/EOF */
1265
0
                *len = bytes;
1266
0
                fill_buf = 0;
1267
0
            }
1268
0
        }
1269
1270
0
        if (*len < ID3v2_HEADER_SIZE)
1271
0
            break;
1272
1273
0
        if (ff_id3v2_match(buf, ID3v2_DEFAULT_MAGIC)) {
1274
0
            int64_t maxsize = seg->size >= 0 ? seg->size : 1024*1024;
1275
0
            int taglen = ff_id3v2_tag_len(buf);
1276
0
            int tag_got_bytes = FFMIN(taglen, *len);
1277
0
            int remaining = taglen - tag_got_bytes;
1278
1279
0
            if (taglen > maxsize) {
1280
0
                av_log(pls->parent, AV_LOG_ERROR, "Too large HLS ID3 tag (%d > %"PRId64" bytes)\n",
1281
0
                       taglen, maxsize);
1282
0
                break;
1283
0
            }
1284
1285
            /*
1286
             * Copy the id3 tag to our temporary id3 buffer.
1287
             * We could read a small id3 tag directly without memcpy, but
1288
             * we would still need to copy the large tags, and handling
1289
             * both of those cases together with the possibility for multiple
1290
             * tags would make the handling a bit complex.
1291
             */
1292
0
            pls->id3_buf = av_fast_realloc(pls->id3_buf, &pls->id3_buf_size, id3_buf_pos + taglen);
1293
0
            if (!pls->id3_buf)
1294
0
                break;
1295
0
            memcpy(pls->id3_buf + id3_buf_pos, buf, tag_got_bytes);
1296
0
            id3_buf_pos += tag_got_bytes;
1297
1298
            /* strip the intercepted bytes */
1299
0
            *len -= tag_got_bytes;
1300
0
            memmove(buf, buf + tag_got_bytes, *len);
1301
0
            av_log(pls->parent, AV_LOG_DEBUG, "Stripped %d HLS ID3 bytes\n", tag_got_bytes);
1302
1303
0
            if (remaining > 0) {
1304
                /* read the rest of the tag in */
1305
0
                if (read_from_url(pls, seg, pls->id3_buf + id3_buf_pos, remaining) != remaining)
1306
0
                    break;
1307
0
                id3_buf_pos += remaining;
1308
0
                av_log(pls->parent, AV_LOG_DEBUG, "Stripped additional %d HLS ID3 bytes\n", remaining);
1309
0
            }
1310
1311
0
        } else {
1312
            /* no more ID3 tags */
1313
0
            break;
1314
0
        }
1315
0
    }
1316
1317
    /* re-fill buffer for the caller unless EOF */
1318
0
    if (*len >= 0 && (fill_buf || *len == 0)) {
1319
0
        bytes = read_from_url(pls, seg, buf + *len, buf_size - *len);
1320
1321
        /* ignore error if we already had some data */
1322
0
        if (bytes >= 0)
1323
0
            *len += bytes;
1324
0
        else if (*len == 0)
1325
0
            *len = bytes;
1326
0
    }
1327
1328
0
    if (pls->id3_buf) {
1329
        /* Now parse all the ID3 tags */
1330
0
        FFIOContext id3ioctx;
1331
0
        ffio_init_read_context(&id3ioctx, pls->id3_buf, id3_buf_pos);
1332
0
        handle_id3(&id3ioctx.pub, pls);
1333
0
    }
1334
1335
0
    if (pls->is_id3_timestamped == -1)
1336
0
        pls->is_id3_timestamped = (pls->id3_mpegts_timestamp != AV_NOPTS_VALUE);
1337
0
}
1338
1339
static int read_key(HLSContext *c, struct playlist *pls, struct segment *seg)
1340
42
{
1341
42
    AVIOContext *pb = NULL;
1342
1343
42
    int ret = open_url(pls->parent, &pb, seg->key, &c->avio_opts, NULL, NULL);
1344
42
    if (ret < 0) {
1345
42
        av_log(pls->parent, AV_LOG_ERROR, "Unable to open key file %s, %s\n",
1346
42
               seg->key, av_err2str(ret));
1347
42
        return ret;
1348
42
    }
1349
1350
0
    ret = avio_read(pb, pls->key, sizeof(pls->key));
1351
0
    ff_format_io_close(pls->parent, &pb);
1352
0
    if (ret != sizeof(pls->key)) {
1353
0
        if (ret < 0) {
1354
0
            av_log(pls->parent, AV_LOG_ERROR, "Unable to read key file %s, %s\n",
1355
0
                   seg->key, av_err2str(ret));
1356
0
        } else {
1357
0
            av_log(pls->parent, AV_LOG_ERROR, "Unable to read key file %s, read bytes %d != %zu\n",
1358
0
                   seg->key, ret, sizeof(pls->key));
1359
0
            ret = AVERROR_INVALIDDATA;
1360
0
        }
1361
1362
0
        return ret;
1363
0
    }
1364
1365
0
    av_strlcpy(pls->key_url, seg->key, sizeof(pls->key_url));
1366
1367
0
    return 0;
1368
0
}
1369
1370
static int open_input(HLSContext *c, struct playlist *pls, struct segment *seg, AVIOContext **in)
1371
578
{
1372
578
    AVDictionary *opts = NULL;
1373
578
    int ret;
1374
578
    int is_http = 0;
1375
1376
578
    if (c->http_persistent)
1377
578
        av_dict_set(&opts, "multiple_requests", "1", 0);
1378
1379
578
    if (seg->size >= 0) {
1380
        /* try to restrict the HTTP request to the part we want
1381
         * (if this is in fact a HTTP request) */
1382
68
        av_dict_set_int(&opts, "offset", seg->url_offset, 0);
1383
68
        av_dict_set_int(&opts, "end_offset", seg->url_offset + seg->size, 0);
1384
68
    }
1385
1386
578
    av_log(pls->parent, AV_LOG_VERBOSE, "HLS request for url '%s', offset %"PRId64", playlist %d\n",
1387
578
           seg->url, seg->url_offset, pls->index);
1388
1389
578
    if (seg->key_type == KEY_AES_128 || seg->key_type == KEY_SAMPLE_AES) {
1390
42
        if (strcmp(seg->key, pls->key_url)) {
1391
42
            ret = read_key(c, pls, seg);
1392
42
            if (ret < 0)
1393
42
                goto cleanup;
1394
42
        }
1395
42
    }
1396
1397
536
    if (seg->key_type == KEY_AES_128) {
1398
0
        char iv[33], key[33], url[MAX_URL_SIZE];
1399
0
        ff_data_to_hex(iv, seg->iv, sizeof(seg->iv), 0);
1400
0
        ff_data_to_hex(key, pls->key, sizeof(pls->key), 0);
1401
0
        if (strstr(seg->url, "://"))
1402
0
            snprintf(url, sizeof(url), "crypto+%s", seg->url);
1403
0
        else
1404
0
            snprintf(url, sizeof(url), "crypto:%s", seg->url);
1405
1406
0
        av_dict_set(&opts, "key", key, 0);
1407
0
        av_dict_set(&opts, "iv", iv, 0);
1408
1409
0
        ret = open_url(pls->parent, in, url, &c->avio_opts, opts, &is_http);
1410
0
        if (ret < 0) {
1411
0
            goto cleanup;
1412
0
        }
1413
0
        ret = 0;
1414
536
    } else {
1415
536
        ret = open_url(pls->parent, in, seg->url, &c->avio_opts, opts, &is_http);
1416
536
    }
1417
1418
    /* Seek to the requested position. If this was a HTTP request, the offset
1419
     * should already be where want it to, but this allows e.g. local testing
1420
     * without a HTTP server.
1421
     *
1422
     * This is not done for HTTP at all as avio_seek() does internal bookkeeping
1423
     * of file offset which is out-of-sync with the actual offset when "offset"
1424
     * AVOption is used with http protocol, causing the seek to not be a no-op
1425
     * as would be expected. Wrong offset received from the server will not be
1426
     * noticed without the call, though.
1427
     */
1428
536
    if (ret == 0 && !is_http && seg->url_offset) {
1429
0
        int64_t seekret = avio_seek(*in, seg->url_offset, SEEK_SET);
1430
0
        if (seekret < 0) {
1431
0
            av_log(pls->parent, AV_LOG_ERROR, "Unable to seek to offset %"PRId64" of HLS segment '%s'\n", seg->url_offset, seg->url);
1432
0
            ret = seekret;
1433
0
            ff_format_io_close(pls->parent, in);
1434
0
        }
1435
0
    }
1436
1437
578
cleanup:
1438
578
    av_dict_free(&opts);
1439
578
    pls->cur_seg_offset = 0;
1440
578
    return ret;
1441
536
}
1442
1443
static int update_init_section(struct playlist *pls, struct segment *seg)
1444
578
{
1445
578
    static const int max_init_section_size = 1024*1024;
1446
578
    HLSContext *c = pls->parent->priv_data;
1447
578
    int64_t sec_size;
1448
578
    int64_t urlsize;
1449
578
    int ret;
1450
1451
578
    if (seg->init_section == pls->cur_init_section)
1452
548
        return 0;
1453
1454
30
    pls->cur_init_section = NULL;
1455
1456
30
    if (!seg->init_section)
1457
0
        return 0;
1458
1459
30
    ret = open_input(c, pls, seg->init_section, &pls->input);
1460
30
    if (ret < 0) {
1461
30
        av_log(pls->parent, AV_LOG_WARNING,
1462
30
               "Failed to open an initialization section in playlist %d\n",
1463
30
               pls->index);
1464
30
        return ret;
1465
30
    }
1466
1467
0
    if (seg->init_section->size >= 0)
1468
0
        sec_size = seg->init_section->size;
1469
0
    else if ((urlsize = avio_size(pls->input)) >= 0)
1470
0
        sec_size = urlsize;
1471
0
    else
1472
0
        sec_size = max_init_section_size;
1473
1474
0
    av_log(pls->parent, AV_LOG_DEBUG,
1475
0
           "Downloading an initialization section of size %"PRId64"\n",
1476
0
           sec_size);
1477
1478
0
    sec_size = FFMIN(sec_size, max_init_section_size);
1479
1480
0
    av_fast_malloc(&pls->init_sec_buf, &pls->init_sec_buf_size, sec_size);
1481
1482
0
    ret = read_from_url(pls, seg->init_section, pls->init_sec_buf,
1483
0
                        pls->init_sec_buf_size);
1484
0
    ff_format_io_close(pls->parent, &pls->input);
1485
1486
0
    if (ret < 0)
1487
0
        return ret;
1488
1489
0
    pls->cur_init_section = seg->init_section;
1490
0
    pls->init_sec_data_len = ret;
1491
0
    pls->init_sec_buf_read_offset = 0;
1492
1493
    /* spec says audio elementary streams do not have media initialization
1494
     * sections, so there should be no ID3 timestamps */
1495
0
    pls->is_id3_timestamped = 0;
1496
1497
0
    return 0;
1498
0
}
1499
1500
static int64_t default_reload_interval(struct playlist *pls)
1501
890
{
1502
890
    return pls->n_segments > 0 ?
1503
890
                          pls->segments[pls->n_segments - 1]->duration :
1504
890
                          pls->target_duration;
1505
890
}
1506
1507
static int playlist_needed(struct playlist *pls)
1508
90.1k
{
1509
90.1k
    AVFormatContext *s = pls->parent;
1510
90.1k
    int i, j;
1511
90.1k
    int stream_needed = 0;
1512
90.1k
    int first_st;
1513
1514
    /* If there is no context or streams yet, the playlist is needed */
1515
90.1k
    if ((!pls->ctx || !pls->n_main_streams) && !pls->is_subtitle)
1516
89.5k
        return 1;
1517
1518
    /* check if any of the streams in the playlist are needed */
1519
550
    for (i = 0; i < pls->n_main_streams; i++) {
1520
0
        if (pls->main_streams[i]->discard < AVDISCARD_ALL) {
1521
0
            stream_needed = 1;
1522
0
            break;
1523
0
        }
1524
0
    }
1525
1526
    /* If all streams in the playlist were discarded, the playlist is not
1527
     * needed (regardless of whether whole programs are discarded or not). */
1528
550
    if (!stream_needed)
1529
550
        return 0;
1530
1531
    /* Otherwise, check if all the programs (variants) this playlist is in are
1532
     * discarded. Since all streams in the playlist are part of the same programs
1533
     * we can just check the programs of the first stream. */
1534
1535
0
    first_st = pls->main_streams[0]->index;
1536
1537
0
    for (i = 0; i < s->nb_programs; i++) {
1538
0
        AVProgram *program = s->programs[i];
1539
0
        if (program->discard < AVDISCARD_ALL) {
1540
0
            for (j = 0; j < program->nb_stream_indexes; j++) {
1541
0
                if (program->stream_index[j] == first_st) {
1542
                    /* playlist is in an undiscarded program */
1543
0
                    return 1;
1544
0
                }
1545
0
            }
1546
0
        }
1547
0
    }
1548
1549
    /* some streams were not discarded but all the programs were */
1550
0
    return 0;
1551
0
}
1552
1553
static int reload_playlist(struct playlist *v, HLSContext *c)
1554
890
{
1555
890
    int ret = 0;
1556
890
    int reload_count = 0;
1557
1558
890
    v->needed = playlist_needed(v);
1559
1560
890
    if (!v->needed)
1561
0
        return AVERROR_EOF;
1562
1563
890
    if (!v->input || (c->http_persistent && v->input_read_done)) {
1564
890
        int64_t reload_interval;
1565
1566
        /* Check that the playlist is still needed before opening a new
1567
         * segment. */
1568
890
        v->needed = playlist_needed(v);
1569
1570
890
        if (!v->needed) {
1571
0
            av_log(v->parent, AV_LOG_INFO, "No longer receiving playlist %d ('%s')\n",
1572
0
                   v->index, v->url);
1573
0
            return AVERROR_EOF;
1574
0
        }
1575
1576
        /* If this is a live stream and the reload interval has elapsed since
1577
         * the last playlist reload, reload the playlists now. */
1578
890
        reload_interval = default_reload_interval(v);
1579
1580
1.14k
reload:
1581
1.14k
        reload_count++;
1582
1.14k
        if (reload_count > c->max_reload)
1583
0
            return AVERROR_EOF;
1584
1.14k
        if (!v->finished &&
1585
1.14k
            av_gettime_relative() - v->last_load_time >= reload_interval) {
1586
257
            if ((ret = parse_playlist(c, v->url, v, NULL)) < 0) {
1587
257
                if (ret != AVERROR_EXIT)
1588
257
                    av_log(v->parent, AV_LOG_WARNING, "Failed to reload playlist %d\n",
1589
257
                           v->index);
1590
257
                return ret;
1591
257
            }
1592
            /* If we need to reload the playlist again below (if
1593
             * there's still no more segments), switch to a reload
1594
             * interval of half the target duration. */
1595
0
            reload_interval = v->target_duration / 2;
1596
0
        }
1597
884
        if (v->cur_seq_no < v->start_seq_no) {
1598
0
            av_log(v->parent, AV_LOG_WARNING,
1599
0
                   "skipping %"PRId64" segments ahead, expired from playlists\n",
1600
0
                   v->start_seq_no - v->cur_seq_no);
1601
0
            v->cur_seq_no = v->start_seq_no;
1602
0
        }
1603
884
        if (v->cur_seq_no > v->last_seq_no) {
1604
688
            v->last_seq_no = v->cur_seq_no;
1605
688
            v->m3u8_hold_counters = 0;
1606
688
        } else if (v->last_seq_no == v->cur_seq_no) {
1607
196
            v->m3u8_hold_counters++;
1608
196
            if (v->m3u8_hold_counters >= c->m3u8_hold_counters) {
1609
0
                return AVERROR_EOF;
1610
0
            }
1611
196
        } else {
1612
0
            av_log(v->parent, AV_LOG_WARNING, "The m3u8 list sequence may have been wrapped.\n");
1613
0
        }
1614
884
        if (v->cur_seq_no >= v->start_seq_no + v->n_segments) {
1615
306
            if (v->finished || v->is_subtitle)
1616
0
                return AVERROR_EOF;
1617
1.07k
            while (av_gettime_relative() - v->last_load_time < reload_interval) {
1618
824
                if (ff_check_interrupt(c->interrupt_callback))
1619
55
                    return AVERROR_EXIT;
1620
769
                av_usleep(100*1000);
1621
769
            }
1622
            /* Enough time has elapsed since the last reload */
1623
251
            goto reload;
1624
306
        }
1625
1626
884
    }
1627
578
    return ret;
1628
890
}
1629
1630
static int read_data_continuous(void *opaque, uint8_t *buf, int buf_size)
1631
415
{
1632
415
    struct playlist *v = opaque;
1633
415
    HLSContext *c = v->parent->priv_data;
1634
415
    int ret;
1635
415
    int just_opened = 0;
1636
415
    int segment_retries = 0;
1637
415
    struct segment *seg;
1638
1639
415
    if (c->http_persistent && v->input_read_done) {
1640
0
        ret = reload_playlist(v, c);
1641
0
        if (ret < 0)
1642
0
            return ret;
1643
0
    }
1644
1645
415
    v->input_read_done = 0;
1646
1647
890
restart:
1648
890
    ret = reload_playlist(v, c);
1649
890
    if (ret < 0)
1650
312
        return ret;
1651
1652
578
    seg = current_segment(v);
1653
1654
578
    if (!v->input || (c->http_persistent && v->input_read_done)) {
1655
        /* load/update Media Initialization Section, if any */
1656
578
        ret = update_init_section(v, seg);
1657
578
        if (ret)
1658
30
            return ret;
1659
1660
548
        if (c->http_multiple == 1 && v->input_next_requested) {
1661
0
            FFSWAP(AVIOContext *, v->input, v->input_next);
1662
0
            v->cur_seg_offset = 0;
1663
0
            v->input_next_requested = 0;
1664
0
            ret = 0;
1665
548
        } else {
1666
548
            ret = open_input(c, v, seg, &v->input);
1667
548
        }
1668
548
        if (ret < 0) {
1669
548
            if (ff_check_interrupt(c->interrupt_callback))
1670
73
                return AVERROR_EXIT;
1671
475
            av_log(v->parent, AV_LOG_WARNING, "Failed to open segment %"PRId64" of playlist %d\n",
1672
475
                   v->cur_seq_no,
1673
475
                   v->index);
1674
475
            if (segment_retries >= c->seg_max_retry) {
1675
475
                av_log(v->parent, AV_LOG_WARNING, "Segment %"PRId64" of playlist %d failed too many times, skipping\n",
1676
475
                       v->cur_seq_no,
1677
475
                       v->index);
1678
475
                v->cur_seq_no++;
1679
475
                segment_retries = 0;
1680
475
            } else {
1681
0
                segment_retries++;
1682
0
            }
1683
475
            goto restart;
1684
548
        }
1685
0
        segment_retries = 0;
1686
0
        just_opened = 1;
1687
0
    }
1688
1689
0
    if (c->http_multiple == -1) {
1690
0
        uint8_t *http_version_opt = NULL;
1691
0
        int r = av_opt_get(v->input, "http_version", AV_OPT_SEARCH_CHILDREN, &http_version_opt);
1692
0
        if (r >= 0) {
1693
0
            c->http_multiple = (!strncmp((const char *)http_version_opt, "1.1", 3) || !strncmp((const char *)http_version_opt, "2.0", 3));
1694
0
            av_freep(&http_version_opt);
1695
0
        }
1696
0
    }
1697
1698
0
    seg = next_segment(v);
1699
0
    if (c->http_multiple == 1 && !v->input_next_requested &&
1700
0
        seg && seg->key_type == KEY_NONE && av_strstart(seg->url, "http", NULL)) {
1701
0
        ret = open_input(c, v, seg, &v->input_next);
1702
0
        if (ret < 0) {
1703
0
            if (ff_check_interrupt(c->interrupt_callback))
1704
0
                return AVERROR_EXIT;
1705
0
            av_log(v->parent, AV_LOG_WARNING, "Failed to open segment %"PRId64" of playlist %d\n",
1706
0
                   v->cur_seq_no + 1,
1707
0
                   v->index);
1708
0
        } else {
1709
0
            v->input_next_requested = 1;
1710
0
        }
1711
0
    }
1712
1713
0
    if (v->init_sec_buf_read_offset < v->init_sec_data_len) {
1714
        /* Push init section out first before first actual segment */
1715
0
        int copy_size = FFMIN(v->init_sec_data_len - v->init_sec_buf_read_offset, buf_size);
1716
0
        memcpy(buf, v->init_sec_buf, copy_size);
1717
0
        v->init_sec_buf_read_offset += copy_size;
1718
0
        return copy_size;
1719
0
    }
1720
1721
0
    seg = current_segment(v);
1722
0
    ret = read_from_url(v, seg, buf, buf_size);
1723
0
    if (ret > 0) {
1724
0
        if (just_opened && v->is_id3_timestamped != 0) {
1725
            /* Intercept ID3 tags here, elementary audio streams are required
1726
             * to convey timestamps using them in the beginning of each segment. */
1727
0
            intercept_id3(v, buf, buf_size, &ret);
1728
0
        }
1729
1730
0
        return ret;
1731
0
    }
1732
0
    if (c->http_persistent &&
1733
0
        seg->key_type == KEY_NONE && av_strstart(seg->url, "http", NULL)) {
1734
0
        v->input_read_done = 1;
1735
0
    } else {
1736
0
        ff_format_io_close(v->parent, &v->input);
1737
0
    }
1738
0
    v->cur_seq_no++;
1739
1740
0
    c->cur_seq_no = v->cur_seq_no;
1741
1742
0
    goto restart;
1743
0
}
1744
1745
static int read_data_subtitle_segment(void *opaque, uint8_t *buf, int buf_size)
1746
0
{
1747
0
    struct playlist *v = opaque;
1748
0
    HLSContext *c = v->parent->priv_data;
1749
0
    int ret;
1750
0
    struct segment *seg;
1751
1752
0
    if (!v->needed || v->cur_seq_no - v->start_seq_no >= v->n_segments) {
1753
0
        return AVERROR_EOF;
1754
0
    } else {
1755
0
        seg = current_segment(v);
1756
0
    }
1757
1758
0
    if (!v->input) {
1759
0
        ret = open_input(c, v, seg, &v->input);
1760
0
        if (ret < 0) {
1761
0
            if (ff_check_interrupt(c->interrupt_callback))
1762
0
                return AVERROR_EXIT;
1763
0
            av_log(v->parent, AV_LOG_WARNING, "Failed to open segment of playlist %d\n",
1764
0
                   v->index);
1765
0
            return ret;
1766
0
        }
1767
0
    }
1768
1769
0
    return read_from_url(v, seg, buf, buf_size);
1770
0
}
1771
1772
static int nested_io_open(AVFormatContext *s, AVIOContext **pb, const char *url,
1773
                          int flags, AVDictionary **opts)
1774
0
{
1775
0
    av_log(s, AV_LOG_ERROR,
1776
0
           "A HLS playlist item '%s' referred to an external file '%s'. "
1777
0
           "Opening this file was forbidden for security reasons\n",
1778
0
           s->url, url);
1779
0
    return AVERROR(EPERM);
1780
0
}
1781
1782
static int init_subtitle_context(struct playlist *pls)
1783
0
{
1784
0
    HLSContext *c = pls->parent->priv_data;
1785
0
    const AVInputFormat *in_fmt;
1786
0
    AVDictionary *opts = NULL;
1787
0
    int ret;
1788
1789
0
    if (!(pls->ctx = avformat_alloc_context()))
1790
0
        return AVERROR(ENOMEM);
1791
1792
0
    pls->read_buffer = av_malloc(INITIAL_BUFFER_SIZE);
1793
0
    if (!pls->read_buffer) {
1794
0
        avformat_free_context(pls->ctx);
1795
0
        pls->ctx = NULL;
1796
0
        return AVERROR(ENOMEM);
1797
0
    }
1798
1799
0
    ffio_init_context(&pls->pb, pls->read_buffer, INITIAL_BUFFER_SIZE, 0, pls,
1800
0
                      read_data_subtitle_segment, NULL, NULL);
1801
0
    pls->pb.pub.seekable = 0;
1802
0
    pls->ctx->pb       = &pls->pb.pub;
1803
0
    pls->ctx->io_open  = nested_io_open;
1804
1805
0
    ret = ff_copy_whiteblacklists(pls->ctx, pls->parent);
1806
0
    if (ret < 0)
1807
0
        return ret;
1808
1809
0
    in_fmt = av_find_input_format("webvtt");
1810
0
    av_dict_copy(&opts, c->seg_format_opts, 0);
1811
0
    ret = avformat_open_input(&pls->ctx, current_segment(pls)->url, in_fmt, &opts);
1812
0
    av_dict_free(&opts);
1813
1814
0
    return ret;
1815
0
}
1816
1817
static int read_subtitle_packet(struct playlist *v, AVPacket *pkt)
1818
0
{
1819
0
    HLSContext *c = v->parent->priv_data;
1820
0
    int ret;
1821
1822
0
restart:
1823
0
    ret = reload_playlist(v, c);
1824
0
    if (ret < 0)
1825
0
        return ret;
1826
1827
0
    if (v->input && !v->ctx)
1828
0
        ff_format_io_close(v->parent, &v->input);
1829
1830
0
    if (!v->input && !v->ctx) {
1831
0
        ret = init_subtitle_context(v);
1832
0
        if (ret < 0)
1833
0
            return ret;
1834
0
    }
1835
1836
0
    ret = av_read_frame(v->ctx, v->pkt);
1837
0
    if (!ret) {
1838
0
        return ret;
1839
0
    }
1840
0
    ff_format_io_close(v->parent, &v->input);
1841
0
    v->cur_seq_no++;
1842
0
    c->cur_seq_no = v->cur_seq_no;
1843
1844
0
    avformat_close_input(&v->ctx);
1845
1846
0
    goto restart;
1847
0
}
1848
1849
static void add_renditions_to_variant(HLSContext *c, struct variant *var,
1850
                                      enum AVMediaType type, const char *group_id)
1851
4.16k
{
1852
4.16k
    int i;
1853
1854
1.19M
    for (i = 0; i < c->n_renditions; i++) {
1855
1.19M
        struct rendition *rend = c->renditions[i];
1856
1857
1.19M
        if (rend->type == type && !strcmp(rend->group_id, group_id)) {
1858
1859
4.17k
            if (rend->playlist)
1860
                /* rendition is an external playlist
1861
                 * => add the playlist to the variant */
1862
965
                dynarray_add(&var->playlists, &var->n_playlists, rend->playlist);
1863
3.21k
            else
1864
                /* rendition is part of the variant main Media Playlist
1865
                 * => add the rendition to the main Media Playlist */
1866
3.21k
                dynarray_add(&var->playlists[0]->renditions,
1867
4.17k
                             &var->playlists[0]->n_renditions,
1868
4.17k
                             rend);
1869
4.17k
        }
1870
1.19M
    }
1871
4.16k
}
1872
1873
static void add_metadata_from_renditions(AVFormatContext *s, struct playlist *pls,
1874
                                         enum AVMediaType type)
1875
0
{
1876
0
    int rend_idx = 0;
1877
0
    int i;
1878
1879
0
    for (i = 0; i < pls->n_main_streams; i++) {
1880
0
        AVStream *st = pls->main_streams[i];
1881
1882
0
        if (st->codecpar->codec_type != type)
1883
0
            continue;
1884
1885
0
        for (; rend_idx < pls->n_renditions; rend_idx++) {
1886
0
            struct rendition *rend = pls->renditions[rend_idx];
1887
1888
0
            if (rend->type != type)
1889
0
                continue;
1890
1891
0
            if (rend->language[0])
1892
0
                av_dict_set(&st->metadata, "language", rend->language, 0);
1893
0
            if (rend->name[0])
1894
0
                av_dict_set(&st->metadata, "comment", rend->name, 0);
1895
1896
0
            st->disposition |= rend->disposition;
1897
0
        }
1898
0
        if (rend_idx >=pls->n_renditions)
1899
0
            break;
1900
0
    }
1901
0
}
1902
1903
/* if timestamp was in valid range: returns 1 and sets seq_no
1904
 * if not: returns 0 and sets seq_no to closest segment */
1905
static int find_timestamp_in_playlist(HLSContext *c, struct playlist *pls,
1906
                                      int64_t timestamp, int64_t *seq_no,
1907
                                      int64_t *seg_start_ts)
1908
0
{
1909
0
    int i;
1910
0
    int64_t pos = c->first_timestamp == AV_NOPTS_VALUE ?
1911
0
                  0 : c->first_timestamp;
1912
1913
0
    if (timestamp < pos) {
1914
0
        *seq_no = pls->start_seq_no;
1915
0
        return 0;
1916
0
    }
1917
1918
0
    for (i = 0; i < pls->n_segments; i++) {
1919
0
        int64_t diff = pos + pls->segments[i]->duration - timestamp;
1920
0
        if (diff > 0) {
1921
0
            *seq_no = pls->start_seq_no + i;
1922
0
            if (seg_start_ts) {
1923
0
                *seg_start_ts = pos;
1924
0
            }
1925
0
            return 1;
1926
0
        }
1927
0
        pos += pls->segments[i]->duration;
1928
0
    }
1929
1930
0
    *seq_no = pls->start_seq_no + pls->n_segments - 1;
1931
1932
0
    return 0;
1933
0
}
1934
1935
static int64_t select_cur_seq_no(HLSContext *c, struct playlist *pls)
1936
415
{
1937
415
    int64_t seq_no;
1938
1939
415
    if (!pls->finished && !c->first_packet &&
1940
415
        av_gettime_relative() - pls->last_load_time >= default_reload_interval(pls))
1941
        /* reload the playlist since it was suspended */
1942
0
        parse_playlist(c, pls->url, pls, NULL);
1943
1944
    /* If playback is already in progress (we are just selecting a new
1945
     * playlist) and this is a complete file, find the matching segment
1946
     * by counting durations. */
1947
415
    if (pls->finished && c->cur_timestamp != AV_NOPTS_VALUE) {
1948
0
        find_timestamp_in_playlist(c, pls, c->cur_timestamp, &seq_no, NULL);
1949
0
        return seq_no;
1950
0
    }
1951
1952
415
    if (!pls->finished) {
1953
415
        if (!c->first_packet && /* we are doing a segment selection during playback */
1954
415
            c->cur_seq_no >= pls->start_seq_no &&
1955
415
            c->cur_seq_no < pls->start_seq_no + pls->n_segments)
1956
            /* While spec 3.4.3 says that we cannot assume anything about the
1957
             * content at the same sequence number on different playlists,
1958
             * in practice this seems to work and doing it otherwise would
1959
             * require us to download a segment to inspect its timestamps. */
1960
0
            return c->cur_seq_no;
1961
1962
        /* If this is a live stream, start live_start_index segments from the
1963
         * start or end */
1964
415
        if (c->live_start_index < 0)
1965
415
            seq_no = pls->start_seq_no + FFMAX(pls->n_segments +
1966
415
                                            c->live_start_index, 0);
1967
0
        else
1968
0
            seq_no = pls->start_seq_no + FFMIN(c->live_start_index,
1969
415
                                            pls->n_segments - 1);
1970
1971
        /* If #EXT-X-START in playlist, need to recalculate */
1972
415
        if (pls->time_offset_flag && c->prefer_x_start) {
1973
0
            int64_t start_timestamp;
1974
0
            int64_t playlist_duration = 0;
1975
0
            int64_t cur_timestamp = c->cur_timestamp == AV_NOPTS_VALUE ? 0 :
1976
0
                                    c->cur_timestamp;
1977
1978
0
            for (int i = 0; i < pls->n_segments; i++)
1979
0
                playlist_duration += pls->segments[i]->duration;
1980
1981
            /* If the absolute value of TIME-OFFSET exceeds
1982
             * the duration of the playlist, it indicates either the end of the
1983
             * playlist (if positive) or the beginning of the playlist (if
1984
             * negative). */
1985
0
            if (pls->start_time_offset >=0 &&
1986
0
                pls->start_time_offset > playlist_duration)
1987
0
                start_timestamp = cur_timestamp + playlist_duration;
1988
0
            else if (pls->start_time_offset >= 0 &&
1989
0
                        pls->start_time_offset <= playlist_duration)
1990
0
                start_timestamp = cur_timestamp + pls->start_time_offset;
1991
0
            else if (pls->start_time_offset < 0 &&
1992
0
                        pls->start_time_offset < -playlist_duration)
1993
0
                start_timestamp = cur_timestamp;
1994
0
            else if (pls->start_time_offset < 0 &&
1995
0
                        pls->start_time_offset > -playlist_duration)
1996
0
                start_timestamp = cur_timestamp + playlist_duration +
1997
0
                                    pls->start_time_offset;
1998
0
            else
1999
0
                start_timestamp = cur_timestamp;
2000
2001
0
            find_timestamp_in_playlist(c, pls, start_timestamp, &seq_no, NULL);
2002
0
        }
2003
415
        return seq_no;
2004
415
    }
2005
2006
    /* Otherwise just start on the first segment. */
2007
0
    return pls->start_seq_no;
2008
415
}
2009
2010
static void add_stream_to_programs(AVFormatContext *s, struct playlist *pls, AVStream *stream)
2011
0
{
2012
0
    HLSContext *c = s->priv_data;
2013
0
    int i, j;
2014
0
    int bandwidth = -1;
2015
2016
0
    for (i = 0; i < c->n_variants; i++) {
2017
0
        struct variant *v = c->variants[i];
2018
2019
0
        for (j = 0; j < v->n_playlists; j++) {
2020
0
            if (v->playlists[j] != pls)
2021
0
                continue;
2022
2023
0
            av_program_add_stream_index(s, i, stream->index);
2024
2025
0
            if (bandwidth < 0)
2026
0
                bandwidth = v->bandwidth;
2027
0
            else if (bandwidth != v->bandwidth)
2028
0
                bandwidth = -1; /* stream in multiple variants with different bandwidths */
2029
0
        }
2030
0
    }
2031
2032
0
    if (bandwidth >= 0)
2033
0
        av_dict_set_int(&stream->metadata, "variant_bitrate", bandwidth, 0);
2034
0
}
2035
2036
static int set_stream_info_from_input_stream(AVStream *st, struct playlist *pls, AVStream *ist)
2037
0
{
2038
0
    int err;
2039
2040
0
    err = avcodec_parameters_copy(st->codecpar, ist->codecpar);
2041
0
    if (err < 0)
2042
0
        return err;
2043
2044
0
    if (pls->is_id3_timestamped) /* custom timestamps via id3 */
2045
0
        avpriv_set_pts_info(st, 33, 1, MPEG_TIME_BASE);
2046
0
    else
2047
0
        avpriv_set_pts_info(st, ist->pts_wrap_bits, ist->time_base.num, ist->time_base.den);
2048
2049
    // copy disposition
2050
0
    st->disposition = ist->disposition;
2051
2052
0
    av_dict_copy(&st->metadata, ist->metadata, 0);
2053
2054
0
    ffstream(st)->need_context_update = 1;
2055
2056
0
    return 0;
2057
0
}
2058
2059
/* add new subdemuxer streams to our context, if any */
2060
static int update_streams_from_subdemuxer(AVFormatContext *s, struct playlist *pls)
2061
0
{
2062
0
    int err;
2063
2064
0
    while (pls->n_main_streams < pls->ctx->nb_streams) {
2065
0
        int ist_idx = pls->n_main_streams;
2066
0
        AVStream *st = avformat_new_stream(s, NULL);
2067
0
        AVStream *ist = pls->ctx->streams[ist_idx];
2068
2069
0
        if (!st)
2070
0
            return AVERROR(ENOMEM);
2071
2072
0
        st->id = pls->index;
2073
0
        dynarray_add(&pls->main_streams, &pls->n_main_streams, st);
2074
2075
0
        add_stream_to_programs(s, pls, st);
2076
2077
0
        err = set_stream_info_from_input_stream(st, pls, ist);
2078
0
        if (err < 0)
2079
0
            return err;
2080
0
    }
2081
2082
0
    return 0;
2083
0
}
2084
2085
static void update_noheader_flag(AVFormatContext *s)
2086
800
{
2087
800
    HLSContext *c = s->priv_data;
2088
800
    int flag_needed = 0;
2089
800
    int i;
2090
2091
89.1k
    for (i = 0; i < c->n_playlists; i++) {
2092
88.3k
        struct playlist *pls = c->playlists[i];
2093
2094
88.3k
        if (pls->has_noheader_flag) {
2095
0
            flag_needed = 1;
2096
0
            break;
2097
0
        }
2098
88.3k
    }
2099
2100
800
    if (flag_needed)
2101
0
        s->ctx_flags |= AVFMTCTX_NOHEADER;
2102
800
    else
2103
800
        s->ctx_flags &= ~AVFMTCTX_NOHEADER;
2104
800
}
2105
2106
static int hls_close(AVFormatContext *s)
2107
8.76k
{
2108
8.76k
    HLSContext *c = s->priv_data;
2109
2110
8.76k
    free_playlist_list(c);
2111
8.76k
    free_variant_list(c);
2112
8.76k
    free_rendition_list(c);
2113
2114
8.76k
    if (c->crypto_ctx.aes_ctx)
2115
0
        av_free(c->crypto_ctx.aes_ctx);
2116
2117
8.76k
    av_dict_free(&c->avio_opts);
2118
8.76k
    ff_format_io_close(c->ctx, &c->playlist_pb);
2119
2120
8.76k
    return 0;
2121
8.76k
}
2122
2123
static int hls_read_header(AVFormatContext *s)
2124
8.76k
{
2125
8.76k
    HLSContext *c = s->priv_data;
2126
8.76k
    int ret = 0, i;
2127
8.76k
    int64_t highest_cur_seq_no = 0;
2128
2129
8.76k
    c->ctx                = s;
2130
8.76k
    c->interrupt_callback = &s->interrupt_callback;
2131
2132
8.76k
    c->first_packet = 1;
2133
8.76k
    c->first_timestamp = AV_NOPTS_VALUE;
2134
8.76k
    c->cur_timestamp = AV_NOPTS_VALUE;
2135
2136
8.76k
    if ((ret = ffio_copy_url_options(s->pb, &c->avio_opts)) < 0)
2137
0
        return ret;
2138
2139
    /* XXX: Some HLS servers don't like being sent the range header,
2140
       in this case, we need to set http_seekable = 0 to disable
2141
       the range header */
2142
8.76k
    av_dict_set_int(&c->avio_opts, "seekable", c->http_seekable, 0);
2143
2144
8.76k
    if ((ret = parse_playlist(c, s->url, NULL, s->pb)) < 0)
2145
5.48k
        return ret;
2146
2147
3.27k
    if (c->n_variants == 0) {
2148
1.61k
        av_log(s, AV_LOG_WARNING, "Empty playlist\n");
2149
1.61k
        return AVERROR_EOF;
2150
1.61k
    }
2151
    /* If the playlist only contained playlists (Master Playlist),
2152
     * parse each individual playlist. */
2153
1.66k
    if (c->n_playlists > 1 || c->playlists[0]->n_segments == 0) {
2154
138k
        for (i = 0; i < c->n_playlists; i++) {
2155
137k
            struct playlist *pls = c->playlists[i];
2156
137k
            pls->m3u8_hold_counters = 0;
2157
137k
            if ((ret = parse_playlist(c, pls->url, pls, NULL)) < 0) {
2158
137k
                av_log(s, AV_LOG_WARNING, "parse_playlist error %s [%s]\n", av_err2str(ret), pls->url);
2159
137k
                pls->broken = 1;
2160
137k
                if (c->n_playlists > 1)
2161
136k
                    continue;
2162
448
                return ret;
2163
137k
            }
2164
137k
        }
2165
1.38k
    }
2166
2167
136k
    for (i = 0; i < c->n_variants; i++) {
2168
135k
        if (c->variants[i]->playlists[0]->n_segments == 0) {
2169
134k
            av_log(s, AV_LOG_WARNING, "Empty segment [%s]\n", c->variants[i]->playlists[0]->url);
2170
134k
            c->variants[i]->playlists[0]->broken = 1;
2171
134k
        }
2172
135k
    }
2173
2174
    /* If this isn't a live stream, calculate the total duration of the
2175
     * stream. */
2176
1.21k
    if (c->variants[0]->playlists[0]->finished) {
2177
0
        int64_t duration = 0;
2178
0
        for (i = 0; i < c->variants[0]->playlists[0]->n_segments; i++)
2179
0
            duration += c->variants[0]->playlists[0]->segments[i]->duration;
2180
0
        s->duration = duration;
2181
0
    }
2182
2183
    /* Associate renditions with variants */
2184
136k
    for (i = 0; i < c->n_variants; i++) {
2185
135k
        struct variant *var = c->variants[i];
2186
2187
135k
        if (var->audio_group[0])
2188
1.03k
            add_renditions_to_variant(c, var, AVMEDIA_TYPE_AUDIO, var->audio_group);
2189
135k
        if (var->video_group[0])
2190
2.34k
            add_renditions_to_variant(c, var, AVMEDIA_TYPE_VIDEO, var->video_group);
2191
135k
        if (var->subtitles_group[0])
2192
788
            add_renditions_to_variant(c, var, AVMEDIA_TYPE_SUBTITLE, var->subtitles_group);
2193
135k
    }
2194
2195
    /* Create a program for each variant */
2196
136k
    for (i = 0; i < c->n_variants; i++) {
2197
135k
        struct variant *v = c->variants[i];
2198
135k
        AVProgram *program;
2199
2200
135k
        program = av_new_program(s, i);
2201
135k
        if (!program)
2202
0
            return AVERROR(ENOMEM);
2203
135k
        av_dict_set_int(&program->metadata, "variant_bitrate", v->bandwidth, 0);
2204
135k
    }
2205
2206
    /* Select the starting segments */
2207
138k
    for (i = 0; i < c->n_playlists; i++) {
2208
136k
        struct playlist *pls = c->playlists[i];
2209
2210
136k
        if (pls->n_segments == 0)
2211
136k
            continue;
2212
2213
415
        pls->cur_seq_no = select_cur_seq_no(c, pls);
2214
415
        highest_cur_seq_no = FFMAX(highest_cur_seq_no, pls->cur_seq_no);
2215
415
    }
2216
2217
1.21k
    av_dict_set(&c->seg_format_opts, "prefer_hls_mpegts_pts", "1", 0);
2218
2219
    /* Open the demuxer for each playlist */
2220
121k
    for (i = 0; i < c->n_playlists; i++) {
2221
120k
        struct playlist *pls = c->playlists[i];
2222
120k
        const AVInputFormat *in_fmt = NULL;
2223
120k
        char *url;
2224
120k
        AVDictionary *options = NULL;
2225
120k
        struct segment *seg = NULL;
2226
2227
120k
        if (!(pls->ctx = avformat_alloc_context()))
2228
0
            return AVERROR(ENOMEM);
2229
2230
120k
        if (pls->n_segments == 0)
2231
120k
            continue;
2232
2233
415
        pls->index  = i;
2234
415
        pls->needed = 1;
2235
415
        pls->parent = s;
2236
2237
        /*
2238
         * If this is a live stream and this playlist looks like it is one segment
2239
         * behind, try to sync it up so that every substream starts at the same
2240
         * time position (so e.g. avformat_find_stream_info() will see packets from
2241
         * all active streams within the first few seconds). This is not very generic,
2242
         * though, as the sequence numbers are technically independent.
2243
         */
2244
415
        if (!pls->finished && pls->cur_seq_no == highest_cur_seq_no - 1 &&
2245
415
            highest_cur_seq_no < pls->start_seq_no + pls->n_segments) {
2246
0
            pls->cur_seq_no = highest_cur_seq_no;
2247
0
        }
2248
2249
415
        pls->read_buffer = av_malloc(INITIAL_BUFFER_SIZE);
2250
415
        if (!pls->read_buffer){
2251
0
            avformat_free_context(pls->ctx);
2252
0
            pls->ctx = NULL;
2253
0
            return AVERROR(ENOMEM);
2254
0
        }
2255
2256
415
        if (pls->is_subtitle)
2257
0
            ffio_init_context(&pls->pb, (unsigned char*)av_strdup("WEBVTT\n"), (int)strlen("WEBVTT\n"), 0, pls,
2258
0
                                       NULL, NULL, NULL);
2259
415
        else
2260
415
            ffio_init_context(&pls->pb, pls->read_buffer, INITIAL_BUFFER_SIZE, 0, pls,
2261
415
                                        read_data_continuous, NULL, NULL);
2262
2263
        /*
2264
         * If encryption scheme is SAMPLE-AES, try to read  ID3 tags of
2265
         * external audio track that contains audio setup information
2266
         */
2267
415
        seg = current_segment(pls);
2268
415
        if (seg && seg->key_type == KEY_SAMPLE_AES && pls->n_renditions > 0 &&
2269
415
            pls->renditions[0]->type == AVMEDIA_TYPE_AUDIO) {
2270
0
            uint8_t buf[HLS_MAX_ID3_TAGS_DATA_LEN];
2271
0
            if ((ret = avio_read(&pls->pb.pub, buf, HLS_MAX_ID3_TAGS_DATA_LEN)) < 0) {
2272
                /* Fail if error was not end of file */
2273
0
                if (ret != AVERROR_EOF) {
2274
0
                    avformat_free_context(pls->ctx);
2275
0
                    pls->ctx = NULL;
2276
0
                    return ret;
2277
0
                }
2278
0
            }
2279
0
            ret = 0;
2280
            /* Reset reading */
2281
0
            ff_format_io_close(pls->parent, &pls->input);
2282
0
            pls->input = NULL;
2283
0
            pls->input_read_done = 0;
2284
0
            ff_format_io_close(pls->parent, &pls->input_next);
2285
0
            pls->input_next = NULL;
2286
0
            pls->input_next_requested = 0;
2287
0
            pls->cur_seg_offset = 0;
2288
0
            pls->cur_init_section = NULL;
2289
            /* Reset EOF flag */
2290
0
            pls->pb.pub.eof_reached = 0;
2291
            /* Clear any buffered data */
2292
0
            pls->pb.pub.buf_end = pls->pb.pub.buf_ptr = pls->pb.pub.buffer;
2293
            /* Reset the position */
2294
0
            pls->pb.pub.pos = 0;
2295
0
        }
2296
2297
        /*
2298
         * If encryption scheme is SAMPLE-AES and audio setup information is present in external audio track,
2299
         * use that information to find the media format, otherwise probe input data
2300
         */
2301
415
        seg = current_segment(pls);
2302
415
        if (seg && seg->key_type == KEY_SAMPLE_AES && pls->is_id3_timestamped &&
2303
415
            pls->audio_setup_info.codec_id != AV_CODEC_ID_NONE) {
2304
0
            av_assert1(pls->audio_setup_info.codec_id == AV_CODEC_ID_AAC ||
2305
0
                       pls->audio_setup_info.codec_id == AV_CODEC_ID_AC3 ||
2306
0
                       pls->audio_setup_info.codec_id == AV_CODEC_ID_EAC3);
2307
            // Keep this list in sync with ff_hls_senc_read_audio_setup_info()
2308
0
            in_fmt = av_find_input_format(pls->audio_setup_info.codec_id == AV_CODEC_ID_AAC ? "aac" :
2309
0
                                          pls->audio_setup_info.codec_id == AV_CODEC_ID_AC3 ? "ac3" : "eac3");
2310
415
        } else {
2311
415
            pls->ctx->probesize = s->probesize > 0 ? s->probesize : 1024 * 4;
2312
415
            pls->ctx->max_analyze_duration = s->max_analyze_duration > 0 ? s->max_analyze_duration : 4 * AV_TIME_BASE;
2313
415
            pls->ctx->interrupt_callback = s->interrupt_callback;
2314
415
            url = av_strdup(pls->segments[0]->url);
2315
415
            ret = av_probe_input_buffer(&pls->pb.pub, &in_fmt, url, NULL, 0, 0);
2316
2317
11.4k
            for (int n = 0; n < pls->n_segments; n++)
2318
11.0k
                if (ret >= 0)
2319
0
                    ret = test_segment(s, in_fmt, pls, pls->segments[n]);
2320
2321
415
            if (ret < 0) {
2322
                /* Free the ctx - it isn't initialized properly at this point,
2323
                * so avformat_close_input shouldn't be called. If
2324
                * avformat_open_input fails below, it frees and zeros the
2325
                * context, so it doesn't need any special treatment like this. */
2326
415
                av_log(s, AV_LOG_ERROR, "Error when loading first segment '%s'\n", url);
2327
415
                avformat_free_context(pls->ctx);
2328
415
                pls->ctx = NULL;
2329
415
                av_free(url);
2330
415
                return ret;
2331
415
            }
2332
0
            av_free(url);
2333
0
        }
2334
2335
0
        seg = current_segment(pls);
2336
0
        if (seg && seg->key_type == KEY_SAMPLE_AES) {
2337
0
            if (strstr(in_fmt->name, "mov")) {
2338
0
                char key[33];
2339
0
                ff_data_to_hex(key, pls->key, sizeof(pls->key), 0);
2340
0
                av_dict_set(&options, "decryption_key", key, 0);
2341
0
            } else if (!c->crypto_ctx.aes_ctx) {
2342
0
                c->crypto_ctx.aes_ctx = av_aes_alloc();
2343
0
                if (!c->crypto_ctx.aes_ctx) {
2344
0
                    avformat_free_context(pls->ctx);
2345
0
                    pls->ctx = NULL;
2346
0
                    return AVERROR(ENOMEM);
2347
0
                }
2348
0
            }
2349
0
        }
2350
2351
0
        pls->ctx->pb       = &pls->pb.pub;
2352
0
        pls->ctx->io_open  = nested_io_open;
2353
0
        pls->ctx->flags   |= s->flags & ~AVFMT_FLAG_CUSTOM_IO;
2354
2355
0
        if ((ret = ff_copy_whiteblacklists(pls->ctx, s)) < 0)
2356
0
            return ret;
2357
2358
0
        av_dict_copy(&options, c->seg_format_opts, 0);
2359
2360
0
        ret = avformat_open_input(&pls->ctx, pls->segments[0]->url, in_fmt, &options);
2361
0
        av_dict_free(&options);
2362
0
        if (ret < 0)
2363
0
            return ret;
2364
2365
0
        if (pls->id3_deferred_extra && pls->ctx->nb_streams == 1) {
2366
0
            ff_id3v2_parse_apic(pls->ctx, pls->id3_deferred_extra);
2367
0
            avformat_queue_attached_pictures(pls->ctx);
2368
0
            ff_id3v2_parse_priv(pls->ctx, pls->id3_deferred_extra);
2369
0
            ff_id3v2_free_extra_meta(&pls->id3_deferred_extra);
2370
0
        }
2371
2372
0
        if (pls->is_id3_timestamped == -1)
2373
0
            av_log(s, AV_LOG_WARNING, "No expected HTTP requests have been made\n");
2374
2375
        /*
2376
         * For ID3 timestamped raw audio streams we need to detect the packet
2377
         * durations to calculate timestamps in fill_timing_for_id3_timestamped_stream(),
2378
         * but for other streams we can rely on our user calling avformat_find_stream_info()
2379
         * on us if they want to.
2380
         */
2381
0
        if (pls->is_id3_timestamped || (pls->n_renditions > 0 && pls->renditions[0]->type == AVMEDIA_TYPE_AUDIO)) {
2382
0
            seg = current_segment(pls);
2383
0
            if (seg && seg->key_type == KEY_SAMPLE_AES && pls->audio_setup_info.setup_data_length > 0 &&
2384
0
                pls->ctx->nb_streams == 1)
2385
0
                ret = ff_hls_senc_parse_audio_setup_info(pls->ctx->streams[0], &pls->audio_setup_info);
2386
0
            else
2387
0
                ret = avformat_find_stream_info(pls->ctx, NULL);
2388
2389
0
            if (ret < 0)
2390
0
                return ret;
2391
0
        }
2392
2393
0
        pls->has_noheader_flag = !!(pls->ctx->ctx_flags & AVFMTCTX_NOHEADER);
2394
2395
        /* Create new AVStreams for each stream in this playlist */
2396
0
        ret = update_streams_from_subdemuxer(s, pls);
2397
0
        if (ret < 0)
2398
0
            return ret;
2399
2400
        /*
2401
         * Copy any metadata from playlist to main streams, but do not set
2402
         * event flags.
2403
         */
2404
0
        if (pls->n_main_streams)
2405
0
            av_dict_copy(&pls->main_streams[0]->metadata, pls->ctx->metadata, 0);
2406
2407
0
        if (pls->is_subtitle) {
2408
0
            avformat_free_context(pls->ctx);
2409
0
            pls->ctx = NULL;
2410
0
            pls->needed = 0;
2411
0
            pls->main_streams[0]->discard = AVDISCARD_ALL;
2412
0
        }
2413
2414
0
        add_metadata_from_renditions(s, pls, AVMEDIA_TYPE_AUDIO);
2415
0
        add_metadata_from_renditions(s, pls, AVMEDIA_TYPE_VIDEO);
2416
0
        add_metadata_from_renditions(s, pls, AVMEDIA_TYPE_SUBTITLE);
2417
0
    }
2418
2419
800
    update_noheader_flag(s);
2420
2421
800
    return 0;
2422
1.21k
}
2423
2424
static int recheck_discard_flags(AVFormatContext *s, int first)
2425
800
{
2426
800
    HLSContext *c = s->priv_data;
2427
800
    int i, changed = 0;
2428
800
    int cur_needed;
2429
2430
    /* Check if any new streams are needed */
2431
89.1k
    for (i = 0; i < c->n_playlists; i++) {
2432
88.3k
        struct playlist *pls = c->playlists[i];
2433
2434
88.3k
        cur_needed = playlist_needed(c->playlists[i]);
2435
2436
88.3k
        if (pls->broken) {
2437
88.3k
            continue;
2438
88.3k
        }
2439
0
        if (cur_needed && !pls->needed) {
2440
0
            pls->needed = 1;
2441
0
            changed = 1;
2442
0
            pls->cur_seq_no = select_cur_seq_no(c, pls);
2443
0
            pls->pb.pub.eof_reached = 0;
2444
0
            if (c->cur_timestamp != AV_NOPTS_VALUE) {
2445
                /* catch up */
2446
0
                pls->seek_timestamp = c->cur_timestamp;
2447
0
                pls->seek_flags = AVSEEK_FLAG_ANY;
2448
0
                pls->seek_stream_index = -1;
2449
0
            }
2450
0
            av_log(s, AV_LOG_INFO, "Now receiving playlist %d, segment %"PRId64"\n", i, pls->cur_seq_no);
2451
0
        } else if (first && !cur_needed && pls->needed) {
2452
0
            ff_format_io_close(pls->parent, &pls->input);
2453
0
            pls->input_read_done = 0;
2454
0
            ff_format_io_close(pls->parent, &pls->input_next);
2455
0
            pls->input_next_requested = 0;
2456
0
            if (pls->is_subtitle)
2457
0
                avformat_close_input(&pls->ctx);
2458
0
            pls->needed = 0;
2459
0
            changed = 1;
2460
0
            av_log(s, AV_LOG_INFO, "No longer receiving playlist %d\n", i);
2461
0
        }
2462
0
    }
2463
800
    return changed;
2464
800
}
2465
2466
static void fill_timing_for_id3_timestamped_stream(struct playlist *pls)
2467
0
{
2468
0
    if (pls->id3_offset >= 0) {
2469
0
        pls->pkt->dts = pls->id3_mpegts_timestamp +
2470
0
                                 av_rescale_q(pls->id3_offset,
2471
0
                                              pls->ctx->streams[pls->pkt->stream_index]->time_base,
2472
0
                                              MPEG_TIME_BASE_Q);
2473
0
        if (pls->pkt->duration)
2474
0
            pls->id3_offset += pls->pkt->duration;
2475
0
        else
2476
0
            pls->id3_offset = -1;
2477
0
    } else {
2478
        /* there have been packets with unknown duration
2479
         * since the last id3 tag, should not normally happen */
2480
0
        pls->pkt->dts = AV_NOPTS_VALUE;
2481
0
    }
2482
2483
0
    if (pls->pkt->duration)
2484
0
        pls->pkt->duration = av_rescale_q(pls->pkt->duration,
2485
0
                                         pls->ctx->streams[pls->pkt->stream_index]->time_base,
2486
0
                                         MPEG_TIME_BASE_Q);
2487
2488
0
    pls->pkt->pts = AV_NOPTS_VALUE;
2489
0
}
2490
2491
static AVRational get_timebase(struct playlist *pls)
2492
0
{
2493
0
    if (pls->is_id3_timestamped)
2494
0
        return MPEG_TIME_BASE_Q;
2495
2496
0
    return pls->ctx->streams[pls->pkt->stream_index]->time_base;
2497
0
}
2498
2499
static int compare_ts_with_wrapdetect(int64_t ts_a, struct playlist *pls_a,
2500
                                      int64_t ts_b, struct playlist *pls_b)
2501
0
{
2502
0
    int64_t scaled_ts_a = av_rescale_q(ts_a, get_timebase(pls_a), MPEG_TIME_BASE_Q);
2503
0
    int64_t scaled_ts_b = av_rescale_q(ts_b, get_timebase(pls_b), MPEG_TIME_BASE_Q);
2504
2505
0
    return av_compare_mod(scaled_ts_a, scaled_ts_b, 1LL << 33);
2506
0
}
2507
2508
static int hls_read_packet(AVFormatContext *s, AVPacket *pkt)
2509
800
{
2510
800
    HLSContext *c = s->priv_data;
2511
800
    int ret, i, minplaylist = -1;
2512
2513
800
    recheck_discard_flags(s, c->first_packet);
2514
800
    c->first_packet = 0;
2515
2516
89.1k
    for (i = 0; i < c->n_playlists; i++) {
2517
88.3k
        struct playlist *pls = c->playlists[i];
2518
        /* Make sure we've got one buffered packet from each open playlist
2519
         * stream */
2520
88.3k
        if (pls->needed && !pls->pkt->data) {
2521
0
            while (1) {
2522
0
                int64_t ts_diff;
2523
0
                AVRational tb;
2524
0
                struct segment *seg = NULL;
2525
0
                if (pls->is_subtitle)
2526
0
                    ret = read_subtitle_packet(pls, pls->pkt);
2527
0
                else
2528
0
                    ret = av_read_frame(pls->ctx, pls->pkt);
2529
0
                if (ret < 0) {
2530
0
                    if (!avio_feof(&pls->pb.pub) && ret != AVERROR_EOF)
2531
0
                        return ret;
2532
0
                    break;
2533
0
                } else {
2534
                    /* stream_index check prevents matching picture attachments etc. */
2535
0
                    if (pls->is_id3_timestamped && pls->pkt->stream_index == 0) {
2536
                        /* audio elementary streams are id3 timestamped */
2537
0
                        fill_timing_for_id3_timestamped_stream(pls);
2538
0
                    }
2539
2540
0
                    if (c->first_timestamp == AV_NOPTS_VALUE &&
2541
0
                        pls->pkt->dts       != AV_NOPTS_VALUE)
2542
0
                        c->first_timestamp = av_rescale_q(pls->pkt->dts,
2543
0
                            get_timebase(pls), AV_TIME_BASE_Q);
2544
0
                }
2545
2546
0
                seg = current_segment(pls);
2547
0
                if (seg && seg->key_type == KEY_SAMPLE_AES && !strstr(pls->ctx->iformat->name, "mov")) {
2548
0
                    enum AVCodecID codec_id = pls->ctx->streams[pls->pkt->stream_index]->codecpar->codec_id;
2549
0
                    memcpy(c->crypto_ctx.iv, seg->iv, sizeof(seg->iv));
2550
0
                    memcpy(c->crypto_ctx.key, pls->key, sizeof(pls->key));
2551
0
                    ff_hls_senc_decrypt_frame(codec_id, &c->crypto_ctx, pls->pkt);
2552
0
                }
2553
2554
0
                if (pls->seek_timestamp == AV_NOPTS_VALUE)
2555
0
                    break;
2556
2557
0
                if (pls->seek_stream_index < 0 ||
2558
0
                    pls->seek_stream_index == pls->pkt->stream_index) {
2559
2560
0
                    if (pls->pkt->dts == AV_NOPTS_VALUE) {
2561
0
                        pls->seek_timestamp = AV_NOPTS_VALUE;
2562
0
                        break;
2563
0
                    }
2564
2565
0
                    tb = get_timebase(pls);
2566
0
                    ts_diff = av_rescale_rnd(pls->pkt->dts, AV_TIME_BASE,
2567
0
                                            tb.den, AV_ROUND_DOWN) -
2568
0
                            pls->seek_timestamp;
2569
0
                    if (ts_diff >= 0 && (pls->seek_flags  & AVSEEK_FLAG_ANY ||
2570
0
                                        pls->pkt->flags & AV_PKT_FLAG_KEY)) {
2571
0
                        pls->seek_timestamp = AV_NOPTS_VALUE;
2572
0
                        break;
2573
0
                    }
2574
0
                }
2575
0
                av_packet_unref(pls->pkt);
2576
0
            }
2577
0
        }
2578
        /* Check if this stream has the packet with the lowest dts */
2579
88.3k
        if (pls->pkt->data) {
2580
0
            struct playlist *minpls = minplaylist < 0 ?
2581
0
                                     NULL : c->playlists[minplaylist];
2582
0
            if (minplaylist < 0) {
2583
0
                minplaylist = i;
2584
0
            } else {
2585
0
                int64_t dts     =    pls->pkt->dts;
2586
0
                int64_t mindts  = minpls->pkt->dts;
2587
2588
0
                if (dts == AV_NOPTS_VALUE ||
2589
0
                    (mindts != AV_NOPTS_VALUE && compare_ts_with_wrapdetect(dts, pls, mindts, minpls) < 0))
2590
0
                    minplaylist = i;
2591
0
            }
2592
0
        }
2593
88.3k
    }
2594
2595
    /* If we got a packet, return it */
2596
800
    if (minplaylist >= 0) {
2597
0
        struct playlist *pls = c->playlists[minplaylist];
2598
0
        AVStream *ist;
2599
0
        AVStream *st;
2600
2601
0
        ret = update_streams_from_subdemuxer(s, pls);
2602
0
        if (ret < 0) {
2603
0
            av_packet_unref(pls->pkt);
2604
0
            return ret;
2605
0
        }
2606
2607
        // If sub-demuxer reports updated metadata, copy it to the first stream
2608
        // and set its AVSTREAM_EVENT_FLAG_METADATA_UPDATED flag.
2609
0
        if (pls->ctx->event_flags & AVFMT_EVENT_FLAG_METADATA_UPDATED) {
2610
0
            if (pls->n_main_streams) {
2611
0
                st = pls->main_streams[0];
2612
0
                av_dict_copy(&st->metadata, pls->ctx->metadata, 0);
2613
0
                st->event_flags |= AVSTREAM_EVENT_FLAG_METADATA_UPDATED;
2614
0
            }
2615
0
            pls->ctx->event_flags &= ~AVFMT_EVENT_FLAG_METADATA_UPDATED;
2616
0
        }
2617
2618
        /* check if noheader flag has been cleared by the subdemuxer */
2619
0
        if (pls->has_noheader_flag && !(pls->ctx->ctx_flags & AVFMTCTX_NOHEADER)) {
2620
0
            pls->has_noheader_flag = 0;
2621
0
            update_noheader_flag(s);
2622
0
        }
2623
2624
0
        if (pls->pkt->stream_index >= pls->n_main_streams) {
2625
0
            av_log(s, AV_LOG_ERROR, "stream index inconsistency: index %d, %d main streams, %d subdemuxer streams\n",
2626
0
                   pls->pkt->stream_index, pls->n_main_streams, pls->ctx->nb_streams);
2627
0
            av_packet_unref(pls->pkt);
2628
0
            return AVERROR_BUG;
2629
0
        }
2630
2631
0
        ist = pls->ctx->streams[pls->pkt->stream_index];
2632
0
        st = pls->main_streams[pls->pkt->stream_index];
2633
2634
0
        av_packet_move_ref(pkt, pls->pkt);
2635
0
        pkt->stream_index = st->index;
2636
2637
0
        if (pkt->dts != AV_NOPTS_VALUE)
2638
0
            c->cur_timestamp = av_rescale_q(pkt->dts,
2639
0
                                            ist->time_base,
2640
0
                                            AV_TIME_BASE_Q);
2641
2642
        /* There may be more situations where this would be useful, but this at least
2643
         * handles newly probed codecs properly (i.e. request_probe by mpegts). */
2644
0
        if (ist->codecpar->codec_id != st->codecpar->codec_id) {
2645
0
            ret = set_stream_info_from_input_stream(st, pls, ist);
2646
0
            if (ret < 0) {
2647
0
                return ret;
2648
0
            }
2649
0
        }
2650
2651
0
        return 0;
2652
0
    }
2653
800
    return AVERROR_EOF;
2654
800
}
2655
2656
static int hls_read_seek(AVFormatContext *s, int stream_index,
2657
                               int64_t timestamp, int flags)
2658
0
{
2659
0
    HLSContext *c = s->priv_data;
2660
0
    struct playlist *seek_pls = NULL;
2661
0
    int i, j;
2662
0
    int stream_subdemuxer_index;
2663
0
    int64_t first_timestamp, seek_timestamp, duration;
2664
0
    int64_t seq_no, seg_start_ts;
2665
2666
0
    if ((flags & AVSEEK_FLAG_BYTE) || (c->ctx->ctx_flags & AVFMTCTX_UNSEEKABLE))
2667
0
        return AVERROR(ENOSYS);
2668
2669
0
    first_timestamp = c->first_timestamp == AV_NOPTS_VALUE ?
2670
0
                      0 : c->first_timestamp;
2671
2672
0
    seek_timestamp = av_rescale_rnd(timestamp, AV_TIME_BASE,
2673
0
                                    s->streams[stream_index]->time_base.den,
2674
0
                                    AV_ROUND_DOWN);
2675
2676
0
    duration = s->duration == AV_NOPTS_VALUE ?
2677
0
               0 : s->duration;
2678
2679
0
    if (0 < duration && duration < seek_timestamp - first_timestamp)
2680
0
        return AVERROR(EIO);
2681
2682
    /* find the playlist with the specified stream */
2683
0
    for (i = 0; i < c->n_playlists; i++) {
2684
0
        struct playlist *pls = c->playlists[i];
2685
0
        for (j = 0; j < pls->n_main_streams; j++) {
2686
0
            if (pls->main_streams[j] == s->streams[stream_index]) {
2687
0
                seek_pls = pls;
2688
0
                stream_subdemuxer_index = j;
2689
0
                break;
2690
0
            }
2691
0
        }
2692
0
    }
2693
    /* check if the timestamp is valid for the playlist with the
2694
     * specified stream index */
2695
0
    if (!seek_pls || !find_timestamp_in_playlist(c, seek_pls, seek_timestamp, &seq_no, &seg_start_ts))
2696
0
        return AVERROR(EIO);
2697
2698
0
    if (s->streams[stream_index]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO &&
2699
0
        flags & AVSEEK_FLAG_BACKWARD && !(flags & AVSEEK_FLAG_ANY)) {
2700
        /* Seeking to start of segment ensures we seek to a keyframe located
2701
         * before the given timestamp. */
2702
0
        seek_timestamp = seg_start_ts;
2703
0
    }
2704
2705
    /* set segment now so we do not need to search again below */
2706
0
    seek_pls->cur_seq_no = seq_no;
2707
0
    seek_pls->seek_stream_index = stream_subdemuxer_index;
2708
2709
0
    for (i = 0; i < c->n_playlists; i++) {
2710
        /* Reset reading */
2711
0
        struct playlist *pls = c->playlists[i];
2712
0
        AVIOContext *const pb = &pls->pb.pub;
2713
0
        ff_format_io_close(pls->parent, &pls->input);
2714
0
        pls->input_read_done = 0;
2715
0
        ff_format_io_close(pls->parent, &pls->input_next);
2716
0
        pls->input_next_requested = 0;
2717
0
        av_packet_unref(pls->pkt);
2718
0
        pb->eof_reached = 0;
2719
        /* Clear any buffered data */
2720
0
        pb->buf_end = pb->buf_ptr = pb->buffer;
2721
        /* Reset the pos, to let the mpegts/mov demuxer know we've seeked. */
2722
0
        pb->pos = 0;
2723
        /* Flush the packet queue of the subdemuxer. */
2724
0
        if (pls->ctx)
2725
0
            ff_read_frame_flush(pls->ctx);
2726
0
        if (pls->is_subtitle)
2727
0
            avformat_close_input(&pls->ctx);
2728
2729
        /* Reset the init segment so it's re-fetched and served appropriately */
2730
0
        pls->cur_init_section = NULL;
2731
2732
0
        pls->seek_timestamp = seek_timestamp;
2733
0
        pls->seek_flags = flags;
2734
2735
0
        if (pls != seek_pls) {
2736
            /* set closest segment seq_no for playlists not handled above */
2737
0
            find_timestamp_in_playlist(c, pls, seek_timestamp, &pls->cur_seq_no, NULL);
2738
            /* seek the playlist to the given position without taking
2739
             * keyframes into account since this playlist does not have the
2740
             * specified stream where we should look for the keyframes */
2741
0
            pls->seek_stream_index = -1;
2742
0
            pls->seek_flags |= AVSEEK_FLAG_ANY;
2743
0
        }
2744
0
    }
2745
2746
0
    c->cur_timestamp = seek_timestamp;
2747
2748
0
    return 0;
2749
0
}
2750
2751
static int hls_probe(const AVProbeData *p)
2752
924k
{
2753
    /* Require #EXTM3U at the start, and either one of the ones below
2754
     * somewhere for a proper match. */
2755
924k
    if (strncmp(p->buf, "#EXTM3U", 7))
2756
921k
        return 0;
2757
2758
2.82k
    if (strstr(p->buf, "#EXT-X-STREAM-INF:")     ||
2759
2.82k
        strstr(p->buf, "#EXT-X-TARGETDURATION:") ||
2760
2.82k
        strstr(p->buf, "#EXT-X-MEDIA-SEQUENCE:")) {
2761
2762
1.10k
        int mime_ok = p->mime_type && !(
2763
0
            av_strcasecmp(p->mime_type, "application/vnd.apple.mpegurl") &&
2764
0
            av_strcasecmp(p->mime_type, "audio/mpegurl")
2765
0
            );
2766
2767
1.10k
        int mime_x = p->mime_type && !(
2768
0
            av_strcasecmp(p->mime_type, "audio/x-mpegurl") &&
2769
0
            av_strcasecmp(p->mime_type, "application/x-mpegurl")
2770
0
            );
2771
2772
1.10k
        if (!mime_ok &&
2773
1.10k
            !mime_x &&
2774
1.10k
            !av_match_ext    (p->filename, "m3u8,m3u") &&
2775
1.10k
             ff_match_url_ext(p->filename, "m3u8,m3u") <= 0) {
2776
1.10k
            av_log(NULL, AV_LOG_ERROR, "Not detecting m3u8/hls with non standard extension and non standard mime type\n");
2777
1.10k
            return 0;
2778
1.10k
        }
2779
0
        if (mime_x)
2780
0
            av_log(NULL, AV_LOG_WARNING, "mime type is not rfc8216 compliant\n");
2781
2782
0
        return AVPROBE_SCORE_MAX;
2783
1.10k
    }
2784
1.72k
    return 0;
2785
2.82k
}
2786
2787
#define OFFSET(x) offsetof(HLSContext, x)
2788
#define FLAGS AV_OPT_FLAG_DECODING_PARAM
2789
static const AVOption hls_options[] = {
2790
    {"live_start_index", "segment index to start live streams at (negative values are from the end)",
2791
        OFFSET(live_start_index), AV_OPT_TYPE_INT, {.i64 = -3}, INT_MIN, INT_MAX, FLAGS},
2792
    {"prefer_x_start", "prefer to use #EXT-X-START if it's in playlist instead of live_start_index",
2793
        OFFSET(prefer_x_start), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, FLAGS},
2794
    {"allowed_extensions", "List of file extensions that hls is allowed to access",
2795
        OFFSET(allowed_extensions), AV_OPT_TYPE_STRING,
2796
        {.str = "3gp,aac,avi,ac3,eac3,flac,mkv,m3u8,m4a,m4s,m4v,mpg,mov,mp2,mp3,mp4,mpeg,mpegts,ogg,ogv,oga,ts,vob,vtt,wav,webvtt"
2797
            ",cmfv,cmfa" // Ticket11526 www.nicovideo.jp
2798
            ",ec3"       // part of Ticket11435 (Elisa Viihde (Finnish online recording service))
2799
            ",fmp4"      // https://github.com/yt-dlp/yt-dlp/issues/12700
2800
        },
2801
        INT_MIN, INT_MAX, FLAGS},
2802
    {"allowed_segment_extensions", "List of file extensions that hls is allowed to access",
2803
        OFFSET(allowed_segment_extensions), AV_OPT_TYPE_STRING,
2804
        {.str = "3gp,aac,avi,ac3,eac3,flac,mkv,m3u8,m4a,m4s,m4v,mpg,mov,mp2,mp3,mp4,mpeg,mpegts,ogg,ogv,oga,ts,vob,vtt,wav,webvtt"
2805
            ",cmfv,cmfa" // Ticket11526 www.nicovideo.jp
2806
            ",ec3"       // part of Ticket11435 (Elisa Viihde (Finnish online recording service))
2807
            ",fmp4"      // https://github.com/yt-dlp/yt-dlp/issues/12700
2808
            ",html"      // https://flash1.bogulus.cfd/
2809
        },
2810
        INT_MIN, INT_MAX, FLAGS},
2811
    {"extension_picky", "Be picky with all extensions matching",
2812
        OFFSET(extension_picky), AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, FLAGS},
2813
    {"max_reload", "Maximum number of times a insufficient list is attempted to be reloaded",
2814
        OFFSET(max_reload), AV_OPT_TYPE_INT, {.i64 = 100}, 0, INT_MAX, FLAGS},
2815
    {"m3u8_hold_counters", "The maximum number of times to load m3u8 when it refreshes without new segments",
2816
        OFFSET(m3u8_hold_counters), AV_OPT_TYPE_INT, {.i64 = 1000}, 0, INT_MAX, FLAGS},
2817
    {"http_persistent", "Use persistent HTTP connections",
2818
        OFFSET(http_persistent), AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, FLAGS },
2819
    {"http_multiple", "Use multiple HTTP connections for fetching segments",
2820
        OFFSET(http_multiple), AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1, FLAGS},
2821
    {"http_seekable", "Use HTTP partial requests, 0 = disable, 1 = enable, -1 = auto",
2822
        OFFSET(http_seekable), AV_OPT_TYPE_BOOL, { .i64 = -1}, -1, 1, FLAGS},
2823
    {"seg_format_options", "Set options for segment demuxer",
2824
        OFFSET(seg_format_opts), AV_OPT_TYPE_DICT, {.str = NULL}, 0, 0, FLAGS},
2825
    {"seg_max_retry", "Maximum number of times to reload a segment on error.",
2826
     OFFSET(seg_max_retry), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, FLAGS},
2827
    {NULL}
2828
};
2829
2830
static const AVClass hls_class = {
2831
    .class_name = "hls demuxer",
2832
    .item_name  = av_default_item_name,
2833
    .option     = hls_options,
2834
    .version    = LIBAVUTIL_VERSION_INT,
2835
};
2836
2837
const FFInputFormat ff_hls_demuxer = {
2838
    .p.name         = "hls",
2839
    .p.long_name    = NULL_IF_CONFIG_SMALL("Apple HTTP Live Streaming"),
2840
    .p.priv_class   = &hls_class,
2841
    .p.flags        = AVFMT_NOGENSEARCH | AVFMT_TS_DISCONT | AVFMT_NO_BYTE_SEEK,
2842
    .priv_data_size = sizeof(HLSContext),
2843
    .flags_internal = FF_INFMT_FLAG_INIT_CLEANUP,
2844
    .read_probe     = hls_probe,
2845
    .read_header    = hls_read_header,
2846
    .read_packet    = hls_read_packet,
2847
    .read_close     = hls_close,
2848
    .read_seek      = hls_read_seek,
2849
};