mirror of
https://github.com/tailscale/tailscale.git
synced 2024-11-26 11:35:35 +00:00
a5ae21a832
Add an enable/disable argument to setDontFragment() in preparation for dynamic enable/disable of peer path MTU discovery. Add getDontFragment() to get the status of the don't fragment bit from a socket. Updates #311 Co-authored-by: James Tucker <james@tailscale.com> Signed-off-by: Val <valerie@tailscale.com>
52 lines
973 B
Go
52 lines
973 B
Go
// Copyright (c) Tailscale Inc & AUTHORS
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
//go:build darwin && !ios
|
|
|
|
package magicsock
|
|
|
|
import (
|
|
"syscall"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
func getDontFragOpt(network string) int {
|
|
if network == "udp4" {
|
|
return unix.IP_DONTFRAG
|
|
}
|
|
return unix.IPV6_DONTFRAG
|
|
}
|
|
|
|
func (c *Conn) setDontFragment(network string, enable bool) error {
|
|
optArg := 1
|
|
if enable == false {
|
|
optArg = 0
|
|
}
|
|
var err error
|
|
rcErr := c.connControl(network, func(fd uintptr) {
|
|
err = syscall.SetsockoptInt(int(fd), getIPProto(network), getDontFragOpt(network), optArg)
|
|
})
|
|
|
|
if rcErr != nil {
|
|
return rcErr
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (c *Conn) getDontFragment(network string) (bool, error) {
|
|
var v int
|
|
var err error
|
|
rcErr := c.connControl(network, func(fd uintptr) {
|
|
v, err = syscall.GetsockoptInt(int(fd), getIPProto(network), getDontFragOpt(network))
|
|
})
|
|
|
|
if rcErr != nil {
|
|
return false, rcErr
|
|
}
|
|
if v == 1 {
|
|
return true, err
|
|
}
|
|
return false, err
|
|
}
|