libavformat/vorbiscomment.c

   1 /*
   2  * VorbisComment writer
   3  * Copyright (c) 2009 James Darnley
   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 #include "avio.h"
  23 #include "avformat.h"
  24 #include "metadata.h"
  25 #include "vorbiscomment.h"
  26 #include "libavutil/dict.h"
  27
  28 /**
  29  * VorbisComment metadata conversion mapping.
  30  * from Ogg Vorbis I format specification: comment field and header specification
  31  * http://xiph.org/vorbis/doc/v-comment.html
  32  */
  33 const AVMetadataConv ff_vorbiscomment_metadata_conv[] = {
  34     { "ALBUMARTIST", "album_artist"},
  35     { "TRACKNUMBER", "track"  },
  36     { "DISCNUMBER",  "disc"   },
  37     { "DESCRIPTION", "comment" },
  38     { 0 }
  39 };
  40
  41 int64_t ff_vorbiscomment_length(const AVDictionary *m, const char *vendor_string,
  42                                 AVChapter **chapters, unsigned int nb_chapters)
  43 {
  44     int64_t len = 8;
  45     len += strlen(vendor_string);
  46     if (chapters && nb_chapters) {
  47         for (int i = 0; i < nb_chapters; i++) {
  48             const AVDictionaryEntry *tag = NULL;
  49             len += 4 + 12 + 1 + 10;
  50             while ((tag = av_dict_iterate(chapters[i]->metadata, tag))) {
  51                 int64_t len1 = !strcmp(tag->key, "title") ? 4 : strlen(tag->key);
  52                 len += 4 + 10 + len1 + 1 + strlen(tag->value);
  53             }
  54         }
  55     }
  56     if (m) {
  57         const AVDictionaryEntry *tag = NULL;
  58         while ((tag = av_dict_iterate(m, tag))) {
  59             len += 4 +strlen(tag->key) + 1 + strlen(tag->value);
  60         }
  61     }
  62     return len;
  63 }
  64
  65 int ff_vorbiscomment_write(AVIOContext *pb, const AVDictionary *m,
  66                            const char *vendor_string,
  67                            AVChapter **chapters, unsigned int nb_chapters)
  68 {
  69     size_t vendor_string_length = strlen(vendor_string);
  70     int cm_count = 0;
  71     avio_wl32(pb, vendor_string_length);
  72     avio_write(pb, vendor_string, vendor_string_length);
  73     if (chapters && nb_chapters) {
  74         for (int i = 0; i < nb_chapters; i++) {
  75             cm_count += av_dict_count(chapters[i]->metadata) + 1;
  76         }
  77     }
  78     if (m) {
  79         int count = av_dict_count(m) + cm_count;
  80         const AVDictionaryEntry *tag = NULL;
  81         avio_wl32(pb, count);
  82         while ((tag = av_dict_iterate(m, tag))) {
  83             int64_t len1 = strlen(tag->key);
  84             int64_t len2 = strlen(tag->value);
  85             if (len1+1+len2 > UINT32_MAX)
  86                 return AVERROR(EINVAL);
  87             avio_wl32(pb, len1 + 1 + len2);
  88             avio_write(pb, tag->key, len1);
  89             avio_w8(pb, '=');
  90             avio_write(pb, tag->value, len2);
  91         }
  92         for (int i = 0; i < nb_chapters; i++) {
  93             AVChapter *chp = chapters[i];
  94             char chapter_time[13];
  95             char chapter_number[4];
  96             int h, m, s, ms;
  97
  98             s  = av_rescale(chp->start, chp->time_base.num, chp->time_base.den);
  99             h  = s / 3600;
 100             m  = (s / 60) % 60;
 101             ms = av_rescale_q(chp->start, chp->time_base, av_make_q(   1, 1000)) % 1000;
 102             s  = s % 60;
 103             snprintf(chapter_number, sizeof(chapter_number), "%03d", i);
 104             snprintf(chapter_time, sizeof(chapter_time), "%02d:%02d:%02d.%03d", h, m, s, ms);
 105             avio_wl32(pb, 10 + 1 + 12);
 106             avio_write(pb, "CHAPTER", 7);
 107             avio_write(pb, chapter_number, 3);
 108             avio_w8(pb, '=');
 109             avio_write(pb, chapter_time, 12);
 110
 111             tag = NULL;
 112             while ((tag = av_dict_iterate(chapters[i]->metadata, tag))) {
 113                 int64_t len1 = !strcmp(tag->key, "title") ? 4 : strlen(tag->key);
 114                 int64_t len2 = strlen(tag->value);
 115                 if (len1+1+len2+10 > UINT32_MAX)
 116                     return AVERROR(EINVAL);
 117                 avio_wl32(pb, 10 + len1 + 1 + len2);
 118                 avio_write(pb, "CHAPTER", 7);
 119                 avio_write(pb, chapter_number, 3);
 120                 if (!strcmp(tag->key, "title"))
 121                     avio_write(pb, "NAME", 4);
 122                 else
 123                     avio_write(pb, tag->key, len1);
 124                 avio_w8(pb, '=');
 125                 avio_write(pb, tag->value, len2);
 126             }
 127         }
 128     } else
 129         avio_wl32(pb, 0);
 130     return 0;
 131 }