mp4.c

   1 /*
   2  * Copyright (C) 2003-2005 M. Bakker, Nero AG, http://www.nero.com
   3  * FAAD2 - Freeware Advanced Audio (AAC) Decoder including SBR decoding
   4  *
   5  * See file COPYING.
   6  */
   7
   8 #include <regex.h>
   9
  10 #include "para.h"
  11 #include "error.h"
  12 #include "portable_io.h"
  13 #include "string.h"
  14 #include "mp4.h"
  15
  16 struct mp4_track {
  17         uint16_t channelCount;
  18         uint16_t sampleRate;
  19
  20         /* stsz */
  21         uint32_t stsz_sample_size;
  22         uint32_t stsz_sample_count;
  23         uint32_t *stsz_table;
  24
  25         /* stts */
  26         uint32_t stts_entry_count;
  27         uint32_t *stts_sample_count;
  28         uint32_t *stts_sample_delta;
  29
  30         /* stsc */
  31         uint32_t stsc_entry_count;
  32         uint32_t *stsc_first_chunk;
  33         uint32_t *stsc_samples_per_chunk;
  34         uint32_t *stsc_sample_desc_index;
  35
  36         /* stsc */
  37         uint32_t stco_entry_count;
  38         uint32_t *stco_chunk_offset;
  39
  40         uint32_t timeScale;
  41         uint64_t duration;
  42 };
  43
  44 #define MAX_TRACKS 1024
  45
  46 struct mp4 {
  47         const struct mp4_callback *cb;
  48         int64_t current_position;
  49
  50         uint64_t moov_offset;
  51         uint64_t moov_size;
  52         uint8_t last_atom;
  53         uint64_t file_size;
  54
  55         /* incremental track index while reading the file */
  56         int32_t total_tracks;
  57         /* track data */
  58         struct mp4_track *track[MAX_TRACKS];
  59         /* the first audio track found */
  60         struct mp4_track *audio_track;
  61
  62         /* metadata */
  63         struct mp4_metadata meta;
  64 };
  65
  66 /*
  67  * Returns -1, 0, or 1 on errors/EOF/success. Partial reads followed by EOF or
  68  * read errors are treated as errors.
  69  */
  70 static int read_data(struct mp4 *f, void *data, size_t size)
  71 {
  72         while (size > 0) {
  73                 ssize_t ret = f->cb->read(f->cb->user_data, data, size);
  74                 if (ret < 0 && errno == EINTR)
  75                         continue;
  76                 /* regard EAGAIN as an error as reads should be blocking. */
  77                 if (ret <= 0)
  78                         return ret < 0? -1 : 0;
  79                 f->current_position += ret;
  80                 size -= ret;
  81         }
  82         return 1;
  83 }
  84
  85 static int read_int64(struct mp4 *f, uint64_t *result)
  86 {
  87         uint8_t data[8];
  88         int ret = read_data(f, data, 8);
  89
  90         if (ret > 0 && result)
  91                 *result = read_u64_be(data);
  92         return ret;
  93 }
  94
  95 static int read_int32(struct mp4 *f, uint32_t *result)
  96 {
  97         uint8_t data[4];
  98         int ret = read_data(f, data, 4);
  99
 100         if (ret > 0 && result)
 101                 *result = read_u32_be(data);
 102         return ret;
 103 }
 104
 105 static int read_int24(struct mp4 *f, uint32_t *result)
 106 {
 107         uint8_t data[3];
 108         int ret = read_data(f, data, 3);
 109
 110         if (ret > 0 && result)
 111                 *result = read_u24_be(data);
 112         return ret;
 113 }
 114
 115 static int read_int16(struct mp4 *f, uint16_t *result)
 116 {
 117         uint8_t data[2];
 118         int ret = read_data(f, data, 2);
 119
 120         if (ret > 0 && result)
 121                 *result = read_u16_be(data);
 122         return ret;
 123 }
 124
 125 static uint8_t read_int8(struct mp4 *f, uint8_t *result)
 126 {
 127         uint8_t data[1];
 128         int ret = read_data(f, data, 1);
 129
 130         if (ret > 0 && result)
 131                 *result = data[0];
 132         return ret;
 133 }
 134
 135 static bool atom_compare(int8_t a1, int8_t b1, int8_t c1, int8_t d1,
 136                 int8_t a2, int8_t b2, int8_t c2, int8_t d2)
 137 {
 138         return a1 == a2 && b1 == b2 && c1 == c2 && d1 == d2;
 139 }
 140
 141 enum atoms {
 142         /* atoms with subatoms */
 143         ATOM_MOOV = 1,
 144         ATOM_TRAK = 2,
 145         ATOM_EDTS = 3,
 146         ATOM_MDIA = 4,
 147         ATOM_MINF = 5,
 148         ATOM_STBL = 6,
 149         ATOM_UDTA = 7,
 150         ATOM_ILST = 8, /* iTunes Metadata list */
 151         ATOM_TITLE = 9,
 152         ATOM_ARTIST = 10,
 153         ATOM_WRITER = 11,
 154         ATOM_ALBUM = 12,
 155         ATOM_DATE = 13,
 156         ATOM_TOOL = 14,
 157         ATOM_COMMENT = 15,
 158         ATOM_GENRE1 = 16,
 159         ATOM_TRACK = 17,
 160         ATOM_DISC = 18,
 161         ATOM_COMPILATION = 19,
 162         ATOM_GENRE2 = 20,
 163         ATOM_TEMPO = 21,
 164         ATOM_COVER = 22,
 165         ATOM_DRMS = 23,
 166         ATOM_SINF = 24,
 167         ATOM_SCHI = 25,
 168
 169         SUBATOMIC = 128,
 170
 171         /* atoms without subatoms */
 172         ATOM_FTYP = 129,
 173         ATOM_MDAT = 130,
 174         ATOM_MVHD = 131,
 175         ATOM_TKHD = 132,
 176         ATOM_TREF = 133,
 177         ATOM_MDHD = 134, /* track header */
 178         ATOM_VMHD = 135,
 179         ATOM_SMHD = 136,
 180         ATOM_HMHD = 137,
 181         ATOM_STSD = 138, /* sample description box */
 182         ATOM_STTS = 139, /* time to sample box */
 183         ATOM_STSZ = 140, /* sample size box */
 184         ATOM_STZ2 = 141,
 185         ATOM_STCO = 142, /* chunk offset box */
 186         ATOM_STSC = 143, /* sample to chunk box */
 187         ATOM_MP4A = 144,
 188         ATOM_MP4V = 145,
 189         ATOM_MP4S = 146,
 190         ATOM_ESDS = 147,
 191         ATOM_META = 148, /* iTunes Metadata box */
 192         ATOM_NAME = 149, /* iTunes Metadata name box */
 193         ATOM_DATA = 150, /* iTunes Metadata data box */
 194         ATOM_CTTS = 151,
 195         ATOM_FRMA = 152,
 196         ATOM_IVIV = 153,
 197         ATOM_PRIV = 154,
 198         ATOM_USER = 155,
 199         ATOM_KEY = 156,
 200         ATOM_ALBUM_ARTIST = 157,
 201         ATOM_CONTENTGROUP = 158,
 202         ATOM_LYRICS = 159,
 203         ATOM_DESCRIPTION = 160,
 204         ATOM_NETWORK = 161,
 205         ATOM_SHOW = 162,
 206         ATOM_EPISODENAME = 163,
 207         ATOM_SORTTITLE = 164,
 208         ATOM_SORTALBUM = 165,
 209         ATOM_SORTARTIST = 166,
 210         ATOM_SORTALBUMARTIST = 167,
 211         ATOM_SORTWRITER = 168,
 212         ATOM_SORTSHOW = 169,
 213         ATOM_SEASON = 170,
 214         ATOM_EPISODE = 171,
 215         ATOM_PODCAST = 172,
 216
 217         ATOM_UNKNOWN = 255
 218 };
 219
 220 #define ATOM_FREE ATOM_UNKNOWN
 221 #define ATOM_SKIP ATOM_UNKNOWN
 222
 223 #define COPYRIGHT_SYMBOL ((int8_t)0xA9)
 224
 225 static uint8_t atom_name_to_type(int8_t a, int8_t b, int8_t c, int8_t d)
 226 {
 227         if (a == 'm') {
 228                 if (atom_compare(a, b, c, d, 'm', 'o', 'o', 'v'))
 229                         return ATOM_MOOV;
 230                 else if (atom_compare(a, b, c, d, 'm', 'i', 'n', 'f'))
 231                         return ATOM_MINF;
 232                 else if (atom_compare(a, b, c, d, 'm', 'd', 'i', 'a'))
 233                         return ATOM_MDIA;
 234                 else if (atom_compare(a, b, c, d, 'm', 'd', 'a', 't'))
 235                         return ATOM_MDAT;
 236                 else if (atom_compare(a, b, c, d, 'm', 'd', 'h', 'd'))
 237                         return ATOM_MDHD;
 238                 else if (atom_compare(a, b, c, d, 'm', 'v', 'h', 'd'))
 239                         return ATOM_MVHD;
 240                 else if (atom_compare(a, b, c, d, 'm', 'p', '4', 'a'))
 241                         return ATOM_MP4A;
 242                 else if (atom_compare(a, b, c, d, 'm', 'p', '4', 'v'))
 243                         return ATOM_MP4V;
 244                 else if (atom_compare(a, b, c, d, 'm', 'p', '4', 's'))
 245                         return ATOM_MP4S;
 246                 else if (atom_compare(a, b, c, d, 'm', 'e', 't', 'a'))
 247                         return ATOM_META;
 248         } else if (a == 't') {
 249                 if (atom_compare(a, b, c, d, 't', 'r', 'a', 'k'))
 250                         return ATOM_TRAK;
 251                 else if (atom_compare(a, b, c, d, 't', 'k', 'h', 'd'))
 252                         return ATOM_TKHD;
 253                 else if (atom_compare(a, b, c, d, 't', 'r', 'e', 'f'))
 254                         return ATOM_TREF;
 255                 else if (atom_compare(a, b, c, d, 't', 'r', 'k', 'n'))
 256                         return ATOM_TRACK;
 257                 else if (atom_compare(a, b, c, d, 't', 'm', 'p', 'o'))
 258                         return ATOM_TEMPO;
 259                 else if (atom_compare(a, b, c, d, 't', 'v', 'n', 'n'))
 260                         return ATOM_NETWORK;
 261                 else if (atom_compare(a, b, c, d, 't', 'v', 's', 'h'))
 262                         return ATOM_SHOW;
 263                 else if (atom_compare(a, b, c, d, 't', 'v', 'e', 'n'))
 264                         return ATOM_EPISODENAME;
 265                 else if (atom_compare(a, b, c, d, 't', 'v', 's', 'n'))
 266                         return ATOM_SEASON;
 267                 else if (atom_compare(a, b, c, d, 't', 'v', 'e', 's'))
 268                         return ATOM_EPISODE;
 269         } else if (a == 's') {
 270                 if (atom_compare(a, b, c, d, 's', 't', 'b', 'l'))
 271                         return ATOM_STBL;
 272                 else if (atom_compare(a, b, c, d, 's', 'm', 'h', 'd'))
 273                         return ATOM_SMHD;
 274                 else if (atom_compare(a, b, c, d, 's', 't', 's', 'd'))
 275                         return ATOM_STSD;
 276                 else if (atom_compare(a, b, c, d, 's', 't', 't', 's'))
 277                         return ATOM_STTS;
 278                 else if (atom_compare(a, b, c, d, 's', 't', 'c', 'o'))
 279                         return ATOM_STCO;
 280                 else if (atom_compare(a, b, c, d, 's', 't', 's', 'c'))
 281                         return ATOM_STSC;
 282                 else if (atom_compare(a, b, c, d, 's', 't', 's', 'z'))
 283                         return ATOM_STSZ;
 284                 else if (atom_compare(a, b, c, d, 's', 't', 'z', '2'))
 285                         return ATOM_STZ2;
 286                 else if (atom_compare(a, b, c, d, 's', 'k', 'i', 'p'))
 287                         return ATOM_SKIP;
 288                 else if (atom_compare(a, b, c, d, 's', 'i', 'n', 'f'))
 289                         return ATOM_SINF;
 290                 else if (atom_compare(a, b, c, d, 's', 'c', 'h', 'i'))
 291                         return ATOM_SCHI;
 292                 else if (atom_compare(a, b, c, d, 's', 'o', 'n', 'm'))
 293                         return ATOM_SORTTITLE;
 294                 else if (atom_compare(a, b, c, d, 's', 'o', 'a', 'l'))
 295                         return ATOM_SORTALBUM;
 296                 else if (atom_compare(a, b, c, d, 's', 'o', 'a', 'r'))
 297                         return ATOM_SORTARTIST;
 298                 else if (atom_compare(a, b, c, d, 's', 'o', 'a', 'a'))
 299                         return ATOM_SORTALBUMARTIST;
 300                 else if (atom_compare(a, b, c, d, 's', 'o', 'c', 'o'))
 301                         return ATOM_SORTWRITER;
 302                 else if (atom_compare(a, b, c, d, 's', 'o', 's', 'n'))
 303                         return ATOM_SORTSHOW;
 304         } else if (a == COPYRIGHT_SYMBOL) {
 305                 if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'n', 'a', 'm'))
 306                         return ATOM_TITLE;
 307                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'A', 'R', 'T'))
 308                         return ATOM_ARTIST;
 309                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'w', 'r', 't'))
 310                         return ATOM_WRITER;
 311                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'a', 'l', 'b'))
 312                         return ATOM_ALBUM;
 313                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'd', 'a', 'y'))
 314                         return ATOM_DATE;
 315                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 't', 'o', 'o'))
 316                         return ATOM_TOOL;
 317                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'c', 'm', 't'))
 318                         return ATOM_COMMENT;
 319                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'g', 'e', 'n'))
 320                         return ATOM_GENRE1;
 321                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'g', 'r', 'p'))
 322                         return ATOM_CONTENTGROUP;
 323                 else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'l', 'y', 'r'))
 324                         return ATOM_LYRICS;
 325         }
 326
 327         if (atom_compare(a, b, c, d, 'e', 'd', 't', 's'))
 328                 return ATOM_EDTS;
 329         else if (atom_compare(a, b, c, d, 'e', 's', 'd', 's'))
 330                 return ATOM_ESDS;
 331         else if (atom_compare(a, b, c, d, 'f', 't', 'y', 'p'))
 332                 return ATOM_FTYP;
 333         else if (atom_compare(a, b, c, d, 'f', 'r', 'e', 'e'))
 334                 return ATOM_FREE;
 335         else if (atom_compare(a, b, c, d, 'h', 'm', 'h', 'd'))
 336                 return ATOM_HMHD;
 337         else if (atom_compare(a, b, c, d, 'v', 'm', 'h', 'd'))
 338                 return ATOM_VMHD;
 339         else if (atom_compare(a, b, c, d, 'u', 'd', 't', 'a'))
 340                 return ATOM_UDTA;
 341         else if (atom_compare(a, b, c, d, 'i', 'l', 's', 't'))
 342                 return ATOM_ILST;
 343         else if (atom_compare(a, b, c, d, 'n', 'a', 'm', 'e'))
 344                 return ATOM_NAME;
 345         else if (atom_compare(a, b, c, d, 'd', 'a', 't', 'a'))
 346                 return ATOM_DATA;
 347         else if (atom_compare(a, b, c, d, 'd', 'i', 's', 'k'))
 348                 return ATOM_DISC;
 349         else if (atom_compare(a, b, c, d, 'g', 'n', 'r', 'e'))
 350                 return ATOM_GENRE2;
 351         else if (atom_compare(a, b, c, d, 'c', 'o', 'v', 'r'))
 352                 return ATOM_COVER;
 353         else if (atom_compare(a, b, c, d, 'c', 'p', 'i', 'l'))
 354                 return ATOM_COMPILATION;
 355         else if (atom_compare(a, b, c, d, 'c', 't', 't', 's'))
 356                 return ATOM_CTTS;
 357         else if (atom_compare(a, b, c, d, 'd', 'r', 'm', 's'))
 358                 return ATOM_DRMS;
 359         else if (atom_compare(a, b, c, d, 'f', 'r', 'm', 'a'))
 360                 return ATOM_FRMA;
 361         else if (atom_compare(a, b, c, d, 'p', 'r', 'i', 'v'))
 362                 return ATOM_PRIV;
 363         else if (atom_compare(a, b, c, d, 'i', 'v', 'i', 'v'))
 364                 return ATOM_IVIV;
 365         else if (atom_compare(a, b, c, d, 'u', 's', 'e', 'r'))
 366                 return ATOM_USER;
 367         else if (atom_compare(a, b, c, d, 'k', 'e', 'y', ' '))
 368                 return ATOM_KEY;
 369         else if (atom_compare(a, b, c, d, 'a', 'A', 'R', 'T'))
 370                 return ATOM_ALBUM_ARTIST;
 371         else if (atom_compare(a, b, c, d, 'd', 'e', 's', 'c'))
 372                 return ATOM_DESCRIPTION;
 373         else if (atom_compare(a, b, c, d, 'p', 'c', 's', 't'))
 374                 return ATOM_PODCAST;
 375         else
 376                 return ATOM_UNKNOWN;
 377 }
 378
 379 /* read atom header, atom size is returned with header included. */
 380 static int atom_read_header(struct mp4 *f, uint8_t *atom_type,
 381                 uint8_t *header_size, uint64_t *atom_size)
 382 {
 383         uint32_t size;
 384         int ret;
 385         int8_t atom_header[8];
 386
 387         ret = read_data(f, atom_header, 8);
 388         if (ret <= 0)
 389                 return ret;
 390         size = read_u32_be(atom_header);
 391         if (size == 1) { /* 64 bit atom size */
 392                 if (header_size)
 393                         *header_size = 16;
 394                 ret = read_int64(f, atom_size);
 395                 if (ret <= 0)
 396                         return ret;
 397         } else {
 398                 if (header_size)
 399                         *header_size = 8;
 400                 if (atom_size)
 401                         *atom_size = size;
 402         }
 403         *atom_type = atom_name_to_type(atom_header[4], atom_header[5],
 404                 atom_header[6], atom_header[7]);
 405         return 1;
 406 }
 407
 408 static int64_t get_position(const struct mp4 *f)
 409 {
 410         return f->current_position;
 411 }
 412
 413 static int need_parse_when_meta_only(uint8_t atom_type)
 414 {
 415         switch (atom_type) {
 416         case ATOM_EDTS:
 417         case ATOM_DRMS:
 418         case ATOM_SINF:
 419         case ATOM_SCHI:
 420         case ATOM_STTS:
 421         case ATOM_STSZ:
 422         case ATOM_STZ2:
 423         case ATOM_STCO:
 424         case ATOM_STSC:
 425         case ATOM_FRMA:
 426         case ATOM_IVIV:
 427         case ATOM_PRIV:
 428                 return 0;
 429         default:
 430                 return 1;
 431         }
 432 }
 433
 434 static int32_t set_position(struct mp4 *f, int64_t position)
 435 {
 436         f->cb->seek(f->cb->user_data, position);
 437         f->current_position = position;
 438
 439         return 0;
 440 }
 441
 442 static int read_stsz(struct mp4 *f)
 443 {
 444         int ret;
 445         int32_t i;
 446         struct mp4_track *t;
 447
 448         if (f->total_tracks == 0)
 449                 return -1;
 450         t = f->track[f->total_tracks - 1];
 451         ret = read_int8(f, NULL); /* version */
 452         if (ret <= 0)
 453                 return ret;
 454         ret = read_int24(f, NULL); /* flags */
 455         if (ret <= 0)
 456                 return ret;
 457         ret = read_int32(f, &t->stsz_sample_size);
 458         if (ret <= 0)
 459                 return ret;
 460         ret = read_int32(f, &t->stsz_sample_count);
 461         if (ret <= 0)
 462                 return ret;
 463         if (t->stsz_sample_size != 0)
 464                 return 1;
 465         t->stsz_table = para_malloc(t->stsz_sample_count * sizeof(int32_t));
 466         for (i = 0; i < t->stsz_sample_count; i++) {
 467                 ret = read_int32(f, &t->stsz_table[i]);
 468                 if (ret <= 0)
 469                         return ret;
 470         }
 471         return 1;
 472 }
 473
 474 static int read_stts(struct mp4 *f)
 475 {
 476         int ret;
 477         int32_t i;
 478         struct mp4_track *t;
 479
 480         if (f->total_tracks == 0)
 481                 return -1;
 482         t = f->track[f->total_tracks - 1];
 483         if (t->stts_entry_count)
 484                 return 0;
 485         ret = read_int8(f, NULL); /* version */
 486         if (ret <= 0)
 487                 return ret;
 488         ret = read_int24(f, NULL); /* flags */
 489         if (ret <= 0)
 490                 return ret;
 491         ret = read_int32(f, &t->stts_entry_count);
 492         if (ret <= 0)
 493                 return ret;
 494         t->stts_sample_count = para_malloc(t->stts_entry_count
 495                 * sizeof(int32_t));
 496         t->stts_sample_delta = para_malloc(t->stts_entry_count
 497                 * sizeof (int32_t));
 498         for (i = 0; i < t->stts_entry_count; i++) {
 499                 ret = read_int32(f, &t->stts_sample_count[i]);
 500                 if (ret <= 0)
 501                         return ret;
 502                 ret = read_int32(f, &t->stts_sample_delta[i]);
 503                 if (ret <= 0)
 504                         return ret;
 505         }
 506         return 1;
 507 }
 508
 509 static int read_stsc(struct mp4 *f)
 510 {
 511         int ret;
 512         int32_t i;
 513         struct mp4_track *t;
 514
 515         if (f->total_tracks == 0)
 516                 return -1;
 517         t = f->track[f->total_tracks - 1];
 518
 519         ret = read_int8(f, NULL); /* version */
 520         if (ret <= 0)
 521                 return ret;
 522         ret = read_int24(f, NULL); /* flags */
 523         if (ret <= 0)
 524                 return ret;
 525         ret = read_int32(f, &t->stsc_entry_count);
 526         if (ret <= 0)
 527                 return ret;
 528         t->stsc_first_chunk = para_malloc(t->stsc_entry_count * sizeof(int32_t));
 529         t->stsc_samples_per_chunk = para_malloc(t->stsc_entry_count
 530                 * sizeof (int32_t));
 531         t->stsc_sample_desc_index = para_malloc(t->stsc_entry_count *
 532                 sizeof (int32_t));
 533
 534         for (i = 0; i < t->stsc_entry_count; i++) {
 535                 ret = read_int32(f, &t->stsc_first_chunk[i]);
 536                 if (ret <= 0)
 537                         return ret;
 538                 ret = read_int32(f, &t->stsc_samples_per_chunk[i]);
 539                 if (ret <= 0)
 540                         return ret;
 541                 ret = read_int32(f, &t->stsc_sample_desc_index[i]);
 542                 if (ret <= 0)
 543                         return ret;
 544         }
 545         return 1;
 546 }
 547
 548 static int read_stco(struct mp4 *f)
 549 {
 550         int ret;
 551         int32_t i;
 552         struct mp4_track *t;
 553
 554         if (f->total_tracks == 0)
 555                 return -1;
 556         t = f->track[f->total_tracks - 1];
 557
 558         ret = read_int8(f, NULL); /* version */
 559         if (ret <= 0)
 560                 return ret;
 561         ret = read_int24(f, NULL); /* flags */
 562         if (ret <= 0)
 563                 return ret;
 564         ret = read_int32(f, &t->stco_entry_count);
 565         if (ret <= 0)
 566                 return ret;
 567         t->stco_chunk_offset = para_malloc(t->stco_entry_count
 568                 * sizeof(int32_t));
 569         for (i = 0; i < t->stco_entry_count; i++) {
 570                 ret = read_int32(f, &t->stco_chunk_offset[i]);
 571                 if (ret <= 0)
 572                         return ret;
 573         }
 574         return 1;
 575 }
 576
 577 static int read_mp4a(struct mp4 *f)
 578 {
 579         int ret;
 580         int32_t i;
 581         uint8_t atom_type = 0;
 582         uint8_t header_size = 0;
 583         struct mp4_track *t;
 584
 585         if (f->total_tracks == 0)
 586                 return -1;
 587         t = f->track[f->total_tracks - 1];
 588
 589         for (i = 0; i < 6; i++) {
 590                 ret = read_int8(f, NULL); /* reserved */
 591                 if (ret <= 0)
 592                         return ret;
 593         }
 594         ret = read_int16(f, NULL); /* data_reference_index */
 595         if (ret <= 0)
 596                 return ret;
 597         ret = read_int32(f, NULL); /* reserved */
 598         if (ret <= 0)
 599                 return ret;
 600         ret = read_int32(f, NULL); /* reserved */
 601         if (ret <= 0)
 602                 return ret;
 603         ret = read_int16(f, &t->channelCount);
 604         if (ret <= 0)
 605                 return ret;
 606         ret = read_int16(f, NULL);
 607         if (ret <= 0)
 608                 return ret;
 609         ret = read_int16(f, NULL);
 610         if (ret <= 0)
 611                 return ret;
 612         ret = read_int16(f, NULL);
 613         if (ret <= 0)
 614                 return ret;
 615         ret = read_int16(f, &t->sampleRate);
 616         if (ret <= 0)
 617                 return ret;
 618         ret = read_int16(f, NULL);
 619         if (ret <= 0)
 620                 return ret;
 621         return atom_read_header(f, &atom_type, &header_size, NULL);
 622 }
 623
 624 static int read_stsd(struct mp4 *f)
 625 {
 626         int ret;
 627         uint32_t i, entry_count;
 628         uint8_t header_size = 0;
 629         struct mp4_track *t;
 630
 631         if (f->total_tracks == 0)
 632                 return -1;
 633         t = f->track[f->total_tracks - 1];
 634         ret = read_int8(f, NULL); /* version */
 635         if (ret <= 0)
 636                 return ret;
 637         ret = read_int24(f, NULL); /* flags */
 638         if (ret <= 0)
 639                 return ret;
 640         ret = read_int32(f, &entry_count);
 641         if (ret <= 0)
 642                 return ret;
 643         for (i = 0; i < entry_count; i++) {
 644                 uint64_t skip = get_position(f);
 645                 uint64_t size;
 646                 uint8_t atom_type = 0;
 647                 ret = atom_read_header(f, &atom_type, &header_size, &size);
 648                 if (ret <= 0)
 649                         return ret;
 650                 skip += size;
 651                 if (!f->audio_track && atom_type == ATOM_MP4A) {
 652                         f->audio_track = t;
 653                         read_mp4a(f);
 654                 }
 655                 set_position(f, skip);
 656         }
 657         return 1;
 658 }
 659
 660 static int32_t tag_add_field(struct mp4_metadata *meta, const char *item,
 661                 const char *value, int32_t len)
 662 {
 663         meta->tags = para_realloc(meta->tags,
 664                 (meta->count + 1) * sizeof(struct mp4_tag));
 665         meta->tags[meta->count].item = para_strdup(item);
 666         meta->tags[meta->count].len = len;
 667         if (len >= 0) {
 668                 meta->tags[meta->count].value = para_malloc(len + 1);
 669                 memcpy(meta->tags[meta->count].value, value, len);
 670                 meta->tags[meta->count].value[len] = 0;
 671         } else {
 672                 meta->tags[meta->count].value = para_strdup(value);
 673         }
 674         meta->count++;
 675         return 1;
 676 }
 677
 678 static int read_string(struct mp4 *f, uint32_t length, char **result)
 679 {
 680         char *str = para_malloc(length + 1);
 681         int ret = read_data(f, str, length);
 682
 683         if (ret <= 0) {
 684                 free(str);
 685                 *result = NULL;
 686         } else {
 687                 str[length] = '\0';
 688                 *result = str;
 689         }
 690         return ret;
 691 }
 692
 693 static const char *get_metadata_name(uint8_t atom_type)
 694 {
 695         switch (atom_type) {
 696         case ATOM_TITLE: return "title";
 697         case ATOM_ARTIST: return "artist";
 698         case ATOM_ALBUM: return "album";
 699         case ATOM_DATE: return "date";
 700         case ATOM_COMMENT: return "comment";
 701         default: return "unknown";
 702         }
 703 }
 704
 705 static int parse_tag(struct mp4 *f, uint8_t parent, int32_t size)
 706 {
 707         int ret;
 708         uint64_t subsize, sumsize;
 709         char *data = NULL;
 710         uint32_t len = 0;
 711         uint64_t destpos;
 712
 713         for (
 714                 sumsize = 0;
 715                 sumsize < size;
 716                 set_position(f, destpos), sumsize += subsize
 717         ) {
 718                 uint8_t atom_type;
 719                 uint8_t header_size = 0;
 720                 ret = atom_read_header(f, &atom_type, &header_size, &subsize);
 721                 if (ret <= 0)
 722                         return ret;
 723                 destpos = get_position(f) + subsize - header_size;
 724                 if (atom_type != ATOM_DATA)
 725                         continue;
 726                 ret = read_int8(f, NULL); /* version */
 727                 if (ret <= 0)
 728                         return ret;
 729                 ret = read_int24(f, NULL); /* flags */
 730                 if (ret <= 0)
 731                         return ret;
 732                 ret = read_int32(f, NULL); /* reserved */
 733                 if (ret <= 0)
 734                         return ret;
 735                 free(data);
 736                 ret = read_string(f, subsize - (header_size + 8), &data);
 737                 if (ret <= 0)
 738                         return ret;
 739                 len = subsize - (header_size + 8);
 740         }
 741         if (!data)
 742                 return -1;
 743         tag_add_field(&f->meta, get_metadata_name(parent), data, len);
 744         free(data);
 745         return 1;
 746 }
 747
 748 static int read_mdhd(struct mp4 *f)
 749 {
 750         int ret;
 751         uint32_t version;
 752         struct mp4_track *t;
 753
 754         if (f->total_tracks == 0)
 755                 return -1;
 756         t = f->track[f->total_tracks - 1];
 757
 758         ret = read_int32(f, &version);
 759         if (ret <= 0)
 760                 return ret;
 761         if (version == 1) {
 762                 ret = read_int64(f, NULL); /* creation-time */
 763                 if (ret <= 0)
 764                         return ret;
 765                 ret = read_int64(f, NULL); /* modification-time */
 766                 if (ret <= 0)
 767                         return ret;
 768                 ret = read_int32(f, &t->timeScale);
 769                 if (ret <= 0)
 770                         return ret;
 771                 ret = read_int64(f, &t->duration);
 772                 if (ret <= 0)
 773                         return ret;
 774         } else { //version == 0
 775                 uint32_t temp;
 776
 777                 ret = read_int32(f, NULL); /* creation-time */
 778                 if (ret <= 0)
 779                         return ret;
 780                 ret = read_int32(f, NULL); /* modification-time */
 781                 if (ret <= 0)
 782                         return ret;
 783                 ret = read_int32(f, &t->timeScale);
 784                 if (ret <= 0)
 785                         return ret;
 786                 ret = read_int32(f, &temp);
 787                 if (ret <= 0)
 788                         return ret;
 789                 t->duration = (temp == (uint32_t) (-1))?
 790                         (uint64_t) (-1) : (uint64_t) (temp);
 791         }
 792         ret = read_int16(f, NULL);
 793         if (ret <= 0)
 794                 return ret;
 795         ret = read_int16(f, NULL);
 796         if (ret <= 0)
 797                 return ret;
 798         return 1;
 799 }
 800
 801 static int32_t read_ilst(struct mp4 *f, int32_t size)
 802 {
 803         int ret;
 804         uint64_t sumsize = 0;
 805
 806         while (sumsize < size) {
 807                 uint8_t atom_type;
 808                 uint64_t subsize, destpos;
 809                 uint8_t header_size = 0;
 810                 ret = atom_read_header(f, &atom_type, &header_size, &subsize);
 811                 if (ret <= 0)
 812                         return ret;
 813                 destpos = get_position(f) + subsize - header_size;
 814                 switch (atom_type) {
 815                 case ATOM_ARTIST:
 816                 case ATOM_TITLE:
 817                 case ATOM_ALBUM:
 818                 case ATOM_COMMENT:
 819                 case ATOM_DATE:
 820                         parse_tag(f, atom_type, subsize - header_size);
 821                 }
 822                 set_position(f, destpos);
 823                 sumsize += subsize;
 824         }
 825         return 1;
 826 }
 827
 828 static int32_t read_meta(struct mp4 *f, uint64_t size)
 829 {
 830         int ret;
 831         uint64_t subsize, sumsize = 0;
 832         uint8_t atom_type;
 833         uint8_t header_size = 0;
 834
 835         ret = read_int8(f, NULL); /* version */
 836         if (ret <= 0)
 837                 return ret;
 838         ret = read_int24(f, NULL); /* flags */
 839         if (ret <= 0)
 840                 return ret;
 841         while (sumsize < (size - (header_size + 4))) {
 842                 ret = atom_read_header(f, &atom_type, &header_size, &subsize);
 843                 if (ret <= 0)
 844                         return ret;
 845                 if (subsize <= header_size + 4)
 846                         return 1;
 847                 if (atom_type == ATOM_ILST)
 848                         read_ilst(f, subsize - (header_size + 4));
 849                 else
 850                         set_position(f, get_position(f) + subsize - header_size);
 851                 sumsize += subsize;
 852         }
 853         return 1;
 854 }
 855
 856 static int atom_read(struct mp4 *f, uint64_t size, uint8_t atom_type)
 857 {
 858         uint64_t dest_position = get_position(f) + size - 8;
 859         int ret = 1; /* return success for atoms we don't care about */
 860
 861         switch (atom_type) {
 862         case ATOM_STSZ: ret = read_stsz(f); break;
 863         case ATOM_STTS: ret = read_stts(f); break;
 864         case ATOM_STSC: ret = read_stsc(f); break;
 865         case ATOM_STCO: ret = read_stco(f); break;
 866         case ATOM_STSD: ret = read_stsd(f); break;
 867         case ATOM_MDHD: ret = read_mdhd(f); break;
 868         case ATOM_META: ret = read_meta(f, size); break;
 869         }
 870         set_position(f, dest_position);
 871         return ret;
 872 }
 873
 874 /* parse atoms that are sub atoms of other atoms */
 875 static int parse_sub_atoms(struct mp4 *f, uint64_t total_size, int meta_only)
 876 {
 877         int ret;
 878         uint64_t size;
 879         uint8_t atom_type = 0;
 880         uint64_t counted_size = 0;
 881         uint8_t header_size = 0;
 882
 883         while (counted_size < total_size) {
 884                 ret = atom_read_header(f, &atom_type, &header_size, &size);
 885                 if (ret <= 0)
 886                         return ret;
 887                 if (size == 0)
 888                         return -1;
 889                 counted_size += size;
 890                 if (atom_type == ATOM_TRAK) {
 891                         if (f->total_tracks >= MAX_TRACKS)
 892                                 return -1;
 893                         f->total_tracks++;
 894                         f->track[f->total_tracks - 1] = para_calloc(
 895                                 sizeof(struct mp4_track));
 896                 }
 897                 /* parse subatoms */
 898                 if (meta_only && !need_parse_when_meta_only(atom_type)) {
 899                         set_position(f, get_position(f) + size - header_size);
 900                 } else if (atom_type < SUBATOMIC) {
 901                         ret = parse_sub_atoms(f, size - header_size, meta_only);
 902                         if (ret <= 0)
 903                                 return ret;
 904                 } else {
 905                         ret = atom_read(f, size, atom_type);
 906                         if (ret <= 0)
 907                                 return ret;
 908                 }
 909         }
 910         return 1;
 911 }
 912
 913 /* parse root atoms */
 914 static int parse_atoms(struct mp4 *f, int meta_only)
 915 {
 916         int ret;
 917         uint64_t size;
 918         uint8_t atom_type = 0;
 919         uint8_t header_size = 0;
 920
 921         f->file_size = 0;
 922
 923         while ((ret = atom_read_header(f, &atom_type, &header_size, &size)) > 0) {
 924                 f->file_size += size;
 925                 f->last_atom = atom_type;
 926
 927                 if (atom_type == ATOM_MOOV && size > header_size) {
 928                         f->moov_offset = get_position(f) - header_size;
 929                         f->moov_size = size;
 930                 }
 931
 932                 /* parse subatoms */
 933                 if (meta_only && !need_parse_when_meta_only(atom_type)) {
 934                         set_position(f, get_position(f) + size - header_size);
 935                 } else if (atom_type < SUBATOMIC) {
 936                         ret = parse_sub_atoms(f, size - header_size, meta_only);
 937                         if (ret <= 0)
 938                                 break;
 939                 } else {
 940                         /* skip this atom */
 941                         set_position(f, get_position(f) + size - header_size);
 942                 }
 943         }
 944         if (ret < 0)
 945                 return ret;
 946         if (!f->audio_track)
 947                 return -E_MP4_TRACK;
 948         return ret;
 949 }
 950
 951 struct mp4 *mp4_open_read(const struct mp4_callback *cb)
 952 {
 953         int ret;
 954         struct mp4 *f = para_calloc(sizeof(struct mp4));
 955
 956         f->cb = cb;
 957         ret = parse_atoms(f, 0);
 958         if (ret < 0) {
 959                 free(f);
 960                 return NULL;
 961         }
 962         return f;
 963 }
 964
 965 void mp4_close(struct mp4 *f)
 966 {
 967         int32_t i;
 968
 969         for (i = 0; i < f->total_tracks; i++) {
 970                 if (f->track[i]) {
 971                         free(f->track[i]->stsz_table);
 972                         free(f->track[i]->stts_sample_count);
 973                         free(f->track[i]->stts_sample_delta);
 974                         free(f->track[i]->stsc_first_chunk);
 975                         free(f->track[i]->stsc_samples_per_chunk);
 976                         free(f->track[i]->stsc_sample_desc_index);
 977                         free(f->track[i]->stco_chunk_offset);
 978                         free(f->track[i]);
 979                 }
 980         }
 981         for (i = 0; i < f->meta.count; i++) {
 982                 free(f->meta.tags[i].item);
 983                 free(f->meta.tags[i].value);
 984         }
 985         free(f->meta.tags);
 986         free(f);
 987 }
 988
 989 static int32_t chunk_of_sample(const struct mp4 *f, int32_t sample,
 990                 int32_t *chunk)
 991 {
 992         const struct mp4_track *t = f->audio_track;
 993         uint32_t *fc = t->stsc_first_chunk, *spc = t->stsc_samples_per_chunk;
 994         int32_t chunk1, chunk1samples, n, total, i;
 995
 996         for (i = 1, total = 0; i < t->stsc_entry_count; i++, total += n) {
 997                 n = (fc[i] - fc[i - 1]) * spc[i - 1]; /* number of samples */
 998                 if (sample < total + n)
 999                         break;
1000         }
1001         chunk1 = fc[i - 1];
1002         chunk1samples = spc[i - 1];
1003         if (chunk1samples != 0)
1004                 *chunk = (sample - total) / chunk1samples + chunk1;
1005         else
1006                 *chunk = 1;
1007         return total + (*chunk - chunk1) * chunk1samples;
1008 }
1009
1010 /**
1011  * Return the number of milliseconds of the audio track.
1012  *
1013  * \param f As returned by \ref mp4_open_read(), must not be NULL.
1014  */
1015 uint64_t mp4_get_duration(const struct mp4 *f)
1016 {
1017         const struct mp4_track *t = f->audio_track;
1018
1019         if (t->timeScale == 0)
1020                 return 0;
1021         return t->duration * 1000 / t->timeScale;
1022 }
1023
1024 int mp4_set_sample_position(struct mp4 *f, int32_t sample)
1025 {
1026         const struct mp4_track *t = f->audio_track;
1027         int32_t offset, chunk, chunk_sample;
1028         uint32_t n, srs; /* sample range size */
1029
1030         if (sample >= t->stsz_sample_count)
1031                 return -ERRNO_TO_PARA_ERROR(EINVAL);
1032         chunk_sample = chunk_of_sample(f, sample, &chunk);
1033         if (t->stsz_sample_size > 0)
1034                 srs = (sample - chunk_sample) * t->stsz_sample_size;
1035         else {
1036                 for (srs = 0, n = chunk_sample; n < sample; n++)
1037                         srs += t->stsz_table[n];
1038         }
1039         if (t->stco_entry_count > 0 && chunk > t->stco_entry_count)
1040                 offset = t->stco_chunk_offset[t->stco_entry_count - 1];
1041         else if (t->stco_entry_count > 0)
1042                 offset = t->stco_chunk_offset[chunk - 1];
1043         else
1044                 offset = 8;
1045         set_position(f, offset + srs);
1046         return 1;
1047 }
1048
1049 int32_t mp4_get_sample_size(const struct mp4 *f, int sample)
1050 {
1051         const struct mp4_track *t = f->audio_track;
1052
1053         if (t->stsz_sample_size != 0)
1054                 return t->stsz_sample_size;
1055         return t->stsz_table[sample];
1056 }
1057
1058 uint32_t mp4_get_sample_rate(const struct mp4 *f)
1059 {
1060         return f->audio_track->sampleRate;
1061 }
1062
1063 uint32_t mp4_get_channel_count(const struct mp4 *f)
1064 {
1065         return f->audio_track->channelCount ;
1066 }
1067
1068 int32_t mp4_num_samples(const struct mp4 *f)
1069 {
1070         const struct mp4_track *t = f->audio_track;
1071         int32_t i;
1072         int32_t total = 0;
1073
1074         for (i = 0; i < t->stts_entry_count; i++)
1075                 total += t->stts_sample_count[i];
1076         return total;
1077 }
1078
1079 struct mp4 *mp4_open_meta(const struct mp4_callback *cb)
1080 {
1081         int ret;
1082         struct mp4 *f = para_calloc(sizeof(struct mp4));
1083
1084         f->cb = cb;
1085         ret = parse_atoms(f, 1);
1086         if (ret < 0) {
1087                 free(f);
1088                 return NULL;
1089         }
1090         return f;
1091 }
1092
1093 /**
1094  * Return the metadata of an mp4 file.
1095  *
1096  * \param f As returned by either \ref mp4_open_read() or \ref mp4_open_meta().
1097  *
1098  * The caller is allowed to add, delete or modify the entries of the returned
1099  * structure in order to pass the modified version to \ref mp4_meta_update().
1100  */
1101 struct mp4_metadata *mp4_get_meta(struct mp4 *f)
1102 {
1103         return &f->meta;
1104 }
1105
1106 static int find_atom(struct mp4 *f, uint64_t base, uint32_t size,
1107         const char *name)
1108 {
1109         uint32_t remaining = size;
1110         uint64_t atom_offset = base;
1111
1112         for (;;) {
1113                 int ret;
1114                 char atom_name[4];
1115                 uint32_t atom_size;
1116
1117                 set_position(f, atom_offset);
1118
1119                 if (remaining < 8)
1120                         return -1;
1121                 ret = read_int32(f, &atom_size);
1122                 if (ret <= 0)
1123                         return ret;
1124                 if (atom_size > remaining || atom_size < 8)
1125                         return -1;
1126                 ret = read_data(f, atom_name, 4);
1127                 if (ret <= 0)
1128                         return ret;
1129                 if (!memcmp(atom_name, name, 4)) {
1130                         set_position(f, atom_offset);
1131                         return 1;
1132                 }
1133                 remaining -= atom_size;
1134                 atom_offset += atom_size;
1135         }
1136 }
1137
1138 /*
1139  * Try to find atom <name> with atom <name_inside> in it. Besides -1/0/1 for
1140  * error, EOF and success, this function may return 2 to indicate that the
1141  * desired atoms were not found.
1142  */
1143 static int find_atom_v2(struct mp4 *f, uint64_t base, uint32_t size,
1144                 const char *name, uint32_t extraheaders, const char *name_inside)
1145 {
1146         uint64_t first_base = (uint64_t) (-1);
1147
1148         for (;;) {
1149                 uint64_t mybase;
1150                 uint32_t mysize;
1151                 int ret = find_atom(f, base, size, name);
1152
1153                 if (ret <= 0)
1154                         return ret;
1155                 mybase = get_position(f);
1156                 ret = read_int32(f, &mysize);
1157                 if (ret <= 0)
1158                         return ret;
1159                 if (first_base == (uint64_t) (-1))
1160                         first_base = mybase;
1161
1162                 if (mysize < 8 + extraheaders)
1163                         break;
1164
1165                 if (find_atom (f, mybase + (8 + extraheaders),
1166                                 mysize - (8 + extraheaders), name_inside)) {
1167                         set_position(f, mybase);
1168                         return 1;
1169                 }
1170                 base += mysize;
1171                 if (size <= mysize)
1172                         break;
1173                 size -= mysize;
1174         }
1175         if (first_base != (uint64_t)(-1)) {
1176                 set_position(f, first_base);
1177                 return 1;
1178         }
1179         /* wanted atom inside not found */
1180         return 2;
1181 }
1182
1183 struct membuffer {
1184         void *data;
1185         unsigned written;
1186         unsigned allocated;
1187 };
1188
1189 static struct membuffer *membuffer_create(void)
1190 {
1191         struct membuffer *buf = para_calloc(sizeof(*buf));
1192
1193         buf->allocated = 256;
1194         buf->data = para_malloc(buf->allocated);
1195         return buf;
1196 }
1197
1198 static void membuffer_write(struct membuffer *buf, const void *ptr,
1199                 unsigned bytes)
1200 {
1201         unsigned dest_size = buf->written + bytes;
1202
1203         if (dest_size > buf->allocated) {
1204                 do {
1205                         buf->allocated <<= 1;
1206                 } while (dest_size > buf->allocated);
1207                 buf->data = para_realloc(buf->data, buf->allocated);
1208         }
1209
1210         if (ptr)
1211                 memcpy((char *) buf->data + buf->written, ptr, bytes);
1212         buf->written += bytes;
1213 }
1214
1215 static void membuffer_write_atom_name(struct membuffer *buf, const char *data)
1216 {
1217         membuffer_write(buf, data, 4);
1218 }
1219
1220 static void membuffer_write_int32(struct membuffer *buf, uint32_t data)
1221 {
1222         uint8_t temp[4];
1223         write_u32_be(temp, data);
1224         membuffer_write(buf, temp, 4);
1225 }
1226
1227 static void membuffer_write_std_tag(struct membuffer *buf, const char *name,
1228                 const char *value)
1229 {
1230         uint32_t len = strlen(value);
1231         membuffer_write_int32(buf, 8 /* atom header */
1232                 + 8 /* data atom header */
1233                 + 8 /* flags + reserved */
1234                 + len);
1235         membuffer_write_atom_name(buf, name);
1236         membuffer_write_int32(buf, 8 /* data atom header */
1237                 + 8 /* flags + reserved */
1238                 + len);
1239         membuffer_write_atom_name(buf, "data");
1240         membuffer_write_int32(buf, 1);  /* flags */
1241         membuffer_write_int32(buf, 0);  /* reserved */
1242         membuffer_write(buf, value, len);
1243 }
1244
1245 static unsigned membuffer_get_size(const struct membuffer *buf)
1246 {
1247         return buf->written;
1248 }
1249
1250 static void *membuffer_detach(struct membuffer *buf)
1251 {
1252         void *ret = para_realloc(buf->data, buf->written);
1253         free(buf);
1254         return ret;
1255 }
1256
1257 struct stdmeta_entry {
1258         const char *atom;
1259         const char *name;
1260 };
1261
1262 static const char *find_standard_meta(const char *name)
1263 {
1264         const struct stdmeta_entry stdmetas[] = {
1265                 {"\xA9" "nam", "title"},
1266                 {"\xA9" "ART", "artist"},
1267                 {"\xA9" "alb", "album"},
1268                 {"\xA9" "day", "date"},
1269                 {"\xA9" "cmt", "comment"},
1270         };
1271
1272         for (unsigned n = 0; n < ARRAY_SIZE(stdmetas); n++)
1273                 if (!strcasecmp(name, stdmetas[n].name))
1274                         return stdmetas[n].atom;
1275         return NULL;
1276 }
1277
1278 static uint32_t create_ilst(const struct mp4_metadata *meta, void **out_buffer,
1279                 uint32_t * out_size)
1280 {
1281         struct membuffer *buf = membuffer_create();
1282         unsigned metaptr;
1283
1284         for (metaptr = 0; metaptr < meta->count; metaptr++) {
1285                 struct mp4_tag *tag = meta->tags + metaptr;
1286                 const char *std_meta_atom = find_standard_meta(tag->item);
1287                 if (std_meta_atom)
1288                         membuffer_write_std_tag(buf, std_meta_atom, tag->value);
1289                 else
1290                         PARA_ERROR_LOG("invalid tag item: %s\n", tag->item);
1291         }
1292         *out_size = membuffer_get_size(buf);
1293         *out_buffer = membuffer_detach(buf);
1294         return 1;
1295 }
1296
1297 static void membuffer_write_atom(struct membuffer *buf, const char *name, unsigned size,
1298                           const void *data)
1299 {
1300         membuffer_write_int32(buf, size + 8);
1301         membuffer_write_atom_name(buf, name);
1302         membuffer_write(buf, data, size);
1303 }
1304
1305 static void *membuffer_get_ptr(const struct membuffer *buf)
1306 {
1307         return buf->data;
1308 }
1309
1310 static bool membuffer_transfer_from_file(struct membuffer *buf, struct mp4 *src,
1311                 unsigned bytes)
1312 {
1313         unsigned oldsize = membuffer_get_size(buf);
1314         char *bufptr;
1315
1316         membuffer_write(buf, 0, bytes);
1317         bufptr = membuffer_get_ptr(buf);
1318         if (read_data(src, bufptr + oldsize, bytes) != 1) {
1319                 free(buf->data);
1320                 free(buf);
1321                 return false;
1322         }
1323         return true;
1324 }
1325
1326 static uint32_t create_meta(const struct mp4_metadata *meta, void **out_buffer,
1327                 uint32_t * out_size)
1328 {
1329         struct membuffer *buf;
1330         uint32_t ilst_size;
1331         void *ilst_buffer;
1332
1333         if (!create_ilst(meta, &ilst_buffer, &ilst_size))
1334                 return 0;
1335
1336         buf = membuffer_create();
1337
1338         membuffer_write_int32(buf, 0);
1339         membuffer_write_atom(buf, "ilst", ilst_size, ilst_buffer);
1340         free(ilst_buffer);
1341
1342         *out_size = membuffer_get_size(buf);
1343         *out_buffer = membuffer_detach(buf);
1344         return 1;
1345 }
1346
1347 static uint32_t create_udta(const struct mp4_metadata *meta, void **out_buffer,
1348 uint32_t * out_size)
1349 {
1350         struct membuffer *buf;
1351         uint32_t meta_size;
1352         void *meta_buffer;
1353
1354         if (!create_meta(meta, &meta_buffer, &meta_size))
1355                 return 0;
1356
1357         buf = membuffer_create();
1358
1359         membuffer_write_atom(buf, "meta", meta_size, meta_buffer);
1360
1361         free(meta_buffer);
1362
1363         *out_size = membuffer_get_size(buf);
1364         *out_buffer = membuffer_detach(buf);
1365         return 1;
1366 }
1367
1368 static uint32_t fix_byte_order_32(uint32_t src)
1369 {
1370         return read_u32_be(&src);
1371 }
1372
1373 static void *modify_moov(struct mp4 *f, uint32_t *out_size)
1374 {
1375         int ret;
1376         uint64_t total_base = f->moov_offset + 8;
1377         uint32_t total_size = (uint32_t) (f->moov_size - 8);
1378         uint64_t udta_offset, meta_offset, ilst_offset;
1379         uint32_t udta_size, meta_size, ilst_size;
1380         uint32_t new_ilst_size;
1381         void *new_ilst_buffer, *out_buffer;
1382         uint8_t *p_out;
1383         int32_t size_delta;
1384         uint32_t tmp;
1385
1386         ret = find_atom_v2(f, total_base, total_size, "udta", 0, "meta");
1387         if (ret <= 0)
1388                 return NULL;
1389         if (ret == 2) {
1390                 struct membuffer *buf;
1391                 void *new_udta_buffer;
1392                 uint32_t new_udta_size;
1393                 if (!create_udta(&f->meta, &new_udta_buffer, &new_udta_size))
1394                         return NULL;
1395
1396                 buf = membuffer_create();
1397                 set_position(f, total_base);
1398                 if (!membuffer_transfer_from_file(buf, f, total_size)) {
1399                         free(new_udta_buffer);
1400                         return NULL;
1401                 }
1402                 membuffer_write_atom(buf, "udta", new_udta_size,
1403                         new_udta_buffer);
1404
1405                 free(new_udta_buffer);
1406
1407                 *out_size = membuffer_get_size(buf);
1408                 return membuffer_detach(buf);
1409         }
1410         udta_offset = get_position(f);
1411         ret = read_int32(f, &udta_size);
1412         if (ret <= 0)
1413                 return NULL;
1414         ret = find_atom_v2(f, udta_offset + 8, udta_size - 8, "meta", 4, "ilst");
1415         if (ret <= 0)
1416                 return NULL;
1417         if (ret == 2) {
1418                 struct membuffer *buf;
1419                 void *new_meta_buffer;
1420                 uint32_t new_meta_size;
1421
1422                 if (!create_meta(&f->meta, &new_meta_buffer, &new_meta_size))
1423                         return NULL;
1424
1425                 buf = membuffer_create();
1426                 set_position(f, total_base);
1427                 if (!membuffer_transfer_from_file(buf, f,
1428                                 udta_offset - total_base)) {
1429                         free(new_meta_buffer);
1430                         return NULL;
1431                 }
1432
1433                 membuffer_write_int32(buf, udta_size + 8 + new_meta_size);
1434                 membuffer_write_atom_name(buf, "udta");
1435                 if (!membuffer_transfer_from_file(buf, f, udta_size)) {
1436                         free(new_meta_buffer);
1437                         return NULL;
1438                 }
1439                 membuffer_write_atom(buf, "meta", new_meta_size,
1440                         new_meta_buffer);
1441                 free(new_meta_buffer);
1442
1443                 *out_size = membuffer_get_size(buf);
1444                 return membuffer_detach(buf);
1445         }
1446         meta_offset = get_position(f);
1447         ret = read_int32(f, &meta_size);
1448         if (ret <= 0)
1449                 return NULL;
1450         /* shouldn't happen, find_atom_v2 above takes care of it */
1451         if (!find_atom(f, meta_offset + 12, meta_size - 12, "ilst"))
1452                 return NULL;
1453         ilst_offset = get_position(f);
1454         ret = read_int32(f, &ilst_size);
1455         if (ret <= 0)
1456                 return NULL;
1457         if (!create_ilst(&f->meta, &new_ilst_buffer, &new_ilst_size))
1458                 return NULL;
1459         size_delta = new_ilst_size - (ilst_size - 8);
1460         *out_size = total_size + size_delta;
1461         out_buffer = para_malloc(*out_size);
1462         p_out = out_buffer;
1463         set_position(f, total_base);
1464         ret = read_data(f, p_out, udta_offset - total_base);
1465         if (ret <= 0)
1466                 return NULL;
1467         p_out += (uint32_t) (udta_offset - total_base);
1468         ret = read_int32(f, &tmp);
1469         if (ret <= 0)
1470                 return NULL;
1471         *(uint32_t *)p_out = fix_byte_order_32(tmp + size_delta);
1472         p_out += 4;
1473         ret = read_data(f, p_out, 4);
1474         if (ret <= 0)
1475                 return NULL;
1476         p_out += 4;
1477         ret = read_data(f, p_out, meta_offset - udta_offset - 8);
1478         if (ret <= 0)
1479                 return NULL;
1480         p_out += (uint32_t) (meta_offset - udta_offset - 8);
1481         ret = read_int32(f, &tmp);
1482         if (ret <= 0)
1483                 return NULL;
1484         *(uint32_t *)p_out = fix_byte_order_32(tmp + size_delta);
1485         p_out += 4;
1486         ret = read_data(f, p_out, 4);
1487         if (ret <= 0)
1488                 return NULL;
1489         p_out += 4;
1490         ret = read_data(f, p_out, ilst_offset - meta_offset - 8);
1491         if (ret <= 0)
1492                 return NULL;
1493         p_out += (uint32_t) (ilst_offset - meta_offset - 8);
1494         ret = read_int32(f, &tmp);
1495         if (ret <= 0)
1496                 return NULL;
1497         *(uint32_t *)p_out = fix_byte_order_32(tmp + size_delta);
1498         p_out += 4;
1499         ret = read_data(f, p_out, 4);
1500         if (ret <= 0)
1501                 return NULL;
1502         p_out += 4;
1503         memcpy(p_out, new_ilst_buffer, new_ilst_size);
1504         p_out += new_ilst_size;
1505         set_position(f, ilst_offset + ilst_size);
1506         ret = read_data(f, p_out, total_size
1507                 - (ilst_offset - total_base) - ilst_size);
1508         if (ret <= 0)
1509                 return NULL;
1510         free(new_ilst_buffer);
1511         return out_buffer;
1512 }
1513
1514 static int32_t write_data(struct mp4 *f, void *data, uint32_t size)
1515 {
1516         int32_t result = 1;
1517
1518         result = f->cb->write(f->cb->user_data, data, size);
1519
1520         f->current_position += size;
1521
1522         return result;
1523 }
1524
1525 static int32_t write_int32(struct mp4 *f, uint32_t data)
1526 {
1527         int8_t temp[4];
1528         write_u32_be(temp, data);
1529         return write_data(f, temp, sizeof(temp));
1530 }
1531
1532 int32_t mp4_meta_update(struct mp4 *f)
1533 {
1534         void *new_moov_data;
1535         uint32_t new_moov_size;
1536
1537         set_position(f, 0);
1538         new_moov_data = modify_moov(f, &new_moov_size);
1539         if (!new_moov_data ) {
1540                 mp4_close(f);
1541                 return 0;
1542         }
1543         /* copy moov atom to end of the file */
1544         if (f->last_atom != ATOM_MOOV) {
1545                 char *free_data = "free";
1546
1547                 /* rename old moov to free */
1548                 set_position(f, f->moov_offset + 4);
1549                 write_data(f, free_data, 4);
1550
1551                 set_position(f, f->file_size);
1552                 write_int32(f, new_moov_size + 8);
1553                 write_data(f, "moov", 4);
1554                 write_data(f, new_moov_data, new_moov_size);
1555         } else {
1556                 set_position(f, f->moov_offset);
1557                 write_int32(f, new_moov_size + 8);
1558                 write_data(f, "moov", 4);
1559                 write_data(f, new_moov_data, new_moov_size);
1560         }
1561         free(new_moov_data);
1562         f->cb->truncate(f->cb->user_data);
1563         return 1;
1564 }
1565
1566 static char *meta_find_by_name(const struct mp4 *f, const char *item)
1567 {
1568         uint32_t i;
1569
1570         for (i = 0; i < f->meta.count; i++)
1571                 if (!strcasecmp(f->meta.tags[i].item, item))
1572                         return para_strdup(f->meta.tags[i].value);
1573         return NULL;
1574 }
1575
1576 /**
1577  * Return the value of the artist meta tag of an mp4 file.
1578  *
1579  * \param f Must not be NULL.
1580  *
1581  * \return If the file does not contain this metadata tag, the function returns
1582  * NULL. Otherwise, a copy of the tag value is returned. The caller should free
1583  * this memory when it is no longer needed.
1584  */
1585 char *mp4_meta_get_artist(const struct mp4 *f)
1586 {
1587         return meta_find_by_name(f, "artist");
1588 }
1589
1590 /**
1591  * Return the value of the title meta tag of an mp4 file.
1592  *
1593  * \param f See \ref mp4_meta_get_artist().
1594  * \return See \ref mp4_meta_get_artist().
1595  */
1596 char *mp4_meta_get_title(const struct mp4 *f)
1597 {
1598         return meta_find_by_name(f, "title");
1599 }
1600
1601 /**
1602  * Return the value of the date meta tag of an mp4 file.
1603  *
1604  * \param f See \ref mp4_meta_get_artist().
1605  * \return See \ref mp4_meta_get_artist().
1606  */
1607 char *mp4_meta_get_date(const struct mp4 *f)
1608 {
1609         return meta_find_by_name(f, "date");
1610 }
1611
1612 /**
1613  * Return the value of the album meta tag of an mp4 file.
1614  *
1615  * \param f See \ref mp4_meta_get_artist().
1616  * \return See \ref mp4_meta_get_artist().
1617  */
1618 char *mp4_meta_get_album(const struct mp4 *f)
1619 {
1620         return meta_find_by_name(f, "album");
1621 }
1622
1623 /**
1624  * Return the value of the comment meta tag of an mp4 file.
1625  *
1626  * \param f See \ref mp4_meta_get_artist().
1627  * \return See \ref mp4_meta_get_artist().
1628  */
1629 char *mp4_meta_get_comment(const struct mp4 *f)
1630 {
1631         return meta_find_by_name(f, "comment");
1632 }