#include "mp4.h"
struct mp4_track {
- uint16_t channelCount;
- uint16_t sampleRate;
+ /* mp4a */
+ uint16_t channel_count;
+ uint16_t sample_rate;
/* stsz */
uint32_t stsz_sample_size;
uint32_t stco_entry_count;
uint32_t *stco_chunk_offset;
- uint32_t timeScale;
+ /* mdhd */
+ uint32_t time_scale;
uint64_t duration;
};
return ret;
}
-static bool atom_compare(int8_t a1, int8_t b1, int8_t c1, int8_t d1,
- int8_t a2, int8_t b2, int8_t c2, int8_t d2)
+#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)
{
- return a1 == a2 && b1 == b2 && c1 == c2 && d1 == d2;
-}
-
-enum atoms {
- /* atoms with subatoms */
- ATOM_MOOV = 1,
- ATOM_TRAK = 2,
- ATOM_EDTS = 3,
- ATOM_MDIA = 4,
- ATOM_MINF = 5,
- ATOM_STBL = 6,
- ATOM_UDTA = 7,
- ATOM_ILST = 8, /* iTunes Metadata list */
- ATOM_TITLE = 9,
- ATOM_ARTIST = 10,
- ATOM_WRITER = 11,
- ATOM_ALBUM = 12,
- ATOM_DATE = 13,
- ATOM_TOOL = 14,
- ATOM_COMMENT = 15,
- ATOM_GENRE1 = 16,
- ATOM_TRACK = 17,
- ATOM_DISC = 18,
- ATOM_COMPILATION = 19,
- ATOM_GENRE2 = 20,
- ATOM_TEMPO = 21,
- ATOM_COVER = 22,
- ATOM_DRMS = 23,
- ATOM_SINF = 24,
- ATOM_SCHI = 25,
-
- SUBATOMIC = 128,
-
- /* atoms without subatoms */
- ATOM_FTYP = 129,
- ATOM_MDAT = 130,
- ATOM_MVHD = 131,
- ATOM_TKHD = 132,
- ATOM_TREF = 133,
- ATOM_MDHD = 134, /* track header */
- ATOM_VMHD = 135,
- ATOM_SMHD = 136,
- ATOM_HMHD = 137,
- ATOM_STSD = 138, /* sample description box */
- ATOM_STTS = 139, /* time to sample box */
- ATOM_STSZ = 140, /* sample size box */
- ATOM_STZ2 = 141,
- ATOM_STCO = 142, /* chunk offset box */
- ATOM_STSC = 143, /* sample to chunk box */
- ATOM_MP4A = 144,
- ATOM_MP4V = 145,
- ATOM_MP4S = 146,
- ATOM_ESDS = 147,
- ATOM_META = 148, /* iTunes Metadata box */
- ATOM_NAME = 149, /* iTunes Metadata name box */
- ATOM_DATA = 150, /* iTunes Metadata data box */
- ATOM_CTTS = 151,
- ATOM_FRMA = 152,
- ATOM_IVIV = 153,
- ATOM_PRIV = 154,
- ATOM_USER = 155,
- ATOM_KEY = 156,
- ATOM_ALBUM_ARTIST = 157,
- ATOM_CONTENTGROUP = 158,
- ATOM_LYRICS = 159,
- ATOM_DESCRIPTION = 160,
- ATOM_NETWORK = 161,
- ATOM_SHOW = 162,
- ATOM_EPISODENAME = 163,
- ATOM_SORTTITLE = 164,
- ATOM_SORTALBUM = 165,
- ATOM_SORTARTIST = 166,
- ATOM_SORTALBUMARTIST = 167,
- ATOM_SORTWRITER = 168,
- ATOM_SORTSHOW = 169,
- ATOM_SEASON = 170,
- ATOM_EPISODE = 171,
- ATOM_PODCAST = 172,
-
- ATOM_UNKNOWN = 255
-};
-
-#define ATOM_FREE ATOM_UNKNOWN
-#define ATOM_SKIP ATOM_UNKNOWN
-
-#define COPYRIGHT_SYMBOL ((int8_t)0xA9)
-
-static uint8_t atom_name_to_type(int8_t a, int8_t b, int8_t c, int8_t d)
-{
- if (a == 'm') {
- if (atom_compare(a, b, c, d, 'm', 'o', 'o', 'v'))
- return ATOM_MOOV;
- else if (atom_compare(a, b, c, d, 'm', 'i', 'n', 'f'))
- return ATOM_MINF;
- else if (atom_compare(a, b, c, d, 'm', 'd', 'i', 'a'))
- return ATOM_MDIA;
- else if (atom_compare(a, b, c, d, 'm', 'd', 'a', 't'))
- return ATOM_MDAT;
- else if (atom_compare(a, b, c, d, 'm', 'd', 'h', 'd'))
- return ATOM_MDHD;
- else if (atom_compare(a, b, c, d, 'm', 'v', 'h', 'd'))
- return ATOM_MVHD;
- else if (atom_compare(a, b, c, d, 'm', 'p', '4', 'a'))
- return ATOM_MP4A;
- else if (atom_compare(a, b, c, d, 'm', 'p', '4', 'v'))
- return ATOM_MP4V;
- else if (atom_compare(a, b, c, d, 'm', 'p', '4', 's'))
- return ATOM_MP4S;
- else if (atom_compare(a, b, c, d, 'm', 'e', 't', 'a'))
- return ATOM_META;
- } else if (a == 't') {
- if (atom_compare(a, b, c, d, 't', 'r', 'a', 'k'))
- return ATOM_TRAK;
- else if (atom_compare(a, b, c, d, 't', 'k', 'h', 'd'))
- return ATOM_TKHD;
- else if (atom_compare(a, b, c, d, 't', 'r', 'e', 'f'))
- return ATOM_TREF;
- else if (atom_compare(a, b, c, d, 't', 'r', 'k', 'n'))
- return ATOM_TRACK;
- else if (atom_compare(a, b, c, d, 't', 'm', 'p', 'o'))
- return ATOM_TEMPO;
- else if (atom_compare(a, b, c, d, 't', 'v', 'n', 'n'))
- return ATOM_NETWORK;
- else if (atom_compare(a, b, c, d, 't', 'v', 's', 'h'))
- return ATOM_SHOW;
- else if (atom_compare(a, b, c, d, 't', 'v', 'e', 'n'))
- return ATOM_EPISODENAME;
- else if (atom_compare(a, b, c, d, 't', 'v', 's', 'n'))
- return ATOM_SEASON;
- else if (atom_compare(a, b, c, d, 't', 'v', 'e', 's'))
- return ATOM_EPISODE;
- } else if (a == 's') {
- if (atom_compare(a, b, c, d, 's', 't', 'b', 'l'))
- return ATOM_STBL;
- else if (atom_compare(a, b, c, d, 's', 'm', 'h', 'd'))
- return ATOM_SMHD;
- else if (atom_compare(a, b, c, d, 's', 't', 's', 'd'))
- return ATOM_STSD;
- else if (atom_compare(a, b, c, d, 's', 't', 't', 's'))
- return ATOM_STTS;
- else if (atom_compare(a, b, c, d, 's', 't', 'c', 'o'))
- return ATOM_STCO;
- else if (atom_compare(a, b, c, d, 's', 't', 's', 'c'))
- return ATOM_STSC;
- else if (atom_compare(a, b, c, d, 's', 't', 's', 'z'))
- return ATOM_STSZ;
- else if (atom_compare(a, b, c, d, 's', 't', 'z', '2'))
- return ATOM_STZ2;
- else if (atom_compare(a, b, c, d, 's', 'k', 'i', 'p'))
- return ATOM_SKIP;
- else if (atom_compare(a, b, c, d, 's', 'i', 'n', 'f'))
- return ATOM_SINF;
- else if (atom_compare(a, b, c, d, 's', 'c', 'h', 'i'))
- return ATOM_SCHI;
- else if (atom_compare(a, b, c, d, 's', 'o', 'n', 'm'))
- return ATOM_SORTTITLE;
- else if (atom_compare(a, b, c, d, 's', 'o', 'a', 'l'))
- return ATOM_SORTALBUM;
- else if (atom_compare(a, b, c, d, 's', 'o', 'a', 'r'))
- return ATOM_SORTARTIST;
- else if (atom_compare(a, b, c, d, 's', 'o', 'a', 'a'))
- return ATOM_SORTALBUMARTIST;
- else if (atom_compare(a, b, c, d, 's', 'o', 'c', 'o'))
- return ATOM_SORTWRITER;
- else if (atom_compare(a, b, c, d, 's', 'o', 's', 'n'))
- return ATOM_SORTSHOW;
- } else if (a == COPYRIGHT_SYMBOL) {
- if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'n', 'a', 'm'))
- return ATOM_TITLE;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'A', 'R', 'T'))
- return ATOM_ARTIST;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'w', 'r', 't'))
- return ATOM_WRITER;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'a', 'l', 'b'))
- return ATOM_ALBUM;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'd', 'a', 'y'))
- return ATOM_DATE;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 't', 'o', 'o'))
- return ATOM_TOOL;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'c', 'm', 't'))
- return ATOM_COMMENT;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'g', 'e', 'n'))
- return ATOM_GENRE1;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'g', 'r', 'p'))
- return ATOM_CONTENTGROUP;
- else if (atom_compare(a, b, c, d, COPYRIGHT_SYMBOL, 'l', 'y', 'r'))
- return ATOM_LYRICS;
- }
-
- if (atom_compare(a, b, c, d, 'e', 'd', 't', 's'))
- return ATOM_EDTS;
- else if (atom_compare(a, b, c, d, 'e', 's', 'd', 's'))
- return ATOM_ESDS;
- else if (atom_compare(a, b, c, d, 'f', 't', 'y', 'p'))
- return ATOM_FTYP;
- else if (atom_compare(a, b, c, d, 'f', 'r', 'e', 'e'))
- return ATOM_FREE;
- else if (atom_compare(a, b, c, d, 'h', 'm', 'h', 'd'))
- return ATOM_HMHD;
- else if (atom_compare(a, b, c, d, 'v', 'm', 'h', 'd'))
- return ATOM_VMHD;
- else if (atom_compare(a, b, c, d, 'u', 'd', 't', 'a'))
- return ATOM_UDTA;
- else if (atom_compare(a, b, c, d, 'i', 'l', 's', 't'))
- return ATOM_ILST;
- else if (atom_compare(a, b, c, d, 'n', 'a', 'm', 'e'))
- return ATOM_NAME;
- else if (atom_compare(a, b, c, d, 'd', 'a', 't', 'a'))
- return ATOM_DATA;
- else if (atom_compare(a, b, c, d, 'd', 'i', 's', 'k'))
- return ATOM_DISC;
- else if (atom_compare(a, b, c, d, 'g', 'n', 'r', 'e'))
- return ATOM_GENRE2;
- else if (atom_compare(a, b, c, d, 'c', 'o', 'v', 'r'))
- return ATOM_COVER;
- else if (atom_compare(a, b, c, d, 'c', 'p', 'i', 'l'))
- return ATOM_COMPILATION;
- else if (atom_compare(a, b, c, d, 'c', 't', 't', 's'))
- return ATOM_CTTS;
- else if (atom_compare(a, b, c, d, 'd', 'r', 'm', 's'))
- return ATOM_DRMS;
- else if (atom_compare(a, b, c, d, 'f', 'r', 'm', 'a'))
- return ATOM_FRMA;
- else if (atom_compare(a, b, c, d, 'p', 'r', 'i', 'v'))
- return ATOM_PRIV;
- else if (atom_compare(a, b, c, d, 'i', 'v', 'i', 'v'))
- return ATOM_IVIV;
- else if (atom_compare(a, b, c, d, 'u', 's', 'e', 'r'))
- return ATOM_USER;
- else if (atom_compare(a, b, c, d, 'k', 'e', 'y', ' '))
- return ATOM_KEY;
- else if (atom_compare(a, b, c, d, 'a', 'A', 'R', 'T'))
- return ATOM_ALBUM_ARTIST;
- else if (atom_compare(a, b, c, d, 'd', 'e', 's', 'c'))
- return ATOM_DESCRIPTION;
- else if (atom_compare(a, b, c, d, 'p', 'c', 's', 't'))
- return ATOM_PODCAST;
- else
- return ATOM_UNKNOWN;
+ #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. */
{
uint32_t size;
int ret;
- int8_t atom_header[8];
+ uint8_t atom_header[8];
ret = read_data(f, atom_header, 8);
if (ret <= 0)
if (atom_size)
*atom_size = size;
}
- *atom_type = atom_name_to_type(atom_header[4], atom_header[5],
- atom_header[6], atom_header[7]);
+ *atom_type = atom_name_to_type(atom_header + 4);
return 1;
}
ret = read_int32(f, NULL); /* reserved */
if (ret <= 0)
return ret;
- ret = read_int16(f, &t->channelCount);
+ ret = read_int16(f, &t->channel_count);
if (ret <= 0)
return ret;
ret = read_int16(f, NULL);
ret = read_int16(f, NULL);
if (ret <= 0)
return ret;
- return read_int16(f, &t->sampleRate);
+ return read_int16(f, &t->sample_rate);
}
static int read_stsd(struct mp4 *f)
ret = read_int64(f, NULL); /* modification-time */
if (ret <= 0)
return ret;
- ret = read_int32(f, &t->timeScale);
+ ret = read_int32(f, &t->time_scale);
if (ret <= 0)
return ret;
ret = read_int64(f, &t->duration);
ret = read_int32(f, NULL); /* modification-time */
if (ret <= 0)
return ret;
- ret = read_int32(f, &t->timeScale);
+ ret = read_int32(f, &t->time_scale);
if (ret <= 0)
return ret;
ret = read_int32(f, &temp);
return 1;
}
-static int parse_leaf_atom(struct mp4 *f, uint64_t size, uint8_t header_size,
- uint8_t atom_type)
-{
- uint64_t dest_position = get_position(f) + size - 8;
- int ret = 1; /* return success for atoms we don't care about */
-
- 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;
- }
- set_position(f, dest_position);
- return ret;
-}
-
static bool need_atom(uint8_t atom_type, bool meta_only)
{
/* these are needed in any case */
static int parse_sub_atoms(struct mp4 *f, uint64_t total_size, bool meta_only)
{
int ret;
- uint64_t size;
- uint8_t atom_type = 0;
- uint64_t counted_size = 0;
- uint8_t header_size = 0;
+ uint64_t dest, size, end = get_position(f) + total_size;
- while (counted_size < total_size) {
+ 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)
return -1;
- counted_size += size;
+ dest = get_position(f) + size - header_size;
if (atom_type == ATOM_TRAK) {
if (f->total_tracks >= MAX_TRACKS)
return -1;
f->udta_offset = get_position(f) - header_size;
f->udta_size = size;
}
- if (!need_atom(atom_type, meta_only)) {
- set_position(f, get_position(f) + size - header_size);
+ if (!need_atom(atom_type, meta_only))
continue;
- }
- if (atom_type < SUBATOMIC) /* atom contains subatoms */
+ 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);
- else
- ret = parse_leaf_atom(f, size, header_size, atom_type);
+ }
if (ret <= 0)
return ret;
}
{
const struct mp4_track *t = f->audio_track;
- if (t->timeScale == 0)
+ if (t->time_scale == 0)
return 0;
- return t->duration * 1000 / t->timeScale;
+ return t->duration * 1000 / t->time_scale;
}
int mp4_set_sample_position(struct mp4 *f, int32_t sample)
uint32_t mp4_get_sample_rate(const struct mp4 *f)
{
- return f->audio_track->sampleRate;
+ return f->audio_track->sample_rate;
}
uint32_t mp4_get_channel_count(const struct mp4 *f)
{
- return f->audio_track->channelCount ;
+ return f->audio_track->channel_count;
}
int32_t mp4_num_samples(const struct mp4 *f)
}
}
-static uint32_t fix_byte_order_32(uint32_t src)
-{
- return read_u32_be(&src);
-}
-
static void *modify_moov(struct mp4 *f, uint32_t *out_size)
{
int ret;
ret = read_int32(f, &tmp);
if (ret <= 0)
return NULL;
- *(uint32_t *)p_out = fix_byte_order_32(tmp + size_delta);
+ write_u32_be(p_out, tmp + size_delta);
p_out += 4;
ret = read_data(f, p_out, 4);
if (ret <= 0)
ret = read_int32(f, &tmp);
if (ret <= 0)
return NULL;
- *(uint32_t *)p_out = fix_byte_order_32(tmp + size_delta);
+ write_u32_be(p_out, tmp + size_delta);
p_out += 4;
ret = read_data(f, p_out, 4);
if (ret <= 0)
ret = read_int32(f, &tmp);
if (ret <= 0)
return NULL;
- *(uint32_t *)p_out = fix_byte_order_32(tmp + size_delta);
+ write_u32_be(p_out, tmp + size_delta);
p_out += 4;
ret = read_data(f, p_out, 4);
if (ret <= 0)
mp4_close(f);
return 0;
}
- /* copy moov atom to end of the file */
if (f->last_atom != ATOM_MOOV) {
- char *free_data = "free";
-
- /* rename old moov to free */
set_position(f, f->moov_offset + 4);
- write_data(f, free_data, 4);
-
- set_position(f, f->file_size);
- write_int32(f, new_moov_size + 8);
- write_data(f, "moov", 4);
- write_data(f, new_moov_data, new_moov_size);
- } else {
+ write_data(f, "free", 4); /* rename old moov to free */
+ set_position(f, f->file_size); /* write new moov atom at EOF */
+ } else /* overwrite old moov atom */
set_position(f, f->moov_offset);
- write_int32(f, new_moov_size + 8);
- write_data(f, "moov", 4);
- write_data(f, new_moov_data, new_moov_size);
- }
+ write_int32(f, new_moov_size + 8);
+ write_data(f, "moov", 4);
+ write_data(f, new_moov_data, new_moov_size);
free(new_moov_data);
f->cb->truncate(f->cb->user_data);
return 1;