894 lines · 31.5 KB · C
RawDownload
/* * Copyright (c) 2008 Jaikrishnan Menon <realityman@gmx.net> * Copyright (c) 2010 Peter Ross <pross@xvid.org> * Copyright (c) 2010 Sebastian Vater <cdgs.basty@googlemail.com> * * This file is part of FFmpeg. * * FFmpeg is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * FFmpeg is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with FFmpeg; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ /** * @file * IFF file demuxer * by Jaikrishnan Menon * for more information on the .iff file format, visit: * http://wiki.multimedia.cx/index.php?title=IFF */ #include <inttypes.h> #include "libavutil/avassert.h"#include "libavutil/channel_layout.h"#include "libavutil/intreadwrite.h"#include "libavutil/dict.h"#include "libavcodec/bytestream.h"#include "avformat.h"#include "id3v2.h"#include "internal.h" #define ID_8SVX       MKTAG('8','S','V','X')#define ID_16SV       MKTAG('1','6','S','V')#define ID_MAUD       MKTAG('M','A','U','D')#define ID_MHDR       MKTAG('M','H','D','R')#define ID_MDAT       MKTAG('M','D','A','T')#define ID_VHDR       MKTAG('V','H','D','R')#define ID_ATAK       MKTAG('A','T','A','K')#define ID_RLSE       MKTAG('R','L','S','E')#define ID_CHAN       MKTAG('C','H','A','N')#define ID_PBM        MKTAG('P','B','M',' ')#define ID_ILBM       MKTAG('I','L','B','M')#define ID_BMHD       MKTAG('B','M','H','D')#define ID_DGBL       MKTAG('D','G','B','L')#define ID_CAMG       MKTAG('C','A','M','G')#define ID_CMAP       MKTAG('C','M','A','P')#define ID_ACBM       MKTAG('A','C','B','M')#define ID_DEEP       MKTAG('D','E','E','P')#define ID_RGB8       MKTAG('R','G','B','8')#define ID_RGBN       MKTAG('R','G','B','N')#define ID_DSD        MKTAG('D','S','D',' ')#define ID_DST        MKTAG('D','S','T',' ')#define ID_DSTC       MKTAG('D','S','T','C')#define ID_DSTF       MKTAG('D','S','T','F')#define ID_FRTE       MKTAG('F','R','T','E')#define ID_ANIM       MKTAG('A','N','I','M')#define ID_ANHD       MKTAG('A','N','H','D')#define ID_DLTA       MKTAG('D','L','T','A')#define ID_DPAN       MKTAG('D','P','A','N') #define ID_FORM       MKTAG('F','O','R','M')#define ID_FRM8       MKTAG('F','R','M','8')#define ID_ANNO       MKTAG('A','N','N','O')#define ID_AUTH       MKTAG('A','U','T','H')#define ID_CHRS       MKTAG('C','H','R','S')#define ID_COPYRIGHT  MKTAG('(','c',')',' ')#define ID_CSET       MKTAG('C','S','E','T')#define ID_FVER       MKTAG('F','V','E','R')#define ID_NAME       MKTAG('N','A','M','E')#define ID_TEXT       MKTAG('T','E','X','T')#define ID_ABIT       MKTAG('A','B','I','T')#define ID_BODY       MKTAG('B','O','D','Y')#define ID_DBOD       MKTAG('D','B','O','D')#define ID_DPEL       MKTAG('D','P','E','L')#define ID_DLOC       MKTAG('D','L','O','C')#define ID_TVDC       MKTAG('T','V','D','C') #define LEFT    2#define RIGHT   4#define STEREO  6 /** * This number of bytes if added at the beginning of each AVPacket * which contain additional information about video properties * which has to be shared between demuxer and decoder. * This number may change between frames, e.g. the demuxer might * set it to smallest possible size of 2 to indicate that there's * no extradata changing in this frame. */#define IFF_EXTRA_VIDEO_SIZE 41 typedef enum {    COMP_NONE,    COMP_FIB,    COMP_EXP} svx8_compression_type; typedef struct IffDemuxContext {    int      is_64bit;  ///< chunk size is 64-bit    int64_t  body_pos;    int64_t  body_end;    uint32_t  body_size;    svx8_compression_type   svx8_compression;    unsigned  maud_bits;    unsigned  maud_compression;    unsigned  bitmap_compression;  ///< delta compression method used    unsigned  bpp;          ///< bits per plane to decode (differs from bits_per_coded_sample if HAM)    unsigned  ham;          ///< 0 if non-HAM or number of hold bits (6 for bpp > 6, 4 otherwise)    unsigned  flags;        ///< 1 for EHB, 0 is no extra half darkening    unsigned  transparency; ///< transparency color index in palette    unsigned  masking;      ///< masking method used    uint8_t   tvdc[32];     ///< TVDC lookup table    int64_t   pts;} IffDemuxContext; /* Metadata string read */static int get_metadata(AVFormatContext *s,                        const char *const tag,                        const unsigned data_size){    uint8_t *buf = ((data_size + 1) == 0) ? NULL : av_malloc(data_size + 1);     if (!buf)        return AVERROR(ENOMEM);     if (avio_read(s->pb, buf, data_size) != data_size) {        av_free(buf);        return AVERROR(EIO);    }    buf[data_size] = 0;    av_dict_set(&s->metadata, tag, buf, AV_DICT_DONT_STRDUP_VAL);    return 0;} static int iff_probe(AVProbeData *p){    const uint8_t *d = p->buf;     if ( (AV_RL32(d)   == ID_FORM &&         (AV_RL32(d+8) == ID_8SVX ||          AV_RL32(d+8) == ID_16SV ||          AV_RL32(d+8) == ID_MAUD ||          AV_RL32(d+8) == ID_PBM  ||          AV_RL32(d+8) == ID_ACBM ||          AV_RL32(d+8) == ID_DEEP ||          AV_RL32(d+8) == ID_ILBM ||          AV_RL32(d+8) == ID_RGB8 ||          AV_RL32(d+8) == ID_ANIM ||          AV_RL32(d+8) == ID_RGBN)) ||         (AV_RL32(d) == ID_FRM8 && AV_RL32(d+12) == ID_DSD))        return AVPROBE_SCORE_MAX;    return 0;} static const AVCodecTag dsd_codec_tags[] = {    { AV_CODEC_ID_DSD_MSBF, ID_DSD },    { AV_CODEC_ID_DST,      ID_DST },    { AV_CODEC_ID_NONE, 0 },};  #define DSD_SLFT MKTAG('S','L','F','T')#define DSD_SRGT MKTAG('S','R','G','T')#define DSD_MLFT MKTAG('M','L','F','T')#define DSD_MRGT MKTAG('M','R','G','T')#define DSD_C    MKTAG('C',' ',' ',' ')#define DSD_LS   MKTAG('L','S',' ',' ')#define DSD_RS   MKTAG('R','S',' ',' ')#define DSD_LFE  MKTAG('L','F','E',' ') static const uint32_t dsd_stereo[]  = { DSD_SLFT, DSD_SRGT };static const uint32_t dsd_5point0[] = { DSD_MLFT, DSD_MRGT, DSD_C, DSD_LS, DSD_RS };static const uint32_t dsd_5point1[] = { DSD_MLFT, DSD_MRGT, DSD_C, DSD_LFE, DSD_LS, DSD_RS }; typedef struct {    uint64_t layout;    const uint32_t * dsd_layout;} DSDLayoutDesc; static const DSDLayoutDesc dsd_channel_layout[] = {    { AV_CH_LAYOUT_STEREO,  dsd_stereo },    { AV_CH_LAYOUT_5POINT0, dsd_5point0 },    { AV_CH_LAYOUT_5POINT1, dsd_5point1 },}; static const uint64_t dsd_loudspeaker_config[] = {    AV_CH_LAYOUT_STEREO,    0, 0,    AV_CH_LAYOUT_5POINT0, AV_CH_LAYOUT_5POINT1,}; static const char * dsd_source_comment[] = {    "dsd_source_comment",    "analogue_source_comment",    "pcm_source_comment",}; static const char * dsd_history_comment[] = {    "general_remark",    "operator_name",    "creating_machine",    "timezone",    "file_revision"}; static int parse_dsd_diin(AVFormatContext *s, AVStream *st, uint64_t eof){    AVIOContext *pb = s->pb;     while (avio_tell(pb) + 12 <= eof && !avio_feof(pb)) {        uint32_t tag      = avio_rl32(pb);        uint64_t size     = avio_rb64(pb);        uint64_t orig_pos = avio_tell(pb);        const char * metadata_tag = NULL;         switch(tag) {        case MKTAG('D','I','A','R'): metadata_tag = "artist"; break;        case MKTAG('D','I','T','I'): metadata_tag = "title";  break;        }         if (metadata_tag && size > 4) {            unsigned int tag_size = avio_rb32(pb);            int ret = get_metadata(s, metadata_tag, FFMIN(tag_size, size - 4));            if (ret < 0) {                av_log(s, AV_LOG_ERROR, "cannot allocate metadata tag %s!\n", metadata_tag);                return ret;            }        }         avio_skip(pb, size - (avio_tell(pb) - orig_pos) + (size & 1));    }     return 0;} static int parse_dsd_prop(AVFormatContext *s, AVStream *st, uint64_t eof){    AVIOContext *pb = s->pb;    char abss[24];    int hour, min, sec, i, ret, config;    int dsd_layout[6];    ID3v2ExtraMeta *id3v2_extra_meta;     while (avio_tell(pb) + 12 <= eof && !avio_feof(pb)) {        uint32_t tag      = avio_rl32(pb);        uint64_t size     = avio_rb64(pb);        uint64_t orig_pos = avio_tell(pb);         switch(tag) {        case MKTAG('A','B','S','S'):            if (size < 8)                return AVERROR_INVALIDDATA;            hour = avio_rb16(pb);            min  = avio_r8(pb);            sec  = avio_r8(pb);            snprintf(abss, sizeof(abss), "%02dh:%02dm:%02ds:%d", hour, min, sec, avio_rb32(pb));            av_dict_set(&st->metadata, "absolute_start_time", abss, 0);            break;         case MKTAG('C','H','N','L'):            if (size < 2)                return AVERROR_INVALIDDATA;            st->codecpar->channels       = avio_rb16(pb);            if (size < 2 + st->codecpar->channels * 4)                return AVERROR_INVALIDDATA;            st->codecpar->channel_layout = 0;            if (st->codecpar->channels > FF_ARRAY_ELEMS(dsd_layout)) {                avpriv_request_sample(s, "channel layout");                break;            }            for (i = 0; i < st->codecpar->channels; i++)                dsd_layout[i] = avio_rl32(pb);            for (i = 0; i < FF_ARRAY_ELEMS(dsd_channel_layout); i++) {                const DSDLayoutDesc * d = &dsd_channel_layout[i];                if (av_get_channel_layout_nb_channels(d->layout) == st->codecpar->channels &&                    !memcmp(d->dsd_layout, dsd_layout, st->codecpar->channels * sizeof(uint32_t))) {                    st->codecpar->channel_layout = d->layout;                    break;                }            }            break;         case MKTAG('C','M','P','R'):            if (size < 4)                return AVERROR_INVALIDDATA;            st->codecpar->codec_tag = tag = avio_rl32(pb);            st->codecpar->codec_id = ff_codec_get_id(dsd_codec_tags, tag);            if (!st->codecpar->codec_id) {                av_log(s, AV_LOG_ERROR, "'%s' compression is not supported\n",                       av_fourcc2str(tag));                return AVERROR_PATCHWELCOME;            }            break;         case MKTAG('F','S',' ',' '):            if (size < 4)                return AVERROR_INVALIDDATA;            st->codecpar->sample_rate = avio_rb32(pb) / 8;            break;         case MKTAG('I','D','3',' '):            id3v2_extra_meta = NULL;            ff_id3v2_read(s, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta, size);            if (id3v2_extra_meta) {                if ((ret = ff_id3v2_parse_apic(s, &id3v2_extra_meta)) < 0) {                    ff_id3v2_free_extra_meta(&id3v2_extra_meta);                    return ret;                }                ff_id3v2_free_extra_meta(&id3v2_extra_meta);            }             if (size < avio_tell(pb) - orig_pos) {                av_log(s, AV_LOG_ERROR, "id3 exceeds chunk size\n");                return AVERROR_INVALIDDATA;            }            break;         case MKTAG('L','S','C','O'):            if (size < 2)                return AVERROR_INVALIDDATA;            config = avio_rb16(pb);            if (config != 0xFFFF) {                if (config < FF_ARRAY_ELEMS(dsd_loudspeaker_config))                    st->codecpar->channel_layout = dsd_loudspeaker_config[config];                if (!st->codecpar->channel_layout)                    avpriv_request_sample(s, "loudspeaker configuration %d", config);            }            break;        }         avio_skip(pb, size - (avio_tell(pb) - orig_pos) + (size & 1));    }     return 0;} static int read_dst_frame(AVFormatContext *s, AVPacket *pkt){    IffDemuxContext *iff = s->priv_data;    AVIOContext *pb = s->pb;    uint32_t chunk_id;    uint64_t chunk_pos, data_pos, data_size;    int ret = AVERROR_EOF;     while (!avio_feof(pb)) {        chunk_pos = avio_tell(pb);        if (chunk_pos >= iff->body_end)            return AVERROR_EOF;         chunk_id = avio_rl32(pb);        data_size = iff->is_64bit ? avio_rb64(pb) : avio_rb32(pb);        data_pos = avio_tell(pb);         if (data_size < 1)            return AVERROR_INVALIDDATA;         switch (chunk_id) {        case ID_DSTF:            if (!pkt) {                iff->body_pos  = avio_tell(pb) - (iff->is_64bit ? 12 : 8);                iff->body_size = iff->body_end - iff->body_pos;                return 0;            }            ret = av_get_packet(pb, pkt, data_size);            if (ret < 0)                return ret;            if (data_size & 1)                avio_skip(pb, 1);            pkt->flags |= AV_PKT_FLAG_KEY;            pkt->stream_index = 0;            pkt->duration = 588 * s->streams[0]->codecpar->sample_rate / 44100;            pkt->pos = chunk_pos;             chunk_pos = avio_tell(pb);            if (chunk_pos >= iff->body_end)                return 0;             avio_seek(pb, chunk_pos, SEEK_SET);            return 0;         case ID_FRTE:            if (data_size < 4)                return AVERROR_INVALIDDATA;            s->streams[0]->duration = avio_rb32(pb) * 588LL * s->streams[0]->codecpar->sample_rate / 44100;            break;        }         avio_skip(pb, data_size - (avio_tell(pb) - data_pos) + (data_size & 1));    }     return ret;} static const uint8_t deep_rgb24[] = {0, 0, 0, 3, 0, 1, 0, 8, 0, 2, 0, 8, 0, 3, 0, 8};static const uint8_t deep_rgba[]  = {0, 0, 0, 4, 0, 1, 0, 8, 0, 2, 0, 8, 0, 3, 0, 8};static const uint8_t deep_bgra[]  = {0, 0, 0, 4, 0, 3, 0, 8, 0, 2, 0, 8, 0, 1, 0, 8};static const uint8_t deep_argb[]  = {0, 0, 0, 4, 0,17, 0, 8, 0, 1, 0, 8, 0, 2, 0, 8};static const uint8_t deep_abgr[]  = {0, 0, 0, 4, 0,17, 0, 8, 0, 3, 0, 8, 0, 2, 0, 8}; static int iff_read_header(AVFormatContext *s){    IffDemuxContext *iff = s->priv_data;    AVIOContext *pb = s->pb;    AVStream *st;    uint8_t *buf;    uint32_t chunk_id;    uint64_t data_size;    uint32_t screenmode = 0, num, den;    unsigned transparency = 0;    unsigned masking = 0; // no mask    uint8_t fmt[16];    int fmt_size;     st = avformat_new_stream(s, NULL);    if (!st)        return AVERROR(ENOMEM);     st->codecpar->channels = 1;    st->codecpar->channel_layout = AV_CH_LAYOUT_MONO;    iff->is_64bit = avio_rl32(pb) == ID_FRM8;    avio_skip(pb, iff->is_64bit ? 8 : 4);    // codec_tag used by ByteRun1 decoder to distinguish progressive (PBM) and interlaced (ILBM) content    st->codecpar->codec_tag = avio_rl32(pb);    if (st->codecpar->codec_tag == ID_ANIM) {        avio_skip(pb, 12);    }    iff->bitmap_compression = -1;    iff->svx8_compression = -1;    iff->maud_bits = -1;    iff->maud_compression = -1;     while(!avio_feof(pb)) {        uint64_t orig_pos;        int res;        const char *metadata_tag = NULL;        int version, nb_comments, i;        chunk_id = avio_rl32(pb);        data_size = iff->is_64bit ? avio_rb64(pb) : avio_rb32(pb);        orig_pos = avio_tell(pb);         switch(chunk_id) {        case ID_VHDR:            st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;             if (data_size < 14)                return AVERROR_INVALIDDATA;            avio_skip(pb, 12);            st->codecpar->sample_rate = avio_rb16(pb);            if (data_size >= 16) {                avio_skip(pb, 1);                iff->svx8_compression = avio_r8(pb);            }            break;         case ID_MHDR:            st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;             if (data_size < 32)                return AVERROR_INVALIDDATA;            avio_skip(pb, 4);            iff->maud_bits = avio_rb16(pb);            avio_skip(pb, 2);            num = avio_rb32(pb);            den = avio_rb16(pb);            if (!den)                return AVERROR_INVALIDDATA;            avio_skip(pb, 2);            st->codecpar->sample_rate = num / den;            st->codecpar->channels = avio_rb16(pb);            iff->maud_compression = avio_rb16(pb);            if (st->codecpar->channels == 1)                st->codecpar->channel_layout = AV_CH_LAYOUT_MONO;            else if (st->codecpar->channels == 2)                st->codecpar->channel_layout = AV_CH_LAYOUT_STEREO;            break;         case ID_ABIT:        case ID_BODY:        case ID_DBOD:        case ID_DSD:        case ID_DST:        case ID_MDAT:            iff->body_pos = avio_tell(pb);            iff->body_end = iff->body_pos + data_size;            iff->body_size = data_size;            if (chunk_id == ID_DST) {                int ret = read_dst_frame(s, NULL);                if (ret < 0)                    return ret;            }

Lines 1–500 of 894.

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t