FFmpeg coverage

Directory: ../../../ffmpeg/
File: src/libavformat/webmdashenc.c
Date: 2026-10-14 14:16:27
Exec Total Coverage
Lines: 276 311 88.7%
Functions: 15 16 93.8%
Branches: 188 244 77.0%
Line Branch Exec Source
1 /*
2 * WebM DASH Manifest XML muxer
3 * Copyright (c) 2014 Vignesh Venkatasubramanian
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 /*
23 * WebM DASH Specification:
24 * https://sites.google.com/a/webmproject.org/wiki/adaptive-streaming/webm-dash-specification
25 * ISO DASH Specification:
26 * http://standards.iso.org/ittf/PubliclyAvailableStandards/c065274_ISO_IEC_23009-1_2014.zip
27 */
28
29 #include <float.h>
30 #include <stdint.h>
31 #include <string.h>
32 #include <time.h>
33
34 #include "avformat.h"
35 #include "matroska.h"
36 #include "mux.h"
37
38 #include "libavutil/avstring.h"
39 #include "libavutil/dict.h"
40 #include "libavutil/mem.h"
41 #include "libavutil/opt.h"
42 #include "libavutil/time_internal.h"
43
44 #include "libavcodec/codec_desc.h"
45
46 typedef struct AdaptationSet {
47 char id[10];
48 int *streams;
49 int nb_streams;
50 } AdaptationSet;
51
52 typedef struct WebMDashMuxContext {
53 const AVClass *class;
54 char *adaptation_sets;
55 AdaptationSet *as;
56 int nb_as;
57 int representation_id;
58 int is_live;
59 int chunk_start_index;
60 int chunk_duration;
61 char *utc_timing_url;
62 double time_shift_buffer_depth;
63 int minimum_update_period;
64 AVDictionary **escaped_metadata;
65 } WebMDashMuxContext;
66
67 13 static const char *get_codec_name(int codec_id)
68 {
69 13 return avcodec_descriptor_get(codec_id)->name;
70 }
71
72 8 static double get_duration(AVFormatContext *s)
73 {
74 8 int i = 0;
75 8 double max = 0.0;
76
2/2
✓ Branch 0 taken 20 times.
✓ Branch 1 taken 8 times.
 28 for (i = 0; i < s->nb_streams; i++) {
77 20 AVDictionaryEntry *duration = av_dict_get(s->streams[i]->metadata,
78 DURATION, NULL, 0);
79
2/4
✓ Branch 0 taken 20 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 20 times.
 20 if (!duration || atof(duration->value) < 0) continue;
80
2/2
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 10 times.
 20 if (atof(duration->value) > max) max = atof(duration->value);
81 }
82 8 return max / 1000;
83 }
84
85 6 static int write_header(AVFormatContext *s)
86 {
87 6 WebMDashMuxContext *w = s->priv_data;
88 6 AVIOContext *pb = s->pb;
89 6 double min_buffer_time = 1.0;
90 6 avio_printf(pb, "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
91 6 avio_printf(pb, "<MPD\n");
92 6 avio_printf(pb, " xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n");
93 6 avio_printf(pb, " xmlns=\"urn:mpeg:DASH:schema:MPD:2011\"\n");
94 6 avio_printf(pb, " xsi:schemaLocation=\"urn:mpeg:DASH:schema:MPD:2011\"\n");
95
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 4 times.
 6 avio_printf(pb, " type=\"%s\"\n", w->is_live ? "dynamic" : "static");
96
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 2 times.
 6 if (!w->is_live) {
97 4 avio_printf(pb, " mediaPresentationDuration=\"PT%gS\"\n",
98 get_duration(s));
99 }
100 6 avio_printf(pb, " minBufferTime=\"PT%gS\"\n", min_buffer_time);
101 12 avio_printf(pb, " profiles=\"%s\"%s",
102
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 4 times.
 6 w->is_live ? "urn:mpeg:dash:profile:isoff-live:2011" : "urn:mpeg:dash:profile:webm-on-demand:2012",
103
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 4 times.
 6 w->is_live ? "\n" : ">\n");
104
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 4 times.
 6 if (w->is_live) {
105 2 time_t local_time = time(NULL);
106 struct tm gmt_buffer;
107 2 struct tm *gmt = gmtime_r(&local_time, &gmt_buffer);
108 char gmt_iso[21];
109
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (!strftime(gmt_iso, 21, "%Y-%m-%dT%H:%M:%SZ", gmt)) {
110 ✗ return AVERROR_UNKNOWN;
111 }
112
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 if (s->flags & AVFMT_FLAG_BITEXACT) {
113 2 av_strlcpy(gmt_iso, "", 1);
114 }
115 2 avio_printf(pb, " availabilityStartTime=\"%s\"\n", gmt_iso);
116 2 avio_printf(pb, " timeShiftBufferDepth=\"PT%gS\"\n", w->time_shift_buffer_depth);
117 2 avio_printf(pb, " minimumUpdatePeriod=\"PT%dS\"", w->minimum_update_period);
118 2 avio_printf(pb, ">\n");
119
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (w->utc_timing_url) {
120 ✗ avio_printf(pb, "<UTCTiming\n");
121 ✗ avio_printf(pb, " schemeIdUri=\"urn:mpeg:dash:utc:http-iso:2014\"\n");
122 ✗ avio_printf(pb, " value=\"%s\"/>\n", w->utc_timing_url);
123 }
124 }
125 6 return 0;
126 }
127
128 6 static void write_footer(AVFormatContext *s)
129 {
130 6 avio_printf(s->pb, "</MPD>\n");
131 6 }
132
133 5 static int subsegment_alignment(AVFormatContext *s, const AdaptationSet *as)
134 {
135 int i;
136 5 AVDictionaryEntry *gold = av_dict_get(s->streams[as->streams[0]]->metadata,
137 CUE_TIMESTAMPS, NULL, 0);
138
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 5 times.
 5 if (!gold) return 0;
139
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 1 times.
 6 for (i = 1; i < as->nb_streams; i++) {
140 5 AVDictionaryEntry *ts = av_dict_get(s->streams[as->streams[i]]->metadata,
141 CUE_TIMESTAMPS, NULL, 0);
142
3/4
✓ Branch 0 taken 5 times.
✗ Branch 1 not taken.
✓ Branch 3 taken 4 times.
✓ Branch 4 taken 1 times.
 5 if (!ts || !av_strstart(ts->value, gold->value, NULL)) return 0;
143 }
144 1 return 1;
145 }
146
147 9 static int bitstream_switching(AVFormatContext *s, const AdaptationSet *as)
148 {
149 int i;
150 9 const AVStream *gold_st = s->streams[as->streams[0]];
151 9 AVDictionaryEntry *gold_track_num = av_dict_get(gold_st->metadata,
152 TRACK_NUMBER, NULL, 0);
153 9 AVCodecParameters *gold_par = gold_st->codecpar;
154
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
 9 if (!gold_track_num) return 0;
155
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 8 times.
 13 for (i = 1; i < as->nb_streams; i++) {
156 5 const AVStream *st = s->streams[as->streams[i]];
157 5 AVDictionaryEntry *track_num = av_dict_get(st->metadata,
158 TRACK_NUMBER, NULL, 0);
159 5 AVCodecParameters *par = st->codecpar;
160
3/4
✓ Branch 0 taken 5 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 1 times.
 10 if (!track_num ||
161 5 !av_strstart(track_num->value, gold_track_num->value, NULL) ||
162
1/2
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
 4 gold_par->codec_id != par->codec_id ||
163
1/2
✓ Branch 0 taken 4 times.
✗ Branch 1 not taken.
 4 gold_par->extradata_size != par->extradata_size ||
164
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 3 times.
 4 (par->extradata_size > 0 &&
165
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
 1 memcmp(gold_par->extradata, par->extradata, par->extradata_size))) {
166 1 return 0;
167 }
168 }
169 8 return 1;
170 }
171
172 /*
173 * Writes a Representation within an Adaptation Set. Returns 0 on success and
174 * < 0 on failure.
175 */
176 14 static int write_representation(AVFormatContext *s, AVStream *st, char *id,
177 int output_width, int output_height,
178 int output_sample_rate)
179 {
180 14 WebMDashMuxContext *w = s->priv_data;
181 14 AVIOContext *pb = s->pb;
182 14 const AVCodecParameters *par = st->codecpar;
183 14 AVDictionary *metadata = w->escaped_metadata[st->index];
184 14 AVDictionaryEntry *bandwidth = av_dict_get(metadata, BANDWIDTH, NULL, 0);
185 const char *bandwidth_str;
186 14 avio_printf(pb, "<Representation id=\"%s\"", id);
187
2/2
✓ Branch 0 taken 12 times.
✓ Branch 1 taken 2 times.
 14 if (bandwidth) {
188 12 bandwidth_str = bandwidth->value;
189
1/2
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
 2 } else if (w->is_live) {
190 // if bandwidth for live was not provided, use a default
191
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 1 times.
 2 bandwidth_str = (par->codec_type == AVMEDIA_TYPE_AUDIO) ? "128000" : "1000000";
192 } else {
193 ✗ return AVERROR(EINVAL);
194 }
195 14 avio_printf(pb, " bandwidth=\"%s\"", bandwidth_str);
196
4/4
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 4 times.
 14 if (par->codec_type == AVMEDIA_TYPE_VIDEO && output_width)
197 4 avio_printf(pb, " width=\"%d\"", par->width);
198
4/4
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 4 times.
✓ Branch 3 taken 4 times.
 14 if (par->codec_type == AVMEDIA_TYPE_VIDEO && output_height)
199 4 avio_printf(pb, " height=\"%d\"", par->height);
200
4/4
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 8 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 4 times.
 14 if (par->codec_type == AVMEDIA_TYPE_AUDIO && output_sample_rate)
201 2 avio_printf(pb, " audioSamplingRate=\"%d\"", par->sample_rate);
202
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 10 times.
 14 if (w->is_live) {
203 // For live streams, Codec and Mime Type always go in the Representation tag.
204 4 avio_printf(pb, " codecs=\"%s\"", get_codec_name(par->codec_id));
205 4 avio_printf(pb, " mimeType=\"%s/webm\"",
206
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
 4 par->codec_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio");
207 // For live streams, subsegments always start with key frames. So this
208 // is always 1.
209 4 avio_printf(pb, " startsWithSAP=\"1\"");
210 4 avio_printf(pb, ">");
211 } else {
212 10 AVDictionaryEntry *irange = av_dict_get(metadata, INITIALIZATION_RANGE, NULL, 0);
213 10 AVDictionaryEntry *cues_start = av_dict_get(metadata, CUES_START, NULL, 0);
214 10 AVDictionaryEntry *cues_end = av_dict_get(metadata, CUES_END, NULL, 0);
215 10 AVDictionaryEntry *filename = av_dict_get(metadata, FILENAME, NULL, 0);
216
4/8
✓ Branch 0 taken 10 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 10 times.
✗ Branch 5 not taken.
✗ Branch 6 not taken.
✓ Branch 7 taken 10 times.
 10 if (!irange || !cues_start || !cues_end || !filename)
217 ✗ return AVERROR(EINVAL);
218
219 10 avio_printf(pb, ">\n");
220 10 avio_printf(pb, "<BaseURL>%s</BaseURL>\n", filename->value);
221 10 avio_printf(pb, "<SegmentBase\n");
222 10 avio_printf(pb, " indexRange=\"%s-%s\">\n", cues_start->value, cues_end->value);
223 10 avio_printf(pb, "<Initialization\n");
224 10 avio_printf(pb, " range=\"0-%s\" />\n", irange->value);
225 10 avio_printf(pb, "</SegmentBase>\n");
226 }
227 14 avio_printf(pb, "</Representation>\n");
228 14 return 0;
229 }
230
231 /*
232 * Checks if width of all streams are the same. Returns 1 if true, 0 otherwise.
233 */
234 3 static int check_matching_width(AVFormatContext *s, const AdaptationSet *as)
235 {
236 int first_width, i;
237
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
 3 if (as->nb_streams < 2) return 1;
238 3 first_width = s->streams[as->streams[0]]->codecpar->width;
239
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 2 times.
 5 for (i = 1; i < as->nb_streams; i++)
240
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2 times.
 3 if (first_width != s->streams[as->streams[i]]->codecpar->width)
241 1 return 0;
242 2 return 1;
243 }
244
245 /*
246 * Checks if height of all streams are the same. Returns 1 if true, 0 otherwise.
247 */
248 3 static int check_matching_height(AVFormatContext *s, const AdaptationSet *as)
249 {
250 int first_height, i;
251
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
 3 if (as->nb_streams < 2) return 1;
252 3 first_height = s->streams[as->streams[0]]->codecpar->height;
253
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 2 times.
 5 for (i = 1; i < as->nb_streams; i++)
254
2/2
✓ Branch 0 taken 1 times.
✓ Branch 1 taken 2 times.
 3 if (first_height != s->streams[as->streams[i]]->codecpar->height)
255 1 return 0;
256 2 return 1;
257 }
258
259 /*
260 * Checks if sample rate of all streams are the same. Returns 1 if true, 0 otherwise.
261 */
262 2 static int check_matching_sample_rate(AVFormatContext *s, const AdaptationSet *as)
263 {
264 int first_sample_rate, i;
265
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (as->nb_streams < 2) return 1;
266 2 first_sample_rate = s->streams[as->streams[0]]->codecpar->sample_rate;
267
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
 4 for (i = 1; i < as->nb_streams; i++)
268
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
 2 if (first_sample_rate != s->streams[as->streams[i]]->codecpar->sample_rate)
269 ✗ return 0;
270 2 return 1;
271 }
272
273 6 static void free_adaptation_sets(AVFormatContext *s)
274 {
275 6 WebMDashMuxContext *w = s->priv_data;
276 int i;
277
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 6 times.
 15 for (i = 0; i < w->nb_as; i++) {
278 9 av_freep(&w->as[i].streams);
279 }
280 6 av_freep(&w->as);
281 6 w->nb_as = 0;
282 6 }
283
284 /*
285 * Parses a live header filename and returns the position of the '_' and '.'
286 * delimiting <file_description> and <representation_id>.
287 *
288 * Name of the header file should conform to the following pattern:
289 * <file_description>_<representation_id>.hdr where <file_description> can be
290 * anything. The chunks should be named according to the following pattern:
291 * <file_description>_<representation_id>_<chunk_number>.chk
292 */
293 8 static int split_filename(char *filename, char **underscore_pos,
294 char **period_pos)
295 {
296 8 *underscore_pos = strrchr(filename, '_');
297
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
 8 if (!*underscore_pos)
298 ✗ return AVERROR(EINVAL);
299 8 *period_pos = strchr(*underscore_pos, '.');
300
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
 8 if (!*period_pos)
301 ✗ return AVERROR(EINVAL);
302 8 return 0;
303 }
304
305 /*
306 * Writes an Adaptation Set. Returns 0 on success and < 0 on failure.
307 */
308 9 static int write_adaptation_set(AVFormatContext *s, int as_index)
309 {
310 9 WebMDashMuxContext *w = s->priv_data;
311 9 AdaptationSet *as = &w->as[as_index];
312 9 const AVStream *st = s->streams[as->streams[0]];
313 9 AVCodecParameters *par = st->codecpar;
314 AVDictionaryEntry *lang;
315 9 AVIOContext *pb = s->pb;
316 int i;
317 static const char boolean[2][6] = { "false", "true" };
318 9 int subsegmentStartsWithSAP = 1;
319
320 // Width, Height and Sample Rate will go in the AdaptationSet tag if they
321 // are the same for all contained Representations. otherwise, they will go
322 // on their respective Representation tag. For live streams, they always go
323 // in the Representation tag.
324 9 int width_in_as = 1, height_in_as = 1, sample_rate_in_as = 1;
325
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 4 times.
 9 if (par->codec_type == AVMEDIA_TYPE_VIDEO) {
326
4/4
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 2 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 1 times.
 5 width_in_as = !w->is_live && check_matching_width (s, as);
327
4/4
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 2 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 1 times.
 5 height_in_as = !w->is_live && check_matching_height(s, as);
328 } else {
329
3/4
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
✓ Branch 3 taken 2 times.
✗ Branch 4 not taken.
 4 sample_rate_in_as = !w->is_live && check_matching_sample_rate(s, as);
330 }
331
332 9 avio_printf(pb, "<AdaptationSet id=\"%s\"", as->id);
333 9 avio_printf(pb, " mimeType=\"%s/webm\"",
334
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 4 times.
 9 par->codec_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio");
335 9 avio_printf(pb, " codecs=\"%s\"", get_codec_name(par->codec_id));
336
337 9 lang = av_dict_get(w->escaped_metadata[st->index], "language", NULL, 0);
338
2/2
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 4 times.
 9 if (lang)
339 5 avio_printf(pb, " lang=\"%s\"", lang->value);
340
341
4/4
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 3 times.
 9 if (par->codec_type == AVMEDIA_TYPE_VIDEO && width_in_as)
342 2 avio_printf(pb, " width=\"%d\"", par->width);
343
4/4
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 3 times.
 9 if (par->codec_type == AVMEDIA_TYPE_VIDEO && height_in_as)
344 2 avio_printf(pb, " height=\"%d\"", par->height);
345
4/4
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 5 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 2 times.
 9 if (par->codec_type == AVMEDIA_TYPE_AUDIO && sample_rate_in_as)
346 2 avio_printf(pb, " audioSamplingRate=\"%d\"", par->sample_rate);
347
348 9 avio_printf(pb, " bitstreamSwitching=\"%s\"",
349 9 boolean[bitstream_switching(s, as)]);
350 9 avio_printf(pb, " subsegmentAlignment=\"%s\"",
351
4/4
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 4 times.
✓ Branch 3 taken 1 times.
✓ Branch 4 taken 4 times.
 9 boolean[w->is_live || subsegment_alignment(s, as)]);
352
353
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 9 times.
 23 for (i = 0; i < as->nb_streams; i++) {
354 14 AVDictionaryEntry *kf = av_dict_get(s->streams[as->streams[i]]->metadata,
355 CLUSTER_KEYFRAME, NULL, 0);
356
5/6
✓ Branch 0 taken 10 times.
✓ Branch 1 taken 4 times.
✓ Branch 2 taken 10 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 1 times.
✓ Branch 5 taken 9 times.
 14 if (!w->is_live && (!kf || !strncmp(kf->value, "0", 1))) subsegmentStartsWithSAP = 0;
357 }
358 9 avio_printf(pb, " subsegmentStartsWithSAP=\"%d\"", subsegmentStartsWithSAP);
359 9 avio_printf(pb, ">\n");
360
361
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 5 times.
 9 if (w->is_live) {
362 AVDictionaryEntry *filename =
363 4 av_dict_get(w->escaped_metadata[st->index], FILENAME, NULL, 0);
364 char *underscore_pos, *period_pos;
365 int ret;
366
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (!filename)
367 ✗ return AVERROR(EINVAL);
368 4 ret = split_filename(filename->value, &underscore_pos, &period_pos);
369
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (ret) return ret;
370 4 *underscore_pos = '\0';
371 4 avio_printf(pb, "<ContentComponent id=\"1\" type=\"%s\"/>\n",
372
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 2 times.
 4 par->codec_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio");
373 4 avio_printf(pb, "<SegmentTemplate");
374 4 avio_printf(pb, " timescale=\"1000\"");
375 4 avio_printf(pb, " duration=\"%d\"", w->chunk_duration);
376 4 avio_printf(pb, " media=\"%s_$RepresentationID$_$Number$.chk\"",
377 filename->value);
378 4 avio_printf(pb, " startNumber=\"%d\"", w->chunk_start_index);
379 4 avio_printf(pb, " initialization=\"%s_$RepresentationID$.hdr\"",
380 filename->value);
381 4 avio_printf(pb, "/>\n");
382 4 *underscore_pos = '_';
383 }
384
385
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 9 times.
 23 for (i = 0; i < as->nb_streams; i++) {
386 14 char buf[25], *representation_id = buf, *underscore_pos, *period_pos;
387 14 AVStream *st = s->streams[as->streams[i]];
388 int ret;
389
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 10 times.
 14 if (w->is_live) {
390 AVDictionaryEntry *filename =
391 4 av_dict_get(w->escaped_metadata[st->index], FILENAME, NULL, 0);
392
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (!filename)
393 ✗ return AVERROR(EINVAL);
394 4 ret = split_filename(filename->value, &underscore_pos, &period_pos);
395
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 4 times.
 4 if (ret < 0)
396 ✗ return ret;
397 4 representation_id = underscore_pos + 1;
398 4 *period_pos = '\0';
399 } else {
400 10 snprintf(buf, sizeof(buf), "%d", w->representation_id++);
401 }
402 14 ret = write_representation(s, st, representation_id, !width_in_as,
403 !height_in_as, !sample_rate_in_as);
404
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
 14 if (ret) return ret;
405
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 10 times.
 14 if (w->is_live)
406 4 *period_pos = '.';
407 }
408 9 avio_printf(s->pb, "</AdaptationSet>\n");
409 9 return 0;
410 }
411
412 6 static int parse_adaptation_sets(AVFormatContext *s)
413 {
414 6 WebMDashMuxContext *w = s->priv_data;
415 6 char *p = w->adaptation_sets;
416 char *q;
417 enum { new_set, parsed_id, parsing_streams } state;
418
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (!w->adaptation_sets) {
419 ✗ av_log(s, AV_LOG_ERROR, "The 'adaptation_sets' option must be set.\n");
420 ✗ return AVERROR(EINVAL);
421 }
422 // syntax id=0,streams=0,1,2 id=1,streams=3,4 and so on
423 6 state = new_set;
424 while (1) {
425
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 32 times.
 32 if (*p == '\0') {
426 ✗ if (state == new_set)
427 ✗ break;
428 else
429 ✗ return AVERROR(EINVAL);
430
3/4
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 23 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 9 times.
 32 } else if (state == new_set && *p == ' ') {
431 ✗ p++;
432 ✗ continue;
433
3/4
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 23 times.
✓ Branch 2 taken 9 times.
✗ Branch 3 not taken.
 32 } else if (state == new_set && !strncmp(p, "id=", 3)) {
434 9 void *mem = av_realloc(w->as, sizeof(*w->as) * (w->nb_as + 1));
435 const char *comma;
436
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
 9 if (mem == NULL)
437 ✗ return AVERROR(ENOMEM);
438 9 w->as = mem;
439 9 ++w->nb_as;
440 9 w->as[w->nb_as - 1].nb_streams = 0;
441 9 w->as[w->nb_as - 1].streams = NULL;
442 9 p += 3; // consume "id="
443 9 q = w->as[w->nb_as - 1].id;
444 9 comma = strchr(p, ',');
445
2/4
✓ Branch 0 taken 9 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 9 times.
 9 if (!comma || comma - p >= sizeof(w->as[w->nb_as - 1].id)) {
446 ✗ av_log(s, AV_LOG_ERROR, "'id' in 'adaptation_sets' is malformed.\n");
447 ✗ return AVERROR(EINVAL);
448 }
449
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 9 times.
 18 while (*p != ',') *q++ = *p++;
450 9 *q = 0;
451 9 p++;
452 9 state = parsed_id;
453
3/4
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 14 times.
✓ Branch 2 taken 9 times.
✗ Branch 3 not taken.
 23 } else if (state == parsed_id && !strncmp(p, "streams=", 8)) {
454 9 p += 8; // consume "streams="
455 9 state = parsing_streams;
456
1/2
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
 14 } else if (state == parsing_streams) {
457 14 struct AdaptationSet *as = &w->as[w->nb_as - 1];
458 int64_t num;
459 14 int ret = av_reallocp_array(&as->streams, ++as->nb_streams,
460 sizeof(*as->streams));
461
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
 14 if (ret < 0)
462 ✗ return ret;
463 14 num = strtoll(p, &q, 10);
464
7/10
✓ Branch 0 taken 14 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 11 times.
✓ Branch 3 taken 3 times.
✓ Branch 4 taken 5 times.
✓ Branch 5 taken 6 times.
✓ Branch 6 taken 5 times.
✗ Branch 7 not taken.
✓ Branch 8 taken 14 times.
✗ Branch 9 not taken.
 14 if (!av_isdigit(*p) || (*q != ' ' && *q != '\0' && *q != ',') ||
465
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 14 times.
 14 num < 0 || num >= s->nb_streams) {
466 ✗ av_log(s, AV_LOG_ERROR, "Invalid value for 'streams' in adapation_sets.\n");
467 ✗ return AVERROR(EINVAL);
468 }
469 14 as->streams[as->nb_streams - 1] = num;
470
2/2
✓ Branch 0 taken 6 times.
✓ Branch 1 taken 8 times.
 14 if (*q == '\0') break;
471
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 5 times.
 8 if (*q == ' ') state = new_set;
472 8 p = ++q;
473 } else {
474 ✗ return -1;
475 }
476 }
477 6 return 0;
478 }
479
480 6 static int webm_dash_manifest_write_header(AVFormatContext *s)
481 {
482 int i;
483 6 double start = 0.0;
484 int ret;
485 6 WebMDashMuxContext *w = s->priv_data;
486
487
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 6 times.
 20 for (unsigned i = 0; i < s->nb_streams; i++) {
488 14 enum AVCodecID codec_id = s->streams[i]->codecpar->codec_id;
489
5/6
✓ Branch 0 taken 8 times.
✓ Branch 1 taken 6 times.
✓ Branch 2 taken 6 times.
✓ Branch 3 taken 2 times.
✓ Branch 4 taken 6 times.
✗ Branch 5 not taken.
 14 if (codec_id != AV_CODEC_ID_VP8 && codec_id != AV_CODEC_ID_VP9 &&
490
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
 6 codec_id != AV_CODEC_ID_AV1 && codec_id != AV_CODEC_ID_VORBIS &&
491 codec_id != AV_CODEC_ID_OPUS)
492 ✗ return AVERROR(EINVAL);
493 }
494
495 6 w->escaped_metadata = av_calloc(s->nb_streams, sizeof(*w->escaped_metadata));
496
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (!w->escaped_metadata)
497 ✗ return AVERROR(ENOMEM);
498
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 6 times.
 20 for (i = 0; i < s->nb_streams; i++) {
499 14 const AVDictionaryEntry *e = NULL;
500
2/2
✓ Branch 1 taken 110 times.
✓ Branch 2 taken 14 times.
 124 while ((e = av_dict_iterate(s->streams[i]->metadata, e))) {
501 char *value;
502 110 ret = av_escape(&value, e->value, NULL, AV_ESCAPE_MODE_XML,
503 AV_ESCAPE_FLAG_XML_DOUBLE_QUOTES);
504
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 110 times.
 110 if (ret < 0)
505 ✗ goto fail;
506 110 ret = av_dict_set(&w->escaped_metadata[i], e->key, value,
507 AV_DICT_DONT_STRDUP_VAL | AV_DICT_MULTIKEY);
508
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 110 times.
 110 if (ret < 0)
509 ✗ goto fail;
510 }
511 }
512
513 6 ret = parse_adaptation_sets(s);
514
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (ret < 0) {
515 ✗ goto fail;
516 }
517 6 ret = write_header(s);
518
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 6 times.
 6 if (ret < 0) {
519 ✗ goto fail;
520 }
521 6 avio_printf(s->pb, "<Period id=\"0\"");
522 6 avio_printf(s->pb, " start=\"PT%gS\"", start);
523
2/2
✓ Branch 0 taken 4 times.
✓ Branch 1 taken 2 times.
 6 if (!w->is_live) {
524 4 avio_printf(s->pb, " duration=\"PT%gS\"", get_duration(s));
525 }
526 6 avio_printf(s->pb, " >\n");
527
528
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 6 times.
 15 for (i = 0; i < w->nb_as; i++) {
529 9 ret = write_adaptation_set(s, i);
530
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9 times.
 9 if (ret < 0) {
531 ✗ goto fail;
532 }
533 }
534
535 6 avio_printf(s->pb, "</Period>\n");
536 6 write_footer(s);
537 6 fail:
538 6 free_adaptation_sets(s);
539
2/2
✓ Branch 0 taken 14 times.
✓ Branch 1 taken 6 times.
 20 for (i = 0; i < s->nb_streams; i++)
540 14 av_dict_free(&w->escaped_metadata[i]);
541 6 av_freep(&w->escaped_metadata);
542 6 return ret < 0 ? ret : 0;
543 }
544
545 ✗ static int webm_dash_manifest_write_packet(AVFormatContext *s, AVPacket *pkt)
546 {
547 ✗ return AVERROR_EOF;
548 }
549
550 #define OFFSET(x) offsetof(WebMDashMuxContext, x)
551 static const AVOption options[] = {
552 { "adaptation_sets", "Adaptation sets. Syntax: id=0,streams=0,1,2 id=1,streams=3,4 and so on", OFFSET(adaptation_sets), AV_OPT_TYPE_STRING, { 0 }, 0, 0, AV_OPT_FLAG_ENCODING_PARAM },
553 { "live", "create a live stream manifest", OFFSET(is_live), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM },
554 { "chunk_start_index", "start index of the chunk", OFFSET(chunk_start_index), AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM },
555 { "chunk_duration_ms", "duration of each chunk (in milliseconds)", OFFSET(chunk_duration), AV_OPT_TYPE_INT, {.i64 = 1000}, 0, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM },
556 { "utc_timing_url", "URL of the page that will return the UTC timestamp in ISO format", OFFSET(utc_timing_url), AV_OPT_TYPE_STRING, { 0 }, 0, 0, AV_OPT_FLAG_ENCODING_PARAM },
557 { "time_shift_buffer_depth", "Smallest time (in seconds) shifting buffer for which any Representation is guaranteed to be available.", OFFSET(time_shift_buffer_depth), AV_OPT_TYPE_DOUBLE, { .dbl = 60.0 }, 1.0, DBL_MAX, AV_OPT_FLAG_ENCODING_PARAM },
558 { "minimum_update_period", "Minimum Update Period (in seconds) of the manifest.", OFFSET(minimum_update_period), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM },
559 { NULL },
560 };
561
562 static const AVClass webm_dash_class = {
563 .class_name = "WebM DASH Manifest muxer",
564 .item_name = av_default_item_name,
565 .option = options,
566 .version = LIBAVUTIL_VERSION_INT,
567 };
568
569 const FFOutputFormat ff_webm_dash_manifest_muxer = {
570 .p.name = "webm_dash_manifest",
571 .p.long_name = NULL_IF_CONFIG_SMALL("WebM DASH Manifest"),
572 .p.mime_type = "application/xml",
573 .p.extensions = "xml",
574 .priv_data_size = sizeof(WebMDashMuxContext),
575 .write_header = webm_dash_manifest_write_header,
576 .write_packet = webm_dash_manifest_write_packet,
577 .p.priv_class = &webm_dash_class,
578 };
579