[IO] Replace the `io::Cursor` re-export with our own `Cursor`
This commit is contained in:
parent
141343edb4
commit
3caaadf9bb
|
@ -752,7 +752,6 @@ mod tests {
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::address::Address;
|
use crate::address::Address;
|
||||||
use crate::io;
|
|
||||||
use crate::network::Network::{Bitcoin, Testnet};
|
use crate::network::Network::{Bitcoin, Testnet};
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
|
|
@ -38,10 +38,10 @@ pub mod io {
|
||||||
compile_error!("At least one of std or core2 must be enabled");
|
compile_error!("At least one of std or core2 must be enabled");
|
||||||
|
|
||||||
#[cfg(feature = "std")]
|
#[cfg(feature = "std")]
|
||||||
pub use std::io::{Cursor, Error, ErrorKind, Result};
|
pub use std::io::{Error, ErrorKind, Result};
|
||||||
|
|
||||||
#[cfg(not(feature = "std"))]
|
#[cfg(not(feature = "std"))]
|
||||||
pub use core2::io::{Cursor, Error, ErrorKind, Result};
|
pub use core2::io::{Error, ErrorKind, Result};
|
||||||
|
|
||||||
/// A generic trait describing an input stream. See [`std::io::Read`] for more info.
|
/// A generic trait describing an input stream. See [`std::io::Read`] for more info.
|
||||||
pub trait Read {
|
pub trait Read {
|
||||||
|
@ -86,11 +86,44 @@ pub mod io {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(all(feature = "core2", not(feature = "std")))]
|
#[cfg(not(feature = "std"))]
|
||||||
impl<R: core2::io::Read> Read for R {
|
impl Read for &[u8] {
|
||||||
#[inline]
|
#[inline]
|
||||||
fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
|
fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
|
||||||
<R as core2::io::Read>::read(self, buf)
|
let cnt = core::cmp::min(self.len(), buf.len());
|
||||||
|
buf[..cnt].copy_from_slice(&self[..cnt]);
|
||||||
|
*self = &self[cnt..];
|
||||||
|
Ok(cnt)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct Cursor<T> {
|
||||||
|
inner: T,
|
||||||
|
pos: u64,
|
||||||
|
}
|
||||||
|
impl<T: AsRef<[u8]>> Cursor<T> {
|
||||||
|
#[inline]
|
||||||
|
pub fn new(inner: T) -> Self {
|
||||||
|
Cursor { inner, pos: 0 }
|
||||||
|
}
|
||||||
|
#[inline]
|
||||||
|
pub fn position(&self) -> u64 {
|
||||||
|
self.pos
|
||||||
|
}
|
||||||
|
#[inline]
|
||||||
|
pub fn into_inner(self) -> T {
|
||||||
|
self.inner
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl<T: AsRef<[u8]>> Read for Cursor<T> {
|
||||||
|
#[inline]
|
||||||
|
fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
|
||||||
|
let inner: &[u8] = self.inner.as_ref();
|
||||||
|
let start_pos = self.pos.try_into().unwrap_or(inner.len());
|
||||||
|
let read = core::cmp::min(inner.len().saturating_sub(start_pos), buf.len());
|
||||||
|
buf[..read].copy_from_slice(&inner[start_pos..start_pos + read]);
|
||||||
|
self.pos = self.pos.saturating_add(read.try_into().unwrap_or(u64::max_value() /* unreachable */));
|
||||||
|
Ok(read)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue