X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=sndplay%2Fsrc%2Fdemux.rs;h=c35167c97ae7bbae7803fe054b833f3b5974a2cc;hb=0d3a25c7795795c4590e496bf7f58fe9ede2b4af;hp=ac483da389eea1c00d049e1124c82ce4651fb2f8;hpb=dbca8a6d522fde14d236bb09e09d199259288267;p=nihav-player.git diff --git a/sndplay/src/demux.rs b/sndplay/src/demux.rs index ac483da..c35167c 100644 --- a/sndplay/src/demux.rs +++ b/sndplay/src/demux.rs @@ -81,7 +81,7 @@ impl<'a> DemuxerObject<'a> { if !is_raw { let res = detect::detect_format(name, br); let (dmx_name, _) = res.unwrap_or(("", detect::DetectionScore::No)); - if dmx_name != "" { + if !dmx_name.is_empty() { println!("trying demuxer {} on {}", dmx_name, name); if let Some(dmx_fact) = reg.dmx_reg.find_demuxer(dmx_name) { br.seek(SeekFrom::Start(0)).unwrap(); @@ -89,7 +89,7 @@ impl<'a> DemuxerObject<'a> { return DemuxerObject::Normal(dmx); } } - if dmx_name != "" { + if !dmx_name.is_empty() { println!("trying raw demuxer {} on {}", dmx_name, name); if let Some(rdmx_fact) = reg.rdmx_reg.find_demuxer(dmx_name) { br.seek(SeekFrom::Start(0)).unwrap(); @@ -126,7 +126,8 @@ impl<'a> DemuxerObject<'a> { } br.seek(SeekFrom::Start(0)).unwrap(); let mut buf = vec![0; 1048576]; - let size = br.peek_buf(&mut buf).unwrap(); + let size = br.read_buf_some(&mut buf).unwrap(); + br.seek(SeekFrom::Start(0)).unwrap(); let mut pname = ""; for pinfo in reg.pkt_reg.iter() { @@ -137,7 +138,7 @@ impl<'a> DemuxerObject<'a> { break; } } - if pname != "" { + if !pname.is_empty() { println!("found raw stream of type {} for {}", pname, name); let pcreate = reg.pkt_reg.find_packetiser(pname).unwrap(); let mut packetiser = (pcreate)(); @@ -150,10 +151,7 @@ impl<'a> DemuxerObject<'a> { } } pub fn is_none(&self) -> bool { - match *self { - DemuxerObject::None => true, - _ => false, - } + matches!(*self, DemuxerObject::None) } pub fn get_duration(&self) -> u64 { match *self { @@ -281,20 +279,31 @@ pub fn detect_tags(br: &mut ByteReader) -> (bool, u64, Option) { // check for ID3v{2-4} let mut buf = [0; 5]; - br.peek_buf(&mut buf).unwrap(); - if &buf[0..3] == b"ID3" && buf[3] > 0 && buf[3] < 5 && buf[4] == 0 { //ID3 tag found, must be a raw stream - br.read_skip(6).unwrap(); - let mut size = 0; - for _ in 0..4 { - let b = br.read_byte().unwrap(); - if (b & 0x80) != 0 { - println!("Invalid ID3 size"); - break; + loop { + if br.peek_buf(&mut buf).is_err() { + break; + } + if &buf[0..3] == b"ID3" && buf[3] > 0 && buf[3] < 5 && buf[4] == 0 { //ID3 tag found, must be a raw stream + br.read_skip(6).unwrap(); + let mut size = 0; + for _ in 0..4 { + let b = br.read_byte().unwrap(); + if (b & 0x80) != 0 { + println!("Invalid ID3 size"); + break; + } + size = (size << 7) | u64::from(b); } - size = (size << 7) | u64::from(b); + start += size + 10; + br.read_skip(size as usize).unwrap(); + while let Ok(0) = br.read_byte() { + start += 1; + } + br.seek(SeekFrom::Start(start)).unwrap(); + is_raw = true; + } else { + break; } - start = size + 10; - is_raw = true; } // check for ID3v1 br.seek(SeekFrom::End(-128)).unwrap(); @@ -302,6 +311,20 @@ pub fn detect_tags(br: &mut ByteReader) -> (bool, u64, Option) { br.peek_buf(&mut buf[..3]).unwrap(); if &buf[0..3] == b"TAG" { end = Some(off); + // check for Lyrics v2 + let mut sig = [0; 9]; + br.seek(SeekFrom::End(-128 - 9)).unwrap(); + br.peek_buf(&mut sig).unwrap(); + if &sig == b"LYRICS200" { + br.seek(SeekFrom::Current(-6)).unwrap(); + let mut sizestr = [0; 6]; + br.peek_buf(&mut sizestr).unwrap(); + if let Ok(sstr) = std::str::from_utf8(&sizestr) { + if let Ok(size) = sstr.parse::() { + end = Some(br.tell() - size); + } + } + } } // check for APETAG let mut buf = [0; 8];