octo
/
Zebra
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
zebra-chain/src/equihash_solution.rs
View File

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