/* * FSB demuxer * Copyright (c) 2015 Paul B Mahol * * 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 */ #include "libavutil/avassert.h"#include "libavutil/intreadwrite.h"#include "avformat.h"#include "avio.h"#include "internal.h" static int fsb_probe(AVProbeData *p){    if (memcmp(p->buf, "FSB", 3) || p->buf[3] - '0' < 1 || p->buf[3] - '0' > 5)        return 0;    if (AV_RL32(p->buf + 4) != 1)        return 0;    return AVPROBE_SCORE_MAX;} static int fsb_read_header(AVFormatContext *s){    AVIOContext *pb = s->pb;    unsigned format, version, c;    int64_t offset;    AVCodecParameters *par;    AVStream *st = avformat_new_stream(s, NULL);     avio_skip(pb, 3); // "FSB"    version = avio_r8(pb) - '0';    if (version != 4 && version != 3) {        avpriv_request_sample(s, "version %d", version);        return AVERROR_PATCHWELCOME;    }     avio_skip(pb, 4);     if (!st)        return AVERROR(ENOMEM);    par = st->codecpar;    par->codec_type  = AVMEDIA_TYPE_AUDIO;    par->codec_tag   = 0;     if (version == 3) {        offset = avio_rl32(pb) + 0x18;        avio_skip(pb, 44);        st->duration = avio_rl32(pb);        avio_skip(pb, 12);        format = avio_rl32(pb);        par->sample_rate = avio_rl32(pb);        if (par->sample_rate <= 0)            return AVERROR_INVALIDDATA;        avio_skip(pb, 6);        par->channels    = avio_rl16(pb);        if (!par->channels)            return AVERROR_INVALIDDATA;         if (format & 0x00000100) {            par->codec_id    = AV_CODEC_ID_PCM_S16LE;            par->block_align = 4096 * par->channels;        } else if (format & 0x00400000) {            par->bits_per_coded_sample = 4;            par->codec_id    = AV_CODEC_ID_ADPCM_IMA_WAV;            par->block_align = 36 * par->channels;        } else if (format & 0x00800000) {            par->codec_id    = AV_CODEC_ID_ADPCM_PSX;            par->block_align = 16 * par->channels;        } else if (format & 0x02000000) {            par->codec_id    = AV_CODEC_ID_ADPCM_THP;            par->block_align = 8 * par->channels;            if (par->channels > INT_MAX / 32)                return AVERROR_INVALIDDATA;            ff_alloc_extradata(par, 32 * par->channels);            if (!par->extradata)                return AVERROR(ENOMEM);            avio_seek(pb, 0x68, SEEK_SET);            for (c = 0; c < par->channels; c++) {                avio_read(pb, par->extradata + 32 * c, 32);                avio_skip(pb, 14);            }        } else {            avpriv_request_sample(s, "format 0x%X", format);            return AVERROR_PATCHWELCOME;        }    } else if (version == 4) {        offset = avio_rl32(pb) + 0x30;        avio_skip(pb, 80);        st->duration = avio_rl32(pb);         format = avio_rb32(pb);        switch(format) {        case 0x40001001:        case 0x00001005:        case 0x40001081:        case 0x40200001:            par->codec_id = AV_CODEC_ID_XMA2;            break;        case 0x40000802:            par->codec_id = AV_CODEC_ID_ADPCM_THP;            break;        default:            avpriv_request_sample(s, "format 0x%X", format);            return AVERROR_PATCHWELCOME;        }         par->sample_rate = avio_rl32(pb);        if (par->sample_rate <= 0)            return AVERROR_INVALIDDATA;        avio_skip(pb, 6);         par->channels    = avio_rl16(pb);        if (!par->channels)            return AVERROR_INVALIDDATA;         switch (par->codec_id) {        case AV_CODEC_ID_XMA2:            ff_alloc_extradata(par, 34);            if (!par->extradata)                return AVERROR(ENOMEM);            memset(par->extradata, 0, 34);            par->block_align = 2048;            break;        case AV_CODEC_ID_ADPCM_THP:            if (par->channels > INT_MAX / 32)                return AVERROR_INVALIDDATA;            ff_alloc_extradata(par, 32 * par->channels);            if (!par->extradata)                return AVERROR(ENOMEM);            avio_seek(pb, 0x80, SEEK_SET);            for (c = 0; c < par->channels; c++) {                avio_read(pb, par->extradata + 32 * c, 32);                avio_skip(pb, 14);            }            par->block_align = 8 * par->channels;            break;        }    } else {        av_assert0(0);    }     avio_skip(pb, offset - avio_tell(pb));    s->internal->data_offset = avio_tell(pb);     avpriv_set_pts_info(st, 64, 1, par->sample_rate);     return 0;} static int fsb_read_packet(AVFormatContext *s, AVPacket *pkt){    AVCodecParameters *par = s->streams[0]->codecpar;    int64_t pos;    int ret;     if (avio_feof(s->pb))        return AVERROR_EOF;     pos = avio_tell(s->pb);    if (par->codec_id == AV_CODEC_ID_ADPCM_THP &&               par->channels > 1) {        int i, ch;         ret = av_new_packet(pkt, par->block_align);        if (ret < 0)            return ret;        for (i = 0; i < 4; i++) {            for (ch = 0; ch < par->channels; ch++) {                pkt->data[ch * 8 + i * 2 + 0] = avio_r8(s->pb);                pkt->data[ch * 8 + i * 2 + 1] = avio_r8(s->pb);            }        }        ret = 0;    } else {        ret = av_get_packet(s->pb, pkt, par->block_align);    }     if (par->codec_id == AV_CODEC_ID_XMA2 && pkt->size >= 1)        pkt->duration = (pkt->data[0] >> 2) * 512;     pkt->pos = pos;    pkt->stream_index = 0;     return ret;} AVInputFormat ff_fsb_demuxer = {    .name        = "fsb",    .long_name   = NULL_IF_CONFIG_SMALL("FMOD Sample Bank"),    .read_probe  = fsb_probe,    .read_header = fsb_read_header,    .read_packet = fsb_read_packet,    .extensions  = "fsb",    .flags       = AVFMT_GENERIC_INDEX,};