74 lines
1.8 KiB
Rust
74 lines
1.8 KiB
Rust
use ssz::{Decode, DecodeError, Encode, SszDecoderBuilder, SszEncoder};
|
|
|
|
#[derive(Debug, PartialEq)]
|
|
pub struct Foo {
|
|
a: u16,
|
|
b: Vec<u8>,
|
|
c: u16,
|
|
}
|
|
|
|
impl Encode for Foo {
|
|
fn is_ssz_fixed_len() -> bool {
|
|
<u16 as Encode>::is_ssz_fixed_len() && <Vec<u16> as Encode>::is_ssz_fixed_len()
|
|
}
|
|
|
|
fn ssz_bytes_len(&self) -> usize {
|
|
<u16 as Encode>::ssz_fixed_len()
|
|
+ ssz::BYTES_PER_LENGTH_OFFSET
|
|
+ <u16 as Encode>::ssz_fixed_len()
|
|
+ self.b.ssz_bytes_len()
|
|
}
|
|
|
|
fn ssz_append(&self, buf: &mut Vec<u8>) {
|
|
let offset = <u16 as Encode>::ssz_fixed_len()
|
|
+ <Vec<u16> as Encode>::ssz_fixed_len()
|
|
+ <u16 as Encode>::ssz_fixed_len();
|
|
|
|
let mut encoder = SszEncoder::container(buf, offset);
|
|
|
|
encoder.append(&self.a);
|
|
encoder.append(&self.b);
|
|
encoder.append(&self.c);
|
|
|
|
encoder.finalize();
|
|
}
|
|
}
|
|
|
|
impl Decode for Foo {
|
|
fn is_ssz_fixed_len() -> bool {
|
|
<u16 as Decode>::is_ssz_fixed_len() && <Vec<u16> as Decode>::is_ssz_fixed_len()
|
|
}
|
|
|
|
fn from_ssz_bytes(bytes: &[u8]) -> Result<Self, DecodeError> {
|
|
let mut builder = SszDecoderBuilder::new(bytes);
|
|
|
|
builder.register_type::<u16>()?;
|
|
builder.register_type::<Vec<u8>>()?;
|
|
builder.register_type::<u16>()?;
|
|
|
|
let mut decoder = builder.build()?;
|
|
|
|
Ok(Self {
|
|
a: decoder.decode_next()?,
|
|
b: decoder.decode_next()?,
|
|
c: decoder.decode_next()?,
|
|
})
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let foo = Foo {
|
|
a: 42,
|
|
b: vec![0, 1, 2, 3],
|
|
c: 11,
|
|
};
|
|
|
|
let bytes = vec![42, 0, 8, 0, 0, 0, 11, 0, 0, 1, 2, 3];
|
|
|
|
assert_eq!(foo.as_ssz_bytes(), bytes);
|
|
|
|
let decoded_foo = Foo::from_ssz_bytes(&bytes).unwrap();
|
|
|
|
assert_eq!(foo, decoded_foo);
|
|
}
|