git.sesse.net Git - ffmpeg/blob - 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 }