122 lines
2.3 KiB
Go
122 lines
2.3 KiB
Go
package p2p
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
tec "github.com/jbenet/go-temp-err-catcher"
|
|
"github.com/libp2p/go-libp2p-net"
|
|
"github.com/libp2p/go-libp2p-peer"
|
|
"github.com/libp2p/go-libp2p-protocol"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
"github.com/multiformats/go-multiaddr-net"
|
|
)
|
|
|
|
// localListener manet streams and proxies them to libp2p services
|
|
type localListener struct {
|
|
ctx context.Context
|
|
|
|
p2p *P2P
|
|
|
|
proto protocol.ID
|
|
laddr ma.Multiaddr
|
|
peer peer.ID
|
|
|
|
listener manet.Listener
|
|
}
|
|
|
|
// ForwardLocal creates new P2P stream to a remote listener
|
|
func (p2p *P2P) ForwardLocal(ctx context.Context, peer peer.ID, proto protocol.ID, bindAddr ma.Multiaddr) (Listener, error) {
|
|
listener := &localListener{
|
|
ctx: ctx,
|
|
p2p: p2p,
|
|
proto: proto,
|
|
peer: peer,
|
|
}
|
|
|
|
maListener, err := manet.Listen(bindAddr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
listener.listener = maListener
|
|
listener.laddr = maListener.Multiaddr()
|
|
|
|
if err := p2p.ListenersLocal.Register(listener); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
go listener.acceptConns()
|
|
|
|
return listener, nil
|
|
}
|
|
|
|
func (l *localListener) dial(ctx context.Context) (net.Stream, error) {
|
|
cctx, cancel := context.WithTimeout(ctx, time.Second*30) //TODO: configurable?
|
|
defer cancel()
|
|
|
|
return l.p2p.peerHost.NewStream(cctx, l.peer, l.proto)
|
|
}
|
|
|
|
func (l *localListener) acceptConns() {
|
|
for {
|
|
local, err := l.listener.Accept()
|
|
if err != nil {
|
|
if tec.ErrIsTemporary(err) {
|
|
continue
|
|
}
|
|
return
|
|
}
|
|
|
|
go l.setupStream(local)
|
|
}
|
|
}
|
|
|
|
func (l *localListener) setupStream(local manet.Conn) {
|
|
remote, err := l.dial(l.ctx)
|
|
if err != nil {
|
|
local.Close()
|
|
log.Warningf("failed to dial to remote %s/%s", l.peer.Pretty(), l.proto)
|
|
return
|
|
}
|
|
|
|
stream := &Stream{
|
|
Protocol: l.proto,
|
|
|
|
OriginAddr: local.RemoteMultiaddr(),
|
|
TargetAddr: l.TargetAddress(),
|
|
peer: l.peer,
|
|
|
|
Local: local,
|
|
Remote: remote,
|
|
|
|
Registry: l.p2p.Streams,
|
|
}
|
|
|
|
l.p2p.Streams.Register(stream)
|
|
}
|
|
|
|
func (l *localListener) close() {
|
|
l.listener.Close()
|
|
}
|
|
|
|
func (l *localListener) Protocol() protocol.ID {
|
|
return l.proto
|
|
}
|
|
|
|
func (l *localListener) ListenAddress() ma.Multiaddr {
|
|
return l.laddr
|
|
}
|
|
|
|
func (l *localListener) TargetAddress() ma.Multiaddr {
|
|
addr, err := ma.NewMultiaddr(maPrefix + l.peer.Pretty())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return addr
|
|
}
|
|
|
|
func (l *localListener) key() string {
|
|
return l.ListenAddress().String()
|
|
}
|