lzc256
/
tailscale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add a knob for maxInFlightConnectionAttempts on TCP connectons 

Signed-off-by: Roy Williams <roy@anthropic.com>
pull/8210/head
Roy Williams 2023-05-25 11:48:32 -04:00
parent 66f97f4bea
commit 35ac4cfbfc
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
wgengine/netstack/netstack.go
View File

@ -246,6 +246,15 @@ func (ns *Impl) wrapProtoHandler(h func(stack.TransportEndpointID, stack.PacketB
} }
} }
var tcpMaxInFlightConnectionAttemptsVar = envknob.RegisterInt("TS_TCP_MAX_IN_FLIGHT_CONNETION_ATTEMPTS")
func () tcpMaxInFlightConnectionAttempts int {
if x := tcpMaxInFlightConnectionAttemptsVar(); x != 0 {
return x
}
return 16
}
// Start sets up all the handlers so netstack can start working. Implements // Start sets up all the handlers so netstack can start working. Implements
// wgengine.FakeImpl. // wgengine.FakeImpl.
func (ns *Impl) Start(lb *ipnlocal.LocalBackend) error { func (ns *Impl) Start(lb *ipnlocal.LocalBackend) error {
@ -256,7 +265,8 @@ func (ns *Impl) Start(lb *ipnlocal.LocalBackend) error {
ns.e.AddNetworkMapCallback(ns.updateIPs) ns.e.AddNetworkMapCallback(ns.updateIPs)
// size = 0 means use default buffer size // size = 0 means use default buffer size
const tcpReceiveBufferSize = 0 const tcpReceiveBufferSize = 0
const maxInFlightConnectionAttempts = 16
const maxInFlightConnectionAttempts = tcpMaxInFlightConnectionAttempts()
tcpFwd := tcp.NewForwarder(ns.ipstack, tcpReceiveBufferSize, maxInFlightConnectionAttempts, ns.acceptTCP) tcpFwd := tcp.NewForwarder(ns.ipstack, tcpReceiveBufferSize, maxInFlightConnectionAttempts, ns.acceptTCP)
udpFwd := udp.NewForwarder(ns.ipstack, ns.acceptUDP) udpFwd := udp.NewForwarder(ns.ipstack, ns.acceptUDP)
ns.ipstack.SetTransportProtocolHandler(tcp.ProtocolNumber, ns.wrapProtoHandler(tcpFwd.HandlePacket)) ns.ipstack.SetTransportProtocolHandler(tcp.ProtocolNumber, ns.wrapProtoHandler(tcpFwd.HandlePacket))