X-Git-Url: https://git.nihav.org/?a=blobdiff_plain;f=src%2Fdemuxers%2Favi.rs;h=3810bc82d99e49989d373660c07cf6ff6b64e535;hb=3ff4cebbb4d6d7eefa9879eeef59de02481a00cc;hp=e5f688afd6a73a41c39864f362a6e7a4c6b124c4;hpb=e189501ed92ffd55123f9fceb7e4c78ddf1be936;p=nihav.git diff --git a/src/demuxers/avi.rs b/src/demuxers/avi.rs index e5f688a..3810bc8 100644 --- a/src/demuxers/avi.rs +++ b/src/demuxers/avi.rs @@ -78,6 +78,11 @@ impl<'a> Demux<'a> for AVIDemuxer<'a> { if self.movi_size == 0 { return Err(EOF); } let mut tag: [u8; 4] = [0; 4]; loop { + if (self.src.tell() & 1) == 1 { + self.src.read_skip(1)?; + self.movi_size -= 1; + if self.movi_size == 0 { return Err(EOF); } + } self.src.read_buf(&mut tag)?; let size = self.src.read_u32le()? as usize; if mktag!(tag) == mktag!(b"JUNK") { @@ -135,7 +140,8 @@ impl<'a> AVIDemuxer<'a> { if RIFFTag::Chunk(tag) == end_tag { return Ok((size, true)); } - let ltag = if is_list_tag(tag) { self.src.read_u32be()? } else { 0 }; + let is_list = is_list_tag(tag); + let ltag = if is_list { self.src.read_u32be()? } else { 0 }; if RIFFTag::List(tag, ltag) == end_tag { return Ok((size, true)); } @@ -144,6 +150,7 @@ impl<'a> AVIDemuxer<'a> { if RIFFTag::Chunk(tag) == CHUNKS[i].tag { let psize = (CHUNKS[i].parse)(self, size)?; if psize != size { return Err(InvalidData); } + if (psize & 1) == 1 { self.src.read_skip(1)?; } return Ok((size + 8, false)); } if RIFFTag::List(tag, ltag) == CHUNKS[i].tag { @@ -155,12 +162,23 @@ impl<'a> AVIDemuxer<'a> { let (psize, _) = self.parse_chunk(end_tag, rest_size, depth+1)?; if psize > rest_size { return Err(InvalidData); } rest_size -= psize; + if (psize & 1) == 1 { + if rest_size > 0 { + rest_size -= 1; + } + } } return Ok((size + 8, false)); } } - self.src.read_skip(size)?; + if !is_list { + self.src.read_skip(size)?; + } else { + if size < 4 { return Err(InvalidData); } + self.src.read_skip(size - 4)?; + } + if (size & 1) == 1 { self.src.read_skip(1)?; } return Ok((size + 8, false)); } @@ -311,7 +329,7 @@ fn parse_strf_auds(dmx: &mut AVIDemuxer, size: usize) -> DemuxerResult { let samplespersec = dmx.src.read_u32le()?; let avgbytespersec = dmx.src.read_u32le()?; let block_align = dmx.src.read_u16le()?; - let bits_per_sample = dmx.src.read_u16le()?; + let bits_per_sample = dmx.src.read_u16le()? * 8; let soniton = NASoniton::new(bits_per_sample as u8, SONITON_FLAG_SIGNED); let ahdr = NAAudioInfo::new(samplespersec, channels as u8, soniton, block_align as usize);