Write and read the equihash solution compactsize on (de)serialize

This commit is contained in:
Deirdre Connolly 2020-02-04 04:23:33 -05:00 committed by Deirdre Connolly
parent a79969b38f
commit 6dedb7e101
1 changed files with 5 additions and 1 deletions

View File

@ -5,7 +5,9 @@ use std::{fmt, io};
#[cfg(test)]
use proptest::{arbitrary::Arbitrary, collection::vec, prelude::*};
use crate::serialization::{SerializationError, ZcashDeserialize, ZcashSerialize};
use crate::serialization::{
ReadZcashExt, SerializationError, WriteZcashExt, ZcashDeserialize, ZcashSerialize,
};
/// The size of an Equihash solution in bytes (always 1344).
const EQUIHASH_SOLUTION_SIZE: usize = 1344;
@ -50,6 +52,7 @@ impl Eq for EquihashSolution {}
impl ZcashSerialize for EquihashSolution {
fn zcash_serialize<W: io::Write>(&self, mut writer: W) -> Result<(), SerializationError> {
writer.write_compactsize(EQUIHASH_SOLUTION_SIZE as u64)?;
writer.write_all(&self.0[..])?;
Ok(())
}
@ -57,6 +60,7 @@ impl ZcashSerialize for EquihashSolution {
impl ZcashDeserialize for EquihashSolution {
fn zcash_deserialize<R: io::Read>(mut reader: R) -> Result<Self, SerializationError> {
reader.read_compactsize()?;
let mut bytes = [0; EQUIHASH_SOLUTION_SIZE];
reader.read_exact(&mut bytes[..])?;
Ok(Self(bytes))