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