mp4.c

   1 /*
   2  * Copyright 2006 dnk <dnk@bjum.net>
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License as
   6  * published by the Free Software Foundation; either version 2 of the
   7  * License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  17  * 02111-1307, USA.
  18  */
  19
  20 #include "ip.h"
  21 #include "xmalloc.h"
  22 #include "debug.h"
  23 #include "file.h"
  24
  25 #include <mp4.h>
  26 #include <faad.h>
  27
  28 #include <errno.h>
  29 #include <string.h>
  30 #include <sys/types.h>
  31 #include <unistd.h>
  32
  33 struct mp4_private {
  34         char *overflow_buf;
  35         int overflow_buf_len;
  36
  37         unsigned char channels;
  38         unsigned long sample_rate;
  39
  40         faacDecHandle decoder;          /* typedef void * */
  41
  42         struct {
  43                 MP4FileHandle handle;   /* typedef void * */
  44
  45                 MP4TrackId track;
  46                 MP4SampleId sample;
  47                 MP4SampleId num_samples;
  48         } mp4;
  49 };
  50
  51
  52 static MP4TrackId mp4_get_track(MP4FileHandle *handle)
  53 {
  54         MP4TrackId num_tracks;
  55         const char *track_type;
  56         uint8_t obj_type;
  57         MP4TrackId i;
  58
  59         num_tracks = MP4GetNumberOfTracks(handle, NULL, 0);
  60
  61         for (i = 1; i <= num_tracks; i++) {
  62                 track_type = MP4GetTrackType(handle, i);
  63                 if (!track_type)
  64                         continue;
  65
  66                 if (!MP4_IS_AUDIO_TRACK_TYPE(track_type))
  67                         continue;
  68
  69                 /* MP4GetTrackAudioType */
  70                 obj_type = MP4GetTrackEsdsObjectTypeId(handle, i);
  71                 if (obj_type == MP4_INVALID_AUDIO_TYPE)
  72                         continue;
  73
  74                 if (obj_type == MP4_MPEG4_AUDIO_TYPE) {
  75                         obj_type = MP4GetTrackAudioMpeg4Type(handle, i);
  76
  77                         if (MP4_IS_MPEG4_AAC_AUDIO_TYPE(obj_type))
  78                                 return i;
  79                 } else {
  80                         if (MP4_IS_AAC_AUDIO_TYPE(obj_type))
  81                                 return i;
  82                 }
  83         }
  84
  85         return MP4_INVALID_TRACK_ID;
  86 }
  87
  88 static int mp4_open(struct input_plugin_data *ip_data)
  89 {
  90         struct mp4_private *priv;
  91         faacDecConfigurationPtr neaac_cfg;
  92         unsigned char *buf;
  93         unsigned int buf_size;
  94
  95
  96         /* http://sourceforge.net/forum/message.php?msg_id=3578887 */
  97         if (ip_data->remote)
  98                 return -IP_ERROR_FUNCTION_NOT_SUPPORTED;
  99
 100         /* init private struct */
 101         priv = xnew0(struct mp4_private, 1);
 102         ip_data->private = priv;
 103
 104         priv->decoder = faacDecOpen();
 105
 106         /* set decoder config */
 107         neaac_cfg = faacDecGetCurrentConfiguration(priv->decoder);
 108         neaac_cfg->outputFormat = FAAD_FMT_16BIT;       /* force 16 bit audio */
 109         neaac_cfg->downMatrix = 1;                      /* 5.1 -> stereo */
 110         faacDecSetConfiguration(priv->decoder, neaac_cfg);
 111
 112         /* open mpeg-4 file */
 113         priv->mp4.handle = MP4Read(ip_data->filename, 0);
 114         if (!priv->mp4.handle) {
 115                 d_print("MP4Read failed\n");
 116                 goto out;
 117         }
 118
 119         /* find aac audio track */
 120         priv->mp4.track = mp4_get_track(priv->mp4.handle);
 121         if (priv->mp4.track == MP4_INVALID_TRACK_ID) {
 122                 d_print("MP4FindTrackId failed\n");
 123                 goto out;
 124         }
 125
 126         priv->mp4.num_samples = MP4GetTrackNumberOfSamples(priv->mp4.handle, priv->mp4.track);
 127
 128         priv->mp4.sample = 1;
 129
 130         buf = NULL;
 131         buf_size = 0;
 132         if (!MP4GetTrackESConfiguration(priv->mp4.handle, priv->mp4.track, &buf, &buf_size)) {
 133                 /* failed to get mpeg-4 audio config... this is ok.
 134                  * faacDecInit2() will simply use default values instead.
 135                  */
 136                 buf = NULL;
 137                 buf_size = 0;
 138         }
 139
 140         /* init decoder according to mpeg-4 audio config */
 141         if (faacDecInit2(priv->decoder, buf, buf_size, &priv->sample_rate, &priv->channels) < 0) {
 142                 free(buf);
 143                 goto out;
 144         }
 145
 146         free(buf);
 147
 148         d_print("sample rate %luhz, channels %u\n", priv->sample_rate, priv->channels);
 149
 150         ip_data->sf = sf_rate(priv->sample_rate) | sf_channels(priv->channels) | sf_bits(16) | sf_signed(1);
 151 #if defined(WORDS_BIGENDIAN)
 152         ip_data->sf |= sf_bigendian(1);
 153 #endif
 154
 155         return 0;
 156
 157 out:
 158         if (priv->mp4.handle)
 159                 MP4Close(priv->mp4.handle);
 160         if (priv->decoder)
 161                 faacDecClose(priv->decoder);
 162         free(priv);
 163         return -IP_ERROR_FILE_FORMAT;
 164 }
 165
 166 static int mp4_close(struct input_plugin_data *ip_data)
 167 {
 168         struct mp4_private *priv;
 169
 170         priv = ip_data->private;
 171
 172         if (priv->mp4.handle)
 173                 MP4Close(priv->mp4.handle);
 174
 175         if (priv->decoder)
 176                 faacDecClose(priv->decoder);
 177
 178         free(priv);
 179         ip_data->private = NULL;
 180
 181         return 0;
 182 }
 183
 184 /* returns -1 on fatal errors
 185  * returns -2 on non-fatal errors
 186  * 0 on eof
 187  * number of bytes put in 'buffer' on success */
 188 static int decode_one_frame(struct input_plugin_data *ip_data, void *buffer, int count)
 189 {
 190         struct mp4_private *priv;
 191         unsigned char *aac_data = NULL;
 192         unsigned int aac_data_len = 0;
 193         faacDecFrameInfo frame_info;
 194         char *sample_buf;
 195         int bytes;
 196
 197         priv = ip_data->private;
 198
 199         BUG_ON(priv->overflow_buf_len);
 200
 201         if (priv->mp4.sample > priv->mp4.num_samples)
 202                 return 0; /* EOF */
 203
 204         if (MP4ReadSample(priv->mp4.handle, priv->mp4.track, priv->mp4.sample,
 205                 &aac_data, &aac_data_len, NULL, NULL, NULL, NULL) == 0) {
 206                 d_print("error reading mp4 sample %d\n", priv->mp4.sample);
 207                 errno = EINVAL;
 208                 return -1;
 209         }
 210
 211         priv->mp4.sample++;
 212
 213         if (!aac_data) {
 214                 d_print("aac_data == NULL\n");
 215                 errno = EINVAL;
 216                 return -1;
 217         }
 218
 219         sample_buf = faacDecDecode(priv->decoder, &frame_info, aac_data, aac_data_len);
 220
 221         free(aac_data);
 222
 223         if (!sample_buf || frame_info.bytesconsumed <= 0) {
 224                 d_print("fatal error: %s\n", faacDecGetErrorMessage(frame_info.error));
 225                 errno = EINVAL;
 226                 return -1;
 227         }
 228
 229         if (frame_info.error != 0) {
 230                 d_print("frame error: %s\n", faacDecGetErrorMessage(frame_info.error));
 231                 return -2;
 232         }
 233
 234         if (frame_info.samples <= 0)
 235                 return -2;
 236
 237         if (frame_info.channels != priv->channels || frame_info.samplerate != priv->sample_rate) {
 238                 d_print("invalid channel or sample_rate count\n");
 239                 return -2;
 240         }
 241
 242         /* 16-bit samples */
 243         bytes = frame_info.samples * 2;
 244
 245         if (bytes > count) {
 246                 /* decoded too much; keep overflow. */
 247                 priv->overflow_buf = sample_buf + count;
 248                 priv->overflow_buf_len = bytes - count;
 249                 memcpy(buffer, sample_buf, count);
 250                 return count;
 251         } else {
 252                 memcpy(buffer, sample_buf, bytes);
 253         }
 254
 255         return bytes;
 256 }
 257
 258 static int mp4_read(struct input_plugin_data *ip_data, char *buffer, int count)
 259 {
 260         struct mp4_private *priv;
 261         int rc;
 262
 263         priv = ip_data->private;
 264
 265         /* use overflow from previous call (if any) */
 266         if (priv->overflow_buf_len > 0) {
 267                 int len = priv->overflow_buf_len;
 268
 269                 if (len > count)
 270                         len = count;
 271
 272                 memcpy(buffer, priv->overflow_buf, len);
 273                 priv->overflow_buf += len;
 274                 priv->overflow_buf_len -= len;
 275
 276                 return len;
 277         }
 278
 279         do {
 280                 rc = decode_one_frame(ip_data, buffer, count);
 281         } while (rc == -2);
 282
 283         return rc;
 284 }
 285
 286 static int mp4_seek(struct input_plugin_data *ip_data, double offset)
 287 {
 288         struct mp4_private *priv;
 289         MP4SampleId sample;
 290         uint32_t scale;
 291
 292         priv = ip_data->private;
 293
 294         scale = MP4GetTrackTimeScale(priv->mp4.handle, priv->mp4.track);
 295         if (scale == 0)
 296                 return -IP_ERROR_INTERNAL;
 297
 298         sample = MP4GetSampleIdFromTime(priv->mp4.handle, priv->mp4.track,
 299                 (MP4Timestamp)(offset * (double)scale), 0);
 300         if (sample == MP4_INVALID_SAMPLE_ID)
 301                 return -IP_ERROR_INTERNAL;
 302
 303         priv->mp4.sample = sample;
 304
 305         d_print("seeking to sample %d\n", sample);
 306
 307         return 0;
 308 }
 309
 310 static int mp4_read_comments(struct input_plugin_data *ip_data,
 311                 struct keyval **comments)
 312 {
 313         struct mp4_private *priv;
 314         uint16_t meta_num, meta_total;
 315         uint8_t val;
 316         /*uint8_t *ustr;
 317         uint32_t size;*/
 318         char *str;
 319         GROWING_KEYVALS(c);
 320
 321         priv = ip_data->private;
 322
 323         /* MP4GetMetadata* provides malloced pointers, and the data
 324          * is in UTF-8 (or at least it should be). */
 325         if (MP4GetMetadataArtist(priv->mp4.handle, &str))
 326                 comments_add(&c, "artist", str);
 327         if (MP4GetMetadataAlbum(priv->mp4.handle, &str))
 328                 comments_add(&c, "album", str);
 329         if (MP4GetMetadataName(priv->mp4.handle, &str))
 330                 comments_add(&c, "title", str);
 331         if (MP4GetMetadataGenre(priv->mp4.handle, &str))
 332                 comments_add(&c, "genre", str);
 333         if (MP4GetMetadataYear(priv->mp4.handle, &str))
 334                 comments_add(&c, "date", str);
 335
 336         if (MP4GetMetadataCompilation(priv->mp4.handle, &val))
 337                 comments_add_const(&c, "compilation", val ? "yes" : "no");
 338 #if 0
 339         if (MP4GetBytesProperty(priv->mp4.handle, "moov.udta.meta.ilst.aART.data", &ustr, &size)) {
 340                 char *xstr;
 341
 342                 /* What's this?
 343                  * This is the result from lack of documentation.
 344                  * It's supposed to return just a string, but it
 345                  * returns an additional 16 bytes of junk at the
 346                  * beginning. Could be a bug. Could be intentional.
 347                  * Hopefully this works around it:
 348                  */
 349                 if (ustr[0] == 0 && size > 16) {
 350                         ustr += 16;
 351                         size -= 16;
 352                 }
 353                 xstr = xmalloc(size + 1);
 354                 memcpy(xstr, ustr, size);
 355                 xstr[size] = 0;
 356                 comments_add(&c, "albumartist", xstr);
 357                 free(xstr);
 358         }
 359 #endif
 360         if (MP4GetMetadataTrack(priv->mp4.handle, &meta_num, &meta_total)) {
 361                 char buf[6];
 362                 snprintf(buf, 6, "%u", meta_num);
 363                 comments_add_const(&c, "tracknumber", buf);
 364         }
 365         if (MP4GetMetadataDisk(priv->mp4.handle, &meta_num, &meta_total)) {
 366                 char buf[6];
 367                 snprintf(buf, 6, "%u", meta_num);
 368                 comments_add_const(&c, "discnumber", buf);
 369         }
 370
 371         keyvals_terminate(&c);
 372         *comments = c.keyvals;
 373         return 0;
 374 }
 375
 376 static int mp4_duration(struct input_plugin_data *ip_data)
 377 {
 378         struct mp4_private *priv;
 379         uint32_t scale;
 380         uint64_t duration;
 381
 382         priv = ip_data->private;
 383
 384         scale = MP4GetTrackTimeScale(priv->mp4.handle, priv->mp4.track);
 385         if (scale == 0)
 386                 return 0;
 387
 388         duration = MP4GetTrackDuration(priv->mp4.handle, priv->mp4.track);
 389
 390         return duration / scale;
 391 }
 392
 393 const struct input_plugin_ops ip_ops = {
 394         .open = mp4_open,
 395         .close = mp4_close,
 396         .read = mp4_read,
 397         .seek = mp4_seek,
 398         .read_comments = mp4_read_comments,
 399         .duration = mp4_duration
 400 };
 401
 402 const char * const ip_extensions[] = { "mp4", "m4a", "m4b", NULL };
 403 const char * const ip_mime_types[] = { /*"audio/mp4", "audio/mp4a-latm",*/ NULL };