65 lines
1.3 KiB
Rust
65 lines
1.3 KiB
Rust
use anyhow::{anyhow, Result};
|
|
use byteorder::{LittleEndian, WriteBytesExt};
|
|
use nom::{number::complete::*, IResult};
|
|
|
|
//------------------------------------------
|
|
|
|
pub trait Unpack {
|
|
// The size of the value when on disk.
|
|
fn disk_size() -> u32;
|
|
fn unpack(data: &[u8]) -> IResult<&[u8], Self>
|
|
where
|
|
Self: std::marker::Sized;
|
|
}
|
|
|
|
pub fn unpack<U: Unpack>(data: &[u8]) -> Result<U> {
|
|
match U::unpack(data) {
|
|
Err(_e) => Err(anyhow!("couldn't parse SMRoot")),
|
|
Ok((_i, v)) => Ok(v),
|
|
}
|
|
}
|
|
|
|
//------------------------------------------
|
|
|
|
pub trait Pack {
|
|
fn pack<W: WriteBytesExt>(&self, data: &mut W) -> Result<()>;
|
|
}
|
|
|
|
//------------------------------------------
|
|
|
|
impl Unpack for u64 {
|
|
fn disk_size() -> u32 {
|
|
8
|
|
}
|
|
|
|
fn unpack(i: &[u8]) -> IResult<&[u8], u64> {
|
|
le_u64(i)
|
|
}
|
|
}
|
|
|
|
impl Pack for u64 {
|
|
fn pack<W: WriteBytesExt>(&self, out: &mut W) -> Result<()> {
|
|
out.write_u64::<LittleEndian>(*self)?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl Unpack for u32 {
|
|
fn disk_size() -> u32 {
|
|
4
|
|
}
|
|
|
|
fn unpack(i: &[u8]) -> IResult<&[u8], u32> {
|
|
le_u32(i)
|
|
}
|
|
}
|
|
|
|
impl Pack for u32 {
|
|
fn pack<W: WriteBytesExt>(&self, out: &mut W) -> Result<()> {
|
|
out.write_u32::<LittleEndian>(*self)?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
//------------------------------------------
|