avcodec/cbs_h264_syntax_template: fix off by 1 error with slice_group_change_cycle
[ffmpeg.git] / libavformat / flac_picture.c
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
30 int ff_flac_parse_picture(AVFormatContext *s, uint8_t *buf, int buf_size)
31 {
32     const CodecMime *mime = ff_id3v2_mime_tags;
33     enum AVCodecID id = AV_CODEC_ID_NONE;
34     AVBufferRef *data = NULL;
35     uint8_t mimetype[64], *desc = NULL;
36     GetByteContext g;
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");
43         if (s->error_recognition & AV_EF_EXPLODE)
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);
52     if (type >= FF_ARRAY_ELEMS(ff_id3v2_picture_types)) {
53         av_log(s, AV_LOG_ERROR, "Invalid picture type: %d.\n", type);
54         if (s->error_recognition & AV_EF_EXPLODE) {
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");
65         if (s->error_recognition & AV_EF_EXPLODE)
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");
71         if (s->error_recognition & AV_EF_EXPLODE)
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);
88         if (s->error_recognition & AV_EF_EXPLODE)
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");
97         if (s->error_recognition & AV_EF_EXPLODE)
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");
119         if (s->error_recognition & AV_EF_EXPLODE)
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
137     av_init_packet(&st->attached_pic);
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;
142     st->attached_pic.flags       |= AV_PKT_FLAG_KEY;
143
144     st->disposition      |= AV_DISPOSITION_ATTACHED_PIC;
145     st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
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 }