use std::{
cmp,
io::{self, Write},
};
use crate::rialo_s_program::program_memory::rlo_memcpy;
#[derive(Debug, Default)]
pub struct BpfWriter<T> {
inner: T,
pos: u64,
}
impl<T> BpfWriter<T> {
pub fn new(inner: T) -> Self {
Self { inner, pos: 0 }
}
}
impl Write for BpfWriter<&mut [u8]> {
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
if self.pos >= self.inner.len() as u64 {
return Ok(0);
}
let amt = cmp::min(
self.inner.len().saturating_sub(self.pos as usize),
buf.len(),
);
rlo_memcpy(&mut self.inner[(self.pos as usize)..], buf, amt);
self.pos += amt as u64;
Ok(amt)
}
fn write_all(&mut self, buf: &[u8]) -> io::Result<()> {
if self.write(buf)? == buf.len() {
Ok(())
} else {
Err(io::Error::new(
io::ErrorKind::WriteZero,
"failed to write whole buffer",
))
}
}
fn flush(&mut self) -> io::Result<()> {
Ok(())
}
}