status-go/vendor/github.com/anacrolix/torrent/dht.go

63 lines
1.6 KiB
Go
Raw Normal View History

2022-03-10 09:44:48 +00:00
package torrent
import (
"io"
"net"
"github.com/anacrolix/dht/v2"
"github.com/anacrolix/dht/v2/krpc"
peer_store "github.com/anacrolix/dht/v2/peer-store"
)
// DHT server interface for use by a Torrent or Client. It's reasonable for this to make assumptions
// for torrent-use that might not be the default behaviour for the DHT server.
type DhtServer interface {
Stats() interface{}
ID() [20]byte
Addr() net.Addr
AddNode(ni krpc.NodeInfo) error
// This is called asynchronously when receiving PORT messages.
Ping(addr *net.UDPAddr)
Announce(hash [20]byte, port int, impliedPort bool) (DhtAnnounce, error)
WriteStatus(io.Writer)
}
// Optional interface for DhtServer's that can expose their peer store (if any).
type PeerStorer interface {
PeerStore() peer_store.Interface
}
type DhtAnnounce interface {
Close()
Peers() <-chan dht.PeersValues
}
type AnacrolixDhtServerWrapper struct {
*dht.Server
}
func (me AnacrolixDhtServerWrapper) Stats() interface{} {
return me.Server.Stats()
}
type anacrolixDhtAnnounceWrapper struct {
*dht.Announce
}
func (me anacrolixDhtAnnounceWrapper) Peers() <-chan dht.PeersValues {
return me.Announce.Peers
}
func (me AnacrolixDhtServerWrapper) Announce(hash [20]byte, port int, impliedPort bool) (DhtAnnounce, error) {
ann, err := me.Server.Announce(hash, port, impliedPort)
return anacrolixDhtAnnounceWrapper{ann}, err
}
func (me AnacrolixDhtServerWrapper) Ping(addr *net.UDPAddr) {
me.Server.PingQueryInput(addr, dht.QueryInput{
RateLimiting: dht.QueryRateLimiting{NoWaitFirst: true},
})
}
var _ DhtServer = AnacrolixDhtServerWrapper{}