git.sesse.net Git - ffmpeg/blob - libavformat/musx.c

   1 /*
   2  * MUSX demuxer
   3  * Copyright (c) 2016 Paul B Mahol
   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/avassert.h"
  23 #include "libavutil/intreadwrite.h"
  24 #include "avformat.h"
  25 #include "internal.h"
  26
  27 static int musx_probe(const AVProbeData *p)
  28 {
  29     unsigned version;
  30
  31     if (AV_RB32(p->buf) != MKBETAG('M','U','S','X'))
  32         return 0;
  33
  34     version = AV_RL32(p->buf + 8);
  35     if (version != 10 &&
  36         version != 6 &&
  37         version != 5 &&
  38         version != 4 &&
  39         version != 201)
  40         return 0;
  41
  42     return AVPROBE_SCORE_MAX / 5 * 2;
  43 }
  44
  45 static int musx_read_header(AVFormatContext *s)
  46 {
  47     unsigned type, version, coding, offset;
  48     AVStream *st;
  49
  50     avio_skip(s->pb, 8);
  51     version = avio_rl32(s->pb);
  52     if (version != 10 &&
  53         version != 6 &&
  54         version != 5 &&
  55         version != 4 &&
  56         version != 201) {
  57         avpriv_request_sample(s, "Unsupported version: %d", version);
  58         return AVERROR_PATCHWELCOME;
  59     }
  60     avio_skip(s->pb, 4);
  61
  62     st = avformat_new_stream(s, NULL);
  63     if (!st)
  64         return AVERROR(ENOMEM);
  65
  66     if (version == 201) {
  67         avio_skip(s->pb, 8);
  68         offset = avio_rl32(s->pb);
  69         st->codecpar->codec_type  = AVMEDIA_TYPE_AUDIO;
  70         st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_PSX;
  71         st->codecpar->channels    = 2;
  72         st->codecpar->sample_rate = 32000;
  73         st->codecpar->block_align = 0x80 * st->codecpar->channels;
  74     }  else if (version == 10) {
  75         type = avio_rl32(s->pb);
  76         st->codecpar->codec_type  = AVMEDIA_TYPE_AUDIO;
  77         offset = 0x800;
  78         switch (type) {
  79         case MKTAG('P', 'S', '3', '_'):
  80             st->codecpar->channels    = 2;
  81             st->codecpar->sample_rate = 44100;
  82             avio_skip(s->pb, 44);
  83             coding = avio_rl32(s->pb);
  84             if (coding == MKTAG('D', 'A', 'T', '4') ||
  85                 coding == MKTAG('D', 'A', 'T', '8')) {
  86                 avio_skip(s->pb, 4);
  87                 st->codecpar->channels   = avio_rl32(s->pb);
  88                 if (st->codecpar->channels <= 0 ||
  89                     st->codecpar->channels > INT_MAX / 0x20)
  90                     return AVERROR_INVALIDDATA;
  91                 st->codecpar->sample_rate = avio_rl32(s->pb);
  92             }
  93             st->codecpar->codec_id   = AV_CODEC_ID_ADPCM_IMA_DAT4;
  94             st->codecpar->block_align = 0x20 * st->codecpar->channels;
  95             break;
  96         case MKTAG('W', 'I', 'I', '_'):
  97             avio_skip(s->pb, 44);
  98             coding = avio_rl32(s->pb);
  99             if (coding != MKTAG('D', 'A', 'T', '4') &&
 100                 coding != MKTAG('D', 'A', 'T', '8')) {
 101                 avpriv_request_sample(s, "Unsupported coding: %X", coding);
 102                 return AVERROR_PATCHWELCOME;
 103             }
 104             avio_skip(s->pb, 4);
 105             st->codecpar->codec_id   = AV_CODEC_ID_ADPCM_IMA_DAT4;
 106             st->codecpar->channels   = avio_rl32(s->pb);
 107             if (st->codecpar->channels <= 0 ||
 108                 st->codecpar->channels > INT_MAX / 0x20)
 109                 return AVERROR_INVALIDDATA;
 110             st->codecpar->sample_rate = avio_rl32(s->pb);
 111             st->codecpar->block_align = 0x20 * st->codecpar->channels;
 112             break;
 113         case MKTAG('X', 'E', '_', '_'):
 114             st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_IMA_DAT4;
 115             st->codecpar->channels    = 2;
 116             st->codecpar->sample_rate = 32000;
 117             st->codecpar->block_align = 0x20 * st->codecpar->channels;
 118             break;
 119         case MKTAG('P', 'S', 'P', '_'):
 120             st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_PSX;
 121             st->codecpar->channels    = 2;
 122             st->codecpar->sample_rate = 32768;
 123             st->codecpar->block_align = 0x80 * st->codecpar->channels;
 124             break;
 125         case MKTAG('P', 'S', '2', '_'):
 126             st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_PSX;
 127             st->codecpar->channels    = 2;
 128             st->codecpar->sample_rate = 32000;
 129             st->codecpar->block_align = 0x80 * st->codecpar->channels;
 130             break;
 131         default:
 132             avpriv_request_sample(s, "Unsupported type: %X", type);
 133             return AVERROR_PATCHWELCOME;
 134         }
 135     } else if (version == 6 || version == 5 || version == 4) {
 136         type = avio_rl32(s->pb);
 137         avio_skip(s->pb, 20);
 138         st->codecpar->codec_type  = AVMEDIA_TYPE_AUDIO;
 139         st->codecpar->channels    = 2;
 140         switch (type) {
 141         case MKTAG('G', 'C', '_', '_'):
 142             st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_IMA_DAT4;
 143             st->codecpar->block_align = 0x20 * st->codecpar->channels;
 144             st->codecpar->sample_rate = 32000;
 145             offset = avio_rb32(s->pb);
 146             break;
 147         case MKTAG('P', 'S', '2', '_'):
 148             st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_PSX;
 149             st->codecpar->block_align = 0x80 * st->codecpar->channels;
 150             st->codecpar->sample_rate = 32000;
 151             offset = avio_rl32(s->pb);
 152             break;
 153         case MKTAG('X', 'B', '_', '_'):
 154             st->codecpar->codec_id    = AV_CODEC_ID_ADPCM_IMA_DAT4;
 155             st->codecpar->block_align = 0x20 * st->codecpar->channels;
 156             st->codecpar->sample_rate = 44100;
 157             offset = avio_rl32(s->pb);
 158             break;
 159         default:
 160             avpriv_request_sample(s, "Unsupported type: %X", type);
 161             return AVERROR_PATCHWELCOME;
 162         }
 163     } else {
 164         av_assert0(0);
 165     }
 166
 167     avio_seek(s->pb, offset, SEEK_SET);
 168
 169     avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
 170
 171     return 0;
 172 }
 173
 174 static int musx_read_packet(AVFormatContext *s, AVPacket *pkt)
 175 {
 176     AVCodecParameters *par = s->streams[0]->codecpar;
 177
 178     return av_get_packet(s->pb, pkt, par->block_align);
 179 }
 180
 181 const AVInputFormat ff_musx_demuxer = {
 182     .name           = "musx",
 183     .long_name      = NULL_IF_CONFIG_SMALL("Eurocom MUSX"),
 184     .read_probe     = musx_probe,
 185     .read_header    = musx_read_header,
 186     .read_packet    = musx_read_packet,
 187     .extensions     = "musx",
 188 };