1/* 2 * id RoQ (.roq) File Demuxer 3 * Copyright (c) 2003 The FFmpeg project 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/** 23 * @file 24 * id RoQ format file demuxer 25 * by Mike Melanson (melanson@pcisys.net) 26 * for more information on the .roq file format, visit: 27 * http://www.csse.monash.edu.au/~timf/ 28 */ 29 30#include "libavutil/channel_layout.h" 31#include "libavutil/intreadwrite.h" 32#include "avformat.h" 33#include "internal.h" 34#include "avio_internal.h" 35 36#define RoQ_MAGIC_NUMBER 0x1084 37#define RoQ_CHUNK_PREAMBLE_SIZE 8 38#define RoQ_AUDIO_SAMPLE_RATE 22050 39#define RoQ_CHUNKS_TO_SCAN 30 40 41#define RoQ_INFO 0x1001 42#define RoQ_QUAD_CODEBOOK 0x1002 43#define RoQ_QUAD_VQ 0x1011 44#define RoQ_SOUND_MONO 0x1020 45#define RoQ_SOUND_STEREO 0x1021 46 47typedef struct RoqDemuxContext { 48 49 int frame_rate; 50 int width; 51 int height; 52 int audio_channels; 53 54 int video_stream_index; 55 int audio_stream_index; 56 57 int64_t video_pts; 58 unsigned int audio_frame_count; 59 60} RoqDemuxContext; 61 62static int roq_probe(const AVProbeData *p) 63{ 64 if ((AV_RL16(&p->buf[0]) != RoQ_MAGIC_NUMBER) || 65 (AV_RL32(&p->buf[2]) != 0xFFFFFFFF)) 66 return 0; 67 68 return AVPROBE_SCORE_MAX; 69} 70 71static int roq_read_header(AVFormatContext *s) 72{ 73 RoqDemuxContext *roq = s->priv_data; 74 AVIOContext *pb = s->pb; 75 unsigned char preamble[RoQ_CHUNK_PREAMBLE_SIZE]; 76 77 /* get the main header */ 78 if (avio_read(pb, preamble, RoQ_CHUNK_PREAMBLE_SIZE) != 79 RoQ_CHUNK_PREAMBLE_SIZE) 80 return AVERROR(EIO); 81 roq->frame_rate = AV_RL16(&preamble[6]); 82 83 /* init private context parameters */ 84 roq->width = roq->height = roq->audio_channels = roq->video_pts = 85 roq->audio_frame_count = 0; 86 roq->audio_stream_index = -1; 87 roq->video_stream_index = -1; 88 89 s->ctx_flags |= AVFMTCTX_NOHEADER; 90 91 return 0; 92} 93 94static int roq_read_packet(AVFormatContext *s, 95 AVPacket *pkt) 96{ 97 RoqDemuxContext *roq = s->priv_data; 98 AVIOContext *pb = s->pb; 99 int ret = 0; 100 unsigned int chunk_size; 101 unsigned int chunk_type; 102 unsigned int codebook_size; 103 unsigned char preamble[RoQ_CHUNK_PREAMBLE_SIZE]; 104 int packet_read = 0; 105 int64_t codebook_offset; 106 107 while (!packet_read) { 108 109 if (avio_feof(s->pb)) 110 return AVERROR(EIO); 111 112 /* get the next chunk preamble */ 113 if ((ret = avio_read(pb, preamble, RoQ_CHUNK_PREAMBLE_SIZE)) != 114 RoQ_CHUNK_PREAMBLE_SIZE) 115 return AVERROR(EIO); 116 117 chunk_type = AV_RL16(&preamble[0]); 118 chunk_size = AV_RL32(&preamble[2]); 119 if(chunk_size > INT_MAX) 120 return AVERROR_INVALIDDATA; 121 122 chunk_size = ffio_limit(pb, chunk_size); 123 124 switch (chunk_type) { 125 126 case RoQ_INFO: 127 if (roq->video_stream_index == -1) { 128 AVStream *st = avformat_new_stream(s, NULL); 129 if (!st) 130 return AVERROR(ENOMEM); 131 avpriv_set_pts_info(st, 63, 1, roq->frame_rate); 132 roq->video_stream_index = st->index; 133 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; 134 st->codecpar->codec_id = AV_CODEC_ID_ROQ; 135 st->codecpar->codec_tag = 0; /* no fourcc */ 136 137 if (avio_read(pb, preamble, RoQ_CHUNK_PREAMBLE_SIZE) != RoQ_CHUNK_PREAMBLE_SIZE) 138 return AVERROR(EIO); 139 st->codecpar->width = roq->width = AV_RL16(preamble); 140 st->codecpar->height = roq->height = AV_RL16(preamble + 2); 141 break; 142 } 143 /* don't care about this chunk anymore */ 144 avio_skip(pb, RoQ_CHUNK_PREAMBLE_SIZE); 145 break; 146 147 case RoQ_QUAD_CODEBOOK: 148 if (roq->video_stream_index < 0) 149 return AVERROR_INVALIDDATA; 150 /* packet needs to contain both this codebook and next VQ chunk */ 151 codebook_offset = avio_tell(pb) - RoQ_CHUNK_PREAMBLE_SIZE; 152 codebook_size = chunk_size; 153 avio_skip(pb, codebook_size); 154 if (avio_read(pb, preamble, RoQ_CHUNK_PREAMBLE_SIZE) != 155 RoQ_CHUNK_PREAMBLE_SIZE) 156 return AVERROR(EIO); 157 chunk_size = AV_RL32(&preamble[2]) + RoQ_CHUNK_PREAMBLE_SIZE * 2 + 158 codebook_size; 159 160 if (chunk_size > INT_MAX) 161 return AVERROR_INVALIDDATA; 162 163 /* rewind */ 164 avio_seek(pb, codebook_offset, SEEK_SET); 165 166 /* load up the packet */ 167 ret= av_get_packet(pb, pkt, chunk_size); 168 if (ret != chunk_size) 169 return AVERROR(EIO); 170 pkt->stream_index = roq->video_stream_index; 171 pkt->pts = roq->video_pts++; 172 173 packet_read = 1; 174 break; 175 176 case RoQ_SOUND_MONO: 177 case RoQ_SOUND_STEREO: 178 if (roq->audio_stream_index == -1) { 179 AVStream *st = avformat_new_stream(s, NULL); 180 if (!st) 181 return AVERROR(ENOMEM); 182 avpriv_set_pts_info(st, 32, 1, RoQ_AUDIO_SAMPLE_RATE); 183 roq->audio_stream_index = st->index; 184 st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; 185 st->codecpar->codec_id = AV_CODEC_ID_ROQ_DPCM; 186 st->codecpar->codec_tag = 0; /* no tag */ 187 if (chunk_type == RoQ_SOUND_STEREO) { 188 st->codecpar->ch_layout = (AVChannelLayout)AV_CHANNEL_LAYOUT_STEREO; 189 } else { 190 st->codecpar->ch_layout = (AVChannelLayout)AV_CHANNEL_LAYOUT_MONO; 191 } 192 roq->audio_channels = st->codecpar->ch_layout.nb_channels; 193 st->codecpar->sample_rate = RoQ_AUDIO_SAMPLE_RATE; 194 st->codecpar->bits_per_coded_sample = 16; 195 st->codecpar->bit_rate = roq->audio_channels * st->codecpar->sample_rate * 196 st->codecpar->bits_per_coded_sample; 197 st->codecpar->block_align = roq->audio_channels * st->codecpar->bits_per_coded_sample; 198 } 199 case RoQ_QUAD_VQ: 200 if (chunk_type == RoQ_QUAD_VQ) { 201 if (roq->video_stream_index < 0) 202 return AVERROR_INVALIDDATA; 203 } 204 205 /* load up the packet */ 206 ret = av_new_packet(pkt, chunk_size + RoQ_CHUNK_PREAMBLE_SIZE); 207 if (ret < 0) 208 return ret; 209 /* copy over preamble */ 210 memcpy(pkt->data, preamble, RoQ_CHUNK_PREAMBLE_SIZE); 211 212 if (chunk_type == RoQ_QUAD_VQ) { 213 pkt->stream_index = roq->video_stream_index; 214 pkt->pts = roq->video_pts++; 215 } else { 216 pkt->stream_index = roq->audio_stream_index; 217 pkt->pts = roq->audio_frame_count; 218 roq->audio_frame_count += (chunk_size / roq->audio_channels); 219 } 220 221 pkt->pos= avio_tell(pb); 222 ret = avio_read(pb, pkt->data + RoQ_CHUNK_PREAMBLE_SIZE, 223 chunk_size); 224 if (ret != chunk_size) { 225 return AVERROR(EIO); 226 } 227 228 packet_read = 1; 229 break; 230 231 default: 232 av_log(s, AV_LOG_ERROR, " unknown RoQ chunk (%04X)\n", chunk_type); 233 return AVERROR_INVALIDDATA; 234 } 235 } 236 237 return ret; 238} 239 240const AVInputFormat ff_roq_demuxer = { 241 .name = "roq", 242 .long_name = NULL_IF_CONFIG_SMALL("id RoQ"), 243 .priv_data_size = sizeof(RoqDemuxContext), 244 .read_probe = roq_probe, 245 .read_header = roq_read_header, 246 .read_packet = roq_read_packet, 247}; 248