status-go/vendor/github.com/libp2p/go-libp2p-circuit/notify.go

55 lines
1.3 KiB
Go
Raw Normal View History

package relay
import (
"context"
"time"
2019-06-09 07:24:20 +00:00
inet "github.com/libp2p/go-libp2p-core/network"
peer "github.com/libp2p/go-libp2p-core/peer"
ma "github.com/multiformats/go-multiaddr"
)
var _ inet.Notifiee = (*RelayNotifiee)(nil)
type RelayNotifiee Relay
2019-06-09 07:24:20 +00:00
func (r *Relay) notifiee() inet.Notifiee {
return (*RelayNotifiee)(r)
}
func (n *RelayNotifiee) Relay() *Relay {
return (*Relay)(n)
}
func (n *RelayNotifiee) Listen(net inet.Network, a ma.Multiaddr) {}
func (n *RelayNotifiee) ListenClose(net inet.Network, a ma.Multiaddr) {}
func (n *RelayNotifiee) OpenedStream(net inet.Network, s inet.Stream) {}
func (n *RelayNotifiee) ClosedStream(net inet.Network, s inet.Stream) {}
func (n *RelayNotifiee) Connected(s inet.Network, c inet.Conn) {
if n.Relay().Matches(c.RemoteMultiaddr()) {
return
}
go func(id peer.ID) {
ctx, cancel := context.WithTimeout(n.ctx, time.Second)
defer cancel()
canhop, err := n.Relay().CanHop(ctx, id)
if err != nil {
log.Debugf("Error testing relay hop: %s", err.Error())
return
}
if canhop {
log.Debugf("Discovered hop relay %s", id.Pretty())
n.mx.Lock()
n.relays[id] = struct{}{}
n.mx.Unlock()
n.host.ConnManager().TagPeer(id, "relay-hop", 2)
}
}(c.RemotePeer())
}
func (n *RelayNotifiee) Disconnected(s inet.Network, c inet.Conn) {}