You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
263 lines
5.6 KiB
263 lines
5.6 KiB
package merkle
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/sha256"
|
|
"fmt"
|
|
|
|
. "github.com/tendermint/tendermint/binary"
|
|
)
|
|
|
|
func HashFromTwoHashes(left []byte, right []byte) []byte {
|
|
var n int64
|
|
var err error
|
|
var hasher = sha256.New()
|
|
WriteByteSlice(hasher, left, &n, &err)
|
|
WriteByteSlice(hasher, right, &n, &err)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return hasher.Sum(nil)
|
|
}
|
|
|
|
/*
|
|
Computes a deterministic minimal height merkle tree hash.
|
|
If the number of items is not a power of two, some leaves
|
|
will be at different levels.
|
|
|
|
*
|
|
/ \
|
|
/ \
|
|
/ \
|
|
/ \
|
|
* *
|
|
/ \ / \
|
|
/ \ / \
|
|
/ \ / \
|
|
* h2 * *
|
|
/ \ / \ / \
|
|
h0 h1 h3 h4 h5 h6
|
|
|
|
*/
|
|
func HashFromHashes(hashes [][]byte) []byte {
|
|
switch len(hashes) {
|
|
case 0:
|
|
return nil
|
|
case 1:
|
|
return hashes[0]
|
|
default:
|
|
left := HashFromHashes(hashes[:len(hashes)/2])
|
|
right := HashFromHashes(hashes[len(hashes)/2:])
|
|
return HashFromTwoHashes(left, right)
|
|
}
|
|
}
|
|
|
|
// Convenience for HashFromHashes.
|
|
func HashFromBinaries(items []Binary) []byte {
|
|
hashes := [][]byte{}
|
|
for _, item := range items {
|
|
hasher := sha256.New()
|
|
_, err := item.WriteTo(hasher)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
hash := hasher.Sum(nil)
|
|
hashes = append(hashes, hash)
|
|
}
|
|
return HashFromHashes(hashes)
|
|
}
|
|
|
|
// Convenience for HashFromHashes.
|
|
func HashFromHashables(items []Hashable) []byte {
|
|
hashes := [][]byte{}
|
|
for _, item := range items {
|
|
hash := item.Hash()
|
|
hashes = append(hashes, hash)
|
|
}
|
|
return HashFromHashes(hashes)
|
|
}
|
|
|
|
/*
|
|
Calculates an array of hashes, useful for deriving hash trails.
|
|
|
|
7
|
|
/ \
|
|
/ \
|
|
/ \
|
|
/ \
|
|
3 11
|
|
/ \ / \
|
|
/ \ / \
|
|
/ \ / \
|
|
1 5 9 13
|
|
/ \ / \ / \ / \
|
|
0 2 4 6 8 10 12 14
|
|
h0 h1 h2 h3 h4 h5 h6 h7
|
|
|
|
(diagram and idea borrowed from libswift)
|
|
|
|
The hashes provided get assigned to even indices.
|
|
The derived merkle hashes get assigned to odd indices.
|
|
If "hashes" is not of length power of 2, it is padded
|
|
with blank (zeroed) hashes.
|
|
*/
|
|
func HashTreeFromHashes(hashes [][]byte) [][]byte {
|
|
|
|
// Make length of "hashes" a power of 2
|
|
hashesLen := uint32(len(hashes))
|
|
fullLen := uint32(1)
|
|
for {
|
|
if fullLen >= hashesLen {
|
|
break
|
|
} else {
|
|
fullLen <<= 1
|
|
}
|
|
}
|
|
blank := make([]byte, len(hashes[0]))
|
|
for i := hashesLen; i < fullLen; i++ {
|
|
hashes = append(hashes, blank)
|
|
}
|
|
|
|
// The result is twice the length minus one.
|
|
res := make([][]byte, len(hashes)*2-1)
|
|
for i, hash := range hashes {
|
|
res[i*2] = hash
|
|
}
|
|
|
|
// Fill all the hashes recursively.
|
|
fillTreeRoot(res, 0, len(res)-1)
|
|
return res
|
|
}
|
|
|
|
// Fill in the blanks.
|
|
func fillTreeRoot(res [][]byte, start, end int) []byte {
|
|
if start == end {
|
|
return res[start]
|
|
} else {
|
|
mid := (start + end) / 2
|
|
left := fillTreeRoot(res, start, mid-1)
|
|
right := fillTreeRoot(res, mid+1, end)
|
|
root := HashFromTwoHashes(left, right)
|
|
res[mid] = root
|
|
return root
|
|
}
|
|
}
|
|
|
|
// Convenience for HashTreeFromHashes.
|
|
func HashTreeFromHashables(items []Hashable) [][]byte {
|
|
hashes := [][]byte{}
|
|
for _, item := range items {
|
|
hash := item.Hash()
|
|
hashes = append(hashes, hash)
|
|
}
|
|
return HashTreeFromHashes(hashes)
|
|
}
|
|
|
|
/*
|
|
Given the original index of an item,
|
|
(e.g. for h5 in the diagram above, the index is 5, not 10)
|
|
returns a trail of hashes, which along with the index can be
|
|
used to calculate the merkle root.
|
|
|
|
See VerifyHashTrailForIndex()
|
|
*/
|
|
func HashTrailForIndex(hashTree [][]byte, index int) [][]byte {
|
|
trail := [][]byte{}
|
|
index *= 2
|
|
|
|
// We start from the leaf layer and work our way up.
|
|
// Notice the indices in the diagram:
|
|
// 0 2 4 ... offset 0, stride 2
|
|
// 1 5 9 ... offset 1, stride 4
|
|
// 3 11 19 ... offset 3, stride 8
|
|
// 7 23 39 ... offset 7, stride 16 etc.
|
|
|
|
offset := 0
|
|
stride := 2
|
|
|
|
for {
|
|
// Calculate sibling of index.
|
|
var next int
|
|
if ((index-offset)/stride)%2 == 0 {
|
|
next = index + stride
|
|
} else {
|
|
next = index - stride
|
|
}
|
|
if next >= len(hashTree) {
|
|
break
|
|
}
|
|
// Insert sibling hash to trail.
|
|
trail = append(trail, hashTree[next])
|
|
|
|
index = (index + next) / 2
|
|
offset += stride
|
|
stride *= 2
|
|
}
|
|
|
|
return trail
|
|
}
|
|
|
|
// Ensures that leafHash is part of rootHash.
|
|
func VerifyHashTrailForIndex(index int, leafHash []byte, trail [][]byte, rootHash []byte) bool {
|
|
index *= 2
|
|
offset := 0
|
|
stride := 2
|
|
|
|
tempHash := make([]byte, len(leafHash))
|
|
copy(tempHash, leafHash)
|
|
|
|
for i := 0; i < len(trail); i++ {
|
|
var next int
|
|
if ((index-offset)/stride)%2 == 0 {
|
|
next = index + stride
|
|
tempHash = HashFromTwoHashes(tempHash, trail[i])
|
|
} else {
|
|
next = index - stride
|
|
tempHash = HashFromTwoHashes(trail[i], tempHash)
|
|
}
|
|
index = (index + next) / 2
|
|
offset += stride
|
|
stride *= 2
|
|
}
|
|
|
|
return bytes.Equal(rootHash, tempHash)
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
func PrintIAVLNode(node *IAVLNode) {
|
|
fmt.Println("==== NODE")
|
|
if node != nil {
|
|
printIAVLNode(node, 0)
|
|
}
|
|
fmt.Println("==== END")
|
|
}
|
|
|
|
func printIAVLNode(node *IAVLNode, indent int) {
|
|
indentPrefix := ""
|
|
for i := 0; i < indent; i++ {
|
|
indentPrefix += " "
|
|
}
|
|
|
|
if node.rightNode != nil {
|
|
printIAVLNode(node.rightNode, indent+1)
|
|
} else if node.rightHash != nil {
|
|
fmt.Printf("%s %X\n", indentPrefix, node.rightHash)
|
|
}
|
|
|
|
fmt.Printf("%s%v:%v\n", indentPrefix, node.key, node.height)
|
|
|
|
if node.leftNode != nil {
|
|
printIAVLNode(node.leftNode, indent+1)
|
|
} else if node.leftHash != nil {
|
|
fmt.Printf("%s %X\n", indentPrefix, node.leftHash)
|
|
}
|
|
|
|
}
|
|
|
|
func maxUint8(a, b uint8) uint8 {
|
|
if a > b {
|
|
return a
|
|
}
|
|
return b
|
|
}
|