| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107 |
- package internet
- import (
- "net"
- "syscall"
- "golang.org/x/sys/windows"
- )
- const (
- TCP_FASTOPEN = 15 // nolint: revive,stylecheck
- IP_UNICAST_IF = 31 // nolint: revive,stylecheck
- IPV6_UNICAST_IF = 31 // nolint: revive,stylecheck
- )
- 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
- }
- func applyOutboundSocketOptions(network string, address string, fd uintptr, config *SocketConfig) error {
- if isTCPSocket(network) {
- if err := setTFO(syscall.Handle(fd), config.Tfo); err != nil {
- return err
- }
- if config.TcpKeepAliveIdle > 0 {
- if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1); err != nil {
- return newError("failed to set SO_KEEPALIVE", err)
- }
- }
- }
- 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)
- }
- }
- if config.TxBufSize != 0 {
- if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_SNDBUF, int(config.TxBufSize)); err != nil {
- return newError("failed to set SO_SNDBUF").Base(err)
- }
- }
- if config.RxBufSize != 0 {
- if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_RCVBUF, int(config.TxBufSize)); err != nil {
- return newError("failed to set SO_RCVBUF").Base(err)
- }
- }
- return nil
- }
- func applyInboundSocketOptions(network string, fd uintptr, config *SocketConfig) error {
- if isTCPSocket(network) {
- if err := setTFO(syscall.Handle(fd), config.Tfo); err != nil {
- return err
- }
- if config.TcpKeepAliveIdle > 0 {
- if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1); err != nil {
- return newError("failed to set SO_KEEPALIVE", err)
- }
- }
- }
- if config.TxBufSize != 0 {
- if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_SNDBUF, int(config.TxBufSize)); err != nil {
- return newError("failed to set SO_SNDBUF").Base(err)
- }
- }
- if config.RxBufSize != 0 {
- if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_RCVBUF, int(config.TxBufSize)); err != nil {
- return newError("failed to set SO_RCVBUF").Base(err)
- }
- }
- return nil
- }
- func bindAddr(fd uintptr, ip []byte, port uint32) error {
- return nil
- }
- func setReuseAddr(fd uintptr) error {
- return nil
- }
- func setReusePort(fd uintptr) error {
- return nil
- }
|