1/* 2 * GENH demuxer 3 * Copyright (c) 2015 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/channel_layout.h" 23#include "libavutil/intreadwrite.h" 24#include "avformat.h" 25#include "internal.h" 26 27typedef struct GENHDemuxContext { 28 unsigned dsp_int_type; 29 unsigned interleave_size; 30} GENHDemuxContext; 31 32static int genh_probe(const AVProbeData *p) 33{ 34 if (AV_RL32(p->buf) != MKTAG('G','E','N','H')) 35 return 0; 36 if (AV_RL32(p->buf+4) <= 0 || AV_RL32(p->buf+4) > 0xFFFF) // channels 37 return 0; 38 39 return AVPROBE_SCORE_MAX / 3 * 2; 40} 41 42static int genh_read_header(AVFormatContext *s) 43{ 44 unsigned start_offset, header_size, codec, coef_type, coef[2]; 45 GENHDemuxContext *c = s->priv_data; 46 av_unused unsigned coef_splitted[2]; 47 int align, ch, ret; 48 AVStream *st; 49 50 avio_skip(s->pb, 4); 51 52 st = avformat_new_stream(s, NULL); 53 if (!st) 54 return AVERROR(ENOMEM); 55 56 st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; 57 st->codecpar->ch_layout.nb_channels = avio_rl32(s->pb); 58 if (st->codecpar->ch_layout.nb_channels <= 0) 59 return AVERROR_INVALIDDATA; 60 if (st->codecpar->ch_layout.nb_channels == 1) 61 st->codecpar->ch_layout = (AVChannelLayout)AV_CHANNEL_LAYOUT_MONO; 62 else if (st->codecpar->ch_layout.nb_channels == 2) 63 st->codecpar->ch_layout = (AVChannelLayout)AV_CHANNEL_LAYOUT_STEREO; 64 align = 65 c->interleave_size = avio_rl32(s->pb); 66 if (align < 0 || align > INT_MAX / st->codecpar->ch_layout.nb_channels) 67 return AVERROR_INVALIDDATA; 68 st->codecpar->block_align = align * st->codecpar->ch_layout.nb_channels; 69 st->codecpar->sample_rate = avio_rl32(s->pb); 70 if (st->codecpar->sample_rate < 0) 71 return AVERROR_INVALIDDATA; 72 73 avio_skip(s->pb, 4); 74 st->duration = avio_rl32(s->pb); 75 76 codec = avio_rl32(s->pb); 77 switch (codec) { 78 case 0: st->codecpar->codec_id = AV_CODEC_ID_ADPCM_PSX; break; 79 case 1: 80 case 11: st->codecpar->bits_per_coded_sample = 4; 81 if (st->codecpar->ch_layout.nb_channels > INT_MAX / 36) 82 return AVERROR_INVALIDDATA; 83 st->codecpar->block_align = 36 * st->codecpar->ch_layout.nb_channels; 84 st->codecpar->codec_id = AV_CODEC_ID_ADPCM_IMA_WAV; break; 85 case 2: st->codecpar->codec_id = AV_CODEC_ID_ADPCM_DTK; break; 86 case 3: st->codecpar->codec_id = st->codecpar->block_align > 0 ? 87 AV_CODEC_ID_PCM_S16BE_PLANAR : 88 AV_CODEC_ID_PCM_S16BE; break; 89 case 4: st->codecpar->codec_id = st->codecpar->block_align > 0 ? 90 AV_CODEC_ID_PCM_S16LE_PLANAR : 91 AV_CODEC_ID_PCM_S16LE; break; 92 case 5: st->codecpar->codec_id = st->codecpar->block_align > 0 ? 93 AV_CODEC_ID_PCM_S8_PLANAR : 94 AV_CODEC_ID_PCM_S8; break; 95 case 6: if (st->codecpar->block_align > INT_MAX/1024) 96 return AVERROR_INVALIDDATA; 97 st->codecpar->codec_id = AV_CODEC_ID_SDX2_DPCM; break; 98 case 7: ret = ff_alloc_extradata(st->codecpar, 2); 99 if (ret < 0) 100 return ret; 101 AV_WL16(st->codecpar->extradata, 3); 102 st->codecpar->codec_id = AV_CODEC_ID_ADPCM_IMA_WS; break; 103 case 10: st->codecpar->codec_id = AV_CODEC_ID_ADPCM_AICA; break; 104 case 12: st->codecpar->codec_id = AV_CODEC_ID_ADPCM_THP; break; 105 case 13: st->codecpar->codec_id = AV_CODEC_ID_PCM_U8; break; 106 case 17: st->codecpar->codec_id = AV_CODEC_ID_ADPCM_IMA_QT; break; 107 default: 108 avpriv_request_sample(s, "codec %d", codec); 109 return AVERROR_PATCHWELCOME; 110 } 111 112 start_offset = avio_rl32(s->pb); 113 header_size = avio_rl32(s->pb); 114 115 if (header_size > start_offset) 116 return AVERROR_INVALIDDATA; 117 118 if (header_size == 0) 119 start_offset = 0x800; 120 121 coef[0] = avio_rl32(s->pb); 122 coef[1] = avio_rl32(s->pb); 123 c->dsp_int_type = avio_rl32(s->pb); 124 coef_type = avio_rl32(s->pb); 125 coef_splitted[0] = avio_rl32(s->pb); 126 coef_splitted[1] = avio_rl32(s->pb); 127 128 if (st->codecpar->codec_id == AV_CODEC_ID_ADPCM_THP) { 129 if (st->codecpar->ch_layout.nb_channels > 2) { 130 avpriv_request_sample(s, "channels %d>2", st->codecpar->ch_layout.nb_channels); 131 return AVERROR_PATCHWELCOME; 132 } 133 134 ff_alloc_extradata(st->codecpar, 32 * st->codecpar->ch_layout.nb_channels); 135 for (ch = 0; ch < st->codecpar->ch_layout.nb_channels; ch++) { 136 if (coef_type & 1) { 137 avpriv_request_sample(s, "coef_type & 1"); 138 return AVERROR_PATCHWELCOME; 139 } else { 140 avio_seek(s->pb, coef[ch], SEEK_SET); 141 avio_read(s->pb, st->codecpar->extradata + 32 * ch, 32); 142 } 143 } 144 145 if (c->dsp_int_type == 1) { 146 st->codecpar->block_align = 8 * st->codecpar->ch_layout.nb_channels; 147 if (c->interleave_size != 1 && 148 c->interleave_size != 2 && 149 c->interleave_size != 4) 150 return AVERROR_INVALIDDATA; 151 } 152 } 153 154 if (st->codecpar->block_align <= 0) 155 return AVERROR_INVALIDDATA; 156 157 avio_skip(s->pb, start_offset - avio_tell(s->pb)); 158 159 avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate); 160 161 return 0; 162} 163 164static int genh_read_packet(AVFormatContext *s, AVPacket *pkt) 165{ 166 AVCodecParameters *par = s->streams[0]->codecpar; 167 GENHDemuxContext *c = s->priv_data; 168 int ret; 169 170 if (c->dsp_int_type == 1 && par->codec_id == AV_CODEC_ID_ADPCM_THP && 171 par->ch_layout.nb_channels > 1) { 172 int i, ch; 173 174 if (avio_feof(s->pb)) 175 return AVERROR_EOF; 176 ret = av_new_packet(pkt, 8 * par->ch_layout.nb_channels); 177 if (ret < 0) 178 return ret; 179 for (i = 0; i < 8 / c->interleave_size; i++) { 180 for (ch = 0; ch < par->ch_layout.nb_channels; ch++) { 181 pkt->data[ch * 8 + i*c->interleave_size+0] = avio_r8(s->pb); 182 pkt->data[ch * 8 + i*c->interleave_size+1] = avio_r8(s->pb); 183 } 184 } 185 ret = 0; 186 } else if (par->codec_id == AV_CODEC_ID_SDX2_DPCM) { 187 ret = av_get_packet(s->pb, pkt, par->block_align * 1024); 188 189 } else { 190 ret = av_get_packet(s->pb, pkt, par->block_align ? par->block_align : 191 1024 * par->ch_layout.nb_channels); 192 } 193 194 pkt->stream_index = 0; 195 return ret; 196} 197 198const AVInputFormat ff_genh_demuxer = { 199 .name = "genh", 200 .long_name = NULL_IF_CONFIG_SMALL("GENeric Header"), 201 .priv_data_size = sizeof(GENHDemuxContext), 202 .read_probe = genh_probe, 203 .read_header = genh_read_header, 204 .read_packet = genh_read_packet, 205 .extensions = "genh", 206}; 207