mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-15 19:02:42 +00:00
c1a2238f1a
* Implement basic tree hash caching * Use spaces to indent top-level Cargo.toml * Optimize BLS tree hash by hashing bytes directly * Implement tree hash caching for validator registry * Persist BeaconState tree hash cache to disk * Address Paul's review comments
83 lines
2.3 KiB
Rust
83 lines
2.3 KiB
Rust
//! Provides a simple hash function utilizing `ring::digest::SHA256`.
|
|
//!
|
|
//! The purpose of this crate is to provide an abstraction to whatever hash function Ethereum
|
|
//! 2.0 is using. The hash function has been subject to change during the specification process, so
|
|
//! defining it once in this crate makes it easy to replace.
|
|
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
use ring::digest::{digest, SHA256};
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
use sha2::{Digest, Sha256};
|
|
|
|
#[cfg(feature = "zero_hash_cache")]
|
|
use lazy_static::lazy_static;
|
|
|
|
/// Returns the digest of `input`.
|
|
///
|
|
/// Uses `ring::digest::SHA256`.
|
|
pub fn hash(input: &[u8]) -> Vec<u8> {
|
|
#[cfg(not(target_arch = "wasm32"))]
|
|
let h = digest(&SHA256, input).as_ref().into();
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
let h = Sha256::digest(input).as_ref().into();
|
|
|
|
h
|
|
}
|
|
|
|
/// Compute the hash of two slices concatenated.
|
|
pub fn hash_concat(h1: &[u8], h2: &[u8]) -> Vec<u8> {
|
|
let mut vec1 = h1.to_vec();
|
|
vec1.extend_from_slice(h2);
|
|
hash(&vec1)
|
|
}
|
|
|
|
/// The max index that can be used with `ZERO_HASHES`.
|
|
#[cfg(feature = "zero_hash_cache")]
|
|
pub const ZERO_HASHES_MAX_INDEX: usize = 48;
|
|
|
|
#[cfg(feature = "zero_hash_cache")]
|
|
lazy_static! {
|
|
/// Cached zero hashes where `ZERO_HASHES[i]` is the hash of a Merkle tree with 2^i zero leaves.
|
|
pub static ref ZERO_HASHES: Vec<Vec<u8>> = {
|
|
let mut hashes = vec![vec![0; 32]; ZERO_HASHES_MAX_INDEX + 1];
|
|
|
|
for i in 0..ZERO_HASHES_MAX_INDEX {
|
|
hashes[i + 1] = hash_concat(&hashes[i], &hashes[i]);
|
|
}
|
|
|
|
hashes
|
|
};
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use rustc_hex::FromHex;
|
|
|
|
#[cfg(target_arch = "wasm32")]
|
|
use wasm_bindgen_test::*;
|
|
|
|
#[cfg_attr(not(target_arch = "wasm32"), test)]
|
|
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
|
|
fn test_hashing() {
|
|
let input: Vec<u8> = b"hello world".as_ref().into();
|
|
|
|
let output = hash(input.as_ref());
|
|
let expected_hex = "b94d27b9934d3e08a52e52d7da7dabfac484efe37a5380ee9088f7ace2efcde9";
|
|
let expected: Vec<u8> = expected_hex.from_hex().unwrap();
|
|
assert_eq!(expected, output);
|
|
}
|
|
|
|
#[cfg(feature = "zero_hash_cache")]
|
|
mod zero_hash {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn zero_hash_zero() {
|
|
assert_eq!(ZERO_HASHES[0], vec![0; 32]);
|
|
}
|
|
}
|
|
}
|