Fix MoreDataNeeded containing the wrong size

rust-av · May 15, 2023 · 04b86b9 · 04b86b9
1 parent 7f15b7c
commit 04b86b9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/demuxer.rs b/src/demuxer.rs
@@ -173,10 +173,12 @@ impl Demuxer for MkvDemuxer {
                     Ok((seek, Event::MoreDataNeeded(0)))
                 }
                 Err(Err::Incomplete(Needed::Size(size))) => {
-                    Err(Error::MoreDataNeeded(usize::from(size)))
+                    // Needed::Size(size) describes only the missing number of bytes,
+                    // but Error::MoreDataNeeded(size) wants the entire size of the Element.
+                    Err(Error::MoreDataNeeded(buf.data().len() + size.get()))
                 }
                 e => {
-                    error!("{:?}", e);
+                    error!("{e:?}");
                     Err(Error::InvalidData)
                 }
             }