git.sesse.net Git - ffmpeg/blob - libavformat/oggparsevorbis.c

   1 /**
   2       Copyright (C) 2005  Michael Ahlberg, Måns Rullgård
   3
   4       Permission is hereby granted, free of charge, to any person
   5       obtaining a copy of this software and associated documentation
   6       files (the "Software"), to deal in the Software without
   7       restriction, including without limitation the rights to use, copy,
   8       modify, merge, publish, distribute, sublicense, and/or sell copies
   9       of the Software, and to permit persons to whom the Software is
  10       furnished to do so, subject to the following conditions:
  11
  12       The above copyright notice and this permission notice shall be
  13       included in all copies or substantial portions of the Software.
  14
  15       THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  16       EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  17       MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  18       NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
  19       HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  20       WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21       OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  22       DEALINGS IN THE SOFTWARE.
  23 **/
  24
  25 #include <stdlib.h>
  26 #include "avformat.h"
  27 #include "bitstream.h"
  28 #include "bswap.h"
  29 #include "ogg2.h"
  30 #include "avstring.h"
  31
  32 extern int
  33 vorbis_comment(AVFormatContext * as, uint8_t *buf, int size)
  34 {
  35     char *p = buf;
  36     int s, n, j;
  37
  38     if (size < 4)
  39         return -1;
  40
  41     s = AV_RL32(p);
  42     p += 4;
  43     size -= 4;
  44
  45     if (size < s + 4)
  46         return -1;
  47
  48     p += s;
  49     size -= s;
  50
  51     n = AV_RL32(p);
  52     p += 4;
  53     size -= 4;
  54
  55     while (size >= 4) {
  56         char *t, *v;
  57         int tl, vl;
  58
  59         s = AV_RL32(p);
  60         p += 4;
  61         size -= 4;
  62
  63         if (size < s)
  64             break;
  65
  66         t = p;
  67         p += s;
  68         size -= s;
  69         n--;
  70
  71         v = memchr(t, '=', s);
  72         if (!v)
  73             continue;
  74
  75         tl = v - t;
  76         vl = s - tl - 1;
  77         v++;
  78
  79         if (tl && vl) {
  80             char tt[tl + 1];
  81             char ct[vl + 1];
  82
  83             for (j = 0; j < tl; j++)
  84                 tt[j] = toupper(t[j]);
  85             tt[tl] = 0;
  86
  87             memcpy(ct, v, vl);
  88             ct[vl] = 0;
  89
  90             // took from Vorbis_I_spec
  91             if (!strcmp(tt, "AUTHOR") || !strcmp(tt, "ARTIST"))
  92                 av_strlcpy(as->author, ct, sizeof(as->author));
  93             else if (!strcmp(tt, "TITLE"))
  94                 av_strlcpy(as->title, ct, sizeof(as->title));
  95             else if (!strcmp(tt, "COPYRIGHT"))
  96                 av_strlcpy(as->copyright, ct, sizeof(as->copyright));
  97             else if (!strcmp(tt, "DESCRIPTION"))
  98                 av_strlcpy(as->comment, ct, sizeof(as->comment));
  99             else if (!strcmp(tt, "GENRE"))
 100                 av_strlcpy(as->genre, ct, sizeof(as->genre));
 101             else if (!strcmp(tt, "TRACKNUMBER"))
 102                 as->track = atoi(ct);
 103             else if (!strcmp(tt, "ALBUM"))
 104                 av_strlcpy(as->album, ct, sizeof(as->album));
 105         }
 106     }
 107
 108     if (size > 0)
 109         av_log(as, AV_LOG_INFO, "%i bytes of comment header remain\n", size);
 110     if (n > 0)
 111         av_log(as, AV_LOG_INFO,
 112                "truncated comment header, %i comments not found\n", n);
 113
 114     return 0;
 115 }
 116
 117
 118 /** Parse the vorbis header
 119  * Vorbis Identification header from Vorbis_I_spec.html#vorbis-spec-codec
 120  * [vorbis_version] = read 32 bits as unsigned integer | Not used
 121  * [audio_channels] = read 8 bit integer as unsigned | Used
 122  * [audio_sample_rate] = read 32 bits as unsigned integer | Used
 123  * [bitrate_maximum] = read 32 bits as signed integer | Not used yet
 124  * [bitrate_nominal] = read 32 bits as signed integer | Not used yet
 125  * [bitrate_minimum] = read 32 bits as signed integer | Used as bitrate
 126  * [blocksize_0] = read 4 bits as unsigned integer | Not Used
 127  * [blocksize_1] = read 4 bits as unsigned integer | Not Used
 128  * [framing_flag] = read one bit | Not Used
 129  *    */
 130
 131 typedef struct {
 132     unsigned int len[3];
 133     unsigned char *packet[3];
 134 } oggvorbis_private_t;
 135
 136
 137 static unsigned int
 138 fixup_vorbis_headers(AVFormatContext * as, oggvorbis_private_t *priv,
 139                      uint8_t **buf)
 140 {
 141     int i,offset, len;
 142     unsigned char *ptr;
 143
 144     len = priv->len[0] + priv->len[1] + priv->len[2];
 145     ptr = *buf = av_mallocz(len + len/255 + 64);
 146
 147     ptr[0] = 2;
 148     offset = 1;
 149     offset += av_xiphlacing(&ptr[offset], priv->len[0]);
 150     offset += av_xiphlacing(&ptr[offset], priv->len[1]);
 151     for (i = 0; i < 3; i++) {
 152         memcpy(&ptr[offset], priv->packet[i], priv->len[i]);
 153         offset += priv->len[i];
 154     }
 155     *buf = av_realloc(*buf, offset);
 156     return offset;
 157 }
 158
 159
 160 static int
 161 vorbis_header (AVFormatContext * s, int idx)
 162 {
 163     ogg_t *ogg = s->priv_data;
 164     ogg_stream_t *os = ogg->streams + idx;
 165     AVStream *st = s->streams[idx];
 166     oggvorbis_private_t *priv;
 167
 168     if (os->seq > 2)
 169         return 0;
 170
 171     if (os->seq == 0) {
 172         os->private = av_mallocz(sizeof(oggvorbis_private_t));
 173         if (!os->private)
 174             return 0;
 175     }
 176
 177     priv = os->private;
 178     priv->len[os->seq] = os->psize;
 179     priv->packet[os->seq] = av_mallocz(os->psize);
 180     memcpy(priv->packet[os->seq], os->buf + os->pstart, os->psize);
 181     if (os->buf[os->pstart] == 1) {
 182         uint8_t *p = os->buf + os->pstart + 11; //skip up to the audio channels
 183         st->codec->channels = *p++;
 184         st->codec->sample_rate = AV_RL32(p);
 185         p += 8; //skip maximum and and nominal bitrate
 186         st->codec->bit_rate = AV_RL32(p); //Minimum bitrate
 187
 188         st->codec->codec_type = CODEC_TYPE_AUDIO;
 189         st->codec->codec_id = CODEC_ID_VORBIS;
 190
 191         st->time_base.num = 1;
 192         st->time_base.den = st->codec->sample_rate;
 193     } else if (os->buf[os->pstart] == 3) {
 194         vorbis_comment (s, os->buf + os->pstart + 7, os->psize - 8);
 195     } else {
 196         st->codec->extradata_size =
 197             fixup_vorbis_headers(s, priv, &st->codec->extradata);
 198     }
 199
 200     return os->seq < 3;
 201 }
 202
 203 ogg_codec_t vorbis_codec = {
 204     .magic = "\001vorbis",
 205     .magicsize = 7,
 206     .header = vorbis_header
 207 };