63 lines
1.6 KiB
Go
63 lines
1.6 KiB
Go
|
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{}
|