FFmpeg
flac_picture.c
Go to the documentation of this file.
1 /*
2  * Raw FLAC picture parser
3  * Copyright (c) 2001 Fabrice Bellard
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "libavutil/intreadwrite.h"
23 #include "libavcodec/bytestream.h"
24 #include "libavcodec/png.h"
25 #include "avformat.h"
26 #include "flac_picture.h"
27 #include "id3v2.h"
28 #include "internal.h"
29 
31 {
32  const CodecMime *mime = ff_id3v2_mime_tags;
33  enum AVCodecID id = AV_CODEC_ID_NONE;
35  uint8_t mimetype[64], *desc = NULL;
37  AVStream *st;
38  int width, height, ret = 0;
39  unsigned int len, type;
40 
41  if (buf_size < 34) {
42  av_log(s, AV_LOG_ERROR, "Attached picture metadata block too short\n");
44  return AVERROR_INVALIDDATA;
45  return 0;
46  }
47 
48  bytestream2_init(&g, buf, buf_size);
49 
50  /* read the picture type */
51  type = bytestream2_get_be32u(&g);
53  av_log(s, AV_LOG_ERROR, "Invalid picture type: %d.\n", type);
55  return AVERROR_INVALIDDATA;
56  }
57  type = 0;
58  }
59 
60  /* picture mimetype */
61  len = bytestream2_get_be32u(&g);
62  if (len <= 0 || len >= sizeof(mimetype)) {
63  av_log(s, AV_LOG_ERROR, "Could not read mimetype from an attached "
64  "picture.\n");
66  return AVERROR_INVALIDDATA;
67  return 0;
68  }
69  if (len + 24 > bytestream2_get_bytes_left(&g)) {
70  av_log(s, AV_LOG_ERROR, "Attached picture metadata block too short\n");
72  return AVERROR_INVALIDDATA;
73  return 0;
74  }
75  bytestream2_get_bufferu(&g, mimetype, len);
76  mimetype[len] = 0;
77 
78  while (mime->id != AV_CODEC_ID_NONE) {
79  if (!strncmp(mime->str, mimetype, sizeof(mimetype))) {
80  id = mime->id;
81  break;
82  }
83  mime++;
84  }
85  if (id == AV_CODEC_ID_NONE) {
86  av_log(s, AV_LOG_ERROR, "Unknown attached picture mimetype: %s.\n",
87  mimetype);
89  return AVERROR_INVALIDDATA;
90  return 0;
91  }
92 
93  /* picture description */
94  len = bytestream2_get_be32u(&g);
95  if (len > bytestream2_get_bytes_left(&g) - 20) {
96  av_log(s, AV_LOG_ERROR, "Attached picture metadata block too short\n");
98  return AVERROR_INVALIDDATA;
99  return 0;
100  }
101  if (len > 0) {
102  if (!(desc = av_malloc(len + 1))) {
103  return AVERROR(ENOMEM);
104  }
105 
106  bytestream2_get_bufferu(&g, desc, len);
107  desc[len] = 0;
108  }
109 
110  /* picture metadata */
111  width = bytestream2_get_be32u(&g);
112  height = bytestream2_get_be32u(&g);
113  bytestream2_skipu(&g, 8);
114 
115  /* picture data */
116  len = bytestream2_get_be32u(&g);
117  if (len <= 0 || len > bytestream2_get_bytes_left(&g)) {
118  av_log(s, AV_LOG_ERROR, "Attached picture metadata block too short\n");
120  ret = AVERROR_INVALIDDATA;
121  goto fail;
122  }
123  if (!(data = av_buffer_alloc(len + AV_INPUT_BUFFER_PADDING_SIZE))) {
124  RETURN_ERROR(AVERROR(ENOMEM));
125  }
126  bytestream2_get_bufferu(&g, data->data, len);
127  memset(data->data + len, 0, AV_INPUT_BUFFER_PADDING_SIZE);
128 
129  if (AV_RB64(data->data) == PNGSIG)
130  id = AV_CODEC_ID_PNG;
131 
132  st = avformat_new_stream(s, NULL);
133  if (!st) {
134  RETURN_ERROR(AVERROR(ENOMEM));
135  }
136 
138  st->attached_pic.buf = data;
139  st->attached_pic.data = data->data;
140  st->attached_pic.size = len;
141  st->attached_pic.stream_index = st->index;
143 
146  st->codecpar->codec_id = id;
147  st->codecpar->width = width;
148  st->codecpar->height = height;
149  av_dict_set(&st->metadata, "comment", ff_id3v2_picture_types[type], 0);
150  if (desc)
151  av_dict_set(&st->metadata, "title", desc, AV_DICT_DONT_STRDUP_VAL);
152 
153  return 0;
154 
155 fail:
156  av_buffer_unref(&data);
157  av_freep(&desc);
158 
159  return ret;
160 }
#define NULL
Definition: coverity.c:32
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
void av_buffer_unref(AVBufferRef **buf)
Free a given reference and automatically free the buffer if there are no more references to it...
Definition: buffer.c:126
ptrdiff_t const GLvoid * data
Definition: opengl_enc.c:100
const char * g
Definition: vf_curves.c:115
const char * desc
Definition: nvenc.c:68
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:4036
int index
stream index in AVFormatContext
Definition: avformat.h:877
int size
Definition: avcodec.h:1534
static av_always_inline void bytestream2_init(GetByteContext *g, const uint8_t *buf, int buf_size)
Definition: bytestream.h:133
GLint GLenum type
Definition: opengl_enc.c:104
static av_always_inline unsigned int bytestream2_get_bufferu(GetByteContext *g, uint8_t *dst, unsigned int size)
Definition: bytestream.h:273
Format I/O context.
Definition: avformat.h:1353
char str[32]
Definition: internal.h:48
uint8_t
#define av_malloc(s)
int width
Video only.
Definition: avcodec.h:4102
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:4524
#define height
uint8_t * data
Definition: avcodec.h:1533
static av_always_inline void bytestream2_skipu(GetByteContext *g, unsigned int size)
Definition: bytestream.h:170
enum AVCodecID id
Definition: internal.h:49
#define av_log(a,...)
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: avcodec.h:1565
static av_always_inline int bytestream2_get_bytes_left(GetByteContext *g)
Definition: bytestream.h:154
AVCodecID
Identify the syntax and semantics of the bitstream.
Definition: avcodec.h:217
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
enum AVMediaType codec_type
General type of the encoded data.
Definition: avcodec.h:4032
AVBufferRef * buf
A reference to the reference-counted buffer where the packet data is stored.
Definition: avcodec.h:1516
#define PNGSIG
Definition: png.h:47
#define fail()
Definition: checkasm.h:122
int flags
A combination of AV_PKT_FLAG values.
Definition: avcodec.h:1539
const CodecMime ff_id3v2_mime_tags[]
Definition: id3v2.c:131
#define width
#define AV_DICT_DONT_STRDUP_VAL
Take ownership of a value that&#39;s been allocated with av_malloc() or another memory allocation functio...
Definition: dict.h:76
#define s(width, name)
Definition: cbs_vp9.c:257
int ff_flac_parse_picture(AVFormatContext *s, uint8_t *buf, int buf_size)
Definition: flac_picture.c:30
#define AV_EF_EXPLODE
abort decoding on minor error detection
Definition: avcodec.h:2761
AVDictionary * metadata
Definition: avformat.h:940
#define FF_ARRAY_ELEMS(a)
#define AV_DISPOSITION_ATTACHED_PIC
The stream is stored in the file as an attached picture/"cover art" (e.g.
Definition: avformat.h:844
Stream structure.
Definition: avformat.h:876
AVBufferRef * av_buffer_alloc(int size)
Allocate an AVBuffer of the given size using av_malloc().
Definition: buffer.c:68
uint8_t * data
The data buffer.
Definition: buffer.h:89
void * buf
Definition: avisynth_c.h:766
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:70
#define RETURN_ERROR(code)
Definition: flac_picture.h:27
A reference to a data buffer.
Definition: buffer.h:81
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_WL16 uint64_t_TMPL AV_RB64
Definition: bytestream.h:87
Main libavformat public API header.
int error_recognition
Error recognition; higher values will detect more errors but may misdetect some more or less valid pa...
Definition: avformat.h:1620
int disposition
AV_DISPOSITION_* bit field.
Definition: avformat.h:929
void av_init_packet(AVPacket *pkt)
Initialize optional fields of a packet with default values.
Definition: avpacket.c:33
#define AV_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding...
Definition: avcodec.h:808
int len
#define av_freep(p)
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1023
const char *const ff_id3v2_picture_types[21]
Definition: id3v2.c:107
int stream_index
Definition: avcodec.h:1535
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
enum AVCodecID id
AVPacket attached_pic
For streams with AV_DISPOSITION_ATTACHED_PIC disposition, this packet will contain the attached pictu...
Definition: avformat.h:958