sockopt_darwin.go 3.7 KB

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