sockopt_darwin.go 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. package internet
  2. import (
  3. "net"
  4. "golang.org/x/sys/unix"
  5. )
  6. const (
  7. // TCP_FASTOPEN_SERVER is the value to enable TCP fast open on darwin for server connections.
  8. TCP_FASTOPEN_SERVER = 0x01 // nolint: revive,stylecheck
  9. // TCP_FASTOPEN_CLIENT is the value to enable TCP fast open on darwin for client connections.
  10. TCP_FASTOPEN_CLIENT = 0x02 // nolint: revive,stylecheck
  11. )
  12. func applyOutboundSocketOptions(network string, address string, fd uintptr, config *SocketConfig) error {
  13. if isTCPSocket(network) {
  14. switch config.Tfo {
  15. case SocketConfig_Enable:
  16. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_FASTOPEN, TCP_FASTOPEN_CLIENT); err != nil {
  17. return err
  18. }
  19. case SocketConfig_Disable:
  20. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_FASTOPEN, 0); err != nil {
  21. return err
  22. }
  23. }
  24. if config.TcpKeepAliveInterval > 0 {
  25. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_KEEPINTVL, int(config.TcpKeepAliveInterval)); err != nil {
  26. return newError("failed to set TCP_KEEPINTVL").Base(err)
  27. }
  28. }
  29. if config.TcpKeepAliveIdle > 0 {
  30. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_KEEPALIVE, int(config.TcpKeepAliveIdle)); err != nil {
  31. return newError("failed to set TCP_KEEPALIVE (TCP keepalive idle time on Darwin)").Base(err)
  32. }
  33. }
  34. if config.TcpKeepAliveInterval > 0 || config.TcpKeepAliveIdle > 0 {
  35. if err := unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_KEEPALIVE, 1); err != nil {
  36. return newError("failed to set SO_KEEPALIVE").Base(err)
  37. }
  38. }
  39. }
  40. if config.BindToDevice != "" {
  41. iface, err := net.InterfaceByName(config.BindToDevice)
  42. if err != nil {
  43. return newError("failed to get interface ", config.BindToDevice).Base(err)
  44. }
  45. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_IP, unix.IP_BOUND_IF, iface.Index); err != nil {
  46. return newError("failed to set IP_BOUND_IF", err)
  47. }
  48. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_IPV6, unix.IPV6_BOUND_IF, iface.Index); err != nil {
  49. return newError("failed to set IPV6_BOUND_IF", err)
  50. }
  51. }
  52. if config.TxBufSize != 0 {
  53. if err := unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_SNDBUF, int(config.TxBufSize)); err != nil {
  54. return newError("failed to set SO_SNDBUF").Base(err)
  55. }
  56. }
  57. if config.RxBufSize != 0 {
  58. if err := unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_RCVBUF, int(config.RxBufSize)); err != nil {
  59. return newError("failed to set SO_RCVBUF").Base(err)
  60. }
  61. }
  62. return nil
  63. }
  64. func applyInboundSocketOptions(network string, fd uintptr, config *SocketConfig) error {
  65. if isTCPSocket(network) {
  66. switch config.Tfo {
  67. case SocketConfig_Enable:
  68. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_FASTOPEN, TCP_FASTOPEN_SERVER); err != nil {
  69. return err
  70. }
  71. case SocketConfig_Disable:
  72. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_FASTOPEN, 0); err != nil {
  73. return err
  74. }
  75. }
  76. if config.TcpKeepAliveInterval > 0 {
  77. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_KEEPINTVL, int(config.TcpKeepAliveInterval)); err != nil {
  78. return newError("failed to set TCP_KEEPINTVL").Base(err)
  79. }
  80. }
  81. if config.TcpKeepAliveIdle > 0 {
  82. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_TCP, unix.TCP_KEEPALIVE, int(config.TcpKeepAliveIdle)); err != nil {
  83. return newError("failed to set TCP_KEEPALIVE (TCP keepalive idle time on Darwin)").Base(err)
  84. }
  85. }
  86. if config.TcpKeepAliveInterval > 0 || config.TcpKeepAliveIdle > 0 {
  87. if err := unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_KEEPALIVE, 1); err != nil {
  88. return newError("failed to set SO_KEEPALIVE").Base(err)
  89. }
  90. }
  91. }
  92. if config.BindToDevice != "" {
  93. iface, err := net.InterfaceByName(config.BindToDevice)
  94. if err != nil {
  95. return newError("failed to get interface ", config.BindToDevice).Base(err)
  96. }
  97. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_IP, unix.IP_BOUND_IF, iface.Index); err != nil {
  98. return newError("failed to set IP_BOUND_IF", err)
  99. }
  100. if err := unix.SetsockoptInt(int(fd), unix.IPPROTO_IPV6, unix.IPV6_BOUND_IF, iface.Index); err != nil {
  101. return newError("failed to set IPV6_BOUND_IF", err)
  102. }
  103. }
  104. if config.TxBufSize != 0 {
  105. if err := unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_SNDBUF, int(config.TxBufSize)); err != nil {
  106. return newError("failed to set SO_SNDBUF/SO_SNDBUFFORCE").Base(err)
  107. }
  108. }
  109. if config.RxBufSize != 0 {
  110. if err := unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, unix.SO_RCVBUF, int(config.RxBufSize)); err != nil {
  111. return newError("failed to set SO_RCVBUF/SO_RCVBUFFORCE").Base(err)
  112. }
  113. }
  114. return nil
  115. }
  116. func bindAddr(fd uintptr, address []byte, port uint32) error {
  117. return nil
  118. }
  119. func setReuseAddr(fd uintptr) error {
  120. return nil
  121. }
  122. func setReusePort(fd uintptr) error {
  123. return nil
  124. }