2018-09-10 08:39:57 -04:00
|
|
|
package internet
|
|
|
|
|
2022-01-16 17:44:09 -05:00
|
|
|
import (
|
2022-10-07 21:03:51 -04:00
|
|
|
"net"
|
2022-01-16 17:44:09 -05:00
|
|
|
"syscall"
|
2022-04-06 12:06:08 -04:00
|
|
|
|
|
|
|
"golang.org/x/sys/windows"
|
2022-01-16 17:44:09 -05:00
|
|
|
)
|
2018-09-10 08:39:57 -04:00
|
|
|
|
|
|
|
const (
|
2022-10-07 21:03:51 -04:00
|
|
|
TCP_FASTOPEN = 15 // nolint: revive,stylecheck
|
|
|
|
IP_UNICAST_IF = 31 // nolint: revive,stylecheck
|
|
|
|
IPV6_UNICAST_IF = 31 // nolint: revive,stylecheck
|
2018-09-10 08:39:57 -04:00
|
|
|
)
|
|
|
|
|
2018-09-11 04:03:26 -04:00
|
|
|
func setTFO(fd syscall.Handle, settings SocketConfig_TCPFastOpenState) error {
|
|
|
|
switch settings {
|
|
|
|
case SocketConfig_Enable:
|
|
|
|
if err := syscall.SetsockoptInt(fd, syscall.IPPROTO_TCP, TCP_FASTOPEN, 1); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
case SocketConfig_Disable:
|
|
|
|
if err := syscall.SetsockoptInt(fd, syscall.IPPROTO_TCP, TCP_FASTOPEN, 0); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2018-09-10 08:39:57 -04:00
|
|
|
func applyOutboundSocketOptions(network string, address string, fd uintptr, config *SocketConfig) error {
|
2018-09-11 04:03:26 -04:00
|
|
|
if isTCPSocket(network) {
|
2018-09-11 04:18:26 -04:00
|
|
|
if err := setTFO(syscall.Handle(fd), config.Tfo); err != nil {
|
2018-09-11 04:03:26 -04:00
|
|
|
return err
|
2018-09-10 08:39:57 -04:00
|
|
|
}
|
2021-12-23 06:02:47 -05:00
|
|
|
if config.TcpKeepAliveIdle > 0 {
|
2021-11-13 17:13:37 -05:00
|
|
|
if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1); err != nil {
|
|
|
|
return newError("failed to set SO_KEEPALIVE", err)
|
|
|
|
}
|
|
|
|
}
|
2018-09-10 08:39:57 -04:00
|
|
|
}
|
|
|
|
|
2022-10-07 21:03:51 -04:00
|
|
|
if config.BindToDevice != "" {
|
|
|
|
iface, err := net.InterfaceByName(config.BindToDevice)
|
|
|
|
if err != nil {
|
|
|
|
return newError("failed to get interface ", config.BindToDevice).Base(err)
|
|
|
|
}
|
|
|
|
if err := windows.SetsockoptInt(windows.Handle(fd), windows.IPPROTO_IP, IP_UNICAST_IF, iface.Index); err != nil {
|
|
|
|
return newError("failed to set IP_UNICAST_IF", err)
|
|
|
|
}
|
|
|
|
if err := windows.SetsockoptInt(windows.Handle(fd), windows.IPPROTO_IPV6, IPV6_UNICAST_IF, iface.Index); err != nil {
|
|
|
|
return newError("failed to set IPV6_UNICAST_IF", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-01-16 18:02:24 -05:00
|
|
|
if config.TxBufSize != 0 {
|
|
|
|
if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_SNDBUF, int(config.TxBufSize)); err != nil {
|
2022-01-16 17:44:09 -05:00
|
|
|
return newError("failed to set SO_SNDBUF").Base(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-01-16 18:02:24 -05:00
|
|
|
if config.RxBufSize != 0 {
|
|
|
|
if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_RCVBUF, int(config.TxBufSize)); err != nil {
|
2022-01-16 17:44:09 -05:00
|
|
|
return newError("failed to set SO_RCVBUF").Base(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-09-10 08:39:57 -04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func applyInboundSocketOptions(network string, fd uintptr, config *SocketConfig) error {
|
2018-09-11 04:03:26 -04:00
|
|
|
if isTCPSocket(network) {
|
2018-09-11 04:18:26 -04:00
|
|
|
if err := setTFO(syscall.Handle(fd), config.Tfo); err != nil {
|
2018-09-11 04:03:26 -04:00
|
|
|
return err
|
2018-09-10 08:39:57 -04:00
|
|
|
}
|
2021-12-23 06:02:47 -05:00
|
|
|
if config.TcpKeepAliveIdle > 0 {
|
2021-11-13 17:13:37 -05:00
|
|
|
if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1); err != nil {
|
|
|
|
return newError("failed to set SO_KEEPALIVE", err)
|
|
|
|
}
|
|
|
|
}
|
2018-09-10 08:39:57 -04:00
|
|
|
}
|
|
|
|
|
2022-01-16 18:02:24 -05:00
|
|
|
if config.TxBufSize != 0 {
|
|
|
|
if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_SNDBUF, int(config.TxBufSize)); err != nil {
|
2022-01-16 17:44:09 -05:00
|
|
|
return newError("failed to set SO_SNDBUF").Base(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-01-16 18:02:24 -05:00
|
|
|
if config.RxBufSize != 0 {
|
|
|
|
if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_RCVBUF, int(config.TxBufSize)); err != nil {
|
2022-01-16 17:44:09 -05:00
|
|
|
return newError("failed to set SO_RCVBUF").Base(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-09-10 08:39:57 -04:00
|
|
|
return nil
|
|
|
|
}
|
2018-09-17 09:12:58 -04:00
|
|
|
|
2018-11-21 08:54:40 -05:00
|
|
|
func bindAddr(fd uintptr, ip []byte, port uint32) error {
|
2018-09-17 09:12:58 -04:00
|
|
|
return nil
|
|
|
|
}
|
2020-07-30 13:20:12 -04:00
|
|
|
|
|
|
|
func setReuseAddr(fd uintptr) error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func setReusePort(fd uintptr) error {
|
|
|
|
return nil
|
|
|
|
}
|