ceremonyclient/go-libp2p/p2p/muxer/yamux/transport.go
2023-08-20 23:07:43 -05:00

64 lines
1.6 KiB
Go

package yamux
import (
"io"
"math"
"net"
"github.com/libp2p/go-libp2p/core/network"
"github.com/libp2p/go-yamux/v4"
)
var DefaultTransport *Transport
const ID = "/yamux/1.0.0"
func init() {
config := yamux.DefaultConfig()
// We've bumped this to 16MiB as this critically limits throughput.
//
// 1MiB means a best case of 10MiB/s (83.89Mbps) on a connection with
// 100ms latency. The default gave us 2.4MiB *best case* which was
// totally unacceptable.
config.MaxStreamWindowSize = uint32(16 * 1024 * 1024)
// don't spam
config.LogOutput = io.Discard
// We always run over a security transport that buffers internally
// (i.e., uses a block cipher).
config.ReadBufSize = 0
// Effectively disable the incoming streams limit.
// This is now dynamically limited by the resource manager.
config.MaxIncomingStreams = math.MaxUint32
DefaultTransport = (*Transport)(config)
}
// Transport implements mux.Multiplexer that constructs
// yamux-backed muxed connections.
type Transport yamux.Config
var _ network.Multiplexer = &Transport{}
func (t *Transport) NewConn(nc net.Conn, isServer bool, scope network.PeerScope) (network.MuxedConn, error) {
var newSpan func() (yamux.MemoryManager, error)
if scope != nil {
newSpan = func() (yamux.MemoryManager, error) { return scope.BeginSpan() }
}
var s *yamux.Session
var err error
if isServer {
s, err = yamux.Server(nc, t.Config(), newSpan)
} else {
s, err = yamux.Client(nc, t.Config(), newSpan)
}
if err != nil {
return nil, err
}
return NewMuxedConn(s), nil
}
func (t *Transport) Config() *yamux.Config {
return (*yamux.Config)(t)
}