forked from perlin-network/noise
-
Notifications
You must be signed in to change notification settings - Fork 0
/
identity.go
72 lines (59 loc) · 1.96 KB
/
identity.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package peer
import (
"bytes"
"encoding/hex"
"fmt"
"math/bits"
"github.com/perlin-network/noise/crypto/blake2b"
"github.com/perlin-network/noise/internal/protobuf"
)
// ID is an identity of nodes, using its public key hash and network address.
type ID protobuf.ID
// CreateID is a factory function creating ID.
func CreateID(address string, publicKey []byte) ID {
return ID{Address: address, PublicKey: publicKey, Id: blake2b.New().HashBytes(publicKey)}
}
// String returns the identity address and public key.
func (id ID) String() string {
return fmt.Sprintf("ID{Address: %v, Id: %v}", id.Address, id.Id)
}
// Equals determines if two peer IDs are equal to each other based on the contents of their public keys.
func (id ID) Equals(other ID) bool {
return bytes.Equal(id.Id, other.Id)
}
// Less determines if this peer ID's public key is less than other ID's public key.
func (id ID) Less(other interface{}) bool {
if other, is := other.(ID); is {
return bytes.Compare(id.Id, other.Id) == -1
}
return false
}
// PublicKeyHex generates a hex-encoded string of public key hash of this given peer ID.
func (id ID) PublicKeyHex() string {
return hex.EncodeToString(id.PublicKey)
}
// Xor performs XOR (^) over another peer ID's public key.
func (id ID) Xor(other ID) ID {
result := make([]byte, len(id.PublicKey))
for i := 0; i < len(id.PublicKey) && i < len(other.PublicKey); i++ {
result[i] = id.PublicKey[i] ^ other.PublicKey[i]
}
return ID{Address: id.Address, PublicKey: result}
}
// XorID performs XOR (^) over another peer ID's public key hash.
func (id ID) XorID(other ID) ID {
result := make([]byte, len(id.Id))
for i := 0; i < len(id.Id) && i < len(other.Id); i++ {
result[i] = id.Id[i] ^ other.Id[i]
}
return ID{Address: id.Address, Id: result}
}
// PrefixLen returns the number of prefixed zeros in a peer ID.
func (id ID) PrefixLen() int {
for i, b := range id.Id {
if b != 0 {
return i*8 + bits.LeadingZeros8(uint8(b))
}
}
return len(id.Id)*8 - 1
}