mirror of
https://github.com/netbirdio/gvisor.git
synced 2026-05-22 17:12:49 -07:00
Bhasker Hariharan
b070e218c6
Linux controls socket send/receive buffers using a few sysctl variables - net.core.rmem_default - net.core.rmem_max - net.core.wmem_max - net.core.wmem_default - net.ipv4.tcp_rmem - net.ipv4.tcp_wmem The first 4 control the default socket buffer sizes for all sockets raw/packet/tcp/udp and also the maximum permitted socket buffer that can be specified in setsockopt(SOL_SOCKET, SO_(RCV|SND)BUF,...). The last two control the TCP auto-tuning limits and override the default specified in rmem_default/wmem_default as well as the max limits. Netstack today only implements tcp_rmem/tcp_wmem and incorrectly uses it to limit the maximum size in setsockopt() as well as uses it for raw/udp sockets. This changelist introduces the other 4 and updates the udp/raw sockets to use the newly introduced variables. The values for min/max match the current tcp_rmem/wmem values and the default value buffers for UDP/RAW sockets is updated to match the linux value of 212KiB up from the really low current value of 32 KiB. Updates #3043 Fixes #3043 PiperOrigin-RevId: 318089805
107 lines
2.7 KiB
Go
107 lines
2.7 KiB
Go
// Copyright 2020 The gVisor Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package stack
|
|
|
|
import "gvisor.dev/gvisor/pkg/tcpip"
|
|
|
|
const (
|
|
// MinBufferSize is the smallest size of a receive or send buffer.
|
|
MinBufferSize = 4 << 10 // 4 KiB
|
|
|
|
// DefaultBufferSize is the default size of the send/recv buffer for a
|
|
// transport endpoint.
|
|
DefaultBufferSize = 212 << 10 // 212 KiB
|
|
|
|
// DefaultMaxBufferSize is the default maximum permitted size of a
|
|
// send/receive buffer.
|
|
DefaultMaxBufferSize = 4 << 20 // 4 MiB
|
|
)
|
|
|
|
// SendBufferSizeOption is used by stack.(Stack*).Option/SetOption to
|
|
// get/set the default, min and max send buffer sizes.
|
|
type SendBufferSizeOption struct {
|
|
Min int
|
|
Default int
|
|
Max int
|
|
}
|
|
|
|
// ReceiveBufferSizeOption is used by stack.(Stack*).Option/SetOption to
|
|
// get/set the default, min and max receive buffer sizes.
|
|
type ReceiveBufferSizeOption struct {
|
|
Min int
|
|
Default int
|
|
Max int
|
|
}
|
|
|
|
// SetOption allows setting stack wide options.
|
|
func (s *Stack) SetOption(option interface{}) *tcpip.Error {
|
|
switch v := option.(type) {
|
|
case SendBufferSizeOption:
|
|
// Make sure we don't allow lowering the buffer below minimum
|
|
// required for stack to work.
|
|
if v.Min < MinBufferSize {
|
|
return tcpip.ErrInvalidOptionValue
|
|
}
|
|
|
|
if v.Default < v.Min || v.Default > v.Max {
|
|
return tcpip.ErrInvalidOptionValue
|
|
}
|
|
|
|
s.mu.Lock()
|
|
s.sendBufferSize = v
|
|
s.mu.Unlock()
|
|
return nil
|
|
|
|
case ReceiveBufferSizeOption:
|
|
// Make sure we don't allow lowering the buffer below minimum
|
|
// required for stack to work.
|
|
if v.Min < MinBufferSize {
|
|
return tcpip.ErrInvalidOptionValue
|
|
}
|
|
|
|
if v.Default < v.Min || v.Default > v.Max {
|
|
return tcpip.ErrInvalidOptionValue
|
|
}
|
|
|
|
s.mu.Lock()
|
|
s.receiveBufferSize = v
|
|
s.mu.Unlock()
|
|
return nil
|
|
|
|
default:
|
|
return tcpip.ErrUnknownProtocolOption
|
|
}
|
|
}
|
|
|
|
// Option allows retrieving stack wide options.
|
|
func (s *Stack) Option(option interface{}) *tcpip.Error {
|
|
switch v := option.(type) {
|
|
case *SendBufferSizeOption:
|
|
s.mu.RLock()
|
|
*v = s.sendBufferSize
|
|
s.mu.RUnlock()
|
|
return nil
|
|
|
|
case *ReceiveBufferSizeOption:
|
|
s.mu.RLock()
|
|
*v = s.receiveBufferSize
|
|
s.mu.RUnlock()
|
|
return nil
|
|
|
|
default:
|
|
return tcpip.ErrUnknownProtocolOption
|
|
}
|
|
}
|