3 * Copyright (c) 2007 Benjamin Zores <ben@geexbox.org>
4 * based upon libdemac from Dave Chapman.
6 * This file is part of Libav.
8 * Libav is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * Libav is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with Libav; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 #include "libavutil/intreadwrite.h"
24 #include "libavutil/dict.h"
28 #define APE_TAG_VERSION 2000
29 #define APE_TAG_FOOTER_BYTES 32
30 #define APE_TAG_FLAG_CONTAINS_HEADER (1 << 31)
31 #define APE_TAG_FLAG_IS_HEADER (1 << 29)
32 #define APE_TAG_FLAG_IS_BINARY (1 << 1)
34 static int ape_tag_read_field(AVFormatContext *s)
36 AVIOContext *pb = s->pb;
37 uint8_t key[1024], *value;
41 size = avio_rl32(pb); /* field size */
42 flags = avio_rl32(pb); /* field flags */
43 for (i = 0; i < sizeof(key) - 1; i++) {
45 if (c < 0x20 || c > 0x7E)
52 av_log(s, AV_LOG_WARNING, "Invalid APE tag key '%s'.\n", key);
57 if (flags & APE_TAG_FLAG_IS_BINARY) {
58 uint8_t filename[1024];
59 AVStream *st = avformat_new_stream(s, NULL);
61 return AVERROR(ENOMEM);
62 avio_get_str(pb, INT_MAX, filename, sizeof(filename));
63 st->codec->extradata = av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE);
64 if (!st->codec->extradata)
65 return AVERROR(ENOMEM);
66 if (avio_read(pb, st->codec->extradata, size) != size) {
67 av_freep(&st->codec->extradata);
70 st->codec->extradata_size = size;
71 av_dict_set(&st->metadata, key, filename, 0);
72 st->codec->codec_type = AVMEDIA_TYPE_ATTACHMENT;
74 value = av_malloc(size+1);
76 return AVERROR(ENOMEM);
77 c = avio_read(pb, value, size);
83 av_dict_set(&s->metadata, key, value, AV_DICT_DONT_STRDUP_VAL);
88 void ff_ape_parse_tag(AVFormatContext *s)
90 AVIOContext *pb = s->pb;
91 int file_size = avio_size(pb);
92 uint32_t val, fields, tag_bytes;
96 if (file_size < APE_TAG_FOOTER_BYTES)
99 avio_seek(pb, file_size - APE_TAG_FOOTER_BYTES, SEEK_SET);
101 avio_read(pb, buf, 8); /* APETAGEX */
102 if (strncmp(buf, "APETAGEX", 8)) {
106 val = avio_rl32(pb); /* APE tag version */
107 if (val > APE_TAG_VERSION) {
108 av_log(s, AV_LOG_ERROR, "Unsupported tag version. (>=%d)\n", APE_TAG_VERSION);
112 tag_bytes = avio_rl32(pb); /* tag size */
113 if (tag_bytes - APE_TAG_FOOTER_BYTES > (1024 * 1024 * 16)) {
114 av_log(s, AV_LOG_ERROR, "Tag size is way too big\n");
118 fields = avio_rl32(pb); /* number of fields */
119 if (fields > 65536) {
120 av_log(s, AV_LOG_ERROR, "Too many tag fields (%d)\n", fields);
124 val = avio_rl32(pb); /* flags */
125 if (val & APE_TAG_FLAG_IS_HEADER) {
126 av_log(s, AV_LOG_ERROR, "APE Tag is a header\n");
130 avio_seek(pb, file_size - tag_bytes, SEEK_SET);
132 for (i=0; i<fields; i++)
133 if (ape_tag_read_field(s) < 0) break;