uncommented triehash

This commit is contained in:
debris 2015-11-29 19:42:03 +01:00
parent 03c1944d9c
commit f4292134c0
1 changed files with 260 additions and 261 deletions

View File

@ -1,300 +1,299 @@
//! Generete trie root //! Generete trie root
//use std::collections::BTreeMap; use std::collections::BTreeMap;
//use std::cmp; use std::cmp;
//use hash::*; use hash::*;
//use sha3::*; use sha3::*;
//use rlp; use rlp;
//use rlp::RlpStream; use rlp::RlpStream;
//use vector::SharedPrefix; use vector::SharedPrefix;
///// Hex-prefix Notation. First nibble has flags: oddness = 2^0 & termination = 2^1. /// Hex-prefix Notation. First nibble has flags: oddness = 2^0 & termination = 2^1.
///// ///
///// The "termination marker" and "leaf-node" specifier are completely equivalent. /// The "termination marker" and "leaf-node" specifier are completely equivalent.
///// ///
///// Input values are in range `[0, 0xf]`. /// Input values are in range `[0, 0xf]`.
///// ///
///// ```markdown /// ```markdown
///// [0,0,1,2,3,4,5] 0x10012345 // 7 > 4 /// [0,0,1,2,3,4,5] 0x10012345 // 7 > 4
///// [0,1,2,3,4,5] 0x00012345 // 6 > 4 /// [0,1,2,3,4,5] 0x00012345 // 6 > 4
///// [1,2,3,4,5] 0x112345 // 5 > 3 /// [1,2,3,4,5] 0x112345 // 5 > 3
///// [0,0,1,2,3,4] 0x00001234 // 6 > 3 /// [0,0,1,2,3,4] 0x00001234 // 6 > 3
///// [0,1,2,3,4] 0x101234 // 5 > 3 /// [0,1,2,3,4] 0x101234 // 5 > 3
///// [1,2,3,4] 0x001234 // 4 > 3 /// [1,2,3,4] 0x001234 // 4 > 3
///// [0,0,1,2,3,4,5,T] 0x30012345 // 7 > 4 /// [0,0,1,2,3,4,5,T] 0x30012345 // 7 > 4
///// [0,0,1,2,3,4,T] 0x20001234 // 6 > 4 /// [0,0,1,2,3,4,T] 0x20001234 // 6 > 4
///// [0,1,2,3,4,5,T] 0x20012345 // 6 > 4 /// [0,1,2,3,4,5,T] 0x20012345 // 6 > 4
///// [1,2,3,4,5,T] 0x312345 // 5 > 3 /// [1,2,3,4,5,T] 0x312345 // 5 > 3
///// [1,2,3,4,T] 0x201234 // 4 > 3 /// [1,2,3,4,T] 0x201234 // 4 > 3
///// ``` /// ```
///// ///
///// ```rust /// ```rust
///// extern crate ethcore_util as util; /// extern crate ethcore_util as util;
///// use util::triehash::*; /// use util::triehash::*;
///// ///
///// fn main() { /// fn main() {
///// let v = vec![0, 0, 1, 2, 3, 4, 5]; /// let v = vec![0, 0, 1, 2, 3, 4, 5];
///// let e = vec![0x10, 0x01, 0x23, 0x45]; /// let e = vec![0x10, 0x01, 0x23, 0x45];
///// let h = hex_prefix_encode(&v, false); /// let h = hex_prefix_encode(&v, false);
///// assert_eq!(h, e); /// assert_eq!(h, e);
///// ///
///// let v = vec![0, 1, 2, 3, 4, 5]; /// let v = vec![0, 1, 2, 3, 4, 5];
///// let e = vec![0x00, 0x01, 0x23, 0x45]; /// let e = vec![0x00, 0x01, 0x23, 0x45];
///// let h = hex_prefix_encode(&v, false); /// let h = hex_prefix_encode(&v, false);
///// assert_eq!(h, e); /// assert_eq!(h, e);
///// ///
///// let v = vec![0, 1, 2, 3, 4, 5]; /// let v = vec![0, 1, 2, 3, 4, 5];
///// let e = vec![0x20, 0x01, 0x23, 0x45]; /// let e = vec![0x20, 0x01, 0x23, 0x45];
///// let h = hex_prefix_encode(&v, true); /// let h = hex_prefix_encode(&v, true);
///// assert_eq!(h, e); /// assert_eq!(h, e);
///// ///
///// let v = vec![1, 2, 3, 4, 5]; /// let v = vec![1, 2, 3, 4, 5];
///// let e = vec![0x31, 0x23, 0x45]; /// let e = vec![0x31, 0x23, 0x45];
///// let h = hex_prefix_encode(&v, true); /// let h = hex_prefix_encode(&v, true);
///// assert_eq!(h, e); /// assert_eq!(h, e);
///// ///
///// let v = vec![1, 2, 3, 4]; /// let v = vec![1, 2, 3, 4];
///// let e = vec![0x00, 0x12, 0x34]; /// let e = vec![0x00, 0x12, 0x34];
///// let h = hex_prefix_encode(&v, false); /// let h = hex_prefix_encode(&v, false);
///// assert_eq!(h, e); /// assert_eq!(h, e);
///// ///
///// let v = vec![4, 1]; /// let v = vec![4, 1];
///// let e = vec![0x20, 0x41]; /// let e = vec![0x20, 0x41];
///// let h = hex_prefix_encode(&v, true); /// let h = hex_prefix_encode(&v, true);
///// assert_eq!(h, e); /// assert_eq!(h, e);
///// } /// }
///// ``` /// ```
///// ///
//pub fn hex_prefix_encode(nibbles: &[u8], leaf: bool) -> Vec<u8> { pub fn hex_prefix_encode(nibbles: &[u8], leaf: bool) -> Vec<u8> {
//let inlen = nibbles.len(); let inlen = nibbles.len();
//let oddness_factor = inlen % 2; let oddness_factor = inlen % 2;
//// next even number divided by two // next even number divided by two
//let reslen = (inlen + 2) >> 1; let reslen = (inlen + 2) >> 1;
//let mut res = vec![]; let mut res = vec![];
//res.reserve(reslen); res.reserve(reslen);
//let first_byte = { let first_byte = {
//let mut bits = ((inlen as u8 & 1) + (2 * leaf as u8)) << 4; let mut bits = ((inlen as u8 & 1) + (2 * leaf as u8)) << 4;
//if oddness_factor == 1 { if oddness_factor == 1 {
//bits += nibbles[0]; bits += nibbles[0];
//} }
//bits bits
//}; };
//res.push(first_byte); res.push(first_byte);
//let mut offset = oddness_factor; let mut offset = oddness_factor;
//while offset < inlen { while offset < inlen {
//let byte = (nibbles[offset] << 4) + nibbles[offset + 1]; let byte = (nibbles[offset] << 4) + nibbles[offset + 1];
//res.push(byte); res.push(byte);
//offset += 2; offset += 2;
//} }
//res res
//} }
///// Converts slice of bytes to nibbles. /// Converts slice of bytes to nibbles.
///// ///
///// ```rust /// ```rust
///// extern crate ethcore_util as util; /// extern crate ethcore_util as util;
///// use util::triehash::*; /// use util::triehash::*;
///// ///
///// fn main () { /// fn main () {
///// let v = vec![0x31, 0x23, 0x45]; /// let v = vec![0x31, 0x23, 0x45];
///// let e = vec![3, 1, 2, 3, 4, 5]; /// let e = vec![3, 1, 2, 3, 4, 5];
///// assert_eq!(as_nibbles(&v), e); /// assert_eq!(as_nibbles(&v), e);
///// ///
///// // A => 65 => 0x41 => [4, 1] /// // A => 65 => 0x41 => [4, 1]
///// let v: Vec<u8> = From::from("A"); /// let v: Vec<u8> = From::from("A");
///// let e = vec![4, 1]; /// let e = vec![4, 1];
///// assert_eq!(as_nibbles(&v), e); /// assert_eq!(as_nibbles(&v), e);
///// } /// }
///// ``` /// ```
//pub fn as_nibbles(bytes: &[u8]) -> Vec<u8> { pub fn as_nibbles(bytes: &[u8]) -> Vec<u8> {
//let mut res = vec![]; let mut res = vec![];
//res.reserve(bytes.len() * 2); res.reserve(bytes.len() * 2);
//for i in 0..bytes.len() { for i in 0..bytes.len() {
//res.push(bytes[i] >> 4); res.push(bytes[i] >> 4);
//res.push((bytes[i] << 4) >> 4); res.push((bytes[i] << 4) >> 4);
//} }
//res res
//} }
//#[derive(Debug)] #[derive(Debug)]
//pub struct NibblePair { pub struct NibblePair {
//nibble: Vec<u8>, nibble: Vec<u8>,
//data: Vec<u8> data: Vec<u8>
//} }
//impl NibblePair { impl NibblePair {
//pub fn new(nibble: Vec<u8>, data: Vec<u8>) -> NibblePair { pub fn new(nibble: Vec<u8>, data: Vec<u8>) -> NibblePair {
//NibblePair { NibblePair {
//nibble: nibble, nibble: nibble,
//data: data data: data
//} }
//} }
//pub fn new_raw(to_nibble: Vec<u8>, data: Vec<u8>) -> NibblePair { pub fn new_raw(to_nibble: Vec<u8>, data: Vec<u8>) -> NibblePair {
//NibblePair::new(as_nibbles(&to_nibble), data) NibblePair::new(as_nibbles(&to_nibble), data)
//} }
//} }
//pub fn ordered_trie_root(data: Vec<Vec<u8>>) -> H256 { pub fn ordered_trie_root(data: Vec<Vec<u8>>) -> H256 {
//let vec: Vec<NibblePair> = data let vec: Vec<NibblePair> = data
//// first put elements into btree to sort them by nibbles // first put elements into btree to sort them by nibbles
//// optimize it later // optimize it later
//.into_iter() .into_iter()
//.fold(BTreeMap::new(), | mut acc, vec | { .fold(BTreeMap::new(), | mut acc, vec | {
//let len = acc.len(); let len = acc.len();
//acc.insert(as_nibbles(&rlp::encode(&len)), vec); acc.insert(as_nibbles(&rlp::encode(&len)), vec);
//acc acc
//}) })
//// then move them to a vector // then move them to a vector
//.into_iter() .into_iter()
//.map(|(k, v)| NibblePair::new(k, v) ) .map(|(k, v)| NibblePair::new(k, v) )
//.collect(); .collect();
//hash256(&vec) hash256(&vec)
//} }
//pub fn hash256(vec: &[NibblePair]) -> H256 { pub fn hash256(vec: &[NibblePair]) -> H256 {
//let out = match vec.len() { let out = match vec.len() {
//0 => rlp::encode(&""), 0 => rlp::encode(&""),
//_ => { _ => {
//let mut stream = RlpStream::new(); let mut stream = RlpStream::new();
//hash256rlp(&vec, 0, &mut stream); hash256rlp(&vec, 0, &mut stream);
//stream.out().unwrap() stream.out().unwrap()
//} }
//}; };
//println!("out: {:?}", out); println!("out: {:?}", out);
//out.sha3() out.sha3()
//} }
//fn hash256rlp(vec: &[NibblePair], pre_len: usize, stream: &mut RlpStream) { fn hash256rlp(vec: &[NibblePair], pre_len: usize, stream: &mut RlpStream) {
//match vec.len() { match vec.len() {
//0 => { 0 => {
//stream.append(&""); stream.append(&"");
//}, },
//1 => { 1 => {
//stream.append_list(2); stream.append_list(2);
//stream.append(&hex_prefix_encode(&vec[0].nibble[pre_len..], true)); stream.append(&hex_prefix_encode(&vec[0].nibble[pre_len..], true));
//stream.append(&vec[0].data); stream.append(&vec[0].data);
//}, },
//_ => { _ => {
//let shared_prefix = vec.iter() let shared_prefix = vec.iter()
//// skip first element // skip first element
//.skip(1) .skip(1)
//// get minimum number of shared nibbles between first and each successive // get minimum number of shared nibbles between first and each successive
//.fold(usize::max_value(), | acc, pair | { .fold(usize::max_value(), | acc, pair | {
//cmp::min(vec[0].nibble.shared_prefix_len(&pair.nibble), acc) cmp::min(vec[0].nibble.shared_prefix_len(&pair.nibble), acc)
//}); });
//match shared_prefix > pre_len { match shared_prefix > pre_len {
//true => { true => {
//stream.append_list(2); stream.append_list(2);
//stream.append(&hex_prefix_encode(&vec[0].nibble[pre_len..shared_prefix], false)); stream.append(&hex_prefix_encode(&vec[0].nibble[pre_len..shared_prefix], false));
//hash256aux(vec, shared_prefix, stream); hash256aux(vec, shared_prefix, stream);
//}, },
//false => { false => {
//stream.append_list(17); stream.append_list(17);
//// every nibble is longer then previous one // every nibble is longer then previous one
//let iter = vec.iter() let iter = vec.iter()
//// move to first element with len different then pre_len // move to first element with len different then pre_len
//.take_while(| pair | { pair.nibble.len() == pre_len }); .take_while(| pair | { pair.nibble.len() == pre_len });
//let mut begin = iter.count(); let mut begin = iter.count();
//for i in 0..16 { for i in 0..16 {
//// cout how many successive elements have same next nibble // cout how many successive elements have same next nibble
//let len = vec[begin..].iter() let len = vec[begin..].iter()
//.map(| pair | pair.nibble[pre_len] ) .map(| pair | pair.nibble[pre_len] )
//.take_while(|&q| q == i).count(); .take_while(|&q| q == i).count();
//match len { match len {
//0 => { stream.append(&""); }, 0 => { stream.append(&""); },
//_ => hash256aux(&vec[begin..begin + len], pre_len + 1, stream) _ => hash256aux(&vec[begin..begin + len], pre_len + 1, stream)
//} }
//begin += len; begin += len;
//} }
//match pre_len == vec[0].nibble.len() { match pre_len == vec[0].nibble.len() {
//true => stream.append(&vec[0].data), true => stream.append(&vec[0].data),
//false => stream.append(&"") false => stream.append(&"")
//}; };
//} }
//} }
//} }
//}; };
//} }
//fn hash256aux(vec: &[NibblePair], pre_len: usize, stream: &mut RlpStream) { fn hash256aux(vec: &[NibblePair], pre_len: usize, stream: &mut RlpStream) {
//let mut s = RlpStream::new(); let mut s = RlpStream::new();
//hash256rlp(vec, pre_len, &mut s); hash256rlp(vec, pre_len, &mut s);
//let out = s.out().unwrap(); let out = s.out().unwrap();
//match out.len() { match out.len() {
//0...31 => stream.append_raw(&out, 1), 0...31 => stream.append_raw(&out, 1),
//_ => stream.append(&out.sha3()) _ => stream.append(&out.sha3())
//}; };
//} }
//#[cfg(test)] #[cfg(test)]
//mod tests { mod tests {
//use std::str::FromStr; use std::str::FromStr;
//use rustc_serialize::hex::FromHex; use rustc_serialize::hex::FromHex;
//use hash::*; use hash::*;
//use triehash::*; use triehash::*;
//#[test] #[test]
//fn empty_trie_root() { fn empty_trie_root() {
//assert_eq!(hash256(&vec![]), H256::from_str("56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421").unwrap()); assert_eq!(hash256(&vec![]), H256::from_str("56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421").unwrap());
//} }
//#[test] #[test]
//fn single_trie_item() { fn single_trie_item() {
//let v = vec![ let v = vec![
//NibblePair::new_raw(From::from("A"), NibblePair::new_raw(From::from("A"),
//From::from("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")) From::from("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"))
//]; ];
//println!("{:?}", v);
//assert_eq!(hash256(&v), H256::from_str("d23786fb4a010da3ce639d66d5e904a11dbc02746d1ce25029e53290cabf28ab").unwrap()); assert_eq!(hash256(&v), H256::from_str("d23786fb4a010da3ce639d66d5e904a11dbc02746d1ce25029e53290cabf28ab").unwrap());
//} }
//#[test] #[test]
//fn test_trie_root() { fn test_trie_root() {
//let v = vec![ let _v = vec![
//NibblePair::new_raw("0000000000000000000000000000000000000000000000000000000000000045".from_hex().unwrap(), NibblePair::new_raw("0000000000000000000000000000000000000000000000000000000000000045".from_hex().unwrap(),
//"22b224a1420a802ab51d326e29fa98e34c4f24ea".from_hex().unwrap()), "22b224a1420a802ab51d326e29fa98e34c4f24ea".from_hex().unwrap()),
//NibblePair::new_raw("0000000000000000000000000000000000000000000000000000000000000046".from_hex().unwrap(), NibblePair::new_raw("0000000000000000000000000000000000000000000000000000000000000046".from_hex().unwrap(),
//"67706c2076330000000000000000000000000000000000000000000000000000".from_hex().unwrap()), "67706c2076330000000000000000000000000000000000000000000000000000".from_hex().unwrap()),
//NibblePair::new_raw("000000000000000000000000697c7b8c961b56f675d570498424ac8de1a918f6".from_hex().unwrap(), NibblePair::new_raw("000000000000000000000000697c7b8c961b56f675d570498424ac8de1a918f6".from_hex().unwrap(),
//"6f6f6f6820736f2067726561742c207265616c6c6c793f000000000000000000".from_hex().unwrap()), "6f6f6f6820736f2067726561742c207265616c6c6c793f000000000000000000".from_hex().unwrap()),
//NibblePair::new_raw("0000000000000000000000007ef9e639e2733cb34e4dfc576d4b23f72db776b2".from_hex().unwrap(), NibblePair::new_raw("0000000000000000000000007ef9e639e2733cb34e4dfc576d4b23f72db776b2".from_hex().unwrap(),
//"4655474156000000000000000000000000000000000000000000000000000000".from_hex().unwrap()), "4655474156000000000000000000000000000000000000000000000000000000".from_hex().unwrap()),
//NibblePair::new_raw("000000000000000000000000ec4f34c97e43fbb2816cfd95e388353c7181dab1".from_hex().unwrap(), NibblePair::new_raw("000000000000000000000000ec4f34c97e43fbb2816cfd95e388353c7181dab1".from_hex().unwrap(),
//"4e616d6552656700000000000000000000000000000000000000000000000000".from_hex().unwrap()), "4e616d6552656700000000000000000000000000000000000000000000000000".from_hex().unwrap()),
//NibblePair::new_raw("4655474156000000000000000000000000000000000000000000000000000000".from_hex().unwrap(), NibblePair::new_raw("4655474156000000000000000000000000000000000000000000000000000000".from_hex().unwrap(),
//"7ef9e639e2733cb34e4dfc576d4b23f72db776b2".from_hex().unwrap()), "7ef9e639e2733cb34e4dfc576d4b23f72db776b2".from_hex().unwrap()),
//NibblePair::new_raw("4e616d6552656700000000000000000000000000000000000000000000000000".from_hex().unwrap(), NibblePair::new_raw("4e616d6552656700000000000000000000000000000000000000000000000000".from_hex().unwrap(),
//"ec4f34c97e43fbb2816cfd95e388353c7181dab1".from_hex().unwrap()), "ec4f34c97e43fbb2816cfd95e388353c7181dab1".from_hex().unwrap()),
//NibblePair::new_raw("6f6f6f6820736f2067726561742c207265616c6c6c793f000000000000000000".from_hex().unwrap(), NibblePair::new_raw("6f6f6f6820736f2067726561742c207265616c6c6c793f000000000000000000".from_hex().unwrap(),
//"697c7b8c961b56f675d570498424ac8de1a918f6".from_hex().unwrap()) "697c7b8c961b56f675d570498424ac8de1a918f6".from_hex().unwrap())
//]; ];
//let root = H256::from_str("9f6221ebb8efe7cff60a716ecb886e67dd042014be444669f0159d8e68b42100").unwrap(); let _root = H256::from_str("9f6221ebb8efe7cff60a716ecb886e67dd042014be444669f0159d8e68b42100").unwrap();
//let res = hash256(&v); //let res = hash256(&v);
//println!("{:?}", res); //println!("{:?}", res);
////assert!(false); //assert_eq!(res, root);
//} }
//} }