#include "mp4.h"
struct mp4_track {
- bool is_audio;
uint16_t channelCount;
uint16_t sampleRate;
/* stts */
uint32_t stts_entry_count;
uint32_t *stts_sample_count;
- uint32_t *stts_sample_delta;
/* stsc */
uint32_t stsc_entry_count;
uint32_t *stsc_first_chunk;
uint32_t *stsc_samples_per_chunk;
- uint32_t *stsc_sample_desc_index;
/* stsc */
uint32_t stco_entry_count;
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;
+
uint8_t last_atom;
uint64_t file_size;
int32_t total_tracks;
/* track data */
struct mp4_track *track[MAX_TRACKS];
+ /* the first audio track found */
+ struct mp4_track *audio_track;
/* metadata */
struct mp4_metadata meta;
return f->current_position;
}
-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;
- }
-}
-
static int32_t set_position(struct mp4 *f, int64_t position)
{
f->cb->seek(f->cb->user_data, position);
return ret;
t->stts_sample_count = para_malloc(t->stts_entry_count
* sizeof(int32_t));
- t->stts_sample_delta = 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, &t->stts_sample_delta[i]);
+ ret = read_int32(f, NULL); /* sample delta */
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));
- t->stsc_sample_desc_index = 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)
ret = read_int32(f, &t->stsc_samples_per_chunk[i]);
if (ret <= 0)
return ret;
- ret = read_int32(f, &t->stsc_sample_desc_index[i]);
+ ret = read_int32(f, NULL); /* sample desc index */
if (ret <= 0)
return ret;
}
{
int ret;
int32_t i;
- uint8_t atom_type = 0;
- uint8_t header_size = 0;
struct mp4_track *t;
if (f->total_tracks == 0)
ret = read_int16(f, NULL);
if (ret <= 0)
return ret;
- ret = read_int16(f, &t->sampleRate);
- if (ret <= 0)
- return ret;
- ret = read_int16(f, NULL);
- if (ret <= 0)
- return ret;
- return atom_read_header(f, &atom_type, &header_size, NULL);
+ return read_int16(f, &t->sampleRate);
}
static int read_stsd(struct mp4 *f)
{
int ret;
uint32_t i, entry_count;
- uint8_t header_size = 0;
struct mp4_track *t;
if (f->total_tracks == 0)
uint64_t skip = get_position(f);
uint64_t size;
uint8_t atom_type = 0;
- ret = atom_read_header(f, &atom_type, &header_size, &size);
+ ret = atom_read_header(f, &atom_type, NULL, &size);
if (ret <= 0)
return ret;
skip += size;
- t->is_audio = atom_type == ATOM_MP4A;
- if (t->is_audio)
+ if (!f->audio_track && atom_type == ATOM_MP4A) {
+ f->audio_track = t;
read_mp4a(f);
+ }
set_position(f, skip);
}
return 1;
}
-static int32_t tag_add_field(struct mp4_metadata *meta, const char *item,
- const char *value, int32_t len)
-{
- meta->tags = para_realloc(meta->tags,
- (meta->count + 1) * sizeof(struct mp4_tag));
- meta->tags[meta->count].item = para_strdup(item);
- meta->tags[meta->count].len = len;
- if (len >= 0) {
- meta->tags[meta->count].value = para_malloc(len + 1);
- memcpy(meta->tags[meta->count].value, value, len);
- meta->tags[meta->count].value[len] = 0;
- } else {
- meta->tags[meta->count].value = para_strdup(value);
- }
- meta->count++;
- return 1;
-}
-
-static int read_string(struct mp4 *f, uint32_t length, char **result)
-{
- char *str = para_malloc(length + 1);
- int ret = read_data(f, str, length);
-
- if (ret <= 0) {
- free(str);
- *result = NULL;
- } else {
- str[length] = '\0';
- *result = str;
- }
- return ret;
-}
-
static const char *get_metadata_name(uint8_t atom_type)
{
switch (atom_type) {
{
int ret;
uint64_t subsize, sumsize;
- char *data = NULL;
+ char *value = NULL;
uint32_t len = 0;
uint64_t destpos;
+ struct mp4_tag *tag;
for (
sumsize = 0;
uint8_t header_size = 0;
ret = atom_read_header(f, &atom_type, &header_size, &subsize);
if (ret <= 0)
- return ret;
+ goto fail;
destpos = get_position(f) + subsize - header_size;
if (atom_type != ATOM_DATA)
continue;
ret = read_int8(f, NULL); /* version */
if (ret <= 0)
- return ret;
+ goto fail;
ret = read_int24(f, NULL); /* flags */
if (ret <= 0)
- return ret;
+ goto fail;
ret = read_int32(f, NULL); /* reserved */
if (ret <= 0)
- return ret;
- free(data);
- ret = read_string(f, subsize - (header_size + 8), &data);
- if (ret <= 0)
- return ret;
+ 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';
}
- if (!data)
- return -1;
- tag_add_field(&f->meta, get_metadata_name(parent), data, len);
- free(data);
+ 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 int read_mdhd(struct mp4 *f)
case ATOM_ALBUM:
case ATOM_COMMENT:
case ATOM_DATE:
- parse_tag(f, atom_type, subsize - header_size);
+ ret = parse_tag(f, atom_type, subsize - header_size);
+ if (ret <= 0)
+ return ret;
}
set_position(f, destpos);
sumsize += subsize;
return ret;
if (subsize <= header_size + 4)
return 1;
- if (atom_type == ATOM_ILST)
- read_ilst(f, subsize - (header_size + 4));
- else
+ if (atom_type == ATOM_ILST) {
+ 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 1;
}
-static int atom_read(struct mp4 *f, uint64_t size, uint8_t atom_type)
+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 */
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: ret = read_meta(f, size); 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 */
+ 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;
+ }
+ /* 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 int parse_sub_atoms(struct mp4 *f, uint64_t total_size, int meta_only)
+static int parse_sub_atoms(struct mp4 *f, uint64_t total_size, bool meta_only)
{
int ret;
uint64_t size;
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)) {
+ if (!need_atom(atom_type, meta_only)) {
set_position(f, get_position(f) + size - header_size);
- } else if (atom_type < SUBATOMIC) {
- ret = parse_sub_atoms(f, size - header_size, meta_only);
- if (ret <= 0)
- return ret;
- } else {
- ret = atom_read(f, size, atom_type);
- if (ret <= 0)
- return ret;
+ continue;
}
+ if (atom_type < SUBATOMIC) /* atom contains subatoms */
+ 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;
}
return 1;
}
-/* parse root atoms */
-static int parse_atoms(struct mp4 *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;
+ uint8_t atom_type, header_size;
+ struct mp4 *f = para_calloc(sizeof(*f));
+ 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 = get_position(f) - header_size;
- f->moov_size = size;
- }
-
- /* parse subatoms */
- if (meta_only && !need_parse_when_meta_only(atom_type)) {
- set_position(f, get_position(f) + size - header_size);
- } else if (atom_type < SUBATOMIC) {
- ret = parse_sub_atoms(f, size - header_size, meta_only);
- if (ret <= 0)
- return ret;
- } else {
- /* skip this atom */
+ 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;
}
-struct mp4 *mp4_open_read(const struct mp4_callback *cb)
+int mp4_open_read(const struct mp4_callback *cb, struct mp4 **result)
{
- int ret;
- struct mp4 *f = para_calloc(sizeof(struct mp4));
-
- f->cb = cb;
- ret = parse_atoms(f, 0);
- if (ret < 0) {
- free(f);
- return NULL;
- }
- return f;
+ return open_file(cb, false, result);
}
void mp4_close(struct mp4 *f)
if (f->track[i]) {
free(f->track[i]->stsz_table);
free(f->track[i]->stts_sample_count);
- free(f->track[i]->stts_sample_delta);
free(f->track[i]->stsc_first_chunk);
free(f->track[i]->stsc_samples_per_chunk);
- free(f->track[i]->stsc_sample_desc_index);
free(f->track[i]->stco_chunk_offset);
free(f->track[i]);
}
free(f);
}
-static int32_t chunk_of_sample(const struct mp4 *f, int32_t track,
- int32_t sample, int32_t *chunk_sample, int32_t *chunk)
+static int32_t chunk_of_sample(const struct mp4 *f, int32_t sample,
+ int32_t *chunk)
{
- int32_t total_entries = 0;
- int32_t chunk2entry;
- int32_t chunk1, chunk2, chunk1samples, range_samples, total = 0;
-
- *chunk_sample = 0;
- *chunk = 1;
-
- total_entries = f->track[track]->stsc_entry_count;
-
- chunk1 = 1;
- chunk1samples = 0;
- chunk2entry = 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;
- do {
- chunk2 = f->track[track]->stsc_first_chunk[chunk2entry];
- *chunk = chunk2 - chunk1;
- range_samples = *chunk * chunk1samples;
-
- if (sample < total + range_samples)
+ 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;
-
- 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)
+ }
+ chunk1 = fc[i - 1];
+ chunk1samples = spc[i - 1];
+ if (chunk1samples != 0)
*chunk = (sample - total) / chunk1samples + chunk1;
else
*chunk = 1;
-
- *chunk_sample = total + (*chunk - chunk1) * chunk1samples;
-
- return 0;
+ return total + (*chunk - chunk1) * chunk1samples;
}
/**
- * Return the number of milliseconds of the given track.
+ * Return the number of milliseconds of the audio track.
*
* \param f As returned by \ref mp4_open_read(), must not be NULL.
- * \param track Between zero and the value returned by \ref mp4_get_total_tracks().
- *
- * The function returns zero if the audio file is of zero length or contains a
- * corrupt track header.
*/
-uint64_t mp4_get_duration(const struct mp4 *f, int32_t track)
+uint64_t mp4_get_duration(const struct mp4 *f)
{
- const struct mp4_track *t = f->track[track];
+ const struct mp4_track *t = f->audio_track;
if (t->timeScale == 0)
return 0;
return t->duration * 1000 / t->timeScale;
}
-int32_t mp4_get_total_tracks(const struct mp4 *f)
+int mp4_set_sample_position(struct mp4 *f, int32_t sample)
{
- return f->total_tracks;
-}
-
-/**
- * Check whether the given track number corresponds to an audio track.
- *
- * \param f See \ref mp4_get_duration().
- * \param track See \ref mp4_get_duration().
- *
- * Besides audio tracks, an mp4 file may contain video and system tracks. For
- * those the function returns false.
- */
-bool mp4_is_audio_track(const struct mp4 *f, int32_t track)
-{
- return f->track[track]->is_audio;
-}
-
-int mp4_set_sample_position(struct mp4 *f, uint32_t track, int32_t sample)
-{
- const struct mp4_track *t = f->track[track];
+ const struct mp4_track *t = f->audio_track;
int32_t offset, chunk, chunk_sample;
uint32_t n, srs; /* sample range size */
- if (sample >= t->stsz_sample_count || track >= f->total_tracks)
+ if (sample >= t->stsz_sample_count)
return -ERRNO_TO_PARA_ERROR(EINVAL);
- chunk_of_sample(f, track, sample, &chunk_sample, &chunk);
+ chunk_sample = chunk_of_sample(f, sample, &chunk);
if (t->stsz_sample_size > 0)
srs = (sample - chunk_sample) * t->stsz_sample_size;
else {
return 1;
}
-int32_t mp4_get_sample_size(const struct mp4 *f, int track, int sample)
+int32_t mp4_get_sample_size(const struct mp4 *f, int sample)
{
- const struct mp4_track *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 mp4_get_sample_rate(const struct mp4 *f, int32_t track)
+uint32_t mp4_get_sample_rate(const struct mp4 *f)
{
- return f->track[track]->sampleRate;
+ return f->audio_track->sampleRate;
}
-uint32_t mp4_get_channel_count(const struct mp4 *f, int32_t track)
+uint32_t mp4_get_channel_count(const struct mp4 *f)
{
- return f->track[track]->channelCount;
+ return f->audio_track->channelCount ;
}
-int32_t mp4_num_samples(const struct mp4 *f, 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;
}
-struct mp4 *mp4_open_meta(const struct mp4_callback *cb)
+int mp4_open_meta(const struct mp4_callback *cb, struct mp4 **result)
{
- int ret;
- struct mp4 *f = para_calloc(sizeof(struct mp4));
+ struct mp4 *f;
+ int ret = open_file(cb, true, &f);
- f->cb = cb;
- ret = parse_atoms(f, 1);
- if (ret < 0) {
- free(f);
- return NULL;
+ 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;
}
- return f;
+ *result = f;
+ return 1;
}
/**
return &f->meta;
}
-static int find_atom(struct mp4 *f, uint64_t base, uint32_t size,
- const char *name)
-{
- uint32_t remaining = size;
- uint64_t atom_offset = base;
-
- for (;;) {
- int ret;
- char atom_name[4];
- uint32_t atom_size;
-
- set_position(f, atom_offset);
-
- if (remaining < 8)
- return -1;
- ret = read_int32(f, &atom_size);
- if (ret <= 0)
- return ret;
- if (atom_size > remaining || atom_size < 8)
- return -1;
- ret = read_data(f, atom_name, 4);
- if (ret <= 0)
- return ret;
- if (!memcmp(atom_name, name, 4)) {
- set_position(f, atom_offset);
- return 1;
- }
- remaining -= atom_size;
- atom_offset += atom_size;
- }
-}
-
-/*
- * Try to find atom <name> with atom <name_inside> in it. Besides -1/0/1 for
- * error, EOF and success, this function may return 2 to indicate that the
- * desired atoms were not found.
- */
-static int find_atom_v2(struct mp4 *f, uint64_t base, uint32_t size,
- const char *name, uint32_t extraheaders, const char *name_inside)
-{
- uint64_t first_base = (uint64_t) (-1);
-
- for (;;) {
- uint64_t mybase;
- uint32_t mysize;
- int ret = find_atom(f, base, size, name);
-
- if (ret <= 0)
- return ret;
- mybase = get_position(f);
- ret = read_int32(f, &mysize);
- if (ret <= 0)
- return ret;
- 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)) {
- set_position(f, mybase);
- return 1;
- }
- base += mysize;
- if (size <= mysize)
- break;
- size -= mysize;
- }
- if (first_base != (uint64_t)(-1)) {
- set_position(f, first_base);
- return 1;
- }
- /* wanted atom inside not found */
- return 2;
-}
-
struct membuffer {
void *data;
unsigned written;
membuffer_write(buf, temp, 4);
}
-static void membuffer_write_std_tag(struct membuffer *buf, const char *name,
- const char *value)
-{
- uint32_t len = strlen(value);
- membuffer_write_int32(buf, 8 /* atom header */
- + 8 /* data atom header */
- + 8 /* flags + reserved */
- + len);
- membuffer_write_atom_name(buf, name);
- membuffer_write_int32(buf, 8 /* data atom header */
- + 8 /* flags + reserved */
- + len);
- membuffer_write_atom_name(buf, "data");
- membuffer_write_int32(buf, 1); /* flags */
- membuffer_write_int32(buf, 0); /* reserved */
- membuffer_write(buf, value, len);
-}
-
static unsigned membuffer_get_size(const struct membuffer *buf)
{
return buf->written;
return ret;
}
-struct stdmeta_entry {
- const char *atom;
- const char *name;
-};
-
-static const char *find_standard_meta(const char *name)
-{
- const struct stdmeta_entry stdmetas[] = {
- {"\xA9" "nam", "title"},
- {"\xA9" "ART", "artist"},
- {"\xA9" "alb", "album"},
- {"\xA9" "day", "date"},
- {"\xA9" "cmt", "comment"},
- };
-
- for (unsigned n = 0; n < ARRAY_SIZE(stdmetas); n++)
- if (!strcasecmp(name, stdmetas[n].name))
- return stdmetas[n].atom;
- return NULL;
-}
-
-static uint32_t create_ilst(const struct mp4_metadata *meta, void **out_buffer,
- uint32_t * out_size)
+static void *create_ilst(const struct mp4_metadata *meta, uint32_t *out_size)
{
struct membuffer *buf = membuffer_create();
- unsigned metaptr;
-
- for (metaptr = 0; metaptr < meta->count; metaptr++) {
- struct mp4_tag *tag = meta->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);
+ unsigned n;
+
+ for (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
- PARA_ERROR_LOG("invalid tag item: %s\n", tag->item);
- }
- *out_size = membuffer_get_size(buf);
- *out_buffer = membuffer_detach(buf);
- return 1;
-}
-
-static void membuffer_write_atom(struct 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 struct membuffer *buf)
-{
- return buf->data;
-}
-
-static bool membuffer_transfer_from_file(struct membuffer *buf, struct mp4 *src,
- unsigned bytes)
-{
- unsigned oldsize = membuffer_get_size(buf);
- char *bufptr;
-
- membuffer_write(buf, 0, bytes);
- bufptr = membuffer_get_ptr(buf);
- if (read_data(src, bufptr + oldsize, bytes) != 1) {
- free(buf->data);
- free(buf);
- return false;
+ assert(false);
+ membuffer_write_int32(buf, 8 /* atom header */
+ + 8 /* data atom header */
+ + 8 /* flags + reserved */
+ + len);
+ membuffer_write_atom_name(buf, atom_name);
+ membuffer_write_int32(buf, 8 /* data atom header */
+ + 8 /* flags + reserved */
+ + len);
+ membuffer_write_atom_name(buf, "data");
+ membuffer_write_int32(buf, 1); /* flags */
+ membuffer_write_int32(buf, 0); /* reserved */
+ membuffer_write(buf, tag->value, len);
}
- return true;
-}
-
-static uint32_t create_meta(const struct mp4_metadata *meta, void **out_buffer,
- uint32_t * out_size)
-{
- struct membuffer *buf;
- uint32_t ilst_size;
- void *ilst_buffer;
-
- if (!create_ilst(meta, &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);
- return 1;
-}
-
-static uint32_t create_udta(const struct mp4_metadata *meta, void **out_buffer,
-uint32_t * out_size)
-{
- struct membuffer *buf;
- uint32_t meta_size;
- void *meta_buffer;
-
- if (!create_meta(meta, &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);
- return 1;
+ return membuffer_detach(buf);
}
static uint32_t fix_byte_order_32(uint32_t src)
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, *out_buffer;
uint8_t *p_out;
int32_t size_delta;
uint32_t tmp;
- ret = find_atom_v2(f, total_base, total_size, "udta", 0, "meta");
- if (ret <= 0)
- return NULL;
- if (ret == 2) {
- struct membuffer *buf;
- void *new_udta_buffer;
- uint32_t new_udta_size;
- if (!create_udta(&f->meta, &new_udta_buffer, &new_udta_size))
- return NULL;
-
- buf = membuffer_create();
- set_position(f, total_base);
- if (!membuffer_transfer_from_file(buf, f, total_size)) {
- free(new_udta_buffer);
- return NULL;
- }
- membuffer_write_atom(buf, "udta", new_udta_size,
- new_udta_buffer);
-
- free(new_udta_buffer);
-
- *out_size = membuffer_get_size(buf);
- return membuffer_detach(buf);
- }
- udta_offset = get_position(f);
- ret = read_int32(f, &udta_size);
- if (ret <= 0)
- return NULL;
- ret = find_atom_v2(f, udta_offset + 8, udta_size - 8, "meta", 4, "ilst");
- if (ret <= 0)
- return NULL;
- if (ret == 2) {
- struct membuffer *buf;
- void *new_meta_buffer;
- uint32_t new_meta_size;
-
- if (!create_meta(&f->meta, &new_meta_buffer, &new_meta_size))
- return NULL;
-
- buf = membuffer_create();
- set_position(f, total_base);
- if (!membuffer_transfer_from_file(buf, f,
- udta_offset - total_base)) {
- free(new_meta_buffer);
- return NULL;
- }
-
- membuffer_write_int32(buf, udta_size + 8 + new_meta_size);
- membuffer_write_atom_name(buf, "udta");
- if (!membuffer_transfer_from_file(buf, f, udta_size)) {
- free(new_meta_buffer);
- return NULL;
- }
- membuffer_write_atom(buf, "meta", new_meta_size,
- new_meta_buffer);
- free(new_meta_buffer);
-
- *out_size = membuffer_get_size(buf);
- return membuffer_detach(buf);
- }
- meta_offset = get_position(f);
- ret = read_int32(f, &meta_size);
- if (ret <= 0)
- return NULL;
- /* shouldn't happen, find_atom_v2 above takes care of it */
- if (!find_atom(f, meta_offset + 12, meta_size - 12, "ilst"))
- return NULL;
- ilst_offset = get_position(f);
- ret = read_int32(f, &ilst_size);
- if (ret <= 0)
- return NULL;
- if (!create_ilst(&f->meta, &new_ilst_buffer, &new_ilst_size))
- return NULL;
- size_delta = new_ilst_size - (ilst_size - 8);
+ new_ilst_buffer = create_ilst(&f->meta, &new_ilst_size);
+ 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, udta_offset - total_base);
+ ret = read_data(f, p_out, f->udta_offset - total_base);
if (ret <= 0)
return NULL;
- p_out += (uint32_t) (udta_offset - total_base);
+ p_out += f->udta_offset - total_base;
ret = read_int32(f, &tmp);
if (ret <= 0)
return NULL;
if (ret <= 0)
return NULL;
p_out += 4;
- ret = read_data(f, p_out, meta_offset - udta_offset - 8);
+ ret = read_data(f, p_out, f->meta_offset - f->udta_offset - 8);
if (ret <= 0)
return NULL;
- p_out += (uint32_t) (meta_offset - udta_offset - 8);
+ p_out += f->meta_offset - f->udta_offset - 8;
ret = read_int32(f, &tmp);
if (ret <= 0)
return NULL;
if (ret <= 0)
return NULL;
p_out += 4;
- ret = read_data(f, p_out, ilst_offset - meta_offset - 8);
+ ret = read_data(f, p_out, f->ilst_offset - f->meta_offset - 8);
if (ret <= 0)
return NULL;
- p_out += (uint32_t) (ilst_offset - meta_offset - 8);
+ p_out += f->ilst_offset - f->meta_offset - 8;
ret = read_int32(f, &tmp);
if (ret <= 0)
return NULL;
p_out += 4;
memcpy(p_out, new_ilst_buffer, new_ilst_size);
p_out += new_ilst_size;
- set_position(f, ilst_offset + ilst_size);
- ret = read_data(f, p_out, total_size
- - (ilst_offset - total_base) - 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;
free(new_ilst_buffer);