Implement custom Debug impl for BlockHeaderHash

Includes a testcase and write_all implementation for Sha256dWriter.

Related to #63
This commit is contained in:
Deirdre Connolly 2019-10-15 18:51:53 -04:00 committed by Henry de Valence
parent f6e62b0f5e
commit 539a16979b
3 changed files with 48 additions and 3 deletions

View File

@ -11,5 +11,6 @@ edition = "2018"
thiserror = "1"
byteorder = "1.3"
chrono = "0.4"
#hex = "0.4" This conflicts with tracing-subscriber?
failure = "0.1"
hex = "0.4"
sha2 = "0.8"

View File

@ -1,7 +1,8 @@
//! Definitions of block datastructures.
use chrono::{DateTime, Utc};
use std::io;
use hex;
use std::{fmt, io};
use crate::merkle_tree::MerkleTreeRootHash;
use crate::note_commitment_tree::SaplingNoteTreeRootHash;
@ -21,9 +22,17 @@ use crate::transaction::Transaction;
/// the direct bytes of the transactions as well as the header. So
/// for now I want to call it a `BlockHeaderHash` because that's
/// more explicit.
#[derive(Copy, Clone, Debug, Eq, PartialEq)]
#[derive(Copy, Clone, Eq, PartialEq)]
pub struct BlockHeaderHash(pub [u8; 32]);
impl fmt::Debug for BlockHeaderHash {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.debug_tuple("BlockHeaderHash")
.field(&hex::encode(&self.0))
.finish()
}
}
impl From<BlockHeader> for BlockHeaderHash {
fn from(block_header: BlockHeader) -> Self {
let mut hash_writer = Sha256dWriter::default();
@ -141,3 +150,27 @@ impl ZcashDeserialize for Block {
unimplemented!();
}
}
#[cfg(test)]
mod tests {
use std::io::Write;
use super::BlockHeaderHash;
use crate::sha256d_writer::Sha256dWriter;
#[test]
fn test_blockheaderhash_debug() {
let preimage = b"foo bar baz";
let mut sha_writer = Sha256dWriter::default();
let _ = sha_writer.write_all(preimage);
let hash = BlockHeaderHash(sha_writer.finish());
assert_eq!(
format!("{:?}", hash),
"BlockHeaderHash(\"bf46b4b5030752fedac6f884976162bbfb29a9398f104a280b3e34d51b416631\")"
);
}
}

View File

@ -27,6 +27,17 @@ impl Write for Sha256dWriter {
Ok(buf.len())
}
fn write_all(&mut self, buf: &[u8]) -> std::io::Result<()> {
let mut length = 0;
while length != buf.len() {
length += buf.len();
self.hash.input(buf);
}
Ok(())
}
fn flush(&mut self) -> std::io::Result<()> {
Ok(())
}