FFmpeg
apetag.c
Go to the documentation of this file.
1 /*
2  * APE tag handling
3  * Copyright (c) 2007 Benjamin Zores <ben@geexbox.org>
4  * based upon libdemac from Dave Chapman.
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg 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.
12  *
13  * FFmpeg 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.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #include <inttypes.h>
24 
25 #include "libavutil/dict.h"
26 #include "libavutil/mem.h"
27 #include "avformat.h"
28 #include "avio_internal.h"
29 #include "apetag.h"
30 #include "demux.h"
31 #include "internal.h"
32 #include "mux.h"
33 
34 #define APE_TAG_FLAG_CONTAINS_HEADER (1U << 31)
35 #define APE_TAG_FLAG_LACKS_FOOTER (1 << 30)
36 #define APE_TAG_FLAG_IS_HEADER (1 << 29)
37 #define APE_TAG_FLAG_IS_BINARY (1 << 1)
38 
39 static int ape_tag_read_field(AVFormatContext *s)
40 {
41  AVIOContext *pb = s->pb;
42  uint8_t key[1024], *value;
43  int64_t size, flags;
44  int i, c;
45 
46  size = avio_rl32(pb); /* field size */
47  flags = avio_rl32(pb); /* field flags */
48  for (i = 0; i < sizeof(key) - 1; i++) {
49  c = avio_r8(pb);
50  if (c < 0x20 || c > 0x7E)
51  break;
52  else
53  key[i] = c;
54  }
55  key[i] = 0;
56  if (c != 0) {
57  av_log(s, AV_LOG_WARNING, "Invalid APE tag key '%s'.\n", key);
58  return -1;
59  }
60  if (size > INT32_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
61  av_log(s, AV_LOG_ERROR, "APE tag size too large.\n");
62  return AVERROR_INVALIDDATA;
63  }
64  if (flags & APE_TAG_FLAG_IS_BINARY) {
65  uint8_t filename[1024];
66  enum AVCodecID id;
67  int ret;
68  AVStream *st = avformat_new_stream(s, NULL);
69  if (!st)
70  return AVERROR(ENOMEM);
71 
72  ret = avio_get_str(pb, size, filename, sizeof(filename));
73  if (ret < 0)
74  return ret;
75  if (size <= ret) {
76  av_log(s, AV_LOG_WARNING, "Skipping binary tag '%s'.\n", key);
77  return 0;
78  }
79  size -= ret;
80 
81  av_dict_set(&st->metadata, key, filename, 0);
82 
83  if ((id = ff_guess_image2_codec(filename)) != AV_CODEC_ID_NONE) {
84  int ret = ff_add_attached_pic(s, st, s->pb, NULL, size);
85  if (ret < 0) {
86  av_log(s, AV_LOG_ERROR, "Error reading cover art.\n");
87  return ret;
88  }
89  st->codecpar->codec_id = id;
90  } else {
91  if ((ret = ff_get_extradata(s, st->codecpar, s->pb, size)) < 0)
92  return ret;
93  st->codecpar->codec_type = AVMEDIA_TYPE_ATTACHMENT;
94  }
95  } else {
96  value = av_malloc(size+1);
97  if (!value)
98  return AVERROR(ENOMEM);
99  c = avio_read(pb, value, size);
100  if (c < 0) {
101  av_free(value);
102  return c;
103  }
104  value[c] = 0;
105  av_dict_set(&s->metadata, key, value, AV_DICT_DONT_STRDUP_VAL);
106  }
107  return 0;
108 }
109 
110 int64_t ff_ape_parse_tag(AVFormatContext *s)
111 {
112  AVIOContext *pb = s->pb;
113  int64_t file_size = avio_size(pb);
114  uint32_t val, fields, tag_bytes;
115  uint8_t buf[8];
116  int64_t tag_start;
117  int i;
118 
119  if (file_size < APE_TAG_FOOTER_BYTES)
120  return 0;
121 
122  avio_seek(pb, file_size - APE_TAG_FOOTER_BYTES, SEEK_SET);
123 
124  if(avio_read(pb, buf, 8) != 8) /* APETAGEX */
125  return 0;
126  if (strncmp(buf, APE_TAG_PREAMBLE, 8)) {
127  return 0;
128  }
129 
130  val = avio_rl32(pb); /* APE tag version */
131  if (val > APE_TAG_VERSION) {
132  av_log(s, AV_LOG_ERROR, "Unsupported tag version. (>=%d)\n", APE_TAG_VERSION);
133  return 0;
134  }
135 
136  tag_bytes = avio_rl32(pb); /* tag size */
137  if (tag_bytes - APE_TAG_FOOTER_BYTES > (1024 * 1024 * 16)) {
138  av_log(s, AV_LOG_ERROR, "Tag size is way too big\n");
139  return 0;
140  }
141 
142  if (tag_bytes > file_size - APE_TAG_FOOTER_BYTES) {
143  av_log(s, AV_LOG_ERROR, "Invalid tag size %"PRIu32".\n", tag_bytes);
144  return 0;
145  }
146 
147  fields = avio_rl32(pb); /* number of fields */
148  if (fields > 65536) {
149  av_log(s, AV_LOG_ERROR, "Too many tag fields (%"PRIu32")\n", fields);
150  return 0;
151  }
152 
153  val = avio_rl32(pb); /* flags */
154  if (val & APE_TAG_FLAG_IS_HEADER) {
155  av_log(s, AV_LOG_ERROR, "APE Tag is a header\n");
156  return 0;
157  }
158 
159  avio_seek(pb, file_size - tag_bytes, SEEK_SET);
160 
161  if (val & APE_TAG_FLAG_CONTAINS_HEADER)
162  tag_bytes += APE_TAG_HEADER_BYTES;
163 
164  tag_start = file_size - tag_bytes;
165 
166  for (i=0; i<fields; i++)
167  if (ape_tag_read_field(s) < 0) break;
168 
169  return tag_start;
170 }
171 
172 static int string_is_ascii(const uint8_t *str)
173 {
174  while (*str && *str >= 0x20 && *str <= 0x7e ) str++;
175  return !*str;
176 }
177 
178 int ff_ape_write_tag(AVFormatContext *s)
179 {
180  const AVDictionaryEntry *e = NULL;
181  int size, ret, count = 0;
182  AVIOContext *dyn_bc;
183  uint8_t *dyn_buf;
184 
185  if ((ret = avio_open_dyn_buf(&dyn_bc)) < 0)
186  return ret;
187 
188  ff_standardize_creation_time(s);
189  while ((e = av_dict_iterate(s->metadata, e))) {
190  int val_len;
191 
192  if (!string_is_ascii(e->key)) {
193  av_log(s, AV_LOG_WARNING, "Non ASCII keys are not allowed\n");
194  continue;
195  }
196 
197  val_len = strlen(e->value);
198  avio_wl32(dyn_bc, val_len); // value length
199  avio_wl32(dyn_bc, 0); // item flags
200  avio_put_str(dyn_bc, e->key); // key
201  avio_write(dyn_bc, e->value, val_len); // value
202  count++;
203  }
204  if (!count)
205  goto end;
206 
207  size = avio_get_dyn_buf(dyn_bc, &dyn_buf);
208  if (size <= 0)
209  goto end;
210  size += APE_TAG_FOOTER_BYTES;
211 
212  // header
213  avio_write(s->pb, "APETAGEX", 8); // id
214  avio_wl32(s->pb, APE_TAG_VERSION); // version
215  avio_wl32(s->pb, size);
216  avio_wl32(s->pb, count);
217 
218  // flags
219  avio_wl32(s->pb, APE_TAG_FLAG_CONTAINS_HEADER | APE_TAG_FLAG_IS_HEADER);
220  ffio_fill(s->pb, 0, 8); // reserved
221 
222  avio_write(s->pb, dyn_buf, size - APE_TAG_FOOTER_BYTES);
223 
224  // footer
225  avio_write(s->pb, "APETAGEX", 8); // id
226  avio_wl32(s->pb, APE_TAG_VERSION); // version
227  avio_wl32(s->pb, size); // size
228  avio_wl32(s->pb, count); // tag count
229 
230  // flags
231  avio_wl32(s->pb, APE_TAG_FLAG_CONTAINS_HEADER);
232  ffio_fill(s->pb, 0, 8); // reserved
233 
234 end:
235  ffio_free_dyn_buf(&dyn_bc);
236 
237  return ret;
238 }
AV_LOG_WARNING
#define AV_LOG_WARNING
Something somehow does not look correct.
Definition: log.h:186
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
AVCodecParameters::codec_type
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:51
avformat_new_stream
AVStream * avformat_new_stream(AVFormatContext *s, const struct AVCodec *c)
Add a new stream to a media file.
int64_t
long long int64_t
Definition: coverity.c:34
apetag.h
ff_guess_image2_codec
enum AVCodecID ff_guess_image2_codec(const char *filename)
Definition: img2.c:125
APE_TAG_FLAG_IS_BINARY
#define APE_TAG_FLAG_IS_BINARY
Definition: apetag.c:37
avio_size
int64_t avio_size(AVIOContext *s)
Get the filesize.
Definition: aviobuf.c:323
avio_get_dyn_buf
int avio_get_dyn_buf(AVIOContext *s, uint8_t **pbuffer)
Return the written size and a pointer to the buffer.
Definition: aviobuf.c:1374
APE_TAG_PREAMBLE
#define APE_TAG_PREAMBLE
Definition: apetag.h:28
ff_ape_parse_tag
int64_t ff_ape_parse_tag(AVFormatContext *s)
Read and parse an APE tag.
Definition: apetag.c:110
APE_TAG_VERSION
#define APE_TAG_VERSION
Definition: apetag.h:29
av_malloc
#define av_malloc(s)
Definition: tableprint_vlc.h:30
ff_get_extradata
int ff_get_extradata(void *logctx, AVCodecParameters *par, AVIOContext *pb, int size)
Allocate extradata with additional AV_INPUT_BUFFER_PADDING_SIZE at end which is always set to 0 and f...
Definition: demux_utils.c:335
string_is_ascii
static int string_is_ascii(const uint8_t *str)
Definition: apetag.c:172
ape_tag_read_field
static int ape_tag_read_field(AVFormatContext *s)
Definition: apetag.c:39
val
static double val(void *priv, double ch)
Definition: aeval.c:77
APE_TAG_FLAG_CONTAINS_HEADER
#define APE_TAG_FLAG_CONTAINS_HEADER
Definition: apetag.c:34
AV_DICT_DONT_STRDUP_VAL
#define AV_DICT_DONT_STRDUP_VAL
Take ownership of a value that's been allocated with av_malloc() or another memory allocation functio...
Definition: dict.h:79
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
avio_open_dyn_buf
int avio_open_dyn_buf(AVIOContext **s)
Open a write only memory stream.
Definition: aviobuf.c:1362
ff_add_attached_pic
int ff_add_attached_pic(AVFormatContext *s, AVStream *st, AVIOContext *pb, AVBufferRef **buf, int size)
Add an attached pic to an AVStream.
Definition: demux_utils.c:116
s
#define s(width, name)
Definition: cbs_vp9.c:198
AVDictionaryEntry::key
char * key
Definition: dict.h:90
key
const char * key
Definition: hwcontext_opencl.c:189
ff_ape_write_tag
int ff_ape_write_tag(AVFormatContext *s)
Write an APE tag into a file.
Definition: apetag.c:178
fields
the definition of that something depends on the semantic of the filter The callback must examine the status of the filter s links and proceed accordingly The status of output links is stored in the status_in and status_out fields and tested by the then the processing requires a frame on this link and the filter is expected to make efforts in that direction The status of input links is stored by the fifo and status_out fields
Definition: filter_design.txt:155
APE_TAG_FLAG_IS_HEADER
#define APE_TAG_FLAG_IS_HEADER
Definition: apetag.c:36
AVFormatContext
Format I/O context.
Definition: avformat.h:1260
internal.h
AVStream::codecpar
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:771
NULL
#define NULL
Definition: coverity.c:32
AVStream::metadata
AVDictionary * metadata
Definition: avformat.h:828
ffio_fill
void ffio_fill(AVIOContext *s, int b, int64_t count)
Definition: aviobuf.c:187
c
Undefined Behavior In the C some operations are like signed integer dereferencing freed accessing outside allocated Undefined Behavior must not occur in a C it is not safe even if the output of undefined operations is unused The unsafety may seem nit picking but Optimizing compilers have in fact optimized code on the assumption that no undefined Behavior occurs Optimizing code based on wrong assumptions can and has in some cases lead to effects beyond the output of computations The signed integer overflow problem in speed critical code Code which is highly optimized and works with signed integers sometimes has the problem that often the output of the computation does not c
Definition: undefined.txt:32
AVCodecID
AVCodecID
Identify the syntax and semantics of the bitstream.
Definition: codec_id.h:49
avio_rl32
unsigned int avio_rl32(AVIOContext *s)
Definition: aviobuf.c:730
AVIOContext
Bytestream IO Context.
Definition: avio.h:160
ff_standardize_creation_time
int ff_standardize_creation_time(AVFormatContext *s)
Standardize creation_time metadata in AVFormatContext to an ISO-8601 timestamp string.
Definition: mux_utils.c:155
avio_get_str
int avio_get_str(AVIOContext *pb, int maxlen, char *buf, int buflen)
Read a string from pb into buf.
Definition: aviobuf.c:866
size
int size
Definition: twinvq_data.h:10344
avio_write
void avio_write(AVIOContext *s, const unsigned char *buf, int size)
Definition: aviobuf.c:201
avio_r8
int avio_r8(AVIOContext *s)
Definition: aviobuf.c:603
avio_wl32
void avio_wl32(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:357
AV_CODEC_ID_NONE
@ AV_CODEC_ID_NONE
Definition: codec_id.h:50
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
avio_internal.h
value
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default value
Definition: writing_filters.txt:86
AVMEDIA_TYPE_ATTACHMENT
@ AVMEDIA_TYPE_ATTACHMENT
Opaque data information usually sparse.
Definition: avutil.h:205
demux.h
ffio_free_dyn_buf
void ffio_free_dyn_buf(AVIOContext **s)
Free a dynamic buffer.
Definition: aviobuf.c:1435
ret
ret
Definition: filter_design.txt:187
AVStream
Stream structure.
Definition: avformat.h:748
APE_TAG_HEADER_BYTES
#define APE_TAG_HEADER_BYTES
Definition: apetag.h:31
avio_seek
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:231
avformat.h
dict.h
AV_INPUT_BUFFER_PADDING_SIZE
#define AV_INPUT_BUFFER_PADDING_SIZE
Definition: defs.h:40
id
enum AVCodecID id
Definition: dts2pts.c:365
avio_read
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:612
APE_TAG_FOOTER_BYTES
#define APE_TAG_FOOTER_BYTES
Definition: apetag.h:30
mem.h
av_free
#define av_free(p)
Definition: tableprint_vlc.h:33
AVDictionaryEntry
Definition: dict.h:89
AVCodecParameters::codec_id
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:55
av_dict_set
int av_dict_set(AVDictionary **pm, const char *key, const char *value, int flags)
Set the given entry in *pm, overwriting an existing entry.
Definition: dict.c:88
flags
#define flags(name, subs,...)
Definition: cbs_av1.c:482
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
AVDictionaryEntry::value
char * value
Definition: dict.h:91
avio_put_str
int avio_put_str(AVIOContext *s, const char *str)
Write a NULL-terminated string.
Definition: aviobuf.c:373
av_dict_iterate
const AVDictionaryEntry * av_dict_iterate(const AVDictionary *m, const AVDictionaryEntry *prev)
Iterate over a dictionary.
Definition: dict.c:44
mux.h
Generated on Sat Aug 31 2024 19:22:47 for FFmpeg by   doxygen 1.8.17