FFmpeg
sierravmd.c
Go to the documentation of this file.
1 /*
2  * Sierra VMD Format Demuxer
3  * Copyright (c) 2004 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  * Sierra VMD file demuxer
25  * by Vladimir "VAG" Gneushev (vagsoft at mail.ru)
26  * for more information on the Sierra VMD file format, visit:
27  * http://www.pcisys.net/~melanson/codecs/
28  */
29 
31 #include "libavutil/intreadwrite.h"
32 #include "avformat.h"
33 #include "internal.h"
34 #include "avio_internal.h"
35 
36 #define VMD_HEADER_SIZE 0x0330
37 #define BYTES_PER_FRAME_RECORD 16
38 
39 typedef struct vmd_frame {
41  unsigned int frame_size;
42  int64_t frame_offset;
43  int64_t pts;
45 } vmd_frame;
46 
47 typedef struct VmdDemuxContext {
50 
51  unsigned int frame_count;
52  unsigned int frames_per_block;
54  unsigned int current_frame;
55  int is_indeo3;
56 
59  int skiphdr;
60 
61  unsigned char vmd_header[VMD_HEADER_SIZE];
63 
64 static int vmd_probe(const AVProbeData *p)
65 {
66  int w, h, sample_rate;
67  if (p->buf_size < 806)
68  return 0;
69  /* check if the first 2 bytes of the file contain the appropriate size
70  * of a VMD header chunk */
71  if (AV_RL16(&p->buf[0]) != VMD_HEADER_SIZE - 2)
72  return 0;
73  w = AV_RL16(&p->buf[12]);
74  h = AV_RL16(&p->buf[14]);
75  sample_rate = AV_RL16(&p->buf[804]);
76  if ((!w || w > 2048 || !h || h > 2048) &&
77  sample_rate != 22050)
78  return 0;
79 
80  /* only return half certainty since this check is a bit sketchy */
82 }
83 
85 {
86  VmdDemuxContext *vmd = s->priv_data;
87  AVIOContext *pb = s->pb;
88  AVStream *st = NULL, *vst = NULL;
89  unsigned int toc_offset;
90  unsigned char *raw_frame_table;
91  int raw_frame_table_size;
92  int64_t current_offset;
93  int i, j, ret;
94  int width, height;
95  unsigned int total_frames;
96  int64_t current_audio_pts = 0;
97  unsigned char chunk[BYTES_PER_FRAME_RECORD];
98  int num, den;
99  int sound_buffers;
100 
101  /* fetch the main header, including the 2 header length bytes */
102  avio_seek(pb, 0, SEEK_SET);
104  return AVERROR(EIO);
105 
106  width = AV_RL16(&vmd->vmd_header[12]);
107  height = AV_RL16(&vmd->vmd_header[14]);
108  if (width && height) {
109  if(vmd->vmd_header[24] == 'i' && vmd->vmd_header[25] == 'v' && vmd->vmd_header[26] == '3') {
110  vmd->is_indeo3 = 1;
111  } else {
112  vmd->is_indeo3 = 0;
113  }
114  /* start up the decoders */
115  vst = avformat_new_stream(s, NULL);
116  if (!vst)
117  return AVERROR(ENOMEM);
118  avpriv_set_pts_info(vst, 33, 1, 10);
119  vmd->video_stream_index = vst->index;
120  vst->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
121  vst->codecpar->codec_id = vmd->is_indeo3 ? AV_CODEC_ID_INDEO3 : AV_CODEC_ID_VMDVIDEO;
122  vst->codecpar->codec_tag = 0; /* no fourcc */
123  vst->codecpar->width = width;
124  vst->codecpar->height = height;
125  if(vmd->is_indeo3 && vst->codecpar->width > 320){
126  vst->codecpar->width >>= 1;
127  vst->codecpar->height >>= 1;
128  }
129  if ((ret = ff_alloc_extradata(vst->codecpar, VMD_HEADER_SIZE)) < 0)
130  return ret;
131  memcpy(vst->codecpar->extradata, vmd->vmd_header, VMD_HEADER_SIZE);
132  }
133 
134  /* if sample rate is 0, assume no audio */
135  vmd->sample_rate = AV_RL16(&vmd->vmd_header[804]);
136  if (vmd->sample_rate) {
137  int channels;
138  st = avformat_new_stream(s, NULL);
139  if (!st)
140  return AVERROR(ENOMEM);
141  vmd->audio_stream_index = st->index;
144  st->codecpar->codec_tag = 0; /* no fourcc */
145  st->codecpar->sample_rate = vmd->sample_rate;
146  st->codecpar->block_align = AV_RL16(&vmd->vmd_header[806]);
147  if (st->codecpar->block_align & 0x8000) {
149  st->codecpar->block_align = -(st->codecpar->block_align - 0x10000);
150  } else {
152  }
153  if (vmd->vmd_header[811] & 0x80) {
154  channels = 2;
155  } else if (vmd->vmd_header[811] & 0x2) {
156  /* Shivers 2 stereo audio */
157  /* Frame length is for 1 channel */
158  channels = 2;
159  st->codecpar->block_align = st->codecpar->block_align << 1;
160  } else {
161  channels = 1;
162  }
164  st->codecpar->bit_rate = st->codecpar->sample_rate *
166 
167  /* calculate pts */
168  num = st->codecpar->block_align;
169  den = st->codecpar->sample_rate * channels;
170  av_reduce(&num, &den, num, den, (1UL<<31)-1);
171  if (vst)
172  avpriv_set_pts_info(vst, 33, num, den);
173  avpriv_set_pts_info(st, 33, num, den);
174  }
175  if (!s->nb_streams)
176  return AVERROR_INVALIDDATA;
177 
178  toc_offset = AV_RL32(&vmd->vmd_header[812]);
179  vmd->frame_count = AV_RL16(&vmd->vmd_header[6]);
180  vmd->frames_per_block = AV_RL16(&vmd->vmd_header[18]);
181  avio_seek(pb, toc_offset, SEEK_SET);
182 
183  raw_frame_table = NULL;
184  vmd->frame_table = NULL;
185  sound_buffers = AV_RL16(&vmd->vmd_header[808]);
186  raw_frame_table_size = vmd->frame_count * 6;
187  raw_frame_table = av_malloc(raw_frame_table_size);
188  vmd->frame_table = av_malloc_array(vmd->frame_count * vmd->frames_per_block + sound_buffers, sizeof(vmd_frame));
189  if (!raw_frame_table || !vmd->frame_table) {
190  ret = AVERROR(ENOMEM);
191  goto error;
192  }
193  if (avio_read(pb, raw_frame_table, raw_frame_table_size) !=
194  raw_frame_table_size) {
195  ret = AVERROR(EIO);
196  goto error;
197  }
198 
199  total_frames = 0;
200  for (i = 0; i < vmd->frame_count; i++) {
201 
202  current_offset = AV_RL32(&raw_frame_table[6 * i + 2]);
203 
204  /* handle each entry in index block */
205  for (j = 0; j < vmd->frames_per_block; j++) {
206  int type;
207  uint32_t size;
208 
210  av_log(s, AV_LOG_ERROR, "Failed to read frame record\n");
211  if (ret >= 0)
213  goto error;
214  }
215  type = chunk[0];
216  size = AV_RL32(&chunk[2]);
217  if (size > INT_MAX / 2) {
218  av_log(s, AV_LOG_ERROR, "Invalid frame size\n");
220  goto error;
221  }
222  if(!size && type != 1)
223  continue;
224  switch(type) {
225  case 1: /* Audio Chunk */
226  if (!st) break;
227  /* first audio chunk contains several audio buffers */
228  vmd->frame_table[total_frames].frame_offset = current_offset;
229  vmd->frame_table[total_frames].stream_index = vmd->audio_stream_index;
230  vmd->frame_table[total_frames].frame_size = size;
231  memcpy(vmd->frame_table[total_frames].frame_record, chunk, BYTES_PER_FRAME_RECORD);
232  vmd->frame_table[total_frames].pts = current_audio_pts;
233  total_frames++;
234  if(!current_audio_pts)
235  current_audio_pts += sound_buffers - 1;
236  else
237  current_audio_pts++;
238  break;
239  case 2: /* Video Chunk */
240  if (!vst)
241  break;
242  vmd->frame_table[total_frames].frame_offset = current_offset;
243  vmd->frame_table[total_frames].stream_index = vmd->video_stream_index;
244  vmd->frame_table[total_frames].frame_size = size;
245  memcpy(vmd->frame_table[total_frames].frame_record, chunk, BYTES_PER_FRAME_RECORD);
246  vmd->frame_table[total_frames].pts = i;
247  total_frames++;
248  break;
249  }
250  current_offset += size;
251  }
252  }
253 
254 
255  vmd->current_frame = 0;
256  vmd->frame_count = total_frames;
257 
258  ret = 0;
259 error:
260  av_freep(&raw_frame_table);
261  return ret;
262 }
263 
265  AVPacket *pkt)
266 {
267  VmdDemuxContext *vmd = s->priv_data;
268  AVIOContext *pb = s->pb;
269  int ret = 0;
270  vmd_frame *frame;
271 
272  if (vmd->current_frame >= vmd->frame_count)
273  return AVERROR_EOF;
274 
275  frame = &vmd->frame_table[vmd->current_frame];
276  /* position the stream (will probably be there already) */
277  avio_seek(pb, frame->frame_offset, SEEK_SET);
278 
279  if(ffio_limit(pb, frame->frame_size) != frame->frame_size)
280  return AVERROR(EIO);
282  if (ret < 0)
283  return ret;
284  pkt->pos= avio_tell(pb);
285  memcpy(pkt->data, frame->frame_record, BYTES_PER_FRAME_RECORD);
286  if(vmd->is_indeo3 && frame->frame_record[0] == 0x02)
287  ret = avio_read(pb, pkt->data, frame->frame_size);
288  else
290  frame->frame_size);
291 
292  if (ret != frame->frame_size) {
293  ret = AVERROR(EIO);
294  }
295  pkt->stream_index = frame->stream_index;
296  pkt->pts = frame->pts;
297  av_log(s, AV_LOG_DEBUG, " dispatching %s frame with %d bytes and pts %"PRId64"\n",
298  (frame->frame_record[0] == 0x02) ? "video" : "audio",
299  frame->frame_size + BYTES_PER_FRAME_RECORD,
300  pkt->pts);
301 
302  vmd->current_frame++;
303 
304  return ret;
305 }
306 
308 {
309  VmdDemuxContext *vmd = s->priv_data;
310 
311  av_freep(&vmd->frame_table);
312 
313  return 0;
314 }
315 
317  .name = "vmd",
318  .long_name = NULL_IF_CONFIG_SMALL("Sierra VMD"),
319  .priv_data_size = sizeof(VmdDemuxContext),
320  .flags_internal = FF_FMT_INIT_CLEANUP,
325 };
error
static void error(const char *err)
Definition: target_bsf_fuzzer.c:31
FF_FMT_INIT_CLEANUP
#define FF_FMT_INIT_CLEANUP
For an AVInputFormat with this flag set read_close() needs to be called by the caller upon read_heade...
Definition: internal.h:48
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
avformat_new_stream
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: options.c:237
AVCodecParameters::codec_type
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:57
VMD_HEADER_SIZE
#define VMD_HEADER_SIZE
Definition: sierravmd.c:36
vmd_frame
Definition: sierravmd.c:39
AVERROR_EOF
#define AVERROR_EOF
End of file.
Definition: error.h:57
vmd_read_close
static int vmd_read_close(AVFormatContext *s)
Definition: sierravmd.c:307
w
uint8_t w
Definition: llviddspenc.c:38
AV_CODEC_ID_INDEO3
@ AV_CODEC_ID_INDEO3
Definition: codec_id.h:78
AVPacket::data
uint8_t * data
Definition: packet.h:374
VmdDemuxContext::frame_count
unsigned int frame_count
Definition: sierravmd.c:51
ff_vmd_demuxer
const AVInputFormat ff_vmd_demuxer
Definition: sierravmd.c:316
VmdDemuxContext::frames_per_block
unsigned int frames_per_block
Definition: sierravmd.c:52
AVCodecParameters::codec_tag
uint32_t codec_tag
Additional information about the codec (corresponds to the AVI FOURCC).
Definition: codec_par.h:65
AVProbeData::buf_size
int buf_size
Size of buf except extra allocated bytes.
Definition: avformat.h:456
sample_rate
sample_rate
Definition: ffmpeg_filter.c:153
av_malloc
#define av_malloc(s)
Definition: tableprint_vlc.h:30
avpriv_set_pts_info
void avpriv_set_pts_info(AVStream *st, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: avformat.c:697
vmd_frame::stream_index
int stream_index
Definition: sierravmd.c:40
read_close
static av_cold int read_close(AVFormatContext *ctx)
Definition: libcdio.c:143
avio_tell
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:505
type
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf type
Definition: writing_filters.txt:86
av_reduce
int av_reduce(int *dst_num, int *dst_den, int64_t num, int64_t den, int64_t max)
Reduce a fraction.
Definition: rational.c:35
pkt
AVPacket * pkt
Definition: movenc.c:59
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
AVInputFormat
Definition: avformat.h:656
VmdDemuxContext::is_indeo3
int is_indeo3
Definition: sierravmd.c:55
VmdDemuxContext::current_frame
unsigned int current_frame
Definition: sierravmd.c:54
width
#define width
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:256
av_new_packet
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:97
AVInputFormat::name
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:661
AVProbeData::buf
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:455
AVMEDIA_TYPE_AUDIO
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
AV_CODEC_ID_VMDAUDIO
@ AV_CODEC_ID_VMDAUDIO
Definition: codec_id.h:438
AV_LOG_DEBUG
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:201
channels
channels
Definition: aptx.h:32
AV_RL16
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_RL16
Definition: bytestream.h:94
VmdDemuxContext::frame_table
vmd_frame * frame_table
Definition: sierravmd.c:53
vmd_probe
static int vmd_probe(const AVProbeData *p)
Definition: sierravmd.c:64
VmdDemuxContext::vmd_header
unsigned char vmd_header[VMD_HEADER_SIZE]
Definition: sierravmd.c:61
AVFormatContext
Format I/O context.
Definition: avformat.h:1213
internal.h
AVStream::codecpar
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1108
read_header
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:532
NULL
#define NULL
Definition: coverity.c:32
read_probe
static int read_probe(const AVProbeData *pd)
Definition: jvdec.c:55
VmdDemuxContext::skiphdr
int skiphdr
Definition: sierravmd.c:59
VmdDemuxContext::video_stream_index
int video_stream_index
Definition: sierravmd.c:48
AVProbeData
This structure contains the data a format has to probe a file.
Definition: avformat.h:453
av_channel_layout_default
void av_channel_layout_default(AVChannelLayout *ch_layout, int nb_channels)
Get the default channel layout for a given number of channels.
Definition: channel_layout.c:960
AVCodecParameters::ch_layout
AVChannelLayout ch_layout
Audio only.
Definition: codec_par.h:212
AV_CODEC_ID_VMDVIDEO
@ AV_CODEC_ID_VMDVIDEO
Definition: codec_id.h:102
VmdDemuxContext::sample_rate
int sample_rate
Definition: sierravmd.c:57
AVPROBE_SCORE_EXTENSION
#define AVPROBE_SCORE_EXTENSION
score for file extension
Definition: avformat.h:463
AVCodecParameters::sample_rate
int sample_rate
Audio only.
Definition: codec_par.h:177
AVIOContext
Bytestream IO Context.
Definition: avio.h:162
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
vmd_read_header
static int vmd_read_header(AVFormatContext *s)
Definition: sierravmd.c:84
size
int size
Definition: twinvq_data.h:10344
height
#define height
ffio_limit
int ffio_limit(AVIOContext *s, int size)
Definition: aviobuf.c:1093
vmd_frame::pts
int64_t pts
Definition: sierravmd.c:43
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:269
AVPacket::pts
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: packet.h:367
avio_internal.h
AVCodecParameters::block_align
int block_align
Audio only.
Definition: codec_par.h:184
av_malloc_array
#define av_malloc_array(a, b)
Definition: tableprint_vlc.h:31
vmd_read_packet
static int vmd_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: sierravmd.c:264
ret
ret
Definition: filter_design.txt:187
read_packet
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
AVStream
Stream structure.
Definition: avformat.h:948
avio_seek
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:260
frame
these buffered frames must be flushed immediately if a new input produces new the filter must not call request_frame to get more It must just process the frame or queue it The task of requesting more frames is left to the filter s request_frame method or the application If a filter has several the filter must be ready for frames arriving randomly on any input any filter with several inputs will most likely require some kind of queuing mechanism It is perfectly acceptable to have a limited queue and to drop frames when the inputs are too unbalanced request_frame For filters that do not use the this method is called when a frame is wanted on an output For a it should directly call filter_frame on the corresponding output For a if there are queued frames already one of these frames should be pushed If the filter should request a frame on one of its repeatedly until at least one frame has been pushed Return or at least make progress towards producing a frame
Definition: filter_design.txt:264
avformat.h
AV_RL32
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:92
VmdDemuxContext
Definition: sierravmd.c:47
AVStream::index
int index
stream index in AVFormatContext
Definition: avformat.h:956
channel_layout.h
avio_read
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:641
BYTES_PER_FRAME_RECORD
#define BYTES_PER_FRAME_RECORD
Definition: sierravmd.c:37
VmdDemuxContext::audio_stream_index
int audio_stream_index
Definition: sierravmd.c:49
AVPacket::stream_index
int stream_index
Definition: packet.h:376
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
vmd_frame::frame_size
unsigned int frame_size
Definition: sierravmd.c:41
AVCodecParameters::bits_per_coded_sample
int bits_per_coded_sample
The number of bits per sample in the codedwords.
Definition: codec_par.h:103
VmdDemuxContext::audio_sample_counter
int64_t audio_sample_counter
Definition: sierravmd.c:58
AVCodecParameters::codec_id
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:61
AVPacket
This structure stores compressed data.
Definition: packet.h:351
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:34
AVPacket::pos
int64_t pos
byte position in stream, -1 if unknown
Definition: packet.h:394
AVCodecParameters::bit_rate
int64_t bit_rate
The average bitrate of the encoded data (in bits per second).
Definition: codec_par.h:90
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
h
h
Definition: vp9dsp_template.c:2038
vmd_frame::frame_offset
int64_t frame_offset
Definition: sierravmd.c:42
vmd_frame::frame_record
unsigned char frame_record[BYTES_PER_FRAME_RECORD]
Definition: sierravmd.c:44
ff_alloc_extradata
int ff_alloc_extradata(AVCodecParameters *par, int size)
Allocate extradata with additional AV_INPUT_BUFFER_PADDING_SIZE at end which is always set to 0.
Definition: utils.c:238