diff --git a/crypto/Cargo.toml b/crypto/Cargo.toml index 4980086..87560ab 100644 --- a/crypto/Cargo.toml +++ b/crypto/Cargo.toml @@ -13,8 +13,7 @@ edition = "2021" [dependencies] winterfell = { git = "https://github.com/novifinancial/winterfell"} winter_utils = { version = "0.4", package = "winter-utils" } -rand_utils = { version = "0.4", package = "winter-rand-utils" } - [dev-dependencies] -proptest = "1.0.0" \ No newline at end of file +proptest = "1.0.0" +rand_utils = { version = "0.4", package = "winter-rand-utils" } \ No newline at end of file diff --git a/crypto/src/hash/mod.rs b/crypto/src/hash/mod.rs index a051c28..367f415 100644 --- a/crypto/src/hash/mod.rs +++ b/crypto/src/hash/mod.rs @@ -1,10 +1,6 @@ -use winterfell::crypto::hashers::Rp64_256 as Hasher; -use winterfell::crypto::ElementHasher; -use winterfell::crypto::Hasher as HashFn; -use winterfell::math::StarkField; - -mod rpo; -pub use rpo::Rpo; +use crate::StarkField; +pub use winterfell::crypto::hashers::Rp64_256 as Hasher; +pub use winterfell::crypto::{ElementHasher, Hasher as HashFn}; // TYPE ALIASES // ================================================================================================ diff --git a/crypto/src/merkle/merkle_path_set.rs b/crypto/src/merkle/merkle_path_set.rs index 72f705d..8010676 100644 --- a/crypto/src/merkle/merkle_path_set.rs +++ b/crypto/src/merkle/merkle_path_set.rs @@ -1,7 +1,6 @@ -use winter_utils::collections::BTreeMap; - use super::{Felt, FieldElement, MerkleError, Word}; use crate::hash::merge; +use winter_utils::collections::{BTreeMap, Vec}; // MERKLE PATH SET // ================================================================================================ @@ -255,9 +254,7 @@ fn compute_path_root(path: &[Word], depth: u32, index: u64) -> Word { #[cfg(test)] mod tests { - use super::calculate_parent_hash; - - use super::{Felt, FieldElement, Word}; + use super::{calculate_parent_hash, Felt, FieldElement, Word}; #[test] fn get_root() { diff --git a/crypto/src/merkle/merkle_tree.rs b/crypto/src/merkle/merkle_tree.rs index bbe976b..d784d30 100644 --- a/crypto/src/merkle/merkle_tree.rs +++ b/crypto/src/merkle/merkle_tree.rs @@ -1,12 +1,11 @@ -use core::slice; -use winter_utils::uninit_vector; -use winterfell::math::log2; - use super::MerkleError; use crate::{ hash::{merge, Digest}, Felt, FieldElement, Word, }; +use core::slice; +use winter_utils::uninit_vector; +use winterfell::math::log2; // MERKLE TREE // ================================================================================================ @@ -147,7 +146,7 @@ impl MerkleTree { #[cfg(test)] mod tests { use super::{Felt, FieldElement, Word}; - use winterfell::crypto::{hashers::Rp64_256, ElementHasher, Hasher}; + use crate::hash::{ElementHasher, HashFn, Hasher}; const LEAVES4: [Word; 4] = [ int_to_node(1), @@ -248,9 +247,9 @@ mod tests { // -------------------------------------------------------------------------------------------- fn compute_internal_nodes() -> (Word, Word, Word) { - let node2 = Rp64_256::hash_elements(&[LEAVES4[0], LEAVES4[1]].concat()); - let node3 = Rp64_256::hash_elements(&[LEAVES4[2], LEAVES4[3]].concat()); - let root = Rp64_256::merge(&[node2, node3]); + let node2 = Hasher::hash_elements(&[LEAVES4[0], LEAVES4[1]].concat()); + let node3 = Hasher::hash_elements(&[LEAVES4[2], LEAVES4[3]].concat()); + let root = Hasher::merge(&[node2, node3]); (root.into(), node2.into(), node3.into()) } diff --git a/crypto/src/merkle/mod.rs b/crypto/src/merkle/mod.rs index 3ab61e8..290064f 100644 --- a/crypto/src/merkle/mod.rs +++ b/crypto/src/merkle/mod.rs @@ -1,8 +1,8 @@ +use crate::{Felt, FieldElement, Word}; + pub mod merkle_path_set; pub mod merkle_tree; -use crate::{Felt, FieldElement, Word}; - // ERRORS // ================================================================================================