#include <regex.h>
#include "para.h"
+#include "error.h"
+#include "portable_io.h"
#include "string.h"
#include "mp4.h"
-int32_t mp4ff_total_tracks(const mp4ff_t * f)
-{
- return f->total_tracks;
-}
-
-static int32_t mp4ff_read_data(mp4ff_t * f, void *data, uint32_t size)
-{
- int32_t result = 1;
+struct mp4_track {
+ /* mp4a */
+ uint16_t channel_count;
+ uint16_t sample_rate;
- result = f->stream->read(f->stream->user_data, data, size);
+ /* stsz */
+ uint32_t stsz_sample_size;
+ uint32_t stsz_sample_count;
+ uint32_t *stsz_table;
- if (result < size)
- f->stream->read_error++;
+ /* stts */
+ uint32_t stts_entry_count;
+ uint32_t *stts_sample_count;
- f->current_position += size;
+ /* stsc */
+ uint32_t stsc_entry_count;
+ uint32_t *stsc_first_chunk;
+ uint32_t *stsc_samples_per_chunk;
- return result;
-}
+ /* stsc */
+ uint32_t stco_entry_count;
+ uint32_t *stco_chunk_offset;
-/* parse atom header size */
-static int32_t mp4ff_atom_get_size(const int8_t * data)
-{
- uint32_t result;
- uint32_t a, b, c, d;
+ /* mdhd */
+ uint32_t time_scale;
+ uint64_t duration;
+};
- a = (uint8_t) data[0];
- b = (uint8_t) data[1];
- c = (uint8_t) data[2];
- d = (uint8_t) data[3];
+#define MAX_TRACKS 1024
- result = (a << 24) | (b << 16) | (c << 8) | d;
- return (int32_t) result;
-}
+struct mp4 {
+ const struct mp4_callback *cb;
+ int64_t current_position;
-static uint64_t mp4ff_read_int64(mp4ff_t * f)
-{
- uint8_t data[8];
- uint64_t result = 0;
- int8_t i;
+ uint64_t moov_offset;
+ uint64_t moov_size;
+ uint64_t meta_offset;
+ uint32_t meta_size;
+ uint64_t ilst_offset;
+ uint32_t ilst_size;
+ uint64_t udta_offset;
+ uint32_t udta_size;
- mp4ff_read_data(f, data, 8);
+ uint8_t last_atom;
+ uint64_t file_size;
- for (i = 0; i < 8; i++) {
- result |= ((uint64_t) data[i]) << ((7 - i) * 8);
- }
+ /* incremental track index while reading the file */
+ int32_t total_tracks;
+ /* track data */
+ struct mp4_track *track[MAX_TRACKS];
+ /* the first audio track found */
+ struct mp4_track *audio_track;
- return result;
-}
+ /* metadata */
+ struct mp4_metadata meta;
+};
-/* comnapre 2 atom names, returns 1 for equal, 0 for unequal */
-static int32_t mp4ff_atom_compare(const int8_t a1, const int8_t b1,
- const int8_t c1, const int8_t d1,
- const int8_t a2, const int8_t b2,
- const int8_t c2, const int8_t d2)
+/*
+ * Returns -1, 0, or 1 on errors/EOF/success. Partial reads followed by EOF or
+ * read errors are treated as errors.
+ */
+static int read_data(struct mp4 *f, void *data, size_t size)
{
- if (a1 == a2 && b1 == b2 && c1 == c2 && d1 == d2)
- return 1;
- else
- return 0;
+ while (size > 0) {
+ ssize_t ret = f->cb->read(f->cb->user_data, data, size);
+ if (ret < 0 && errno == EINTR)
+ continue;
+ /* regard EAGAIN as an error as reads should be blocking. */
+ if (ret <= 0)
+ return ret < 0? -1 : 0;
+ f->current_position += ret;
+ size -= ret;
+ }
+ return 1;
}
-#define TRACK_AUDIO 1
-#define TRACK_VIDEO 2
-#define TRACK_SYSTEM 3
-#define TRACK_UNKNOWN 0
-
-/* atoms with subatoms */
-#define ATOM_MOOV 1
-#define ATOM_TRAK 2
-#define ATOM_EDTS 3
-#define ATOM_MDIA 4
-#define ATOM_MINF 5
-#define ATOM_STBL 6
-#define ATOM_UDTA 7
-#define ATOM_ILST 8 /* iTunes Metadata list */
-#define ATOM_TITLE 9
-#define ATOM_ARTIST 10
-#define ATOM_WRITER 11
-#define ATOM_ALBUM 12
-#define ATOM_DATE 13
-#define ATOM_TOOL 14
-#define ATOM_COMMENT 15
-#define ATOM_GENRE1 16
-#define ATOM_TRACK 17
-#define ATOM_DISC 18
-#define ATOM_COMPILATION 19
-#define ATOM_GENRE2 20
-#define ATOM_TEMPO 21
-#define ATOM_COVER 22
-#define ATOM_DRMS 23
-#define ATOM_SINF 24
-#define ATOM_SCHI 25
-
-#define SUBATOMIC 128
-
-/* atoms without subatoms */
-#define ATOM_FTYP 129
-#define ATOM_MDAT 130
-#define ATOM_MVHD 131
-#define ATOM_TKHD 132
-#define ATOM_TREF 133
-#define ATOM_MDHD 134
-#define ATOM_VMHD 135
-#define ATOM_SMHD 136
-#define ATOM_HMHD 137
-#define ATOM_STSD 138
-#define ATOM_STTS 139
-#define ATOM_STSZ 140
-#define ATOM_STZ2 141
-#define ATOM_STCO 142
-#define ATOM_STSC 143
-#define ATOM_MP4A 144
-#define ATOM_MP4V 145
-#define ATOM_MP4S 146
-#define ATOM_ESDS 147
-#define ATOM_META 148 /* iTunes Metadata box */
-#define ATOM_NAME 149 /* iTunes Metadata name box */
-#define ATOM_DATA 150 /* iTunes Metadata data box */
-#define ATOM_CTTS 151
-#define ATOM_FRMA 152
-#define ATOM_IVIV 153
-#define ATOM_PRIV 154
-#define ATOM_USER 155
-#define ATOM_KEY 156
-#define ATOM_ALBUM_ARTIST 157
-#define ATOM_CONTENTGROUP 158
-#define ATOM_LYRICS 159
-#define ATOM_DESCRIPTION 160
-#define ATOM_NETWORK 161
-#define ATOM_SHOW 162
-#define ATOM_EPISODENAME 163
-#define ATOM_SORTTITLE 164
-#define ATOM_SORTALBUM 165
-#define ATOM_SORTARTIST 166
-#define ATOM_SORTALBUMARTIST 167
-#define ATOM_SORTWRITER 168
-#define ATOM_SORTSHOW 169
-#define ATOM_SEASON 170
-#define ATOM_EPISODE 171
-#define ATOM_PODCAST 172
-
-#define ATOM_UNKNOWN 255
-#define ATOM_FREE ATOM_UNKNOWN
-#define ATOM_SKIP ATOM_UNKNOWN
-
-#define COPYRIGHT_SYMBOL ((int8_t)0xA9)
-
-static uint8_t mp4ff_atom_name_to_type(const int8_t a, const int8_t b,
- const int8_t c, const int8_t d)
+static int read_int64(struct mp4 *f, uint64_t *result)
{
- if (a == 'm') {
- if (mp4ff_atom_compare(a, b, c, d, 'm', 'o', 'o', 'v'))
- return ATOM_MOOV;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'i', 'n', 'f'))
- return ATOM_MINF;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'd', 'i', 'a'))
- return ATOM_MDIA;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'd', 'a', 't'))
- return ATOM_MDAT;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'd', 'h', 'd'))
- return ATOM_MDHD;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'v', 'h', 'd'))
- return ATOM_MVHD;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'p', '4', 'a'))
- return ATOM_MP4A;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'p', '4', 'v'))
- return ATOM_MP4V;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'p', '4', 's'))
- return ATOM_MP4S;
- else if (mp4ff_atom_compare(a, b, c, d, 'm', 'e', 't', 'a'))
- return ATOM_META;
- } else if (a == 't') {
- if (mp4ff_atom_compare(a, b, c, d, 't', 'r', 'a', 'k'))
- return ATOM_TRAK;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'k', 'h', 'd'))
- return ATOM_TKHD;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'r', 'e', 'f'))
- return ATOM_TREF;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'r', 'k', 'n'))
- return ATOM_TRACK;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'm', 'p', 'o'))
- return ATOM_TEMPO;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'v', 'n', 'n'))
- return ATOM_NETWORK;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'v', 's', 'h'))
- return ATOM_SHOW;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'v', 'e', 'n'))
- return ATOM_EPISODENAME;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'v', 's', 'n'))
- return ATOM_SEASON;
- else if (mp4ff_atom_compare(a, b, c, d, 't', 'v', 'e', 's'))
- return ATOM_EPISODE;
- } else if (a == 's') {
- if (mp4ff_atom_compare(a, b, c, d, 's', 't', 'b', 'l'))
- return ATOM_STBL;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'm', 'h', 'd'))
- return ATOM_SMHD;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 't', 's', 'd'))
- return ATOM_STSD;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 't', 't', 's'))
- return ATOM_STTS;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 't', 'c', 'o'))
- return ATOM_STCO;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 't', 's', 'c'))
- return ATOM_STSC;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 't', 's', 'z'))
- return ATOM_STSZ;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 't', 'z', '2'))
- return ATOM_STZ2;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'k', 'i', 'p'))
- return ATOM_SKIP;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'i', 'n', 'f'))
- return ATOM_SINF;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'c', 'h', 'i'))
- return ATOM_SCHI;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'o', 'n', 'm'))
- return ATOM_SORTTITLE;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'o', 'a', 'l'))
- return ATOM_SORTALBUM;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'o', 'a', 'r'))
- return ATOM_SORTARTIST;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'o', 'a', 'a'))
- return ATOM_SORTALBUMARTIST;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'o', 'c', 'o'))
- return ATOM_SORTWRITER;
- else if (mp4ff_atom_compare(a, b, c, d, 's', 'o', 's', 'n'))
- return ATOM_SORTSHOW;
- } else if (a == COPYRIGHT_SYMBOL) {
- if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'n', 'a', 'm'))
- return ATOM_TITLE;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'A', 'R', 'T'))
- return ATOM_ARTIST;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'w', 'r', 't'))
- return ATOM_WRITER;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'a', 'l', 'b'))
- return ATOM_ALBUM;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'd', 'a', 'y'))
- return ATOM_DATE;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 't', 'o', 'o'))
- return ATOM_TOOL;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'c', 'm', 't'))
- return ATOM_COMMENT;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'g', 'e', 'n'))
- return ATOM_GENRE1;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'g', 'r', 'p'))
- return ATOM_CONTENTGROUP;
- else if (mp4ff_atom_compare
- (a, b, c, d, COPYRIGHT_SYMBOL, 'l', 'y', 'r'))
- return ATOM_LYRICS;
- }
+ uint8_t data[8];
+ int ret = read_data(f, data, 8);
- if (mp4ff_atom_compare(a, b, c, d, 'e', 'd', 't', 's'))
- return ATOM_EDTS;
- else if (mp4ff_atom_compare(a, b, c, d, 'e', 's', 'd', 's'))
- return ATOM_ESDS;
- else if (mp4ff_atom_compare(a, b, c, d, 'f', 't', 'y', 'p'))
- return ATOM_FTYP;
- else if (mp4ff_atom_compare(a, b, c, d, 'f', 'r', 'e', 'e'))
- return ATOM_FREE;
- else if (mp4ff_atom_compare(a, b, c, d, 'h', 'm', 'h', 'd'))
- return ATOM_HMHD;
- else if (mp4ff_atom_compare(a, b, c, d, 'v', 'm', 'h', 'd'))
- return ATOM_VMHD;
- else if (mp4ff_atom_compare(a, b, c, d, 'u', 'd', 't', 'a'))
- return ATOM_UDTA;
- else if (mp4ff_atom_compare(a, b, c, d, 'i', 'l', 's', 't'))
- return ATOM_ILST;
- else if (mp4ff_atom_compare(a, b, c, d, 'n', 'a', 'm', 'e'))
- return ATOM_NAME;
- else if (mp4ff_atom_compare(a, b, c, d, 'd', 'a', 't', 'a'))
- return ATOM_DATA;
- else if (mp4ff_atom_compare(a, b, c, d, 'd', 'i', 's', 'k'))
- return ATOM_DISC;
- else if (mp4ff_atom_compare(a, b, c, d, 'g', 'n', 'r', 'e'))
- return ATOM_GENRE2;
- else if (mp4ff_atom_compare(a, b, c, d, 'c', 'o', 'v', 'r'))
- return ATOM_COVER;
- else if (mp4ff_atom_compare(a, b, c, d, 'c', 'p', 'i', 'l'))
- return ATOM_COMPILATION;
- else if (mp4ff_atom_compare(a, b, c, d, 'c', 't', 't', 's'))
- return ATOM_CTTS;
- else if (mp4ff_atom_compare(a, b, c, d, 'd', 'r', 'm', 's'))
- return ATOM_DRMS;
- else if (mp4ff_atom_compare(a, b, c, d, 'f', 'r', 'm', 'a'))
- return ATOM_FRMA;
- else if (mp4ff_atom_compare(a, b, c, d, 'p', 'r', 'i', 'v'))
- return ATOM_PRIV;
- else if (mp4ff_atom_compare(a, b, c, d, 'i', 'v', 'i', 'v'))
- return ATOM_IVIV;
- else if (mp4ff_atom_compare(a, b, c, d, 'u', 's', 'e', 'r'))
- return ATOM_USER;
- else if (mp4ff_atom_compare(a, b, c, d, 'k', 'e', 'y', ' '))
- return ATOM_KEY;
- else if (mp4ff_atom_compare(a, b, c, d, 'a', 'A', 'R', 'T'))
- return ATOM_ALBUM_ARTIST;
- else if (mp4ff_atom_compare(a, b, c, d, 'd', 'e', 's', 'c'))
- return ATOM_DESCRIPTION;
- else if (mp4ff_atom_compare(a, b, c, d, 'p', 'c', 's', 't'))
- return ATOM_PODCAST;
- else
- return ATOM_UNKNOWN;
+ if (ret > 0 && result)
+ *result = read_u64_be(data);
+ return ret;
}
-/* read atom header, return atom size, atom size is with header included */
-static uint64_t mp4ff_atom_read_header(mp4ff_t * f, uint8_t * atom_type,
- uint8_t * header_size)
+static int read_int32(struct mp4 *f, uint32_t *result)
{
- uint64_t size;
- int32_t ret;
- int8_t atom_header[8];
+ uint8_t data[4];
+ int ret = read_data(f, data, 4);
- ret = mp4ff_read_data(f, atom_header, 8);
- if (ret != 8)
- return 0;
-
- size = mp4ff_atom_get_size(atom_header);
- *header_size = 8;
-
- /* check for 64 bit atom size */
- if (size == 1) {
- *header_size = 16;
- size = mp4ff_read_int64(f);
- }
- *atom_type = mp4ff_atom_name_to_type(atom_header[4], atom_header[5],
- atom_header[6], atom_header[7]);
- return size;
+ if (ret > 0 && result)
+ *result = read_u32_be(data);
+ return ret;
}
-static int64_t mp4ff_position(const mp4ff_t * f)
+static int read_int24(struct mp4 *f, uint32_t *result)
{
- return f->current_position;
-}
+ uint8_t data[3];
+ int ret = read_data(f, data, 3);
-static int need_parse_when_meta_only(uint8_t atom_type)
-{
- switch (atom_type) {
- case ATOM_EDTS:
- case ATOM_DRMS:
- case ATOM_SINF:
- case ATOM_SCHI:
- case ATOM_STTS:
- case ATOM_STSZ:
- case ATOM_STZ2:
- case ATOM_STCO:
- case ATOM_STSC:
- case ATOM_FRMA:
- case ATOM_IVIV:
- case ATOM_PRIV:
- return 0;
- default:
- return 1;
- }
+ if (ret > 0 && result)
+ *result = read_u24_be(data);
+ return ret;
}
-static int32_t mp4ff_set_position(mp4ff_t * f, const int64_t position)
+static int read_int16(struct mp4 *f, uint16_t *result)
{
- f->stream->seek(f->stream->user_data, position);
- f->current_position = position;
+ uint8_t data[2];
+ int ret = read_data(f, data, 2);
- return 0;
+ if (ret > 0 && result)
+ *result = read_u16_be(data);
+ return ret;
}
-static void mp4ff_track_add(mp4ff_t * f)
+static uint8_t read_int8(struct mp4 *f, uint8_t *result)
{
- f->total_tracks++;
+ uint8_t data[1];
+ int ret = read_data(f, data, 1);
- if (f->total_tracks > MAX_TRACKS) {
- f->total_tracks = 0;
- f->error++;
- return;
- }
- f->track[f->total_tracks - 1] = para_calloc(sizeof (mp4ff_track_t));
+ if (ret > 0 && result)
+ *result = data[0];
+ return ret;
}
-static uint8_t mp4ff_read_char(mp4ff_t * f)
-{
- uint8_t output;
- mp4ff_read_data(f, &output, 1);
- return output;
+#define ATOM_ITEMS \
+ ATOM_ITEM(MOOV, 'm', 'o', 'o', 'v') \
+ ATOM_ITEM(TRAK, 't', 'r', 'a', 'k') \
+ ATOM_ITEM(MDIA, 'm', 'd', 'i', 'a') \
+ ATOM_ITEM(MINF, 'm', 'i', 'n', 'f') \
+ ATOM_ITEM(STBL, 's', 't', 'b', 'l') \
+ ATOM_ITEM(UDTA, 'u', 'd', 't', 'a') \
+ ATOM_ITEM(ILST, 'i', 'l', 's', 't') /* iTunes Metadata list */ \
+ ATOM_ITEM(ARTIST, 0xa9, 'A', 'R', 'T') \
+ ATOM_ITEM(TITLE, 0xa9, 'n', 'a', 'm') \
+ ATOM_ITEM(ALBUM, 0xa9, 'a', 'l', 'b') \
+ ATOM_ITEM(DATE, 0xa9, 'd', 'a', 'y') \
+ ATOM_ITEM(COMMENT, 0xa9, 'c', 'm', 't') \
+ ATOM_ITEM(MDHD, 'm', 'd', 'h', 'd') /* track header */ \
+ ATOM_ITEM(STSD, 's', 't', 's', 'd') /* sample description box */ \
+ ATOM_ITEM(STTS, 's', 't', 't', 's') /* time to sample box */ \
+ ATOM_ITEM(STSZ, 's', 't', 's', 'z') /* sample size box */ \
+ ATOM_ITEM(STCO, 's', 't', 'c', 'o') /* chunk offset box */ \
+ ATOM_ITEM(STSC, 's', 't', 's', 'c') /* sample to chunk box */ \
+ ATOM_ITEM(MP4A, 'm', 'p', '4', 'a') \
+ ATOM_ITEM(META, 'm', 'e', 't', 'a') /* iTunes Metadata box */ \
+ ATOM_ITEM(DATA, 'd', 'a', 't', 'a') /* iTunes Metadata data box */ \
+
+#define ATOM_ITEM(_name, a, b, c, d) ATOM_ ## _name,
+enum atom {ATOM_ITEMS};
+#undef ATOM_ITEM
+
+static uint8_t atom_name_to_type(uint8_t *p)
+{
+ #define ATOM_VALUE(a, b, c, d) ((a << 24) + (b << 16) + (c << 8) + d)
+ #define ATOM_ITEM(_name, a, b, c, d) \
+ {.name = # _name, .val = ATOM_VALUE(a, b, c, d)},
+ static const struct {
+ const char *name;
+ uint32_t val;
+ } atom_table[] = {ATOM_ITEMS};
+ #undef ATOM_ITEM
+ uint32_t val = read_u32_be(p);
+
+ for (uint8_t n = 0; n < ARRAY_SIZE(atom_table); n++)
+ if (val == atom_table[n].val)
+ return n;
+ return 255;
+}
+
+/* read atom header, atom size is returned with header included. */
+static int atom_read_header(struct mp4 *f, uint8_t *atom_type,
+ uint8_t *header_size, uint64_t *atom_size)
+{
+ uint32_t size;
+ int ret;
+ uint8_t atom_header[8];
+
+ ret = read_data(f, atom_header, 8);
+ if (ret <= 0)
+ return ret;
+ size = read_u32_be(atom_header);
+ if (size == 1) { /* 64 bit atom size */
+ if (header_size)
+ *header_size = 16;
+ ret = read_int64(f, atom_size);
+ if (ret <= 0)
+ return ret;
+ } else {
+ if (header_size)
+ *header_size = 8;
+ if (atom_size)
+ *atom_size = size;
+ }
+ *atom_type = atom_name_to_type(atom_header + 4);
+ return 1;
}
-static uint32_t mp4ff_read_int24(mp4ff_t * f)
+static int64_t get_position(const struct mp4 *f)
{
- uint32_t result;
- uint32_t a, b, c;
- int8_t data[4];
-
- mp4ff_read_data(f, data, 3);
- a = (uint8_t) data[0];
- b = (uint8_t) data[1];
- c = (uint8_t) data[2];
-
- result = (a << 16) | (b << 8) | c;
- return (uint32_t) result;
+ return f->current_position;
}
-static uint32_t mp4ff_read_int32(mp4ff_t * f)
+static int32_t set_position(struct mp4 *f, int64_t position)
{
- uint32_t result;
- uint32_t a, b, c, d;
- int8_t data[4];
-
- mp4ff_read_data(f, data, 4);
- a = (uint8_t) data[0];
- b = (uint8_t) data[1];
- c = (uint8_t) data[2];
- d = (uint8_t) data[3];
-
- result = (a << 24) | (b << 16) | (c << 8) | d;
- return (uint32_t) result;
-}
+ f->cb->seek(f->cb->user_data, position);
+ f->current_position = position;
-static int32_t mp4ff_read_stsz(mp4ff_t * f)
-{
- if (f->total_tracks == 0)
- return f->error++;
-
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
- f->track[f->total_tracks - 1]->stsz_sample_size = mp4ff_read_int32(f);
- f->track[f->total_tracks - 1]->stsz_sample_count = mp4ff_read_int32(f);
-
- if (f->track[f->total_tracks - 1]->stsz_sample_size == 0) {
- int32_t i;
- f->track[f->total_tracks - 1]->stsz_table =
- para_malloc(f->track[f->total_tracks - 1]->stsz_sample_count
- * sizeof (int32_t));
- for (i = 0; i < f->track[f->total_tracks - 1]->stsz_sample_count
- && !f->stream->read_error; i++) {
- f->track[f->total_tracks - 1]->stsz_table[i] = mp4ff_read_int32(f);
- }
- }
return 0;
}
-static int32_t mp4ff_read_stts(mp4ff_t * f)
+static int read_stsz(struct mp4 *f)
{
+ int ret;
int32_t i;
- mp4ff_track_t *p_track;
+ struct mp4_track *t;
- /* CVE-2017-9223 */
if (f->total_tracks == 0)
- return f->error++;
-
- p_track = f->track[f->total_tracks - 1];
-
- if (p_track->stts_entry_count)
- return 0;
-
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
- p_track->stts_entry_count = mp4ff_read_int32(f);
-
- p_track->stts_sample_count = para_malloc(p_track->stts_entry_count
- * sizeof (int32_t));
- p_track->stts_sample_delta = para_malloc(p_track->stts_entry_count
- * sizeof (int32_t));
- for (i = 0; i < f->track[f->total_tracks - 1]->stts_entry_count && !f->stream->read_error; i++) { /* CVE-2017-9254 */
- p_track->stts_sample_count[i] = mp4ff_read_int32(f);
- p_track->stts_sample_delta[i] = mp4ff_read_int32(f);
+ return -1;
+ t = f->track[f->total_tracks - 1];
+ ret = read_int8(f, NULL); /* version */
+ if (ret <= 0)
+ return ret;
+ ret = read_int24(f, NULL); /* flags */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &t->stsz_sample_size);
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &t->stsz_sample_count);
+ if (ret <= 0)
+ return ret;
+ if (t->stsz_sample_size != 0)
+ return 1;
+ t->stsz_table = para_malloc(t->stsz_sample_count * sizeof(int32_t));
+ for (i = 0; i < t->stsz_sample_count; i++) {
+ ret = read_int32(f, &t->stsz_table[i]);
+ if (ret <= 0)
+ return ret;
}
return 1;
}
-static int32_t mp4ff_read_ctts(mp4ff_t * f)
+static int read_stts(struct mp4 *f)
{
+ int ret;
int32_t i;
- mp4ff_track_t *p_track;
+ struct mp4_track *t;
if (f->total_tracks == 0)
- return f->error++;
-
- p_track = f->track[f->total_tracks - 1];
- if (p_track->ctts_entry_count)
+ return -1;
+ t = f->track[f->total_tracks - 1];
+ if (t->stts_entry_count)
return 0;
-
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
- p_track->ctts_entry_count = mp4ff_read_int32(f);
-
- p_track->ctts_sample_count = para_malloc(p_track->ctts_entry_count
- * sizeof (int32_t));
- p_track->ctts_sample_offset = para_malloc(p_track->ctts_entry_count
- * sizeof (int32_t));
-
- for (i = 0; i < f->track[f->total_tracks - 1]->ctts_entry_count && !f->stream->read_error; i++) { /* CVE-2017-9257 */
- p_track->ctts_sample_count[i] = mp4ff_read_int32(f);
- p_track->ctts_sample_offset[i] = mp4ff_read_int32(f);
+ ret = read_int8(f, NULL); /* version */
+ if (ret <= 0)
+ return ret;
+ ret = read_int24(f, NULL); /* flags */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &t->stts_entry_count);
+ if (ret <= 0)
+ return ret;
+ t->stts_sample_count = para_malloc(t->stts_entry_count
+ * sizeof(int32_t));
+ for (i = 0; i < t->stts_entry_count; i++) {
+ ret = read_int32(f, &t->stts_sample_count[i]);
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, NULL); /* sample delta */
+ if (ret <= 0)
+ return ret;
}
return 1;
}
-static int32_t mp4ff_read_stsc(mp4ff_t * f)
+static int read_stsc(struct mp4 *f)
{
+ int ret;
int32_t i;
+ struct mp4_track *t;
if (f->total_tracks == 0)
- return f->error++;
-
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
- f->track[f->total_tracks - 1]->stsc_entry_count = mp4ff_read_int32(f);
-
- f->track[f->total_tracks - 1]->stsc_first_chunk =
- para_malloc(f->track[f->total_tracks - 1]->stsc_entry_count *
- sizeof (int32_t));
- f->track[f->total_tracks - 1]->stsc_samples_per_chunk =
- para_malloc(f->track[f->total_tracks - 1]->stsc_entry_count *
- sizeof (int32_t));
- f->track[f->total_tracks - 1]->stsc_sample_desc_index =
- para_malloc(f->track[f->total_tracks - 1]->stsc_entry_count *
- sizeof (int32_t));
-
- for (i = 0; i < f->track[f->total_tracks - 1]->stsc_entry_count && !f->stream->read_error; i++) { /* CVE-2017-9255 */
- f->track[f->total_tracks - 1]->stsc_first_chunk[i] =
- mp4ff_read_int32(f);
- f->track[f->total_tracks - 1]->stsc_samples_per_chunk[i] =
- mp4ff_read_int32(f);
- f->track[f->total_tracks - 1]->stsc_sample_desc_index[i] =
- mp4ff_read_int32(f);
+ return -1;
+ t = f->track[f->total_tracks - 1];
+
+ ret = read_int8(f, NULL); /* version */
+ if (ret <= 0)
+ return ret;
+ ret = read_int24(f, NULL); /* flags */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &t->stsc_entry_count);
+ if (ret <= 0)
+ return ret;
+ t->stsc_first_chunk = para_malloc(t->stsc_entry_count * sizeof(int32_t));
+ t->stsc_samples_per_chunk = para_malloc(t->stsc_entry_count
+ * sizeof (int32_t));
+ for (i = 0; i < t->stsc_entry_count; i++) {
+ ret = read_int32(f, &t->stsc_first_chunk[i]);
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &t->stsc_samples_per_chunk[i]);
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, NULL); /* sample desc index */
+ if (ret <= 0)
+ return ret;
}
-
- return 0;
+ return 1;
}
-static int32_t mp4ff_read_stco(mp4ff_t * f)
+static int read_stco(struct mp4 *f)
{
+ int ret;
int32_t i;
+ struct mp4_track *t;
if (f->total_tracks == 0)
- return f->error++;
-
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
- f->track[f->total_tracks - 1]->stco_entry_count = mp4ff_read_int32(f);
-
- f->track[f->total_tracks - 1]->stco_chunk_offset =
- para_malloc(f->track[f->total_tracks - 1]->stco_entry_count *
- sizeof (int32_t));
- for (i = 0; i < f->track[f->total_tracks - 1]->stco_entry_count && !f->stream->read_error; i++) { /* CVE-2017-9256 */
- f->track[f->total_tracks - 1]->stco_chunk_offset[i] =
- mp4ff_read_int32(f);
+ return -1;
+ t = f->track[f->total_tracks - 1];
+
+ ret = read_int8(f, NULL); /* version */
+ if (ret <= 0)
+ return ret;
+ ret = read_int24(f, NULL); /* flags */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &t->stco_entry_count);
+ if (ret <= 0)
+ return ret;
+ t->stco_chunk_offset = para_malloc(t->stco_entry_count
+ * sizeof(int32_t));
+ for (i = 0; i < t->stco_entry_count; i++) {
+ ret = read_int32(f, &t->stco_chunk_offset[i]);
+ if (ret <= 0)
+ return ret;
}
-
- return 0;
-}
-
-static uint16_t mp4ff_read_int16(mp4ff_t * f)
-{
- uint32_t result;
- uint32_t a, b;
- int8_t data[2];
-
- mp4ff_read_data(f, data, 2);
- a = (uint8_t) data[0];
- b = (uint8_t) data[1];
-
- result = (a << 8) | b;
- return (uint16_t) result;
-}
-
-static uint32_t mp4ff_read_mp4_descr_length(mp4ff_t * f)
-{
- uint8_t b;
- uint8_t numBytes = 0;
- uint32_t length = 0;
-
- do {
- b = mp4ff_read_char(f);
- numBytes++;
- length = (length << 7) | (b & 0x7F);
- } while ((b & 0x80) && numBytes < 4);
-
- return length;
+ return 1;
}
-static int32_t mp4ff_read_esds(mp4ff_t * f)
-{
- uint8_t tag;
- uint32_t temp;
-
- if (f->total_tracks == 0)
- return f->error++;
-
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
-
- /* get and verify ES_DescrTag */
- tag = mp4ff_read_char(f);
- if (tag == 0x03) {
- /* read length */
- if (mp4ff_read_mp4_descr_length(f) < 5 + 15) {
- return 1;
- }
- /* skip 3 bytes */
- mp4ff_read_int24(f);
- } else {
- /* skip 2 bytes */
- mp4ff_read_int16(f);
- }
-
- /* get and verify DecoderConfigDescrTab */
- if (mp4ff_read_char(f) != 0x04) {
- return 1;
- }
-
- /* read length */
- temp = mp4ff_read_mp4_descr_length(f);
- if (temp < 13)
- return 1;
- f->track[f->total_tracks - 1]->audioType = mp4ff_read_char(f);
- mp4ff_read_int32(f); //0x15000414 ????
- f->track[f->total_tracks - 1]->maxBitrate = mp4ff_read_int32(f);
- f->track[f->total_tracks - 1]->avgBitrate = mp4ff_read_int32(f);
-
- /* get and verify DecSpecificInfoTag */
- if (mp4ff_read_char(f) != 0x05) {
- return 1;
- }
-
- /* read length */
- f->track[f->total_tracks - 1]->decoderConfigLen =
- mp4ff_read_mp4_descr_length(f);
-
- free(f->track[f->total_tracks - 1]->decoderConfig);
- f->track[f->total_tracks - 1]->decoderConfig =
- para_malloc(f->track[f->total_tracks - 1]->decoderConfigLen);
- if (f->track[f->total_tracks - 1]->decoderConfig) {
- mp4ff_read_data(f, f->track[f->total_tracks - 1]->decoderConfig,
- f->track[f->total_tracks -
- 1]->decoderConfigLen);
- }
- /* will skip the remainder of the atom */
- return 0;
-}
-static int32_t mp4ff_read_mp4a(mp4ff_t * f)
+static int read_mp4a(struct mp4 *f)
{
+ int ret;
int32_t i;
- uint8_t atom_type = 0;
- uint8_t header_size = 0;
+ struct mp4_track *t;
if (f->total_tracks == 0)
- return f->error++;
+ return -1;
+ t = f->track[f->total_tracks - 1];
for (i = 0; i < 6; i++) {
- mp4ff_read_char(f); /* reserved */
- }
- /* data_reference_index */ mp4ff_read_int16(f);
-
- mp4ff_read_int32(f); /* reserved */
- mp4ff_read_int32(f); /* reserved */
-
- f->track[f->total_tracks - 1]->channelCount = mp4ff_read_int16(f);
- f->track[f->total_tracks - 1]->sampleSize = mp4ff_read_int16(f);
-
- mp4ff_read_int16(f);
- mp4ff_read_int16(f);
-
- f->track[f->total_tracks - 1]->sampleRate = mp4ff_read_int16(f);
-
- mp4ff_read_int16(f);
-
- mp4ff_atom_read_header(f, &atom_type, &header_size);
- if (atom_type == ATOM_ESDS) {
- mp4ff_read_esds(f);
+ ret = read_int8(f, NULL); /* reserved */
+ if (ret <= 0)
+ return ret;
}
+ ret = read_int16(f, NULL); /* data_reference_index */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, NULL); /* reserved */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, NULL); /* reserved */
+ if (ret <= 0)
+ return ret;
+ ret = read_int16(f, &t->channel_count);
+ if (ret <= 0)
+ return ret;
+ ret = read_int16(f, NULL);
+ if (ret <= 0)
+ return ret;
+ ret = read_int16(f, NULL);
+ if (ret <= 0)
+ return ret;
+ ret = read_int16(f, NULL);
+ if (ret <= 0)
+ return ret;
+ return read_int16(f, &t->sample_rate);
+}
+
+static int read_stsd(struct mp4 *f)
+{
+ int ret;
+ uint32_t i, entry_count;
+ struct mp4_track *t;
- return 0;
-}
-
-static int32_t mp4ff_read_stsd(mp4ff_t * f)
-{
- int32_t i;
- uint8_t header_size = 0;
-
- /* CVE-2017-9218 */
if (f->total_tracks == 0)
- return f->error++;
-
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
-
- f->track[f->total_tracks - 1]->stsd_entry_count = mp4ff_read_int32(f);
-
- for (i = 0; i < f->track[f->total_tracks - 1]->stsd_entry_count && !f->stream->read_error; i++) { /* CVE-2017-9253 */
- uint64_t skip = mp4ff_position(f);
+ return -1;
+ t = f->track[f->total_tracks - 1];
+ ret = read_int8(f, NULL); /* version */
+ if (ret <= 0)
+ return ret;
+ ret = read_int24(f, NULL); /* flags */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &entry_count);
+ if (ret <= 0)
+ return ret;
+ for (i = 0; i < entry_count; i++) {
+ uint64_t skip = get_position(f);
uint64_t size;
uint8_t atom_type = 0;
- size = mp4ff_atom_read_header(f, &atom_type, &header_size);
+ ret = atom_read_header(f, &atom_type, NULL, &size);
+ if (ret <= 0)
+ return ret;
skip += size;
-
- if (atom_type == ATOM_MP4A) {
- f->track[f->total_tracks - 1]->type = TRACK_AUDIO;
- mp4ff_read_mp4a(f);
- } else if (atom_type == ATOM_MP4V) {
- f->track[f->total_tracks - 1]->type = TRACK_VIDEO;
- } else if (atom_type == ATOM_MP4S) {
- f->track[f->total_tracks - 1]->type = TRACK_SYSTEM;
- } else {
- f->track[f->total_tracks - 1]->type = TRACK_UNKNOWN;
- }
-
- mp4ff_set_position(f, skip);
- }
-
- return 0;
-}
-
-static int32_t mp4ff_read_mvhd(mp4ff_t * f)
-{
- int32_t i;
-
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
- /* creation_time */ mp4ff_read_int32(f);
- /* modification_time */ mp4ff_read_int32(f);
- f->time_scale = mp4ff_read_int32(f);
- f->duration = mp4ff_read_int32(f);
- /* preferred_rate */ mp4ff_read_int32(f);
- /*mp4ff_read_fixed32(f); */
- /* preferred_volume */ mp4ff_read_int16(f);
- /*mp4ff_read_fixed16(f); */
- for (i = 0; i < 10; i++) {
- /* reserved */ mp4ff_read_char(f);
- }
- for (i = 0; i < 9; i++) {
- mp4ff_read_int32(f); /* matrix */
- }
- /* preview_time */ mp4ff_read_int32(f);
- /* preview_duration */ mp4ff_read_int32(f);
- /* poster_time */ mp4ff_read_int32(f);
- /* selection_time */ mp4ff_read_int32(f);
- /* selection_duration */ mp4ff_read_int32(f);
- /* current_time */ mp4ff_read_int32(f);
- /* next_track_id */ mp4ff_read_int32(f);
-
- return 0;
-}
-
-static int32_t mp4ff_tag_add_field(mp4ff_metadata_t * tags, const char *item,
- const char *value, int32_t len)
-{
- void *backup = (void *) tags->tags;
-
- if (!item || (item && !*item) || !value)
- return 0;
-
- tags->tags = (mp4ff_tag_t *) realloc(tags->tags,
- (tags->count + 1) * sizeof (mp4ff_tag_t));
- if (!tags->tags) {
- free(backup);
- return 0;
- } else {
- tags->tags[tags->count].item = para_strdup(item);
- tags->tags[tags->count].len = len;
- if (len >= 0) {
- tags->tags[tags->count].value = para_malloc(len + 1);
- memcpy(tags->tags[tags->count].value, value, len);
- tags->tags[tags->count].value[len] = 0;
- } else {
- tags->tags[tags->count].value = para_strdup(value);
+ if (!f->audio_track && atom_type == ATOM_MP4A) {
+ f->audio_track = t;
+ read_mp4a(f);
}
- tags->count++;
- return 1;
- }
-}
-
-static const char *ID3v1GenreList[] = {
- "Blues", "Classic Rock", "Country", "Dance", "Disco", "Funk",
- "Grunge", "Hip-Hop", "Jazz", "Metal", "New Age", "Oldies",
- "Other", "Pop", "R&B", "Rap", "Reggae", "Rock",
- "Techno", "Industrial", "Alternative", "Ska", "Death Metal", "Pranks",
- "Soundtrack", "Euro-Techno", "Ambient", "Trip-Hop", "Vocal",
- "Jazz+Funk", "Fusion", "Trance", "Classical", "Instrumental", "Acid",
- "House", "Game", "Sound Clip", "Gospel", "Noise", "AlternRock", "Bass",
- "Soul", "Punk", "Space", "Meditative", "Instrumental Pop",
- "Instrumental Rock", "Ethnic", "Gothic", "Darkwave",
- "Techno-Industrial", "Electronic", "Pop-Folk", "Eurodance", "Dream",
- "Southern Rock", "Comedy", "Cult", "Gangsta", "Top 40",
- "Christian Rap", "Pop/Funk", "Jungle", "Native American", "Cabaret",
- "New Wave", "Psychadelic", "Rave", "Showtunes", "Trailer", "Lo-Fi",
- "Tribal", "Acid Punk", "Acid Jazz", "Polka", "Retro", "Musical",
- "Rock & Roll", "Hard Rock", "Folk", "Folk/Rock", "National Folk",
- "Swing", "Fast-Fusion", "Bebob", "Latin", "Revival", "Celtic",
- "Bluegrass", "Avantgarde", "Gothic Rock", "Progressive Rock",
- "Psychedelic Rock", "Symphonic Rock", "Slow Rock", "Big Band",
- "Chorus", "Easy Listening", "Acoustic", "Humour", "Speech", "Chanson",
- "Opera", "Chamber Music", "Sonata", "Symphony", "Booty Bass", "Primus",
- "Porn Groove", "Satire", "Slow Jam", "Club", "Tango", "Samba",
- "Folklore", "Ballad", "Power Ballad", "Rhythmic Soul", "Freestyle",
- "Duet", "Punk Rock", "Drum Solo", "A capella", "Euro-House",
- "Dance Hall", "Goa", "Drum & Bass", "Club House", "Hardcore", "Terror",
- "Indie", "BritPop", "NegerPunk", "Polsk Punk", "Beat",
- "Christian Gangsta", "Heavy Metal", "Black Metal", "Crossover",
- "Contemporary C", "Christian Rock", "Merengue", "Salsa", "Thrash Metal",
- "Anime", "JPop", "SynthPop",
-};
-
-static const char *mp4ff_meta_index_to_genre(uint32_t idx)
-{
- if (idx > 0 && idx <= sizeof (ID3v1GenreList) / sizeof (ID3v1GenreList[0])) {
- return ID3v1GenreList[idx - 1];
- } else {
- return 0;
+ set_position(f, skip);
}
+ return 1;
}
-static char *mp4ff_read_string(mp4ff_t * f, uint32_t length)
-{
- char *str = para_malloc(length + 1);
- if ((uint32_t)mp4ff_read_data(f, str, length) != length) {
- free(str);
- str = NULL;
- } else
- str[length] = 0;
- return str;
-}
-
-static int32_t mp4ff_set_metadata_name(const uint8_t atom_type, char **name)
+static const char *get_metadata_name(uint8_t atom_type)
{
- static char *tag_names[] = {
- "unknown", "title", "artist", "writer", "album",
- "date", "tool", "comment", "genre", "track",
- "disc", "compilation", "genre", "tempo", "cover",
- "album_artist", "contentgroup", "lyrics", "description",
- "network", "show", "episodename",
- "sorttitle", "sortalbum", "sortartist", "sortalbumartist",
- "sortwriter", "sortshow",
- "season", "episode", "podcast"
- };
- uint8_t tag_idx = 0;
-
switch (atom_type) {
- case ATOM_TITLE:
- tag_idx = 1;
- break;
- case ATOM_ARTIST:
- tag_idx = 2;
- break;
- case ATOM_WRITER:
- tag_idx = 3;
- break;
- case ATOM_ALBUM:
- tag_idx = 4;
- break;
- case ATOM_DATE:
- tag_idx = 5;
- break;
- case ATOM_TOOL:
- tag_idx = 6;
- break;
- case ATOM_COMMENT:
- tag_idx = 7;
- break;
- case ATOM_GENRE1:
- tag_idx = 8;
- break;
- case ATOM_TRACK:
- tag_idx = 9;
- break;
- case ATOM_DISC:
- tag_idx = 10;
- break;
- case ATOM_COMPILATION:
- tag_idx = 11;
- break;
- case ATOM_GENRE2:
- tag_idx = 12;
- break;
- case ATOM_TEMPO:
- tag_idx = 13;
- break;
- case ATOM_COVER:
- tag_idx = 14;
- break;
- case ATOM_ALBUM_ARTIST:
- tag_idx = 15;
- break;
- case ATOM_CONTENTGROUP:
- tag_idx = 16;
- break;
- case ATOM_LYRICS:
- tag_idx = 17;
- break;
- case ATOM_DESCRIPTION:
- tag_idx = 18;
- break;
- case ATOM_NETWORK:
- tag_idx = 19;
- break;
- case ATOM_SHOW:
- tag_idx = 20;
- break;
- case ATOM_EPISODENAME:
- tag_idx = 21;
- break;
- case ATOM_SORTTITLE:
- tag_idx = 22;
- break;
- case ATOM_SORTALBUM:
- tag_idx = 23;
- break;
- case ATOM_SORTARTIST:
- tag_idx = 24;
- break;
- case ATOM_SORTALBUMARTIST:
- tag_idx = 25;
- break;
- case ATOM_SORTWRITER:
- tag_idx = 26;
- break;
- case ATOM_SORTSHOW:
- tag_idx = 27;
- break;
- case ATOM_SEASON:
- tag_idx = 28;
- break;
- case ATOM_EPISODE:
- tag_idx = 29;
- break;
- case ATOM_PODCAST:
- tag_idx = 30;
- break;
- default:
- tag_idx = 0;
- break;
- }
-
- *name = para_strdup(tag_names[tag_idx]);
- return 0;
-}
-
-static uint32_t min_body_size(const uint8_t atom_type)
-{
- switch(atom_type) {
- case ATOM_GENRE2:
- case ATOM_TEMPO:
- return 10;
- case ATOM_TRACK:
- return sizeof (char) /* version */
- + sizeof(uint8_t) * 3 /* flags */
- + sizeof(uint32_t) /* reserved */
- + sizeof(uint16_t) /* leading uint16_t */
- + sizeof(uint16_t) /* track */
- + sizeof(uint16_t); /* totaltracks */
- case ATOM_DISC:
- return sizeof (char) /* version */
- + sizeof(uint8_t) * 3 /* flags */
- + sizeof(uint32_t) /* reserved */
- + sizeof(uint16_t) /* disc */
- + sizeof(uint16_t); /* totaldiscs */
- default: assert(false);
+ case ATOM_TITLE: return "title";
+ case ATOM_ARTIST: return "artist";
+ case ATOM_ALBUM: return "album";
+ case ATOM_DATE: return "date";
+ case ATOM_COMMENT: return "comment";
+ default: return "unknown";
}
}
-static int32_t mp4ff_parse_tag(mp4ff_t * f, const uint8_t parent,
- const int32_t size)
+static int parse_tag(struct mp4 *f, uint8_t parent, int32_t size)
{
- uint8_t atom_type;
- uint8_t header_size = 0;
+ int ret;
uint64_t subsize, sumsize;
- char *name = NULL;
- char *data = NULL;
- uint32_t done = 0;
+ char *value = NULL;
uint32_t len = 0;
uint64_t destpos;
+ struct mp4_tag *tag;
for (
sumsize = 0;
- sumsize < size && !f->stream->read_error; /* CVE-2017-9222 */
- mp4ff_set_position(f, destpos), sumsize += subsize
+ sumsize < size;
+ set_position(f, destpos), sumsize += subsize
) {
- subsize = mp4ff_atom_read_header(f, &atom_type, &header_size);
- destpos = mp4ff_position(f) + subsize - header_size;
- if (done)
- continue;
- if (atom_type == ATOM_NAME) {
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
- free(name);
- name = mp4ff_read_string(f, subsize - (header_size + 4));
- continue;
- }
+ uint8_t atom_type;
+ uint8_t header_size = 0;
+ ret = atom_read_header(f, &atom_type, &header_size, &subsize);
+ if (ret <= 0)
+ goto fail;
+ destpos = get_position(f) + subsize - header_size;
if (atom_type != ATOM_DATA)
continue;
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
- mp4ff_read_int32(f); /* reserved */
-
- /* some need special attention */
- if (parent == ATOM_GENRE2 || parent == ATOM_TEMPO) {
- uint16_t val;
- if (subsize - header_size < min_body_size(parent))
- continue;
- val = mp4ff_read_int16(f);
- if (parent == ATOM_TEMPO) {
- char temp[16];
- sprintf(temp, "%.5u BPM", val);
- mp4ff_tag_add_field(&(f-> tags), "tempo",
- temp, -1);
- } else {
- const char *tmp = mp4ff_meta_index_to_genre(val);
- if (tmp)
- mp4ff_tag_add_field (&(f->tags),
- "genre", tmp, -1);
- }
- done = 1;
- } else if (parent == ATOM_TRACK || parent == ATOM_DISC) {
- uint16_t index, total;
- char temp[32];
- if (subsize - header_size < min_body_size(parent))
- continue;
- mp4ff_read_int16(f);
- index = mp4ff_read_int16(f);
- total = mp4ff_read_int16(f);
- if (parent == ATOM_TRACK)
- mp4ff_read_int16(f);
- sprintf(temp, "%d", index);
- mp4ff_tag_add_field(&(f->tags), parent == ATOM_TRACK?
- "track" : "disc", temp, -1);
- if (total > 0) {
- sprintf(temp, "%d", total);
- mp4ff_tag_add_field(& (f-> tags),
- parent == ATOM_TRACK?
- "totaltracks" : "totaldiscs", temp, -1);
- }
- done = 1;
- } else {
- free(data);
- data = mp4ff_read_string(f, subsize - (header_size + 8));
- len = subsize - (header_size + 8);
- }
- }
- if (data) {
- if (!done) {
- if (name == NULL)
- mp4ff_set_metadata_name(parent , &name);
- if (name)
- mp4ff_tag_add_field(&(f->tags), name, data, len);
- }
-
- free(data);
+ ret = read_int8(f, NULL); /* version */
+ if (ret <= 0)
+ goto fail;
+ ret = read_int24(f, NULL); /* flags */
+ if (ret <= 0)
+ goto fail;
+ ret = read_int32(f, NULL); /* reserved */
+ if (ret <= 0)
+ goto fail;
+ ret = -ERRNO_TO_PARA_ERROR(EINVAL);
+ if (subsize < header_size + 8 || subsize > UINT_MAX)
+ goto fail;
+ len = subsize - (header_size + 8);
+ free(value);
+ value = para_malloc(len + 1);
+ ret = read_data(f, value, len);
+ if (ret <= 0)
+ goto fail;
+ value[len] = '\0';
}
- free(name);
+ if (!value)
+ return -ERRNO_TO_PARA_ERROR(EINVAL);
+ f->meta.tags = para_realloc(f->meta.tags, (f->meta.count + 1)
+ * sizeof(struct mp4_tag));
+ tag = f->meta.tags + f->meta.count;
+ tag->item = para_strdup(get_metadata_name(parent));
+ tag->value = value;
+ tag->len = len;
+ f->meta.count++;
return 1;
+fail:
+ free(value);
+ return ret;
}
-static int32_t mp4ff_read_mdhd(mp4ff_t * f)
+static int read_mdhd(struct mp4 *f)
{
+ int ret;
uint32_t version;
+ struct mp4_track *t;
- /* CVE-2017-9221 */
if (f->total_tracks == 0)
- return f->error++;
+ return -1;
+ t = f->track[f->total_tracks - 1];
- version = mp4ff_read_int32(f);
+ ret = read_int32(f, &version);
+ if (ret <= 0)
+ return ret;
if (version == 1) {
- mp4ff_read_int64(f); //creation-time
- mp4ff_read_int64(f); //modification-time
- f->track[f->total_tracks - 1]->timeScale = mp4ff_read_int32(f); //timescale
- f->track[f->total_tracks - 1]->duration = mp4ff_read_int64(f); //duration
- } else //version == 0
- {
+ ret = read_int64(f, NULL); /* creation-time */
+ if (ret <= 0)
+ return ret;
+ ret = read_int64(f, NULL); /* modification-time */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &t->time_scale);
+ if (ret <= 0)
+ return ret;
+ ret = read_int64(f, &t->duration);
+ if (ret <= 0)
+ return ret;
+ } else { //version == 0
uint32_t temp;
- mp4ff_read_int32(f); //creation-time
- mp4ff_read_int32(f); //modification-time
- f->track[f->total_tracks - 1]->timeScale = mp4ff_read_int32(f); //timescale
- temp = mp4ff_read_int32(f);
- f->track[f->total_tracks - 1]->duration = (temp == (uint32_t) (-1)) ? (uint64_t) (-1) : (uint64_t) (temp);
+ ret = read_int32(f, NULL); /* creation-time */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, NULL); /* modification-time */
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &t->time_scale);
+ if (ret <= 0)
+ return ret;
+ ret = read_int32(f, &temp);
+ if (ret <= 0)
+ return ret;
+ t->duration = (temp == (uint32_t) (-1))?
+ (uint64_t) (-1) : (uint64_t) (temp);
}
- mp4ff_read_int16(f);
- mp4ff_read_int16(f);
+ ret = read_int16(f, NULL);
+ if (ret <= 0)
+ return ret;
+ ret = read_int16(f, NULL);
+ if (ret <= 0)
+ return ret;
return 1;
}
-static int32_t mp4ff_parse_metadata(mp4ff_t * f, const int32_t size)
+static int32_t read_ilst(struct mp4 *f, int32_t size)
{
- uint64_t subsize, sumsize = 0;
- uint8_t atom_type;
- uint8_t header_size = 0;
+ int ret;
+ uint64_t sumsize = 0;
while (sumsize < size) {
- subsize = mp4ff_atom_read_header(f, &atom_type, &header_size);
- if (subsize == 0)
- break;
- mp4ff_parse_tag(f, atom_type,
- (uint32_t) (subsize - header_size));
+ uint8_t atom_type;
+ uint64_t subsize, destpos;
+ uint8_t header_size = 0;
+ ret = atom_read_header(f, &atom_type, &header_size, &subsize);
+ if (ret <= 0)
+ return ret;
+ destpos = get_position(f) + subsize - header_size;
+ switch (atom_type) {
+ case ATOM_ARTIST:
+ case ATOM_TITLE:
+ case ATOM_ALBUM:
+ case ATOM_COMMENT:
+ case ATOM_DATE:
+ ret = parse_tag(f, atom_type, subsize - header_size);
+ if (ret <= 0)
+ return ret;
+ }
+ set_position(f, destpos);
sumsize += subsize;
}
-
- return 0;
+ return 1;
}
-static int32_t mp4ff_read_meta(mp4ff_t * f, const uint64_t size)
+static int32_t read_meta(struct mp4 *f, uint64_t size)
{
+ int ret;
uint64_t subsize, sumsize = 0;
uint8_t atom_type;
uint8_t header_size = 0;
- mp4ff_read_char(f); /* version */
- mp4ff_read_int24(f); /* flags */
-
+ ret = read_int8(f, NULL); /* version */
+ if (ret <= 0)
+ return ret;
+ ret = read_int24(f, NULL); /* flags */
+ if (ret <= 0)
+ return ret;
while (sumsize < (size - (header_size + 4))) {
- subsize = mp4ff_atom_read_header(f, &atom_type, &header_size);
+ ret = atom_read_header(f, &atom_type, &header_size, &subsize);
+ if (ret <= 0)
+ return ret;
if (subsize <= header_size + 4)
return 1;
if (atom_type == ATOM_ILST) {
- mp4ff_parse_metadata(f, (uint32_t) (subsize - (header_size + 4)));
- } else {
- mp4ff_set_position(f, mp4ff_position(f) + subsize - header_size);
- }
+ f->ilst_offset = get_position(f) - header_size;
+ f->ilst_size = subsize;
+ ret = read_ilst(f, subsize - (header_size + 4));
+ if (ret <= 0)
+ return ret;
+ } else
+ set_position(f, get_position(f) + subsize - header_size);
sumsize += subsize;
}
-
- return 0;
+ return 1;
}
-static int32_t mp4ff_atom_read(mp4ff_t * f, const int32_t size,
- const uint8_t atom_type)
+static bool need_atom(uint8_t atom_type, bool meta_only)
{
- uint64_t dest_position = mp4ff_position(f) + size - 8;
- if (atom_type == ATOM_STSZ) {
- /* sample size box */
- mp4ff_read_stsz(f);
- } else if (atom_type == ATOM_STTS) {
- /* time to sample box */
- mp4ff_read_stts(f);
- } else if (atom_type == ATOM_CTTS) {
- /* composition offset box */
- mp4ff_read_ctts(f);
- } else if (atom_type == ATOM_STSC) {
- /* sample to chunk box */
- mp4ff_read_stsc(f);
- } else if (atom_type == ATOM_STCO) {
- /* chunk offset box */
- mp4ff_read_stco(f);
- } else if (atom_type == ATOM_STSD) {
- /* sample description box */
- mp4ff_read_stsd(f);
- } else if (atom_type == ATOM_MVHD) {
- /* movie header box */
- mp4ff_read_mvhd(f);
- } else if (atom_type == ATOM_MDHD) {
- /* track header */
- mp4ff_read_mdhd(f);
- } else if (atom_type == ATOM_META) {
- /* iTunes Metadata box */
- mp4ff_read_meta(f, size);
+ /* these are needed in any case */
+ switch (atom_type) {
+ case ATOM_STSD:
+ case ATOM_META:
+ case ATOM_TRAK:
+ case ATOM_MDIA:
+ case ATOM_MINF:
+ case ATOM_STBL:
+ case ATOM_UDTA:
+ return true;
}
-
- mp4ff_set_position(f, dest_position);
- return 0;
+ /* meta-only opens don't need anything else */
+ if (meta_only)
+ return false;
+ /* these are only required for regular opens */
+ switch (atom_type) {
+ case ATOM_STTS:
+ case ATOM_STSZ:
+ case ATOM_STCO:
+ case ATOM_STSC:
+ case ATOM_MDHD:
+ return true;
+ }
+ return false;
}
/* parse atoms that are sub atoms of other atoms */
-static int32_t parse_sub_atoms(mp4ff_t * f, const uint64_t total_size, int meta_only)
+static int parse_sub_atoms(struct mp4 *f, uint64_t total_size, bool meta_only)
{
- uint64_t size;
- uint8_t atom_type = 0;
- uint64_t counted_size = 0;
- uint8_t header_size = 0;
-
- while (counted_size < total_size) {
- size = mp4ff_atom_read_header(f, &atom_type, &header_size);
- counted_size += size;
+ int ret;
+ uint64_t dest, size, end = get_position(f) + total_size;
- /* check for end of file */
+ for (dest = get_position(f); dest < end; set_position(f, dest)) {
+ uint8_t header_size, atom_type;
+ ret = atom_read_header(f, &atom_type, &header_size, &size);
+ if (ret <= 0)
+ return ret;
if (size == 0)
- break;
-
- /* we're starting to read a new track, update index,
- * so that all data and tables get written in the right place
- */
+ return -1;
+ dest = get_position(f) + size - header_size;
if (atom_type == ATOM_TRAK) {
- mp4ff_track_add(f);
+ if (f->total_tracks >= MAX_TRACKS)
+ return -1;
+ f->total_tracks++;
+ f->track[f->total_tracks - 1] = para_calloc(
+ sizeof(struct mp4_track));
+ } else if (atom_type == ATOM_UDTA) {
+ f->udta_offset = get_position(f) - header_size;
+ f->udta_size = size;
}
-
- /* parse subatoms */
- if (meta_only && !need_parse_when_meta_only(atom_type)) {
- mp4ff_set_position(f, mp4ff_position(f) + size - header_size);
- } else if (atom_type < SUBATOMIC) {
- parse_sub_atoms(f, size - header_size, meta_only);
- } else {
- mp4ff_atom_read(f, (uint32_t) size, atom_type);
+ if (!need_atom(atom_type, meta_only))
+ continue;
+ switch (atom_type) {
+ case ATOM_STSZ: ret = read_stsz(f); break;
+ case ATOM_STTS: ret = read_stts(f); break;
+ case ATOM_STSC: ret = read_stsc(f); break;
+ case ATOM_STCO: ret = read_stco(f); break;
+ case ATOM_STSD: ret = read_stsd(f); break;
+ case ATOM_MDHD: ret = read_mdhd(f); break;
+ case ATOM_META:
+ f->meta_offset = get_position(f) - header_size;
+ f->meta_size = size;
+ ret = read_meta(f, size);
+ break;
+ default:
+ ret = parse_sub_atoms(f, size - header_size, meta_only);
}
+ if (ret <= 0)
+ return ret;
}
-
- return 0;
+ return 1;
}
-/* parse root atoms */
-static int32_t parse_atoms(mp4ff_t * f, int meta_only)
+static int open_file(const struct mp4_callback *cb, bool meta_only, struct mp4 **result)
{
+ int ret;
uint64_t size;
- uint8_t atom_type = 0;
- uint8_t header_size = 0;
-
- f->file_size = 0;
- f->stream->read_error = 0;
+ uint8_t atom_type, header_size;
+ struct mp4 *f = para_calloc(sizeof(*f));
- while ((size =
- mp4ff_atom_read_header(f, &atom_type, &header_size)) != 0) {
+ f->cb = cb;
+ while ((ret = atom_read_header(f, &atom_type, &header_size, &size)) > 0) {
f->file_size += size;
f->last_atom = atom_type;
-
- if (atom_type == ATOM_MOOV && size > header_size) {
- f->moov_offset = mp4ff_position(f) - header_size;
- f->moov_size = size;
- }
-
- /* parse subatoms */
- if (meta_only && !need_parse_when_meta_only(atom_type)) {
- mp4ff_set_position(f, mp4ff_position(f) + size - header_size);
- } else if (atom_type < SUBATOMIC) {
- parse_sub_atoms(f, size - header_size, meta_only);
- } else {
- /* skip this atom */
- mp4ff_set_position(f, mp4ff_position(f) + size - header_size);
+ if (atom_type != ATOM_MOOV || size <= header_size) { /* skip */
+ set_position(f, get_position(f) + size - header_size);
+ continue;
}
+ f->moov_offset = get_position(f) - header_size;
+ f->moov_size = size;
+ ret = parse_sub_atoms(f, size - header_size, meta_only);
+ if (ret <= 0)
+ break;
+ }
+ if (ret < 0) {
+ ret = -E_MP4_OPEN;
+ goto fail;
}
+ ret = -E_MP4_TRACK;
+ if (!f->audio_track)
+ goto fail;
+ *result = f;
+ return 1;
+fail:
+ *result = NULL;
+ free(f);
+ return ret;
+}
- return 0;
+int mp4_open_read(const struct mp4_callback *cb, struct mp4 **result)
+{
+ return open_file(cb, false, result);
}
-void mp4ff_get_decoder_config(const mp4ff_t * f, const int track,
- unsigned char **ppBuf, unsigned int *pBufSize)
+void mp4_close(struct mp4 *f)
{
- if (track >= f->total_tracks) {
- *ppBuf = NULL;
- *pBufSize = 0;
- return;
- }
+ int32_t i;
- if (f->track[track]->decoderConfig == NULL
- || f->track[track]->decoderConfigLen == 0) {
- *ppBuf = NULL;
- *pBufSize = 0;
- } else {
- *ppBuf = para_malloc(f->track[track]->decoderConfigLen);
- memcpy(*ppBuf, f->track[track]->decoderConfig,
- f->track[track]->decoderConfigLen);
- *pBufSize = f->track[track]->decoderConfigLen;
+ for (i = 0; i < f->total_tracks; i++) {
+ if (f->track[i]) {
+ free(f->track[i]->stsz_table);
+ free(f->track[i]->stts_sample_count);
+ free(f->track[i]->stsc_first_chunk);
+ free(f->track[i]->stsc_samples_per_chunk);
+ free(f->track[i]->stco_chunk_offset);
+ free(f->track[i]);
+ }
+ }
+ for (i = 0; i < f->meta.count; i++) {
+ free(f->meta.tags[i].item);
+ free(f->meta.tags[i].value);
}
+ free(f->meta.tags);
+ free(f);
}
-mp4ff_t *mp4ff_open_read(mp4ff_callback_t * f)
+static int32_t chunk_of_sample(const struct mp4 *f, int32_t sample,
+ int32_t *chunk)
{
- mp4ff_t *ff = para_calloc(sizeof(mp4ff_t));
-
- ff->stream = f;
-
- parse_atoms(ff, 0);
+ const struct mp4_track *t = f->audio_track;
+ uint32_t *fc = t->stsc_first_chunk, *spc = t->stsc_samples_per_chunk;
+ int32_t chunk1, chunk1samples, n, total, i;
- if (ff->error) {
- free(ff);
- ff = NULL;
+ for (i = 1, total = 0; i < t->stsc_entry_count; i++, total += n) {
+ n = (fc[i] - fc[i - 1]) * spc[i - 1]; /* number of samples */
+ if (sample < total + n)
+ break;
}
-
- return ff;
+ chunk1 = fc[i - 1];
+ chunk1samples = spc[i - 1];
+ if (chunk1samples != 0)
+ *chunk = (sample - total) / chunk1samples + chunk1;
+ else
+ *chunk = 1;
+ return total + (*chunk - chunk1) * chunk1samples;
}
-static int32_t mp4ff_tag_delete(mp4ff_metadata_t * tags)
+/**
+ * Return the number of milliseconds of the audio track.
+ *
+ * \param f As returned by \ref mp4_open_read(), must not be NULL.
+ */
+uint64_t mp4_get_duration(const struct mp4 *f)
{
- uint32_t i;
+ const struct mp4_track *t = f->audio_track;
- for (i = 0; i < tags->count; i++) {
- free(tags->tags[i].item);
- free(tags->tags[i].value);
- }
- free(tags->tags);
- tags->tags = NULL;
- tags->count = 0;
-
- return 0;
-}
-
-void mp4ff_close(mp4ff_t * ff)
-{
- int32_t i;
-
- for (i = 0; i < ff->total_tracks; i++) {
- if (ff->track[i]) {
- free(ff->track[i]->stsz_table);
- free(ff->track[i]->stts_sample_count);
- free(ff->track[i]->stts_sample_delta);
- free(ff->track[i]->stsc_first_chunk);
- free(ff->track[i]->stsc_samples_per_chunk);
- free(ff->track[i]->stsc_sample_desc_index);
- free(ff->track[i]->stco_chunk_offset);
- free(ff->track[i]->decoderConfig);
- free(ff->track[i]->ctts_sample_count);
- free(ff->track[i]->ctts_sample_offset);
- free(ff->track[i]);
- }
- }
-
- mp4ff_tag_delete(&(ff->tags));
- free(ff);
+ if (t->time_scale == 0)
+ return 0;
+ return t->duration * 1000 / t->time_scale;
}
-static int32_t mp4ff_chunk_of_sample(const mp4ff_t * f, const int32_t track,
- const int32_t sample, int32_t * chunk_sample, int32_t * chunk)
+int mp4_set_sample_position(struct mp4 *f, int32_t sample)
{
- int32_t total_entries = 0;
- int32_t chunk2entry;
- int32_t chunk1, chunk2, chunk1samples, range_samples, total = 0;
+ const struct mp4_track *t = f->audio_track;
+ int32_t offset, chunk, chunk_sample;
+ uint32_t n, srs; /* sample range size */
- *chunk_sample = 0;
- *chunk = 1;
- if (f->track[track] == NULL) {
- return -1;
+ if (sample >= t->stsz_sample_count)
+ return -ERRNO_TO_PARA_ERROR(EINVAL);
+ chunk_sample = chunk_of_sample(f, sample, &chunk);
+ if (t->stsz_sample_size > 0)
+ srs = (sample - chunk_sample) * t->stsz_sample_size;
+ else {
+ for (srs = 0, n = chunk_sample; n < sample; n++)
+ srs += t->stsz_table[n];
}
-
- total_entries = f->track[track]->stsc_entry_count;
-
- chunk1 = 1;
- chunk1samples = 0;
- chunk2entry = 0;
-
- do {
- chunk2 = f->track[track]->stsc_first_chunk[chunk2entry];
- *chunk = chunk2 - chunk1;
- range_samples = *chunk * chunk1samples;
-
- if (sample < total + range_samples)
- break;
-
- chunk1samples = f->track[track]->stsc_samples_per_chunk[chunk2entry];
- chunk1 = chunk2;
-
- if (chunk2entry < total_entries) {
- chunk2entry++;
- total += range_samples;
- }
- } while (chunk2entry < total_entries);
-
- if (chunk1samples)
- *chunk = (sample - total) / chunk1samples + chunk1;
+ if (t->stco_entry_count > 0 && chunk > t->stco_entry_count)
+ offset = t->stco_chunk_offset[t->stco_entry_count - 1];
+ else if (t->stco_entry_count > 0)
+ offset = t->stco_chunk_offset[chunk - 1];
else
- *chunk = 1;
-
- *chunk_sample = total + (*chunk - chunk1) * chunk1samples;
-
- return 0;
-}
-
-static int32_t mp4ff_chunk_to_offset(const mp4ff_t * f, const int32_t track,
- const int32_t chunk)
-{
- const mp4ff_track_t *p_track = f->track[track];
-
- if (p_track->stco_entry_count && (chunk > p_track->stco_entry_count)) {
- return p_track->stco_chunk_offset[p_track->stco_entry_count -
- 1];
- } else if (p_track->stco_entry_count) {
- return p_track->stco_chunk_offset[chunk - 1];
- } else {
- return 8;
- }
-
- return 0;
-}
-
-static int32_t mp4ff_sample_range_size(const mp4ff_t * f, const int32_t track,
- const int32_t chunk_sample, const int32_t sample)
-{
- int32_t i, total;
- const mp4ff_track_t *p_track = f->track[track];
-
- if (p_track->stsz_sample_size) {
- return (sample - chunk_sample) * p_track->stsz_sample_size;
- } else {
- if (sample >= p_track->stsz_sample_count)
- return 0; //error
-
- for (i = chunk_sample, total = 0; i < sample; i++) {
- total += p_track->stsz_table[i];
- }
- }
-
- return total;
-}
-static int32_t mp4ff_sample_to_offset(const mp4ff_t * f, const int32_t track,
- const int32_t sample)
-{
- int32_t chunk, chunk_sample, chunk_offset1, chunk_offset2;
-
- mp4ff_chunk_of_sample(f, track, sample, &chunk_sample, &chunk);
-
- chunk_offset1 = mp4ff_chunk_to_offset(f, track, chunk);
- chunk_offset2 = chunk_offset1 + mp4ff_sample_range_size(f,
- track, chunk_sample, sample);
- return chunk_offset2;
-}
-
-void mp4ff_set_sample_position(mp4ff_t *f, const int32_t track,
- const int32_t sample)
-{
- int32_t offset = mp4ff_sample_to_offset(f, track, sample);
- mp4ff_set_position(f, offset);
+ offset = 8;
+ set_position(f, offset + srs);
+ return 1;
}
-int32_t mp4ff_get_sample_size(const mp4ff_t *f, int track, int sample)
+int32_t mp4_get_sample_size(const struct mp4 *f, int sample)
{
- const mp4ff_track_t *t = f->track[track];
+ const struct mp4_track *t = f->audio_track;
if (t->stsz_sample_size != 0)
return t->stsz_sample_size;
return t->stsz_table[sample];
}
-uint32_t mp4ff_get_sample_rate(const mp4ff_t * f, const int32_t track)
+uint32_t mp4_get_sample_rate(const struct mp4 *f)
{
- return f->track[track]->sampleRate;
+ return f->audio_track->sample_rate;
}
-uint32_t mp4ff_get_channel_count(const mp4ff_t * f, const int32_t track)
+uint32_t mp4_get_channel_count(const struct mp4 *f)
{
- return f->track[track]->channelCount;
+ return f->audio_track->channel_count;
}
-int32_t mp4ff_num_samples(const mp4ff_t * f, const int32_t track)
+int32_t mp4_num_samples(const struct mp4 *f)
{
+ const struct mp4_track *t = f->audio_track;
int32_t i;
int32_t total = 0;
- for (i = 0; i < f->track[track]->stts_entry_count; i++) {
- total += f->track[track]->stts_sample_count[i];
- }
+ for (i = 0; i < t->stts_entry_count; i++)
+ total += t->stts_sample_count[i];
return total;
}
-mp4ff_t *mp4ff_open_read_metaonly(mp4ff_callback_t * f)
-{
- mp4ff_t *ff = para_calloc(sizeof(mp4ff_t));
-
- ff->stream = f;
-
- parse_atoms(ff, 1);
-
- if (ff->error) {
- free(ff);
- ff = NULL;
- }
-
- return ff;
-}
-
-int32_t mp4ff_meta_get_num_items(const mp4ff_t * f)
-{
- return f->tags.count;
-}
-
-int32_t mp4ff_meta_get_by_index(const mp4ff_t * f, uint32_t index,
- char **item, char **value)
-{
- if (index >= f->tags.count) {
- *item = NULL;
- *value = NULL;
- return 0;
- } else {
- *item = para_strdup(f->tags.tags[index].item);
- *value = para_strdup(f->tags.tags[index].value);
- return 1;
- }
-}
-
-static uint32_t find_atom(mp4ff_t * f, uint64_t base, uint32_t size,
- const char *name)
-{
- uint32_t remaining = size;
- uint64_t atom_offset = base;
- for (;;) {
- char atom_name[4];
- uint32_t atom_size;
-
- mp4ff_set_position(f, atom_offset);
-
- if (remaining < 8)
- break;
- atom_size = mp4ff_read_int32(f);
- if (atom_size > remaining || atom_size < 8)
- break;
- mp4ff_read_data(f, atom_name, 4);
-
- if (!memcmp(atom_name, name, 4)) {
- mp4ff_set_position(f, atom_offset);
- return 1;
- }
-
- remaining -= atom_size;
- atom_offset += atom_size;
- }
- return 0;
-}
-static uint32_t find_atom_v2(mp4ff_t * f, uint64_t base, uint32_t size,
- const char *name, uint32_t extraheaders, const char *name_inside)
-{
- uint64_t first_base = (uint64_t) (-1);
- while (find_atom(f, base, size, name)) //try to find atom <name> with atom <name_inside> in it
- {
- uint64_t mybase = mp4ff_position(f);
- uint32_t mysize = mp4ff_read_int32(f);
-
- if (first_base == (uint64_t) (-1))
- first_base = mybase;
-
- if (mysize < 8 + extraheaders)
- break;
-
- if (find_atom (f, mybase + (8 + extraheaders),
- mysize - (8 + extraheaders), name_inside)) {
- mp4ff_set_position(f, mybase);
- return 2;
- }
- base += mysize;
- if (size <= mysize) {
- size = 0;
- break;
- }
- size -= mysize;
- }
-
- if (first_base != (uint64_t) (-1)) //wanted atom inside not found
- {
- mp4ff_set_position(f, first_base);
- return 1;
- } else
- return 0;
-}
-
-typedef struct {
- void *data;
- unsigned written;
- unsigned allocated;
- unsigned error;
-} membuffer;
-
-#define stricmp strcasecmp
-
-static membuffer *membuffer_create(void)
-{
- const unsigned initial_size = 256;
-
- membuffer *buf = para_malloc(sizeof (membuffer));
- buf->data = para_malloc(initial_size);
- buf->written = 0;
- buf->allocated = initial_size;
- buf->error = buf->data == 0 ? 1 : 0;
-
- return buf;
-}
-
-static unsigned membuffer_write(membuffer * buf, const void *ptr, unsigned bytes)
-{
- unsigned dest_size = buf->written + bytes;
-
- if (buf->error)
- return 0;
- if (dest_size > buf->allocated) {
- do {
- buf->allocated <<= 1;
- } while (dest_size > buf->allocated);
-
- {
- void *newptr = realloc(buf->data, buf->allocated);
- if (newptr == 0) {
- free(buf->data);
- buf->data = 0;
- buf->error = 1;
- return 0;
- }
- buf->data = newptr;
- }
- }
-
- if (ptr)
- memcpy((char *) buf->data + buf->written, ptr, bytes);
- buf->written += bytes;
- return bytes;
-}
-
-static unsigned membuffer_write_atom_name(membuffer * buf, const char *data)
-{
- return membuffer_write(buf, data, 4) == 4 ? 1 : 0;
-}
-
-static unsigned membuffer_write_int16(membuffer * buf, uint16_t data)
-{
- uint8_t temp[2] = { (uint8_t) (data >> 8), (uint8_t) data };
- return membuffer_write(buf, temp, 2);
-}
-
-static unsigned membuffer_write_int32(membuffer * buf, uint32_t data)
-{
- uint8_t temp[4] = { (uint8_t) (data >> 24), (uint8_t) (data >> 16),
- (uint8_t) (data >> 8), (uint8_t) data };
- return membuffer_write(buf, temp, 4);
-}
-
-static void membuffer_write_track_tag(membuffer * buf, const char *name,
- uint32_t index, uint32_t total)
-{
- membuffer_write_int32(buf,
- 8 /*atom header */ + 8 /*data atom header */ +
- 8 /*flags + reserved */ + 8 /*actual data */ );
- membuffer_write_atom_name(buf, name);
- membuffer_write_int32(buf,
- 8 /*data atom header */ +
- 8 /*flags + reserved */ + 8 /*actual data */ );
- membuffer_write_atom_name(buf, "data");
- membuffer_write_int32(buf, 0); //flags
- membuffer_write_int32(buf, 0); //reserved
- membuffer_write_int16(buf, 0);
- membuffer_write_int16(buf, (uint16_t) index); //track number
- membuffer_write_int16(buf, (uint16_t) total); //total tracks
- membuffer_write_int16(buf, 0);
-}
-
-static void membuffer_write_int16_tag(membuffer * buf, const char *name,
- uint16_t value)
-{
- membuffer_write_int32(buf,
- 8 /*atom header */ + 8 /*data atom header */ +
- 8 /*flags + reserved */ + 2 /*actual data */ );
- membuffer_write_atom_name(buf, name);
- membuffer_write_int32(buf,
- 8 /*data atom header */ +
- 8 /*flags + reserved */ + 2 /*actual data */ );
- membuffer_write_atom_name(buf, "data");
- membuffer_write_int32(buf, 0); //flags
- membuffer_write_int32(buf, 0); //reserved
- membuffer_write_int16(buf, value); //value
-}
-
-static uint32_t myatoi(const char *param)
-{
- return param ? atoi(param) : 0;
-}
-
-static uint32_t mp4ff_meta_genre_to_index(const char *genrestr)
-{
- unsigned n;
- for (n = 0; n < sizeof (ID3v1GenreList) / sizeof (ID3v1GenreList[0]); n++) {
- if (!stricmp(genrestr, ID3v1GenreList[n]))
- return n + 1;
- }
- return 0;
-}
-
-typedef struct {
- const char *atom;
- const char *name;
-} stdmeta_entry;
-
-static stdmeta_entry stdmetas[] = {
- {"\xA9" "nam", "title"},
- {"\xA9" "ART", "artist"},
- {"\xA9" "wrt", "writer"},
- {"\xA9" "alb", "album"},
- {"\xA9" "day", "date"},
- {"\xA9" "too", "tool"},
- {"\xA9" "cmt", "comment"},
- {"cpil", "compilation"},
- {"covr", "cover"},
- {"aART", "album_artist"},
-};
-
-static const char *find_standard_meta(const char *name) //returns atom name if found, 0 if not
-{
- unsigned n;
- for (n = 0; n < sizeof (stdmetas) / sizeof (stdmetas[0]); n++) {
- if (!stricmp(name, stdmetas[n].name))
- return stdmetas[n].atom;
- }
- return 0;
-}
-
-static void membuffer_write_std_tag(membuffer * buf, const char *name,
- const char *value)
-{
- uint32_t flags = 1;
-
- /* special check for compilation flag */
- if (strcmp(name, "cpil") == 0) {
- flags = 21;
- }
-
- membuffer_write_int32(buf,
- 8 /*atom header */ + 8 /*data atom header */ +
- 8 /*flags + reserved */ + strlen(value));
- membuffer_write_atom_name(buf, name);
- membuffer_write_int32(buf,
- 8 /*data atom header */ +
- 8 /*flags + reserved */ + strlen(value));
- membuffer_write_atom_name(buf, "data");
- membuffer_write_int32(buf, flags); //flags
- membuffer_write_int32(buf, 0); //reserved
- membuffer_write(buf, value, strlen(value));
-}
-
-static void membuffer_write_custom_tag(membuffer * buf, const char *name,
- const char *value)
-{
- membuffer_write_int32(buf,
- 8 /*atom header */ +
- 0x1C /*weirdo itunes atom */ +
- 12 /*name atom header */ + strlen(name) +
- 16 /*data atom header + flags */ + strlen(value));
- membuffer_write_atom_name(buf, "----");
- membuffer_write_int32(buf, 0x1C); //weirdo itunes atom
- membuffer_write_atom_name(buf, "mean");
- membuffer_write_int32(buf, 0);
- membuffer_write(buf, "com.apple.iTunes", 16);
- membuffer_write_int32(buf, 12 + strlen(name));
- membuffer_write_atom_name(buf, "name");
- membuffer_write_int32(buf, 0);
- membuffer_write(buf, name, strlen(name));
- membuffer_write_int32(buf,
- 8 /*data atom header */ +
- 8 /*flags + reserved */ + strlen(value));
- membuffer_write_atom_name(buf, "data");
- membuffer_write_int32(buf, 1); //flags
- membuffer_write_int32(buf, 0); //reserved
- membuffer_write(buf, value, strlen(value));
-}
-
-static unsigned membuffer_error(const membuffer * buf)
-{
- return buf->error;
-}
-
-static void membuffer_free(membuffer * buf)
-{
- free(buf->data);
- free(buf);
-}
-
-static unsigned membuffer_get_size(const membuffer * buf)
-{
- return buf->written;
-}
-
-static void *membuffer_detach(membuffer * buf)
-{
- void *ret;
-
- if (buf->error)
- return 0;
-
- ret = realloc(buf->data, buf->written);
-
- if (ret == 0)
- free(buf->data);
-
- buf->data = 0;
- buf->error = 1;
-
- return ret;
-}
-
-static uint32_t create_ilst(const mp4ff_metadata_t * data, void **out_buffer,
- uint32_t * out_size)
+int mp4_open_meta(const struct mp4_callback *cb, struct mp4 **result)
{
- membuffer *buf = membuffer_create();
- unsigned metaptr;
- char *mask = para_calloc(data->count);
- {
- const char *tracknumber_ptr = 0, *totaltracks_ptr = 0;
- const char *discnumber_ptr = 0, *totaldiscs_ptr = 0;
- const char *genre_ptr = 0, *tempo_ptr = 0;
- for (metaptr = 0; metaptr < data->count; metaptr++) {
- mp4ff_tag_t *tag = &data->tags[metaptr];
- if (!stricmp(tag->item, "tracknumber") || !stricmp(tag->item, "track")) {
- if (tracknumber_ptr == 0)
- tracknumber_ptr = tag->value;
- mask[metaptr] = 1;
- } else if (!stricmp(tag->item, "totaltracks")) {
- if (totaltracks_ptr == 0)
- totaltracks_ptr = tag->value;
- mask[metaptr] = 1;
- } else if (!stricmp(tag->item, "discnumber")
- || !stricmp(tag->item, "disc")) {
- if (discnumber_ptr == 0)
- discnumber_ptr = tag->value;
- mask[metaptr] = 1;
- } else if (!stricmp(tag->item, "totaldiscs")) {
- if (totaldiscs_ptr == 0)
- totaldiscs_ptr = tag->value;
- mask[metaptr] = 1;
- } else if (!stricmp(tag->item, "genre")) {
- if (genre_ptr == 0)
- genre_ptr = tag->value;
- mask[metaptr] = 1;
- } else if (!stricmp(tag->item, "tempo")) {
- if (tempo_ptr == 0)
- tempo_ptr = tag->value;
- mask[metaptr] = 1;
- }
-
- }
+ struct mp4 *f;
+ int ret = open_file(cb, true, &f);
- if (tracknumber_ptr)
- membuffer_write_track_tag(buf, "trkn",
- myatoi(tracknumber_ptr),
- myatoi(totaltracks_ptr));
- if (discnumber_ptr)
- membuffer_write_track_tag(buf, "disk",
- myatoi(discnumber_ptr),
- myatoi(totaldiscs_ptr));
- if (tempo_ptr)
- membuffer_write_int16_tag(buf, "tmpo",
- (uint16_t) myatoi(tempo_ptr));
-
- if (genre_ptr) {
- uint32_t index = mp4ff_meta_genre_to_index(genre_ptr);
- if (index == 0)
- membuffer_write_std_tag(buf, "©gen",
- genre_ptr);
- else
- membuffer_write_int16_tag(buf, "gnre",
- (uint16_t) index);
- }
- }
-
- for (metaptr = 0; metaptr < data->count; metaptr++) {
- if (!mask[metaptr]) {
- mp4ff_tag_t *tag = &data->tags[metaptr];
- const char *std_meta_atom = find_standard_meta(tag->item);
- if (std_meta_atom) {
- membuffer_write_std_tag(buf, std_meta_atom,
- tag->value);
- } else {
- membuffer_write_custom_tag(buf, tag->item,
- tag->value);
- }
- }
+ if (ret < 0)
+ return ret;
+ if (f->udta_size == 0 || f->meta_size == 0 || f->ilst_size == 0) {
+ mp4_close(f);
+ *result = NULL;
+ return -E_MP4_MISSING_ATOM;
}
-
- free(mask);
-
- if (membuffer_error(buf)) {
- membuffer_free(buf);
- return 0;
- }
-
- *out_size = membuffer_get_size(buf);
- *out_buffer = membuffer_detach(buf);
- membuffer_free(buf);
-
+ *result = f;
return 1;
}
-static void membuffer_write_atom(membuffer * buf, const char *name, unsigned size,
- const void *data)
-{
- membuffer_write_int32(buf, size + 8);
- membuffer_write_atom_name(buf, name);
- membuffer_write(buf, data, size);
-}
-
-static void *membuffer_get_ptr(const membuffer * buf)
-{
- return buf->data;
-}
-
-static void membuffer_set_error(membuffer * buf)
-{
- buf->error = 1;
-}
-
-static unsigned membuffer_transfer_from_file(membuffer * buf, mp4ff_t * src,
- unsigned bytes)
-{
- unsigned oldsize;
- void *bufptr;
-
- oldsize = membuffer_get_size(buf);
- if (membuffer_write(buf, 0, bytes) != bytes)
- return 0;
-
- bufptr = membuffer_get_ptr(buf);
- if (bufptr == 0)
- return 0;
-
- if ((unsigned) mp4ff_read_data(src, (char *) bufptr + oldsize, bytes) !=
- bytes) {
- membuffer_set_error(buf);
- return 0;
+/**
+ * Return the metadata of an mp4 file.
+ *
+ * \param f As returned by either \ref mp4_open_read() or \ref mp4_open_meta().
+ *
+ * The caller is allowed to add, delete or modify the entries of the returned
+ * structure in order to pass the modified version to \ref mp4_meta_update().
+ */
+struct mp4_metadata *mp4_get_meta(struct mp4 *f)
+{
+ return &f->meta;
+}
+
+/** Total length of an on-disk metadata tag. */
+#define TAG_LEN(_len) (24 + (_len))
+static void create_ilst(const struct mp4_metadata *meta, uint8_t *out)
+{
+ for (unsigned n = 0; n < meta->count; n++) {
+ struct mp4_tag *tag = meta->tags + n;
+ unsigned len = strlen(tag->value);
+ const char *atom_name;
+
+ if (!strcasecmp(tag->item, "title"))
+ atom_name = "\xA9" "nam";
+ else if (!strcasecmp(tag->item, "artist"))
+ atom_name = "\xA9" "ART";
+ else if (!strcasecmp(tag->item, "album"))
+ atom_name = "\xA9" "alb";
+ else if (!strcasecmp(tag->item, "date"))
+ atom_name = "\xA9" "day";
+ else if (!strcasecmp(tag->item, "comment"))
+ atom_name = "\xA9" "cmt";
+ else
+ assert(false);
+ write_u32_be(out, TAG_LEN(len));
+ memcpy(out + 4, atom_name, 4);
+ write_u32_be(out + 8, 8 /* data atom header */
+ + 8 /* flags + reserved */
+ + len);
+ memcpy(out + 12, "data", 4);
+ write_u32_be(out + 16, 1); /* flags */
+ write_u32_be(out + 20, 0); /* reserved */
+ memcpy(out + 24, tag->value, len);
+ out += TAG_LEN(len);
}
-
- return bytes;
}
-static uint32_t create_meta(const mp4ff_metadata_t * data, void **out_buffer,
- uint32_t * out_size)
-{
- membuffer *buf;
- uint32_t ilst_size;
- void *ilst_buffer;
-
- if (!create_ilst(data, &ilst_buffer, &ilst_size))
- return 0;
-
- buf = membuffer_create();
-
- membuffer_write_int32(buf, 0);
- membuffer_write_atom(buf, "ilst", ilst_size, ilst_buffer);
- free(ilst_buffer);
-
- *out_size = membuffer_get_size(buf);
- *out_buffer = membuffer_detach(buf);
- membuffer_free(buf);
- return 1;
-}
-
-static uint32_t create_udta(const mp4ff_metadata_t * data, void **out_buffer,
-uint32_t * out_size)
-{
- membuffer *buf;
- uint32_t meta_size;
- void *meta_buffer;
-
- if (!create_meta(data, &meta_buffer, &meta_size))
- return 0;
-
- buf = membuffer_create();
-
- membuffer_write_atom(buf, "meta", meta_size, meta_buffer);
-
- free(meta_buffer);
-
- *out_size = membuffer_get_size(buf);
- *out_buffer = membuffer_detach(buf);
- membuffer_free(buf);
- return 1;
-}
-
-static uint32_t fix_byte_order_32(uint32_t src)
-{
- uint32_t result;
- uint32_t a, b, c, d;
- int8_t data[4];
-
- memcpy(data, &src, sizeof (src));
- a = (uint8_t) data[0];
- b = (uint8_t) data[1];
- c = (uint8_t) data[2];
- d = (uint8_t) data[3];
-
- result = (a << 24) | (b << 16) | (c << 8) | d;
- return (uint32_t) result;
-}
-
-static uint32_t modify_moov(mp4ff_t * f, const mp4ff_metadata_t * data,
- void **out_buffer, uint32_t * out_size)
+static void *modify_moov(struct mp4 *f, uint32_t *out_size)
{
+ int ret;
uint64_t total_base = f->moov_offset + 8;
uint32_t total_size = (uint32_t) (f->moov_size - 8);
-
- uint64_t udta_offset, meta_offset, ilst_offset;
- uint32_t udta_size, meta_size, ilst_size;
-
- uint32_t new_ilst_size;
- void *new_ilst_buffer;
-
+ uint32_t new_ilst_size = 0;
+ void *out_buffer;
uint8_t *p_out;
int32_t size_delta;
-
- if (!find_atom_v2(f, total_base, total_size, "udta", 0, "meta")) {
- membuffer *buf;
- void *new_udta_buffer;
- uint32_t new_udta_size;
- if (!create_udta(data, &new_udta_buffer, &new_udta_size))
- return 0;
-
- buf = membuffer_create();
- mp4ff_set_position(f, total_base);
- membuffer_transfer_from_file(buf, f, total_size);
-
- membuffer_write_atom(buf, "udta", new_udta_size,
- new_udta_buffer);
-
- free(new_udta_buffer);
-
- *out_size = membuffer_get_size(buf);
- *out_buffer = membuffer_detach(buf);
- membuffer_free(buf);
- return 1;
- } else {
- udta_offset = mp4ff_position(f);
- udta_size = mp4ff_read_int32(f);
- if (!find_atom_v2 (f, udta_offset + 8, udta_size - 8, "meta", 4, "ilst")) {
- membuffer *buf;
- void *new_meta_buffer;
- uint32_t new_meta_size;
- if (!create_meta(data, &new_meta_buffer, &new_meta_size))
- return 0;
-
- buf = membuffer_create();
- mp4ff_set_position(f, total_base);
- membuffer_transfer_from_file(buf, f,
- (uint32_t)(udta_offset - total_base));
-
- membuffer_write_int32(buf, udta_size + 8 + new_meta_size);
- membuffer_write_atom_name(buf, "udta");
- membuffer_transfer_from_file(buf, f, udta_size);
-
- membuffer_write_atom(buf, "meta", new_meta_size,
- new_meta_buffer);
- free(new_meta_buffer);
-
- *out_size = membuffer_get_size(buf);
- *out_buffer = membuffer_detach(buf);
- membuffer_free(buf);
- return 1;
+ uint32_t tmp;
+
+ for (unsigned n = 0; n < f->meta.count; n++)
+ new_ilst_size += TAG_LEN(strlen(f->meta.tags[n].value));
+ size_delta = new_ilst_size - (f->ilst_size - 8);
+ *out_size = total_size + size_delta;
+ out_buffer = para_malloc(*out_size);
+ p_out = out_buffer;
+ set_position(f, total_base);
+ ret = read_data(f, p_out, f->udta_offset - total_base);
+ if (ret <= 0)
+ return NULL;
+ p_out += f->udta_offset - total_base;
+ ret = read_int32(f, &tmp);
+ if (ret <= 0)
+ return NULL;
+ write_u32_be(p_out, tmp + size_delta);
+ p_out += 4;
+ ret = read_data(f, p_out, 4);
+ if (ret <= 0)
+ return NULL;
+ p_out += 4;
+ ret = read_data(f, p_out, f->meta_offset - f->udta_offset - 8);
+ if (ret <= 0)
+ return NULL;
+ p_out += f->meta_offset - f->udta_offset - 8;
+ ret = read_int32(f, &tmp);
+ if (ret <= 0)
+ return NULL;
+ write_u32_be(p_out, tmp + size_delta);
+ p_out += 4;
+ ret = read_data(f, p_out, 4);
+ if (ret <= 0)
+ return NULL;
+ p_out += 4;
+ ret = read_data(f, p_out, f->ilst_offset - f->meta_offset - 8);
+ if (ret <= 0)
+ return NULL;
+ p_out += f->ilst_offset - f->meta_offset - 8;
+ ret = read_int32(f, &tmp);
+ if (ret <= 0)
+ return NULL;
+ write_u32_be(p_out, tmp + size_delta);
+ p_out += 4;
+ ret = read_data(f, p_out, 4);
+ if (ret <= 0)
+ return NULL;
+ p_out += 4;
+ create_ilst(&f->meta, p_out);
+ p_out += new_ilst_size;
+ set_position(f, f->ilst_offset + f->ilst_size);
+ ret = read_data(f, p_out, total_size - (f->ilst_offset - total_base)
+ - f->ilst_size);
+ if (ret <= 0)
+ return NULL;
+ return out_buffer;
+}
+
+static int write_data(struct mp4 *f, void *data, size_t size)
+{
+ while (size > 0) {
+ ssize_t ret = f->cb->write(f->cb->user_data, data, size);
+ if (ret < 0) {
+ if (errno == EINTR)
+ continue;
+ return -ERRNO_TO_PARA_ERROR(errno);
}
- meta_offset = mp4ff_position(f);
- meta_size = mp4ff_read_int32(f);
- if (!find_atom(f, meta_offset + 12, meta_size - 12, "ilst"))
- return 0; //shouldn't happen, find_atom_v2 above takes care of it
- ilst_offset = mp4ff_position(f);
- ilst_size = mp4ff_read_int32(f);
-
- if (!create_ilst(data, &new_ilst_buffer, &new_ilst_size))
- return 0;
-
- size_delta = new_ilst_size - (ilst_size - 8);
-
- *out_size = total_size + size_delta;
- *out_buffer = para_malloc(*out_size);
- p_out = (uint8_t *) * out_buffer;
-
- mp4ff_set_position(f, total_base);
- mp4ff_read_data(f, p_out,
- (uint32_t) (udta_offset - total_base));
- p_out += (uint32_t) (udta_offset - total_base);
- *(uint32_t *) p_out = fix_byte_order_32(mp4ff_read_int32(f) + size_delta);
- p_out += 4;
- mp4ff_read_data(f, p_out, 4);
- p_out += 4;
- mp4ff_read_data(f, p_out,
- (uint32_t) (meta_offset - udta_offset - 8));
- p_out += (uint32_t) (meta_offset - udta_offset - 8);
- *(uint32_t *) p_out = fix_byte_order_32(mp4ff_read_int32(f) + size_delta);
- p_out += 4;
- mp4ff_read_data(f, p_out, 4);
- p_out += 4;
- mp4ff_read_data(f, p_out,
- (uint32_t) (ilst_offset - meta_offset - 8));
- p_out += (uint32_t) (ilst_offset - meta_offset - 8);
- *(uint32_t *) p_out = fix_byte_order_32(mp4ff_read_int32(f) + size_delta);
- p_out += 4;
- mp4ff_read_data(f, p_out, 4);
- p_out += 4;
-
- memcpy(p_out, new_ilst_buffer, new_ilst_size);
- p_out += new_ilst_size;
-
- mp4ff_set_position(f, ilst_offset + ilst_size);
- mp4ff_read_data(f, p_out, (uint32_t) (total_size
- - (ilst_offset - total_base) - ilst_size));
-
- free(new_ilst_buffer);
+ f->current_position += ret;
+ size -= ret;
}
return 1;
}
-static int32_t mp4ff_write_data(mp4ff_t * f, void *data, uint32_t size)
-{
- int32_t result = 1;
-
- result = f->stream->write(f->stream->user_data, data, size);
-
- f->current_position += size;
-
- return result;
-}
-
-static int32_t mp4ff_write_int32(mp4ff_t * f, const uint32_t data)
-{
- uint32_t result;
- uint32_t a, b, c, d;
- int8_t temp[4];
-
- *(uint32_t *) temp = data;
- a = (uint8_t) temp[0];
- b = (uint8_t) temp[1];
- c = (uint8_t) temp[2];
- d = (uint8_t) temp[3];
-
- result = (a << 24) | (b << 16) | (c << 8) | d;
-
- return mp4ff_write_data(f, (uint8_t *) & result, sizeof (result));
-}
-
-static int32_t mp4ff_truncate(mp4ff_t * f)
-{
- return f->stream->truncate(f->stream->user_data);
-}
-
-int32_t mp4ff_meta_update(mp4ff_callback_t * f, const mp4ff_metadata_t * data)
+int mp4_meta_update(struct mp4 *f)
{
void *new_moov_data;
uint32_t new_moov_size;
+ uint8_t buf[8] = "----moov";
+ int ret;
- mp4ff_t *ff = para_calloc(sizeof(mp4ff_t));
- ff->stream = f;
- mp4ff_set_position(ff, 0);
-
- parse_atoms(ff, 1);
-
- if (!modify_moov(ff, data, &new_moov_data, &new_moov_size)) {
- mp4ff_close(ff);
+ set_position(f, 0);
+ new_moov_data = modify_moov(f, &new_moov_size);
+ if (!new_moov_data ) {
+ mp4_close(f);
return 0;
}
-
- /* copy moov atom to end of the file */
- if (ff->last_atom != ATOM_MOOV) {
- char *free_data = "free";
-
- /* rename old moov to free */
- mp4ff_set_position(ff, ff->moov_offset + 4);
- mp4ff_write_data(ff, free_data, 4);
-
- mp4ff_set_position(ff, ff->file_size);
- mp4ff_write_int32(ff, new_moov_size + 8);
- mp4ff_write_data(ff, "moov", 4);
- mp4ff_write_data(ff, new_moov_data, new_moov_size);
- } else {
- mp4ff_set_position(ff, ff->moov_offset);
- mp4ff_write_int32(ff, new_moov_size + 8);
- mp4ff_write_data(ff, "moov", 4);
- mp4ff_write_data(ff, new_moov_data, new_moov_size);
- }
-
- mp4ff_truncate(ff);
-
- mp4ff_close(ff);
- return 1;
+ if (f->last_atom != ATOM_MOOV) {
+ set_position(f, f->moov_offset + 4);
+ ret = write_data(f, "free", 4); /* rename old moov to free */
+ if (ret < 0)
+ goto free_moov;
+ set_position(f, f->file_size); /* write new moov atom at EOF */
+ } else /* overwrite old moov atom */
+ set_position(f, f->moov_offset);
+ write_u32_be(buf, new_moov_size + 8);
+ ret = write_data(f, buf, sizeof(buf));
+ if (ret < 0)
+ goto free_moov;
+ ret = write_data(f, new_moov_data, new_moov_size);
+ if (ret < 0)
+ goto free_moov;
+ f->cb->truncate(f->cb->user_data);
+ ret = 1;
+free_moov:
+ free(new_moov_data);
+ return ret;
}
-/* find a metadata item by name */
-/* returns 0 if item found, 1 if no such item */
-static int32_t mp4ff_meta_find_by_name(const mp4ff_t * f, const char *item,
- char **value)
+static char *meta_find_by_name(const struct mp4 *f, const char *item)
{
uint32_t i;
- for (i = 0; i < f->tags.count; i++) {
- if (!stricmp(f->tags.tags[i].item, item)) {
- *value = para_strdup(f->tags.tags[i].value);
- return 1;
- }
- }
-
- *value = NULL;
-
- /* not found */
- return 0;
+ for (i = 0; i < f->meta.count; i++)
+ if (!strcasecmp(f->meta.tags[i].item, item))
+ return para_strdup(f->meta.tags[i].value);
+ return NULL;
}
-int32_t mp4ff_meta_get_artist(const mp4ff_t * f, char **value)
+/**
+ * Return the value of the artist meta tag of an mp4 file.
+ *
+ * \param f Must not be NULL.
+ *
+ * \return If the file does not contain this metadata tag, the function returns
+ * NULL. Otherwise, a copy of the tag value is returned. The caller should free
+ * this memory when it is no longer needed.
+ */
+char *mp4_meta_get_artist(const struct mp4 *f)
{
- return mp4ff_meta_find_by_name(f, "artist", value);
+ return meta_find_by_name(f, "artist");
}
-int32_t mp4ff_meta_get_title(const mp4ff_t * f, char **value)
+/**
+ * Return the value of the title meta tag of an mp4 file.
+ *
+ * \param f See \ref mp4_meta_get_artist().
+ * \return See \ref mp4_meta_get_artist().
+ */
+char *mp4_meta_get_title(const struct mp4 *f)
{
- return mp4ff_meta_find_by_name(f, "title", value);
+ return meta_find_by_name(f, "title");
}
-int32_t mp4ff_meta_get_date(const mp4ff_t * f, char **value)
+/**
+ * Return the value of the date meta tag of an mp4 file.
+ *
+ * \param f See \ref mp4_meta_get_artist().
+ * \return See \ref mp4_meta_get_artist().
+ */
+char *mp4_meta_get_date(const struct mp4 *f)
{
- return mp4ff_meta_find_by_name(f, "date", value);
+ return meta_find_by_name(f, "date");
}
-int32_t mp4ff_meta_get_album(const mp4ff_t * f, char **value)
+/**
+ * Return the value of the album meta tag of an mp4 file.
+ *
+ * \param f See \ref mp4_meta_get_artist().
+ * \return See \ref mp4_meta_get_artist().
+ */
+char *mp4_meta_get_album(const struct mp4 *f)
{
- return mp4ff_meta_find_by_name(f, "album", value);
+ return meta_find_by_name(f, "album");
}
-int32_t mp4ff_meta_get_comment(const mp4ff_t * f, char **value)
+/**
+ * Return the value of the comment meta tag of an mp4 file.
+ *
+ * \param f See \ref mp4_meta_get_artist().
+ * \return See \ref mp4_meta_get_artist().
+ */
+char *mp4_meta_get_comment(const struct mp4 *f)
{
- return mp4ff_meta_find_by_name(f, "comment", value);
+ return meta_find_by_name(f, "comment");
}