aboutsummaryrefslogtreecommitdiff
path: root/src/unityfs/multi_reader.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/unityfs/multi_reader.rs')
-rw-r--r--src/unityfs/multi_reader.rs10
1 files changed, 8 insertions, 2 deletions
diff --git a/src/unityfs/multi_reader.rs b/src/unityfs/multi_reader.rs
index 3de6cd5..64e1670 100644
--- a/src/unityfs/multi_reader.rs
+++ b/src/unityfs/multi_reader.rs
@@ -1,10 +1,11 @@
+use anyhow::Result;
+use log::trace;
use std::{
io::{Read, Seek, SeekFrom},
sync::{Arc, Mutex},
};
-use anyhow::Result;
-
+/// Splits an reader into many by seeking in between reads.
pub struct MultiReader<T> {
position: u64,
inner: Arc<Mutex<(u64, T)>>,
@@ -30,7 +31,12 @@ impl<T: Read + Seek> Read for MultiReader<T> {
fn read(&mut self, buf: &mut [u8]) -> std::io::Result<usize> {
let mut g = self.inner.lock().unwrap();
if g.0 != self.position {
+ trace!(
+ "seeking to match stream position ({} actual vs {} required)",
+ g.0, self.position
+ );
g.1.seek(SeekFrom::Start(self.position))?;
+ g.0 = self.position;
}
let size = g.1.read(buf)?;
g.0 += size as u64;