45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
|
package torrent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/anacrolix/dht/v2/krpc"
|
||
|
|
|
||
|
|
"github.com/anacrolix/torrent/peer_protocol"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Peer connection info, handed about publicly.
|
||
|
|
type PeerInfo struct {
|
||
|
|
Id [20]byte
|
||
|
|
Addr PeerRemoteAddr
|
||
|
|
Source PeerSource
|
||
|
|
// Peer is known to support encryption.
|
||
|
|
SupportsEncryption bool
|
||
|
|
peer_protocol.PexPeerFlags
|
||
|
|
// Whether we can ignore poor or bad behaviour from the peer.
|
||
|
|
Trusted bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me PeerInfo) equal(other PeerInfo) bool {
|
||
|
|
return me.Id == other.Id &&
|
||
|
|
me.Addr.String() == other.Addr.String() &&
|
||
|
|
me.Source == other.Source &&
|
||
|
|
me.SupportsEncryption == other.SupportsEncryption &&
|
||
|
|
me.PexPeerFlags == other.PexPeerFlags &&
|
||
|
|
me.Trusted == other.Trusted
|
||
|
|
}
|
||
|
|
|
||
|
|
// Generate PeerInfo from peer exchange
|
||
|
|
func (me *PeerInfo) FromPex(na krpc.NodeAddr, fs peer_protocol.PexPeerFlags) {
|
||
|
|
me.Addr = ipPortAddr{append([]byte(nil), na.IP...), na.Port}
|
||
|
|
me.Source = PeerSourcePex
|
||
|
|
// If they prefer encryption, they must support it.
|
||
|
|
if fs.Get(peer_protocol.PexPrefersEncryption) {
|
||
|
|
me.SupportsEncryption = true
|
||
|
|
}
|
||
|
|
me.PexPeerFlags = fs
|
||
|
|
}
|
||
|
|
|
||
|
|
func (me PeerInfo) addr() IpPort {
|
||
|
|
ipPort, _ := tryIpPortFromNetAddr(me.Addr)
|
||
|
|
return IpPort{ipPort.IP, uint16(ipPort.Port)}
|
||
|
|
}
|