Merge pull request #791 from fjl/discover-sha3-distance

p2p/discover: sha3-based node distance

1 files changed, 14 insertions, 1 deletions

diff --git a/common/types.go b/common/types.go
index daefcde11..183d48fb3 100644
--- a/common/types.go
+++ b/common/types.go
@@ -1,6 +1,10 @@
 package common
 
-import "math/big"
+import (
+	"math/big"
+	"math/rand"
+	"reflect"
+)
 
 const (
 	hashLength    = 32
@@ -48,6 +52,15 @@ func (h *Hash) Set(other Hash) {
 	}
 }
 
+// Generate implements testing/quick.Generator.
+func (h Hash) Generate(rand *rand.Rand, size int) reflect.Value {
+	m := rand.Intn(len(h))
+	for i := len(h) - 1; i > m; i-- {
+		h[i] = byte(rand.Uint32())
+	}
+	return reflect.ValueOf(h)
+}
+
 /////////// Address
 func BytesToAddress(b []byte) Address {
 	var a Address