mirror of
https://github.com/ipfs/kubo.git
synced 2025-09-09 17:22:21 +08:00
89f5cd4c94
I think it's time to move a lot of the peer-to-peer networking but-not-ipfs-specific things into its own package: p2p. This could in the future be split off into its own library. The first thing to go is the peer.
63 lines
1.4 KiB
Go
63 lines
1.4 KiB
Go
package kbucket
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/sha256"
|
|
"errors"
|
|
|
|
peer "github.com/jbenet/go-ipfs/p2p/peer"
|
|
ks "github.com/jbenet/go-ipfs/routing/keyspace"
|
|
u "github.com/jbenet/go-ipfs/util"
|
|
)
|
|
|
|
// Returned if a routing table query returns no results. This is NOT expected
|
|
// behaviour
|
|
var ErrLookupFailure = errors.New("failed to find any peer in table")
|
|
|
|
// ID for IpfsDHT is in the XORKeySpace
|
|
//
|
|
// The type dht.ID signifies that its contents have been hashed from either a
|
|
// peer.ID or a util.Key. This unifies the keyspace
|
|
type ID []byte
|
|
|
|
func (id ID) equal(other ID) bool {
|
|
return bytes.Equal(id, other)
|
|
}
|
|
|
|
func (id ID) less(other ID) bool {
|
|
a := ks.Key{Space: ks.XORKeySpace, Bytes: id}
|
|
b := ks.Key{Space: ks.XORKeySpace, Bytes: other}
|
|
return a.Less(b)
|
|
}
|
|
|
|
func xor(a, b ID) ID {
|
|
return ID(u.XOR(a, b))
|
|
}
|
|
|
|
func commonPrefixLen(a, b ID) int {
|
|
return ks.ZeroPrefixLen(u.XOR(a, b))
|
|
}
|
|
|
|
// ConvertPeerID creates a DHT ID by hashing a Peer ID (Multihash)
|
|
func ConvertPeerID(id peer.ID) ID {
|
|
hash := sha256.Sum256([]byte(id))
|
|
return hash[:]
|
|
}
|
|
|
|
// ConvertKey creates a DHT ID by hashing a local key (String)
|
|
func ConvertKey(id u.Key) ID {
|
|
hash := sha256.Sum256([]byte(id))
|
|
return hash[:]
|
|
}
|
|
|
|
// Closer returns true if a is closer to key than b is
|
|
func Closer(a, b peer.ID, key u.Key) bool {
|
|
aid := ConvertPeerID(a)
|
|
bid := ConvertPeerID(b)
|
|
tgt := ConvertKey(key)
|
|
adist := xor(aid, tgt)
|
|
bdist := xor(bid, tgt)
|
|
|
|
return adist.less(bdist)
|
|
}
|