musx.c 6.37 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
/*
 * MUSX demuxer
 * Copyright (c) 2016 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 "avformat.h"
#include "internal.h"

static int musx_probe(AVProbeData *p)
{
    if (memcmp(p->buf, "MUSX", 4))
        return 0;

    return AVPROBE_SCORE_MAX / 5 * 2;
}

static int musx_read_header(AVFormatContext *s)
{
    unsigned type, version, coding, offset;
    AVStream *st;

    avio_skip(s->pb, 8);
    version = avio_rl32(s->pb);
    if (version != 10 &&
        version != 6 &&
        version != 5 &&
        version != 4 &&
        version != 201) {
        avpriv_request_sample(s, "Unsupported version: %d", version);
        return AVERROR_PATCHWELCOME;
    }
    avio_skip(s->pb, 4);

    st = avformat_new_stream(s, NULL);
    if (!st)
        return AVERROR(ENOMEM);

    if (version == 201) {
        avio_skip(s->pb, 8);
        offset = avio_rl32(s->pb);
58 59 60 61 62
        st->codecpar->codec_type  = AVMEDIA_TYPE_AUDIO;
        st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_PSX;
        st->codecpar->channels    = 2;
        st->codecpar->sample_rate = 32000;
        st->codecpar->block_align = 0x80 * st->codecpar->channels;
63 64
    }  else if (version == 10) {
        type = avio_rl32(s->pb);
65
        st->codecpar->codec_type  = AVMEDIA_TYPE_AUDIO;
66 67 68
        offset = 0x800;
        switch (type) {
        case MKTAG('P', 'S', '3', '_'):
69 70
            st->codecpar->channels    = 2;
            st->codecpar->sample_rate = 44100;
71 72 73 74 75
            avio_skip(s->pb, 44);
            coding = avio_rl32(s->pb);
            if (coding == MKTAG('D', 'A', 'T', '4') ||
                coding == MKTAG('D', 'A', 'T', '8')) {
                avio_skip(s->pb, 4);
76 77 78
                st->codecpar->channels   = avio_rl32(s->pb);
                if (st->codecpar->channels <= 0 ||
                    st->codecpar->channels > INT_MAX / 0x20)
79
                    return AVERROR_INVALIDDATA;
80
                st->codecpar->sample_rate = avio_rl32(s->pb);
81
            }
82 83
            st->codecpar->codec_id   = AV_CODEC_ID_ADPCM_IMA_DAT4;
            st->codecpar->block_align = 0x20 * st->codecpar->channels;
84 85 86 87 88 89 90 91 92 93
            break;
        case MKTAG('W', 'I', 'I', '_'):
            avio_skip(s->pb, 44);
            coding = avio_rl32(s->pb);
            if (coding != MKTAG('D', 'A', 'T', '4') &&
                coding != MKTAG('D', 'A', 'T', '8')) {
                avpriv_request_sample(s, "Unsupported coding: %X", coding);
                return AVERROR_PATCHWELCOME;
            }
            avio_skip(s->pb, 4);
94 95 96 97
            st->codecpar->codec_id   = AV_CODEC_ID_ADPCM_IMA_DAT4;
            st->codecpar->channels   = avio_rl32(s->pb);
            if (st->codecpar->channels <= 0 ||
                st->codecpar->channels > INT_MAX / 0x20)
98
                return AVERROR_INVALIDDATA;
99 100
            st->codecpar->sample_rate = avio_rl32(s->pb);
            st->codecpar->block_align = 0x20 * st->codecpar->channels;
101 102
            break;
        case MKTAG('X', 'E', '_', '_'):
103 104 105 106
            st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_IMA_DAT4;
            st->codecpar->channels    = 2;
            st->codecpar->sample_rate = 32000;
            st->codecpar->block_align = 0x20 * st->codecpar->channels;
107 108
            break;
        case MKTAG('P', 'S', 'P', '_'):
109 110 111 112
            st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_PSX;
            st->codecpar->channels    = 2;
            st->codecpar->sample_rate = 32768;
            st->codecpar->block_align = 0x80 * st->codecpar->channels;
113 114
            break;
        case MKTAG('P', 'S', '2', '_'):
115 116 117 118
            st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_PSX;
            st->codecpar->channels    = 2;
            st->codecpar->sample_rate = 32000;
            st->codecpar->block_align = 0x80 * st->codecpar->channels;
119 120 121 122 123 124 125 126
            break;
        default:
            avpriv_request_sample(s, "Unsupported type: %X", type);
            return AVERROR_PATCHWELCOME;
        }
    } else if (version == 6 || version == 5 || version == 4) {
        type = avio_rl32(s->pb);
        avio_skip(s->pb, 20);
127 128
        st->codecpar->codec_type  = AVMEDIA_TYPE_AUDIO;
        st->codecpar->channels    = 2;
129 130
        switch (type) {
        case MKTAG('G', 'C', '_', '_'):
131 132 133
            st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_IMA_DAT4;
            st->codecpar->block_align = 0x20 * st->codecpar->channels;
            st->codecpar->sample_rate = 32000;
134 135 136
            offset = avio_rb32(s->pb);
            break;
        case MKTAG('P', 'S', '2', '_'):
137 138 139
            st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_PSX;
            st->codecpar->block_align = 0x80 * st->codecpar->channels;
            st->codecpar->sample_rate = 32000;
140 141 142
            offset = avio_rl32(s->pb);
            break;
        case MKTAG('X', 'B', '_', '_'):
143 144 145
            st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_IMA_DAT4;
            st->codecpar->block_align = 0x20 * st->codecpar->channels;
            st->codecpar->sample_rate = 44100;
146 147 148 149 150 151 152 153 154 155 156 157
            offset = avio_rl32(s->pb);
            break;
        default:
            avpriv_request_sample(s, "Unsupported type: %X", type);
            return AVERROR_PATCHWELCOME;
        }
    } else {
        av_assert0(0);
    }

    avio_seek(s->pb, offset, SEEK_SET);

158
    avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
159 160 161 162 163 164

    return 0;
}

static int musx_read_packet(AVFormatContext *s, AVPacket *pkt)
{
165
    AVCodecParameters *par = s->streams[0]->codecpar;
166

167
    return av_get_packet(s->pb, pkt, par->block_align);
168 169 170 171 172 173 174 175 176 177
}

AVInputFormat ff_musx_demuxer = {
    .name           = "musx",
    .long_name      = NULL_IF_CONFIG_SMALL("Eurocom MUSX"),
    .read_probe     = musx_probe,
    .read_header    = musx_read_header,
    .read_packet    = musx_read_packet,
    .extensions     = "musx",
};