mirror of https://github.com/tailscale/tailscale/
net/sockopts,wgengine/magicsock: export socket buffer sizing logic (#16909)
For eventual use by net/udprelay.Server Updates tailscale/corp#31164 Signed-off-by: Jordan Whited <jordan@tailscale.com>dependabot/go_modules/github.com/go-viper/mapstructure/v2-2.4.0
Jordan Whited
4 months ago
committed by
GitHub
parent
b48d2de6ab
commit
641a90ea33
@ -0,0 +1,37 @@
|
||||
// Copyright (c) Tailscale Inc & AUTHORS
|
||||
// SPDX-License-Identifier: BSD-3-Clause
|
||||
|
||||
// Package sockopts contains logic for applying socket options.
|
||||
package sockopts
|
||||
|
||||
import (
|
||||
"net"
|
||||
"runtime"
|
||||
|
||||
"tailscale.com/types/nettype"
|
||||
)
|
||||
|
||||
// BufferDirection represents either the read/receive or write/send direction
|
||||
// of a socket buffer.
|
||||
type BufferDirection string
|
||||
|
||||
const (
|
||||
ReadDirection BufferDirection = "read"
|
||||
WriteDirection BufferDirection = "write"
|
||||
)
|
||||
|
||||
func portableSetBufferSize(pconn nettype.PacketConn, direction BufferDirection, size int) error {
|
||||
if runtime.GOOS == "plan9" {
|
||||
// Not supported. Don't try. Avoid logspam.
|
||||
return nil
|
||||
}
|
||||
var err error
|
||||
if c, ok := pconn.(*net.UDPConn); ok {
|
||||
if direction == WriteDirection {
|
||||
err = c.SetWriteBuffer(size)
|
||||
} else {
|
||||
err = c.SetReadBuffer(size)
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
||||
@ -0,0 +1,21 @@
|
||||
// Copyright (c) Tailscale Inc & AUTHORS
|
||||
// SPDX-License-Identifier: BSD-3-Clause
|
||||
|
||||
//go:build !linux
|
||||
|
||||
package sockopts
|
||||
|
||||
import (
|
||||
"tailscale.com/types/nettype"
|
||||
)
|
||||
|
||||
// SetBufferSize sets pconn's buffer to size for direction. size may be silently
|
||||
// capped depending on platform.
|
||||
//
|
||||
// errForce is only relevant for Linux, and will always be nil otherwise,
|
||||
// but we maintain a consistent cross-platform API.
|
||||
//
|
||||
// If pconn is not a [*net.UDPConn], then SetBufferSize is no-op.
|
||||
func SetBufferSize(pconn nettype.PacketConn, direction BufferDirection, size int) (errForce error, errPortable error) {
|
||||
return nil, portableSetBufferSize(pconn, direction, size)
|
||||
}
|
||||
@ -0,0 +1,40 @@
|
||||
// Copyright (c) Tailscale Inc & AUTHORS
|
||||
// SPDX-License-Identifier: BSD-3-Clause
|
||||
|
||||
//go:build linux
|
||||
|
||||
package sockopts
|
||||
|
||||
import (
|
||||
"net"
|
||||
"syscall"
|
||||
|
||||
"tailscale.com/types/nettype"
|
||||
)
|
||||
|
||||
// SetBufferSize sets pconn's buffer to size for direction. It attempts
|
||||
// (errForce) to set SO_SNDBUFFORCE or SO_RECVBUFFORCE which can overcome the
|
||||
// limit of net.core.{r,w}mem_max, but require CAP_NET_ADMIN. It falls back to
|
||||
// the portable implementation (errPortable) if that fails, which may be
|
||||
// silently capped to net.core.{r,w}mem_max.
|
||||
//
|
||||
// If pconn is not a [*net.UDPConn], then SetBufferSize is no-op.
|
||||
func SetBufferSize(pconn nettype.PacketConn, direction BufferDirection, size int) (errForce error, errPortable error) {
|
||||
opt := syscall.SO_RCVBUFFORCE
|
||||
if direction == WriteDirection {
|
||||
opt = syscall.SO_SNDBUFFORCE
|
||||
}
|
||||
if c, ok := pconn.(*net.UDPConn); ok {
|
||||
var rc syscall.RawConn
|
||||
rc, errForce = c.SyscallConn()
|
||||
if errForce == nil {
|
||||
rc.Control(func(fd uintptr) {
|
||||
errForce = syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, opt, size)
|
||||
})
|
||||
}
|
||||
if errForce != nil {
|
||||
errPortable = portableSetBufferSize(pconn, direction, size)
|
||||
}
|
||||
}
|
||||
return errForce, errPortable
|
||||
}
|
||||
Loading…
Reference in New Issue