LCOV - code coverage report
Current view: top level - libavformat - webmdashenc.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 286 323 88.5 %
Date: 2017-12-18 13:19:42 Functions: 17 18 94.4 %

          Line data    Source code
       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             : 
      33             : #include "avformat.h"
      34             : #include "avio_internal.h"
      35             : #include "matroska.h"
      36             : 
      37             : #include "libavutil/avstring.h"
      38             : #include "libavutil/dict.h"
      39             : #include "libavutil/opt.h"
      40             : #include "libavutil/time_internal.h"
      41             : 
      42             : typedef struct AdaptationSet {
      43             :     char id[10];
      44             :     int *streams;
      45             :     int nb_streams;
      46             : } AdaptationSet;
      47             : 
      48             : typedef struct WebMDashMuxContext {
      49             :     const AVClass  *class;
      50             :     char *adaptation_sets;
      51             :     AdaptationSet *as;
      52             :     int nb_as;
      53             :     int representation_id;
      54             :     int is_live;
      55             :     int chunk_start_index;
      56             :     int chunk_duration;
      57             :     char *utc_timing_url;
      58             :     double time_shift_buffer_depth;
      59             :     int minimum_update_period;
      60             :     int debug_mode;
      61             : } WebMDashMuxContext;
      62             : 
      63          13 : static const char *get_codec_name(int codec_id)
      64             : {
      65          13 :     switch (codec_id) {
      66           3 :         case AV_CODEC_ID_VP8:
      67           3 :             return "vp8";
      68           4 :         case AV_CODEC_ID_VP9:
      69           4 :             return "vp9";
      70           6 :         case AV_CODEC_ID_VORBIS:
      71           6 :             return "vorbis";
      72           0 :         case AV_CODEC_ID_OPUS:
      73           0 :             return "opus";
      74             :     }
      75           0 :     return NULL;
      76             : }
      77             : 
      78           8 : static double get_duration(AVFormatContext *s)
      79             : {
      80           8 :     int i = 0;
      81           8 :     double max = 0.0;
      82          28 :     for (i = 0; i < s->nb_streams; i++) {
      83          20 :         AVDictionaryEntry *duration = av_dict_get(s->streams[i]->metadata,
      84             :                                                   DURATION, NULL, 0);
      85          20 :         if (!duration || atof(duration->value) < 0) continue;
      86          20 :         if (atof(duration->value) > max) max = atof(duration->value);
      87             :     }
      88           8 :     return max / 1000;
      89             : }
      90             : 
      91           6 : static int write_header(AVFormatContext *s)
      92             : {
      93           6 :     WebMDashMuxContext *w = s->priv_data;
      94           6 :     double min_buffer_time = 1.0;
      95           6 :     avio_printf(s->pb, "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
      96           6 :     avio_printf(s->pb, "<MPD\n");
      97           6 :     avio_printf(s->pb, "  xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n");
      98           6 :     avio_printf(s->pb, "  xmlns=\"urn:mpeg:DASH:schema:MPD:2011\"\n");
      99           6 :     avio_printf(s->pb, "  xsi:schemaLocation=\"urn:mpeg:DASH:schema:MPD:2011\"\n");
     100           6 :     avio_printf(s->pb, "  type=\"%s\"\n", w->is_live ? "dynamic" : "static");
     101           6 :     if (!w->is_live) {
     102           4 :         avio_printf(s->pb, "  mediaPresentationDuration=\"PT%gS\"\n",
     103             :                     get_duration(s));
     104             :     }
     105           6 :     avio_printf(s->pb, "  minBufferTime=\"PT%gS\"\n", min_buffer_time);
     106          12 :     avio_printf(s->pb, "  profiles=\"%s\"%s",
     107           6 :                 w->is_live ? "urn:mpeg:dash:profile:isoff-live:2011" : "urn:webm:dash:profile:webm-on-demand:2012",
     108           6 :                 w->is_live ? "\n" : ">\n");
     109           6 :     if (w->is_live) {
     110           2 :         time_t local_time = time(NULL);
     111             :         struct tm gmt_buffer;
     112           2 :         struct tm *gmt = gmtime_r(&local_time, &gmt_buffer);
     113             :         char gmt_iso[21];
     114           2 :         if (!strftime(gmt_iso, 21, "%Y-%m-%dT%H:%M:%SZ", gmt)) {
     115           0 :             return AVERROR_UNKNOWN;
     116             :         }
     117           2 :         if (w->debug_mode) {
     118           2 :             av_strlcpy(gmt_iso, "", 1);
     119             :         }
     120           2 :         avio_printf(s->pb, "  availabilityStartTime=\"%s\"\n", gmt_iso);
     121           2 :         avio_printf(s->pb, "  timeShiftBufferDepth=\"PT%gS\"\n", w->time_shift_buffer_depth);
     122           2 :         avio_printf(s->pb, "  minimumUpdatePeriod=\"PT%dS\"", w->minimum_update_period);
     123           2 :         avio_printf(s->pb, ">\n");
     124           2 :         if (w->utc_timing_url) {
     125           0 :             avio_printf(s->pb, "<UTCTiming\n");
     126           0 :             avio_printf(s->pb, "  schemeIdUri=\"urn:mpeg:dash:utc:http-iso:2014\"\n");
     127           0 :             avio_printf(s->pb, "  value=\"%s\"/>\n", w->utc_timing_url);
     128             :         }
     129             :     }
     130           6 :     return 0;
     131             : }
     132             : 
     133           6 : static void write_footer(AVFormatContext *s)
     134             : {
     135           6 :     avio_printf(s->pb, "</MPD>\n");
     136           6 : }
     137             : 
     138           5 : static int subsegment_alignment(AVFormatContext *s, AdaptationSet *as) {
     139             :     int i;
     140           5 :     AVDictionaryEntry *gold = av_dict_get(s->streams[as->streams[0]]->metadata,
     141             :                                           CUE_TIMESTAMPS, NULL, 0);
     142           5 :     if (!gold) return 0;
     143           6 :     for (i = 1; i < as->nb_streams; i++) {
     144           5 :         AVDictionaryEntry *ts = av_dict_get(s->streams[as->streams[i]]->metadata,
     145             :                                             CUE_TIMESTAMPS, NULL, 0);
     146           5 :         if (!ts || strncmp(gold->value, ts->value, strlen(gold->value))) return 0;
     147             :     }
     148           1 :     return 1;
     149             : }
     150             : 
     151           9 : static int bitstream_switching(AVFormatContext *s, AdaptationSet *as) {
     152             :     int i;
     153           9 :     AVDictionaryEntry *gold_track_num = av_dict_get(s->streams[as->streams[0]]->metadata,
     154             :                                                     TRACK_NUMBER, NULL, 0);
     155           9 :     AVCodecParameters *gold_par = s->streams[as->streams[0]]->codecpar;
     156           9 :     if (!gold_track_num) return 0;
     157          13 :     for (i = 1; i < as->nb_streams; i++) {
     158           5 :         AVDictionaryEntry *track_num = av_dict_get(s->streams[as->streams[i]]->metadata,
     159             :                                                    TRACK_NUMBER, NULL, 0);
     160           5 :         AVCodecParameters *par = s->streams[as->streams[i]]->codecpar;
     161          10 :         if (!track_num ||
     162           9 :             strncmp(gold_track_num->value, track_num->value, strlen(gold_track_num->value)) ||
     163           8 :             gold_par->codec_id != par->codec_id ||
     164           8 :             gold_par->extradata_size != par->extradata_size ||
     165           4 :             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 *stream, char *id,
     177             :                                 int output_width, int output_height,
     178             :                                 int output_sample_rate) {
     179          14 :     WebMDashMuxContext *w = s->priv_data;
     180          14 :     AVDictionaryEntry *irange = av_dict_get(stream->metadata, INITIALIZATION_RANGE, NULL, 0);
     181          14 :     AVDictionaryEntry *cues_start = av_dict_get(stream->metadata, CUES_START, NULL, 0);
     182          14 :     AVDictionaryEntry *cues_end = av_dict_get(stream->metadata, CUES_END, NULL, 0);
     183          14 :     AVDictionaryEntry *filename = av_dict_get(stream->metadata, FILENAME, NULL, 0);
     184          14 :     AVDictionaryEntry *bandwidth = av_dict_get(stream->metadata, BANDWIDTH, NULL, 0);
     185             :     const char *bandwidth_str;
     186          28 :     if ((w->is_live && (!filename)) ||
     187          24 :         (!w->is_live && (!irange || !cues_start || !cues_end || !filename || !bandwidth))) {
     188           0 :         return AVERROR_INVALIDDATA;
     189             :     }
     190          14 :     avio_printf(s->pb, "<Representation id=\"%s\"", id);
     191             :     // if bandwidth for live was not provided, use a default
     192          14 :     if (w->is_live && !bandwidth) {
     193           2 :         bandwidth_str = (stream->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) ? "128000" : "1000000";
     194             :     } else {
     195          12 :         bandwidth_str = bandwidth->value;
     196             :     }
     197          14 :     avio_printf(s->pb, " bandwidth=\"%s\"", bandwidth_str);
     198          14 :     if (stream->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && output_width)
     199           4 :         avio_printf(s->pb, " width=\"%d\"", stream->codecpar->width);
     200          14 :     if (stream->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && output_height)
     201           4 :         avio_printf(s->pb, " height=\"%d\"", stream->codecpar->height);
     202          14 :     if (stream->codecpar->codec_type == AVMEDIA_TYPE_AUDIO && output_sample_rate)
     203           2 :         avio_printf(s->pb, " audioSamplingRate=\"%d\"", stream->codecpar->sample_rate);
     204          14 :     if (w->is_live) {
     205             :         // For live streams, Codec and Mime Type always go in the Representation tag.
     206           4 :         avio_printf(s->pb, " codecs=\"%s\"", get_codec_name(stream->codecpar->codec_id));
     207           4 :         avio_printf(s->pb, " mimeType=\"%s/webm\"",
     208           4 :                     stream->codecpar->codec_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio");
     209             :         // For live streams, subsegments always start with key frames. So this
     210             :         // is always 1.
     211           4 :         avio_printf(s->pb, " startsWithSAP=\"1\"");
     212           4 :         avio_printf(s->pb, ">");
     213             :     } else {
     214          10 :         avio_printf(s->pb, ">\n");
     215          10 :         avio_printf(s->pb, "<BaseURL>%s</BaseURL>\n", filename->value);
     216          10 :         avio_printf(s->pb, "<SegmentBase\n");
     217          10 :         avio_printf(s->pb, "  indexRange=\"%s-%s\">\n", cues_start->value, cues_end->value);
     218          10 :         avio_printf(s->pb, "<Initialization\n");
     219          10 :         avio_printf(s->pb, "  range=\"0-%s\" />\n", irange->value);
     220          10 :         avio_printf(s->pb, "</SegmentBase>\n");
     221             :     }
     222          14 :     avio_printf(s->pb, "</Representation>\n");
     223          14 :     return 0;
     224             : }
     225             : 
     226             : /*
     227             :  * Checks if width of all streams are the same. Returns 1 if true, 0 otherwise.
     228             :  */
     229           3 : static int check_matching_width(AVFormatContext *s, AdaptationSet *as) {
     230             :     int first_width, i;
     231           3 :     if (as->nb_streams < 2) return 1;
     232           3 :     first_width = s->streams[as->streams[0]]->codecpar->width;
     233           5 :     for (i = 1; i < as->nb_streams; i++)
     234           3 :         if (first_width != s->streams[as->streams[i]]->codecpar->width)
     235           1 :           return 0;
     236           2 :     return 1;
     237             : }
     238             : 
     239             : /*
     240             :  * Checks if height of all streams are the same. Returns 1 if true, 0 otherwise.
     241             :  */
     242           3 : static int check_matching_height(AVFormatContext *s, AdaptationSet *as) {
     243             :     int first_height, i;
     244           3 :     if (as->nb_streams < 2) return 1;
     245           3 :     first_height = s->streams[as->streams[0]]->codecpar->height;
     246           5 :     for (i = 1; i < as->nb_streams; i++)
     247           3 :         if (first_height != s->streams[as->streams[i]]->codecpar->height)
     248           1 :           return 0;
     249           2 :     return 1;
     250             : }
     251             : 
     252             : /*
     253             :  * Checks if sample rate of all streams are the same. Returns 1 if true, 0 otherwise.
     254             :  */
     255           2 : static int check_matching_sample_rate(AVFormatContext *s, AdaptationSet *as) {
     256             :     int first_sample_rate, i;
     257           2 :     if (as->nb_streams < 2) return 1;
     258           2 :     first_sample_rate = s->streams[as->streams[0]]->codecpar->sample_rate;
     259           4 :     for (i = 1; i < as->nb_streams; i++)
     260           2 :         if (first_sample_rate != s->streams[as->streams[i]]->codecpar->sample_rate)
     261           0 :           return 0;
     262           2 :     return 1;
     263             : }
     264             : 
     265          12 : static void free_adaptation_sets(AVFormatContext *s) {
     266          12 :     WebMDashMuxContext *w = s->priv_data;
     267             :     int i;
     268          21 :     for (i = 0; i < w->nb_as; i++) {
     269           9 :         av_freep(&w->as[i].streams);
     270             :     }
     271          12 :     av_freep(&w->as);
     272          12 :     w->nb_as = 0;
     273          12 : }
     274             : 
     275             : /*
     276             :  * Parses a live header filename and computes the representation id,
     277             :  * initialization pattern and the media pattern. Pass NULL if you don't want to
     278             :  * compute any of those 3. Returns 0 on success and non-zero on failure.
     279             :  *
     280             :  * Name of the header file should conform to the following pattern:
     281             :  * <file_description>_<representation_id>.hdr where <file_description> can be
     282             :  * anything. The chunks should be named according to the following pattern:
     283             :  * <file_description>_<representation_id>_<chunk_number>.chk
     284             :  */
     285           8 : static int parse_filename(char *filename, char **representation_id,
     286             :                           char **initialization_pattern, char **media_pattern) {
     287           8 :     char *underscore_pos = NULL;
     288           8 :     char *period_pos = NULL;
     289           8 :     char *temp_pos = NULL;
     290           8 :     char *filename_str = av_strdup(filename);
     291           8 :     int ret = 0;
     292             : 
     293           8 :     if (!filename_str) {
     294           0 :         ret = AVERROR(ENOMEM);
     295           0 :         goto end;
     296             :     }
     297           8 :     temp_pos = av_stristr(filename_str, "_");
     298          40 :     while (temp_pos) {
     299          24 :         underscore_pos = temp_pos + 1;
     300          24 :         temp_pos = av_stristr(temp_pos + 1, "_");
     301             :     }
     302           8 :     if (!underscore_pos) {
     303           0 :         ret = AVERROR_INVALIDDATA;
     304           0 :         goto end;
     305             :     }
     306           8 :     period_pos = av_stristr(underscore_pos, ".");
     307           8 :     if (!period_pos) {
     308           0 :         ret = AVERROR_INVALIDDATA;
     309           0 :         goto end;
     310             :     }
     311           8 :     *(underscore_pos - 1) = 0;
     312           8 :     if (representation_id) {
     313           4 :         *representation_id = av_malloc(period_pos - underscore_pos + 1);
     314           4 :         if (!(*representation_id)) {
     315           0 :             ret = AVERROR(ENOMEM);
     316           0 :             goto end;
     317             :         }
     318           4 :         av_strlcpy(*representation_id, underscore_pos, period_pos - underscore_pos + 1);
     319             :     }
     320           8 :     if (initialization_pattern) {
     321           4 :         *initialization_pattern = av_asprintf("%s_$RepresentationID$.hdr",
     322             :                                               filename_str);
     323           4 :         if (!(*initialization_pattern)) {
     324           0 :             ret = AVERROR(ENOMEM);
     325           0 :             goto end;
     326             :         }
     327             :     }
     328           8 :     if (media_pattern) {
     329           4 :         *media_pattern = av_asprintf("%s_$RepresentationID$_$Number$.chk",
     330             :                                      filename_str);
     331           4 :         if (!(*media_pattern)) {
     332           0 :             ret = AVERROR(ENOMEM);
     333           0 :             goto end;
     334             :         }
     335             :     }
     336             : 
     337          16 : end:
     338           8 :     av_freep(&filename_str);
     339           8 :     return ret;
     340             : }
     341             : 
     342             : /*
     343             :  * Writes an Adaptation Set. Returns 0 on success and < 0 on failure.
     344             :  */
     345           9 : static int write_adaptation_set(AVFormatContext *s, int as_index)
     346             : {
     347           9 :     WebMDashMuxContext *w = s->priv_data;
     348           9 :     AdaptationSet *as = &w->as[as_index];
     349           9 :     AVCodecParameters *par = s->streams[as->streams[0]]->codecpar;
     350             :     AVDictionaryEntry *lang;
     351             :     int i;
     352             :     static const char boolean[2][6] = { "false", "true" };
     353           9 :     int subsegmentStartsWithSAP = 1;
     354             : 
     355             :     // Width, Height and Sample Rate will go in the AdaptationSet tag if they
     356             :     // are the same for all contained Representations. otherwise, they will go
     357             :     // on their respective Representation tag. For live streams, they always go
     358             :     // in the Representation tag.
     359           9 :     int width_in_as = 1, height_in_as = 1, sample_rate_in_as = 1;
     360           9 :     if (par->codec_type == AVMEDIA_TYPE_VIDEO) {
     361           5 :       width_in_as = !w->is_live && check_matching_width(s, as);
     362           5 :       height_in_as = !w->is_live && check_matching_height(s, as);
     363             :     } else {
     364           4 :       sample_rate_in_as = !w->is_live && check_matching_sample_rate(s, as);
     365             :     }
     366             : 
     367           9 :     avio_printf(s->pb, "<AdaptationSet id=\"%s\"", as->id);
     368           9 :     avio_printf(s->pb, " mimeType=\"%s/webm\"",
     369           9 :                 par->codec_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio");
     370           9 :     avio_printf(s->pb, " codecs=\"%s\"", get_codec_name(par->codec_id));
     371             : 
     372           9 :     lang = av_dict_get(s->streams[as->streams[0]]->metadata, "language", NULL, 0);
     373           9 :     if (lang) avio_printf(s->pb, " lang=\"%s\"", lang->value);
     374             : 
     375           9 :     if (par->codec_type == AVMEDIA_TYPE_VIDEO && width_in_as)
     376           2 :         avio_printf(s->pb, " width=\"%d\"", par->width);
     377           9 :     if (par->codec_type == AVMEDIA_TYPE_VIDEO && height_in_as)
     378           2 :         avio_printf(s->pb, " height=\"%d\"", par->height);
     379           9 :     if (par->codec_type == AVMEDIA_TYPE_AUDIO && sample_rate_in_as)
     380           2 :         avio_printf(s->pb, " audioSamplingRate=\"%d\"", par->sample_rate);
     381             : 
     382           9 :     avio_printf(s->pb, " bitstreamSwitching=\"%s\"",
     383           9 :                 boolean[bitstream_switching(s, as)]);
     384           9 :     avio_printf(s->pb, " subsegmentAlignment=\"%s\"",
     385           9 :                 boolean[w->is_live || subsegment_alignment(s, as)]);
     386             : 
     387          23 :     for (i = 0; i < as->nb_streams; i++) {
     388          14 :         AVDictionaryEntry *kf = av_dict_get(s->streams[as->streams[i]]->metadata,
     389             :                                             CLUSTER_KEYFRAME, NULL, 0);
     390          14 :         if (!w->is_live && (!kf || !strncmp(kf->value, "0", 1))) subsegmentStartsWithSAP = 0;
     391             :     }
     392           9 :     avio_printf(s->pb, " subsegmentStartsWithSAP=\"%d\"", subsegmentStartsWithSAP);
     393           9 :     avio_printf(s->pb, ">\n");
     394             : 
     395           9 :     if (w->is_live) {
     396           4 :         AVDictionaryEntry *filename =
     397           4 :             av_dict_get(s->streams[as->streams[0]]->metadata, FILENAME, NULL, 0);
     398           4 :         char *initialization_pattern = NULL;
     399           4 :         char *media_pattern = NULL;
     400           4 :         int ret = parse_filename(filename->value, NULL, &initialization_pattern,
     401             :                                  &media_pattern);
     402           4 :         if (ret) return ret;
     403           4 :         avio_printf(s->pb, "<ContentComponent id=\"1\" type=\"%s\"/>\n",
     404           4 :                     par->codec_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio");
     405           4 :         avio_printf(s->pb, "<SegmentTemplate");
     406           4 :         avio_printf(s->pb, " timescale=\"1000\"");
     407           4 :         avio_printf(s->pb, " duration=\"%d\"", w->chunk_duration);
     408           4 :         avio_printf(s->pb, " media=\"%s\"", media_pattern);
     409           4 :         avio_printf(s->pb, " startNumber=\"%d\"", w->chunk_start_index);
     410           4 :         avio_printf(s->pb, " initialization=\"%s\"", initialization_pattern);
     411           4 :         avio_printf(s->pb, "/>\n");
     412           4 :         av_free(initialization_pattern);
     413           4 :         av_free(media_pattern);
     414             :     }
     415             : 
     416          46 :     for (i = 0; i < as->nb_streams; i++) {
     417          14 :         char *representation_id = NULL;
     418             :         int ret;
     419          14 :         if (w->is_live) {
     420           4 :             AVDictionaryEntry *filename =
     421           4 :                 av_dict_get(s->streams[as->streams[i]]->metadata, FILENAME, NULL, 0);
     422           4 :             if (!filename)
     423           0 :                 return AVERROR(EINVAL);
     424           4 :             if (ret = parse_filename(filename->value, &representation_id, NULL, NULL))
     425           0 :                 return ret;
     426             :         } else {
     427          10 :             representation_id = av_asprintf("%d", w->representation_id++);
     428          10 :             if (!representation_id) return AVERROR(ENOMEM);
     429             :         }
     430          14 :         ret = write_representation(s, s->streams[as->streams[i]],
     431             :                                    representation_id, !width_in_as,
     432             :                                    !height_in_as, !sample_rate_in_as);
     433          14 :         av_free(representation_id);
     434          14 :         if (ret) return ret;
     435             :     }
     436           9 :     avio_printf(s->pb, "</AdaptationSet>\n");
     437           9 :     return 0;
     438             : }
     439             : 
     440          14 : static int to_integer(char *p, int len)
     441             : {
     442             :     int ret;
     443          14 :     char *q = av_malloc(sizeof(char) * len);
     444          14 :     if (!q)
     445           0 :         return AVERROR(ENOMEM);
     446          14 :     av_strlcpy(q, p, len);
     447          14 :     ret = atoi(q);
     448          14 :     av_free(q);
     449          14 :     return ret;
     450             : }
     451             : 
     452           6 : static int parse_adaptation_sets(AVFormatContext *s)
     453             : {
     454           6 :     WebMDashMuxContext *w = s->priv_data;
     455           6 :     char *p = w->adaptation_sets;
     456             :     char *q;
     457             :     enum { new_set, parsed_id, parsing_streams } state;
     458           6 :     if (!w->adaptation_sets) {
     459           0 :         av_log(s, AV_LOG_ERROR, "The 'adaptation_sets' option must be set.\n");
     460           0 :         return AVERROR(EINVAL);
     461             :     }
     462             :     // syntax id=0,streams=0,1,2 id=1,streams=3,4 and so on
     463           6 :     state = new_set;
     464          38 :     while (p < w->adaptation_sets + strlen(w->adaptation_sets)) {
     465          32 :         if (*p == ' ')
     466           0 :             continue;
     467          41 :         else if (state == new_set && !strncmp(p, "id=", 3)) {
     468           9 :             void *mem = av_realloc(w->as, sizeof(*w->as) * (w->nb_as + 1));
     469           9 :             if (mem == NULL)
     470           0 :                 return AVERROR(ENOMEM);
     471           9 :             w->as = mem;
     472           9 :             ++w->nb_as;
     473           9 :             w->as[w->nb_as - 1].nb_streams = 0;
     474           9 :             w->as[w->nb_as - 1].streams = NULL;
     475           9 :             p += 3; // consume "id="
     476           9 :             q = w->as[w->nb_as - 1].id;
     477           9 :             while (*p != ',') *q++ = *p++;
     478           9 :             *q = 0;
     479           9 :             p++;
     480           9 :             state = parsed_id;
     481          23 :         } else if (state == parsed_id && !strncmp(p, "streams=", 8)) {
     482           9 :             p += 8; // consume "streams="
     483           9 :             state = parsing_streams;
     484          14 :         } else if (state == parsing_streams) {
     485          14 :             struct AdaptationSet *as = &w->as[w->nb_as - 1];
     486          14 :             q = p;
     487          14 :             while (*q != '\0' && *q != ',' && *q != ' ') q++;
     488          14 :             as->streams = av_realloc(as->streams, sizeof(*as->streams) * ++as->nb_streams);
     489          14 :             if (as->streams == NULL)
     490           0 :                 return AVERROR(ENOMEM);
     491          14 :             as->streams[as->nb_streams - 1] = to_integer(p, q - p + 1);
     492          28 :             if (as->streams[as->nb_streams - 1] < 0 ||
     493          14 :                 as->streams[as->nb_streams - 1] >= s->nb_streams) {
     494           0 :                 av_log(s, AV_LOG_ERROR, "Invalid value for 'streams' in adapation_sets.\n");
     495           0 :                 return AVERROR(EINVAL);
     496             :             }
     497          14 :             if (*q == '\0') break;
     498           8 :             if (*q == ' ') state = new_set;
     499           8 :             p = ++q;
     500             :         } else {
     501           0 :             return -1;
     502             :         }
     503             :     }
     504           6 :     return 0;
     505             : }
     506             : 
     507           6 : static int webm_dash_manifest_write_header(AVFormatContext *s)
     508             : {
     509             :     int i;
     510           6 :     double start = 0.0;
     511             :     int ret;
     512           6 :     WebMDashMuxContext *w = s->priv_data;
     513           6 :     ret = parse_adaptation_sets(s);
     514           6 :     if (ret < 0) {
     515           0 :         goto fail;
     516             :     }
     517           6 :     ret = write_header(s);
     518           6 :     if (ret < 0) {
     519           0 :         goto fail;
     520             :     }
     521           6 :     avio_printf(s->pb, "<Period id=\"0\"");
     522           6 :     avio_printf(s->pb, " start=\"PT%gS\"", start);
     523           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          15 :     for (i = 0; i < w->nb_as; i++) {
     529           9 :         ret = write_adaptation_set(s, i);
     530           9 :         if (ret < 0) {
     531           0 :             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           6 :     return ret < 0 ? ret : 0;
     540             : }
     541             : 
     542           0 : static int webm_dash_manifest_write_packet(AVFormatContext *s, AVPacket *pkt)
     543             : {
     544           0 :     return AVERROR_EOF;
     545             : }
     546             : 
     547           6 : static int webm_dash_manifest_write_trailer(AVFormatContext *s)
     548             : {
     549           6 :     free_adaptation_sets(s);
     550           6 :     return 0;
     551             : }
     552             : 
     553             : #define OFFSET(x) offsetof(WebMDashMuxContext, x)
     554             : static const AVOption options[] = {
     555             :     { "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 },
     556             :     { "debug_mode", "[private option - users should never set this]. Create deterministic output", OFFSET(debug_mode), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM },
     557             :     { "live", "create a live stream manifest", OFFSET(is_live), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM },
     558             :     { "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 },
     559             :     { "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 },
     560             :     { "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 },
     561             :     { "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 },
     562             :     { "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 },
     563             :     { NULL },
     564             : };
     565             : 
     566             : #if CONFIG_WEBM_DASH_MANIFEST_MUXER
     567             : static const AVClass webm_dash_class = {
     568             :     .class_name = "WebM DASH Manifest muxer",
     569             :     .item_name  = av_default_item_name,
     570             :     .option     = options,
     571             :     .version    = LIBAVUTIL_VERSION_INT,
     572             : };
     573             : 
     574             : AVOutputFormat ff_webm_dash_manifest_muxer = {
     575             :     .name              = "webm_dash_manifest",
     576             :     .long_name         = NULL_IF_CONFIG_SMALL("WebM DASH Manifest"),
     577             :     .mime_type         = "application/xml",
     578             :     .extensions        = "xml",
     579             :     .priv_data_size    = sizeof(WebMDashMuxContext),
     580             :     .write_header      = webm_dash_manifest_write_header,
     581             :     .write_packet      = webm_dash_manifest_write_packet,
     582             :     .write_trailer     = webm_dash_manifest_write_trailer,
     583             :     .priv_class        = &webm_dash_class,
     584             : };
     585             : #endif

Generated by: LCOV version 1.13