X-Git-Url: https://git.nihav.org/?p=nihav.git;a=blobdiff_plain;f=nihav-core%2Fsrc%2Fio%2Fbyteio.rs;h=0ff54cb13ed9321e31f794dbea03e8365d53dc39;hp=b3d06216d842c2a9790aff6dd7cb499bfd2d2ea0;hb=6a97ae3af1e98b2d7b064c8870aaccae44a47c7d;hpb=f88724a81e18ae9521863faf94ae9d2c2cb60e15 diff --git a/nihav-core/src/io/byteio.rs b/nihav-core/src/io/byteio.rs index b3d0621..0ff54cb 100644 --- a/nihav-core/src/io/byteio.rs +++ b/nihav-core/src/io/byteio.rs @@ -717,6 +717,12 @@ impl ByteIO for BoundedFileReader { } fn read_buf(&mut self, buf: &mut [u8]) -> ByteIOResult { + if let Some(epos) = self.end { + if self.real_tell() >= epos { + self.eof = true; + return Err(ByteIOError::EOF); + } + } let len = self.max_read_len(buf.len()); match self.file.read_exact(&mut buf[..len]) { Ok(()) if len == buf.len() => Ok(buf.len()), @@ -736,16 +742,24 @@ impl ByteIO for BoundedFileReader { } fn read_buf_some(&mut self, buf: &mut [u8]) -> ByteIOResult { + if let Some(epos) = self.end { + if self.real_tell() >= epos { + self.eof = true; + return Err(ByteIOError::EOF); + } + } let len = self.max_read_len(buf.len()); let ret = self.file.read(&mut buf[..len]); if ret.is_err() { return Err(ByteIOError::ReadError); } let sz = ret.unwrap(); if sz < len { - if let Err(_err) = self.file.read(&mut buf[sz..][..1]) { - self.eof = true; - } else { + if let Ok(1) = self.file.read(&mut buf[sz..][..1]) { return Ok(sz + 1); } + self.eof = true; + if sz == 0 { + return Err(ByteIOError::EOF); + } } Ok(sz) }