mirror of
https://github.com/libp2p/go-libp2p.git
synced 2025-09-26 20:21:26 +08:00
6249e685e9
This introduces a new GatedMaListener type which gates conns accepted from a manet.Listener with a gater and creates the rcmgr scope for it. Explicitly passing the scope allows for many guardrails that the previous interface assertion didn't. This breaks the previous responsibility of the upgradeListener method into two, one gating the connection initially, and the other upgrading the connection with a security and muxer selection. This split makes it easy to gate the connection with the resource manager as early as possible. This is especially true for websocket because we want to gate the connection just after the TCP connection is established, and not after the tls handshake + websocket upgrade is completed.
114 lines
3.0 KiB
Go
114 lines
3.0 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/libp2p/go-libp2p/core/host"
|
|
"github.com/libp2p/go-libp2p/core/network"
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
|
"github.com/libp2p/go-libp2p/core/transport"
|
|
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
var circuitProtocol = ma.ProtocolWithCode(ma.P_CIRCUIT)
|
|
var circuitAddr = ma.Cast(circuitProtocol.VCode)
|
|
|
|
// AddTransport constructs a new p2p-circuit/v2 client and adds it as a transport to the
|
|
// host network
|
|
func AddTransport(h host.Host, upgrader transport.Upgrader) error {
|
|
n, ok := h.Network().(transport.TransportNetwork)
|
|
if !ok {
|
|
return fmt.Errorf("%v is not a transport network", h.Network())
|
|
}
|
|
|
|
c, err := New(h, upgrader)
|
|
if err != nil {
|
|
return fmt.Errorf("error constructing circuit client: %w", err)
|
|
}
|
|
|
|
err = n.AddTransport(c)
|
|
if err != nil {
|
|
return fmt.Errorf("error adding circuit transport: %w", err)
|
|
}
|
|
|
|
err = n.Listen(circuitAddr)
|
|
if err != nil {
|
|
return fmt.Errorf("error listening to circuit addr: %w", err)
|
|
}
|
|
|
|
c.Start()
|
|
|
|
return nil
|
|
}
|
|
|
|
// Transport interface
|
|
var _ transport.Transport = (*Client)(nil)
|
|
|
|
// p2p-circuit implements the SkipResolver interface so that the underlying
|
|
// transport can do the address resolution later. If you wrap this transport,
|
|
// make sure you also implement SkipResolver as well.
|
|
var _ transport.SkipResolver = (*Client)(nil)
|
|
var _ io.Closer = (*Client)(nil)
|
|
|
|
// SkipResolve returns true since we always defer to the inner transport for
|
|
// the actual connection. By skipping resolution here, we let the inner
|
|
// transport decide how to resolve the multiaddr
|
|
func (c *Client) SkipResolve(ctx context.Context, maddr ma.Multiaddr) bool {
|
|
return true
|
|
}
|
|
|
|
func (c *Client) Dial(ctx context.Context, a ma.Multiaddr, p peer.ID) (transport.CapableConn, error) {
|
|
connScope, err := c.host.Network().ResourceManager().OpenConnection(network.DirOutbound, false, a)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
conn, err := c.dialAndUpgrade(ctx, a, p, connScope)
|
|
if err != nil {
|
|
connScope.Done()
|
|
return nil, err
|
|
}
|
|
return conn, nil
|
|
}
|
|
|
|
func (c *Client) dialAndUpgrade(ctx context.Context, a ma.Multiaddr, p peer.ID, connScope network.ConnManagementScope) (transport.CapableConn, error) {
|
|
if err := connScope.SetPeer(p); err != nil {
|
|
return nil, err
|
|
}
|
|
conn, err := c.dial(ctx, a, p)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
conn.tagHop()
|
|
cc, err := c.upgrader.Upgrade(ctx, c, conn, network.DirOutbound, p, connScope)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return capableConn{cc.(capableConnWithStat)}, nil
|
|
}
|
|
|
|
func (c *Client) CanDial(addr ma.Multiaddr) bool {
|
|
_, err := addr.ValueForProtocol(ma.P_CIRCUIT)
|
|
return err == nil
|
|
}
|
|
|
|
func (c *Client) Listen(addr ma.Multiaddr) (transport.Listener, error) {
|
|
// TODO connect to the relay and reserve slot if specified
|
|
if _, err := addr.ValueForProtocol(ma.P_CIRCUIT); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return c.upgrader.UpgradeGatedMaListener(c, c.upgrader.GateMaListener(c.Listener())), nil
|
|
}
|
|
|
|
func (c *Client) Protocols() []int {
|
|
return []int{ma.P_CIRCUIT}
|
|
}
|
|
|
|
func (c *Client) Proxy() bool {
|
|
return true
|
|
}
|