github.com/xmplusdev/xmcore@v1.8.11-0.20240412132628-5518b55526af/transport/internet/sockopt_linux.go (about)

     1  package internet
     2  
     3  import (
     4  	"net"
     5  	"syscall"
     6  
     7  	"golang.org/x/sys/unix"
     8  )
     9  
    10  func bindAddr(fd uintptr, ip []byte, port uint32) error {
    11  	setReuseAddr(fd)
    12  	setReusePort(fd)
    13  
    14  	var sockaddr syscall.Sockaddr
    15  
    16  	switch len(ip) {
    17  	case net.IPv4len:
    18  		a4 := &syscall.SockaddrInet4{
    19  			Port: int(port),
    20  		}
    21  		copy(a4.Addr[:], ip)
    22  		sockaddr = a4
    23  	case net.IPv6len:
    24  		a6 := &syscall.SockaddrInet6{
    25  			Port: int(port),
    26  		}
    27  		copy(a6.Addr[:], ip)
    28  		sockaddr = a6
    29  	default:
    30  		return newError("unexpected length of ip")
    31  	}
    32  
    33  	return syscall.Bind(int(fd), sockaddr)
    34  }
    35  
    36  func applyOutboundSocketOptions(network string, address string, fd uintptr, config *SocketConfig) error {
    37  	if config.Mark != 0 {
    38  		if err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_MARK, int(config.Mark)); err != nil {
    39  			return newError("failed to set SO_MARK").Base(err)
    40  		}
    41  	}
    42  
    43  	if config.Interface != "" {
    44  		if err := syscall.BindToDevice(int(fd), config.Interface); err != nil {
    45  			return newError("failed to set Interface").Base(err)
    46  		}
    47  	}
    48  
    49  	if isTCPSocket(network) {
    50  		tfo := config.ParseTFOValue()
    51  		if tfo > 0 {
    52  			tfo = 1
    53  		}
    54  		if tfo >= 0 {
    55  			if err := syscall.SetsockoptInt(int(fd), syscall.SOL_TCP, unix.TCP_FASTOPEN_CONNECT, tfo); err != nil {
    56  				return newError("failed to set TCP_FASTOPEN_CONNECT", tfo).Base(err)
    57  			}
    58  		}
    59  
    60  		if config.TcpKeepAliveInterval > 0 || config.TcpKeepAliveIdle > 0 {
    61  			if config.TcpKeepAliveInterval > 0 {
    62  				if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, syscall.TCP_KEEPINTVL, int(config.TcpKeepAliveInterval)); err != nil {
    63  					return newError("failed to set TCP_KEEPINTVL", err)
    64  				}
    65  			}
    66  			if config.TcpKeepAliveIdle > 0 {
    67  				if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, syscall.TCP_KEEPIDLE, int(config.TcpKeepAliveIdle)); err != nil {
    68  					return newError("failed to set TCP_KEEPIDLE", err)
    69  				}
    70  			}
    71  			if err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1); err != nil {
    72  				return newError("failed to set SO_KEEPALIVE", err)
    73  			}
    74  		} else if config.TcpKeepAliveInterval < 0 || config.TcpKeepAliveIdle < 0 {
    75  			if err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 0); err != nil {
    76  				return newError("failed to unset SO_KEEPALIVE", err)
    77  			}
    78  		}
    79  
    80  		if config.TcpCongestion != "" {
    81  			if err := syscall.SetsockoptString(int(fd), syscall.SOL_TCP, syscall.TCP_CONGESTION, config.TcpCongestion); err != nil {
    82  				return newError("failed to set TCP_CONGESTION", err)
    83  			}
    84  		}
    85  
    86  		if config.TcpWindowClamp > 0 {
    87  			if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, syscall.TCP_WINDOW_CLAMP, int(config.TcpWindowClamp)); err != nil {
    88  				return newError("failed to set TCP_WINDOW_CLAMP", err)
    89  			}
    90  		}
    91  
    92  		if config.TcpUserTimeout > 0 {
    93  			if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, unix.TCP_USER_TIMEOUT, int(config.TcpUserTimeout)); err != nil {
    94  				return newError("failed to set TCP_USER_TIMEOUT", err)
    95  			}
    96  		}
    97  
    98  		if config.TcpMaxSeg > 0 {
    99  			if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, unix.TCP_MAXSEG, int(config.TcpMaxSeg)); err != nil {
   100  				return newError("failed to set TCP_MAXSEG", err)
   101  			}
   102  		}
   103  
   104  		if config.TcpNoDelay {
   105  			if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, unix.TCP_NODELAY, 1); err != nil {
   106  				return newError("failed to set TCP_NODELAY", err)
   107  			}
   108  		}
   109  
   110  	}
   111  
   112  	if config.Tproxy.IsEnabled() {
   113  		if err := syscall.SetsockoptInt(int(fd), syscall.SOL_IP, syscall.IP_TRANSPARENT, 1); err != nil {
   114  			return newError("failed to set IP_TRANSPARENT").Base(err)
   115  		}
   116  	}
   117  
   118  	return nil
   119  }
   120  
   121  func applyInboundSocketOptions(network string, fd uintptr, config *SocketConfig) error {
   122  	if config.Mark != 0 {
   123  		if err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_MARK, int(config.Mark)); err != nil {
   124  			return newError("failed to set SO_MARK").Base(err)
   125  		}
   126  	}
   127  	if isTCPSocket(network) {
   128  		tfo := config.ParseTFOValue()
   129  		if tfo >= 0 {
   130  			if err := syscall.SetsockoptInt(int(fd), syscall.SOL_TCP, unix.TCP_FASTOPEN, tfo); err != nil {
   131  				return newError("failed to set TCP_FASTOPEN", tfo).Base(err)
   132  			}
   133  		}
   134  
   135  		if config.TcpKeepAliveInterval > 0 || config.TcpKeepAliveIdle > 0 {
   136  			if config.TcpKeepAliveInterval > 0 {
   137  				if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, syscall.TCP_KEEPINTVL, int(config.TcpKeepAliveInterval)); err != nil {
   138  					return newError("failed to set TCP_KEEPINTVL", err)
   139  				}
   140  			}
   141  			if config.TcpKeepAliveIdle > 0 {
   142  				if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, syscall.TCP_KEEPIDLE, int(config.TcpKeepAliveIdle)); err != nil {
   143  					return newError("failed to set TCP_KEEPIDLE", err)
   144  				}
   145  			}
   146  			if err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1); err != nil {
   147  				return newError("failed to set SO_KEEPALIVE", err)
   148  			}
   149  		} else if config.TcpKeepAliveInterval < 0 || config.TcpKeepAliveIdle < 0 {
   150  			if err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 0); err != nil {
   151  				return newError("failed to unset SO_KEEPALIVE", err)
   152  			}
   153  		}
   154  
   155  		if config.TcpCongestion != "" {
   156  			if err := syscall.SetsockoptString(int(fd), syscall.SOL_TCP, syscall.TCP_CONGESTION, config.TcpCongestion); err != nil {
   157  				return newError("failed to set TCP_CONGESTION", err)
   158  			}
   159  		}
   160  
   161  		if config.TcpWindowClamp > 0 {
   162  			if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, syscall.TCP_WINDOW_CLAMP, int(config.TcpWindowClamp)); err != nil {
   163  				return newError("failed to set TCP_WINDOW_CLAMP", err)
   164  			}
   165  		}
   166  
   167  		if config.TcpUserTimeout > 0 {
   168  			if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, unix.TCP_USER_TIMEOUT, int(config.TcpUserTimeout)); err != nil {
   169  				return newError("failed to set TCP_USER_TIMEOUT", err)
   170  			}
   171  		}
   172  
   173  		if config.TcpMaxSeg > 0 {
   174  			if err := syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, unix.TCP_MAXSEG, int(config.TcpMaxSeg)); err != nil {
   175  				return newError("failed to set TCP_MAXSEG", err)
   176  			}
   177  		}
   178  	}
   179  
   180  	if config.Tproxy.IsEnabled() {
   181  		if err := syscall.SetsockoptInt(int(fd), syscall.SOL_IP, syscall.IP_TRANSPARENT, 1); err != nil {
   182  			return newError("failed to set IP_TRANSPARENT").Base(err)
   183  		}
   184  	}
   185  
   186  	if config.ReceiveOriginalDestAddress && isUDPSocket(network) {
   187  		err1 := syscall.SetsockoptInt(int(fd), syscall.SOL_IPV6, unix.IPV6_RECVORIGDSTADDR, 1)
   188  		err2 := syscall.SetsockoptInt(int(fd), syscall.SOL_IP, syscall.IP_RECVORIGDSTADDR, 1)
   189  		if err1 != nil && err2 != nil {
   190  			return err1
   191  		}
   192  	}
   193  
   194  	if config.V6Only {
   195  		if err := syscall.SetsockoptInt(int(fd), syscall.SOL_IPV6, syscall.IPV6_V6ONLY, 1); err != nil {
   196  			return newError("failed to set IPV6_V6ONLY", err)
   197  		}
   198  	}
   199  
   200  	return nil
   201  }
   202  
   203  func setReuseAddr(fd uintptr) error {
   204  	if err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_REUSEADDR, 1); err != nil {
   205  		return newError("failed to set SO_REUSEADDR").Base(err).AtWarning()
   206  	}
   207  	return nil
   208  }
   209  
   210  func setReusePort(fd uintptr) error {
   211  	if err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, unix.SO_REUSEPORT, 1); err != nil {
   212  		return newError("failed to set SO_REUSEPORT").Base(err).AtWarning()
   213  	}
   214  	return nil
   215  }