fix clippy warnings
[nihav.git] / nihav-realmedia / src / demuxers / realmedia.rs
CommitLineData
ce52b3b5 1use super::*;
5641dccf 2use nihav_core::demuxers::DemuxerError::*;
ce52b3b5
KS
3use std::io::SeekFrom;
4use std::mem;
418bd0b8 5use std::fmt;
ce52b3b5
KS
6
7macro_rules! mktag {
8 ($a:expr, $b:expr, $c:expr, $d:expr) => ({
9 (($a as u32) << 24) | (($b as u32) << 16) | (($c as u32) << 8) | ($d as u32)
10 });
11 ($arr:expr) => ({
12 (($arr[0] as u32) << 24) | (($arr[1] as u32) << 16) | (($arr[2] as u32) << 8) | ($arr[3] as u32)
13 });
14}
15
87c1d476
KS
16trait ReadSize {
17 fn read_size(&mut self, ver: u16) -> ByteIOResult<u64>;
18}
19
20impl<'a> ReadSize for ByteReader<'a> {
21 fn read_size(&mut self, ver: u16) -> ByteIOResult<u64> {
22 match ver {
23 0 => Ok(u64::from(self.read_u32be()?)),
24 2 => self.read_u64be(),
25 _ => unreachable!(),
26 }
27 }
28}
29
02cb1e03
KS
30const RM_SIPRO_BLOCK_SIZES: [usize; 4] = [ 29, 19, 37, 20 ];
31const RM_SIPRO_SWAPS: [[u8; 2]; 38] = [
32 [ 0, 63 ], [ 1, 22 ], [ 2, 44 ], [ 3, 90 ],
33 [ 5, 81 ], [ 7, 31 ], [ 8, 86 ], [ 9, 58 ],
34 [ 10, 36 ], [ 12, 68 ], [ 13, 39 ], [ 14, 73 ],
35 [ 15, 53 ], [ 16, 69 ], [ 17, 57 ], [ 19, 88 ],
36 [ 20, 34 ], [ 21, 71 ], [ 24, 46 ], [ 25, 94 ],
37 [ 26, 54 ], [ 28, 75 ], [ 29, 50 ], [ 32, 70 ],
38 [ 33, 92 ], [ 35, 74 ], [ 38, 85 ], [ 40, 56 ],
39 [ 42, 87 ], [ 43, 65 ], [ 45, 59 ], [ 48, 79 ],
40 [ 49, 93 ], [ 51, 89 ], [ 55, 95 ], [ 61, 76 ],
41 [ 67, 83 ], [ 77, 80 ]
42];
43
ce52b3b5
KS
44struct RMVideoStream {
45 frame: Vec<u8>,
46 hdr_size: usize,
47 frame_size: usize,
48 frame_pos: usize,
49}
50
51impl RMVideoStream {
52 fn new() -> Self {
53 RMVideoStream {
54 frame: Vec::new(),
55 hdr_size: 0,
56 frame_size: 0,
57 frame_pos: 0,
58 }
59 }
60 fn flush(&mut self) {
37952415 61 self.frame.clear();
ce52b3b5
KS
62 self.frame_size = 0;
63 self.frame_pos = 0;
64 }
65 fn start_slice(&mut self, num_slices: usize, frame_size: usize, data: &[u8]) {
66 self.hdr_size = num_slices * 8 + 1;
67 self.frame.resize(frame_size + self.hdr_size, 0);
68 self.frame[0] = (num_slices - 1) as u8;
69 self.frame_pos = 0;
d254ca17 70 self.add_slice(1, data).unwrap();
ce52b3b5 71 }
d254ca17
KS
72 fn add_slice(&mut self, slice_no: usize, data: &[u8]) -> DemuxerResult<()> {
73 validate!(self.hdr_size + self.frame_pos + data.len() <= self.frame.len());
ce52b3b5 74 self.write_slice_info(slice_no);
9037cf6b 75 let dslice = &mut self.frame[self.hdr_size + self.frame_pos..][..data.len()];
ce52b3b5
KS
76 dslice.copy_from_slice(data);
77 self.frame_pos += data.len();
d254ca17 78 Ok(())
ce52b3b5
KS
79 }
80 fn write_slice_info(&mut self, slice_no: usize) {
81 let off = 1 + (slice_no - 1) * 8;
82 self.frame[off + 0] = 0;
83 self.frame[off + 1] = 0;
84 self.frame[off + 2] = 0;
85 self.frame[off + 3] = 1;
86 self.frame[off + 4] = (self.frame_pos >> 24) as u8;
87 self.frame[off + 5] = (self.frame_pos >> 16) as u8;
88 self.frame[off + 6] = (self.frame_pos >> 8) as u8;
89 self.frame[off + 7] = (self.frame_pos >> 0) as u8;
90 }
91 fn get_frame_data(&mut self) -> Vec<u8> {
92 let mut v: Vec<u8> = Vec::new();
93 mem::swap(&mut v, &mut self.frame);
94 self.flush();
95 v
96 }
97}
98
99#[derive(Clone,Copy,PartialEq)]
100enum Deinterleaver {
101 None,
c19ce782 102 RA28_8,
ce52b3b5
KS
103 Generic,
104 Sipro,
105 VBR,
106}
107
108struct RMAudioStream {
109 deint: Deinterleaver,
c19ce782 110 iinfo: Option<InterleaveInfo>,
fe07b469
KS
111 buf: Vec<u8>,
112 sub_packet: usize,
ce52b3b5
KS
113}
114
c19ce782
KS
115const RM_ILEAVE_INT0: u32 = mktag!(b"Int0");
116const RM_ILEAVE_INT4: u32 = mktag!(b"Int4");
117const RM_ILEAVE_GENR: u32 = mktag!(b"genr");
118const RM_ILEAVE_SIPR: u32 = mktag!(b"sipr");
bb149b3b 119const RM_ILEAVE_VBRF: u32 = mktag!(b"vbrf");
c19ce782
KS
120const RM_ILEAVE_VBRS: u32 = mktag!(b"vbrs");
121
ce52b3b5 122impl RMAudioStream {
c19ce782
KS
123 fn new(iinfo: Option<InterleaveInfo>) -> Self {
124 let deint;
1f4aa1f4 125 let buf: Vec<u8>;
c19ce782
KS
126 if let Some(info) = iinfo {
127 deint = match info.id {
128 RM_ILEAVE_INT0 => Deinterleaver::None,
129 RM_ILEAVE_INT4 => Deinterleaver::RA28_8,
130 RM_ILEAVE_GENR => Deinterleaver::Generic,
131 RM_ILEAVE_SIPR => Deinterleaver::Sipro,
bb149b3b 132 RM_ILEAVE_VBRF => Deinterleaver::VBR,
c19ce782
KS
133 RM_ILEAVE_VBRS => Deinterleaver::VBR,
134 _ => {println!("unknown deint {:X}", info.id); Deinterleaver::None },
135 };
fe07b469
KS
136 match deint {
137 Deinterleaver::None => { buf = Vec::new(); },
6669c73b
KS
138 Deinterleaver::RA28_8 |
139 Deinterleaver::Generic |
140 Deinterleaver::Sipro => {
fe07b469 141 let bsize = (info.frame_size as usize) * (info.factor as usize);
e07387c7 142 buf = vec![0; bsize];
fe07b469 143 },
480d951e
KS
144 Deinterleaver::VBR => {
145 buf = Vec::new();
146 },
fe07b469 147 };
c19ce782
KS
148 } else {
149 deint = Deinterleaver::None;
fe07b469
KS
150 buf = Vec::new();
151 }
e07387c7 152 RMAudioStream { deint, iinfo, buf, sub_packet: 0 }
fe07b469 153 }
70910ac3 154 fn read_apackets(&mut self, queued_packets: &mut Vec<NAPacket>, src: &mut ByteReader, stream: NAStreamRef, ts: u32, keyframe: bool, payload_size: usize) -> DemuxerResult<NAPacket> {
8e14efb0 155 let ts = stream.make_ts(Some(ts as u64), None, None);
fe07b469
KS
156
157 if keyframe {
158 self.sub_packet = 0;
159 }
160 match self.deint {
6669c73b 161 Deinterleaver::None => { return src.read_packet(stream, ts, keyframe, payload_size); },
fe07b469
KS
162 Deinterleaver::RA28_8 => {
163 let iinfo = self.iinfo.unwrap();
164 let factor = iinfo.factor as usize;
165 let halffact = factor >> 1;
166 let fsize = iinfo.frame_size as usize;
167 let bsize = iinfo.block_size as usize;
168 let ppos = self.sub_packet;
169 for sb in 0..halffact {
6f263099
KS
170 let dst = &mut self.buf[sb * 2 * fsize + ppos * bsize..][..bsize];
171 src.read_buf(dst)?;
fe07b469
KS
172 }
173 self.sub_packet += 1;
174 if self.sub_packet == factor {
175 self.sub_packet = 0;
6669c73b 176 return Ok(NAPacket::new(stream, ts, true, self.buf.clone()));
fe07b469 177 } else {
6669c73b 178 return Err(DemuxerError::TryAgain);
fe07b469
KS
179 }
180 },
6669c73b
KS
181 Deinterleaver::Generic => {
182 let iinfo = self.iinfo.unwrap();
183 let factor = iinfo.factor as usize;
184 let fsize = iinfo.frame_size as usize;
185 let bsize = iinfo.block_size as usize;
186 let factor2 = fsize / bsize;
187 let ppos = self.sub_packet;
188
189 for sb in 0..factor2 {
190 let sb_pos = factor * sb + ((factor + 1) >> 1) * (ppos & 1) + (ppos >> 1);
6f263099
KS
191 let dst = &mut self.buf[bsize * sb_pos..][..bsize];
192 src.read_buf(dst)?;
6669c73b
KS
193 }
194 },
195 Deinterleaver::Sipro => {
196 let iinfo = self.iinfo.unwrap();
197 let fsize = iinfo.frame_size as usize;
198 let ppos = self.sub_packet;
199
6f263099
KS
200 let dst = &mut self.buf[fsize * ppos..][..fsize];
201 src.read_buf(dst)?;
6669c73b 202 },
480d951e
KS
203 Deinterleaver::VBR => {
204 validate!(payload_size >= 5);
205 let hdrsizesize = src.read_u16be()?;
206 let num_entries = (hdrsizesize / 16) as usize;
207 validate!(payload_size >= num_entries * 3 + 2);
208 let mut sizes: Vec<usize> = Vec::with_capacity(num_entries);
209 let mut tot_size = 0;
210 for _ in 0..num_entries {
211 let sz = src.read_u16be()? as usize;
212 tot_size += sz;
213 sizes.push(sz);
214 }
215 validate!(tot_size + num_entries * 2 + 2 == payload_size);
8e14efb0 216 let pkt_ts = stream.make_ts(None, None, None);
480d951e
KS
217 let mut first = true;
218 for size in sizes.iter() {
219 let cur_ts = if first { ts } else { pkt_ts };
220 first = false;
221 let pkt = src.read_packet(stream.clone(), cur_ts, true, *size)?;
222 queued_packets.push(pkt);
223 }
224 queued_packets.reverse();
225 let pkt0 = queued_packets.pop().unwrap();
226 return Ok(pkt0);
227 },
6669c73b
KS
228 };
229
230 let iinfo = self.iinfo.unwrap();
231 let factor = iinfo.factor as usize;
5dfd175c 232 let fsize = if iinfo.block_size != 0 { iinfo.block_size } else { iinfo.frame_size } as usize;
6669c73b
KS
233
234 self.sub_packet += 1;
235 if self.sub_packet < factor {
236 return Err(DemuxerError::TryAgain);
237 }
238
239 self.sub_packet = 0;
240
241 if self.deint == Deinterleaver::Sipro {
76c275c9 242 sipro_restore(&mut self.buf, factor, iinfo.frame_size as usize);
6669c73b
KS
243 }
244
245 let mut frames_iter = self.buf.chunks(fsize);
246 let pkt0 = frames_iter.next().unwrap();
247
8e14efb0 248 let pkt_ts = stream.make_ts(None, None, None);
6669c73b
KS
249 for pkts in frames_iter {
250 let pkt = NAPacket::new(stream.clone(), pkt_ts, true, pkts.to_vec());
251 queued_packets.push(pkt);
c19ce782 252 }
6669c73b
KS
253 queued_packets.reverse();
254 Ok(NAPacket::new(stream, ts, true, pkt0.to_vec()))
ce52b3b5
KS
255 }
256}
257
02cb1e03
KS
258fn sipro_restore(buf: &mut [u8], factor: usize, fsize: usize) {
259 let stride = factor * fsize * 2 / 96;
260 for i in 0..38 {
261 let mut sidx = (RM_SIPRO_SWAPS[i][0] as usize) * stride;
262 let mut didx = (RM_SIPRO_SWAPS[i][1] as usize) * stride;
263 for _ in 0..stride {
264 let in0 = buf[sidx >> 1];
265 let in1 = buf[didx >> 1];
266 let nib0 = (in0 >> ((sidx & 1) * 4)) & 0xF;
267 let nib1 = (in1 >> ((didx & 1) * 4)) & 0xF;
268
269 buf[didx >> 1] = (nib0 << (4 * (didx & 1))) | (in1 & (0xF << (4 * (!didx & 1))));
270 buf[sidx >> 1] = (nib1 << (4 * (sidx & 1))) | (in0 & (0xF << (4 * (!sidx & 1))));
271
272 sidx += 1;
273 didx += 1;
274 }
275 }
276}
277
d4df9ee1
KS
278struct SubstreamInfo {
279 id: u32,
280 map: Vec<u16>,
281 str_ids: Vec<u32>,
282}
283
284struct MLTIMapper {
285 sub_info: Vec<SubstreamInfo>,
418bd0b8 286 sstr_id: u32,
d4df9ee1
KS
287}
288
289impl MLTIMapper {
290 fn new() -> Self {
291 MLTIMapper {
292 sub_info: Vec::new(),
418bd0b8 293 sstr_id: 0x10000,
d4df9ee1
KS
294 }
295 }
296 fn add_stream(&mut self, stream_no: u32) {
297 let ssinfo = SubstreamInfo { id: stream_no, map: Vec::new(), str_ids: Vec::new() };
298 self.sub_info.push(ssinfo);
299 }
418bd0b8
KS
300 fn get_substream_no(&self) -> u32 {
301 self.sstr_id - 1
302 }
d4df9ee1
KS
303 fn find_idx(&self, stream_no: u32) -> Option<usize> {
304 self.sub_info.iter().position(|x| x.id == stream_no)
305 }
418bd0b8
KS
306 fn add_map_rule(&mut self, map_ss: u16) {
307 let idx = self.sub_info.len() - 1;
d4df9ee1
KS
308 self.sub_info[idx].map.push(map_ss);
309 }
418bd0b8
KS
310 fn add_substream(&mut self) {
311 let idx = self.sub_info.len() - 1;
312 self.sub_info[idx].str_ids.push(self.sstr_id);
313 self.sstr_id += 1;
d4df9ee1
KS
314 }
315 fn is_mlti_stream(&self, stream_no: u32) -> bool {
316 self.find_idx(stream_no).is_some()
317 }
318 fn find_substream(&self, stream_no: u32, grp: u16) -> Option<u32> {
319 if let Some(idx) = self.find_idx(stream_no) {
320 if (grp as usize) < self.sub_info[idx].map.len() {
321 let sub_id = self.sub_info[idx].map[grp as usize] as usize;
322 if sub_id < self.sub_info[idx].str_ids.len() {
323 return Some(self.sub_info[idx].str_ids[sub_id]);
324 }
325 }
326 }
327 None
328 }
329}
330
ce52b3b5
KS
331enum RMStreamType {
332 Audio(RMAudioStream),
333 Video(RMVideoStream),
334 Logical,
335 Unknown,
336}
337
418bd0b8
KS
338struct CommonStreamData {
339 streams: Vec<RMStreamType>,
340 str_ids: Vec<u32>,
341 mlti_mapper: MLTIMapper,
342}
343
344impl CommonStreamData {
345 fn new() -> Self {
346 CommonStreamData {
347 streams: Vec::new(),
348 str_ids: Vec::new(),
349 mlti_mapper: MLTIMapper::new(),
350 }
351 }
352 fn get_stream_id(&self, str_no: u32, pkt_grp: u16) -> u32 {
353 if !self.mlti_mapper.is_mlti_stream(str_no) {
354 str_no
355 } else {
356 self.mlti_mapper.find_substream(str_no, pkt_grp).unwrap()
357 }
358 }
359 fn find_stream(&self, stream_id: u32) -> Option<usize> {
360 self.str_ids.iter().position(|x| *x == stream_id)
361 }
362}
363
ce52b3b5
KS
364struct RealMediaDemuxer<'a> {
365 src: &'a mut ByteReader<'a>,
366 data_pos: u64,
cdcbaa6c 367 data_end: u64,
770bef6f 368 next_data: u64,
db5cc44b 369 data_ver: u16,
ce52b3b5
KS
370 num_packets: u32,
371 cur_packet: u32,
372
418bd0b8 373 str_data: CommonStreamData,
ce52b3b5 374
cdcbaa6c
KS
375 data_chunks: Vec<(u64, u32, u16)>,
376 cur_data_chunk: usize,
377
ce52b3b5
KS
378 queued_pkts: Vec<NAPacket>,
379 slice_buf: Vec<u8>,
380}
381
382fn find_codec_name(registry: &[(&[u8;4], &'static str)], fcc: u32) -> &'static str {
383 for &(fourcc, name) in registry {
384 if mktag!(fourcc) == fcc { return name; }
385 }
386 "unknown"
387}
388
389fn read_14or30(src: &mut ByteReader) -> DemuxerResult<(bool, u32)> {
390 let tmp = src.read_u16be()?;
391 let flag = (tmp & 0x8000) != 0;
392 if (tmp & 0x4000) == 0x4000 {
393 Ok((flag, ((tmp & 0x3FFF) as u32)))
394 } else {
395 let val = ((tmp as u32) << 16) | (src.read_u16be()? as u32);
396 Ok((flag, val & 0x3FFFFFFF))
397 }
398}
399
70910ac3 400fn read_video_buf(src: &mut ByteReader, stream: NAStreamRef, ts: u32, keyframe: bool, frame_size: usize) -> DemuxerResult<NAPacket> {
e6aaad5c 401 let size = frame_size + 9;
e07387c7 402 let mut vec: Vec<u8> = vec![0; size];
ce52b3b5
KS
403 //v[0] = 0; // 1 slice
404 vec[4] = 1;
405 src.read_buf(&mut vec[9..])?;
406
8e14efb0 407 let ts = stream.make_ts(Some(ts as u64), None, None);
ce52b3b5
KS
408 Ok(NAPacket::new(stream, ts, keyframe, vec))
409}
410
70910ac3 411fn read_multiple_frame(src: &mut ByteReader, stream: NAStreamRef, keyframe: bool, skip_mtype: bool) -> DemuxerResult<NAPacket> {
ce52b3b5
KS
412 if !skip_mtype {
413 let mtype = src.read_byte()?;
414 validate!(mtype == 0xC0);
415 }
416 let (_, frame_size) = read_14or30(src)?;
417 let (_, timestamp) = read_14or30(src)?;
c9aba47e 418 let _seq_no = src.read_byte()?;
ce52b3b5
KS
419
420 read_video_buf(src, stream, timestamp, keyframe, frame_size as usize)
421}
422
418bd0b8
KS
423struct RMDemuxCommon {}
424
425impl RMDemuxCommon {
a480a0de 426 fn parse_stream_info(str_data: &mut CommonStreamData, strmgr: &mut StreamManager, stream_no: u32, edata: &[u8], duration: u32) -> DemuxerResult<bool> {
418bd0b8 427 let mut is_mlti = false;
fa57381e 428 let mut mr = MemoryReader::new_read(edata);
418bd0b8
KS
429 let mut src = ByteReader::new(&mut mr);
430 let tag = src.read_u32be()?;
431 let tag2 = src.peek_u32be()?;
418bd0b8 432 if tag == mktag!('.', 'r', 'a', 0xFD) {
a480a0de 433 Self::parse_audio_stream(strmgr, &mut str_data.streams, stream_no, &mut src, edata, duration)?;
418bd0b8 434 } else if ((tag2 == mktag!('V', 'I', 'D', 'O')) || (tag2 == mktag!('I', 'M', 'A', 'G'))) && ((tag as usize) <= edata.len()) {
a480a0de 435 Self::parse_video_stream(strmgr, &mut str_data.streams, stream_no, &mut src, edata, tag2, duration)?;
418bd0b8 436 } else if tag == mktag!(b"LSD:") {
e07387c7 437 let extradata = Some(edata.to_owned());
418bd0b8
KS
438
439 src.read_skip(4)?; //version
440 let channels = src.read_u16be()?;
441 let samp_size = src.read_u16be()?;
442 let sample_rate = src.read_u32be()?;
443
418bd0b8
KS
444 let soniton = NASoniton::new(samp_size as u8, SONITON_FLAG_SIGNED);
445 let ahdr = NAAudioInfo::new(sample_rate, channels as u8, soniton, 1);
446 let nainfo = NACodecInfo::new("ralf", NACodecTypeInfo::Audio(ahdr), extradata);
e6aaad5c 447 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, stream_no, nainfo, 1, 1000, u64::from(duration)));
418bd0b8
KS
448 if res.is_none() { return Err(MemoryError); }
449 let astr = RMAudioStream::new(None);
450 str_data.streams.push(RMStreamType::Audio(astr));
451 } else if tag == mktag!(b"MLTI") {
452 is_mlti = true;
453 let num_rules = src.read_u16be()? as usize;
454 let mut max_sub = 0;
455 str_data.mlti_mapper.add_stream(stream_no);
456 for _ in 0..num_rules {
457 let substr = src.read_u16be()?;
458 max_sub = max_sub.max(substr);
459 str_data.mlti_mapper.add_map_rule(substr);
460 }
461 let num_substreams = src.read_u16be()? as usize;
462 validate!(num_substreams > (max_sub as usize));
463 for _ in 0..num_substreams {
464 let hdr_size = src.read_u32be()? as usize;
465 validate!(hdr_size > 8);
466 let pos = src.tell() as usize;
467 src.read_skip(hdr_size)?;
468 str_data.mlti_mapper.add_substream();
469 {
470 let hdrsrc = &edata[pos..][..hdr_size];
471 let mut mr = MemoryReader::new_read(hdrsrc);
472 let mut hsrc = ByteReader::new(&mut mr);
473
474 let tag = hsrc.read_u32be()?;
475 let tag2 = hsrc.peek_u32be()?;
476 let stream_no = str_data.mlti_mapper.get_substream_no();
477//todo check that all substreams are of the same type");
478 if tag == mktag!('.', 'r', 'a', 0xFD) {
a480a0de 479 Self::parse_audio_stream(strmgr, &mut str_data.streams, stream_no, &mut hsrc, hdrsrc, duration)?;
418bd0b8 480 } else if (tag2 == mktag!('V', 'I', 'D', 'O')) && ((tag as usize) <= hdr_size) {
a480a0de 481 Self::parse_video_stream(strmgr, &mut str_data.streams, stream_no, &mut hsrc, hdrsrc, tag2, duration)?;
418bd0b8
KS
482 } else {
483println!("unknown MLTI substream {:08X} / {:08X}", tag, tag2);
484 return Err(DemuxerError::InvalidData);
485 }
486 str_data.str_ids.push(stream_no);
487 }
488 }
489 } else {
490 str_data.streams.push(RMStreamType::Logical);
491 }
492 Ok(is_mlti)
493 }
a480a0de 494 fn parse_audio_stream(strmgr: &mut StreamManager, streams: &mut Vec<RMStreamType>, stream_no: u32, src: &mut ByteReader, edata_: &[u8], duration: u32) -> DemuxerResult<()> {
418bd0b8
KS
495 let ver = src.read_u16be()?;
496 let ainfo = match ver {
497 3 => {
498 parse_aformat3(src)?
499 },
500 4 => {
501 parse_aformat4(src)?
502 },
503 5 => {
504 parse_aformat5(src)?
505 },
506 _ => {
507 println!("unknown version {}", ver);
508 return Err(DemuxerError::InvalidData);
509 },
510 };
418bd0b8
KS
511 let cname = find_codec_name(RM_AUDIO_CODEC_REGISTER, ainfo.fcc);
512 let blk_size = if ainfo.fcc != mktag!(b"sipr") {
e614e5e6
KS
513 if let Some(ref iinfo) = ainfo.ileave_info {
514 iinfo.block_size as usize
515 } else {
516 ainfo.granularity as usize
517 }
418bd0b8
KS
518 } else {
519 validate!(ainfo.flavor <= 3);
520 RM_SIPRO_BLOCK_SIZES[ainfo.flavor as usize]
521 };
522 let srate = ainfo.sample_rate;
523 let soniton = NASoniton::new(ainfo.sample_size as u8, SONITON_FLAG_SIGNED);
524 let ahdr = NAAudioInfo::new(srate, ainfo.channels as u8, soniton, blk_size);
525 let extradata = if ainfo.edata_size == 0 {
526 None
527 } else {
528 let eslice = &edata_[(src.tell() as usize)..];
529 Some(eslice.to_vec())
530 };
a480a0de 531 let duration = if duration == 0 { ainfo.get_duration(1000) } else { u64::from(duration) };
418bd0b8 532 let nainfo = NACodecInfo::new(cname, NACodecTypeInfo::Audio(ahdr), extradata);
e6aaad5c 533 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, stream_no, nainfo, 1, 1000, duration));
418bd0b8
KS
534 if res.is_none() { return Err(MemoryError); }
535
536 let astr = RMAudioStream::new(ainfo.ileave_info);
537 streams.push(RMStreamType::Audio(astr));
538 Ok(())
539 }
540#[allow(unused_variables)]
a480a0de 541 fn parse_video_stream(strmgr: &mut StreamManager, streams: &mut Vec<RMStreamType>, stream_no: u32, src: &mut ByteReader, edata_: &[u8], tag2: u32, duration: u32) -> DemuxerResult<()> {
418bd0b8
KS
542 src.read_skip(4)?;
543 let fcc = src.read_u32be()?;
544 let width = src.read_u16be()? as usize;
545 let height = src.read_u16be()? as usize;
546 let bpp = src.read_u16be()?;
547 let pad_w = src.read_u16be()?;
548 let pad_h = src.read_u16be()?;
59dc9182 549 let _fps;
418bd0b8 550 if tag2 == mktag!('V', 'I', 'D', 'O') {
59dc9182 551 _fps = src.read_u32be()?;
418bd0b8 552 } else {
59dc9182 553 _fps = 0x10000;
418bd0b8
KS
554 }
555 let extradata: Option<Vec<u8>>;
556 if src.left() > 0 {
557 let eslice = &edata_[(src.tell() as usize)..];
558 extradata = Some(eslice.to_vec());
559 } else {
560 extradata = None;
561 }
562 let cname = find_codec_name(RM_VIDEO_CODEC_REGISTER, fcc);
563
564 let vhdr = NAVideoInfo::new(width, height, false, RGB24_FORMAT);
565 let vinfo = NACodecInfo::new(cname, NACodecTypeInfo::Video(vhdr), extradata);
e6aaad5c 566 let res = strmgr.add_stream(NAStream::new(StreamType::Video, stream_no, vinfo, 1, 1000, u64::from(duration)));
418bd0b8
KS
567 if res.is_none() { return Err(DemuxerError::MemoryError); }
568
569 let vstr = RMVideoStream::new();
570 streams.push(RMStreamType::Video(vstr));
571 Ok(())
572 }
573#[allow(unused_variables)]
e6aaad5c 574 #[allow(clippy::question_mark)]
70910ac3 575 fn parse_packet_payload(src: &mut ByteReader, rmstream: &mut RMStreamType, stream: NAStreamRef, slice_buf: &mut Vec<u8>, queued_pkts: &mut Vec<NAPacket>, keyframe: bool, ts: u32, payload_size: usize) -> DemuxerResult<NAPacket> {
418bd0b8
KS
576 match rmstream {
577 RMStreamType::Video(ref mut vstr) => {
578
579 let pos = src.tell();
580 let b0 = src.read_byte()?;
581 match b0 >> 6 {
582 0 => { // partial frame
583 let b1 = src.read_byte()?;
584 let hdr1 = ((b0 as u16) << 8) | (b1 as u16);
585 let num_pkts = ((hdr1 >> 7) & 0x7F) as usize;
586 let packet_num = hdr1 & 0x7F;
587 let (_, frame_size) = read_14or30(src)?;
588 let (_, off) = read_14or30(src)?;
589 let seq_no = src.read_byte()?;
418bd0b8
KS
590 let hdr_skip = (src.tell() - pos) as usize;
591
e6aaad5c 592 let slice_size = payload_size - hdr_skip;
418bd0b8
KS
593 slice_buf.resize(slice_size, 0);
594 src.read_buf(slice_buf.as_mut_slice())?;
595 if packet_num == 1 {
596 vstr.start_slice(num_pkts, frame_size as usize, slice_buf.as_slice());
597 } else {
d254ca17 598 vstr.add_slice(packet_num as usize, slice_buf.as_slice())?;
418bd0b8
KS
599 }
600 if (packet_num as usize) < num_pkts {
601 return Err(DemuxerError::TryAgain);
602 }
603 //todo: check if full frame is received
8e14efb0 604 let ts = stream.make_ts(Some(ts as u64), None, None);
418bd0b8
KS
605 let pkt = NAPacket::new(stream, ts, keyframe, vstr.get_frame_data());
606 Ok(pkt)
607 },
608 1 => { // whole frame
609 let seq_no = src.read_byte()?;
418bd0b8
KS
610 read_video_buf(src, stream, ts, keyframe, payload_size - 2)
611 },
612 2 => { // last partial frame
613 let b1 = src.read_byte()?;
614 let hdr1 = ((b0 as u16) << 8) | (b1 as u16);
615 let num_pkts = ((hdr1 >> 7) & 0x7F) as usize;
616 let packet_num = hdr1 & 0x7F;
617 let (_, frame_size) = read_14or30(src)?;
618 let (_, tail_size) = read_14or30(src)?;
619 let seq_no = src.read_byte()?;
418bd0b8
KS
620 slice_buf.resize(tail_size as usize, 0);
621 src.read_buf(slice_buf.as_mut_slice())?;
622 if packet_num == 1 && frame_size == tail_size {
623 vstr.start_slice(num_pkts, frame_size as usize, slice_buf.as_slice());
624 } else {
d254ca17 625 vstr.add_slice(packet_num as usize, slice_buf.as_slice())?;
418bd0b8
KS
626 }
627
628 while src.tell() < pos + (payload_size as u64) {
629 let res = read_multiple_frame(src, stream.clone(), false, false);
630 if res.is_err() { break; }
631 queued_pkts.push(res.unwrap());
632 }
633 queued_pkts.reverse();
8e14efb0 634 let ts = stream.make_ts(Some(ts as u64), None, None);
418bd0b8
KS
635 let pkt = NAPacket::new(stream, ts, keyframe, vstr.get_frame_data());
636 Ok(pkt)
637 },
638 _ => { // multiple frames
418bd0b8
KS
639 let res = read_multiple_frame(src, stream.clone(), keyframe, true);
640 if res.is_err() { return res; }
641 while src.tell() < pos + (payload_size as u64) {
642 let res = read_multiple_frame(src, stream.clone(), false, false);
643 if res.is_err() { break; }
644 queued_pkts.push(res.unwrap());
645 }
646 queued_pkts.reverse();
647 res
648 },
649 }
650 },
651 RMStreamType::Audio(ref mut astr) => {
652 astr.read_apackets(queued_pkts, src, stream, ts, keyframe, payload_size)
653 },
654 _ => {
655 src.read_skip(payload_size)?;
656 Err(DemuxerError::InvalidData)
657 },
658 }
659 }
660}
661
ce52b3b5 662impl<'a> DemuxCore<'a> for RealMediaDemuxer<'a> {
cdcbaa6c
KS
663 fn open(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
664 self.read_header(strmgr, seek_idx)?;
ce52b3b5
KS
665 Ok(())
666 }
667
668#[allow(unused_variables)]
669 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
670 if !self.queued_pkts.is_empty() {
671 let pkt = self.queued_pkts.pop().unwrap();
672 return Ok(pkt);
673 }
674 loop {
cdcbaa6c
KS
675 if (self.cur_packet >= self.num_packets) || (self.src.tell() >= self.data_end) {
676 self.cur_data_chunk += 1;
677 if self.cur_data_chunk < self.data_chunks.len() {
678 let (pos, _, _) = self.data_chunks[self.cur_data_chunk];
679 self.src.seek(SeekFrom::Start(pos))?;
770bef6f
KS
680 let res = read_chunk(self.src);
681 if let Ok((id, size, ver)) = res {
682 self.data_pos = self.src.tell();
683 self.data_ver = ver;
cdcbaa6c 684 self.data_end = self.data_pos + (size as u64);
770bef6f
KS
685 if self.parse_data_start().is_ok() {
686 continue;
687 }
688 }
689 }
690 return Err(DemuxerError::EOF);
691 }
ce52b3b5
KS
692
693 let pkt_start = self.src.tell();
7348405b 694 let ver = self.src.read_u16be()?;
ce52b3b5
KS
695 validate!(ver <= 1);
696 let len = self.src.read_u16be()? as usize;
d4df9ee1 697 let str_no = self.src.read_u16be()? as u32;
ce52b3b5 698 let ts = self.src.read_u32be()?;
d4df9ee1 699 let pkt_grp;
770bef6f 700 let flags;
ce52b3b5 701 if ver == 0 {
d4df9ee1 702 pkt_grp = self.src.read_byte()? as u16;
770bef6f 703 flags = self.src.read_byte()?;
ce52b3b5 704 } else {
7348405b 705 //asm_rule = self.src.read_u16be()?;
d4df9ee1 706 //asm_flags = self.src.read_byte()?;
ce52b3b5 707 self.src.read_skip(2)?;
d4df9ee1 708 pkt_grp = 0;
770bef6f
KS
709 self.src.read_skip(1)?;
710 flags = 0;
ce52b3b5 711 }
ce52b3b5 712 let hdr_size = self.src.tell() - pkt_start;
d4df9ee1 713//println!("packet @{:X} size {} for {} ts {} grp {} flags {:X}", pkt_start, len, str_no, ts, pkt_grp, flags);
ce52b3b5
KS
714 self.cur_packet += 1;
715
716 let payload_size = len - (hdr_size as usize);
717
e6aaad5c 718 let stream_id = self.str_data.get_stream_id(str_no, pkt_grp);
418bd0b8 719 let sr = self.str_data.find_stream(stream_id);
ce52b3b5 720 if sr.is_none() {
ce52b3b5
KS
721 self.src.read_skip(payload_size)?;
722 return Err(DemuxerError::InvalidData);
723 }
724 let str_id = sr.unwrap();
d4df9ee1
KS
725
726 let streamres = strmgr.get_stream_by_id(stream_id);
c9aba47e
KS
727 if streamres.is_none() {
728 self.src.read_skip(payload_size)?;
729 continue;
730 }
731 let stream = streamres.unwrap();
d4df9ee1 732 if strmgr.is_ignored_id(stream_id) {
ce52b3b5
KS
733 self.src.read_skip(payload_size)?;
734 continue;
735 }
736 //todo skip unwanted packet
737 let keyframe = (flags & KEYFRAME_FLAG) != 0;
738
6f263099 739 let ret = RMDemuxCommon::parse_packet_payload(self.src, &mut self.str_data.streams[str_id], stream, &mut self.slice_buf, &mut self.queued_pkts, keyframe, ts, payload_size);
418bd0b8
KS
740 if let Err(DemuxerError::TryAgain) = ret {
741 continue;
742 } else {
743 return ret;
744 }
ce52b3b5
KS
745 }
746 }
747
748 #[allow(unused_variables)]
24d99894 749 fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()> {
cdcbaa6c
KS
750 self.queued_pkts.clear();
751 let ret = seek_idx.find_pos(time);
752 if ret.is_none() {
753 return Err(DemuxerError::SeekError);
754 }
755 let ret = ret.unwrap();
756 let seek_pos = ret.pos;
757 for (pos, size, ver) in self.data_chunks.iter() {
758 if seek_pos < *pos { continue; }
759 let end = *pos + (*size as u64);
760 if seek_pos < end {
761 self.cur_packet = 0;
762 self.data_pos = seek_pos;
763 self.data_ver = *ver;
764 self.data_end = end;
765 self.src.seek(SeekFrom::Start(seek_pos))?;
766 return Ok(());
767 }
768 }
769 Err(DemuxerError::SeekError)
ce52b3b5 770 }
a480a0de
KS
771
772 fn get_duration(&self) -> u64 { 0 }
ce52b3b5
KS
773}
774
787b8d03
KS
775impl<'a> NAOptionHandler for RealMediaDemuxer<'a> {
776 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
777 fn set_options(&mut self, _options: &[NAOption]) { }
778 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
779}
780
ce52b3b5
KS
781fn read_chunk(src: &mut ByteReader) -> DemuxerResult<(u32, u32, u16)> {
782 let id = src.read_u32be()?;
6132225f 783if id == 0 { return Ok((0, 0, 0)); }
ce52b3b5 784 let size = src.read_u32be()?;
9f45e528
KS
785if size == 0 {
786 let ver = src.read_u16be()?;
db5cc44b 787 validate!(ver <= 2);
9f45e528
KS
788 return Ok((id, 0x0FFFFFFF, ver));
789}
ce52b3b5
KS
790 validate!(size >= 10);
791 let ver = src.read_u16be()?;
db5cc44b 792 validate!(ver <= 2);
ce52b3b5
KS
793 Ok((id, size, ver))
794}
795
c19ce782
KS
796#[derive(Clone,Copy,Debug)]
797struct InterleaveInfo {
798 id: u32,
799 factor: u16,
800 block_size: u16,
801 frame_size: u16,
802}
803
804#[derive(Clone,Copy,Debug)]
3518aa85 805#[allow(dead_code)]
c19ce782
KS
806struct RealAudioInfo {
807 fcc: u32,
808 sample_rate: u32,
809 sample_size: u16,
02cb1e03 810 flavor: u16,
c19ce782
KS
811 channels: u16,
812 channel_mask: u32,
813 granularity: u32,
814 bytes_per_minute: u32,
815 total_bytes: u32,
816 edata_size: u32,
817 ileave_info: Option<InterleaveInfo>
818}
819
a480a0de
KS
820impl RealAudioInfo {
821 fn get_duration(&self, base: u32) -> u64 {
822 if self.bytes_per_minute != 0 {
823 u64::from(self.total_bytes) * 60 * u64::from(base) / u64::from(self.bytes_per_minute)
824 } else {
825 0
826 }
827 }
828}
829
c19ce782
KS
830fn skip_ra_metadata(src: &mut ByteReader) -> DemuxerResult<()> {
831 let title_len = src.read_byte()? as usize;
832 src.read_skip(title_len)?;
833 let author_len = src.read_byte()? as usize;
834 src.read_skip(author_len)?;
835 let copywrong_len = src.read_byte()? as usize;
836 src.read_skip(copywrong_len)?;
837 let comment_len = src.read_byte()? as usize;
838 src.read_skip(comment_len)?;
839 Ok(())
840}
841
842#[allow(unused_variables)]
843fn parse_aformat3(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
844 let start = src.tell();
845 let header_len = src.read_u16be()?;
846 validate!(header_len >= 24);
847 let flavor = src.read_u16be()?;
848 let granularity = src.read_u32be()?;
849 let bytes_per_minute = src.read_u32be()?;
850 let total_bytes = src.read_u32be()?;
851
852 skip_ra_metadata(src)?;
853
0d9dc164
KS
854 // the original RealAudio has no such fields
855 let fcc = if src.tell() != start + u64::from(header_len) + 2 {
856 let _can_copy = src.read_byte()?;
857 let fcc_len = src.read_byte()?;
858 validate!(fcc_len == 0 || fcc_len == 4);
859 src.read_u32be()?
860 } else {
861 read_u32be(b"lpcJ")?
862 };
c19ce782
KS
863
864 let end = src.tell();
865 validate!(end - start <= (header_len as u64) + 2);
866
867 Ok(RealAudioInfo {
e07387c7 868 fcc, flavor,
02cb1e03 869 sample_rate: 8000, sample_size: 16, channels: 1, channel_mask: 0,
e07387c7
KS
870 granularity, bytes_per_minute,
871 total_bytes, edata_size: 0,
c19ce782
KS
872 ileave_info: None,
873 })
874}
875
876#[allow(unused_variables)]
877fn parse_aformat4(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
878 let start = src.tell();
879 src.read_skip(2)?; // zeroes
880 let id = src.read_u32be()?;
881 validate!(id == mktag!(b".ra4"));
882 let data_size = src.read_u32be()?;
883 let _ver4 = src.read_u16be()?; // should be 4
884 let header_size = src.read_u32be()?;
02cb1e03 885 let flavor = src.read_u16be()?;
c19ce782
KS
886 let granularity = src.read_u32be()?;
887 let total_bytes = src.read_u32be()?;
888 let bytes_per_minute = src.read_u32be()?;
889 let _bytes_per_minute2 = src.read_u32be()?;
890 let ileave_factor = src.read_u16be()?;
891 let ileave_block_size = src.read_u16be()?;
892 let _user_data = src.read_u16be()?;
893 let sample_rate = src.read_u32be()?;
894 let sample_size = src.read_u32be()?;
895 let channels = src.read_u16be()?;
896 let interleaver_id_len = src.read_byte()?;
897 validate!(interleaver_id_len == 4);
898 let interleaver_id = src.read_u32be()?;
899 let fcc_len = src.read_byte()?;
900 validate!(fcc_len == 4);
901 let fcc = src.read_u32be()?;
902 let is_interleaved = src.read_byte()?;
903 let _can_copy = src.read_byte()?;
904 let _stream_type = src.read_byte()?;
905
906 skip_ra_metadata(src)?;
907
908 let end = src.tell();
909 validate!(end - start <= (header_size as u64) + 10);
910
911 let ileave_info = if is_interleaved != 0 {
912 Some(InterleaveInfo {
fe07b469
KS
913 id: interleaver_id, factor: ileave_factor, block_size: granularity as u16,
914 frame_size: ileave_block_size,
c19ce782
KS
915 })
916 } else {
917 None
918 };
919
920 Ok(RealAudioInfo {
e07387c7
KS
921 fcc, flavor,
922 sample_rate, sample_size: sample_size as u16, channels, channel_mask: 0,
923 granularity, bytes_per_minute,
c19ce782 924 total_bytes: total_bytes & 0xFFFFFF, edata_size: 0,
e07387c7 925 ileave_info,
c19ce782
KS
926 })
927}
928
929#[allow(unused_variables)]
930fn parse_aformat5(src: &mut ByteReader) -> DemuxerResult<RealAudioInfo> {
931 let start = src.tell();
932 src.read_skip(2)?; // zeroes
933 let id = src.read_u32be()?;
db5cc44b 934 validate!((id == mktag!(b".ra5")) || (id == mktag!(b".ra4")));
c19ce782
KS
935 let data_size = src.read_u32be()?;
936 let _ver5 = src.read_u16be()?; // should be 5
937 let header_size = src.read_u32be()?;
02cb1e03 938 let flavor = src.read_u16be()?;
c19ce782
KS
939 let granularity = src.read_u32be()?;
940 let total_bytes = src.read_u32be()?;
941 let bytes_per_minute = src.read_u32be()?;
942 let _bytes_per_minute2 = src.read_u32be()?;
943 let ileave_factor = src.read_u16be()?;
c19ce782 944 let frame_size = src.read_u16be()?;
6669c73b 945 let ileave_block_size = src.read_u16be()?;
c19ce782
KS
946 let user_data = src.read_u32be()?;
947 let _sample_rate1 = src.read_u16be()?;
948 let sample_rate = src.read_u32be()?;
949 let sample_size = src.read_u32be()?;
950 let channels = src.read_u16be()?;
951 let interleaver_id = src.read_u32be()?;
952 let fcc = src.read_u32be()?;
953 let is_interleaved = src.read_byte()?;
954 let _can_copy = src.read_byte()?;
955 let _stream_type = src.read_byte()?;
956 let has_ileave_pattern = src.read_byte()?;
957 if has_ileave_pattern != 0 {
958unimplemented!("ra5 interleave pattern");
959 }
843c8c87 960 let mut edata_size = src.read_u32be()?;
c19ce782 961 let end = src.tell();
db5cc44b
KS
962 if id == mktag!(b".ra5") {
963 validate!(end - start <= (header_size as u64) + 10);
bb149b3b 964// src.read_skip(((end - start) as usize) - (header_size as usize) - 10)?;
db5cc44b
KS
965 } else {
966 validate!(end - start <= (header_size as u64) + 15);
967 }
c19ce782
KS
968
969 let ileave_info = if is_interleaved != 0 {
970 Some(InterleaveInfo {
e07387c7 971 id: interleaver_id, factor: ileave_factor, block_size: ileave_block_size, frame_size,
c19ce782
KS
972 })
973 } else {
974 None
975 };
843c8c87
KS
976 if (fcc == mktag!(b"raac")) || (fcc == mktag!(b"racp")) {
977 validate!(edata_size > 1);
978 edata_size -= 1;
979 src.read_skip(1)?;
980 }
c19ce782
KS
981
982 Ok(RealAudioInfo {
e07387c7
KS
983 fcc, flavor,
984 sample_rate, sample_size: sample_size as u16, channels, channel_mask: 0,
985 granularity, bytes_per_minute,
986 total_bytes: total_bytes & 0xFFFFFF, edata_size,
987 ileave_info,
c19ce782
KS
988 })
989}
990
ce52b3b5
KS
991const RMVB_HDR_SIZE: u32 = 18;
992const RMVB_PROP_SIZE: u32 = 50;
993const KEYFRAME_FLAG: u8 = 0x02;
994
995impl<'a> RealMediaDemuxer<'a> {
996 fn new(io: &'a mut ByteReader<'a>) -> Self {
997 RealMediaDemuxer {
998 src: io,
999 data_pos: 0,
cdcbaa6c 1000 data_end: 0,
770bef6f 1001 next_data: 0,
db5cc44b 1002 data_ver: 0,
ce52b3b5
KS
1003 num_packets: 0,
1004 cur_packet: 0,
418bd0b8 1005 str_data: CommonStreamData::new(),
cdcbaa6c
KS
1006 data_chunks: Vec::new(),
1007 cur_data_chunk: 0,
ce52b3b5
KS
1008 queued_pkts: Vec::new(),
1009 slice_buf: Vec::new(),
1010 }
1011 }
1012#[allow(unused_variables)]
cdcbaa6c 1013 fn read_header(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
ce52b3b5 1014 let (id, size, ver) = read_chunk(self.src)?;
db5cc44b 1015 validate!((id == mktag!(b".RMF")) || (id == mktag!(b".RMP")));
ce52b3b5
KS
1016 validate!(size >= RMVB_HDR_SIZE);
1017 let fver = self.src.read_u32be()?;
1018 validate!(fver <= 1);
1019 let num_hdr = self.src.read_u32be()? as usize;
1020 validate!(num_hdr >= 1);
1021 if size > RMVB_HDR_SIZE {
1022 self.src.read_skip((size - RMVB_HDR_SIZE) as usize)?;
1023 }
1024
1025 let (id, size, ver) = read_chunk(self.src)?;
db5cc44b
KS
1026 let prop_size = if ver == 0 { RMVB_PROP_SIZE } else { RMVB_PROP_SIZE + 4 };
1027 validate!(size >= prop_size);
1028 validate!((ver == 0) || (ver == 2));
ce52b3b5
KS
1029 let maxbr = self.src.read_u32be()?;
1030 let avgbr = self.src.read_u32be()?;
1031 let maxps = self.src.read_u32be()?;
1032 let avgps = self.src.read_u32be()?;
1033 let num_pkt = self.src.read_u32be()? as usize;
1034 let duration = self.src.read_u32be()?;
1035 let preroll = self.src.read_u32be()?;
87c1d476 1036 let idx_off = self.src.read_size(ver)?;
ce52b3b5
KS
1037 let data_off = self.src.read_u32be()?;
1038 let num_streams = self.src.read_u16be()? as usize;
1039 let flags = self.src.read_u16be()?;
db5cc44b
KS
1040 if size > prop_size {
1041 self.src.read_skip((size - prop_size) as usize)?;
ce52b3b5
KS
1042 }
1043
1044 for _ in 0..num_hdr {
c9aba47e
KS
1045 if self.src.is_eof() {
1046 //warn maybe?
1047 break;
1048 }
cdcbaa6c 1049 let res = self.parse_chunk(strmgr, seek_idx);
c9aba47e
KS
1050 match res {
1051 Ok(last) => { if last { break; } },
1052 Err(DemuxerError::IOError) => { break; },
b7b96b4c 1053 Err(etype) => {
fa57381e 1054 if self.data_chunks.is_empty() { // data is not found, report error
b7b96b4c
KS
1055 return Err(etype);
1056 }
1057 },
c9aba47e 1058 };
ce52b3b5 1059 }
fa57381e 1060 validate!(!self.data_chunks.is_empty());
cdcbaa6c
KS
1061 self.cur_data_chunk = 0;
1062 let (pos, size, ver) = self.data_chunks[self.cur_data_chunk];
1063 self.data_pos = pos;
1064 self.data_ver = ver;
1065 self.data_end = pos + (size as u64);
1066 self.src.seek(SeekFrom::Start(self.data_pos + 10))?;
770bef6f
KS
1067 self.parse_data_start()?;
1068 Ok(())
1069 }
1070 fn parse_data_start(&mut self) -> DemuxerResult<()> {
ce52b3b5 1071 let num_packets = self.src.read_u32be()?;
db5cc44b
KS
1072 if self.data_ver == 2 {
1073 self.src.read_skip(12)?; // zeroes?
1074 }
ce52b3b5 1075 let next_data_hdr = self.src.read_u32be()?;
9f45e528 1076 self.num_packets = if num_packets > 0 { num_packets } else { 0xFFFFFF };
ce52b3b5 1077 self.cur_packet = 0;
770bef6f 1078 self.next_data = next_data_hdr as u64;
ce52b3b5
KS
1079 Ok(())
1080 }
cdcbaa6c 1081 fn parse_chunk(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<bool> {
ce52b3b5
KS
1082 let (id, size, ver) = read_chunk(self.src)?;
1083 let end_pos = self.src.tell() - 10 + (size as u64);
1084
db5cc44b 1085 validate!((ver == 0) || (ver == 2));
ce52b3b5
KS
1086 if id == mktag!(b"CONT") { self.parse_content_desc()?; }
1087 else if id == mktag!(b"MDPR") { self.parse_mdpr(strmgr)?; }
cdcbaa6c
KS
1088 else if id == mktag!(b"DATA") {
1089 self.data_chunks.push((self.src.tell() - 10, size, ver));
1090 }
9da33f04
KS
1091 else if id == mktag!(b"INDX") {
1092 if !seek_idx.skip_index {
1093 self.parse_index(seek_idx, (size as usize) - 10, ver)?;
1094 }
1095 }
6132225f 1096 else if id == 0 { return Ok(true); }
ce52b3b5
KS
1097 else { println!("unknown chunk type {:08X}", id); }
1098
1099 let cpos = self.src.tell();
1100 if cpos < end_pos {
1101 self.src.read_skip((end_pos - cpos) as usize)?;
1102 }
6132225f 1103 Ok(false)
ce52b3b5
KS
1104 }
1105#[allow(unused_variables)]
1106 fn parse_content_desc(&mut self) -> DemuxerResult<()> {
1107 let title_len = self.src.read_u16be()? as usize;
1108 self.src.read_skip(title_len)?;
1109 let author_len = self.src.read_u16be()? as usize;
1110 self.src.read_skip(author_len)?;
1111 let copywrong_len = self.src.read_u16be()? as usize;
1112 self.src.read_skip(copywrong_len)?;
1113 let comment_len = self.src.read_u16be()? as usize;
1114 self.src.read_skip(comment_len)?;
1115 Ok(())
1116 }
1117#[allow(unused_variables)]
1118 fn parse_mdpr(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<()> {
d4df9ee1 1119 let stream_no = self.src.read_u16be()? as u32;
ce52b3b5
KS
1120//todo check stream_no for duplicates
1121 let maxbr = self.src.read_u32be()?;
1122 let avgbr = self.src.read_u32be()?;
1123 let maxps = self.src.read_u32be()?;
1124 let avgps = self.src.read_u32be()?;
1125 let start = self.src.read_u32be()?;
1126 let preroll = self.src.read_u32be()?;
1127 let duration = self.src.read_u32be()?;
1128 let sname_size = self.src.read_byte()? as usize;
1129 let sname = read_string_size(self.src, sname_size)?;
ce52b3b5
KS
1130 let mime_size = self.src.read_byte()? as usize;
1131 let mime = read_string_size(self.src, mime_size)?;
ce52b3b5
KS
1132 let edata_size = self.src.read_u32be()? as usize;
1133 let edata: Option<Vec<u8>> = if edata_size == 0 { None } else {
e07387c7 1134 let mut edvec: Vec<u8> = vec![0; edata_size];
ce52b3b5
KS
1135 self.src.read_buf(&mut edvec)?;
1136 Some(edvec)
1137 };
d4df9ee1 1138 let mut is_mlti = false;
ce52b3b5
KS
1139 if edata_size > 8 {
1140 if let Some(edata_) = edata {
a480a0de 1141 is_mlti = RMDemuxCommon::parse_stream_info(&mut self.str_data, strmgr, stream_no, &edata_, duration)?;
ce52b3b5
KS
1142 }
1143 } else {
418bd0b8 1144 self.str_data.streams.push(RMStreamType::Unknown);
ce52b3b5 1145 }
d4df9ee1 1146 if !is_mlti {
418bd0b8 1147 self.str_data.str_ids.push(stream_no);
d4df9ee1 1148 }
d4df9ee1 1149
ce52b3b5
KS
1150 Ok(())
1151 }
cdcbaa6c 1152 fn parse_index(&mut self, seek_idx: &mut SeekIndex, chunk_size: usize, ver: u16) -> DemuxerResult<()> {
60b9641e 1153 if ver != 0 && ver != 2 { return Ok(()); }
cdcbaa6c
KS
1154 let num_entries = self.src.read_u32be()? as usize;
1155 let str_id = self.src.read_u16be()? as u32;
87c1d476 1156 let _next_idx = self.src.read_size(ver)?;
60b9641e
KS
1157 if ver == 0 {
1158 validate!(chunk_size == num_entries * 14 + 10);
1159 } else {
1160 validate!(chunk_size == num_entries * 18 + 14);
1161 }
cdcbaa6c 1162 if num_entries == 0 { return Ok(()); }
d24468d9 1163
cdcbaa6c
KS
1164 seek_idx.add_stream(str_id);
1165 let idx = seek_idx.get_stream_index(str_id).unwrap();
1166 for _ in 0..num_entries {
60b9641e
KS
1167 let iver = self.src.read_u16be()?;
1168 validate!(iver == ver);
cdcbaa6c 1169 let ts = self.src.read_u32be()? as u64;
87c1d476 1170 let pos = self.src.read_size(ver)?;
cdcbaa6c
KS
1171 let _pkt_no = self.src.read_u32be()?;
1172 idx.add_entry(SeekEntry { time: ts, pts: 0, pos });
1173 }
1174 idx.filled = true;
1175 seek_idx.mode = SeekIndexMode::Present;
1176 Ok(())
1177 }
ce52b3b5
KS
1178}
1179
1180fn read_string_size(src: &mut ByteReader, size: usize) -> DemuxerResult<String> {
1181 let mut vec: Vec<u8> = Vec::new();
1182 for _ in 0..size {
1183 let c = src.read_byte()?;
1184 vec.push(c);
1185 }
817e4872 1186 if let Ok(res) = String::from_utf8(vec) {
b7c882c1 1187 Ok(res)
ce52b3b5
KS
1188 } else {
1189 Ok(String::new())
1190 }
1191}
1192
d1304e6e
KS
1193struct RealAudioDemuxer<'a> {
1194 src: &'a mut ByteReader<'a>,
1195 stream: Option<RMAudioStream>,
1196 data_start: u64,
1197 data_end: u64,
1198 blk_size: usize,
1199 queued_pkts: Vec<NAPacket>,
1200}
1201
1202impl<'a> DemuxCore<'a> for RealAudioDemuxer<'a> {
1203 #[allow(unused_variables)]
33b5a8f0 1204 fn open(&mut self, strmgr: &mut StreamManager, _seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
d1304e6e
KS
1205 let magic = self.src.read_u32be()?;
1206 validate!(magic == mktag!(b".ra\xFD"));
1207 let ver = self.src.read_u16be()?;
1208 let ainfo = match ver {
1209 3 => {
6f263099 1210 parse_aformat3(self.src)?
d1304e6e
KS
1211 },
1212 4 => {
6f263099 1213 parse_aformat4(self.src)?
d1304e6e
KS
1214 },
1215 5 => {
6f263099 1216 parse_aformat5(self.src)?
d1304e6e
KS
1217 },
1218 _ => {
1219 println!("unknown version {}", ver);
1220 return Err(DemuxerError::InvalidData);
1221 },
1222 };
d1304e6e
KS
1223 let cname = find_codec_name(RM_AUDIO_CODEC_REGISTER, ainfo.fcc);
1224 let blk_size = if ainfo.fcc != mktag!(b"sipr") {
1225 ainfo.granularity as usize
1226 } else {
1227 validate!(ainfo.flavor <= 3);
1228 RM_SIPRO_BLOCK_SIZES[ainfo.flavor as usize]
1229 };
1230 let srate = ainfo.sample_rate;
1231 let soniton = NASoniton::new(ainfo.sample_size as u8, SONITON_FLAG_SIGNED);
1232 let ahdr = NAAudioInfo::new(srate, ainfo.channels as u8, soniton, blk_size);
1233 let extradata = if ainfo.edata_size == 0 {
1234 None
1235 } else {
e07387c7 1236 let mut dta: Vec<u8> = vec![0; ainfo.edata_size as usize];
d1304e6e
KS
1237 self.src.read_buf(dta.as_mut_slice())?;
1238 Some(dta)
1239 };
1240 let nainfo = NACodecInfo::new(cname, NACodecTypeInfo::Audio(ahdr), extradata);
a480a0de 1241 let res = strmgr.add_stream(NAStream::new(StreamType::Audio, 0, nainfo, 1, srate, ainfo.get_duration(ainfo.sample_rate)));
d1304e6e
KS
1242 if res.is_none() { return Err(MemoryError); }
1243
1244 let astr = RMAudioStream::new(ainfo.ileave_info);
1245 self.data_start = self.src.tell();
1246 self.data_end = if ainfo.total_bytes > 0 { self.src.tell() + (ainfo.total_bytes as u64) } else { 0 };
1247 self.blk_size = blk_size;
1248 self.stream = Some(astr);
1249
1250 Ok(())
1251 }
1252
1253 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
1254 if !self.queued_pkts.is_empty() {
951a626b
KS
1255 let mut pkt = self.queued_pkts.pop().unwrap();
1256 pkt.ts.pts = None;
d1304e6e
KS
1257 return Ok(pkt);
1258 }
1259 if (self.data_end != 0) && (self.src.tell() >= self.data_end) {
1260 return Err(DemuxerError::EOF);
1261 }
1262 let streamres = strmgr.get_stream_by_id(0);
1263 let stream = streamres.unwrap();
1264 if let Some(ref mut astr) = self.stream {
1265 loop {
951a626b 1266 let mut ret = astr.read_apackets(&mut self.queued_pkts, self.src, stream.clone(), 0, false, self.blk_size);
d1304e6e
KS
1267 if let Err(DemuxerError::TryAgain) = ret {
1268 continue;
1269 }
951a626b
KS
1270 if let Ok(ref mut pkt) = ret {
1271 pkt.ts.pts = None;
1272 }
d1304e6e
KS
1273 return ret;
1274 }
1275 }
1276 Err(DemuxerError::NoSuchInput)
1277 }
1278
1279 #[allow(unused_variables)]
24d99894 1280 fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()> {
d1304e6e
KS
1281 Err(NotImplemented)
1282 }
a480a0de
KS
1283
1284 fn get_duration(&self) -> u64 { 0 }
d1304e6e
KS
1285}
1286
787b8d03
KS
1287impl<'a> NAOptionHandler for RealAudioDemuxer<'a> {
1288 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
1289 fn set_options(&mut self, _options: &[NAOption]) { }
1290 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
1291}
1292
d1304e6e
KS
1293impl<'a> RealAudioDemuxer<'a> {
1294 fn new(io: &'a mut ByteReader<'a>) -> Self {
1295 RealAudioDemuxer {
1296 src: io,
1297 data_start: 0,
1298 data_end: 0,
1299 blk_size: 0,
1300 stream: None,
1301 queued_pkts: Vec::new(),
1302 }
1303 }
1304}
1305
418bd0b8
KS
1306enum IVRRecord {
1307 Invalid(u8),
1308 StreamProperties(usize),
817e4872 1309 Packet { ts: u32, strm: u32, flags: u32, len: usize, checksum: u32 },
418bd0b8
KS
1310 IntValue(Vec<u8>, u32),
1311 BinaryData(Vec<u8>, Vec<u8>),
1312 StringData(Vec<u8>, Vec<u8>),
1313 HeaderEnd,
1314 DataStart,
1315 DataEnd,
1316}
1317
1318impl IVRRecord {
1319 fn read_string(src: &mut ByteReader) -> DemuxerResult<Vec<u8>> {
1320 let len = src.read_u32be()? as usize;
e07387c7 1321 let mut val = vec![0; len];
418bd0b8
KS
1322 src.read_buf(val.as_mut_slice())?;
1323 Ok(val)
1324 }
1325
1326 fn read(src: &mut ByteReader) -> DemuxerResult<Self> {
1327 let code = src.read_byte()?;
1328 match code {
1329 1 => {
1330 let val = src.read_u32be()? as usize;
1331 Ok(IVRRecord::StreamProperties(val))
1332 },
1333 2 => {
1334 let ts = src.read_u32be()?;
817e4872 1335 let strm = src.read_u16be()? as u32;
418bd0b8
KS
1336 let flags = src.read_u32be()?;
1337 let len = src.read_u32be()? as usize;
1338 let chk = src.read_u32be()?;
6dc93d35 1339 validate!((len > 0) && (len < (1 << 24)));
817e4872 1340 Ok(IVRRecord::Packet { ts, strm, flags, len, checksum: chk })
418bd0b8
KS
1341 },
1342 3 => {
1343 let name = Self::read_string(src)?;
1344 let len = src.read_u32be()?;
1345 validate!(len == 4);
1346 let val = src.read_u32be()?;
1347 Ok(IVRRecord::IntValue(name, val))
1348 },
1349 4 => {
1350 let name = Self::read_string(src)?;
1351 let len = src.read_u32be()? as usize;
e07387c7 1352 let mut val = vec![0; len];
418bd0b8
KS
1353 src.read_buf(val.as_mut_slice())?;
1354 Ok(IVRRecord::BinaryData(name, val))
1355 },
1356 5 => {
1357 let name = Self::read_string(src)?;
1358 let val = Self::read_string(src)?;
1359 Ok(IVRRecord::StringData(name, val))
1360 },
1361 6 => Ok(IVRRecord::HeaderEnd),
1362 7 => {
1363 src.read_skip(8)?; // always zero?
1364 Ok(IVRRecord::DataEnd)
1365 },
1366 8 => {
1367 src.read_skip(8)?; // always zero?
1368 Ok(IVRRecord::DataStart)
1369 },
1370 _ => Ok(IVRRecord::Invalid(code)),
1371 }
1372 }
1373 fn is_data_start(&self) -> bool {
6f263099 1374 matches!(*self, IVRRecord::DataStart)
418bd0b8
KS
1375 }
1376}
1377
1378impl fmt::Display for IVRRecord {
1379 fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
1380 match *self {
1381 IVRRecord::Invalid(typ) => write!(f, "Invalid({:02X})", typ),
1382 IVRRecord::StreamProperties(num) =>
1383 write!(f, "({} stream properties)", num),
817e4872
KS
1384 IVRRecord::Packet { ts, strm, flags, len, checksum } =>
1385 write!(f, "paket({}, {}, {:X}, {}, {})", ts, strm, flags, len, checksum),
418bd0b8
KS
1386 IVRRecord::IntValue(ref name, val) =>
1387 write!(f, "({} = {})", String::from_utf8_lossy(name), val),
1388 IVRRecord::BinaryData(ref name, ref val) =>
1389 write!(f, "({} = {} bytes)", String::from_utf8_lossy(name), val.len()),
1390 IVRRecord::StringData(ref name, ref val) =>
1391 write!(f, "({} = {})", String::from_utf8_lossy(name), String::from_utf8_lossy(val)),
1392 IVRRecord::HeaderEnd => write!(f, "header end"),
1393 IVRRecord::DataEnd => write!(f, "data end"),
1394 IVRRecord::DataStart => write!(f, "data start"),
1395 }
1396 }
1397}
1398
1399struct RecordDemuxer {
1400 start_pos: u64,
1401 cur_pos: u64,
1402 start_str: u32,
1403 remap_ids: Vec<u32>,
1404}
1405
1406impl RecordDemuxer {
1407 fn new(pos: u64, start_str: u32) -> Self {
1408 RecordDemuxer {
1409 start_pos: pos,
1410 cur_pos: pos,
e07387c7 1411 start_str,
418bd0b8
KS
1412 remap_ids: Vec::new(),
1413 }
1414 }
1415 fn parse_header(&mut self, src: &mut ByteReader, strmgr: &mut StreamManager, str_data: &mut CommonStreamData) -> DemuxerResult<()> {
1416 src.seek(SeekFrom::Start(self.cur_pos))?;
1417 let magic = src.read_u32be()?;
1418 validate!(magic == mktag!(b".REC"));
1419 let _smth = src.read_byte()?;
1420 let num_entries = src.read_u32be()? as usize;
1421 for _ in 0..num_entries {
1422 let _rec = IVRRecord::read(src)?;
418bd0b8
KS
1423 }
1424 let mut has_seek_table = false;
1425 let mut cur_str_no = 0;
1426 loop {
1427 let rec = IVRRecord::read(src)?;
1428 match rec {
1429 IVRRecord::HeaderEnd => { break; },
1430 IVRRecord::StreamProperties(num) => {
1431 let stream_no = cur_str_no + self.start_str;
1432 cur_str_no += 1;
1433 let mut parsed = false;
1434 let mut real_stream_no = 0;
a480a0de 1435 let mut duration = 0;
418bd0b8
KS
1436 for _ in 0..num {
1437 let rec = IVRRecord::read(src)?;
418bd0b8
KS
1438 match rec {
1439 IVRRecord::IntValue(ref name, val) => {
1440 if name == b"StreamNumber\0" {
1441 real_stream_no = val;
1442 }
a480a0de
KS
1443 if name == b"Duration\0" {
1444 duration = val;
1445 }
418bd0b8
KS
1446 },
1447 IVRRecord::BinaryData(ref name, ref val) => {
1448 if name == b"OpaqueData\0" {
1449 validate!(!parsed);
a480a0de 1450 let is_mlti = RMDemuxCommon::parse_stream_info(str_data, strmgr, stream_no, val, duration)?;
418bd0b8
KS
1451 if !is_mlti {
1452 str_data.str_ids.push(stream_no);
1453 }
1454 parsed = true;
1455 }
1456 },
1457 IVRRecord::StringData(ref name, ref val) => {
1458 if (name == b"SeekType\0") && (val != b"None\0") {
1459 has_seek_table = true;
1460 }
1461 },
1462 _ => { return Err(DemuxerError::InvalidData); }
1463 };
1464 }
1465 if !parsed {
1466 str_data.streams.push(RMStreamType::Unknown);
1467 str_data.str_ids.push(stream_no);
1468 }
1469 self.remap_ids.push(real_stream_no);
1470 },
1471 _ => {println!(" unexpected {}", rec); return Err(DemuxerError::InvalidData); }
1472 };
1473 }
418bd0b8
KS
1474 let off0 = src.read_u32be()? as u64;
1475 let _off1 = src.read_u32be()?;
1476 let _off2 = src.read_u32be()?;
1477 validate!(off0 + self.start_pos == src.tell());
418bd0b8
KS
1478 if has_seek_table {
1479 src.read_skip(4)?;
1480 let data_off = src.read_u32be()? as u64;
418bd0b8
KS
1481 let pos = src.tell();
1482 validate!(data_off + self.start_pos > pos);
1483 src.read_skip((data_off + self.start_pos - pos) as usize)?;
1484 let rec = IVRRecord::read(src)?;
1485 validate!(rec.is_data_start());
1486 } else {
1487 let ntype = src.peek_byte()?;
1488 validate!((ntype == 2) || (ntype == 7)); // packet or data end, no start
1489 }
1490
1491 self.cur_pos = src.tell();
1492
1493 Ok(())
1494 }
1495 fn get_packet(&mut self, src: &mut ByteReader, str_data: &mut CommonStreamData, strmgr: &StreamManager, queued_pkts: &mut Vec<NAPacket>, slice_buf: &mut Vec<u8>) -> DemuxerResult<NAPacket> {
1496 src.seek(SeekFrom::Start(self.cur_pos))?;
1497 loop {
1498 let rec = IVRRecord::read(src)?;
1499 match rec {
817e4872 1500 IVRRecord::Packet { ts, strm, flags, len, .. } => {
418bd0b8 1501 let payload_size = len;
817e4872 1502 let sr = self.remap_ids.iter().position(|x| *x == strm);
418bd0b8
KS
1503 validate!(sr.is_some());
1504 let str_no = self.start_str + (sr.unwrap() as u32);
6dc93d35 1505 let pkt_grp = ((flags >> 8) & 0xFF) as u16;
e6aaad5c 1506 let stream_id = str_data.get_stream_id(str_no, pkt_grp);
418bd0b8
KS
1507 let sr = str_data.find_stream(stream_id);
1508 if sr.is_none() {
1509 src.read_skip(payload_size)?;
1510 return Err(DemuxerError::InvalidData);
1511 }
1512 let str_id = sr.unwrap();
1513
1514 let streamres = strmgr.get_stream_by_id(stream_id);
1515 if streamres.is_none() {
1516 src.read_skip(payload_size)?;
1517 continue;
1518 }
1519 let stream = streamres.unwrap();
1520 if strmgr.is_ignored_id(stream_id) {
1521 src.read_skip(payload_size)?;
1522 continue;
1523 }
1524 let keyframe = false;
1525 let ret = RMDemuxCommon::parse_packet_payload(src, &mut str_data.streams[str_id], stream, slice_buf, queued_pkts, keyframe, ts, payload_size);
1526 if let Err(DemuxerError::TryAgain) = ret {
1527 continue;
1528 } else {
1529 self.cur_pos = src.tell();
1530 return ret;
1531 }
1532 },
1533 IVRRecord::DataEnd => return Err(DemuxerError::EOF),
1534 _ => return Err(DemuxerError::InvalidData),
1535 }
1536 }
1537 }
1538}
1539
1540struct RealIVRDemuxer<'a> {
1541 src: &'a mut ByteReader<'a>,
1542 recs: Vec<RecordDemuxer>,
1543 cur_rec: usize,
1544 queued_pkts: Vec<NAPacket>,
1545 slice_buf: Vec<u8>,
1546 str_data: CommonStreamData,
1547}
1548
1549impl<'a> DemuxCore<'a> for RealIVRDemuxer<'a> {
1550 #[allow(unused_variables)]
33b5a8f0 1551 fn open(&mut self, strmgr: &mut StreamManager, seek_idx: &mut SeekIndex) -> DemuxerResult<()> {
418bd0b8
KS
1552 let magic = self.src.peek_u32be()?;
1553 if magic == mktag!(b".REC") {
1554 let mut rec = RecordDemuxer::new(0, 0);
6f263099 1555 rec.parse_header(self.src, strmgr, &mut self.str_data)?;
418bd0b8
KS
1556 self.recs.push(rec);
1557 } else if magic == mktag!(b".R1M") {
418bd0b8
KS
1558 self.src.read_skip(4)?; // magic
1559 self.src.read_skip(3)?; // always 0, 1, 1 ?
6f263099 1560 let _name = IVRRecord::read_string(self.src)?;
418bd0b8
KS
1561 self.src.read_skip(1)?; // always 0?
1562 let len1 = self.src.read_u32be()? as u64;
1563 let off1 = self.src.read_u64be()?;
1564 let cpos = self.src.tell();
1565 validate!(off1 == len1 + cpos - 8);
1566 self.src.read_skip((off1 - cpos) as usize)?;
1567 loop {
1568 let typ = self.src.read_byte()?;
418bd0b8
KS
1569 match typ {
1570 1 => {
1571 let len = self.src.read_u32be()?;
1572 self.src.read_skip(len as usize)?;
1573 },
1574 2 => {
1575 let len = self.src.read_u32be()? as u64;
1576 let pos = self.src.tell();
6dc93d35
KS
1577 if len > 0x20 {
1578 let num_streams = self.str_data.streams.len() as u32;
1579 let mut rec = RecordDemuxer::new(pos + 12, num_streams);
6f263099 1580 rec.parse_header(self.src, strmgr, &mut self.str_data)?;
6dc93d35
KS
1581 self.recs.push(rec);
1582 }
418bd0b8
KS
1583 self.src.seek(SeekFrom::Start(pos + len))?;
1584 },
1585 b'R' => {
1586 let mut buf: [u8; 2] = [0; 2];
1587 self.src.peek_buf(&mut buf)?;
1588 if (buf[0] == b'J') && (buf[1] == b'M') { // RJMx markers at the end of file
1589 break;
1590 } else {
1591 return Err(DemuxerError::InvalidData);
1592 }
1593 },
1594 _ => { return Err(DemuxerError::InvalidData); },
1595 };
1596 }
1597 } else {
1598 return Err(DemuxerError::InvalidData);
1599 }
1600
1601 Ok(())
1602 }
1603
1604 fn get_frame(&mut self, strmgr: &mut StreamManager) -> DemuxerResult<NAPacket> {
1605 if !self.queued_pkts.is_empty() {
1606 let pkt = self.queued_pkts.pop().unwrap();
1607 return Ok(pkt);
1608 }
1609 loop {
1610 if self.cur_rec >= self.recs.len() { return Err(DemuxerError::EOF); }
6f263099 1611 let res = self.recs[self.cur_rec].get_packet(self.src, &mut self.str_data, strmgr, &mut self.queued_pkts, &mut self.slice_buf);
418bd0b8
KS
1612 if let Err(DemuxerError::EOF) = res {
1613 self.cur_rec += 1;
1614 } else {
1615 return res;
1616 }
1617 }
1618 }
1619
1620 #[allow(unused_variables)]
24d99894 1621 fn seek(&mut self, time: NATimePoint, seek_idx: &SeekIndex) -> DemuxerResult<()> {
418bd0b8
KS
1622 Err(NotImplemented)
1623 }
a480a0de
KS
1624
1625 fn get_duration(&self) -> u64 { 0 }
418bd0b8
KS
1626}
1627
787b8d03
KS
1628impl<'a> NAOptionHandler for RealIVRDemuxer<'a> {
1629 fn get_supported_options(&self) -> &[NAOptionDefinition] { &[] }
1630 fn set_options(&mut self, _options: &[NAOption]) { }
1631 fn query_option_value(&self, _name: &str) -> Option<NAValue> { None }
1632}
1633
418bd0b8
KS
1634impl<'a> RealIVRDemuxer<'a> {
1635 fn new(io: &'a mut ByteReader<'a>) -> Self {
1636 RealIVRDemuxer {
1637 src: io,
1638 recs: Vec::new(),
1639 cur_rec: 0,
1640 queued_pkts: Vec::new(),
1641 slice_buf: Vec::new(),
1642 str_data: CommonStreamData::new(),
1643 }
1644 }
1645}
1646
b7c882c1 1647static RM_VIDEO_CODEC_REGISTER: &[(&[u8;4], &str)] = &[
ce52b3b5
KS
1648 (b"RV10", "realvideo1"),
1649 (b"RV20", "realvideo2"),
1650 (b"RVTR", "realvideo2"),
1651 (b"RV30", "realvideo3"),
1652 (b"RV40", "realvideo4"),
db5cc44b 1653 (b"RV60", "realvideo6"),
750b299c 1654 (b"CLV1", "clearvideo_rm"),
ce52b3b5
KS
1655];
1656
b7c882c1 1657static RM_AUDIO_CODEC_REGISTER: &[(&[u8;4], &str)] = &[
ce52b3b5
KS
1658 (b"lpcJ", "ra14.4"),
1659 (b"28_8", "ra28.8"),
1660 (b"cook", "cook"),
1661 (b"dnet", "ac3"),
1662 (b"sipr", "sipro"),
1663 (b"atrc", "atrac3"),
1664 (b"LSD:", "ralf"),
1665 (b"raac", "aac"),
1666 (b"racp", "aac"),
1667];
1668
1669pub struct RealMediaDemuxerCreator { }
1670
1671impl DemuxerCreator for RealMediaDemuxerCreator {
6011e201 1672 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
ce52b3b5
KS
1673 Box::new(RealMediaDemuxer::new(br))
1674 }
1675 fn get_name(&self) -> &'static str { "realmedia" }
1676}
1677
d1304e6e
KS
1678pub struct RealAudioDemuxerCreator { }
1679
1680impl DemuxerCreator for RealAudioDemuxerCreator {
6011e201 1681 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
d1304e6e
KS
1682 Box::new(RealAudioDemuxer::new(br))
1683 }
1684 fn get_name(&self) -> &'static str { "realaudio" }
1685}
1686
418bd0b8 1687pub struct RealIVRDemuxerCreator { }
d1304e6e
KS
1688
1689impl DemuxerCreator for RealIVRDemuxerCreator {
6011e201 1690 fn new_demuxer<'a>(&self, br: &'a mut ByteReader<'a>) -> Box<dyn DemuxCore<'a> + 'a> {
d1304e6e
KS
1691 Box::new(RealIVRDemuxer::new(br))
1692 }
1693 fn get_name(&self) -> &'static str { "real_ivr" }
418bd0b8 1694}
d1304e6e 1695
ce52b3b5
KS
1696#[cfg(test)]
1697mod test {
1698 use super::*;
1699 use std::fs::File;
1700
1701 #[test]
1702 fn test_rm_demux() {
886cde48 1703 // sample from a private collection
ce52b3b5
KS
1704 let mut file =
1705 File::open("assets/RV/rv10_dnet_640x352_realvideo_encoder_4.0.rm").unwrap();
1706// File::open("assets/RV/rv20_cook_640x352_realproducer_plus_8.51.rm").unwrap();
1707// File::open("assets/RV/rv20_svt_atrc_640x352_realproducer_plus_8.51.rm").unwrap();
1708// File::open("assets/RV/rv30_atrc_384x208_realproducer_plus_8.51.rm").unwrap();
1709// File::open("assets/RV/rv30_chroma_drift.rm").unwrap();
1710// File::open("assets/RV/rv30_weighted_mc.rm").unwrap();
1711// File::open("assets/RV/rv40_weighted_mc.rmvb").unwrap();
1712// File::open("assets/RV/rv40_weighted_mc_2.rmvb").unwrap();
1713// File::open("assets/RV/clv1_sipr_384x208_realvideo_encoder_4.0.rm").unwrap();
1714// File::open("assets/RV/luckynight.rmvb").unwrap();
1715// File::open("assets/RV/rv40_ralf.rmvb").unwrap();
1716 let mut fr = FileReader::new_read(&mut file);
1717 let mut br = ByteReader::new(&mut fr);
1718 let mut dmx = RealMediaDemuxer::new(&mut br);
1719 let mut sm = StreamManager::new();
caf0f37e
KS
1720 let mut si = SeekIndex::new();
1721 dmx.open(&mut sm, &mut si).unwrap();
ce52b3b5
KS
1722
1723 loop {
1724 let pktres = dmx.get_frame(&mut sm);
1725 if let Err(e) = pktres {
1726 if e == DemuxerError::EOF { break; }
1727 panic!("error");
1728 }
1729 let pkt = pktres.unwrap();
1730 println!("Got {}", pkt);
1731 }
41f11b2b 1732//panic!("the end");
ce52b3b5 1733 }
d1304e6e
KS
1734 #[test]
1735 fn test_ra_demux() {
886cde48 1736 // sample: https://samples.mplayerhq.hu/real//RA/ra_with_comment_field/diemusik.ra
d1304e6e
KS
1737 let mut file =
1738// File::open("assets/RV/welcome288.ra").unwrap();
1739 File::open("assets/RV/diemusik.ra").unwrap();
1740 let mut fr = FileReader::new_read(&mut file);
1741 let mut br = ByteReader::new(&mut fr);
1742 let mut dmx = RealAudioDemuxer::new(&mut br);
1743 let mut sm = StreamManager::new();
caf0f37e
KS
1744 let mut si = SeekIndex::new();
1745 dmx.open(&mut sm, &mut si).unwrap();
d1304e6e 1746
418bd0b8
KS
1747 loop {
1748 let pktres = dmx.get_frame(&mut sm);
1749 if let Err(e) = pktres {
1750 if e == DemuxerError::EOF { break; }
1751 panic!("error");
1752 }
1753 let pkt = pktres.unwrap();
1754 println!("Got {}", pkt);
1755 }
1756 }
1757 #[test]
1758 fn test_ivr_demux() {
886cde48 1759 // sample: https://samples.mplayerhq.hu/real/ivr/Opener_rm_hi.ivr
418bd0b8
KS
1760 let mut file =
1761 File::open("assets/RV/Opener_rm_hi.ivr").unwrap();
1762// File::open("assets/RV/SherwinWilliamsCommercial.ivr").unwrap();
1763 let mut fr = FileReader::new_read(&mut file);
1764 let mut br = ByteReader::new(&mut fr);
1765 let mut dmx = RealIVRDemuxer::new(&mut br);
1766 let mut sm = StreamManager::new();
caf0f37e
KS
1767 let mut si = SeekIndex::new();
1768 dmx.open(&mut sm, &mut si).unwrap();
418bd0b8 1769
d1304e6e
KS
1770 loop {
1771 let pktres = dmx.get_frame(&mut sm);
1772 if let Err(e) = pktres {
1773 if e == DemuxerError::EOF { break; }
1774 panic!("error");
1775 }
1776 let pkt = pktres.unwrap();
1777 println!("Got {}", pkt);
1778 }
1779 }
ce52b3b5 1780}