sockopt_windows.go 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107
  1. package internet
  2. import (
  3. "net"
  4. "syscall"
  5. "golang.org/x/sys/windows"
  6. )
  7. const (
  8. TCP_FASTOPEN = 15 // nolint: revive,stylecheck
  9. IP_UNICAST_IF = 31 // nolint: revive,stylecheck
  10. IPV6_UNICAST_IF = 31 // nolint: revive,stylecheck
  11. )
  12. func setTFO(fd syscall.Handle, settings SocketConfig_TCPFastOpenState) error {
  13. switch settings {
  14. case SocketConfig_Enable:
  15. if err := syscall.SetsockoptInt(fd, syscall.IPPROTO_TCP, TCP_FASTOPEN, 1); err != nil {
  16. return err
  17. }
  18. case SocketConfig_Disable:
  19. if err := syscall.SetsockoptInt(fd, syscall.IPPROTO_TCP, TCP_FASTOPEN, 0); err != nil {
  20. return err
  21. }
  22. }
  23. return nil
  24. }
  25. func applyOutboundSocketOptions(network string, address string, fd uintptr, config *SocketConfig) error {
  26. if isTCPSocket(network) {
  27. if err := setTFO(syscall.Handle(fd), config.Tfo); err != nil {
  28. return err
  29. }
  30. if config.TcpKeepAliveIdle > 0 {
  31. if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1); err != nil {
  32. return newError("failed to set SO_KEEPALIVE", err)
  33. }
  34. }
  35. }
  36. if config.BindToDevice != "" {
  37. iface, err := net.InterfaceByName(config.BindToDevice)
  38. if err != nil {
  39. return newError("failed to get interface ", config.BindToDevice).Base(err)
  40. }
  41. if err := windows.SetsockoptInt(windows.Handle(fd), windows.IPPROTO_IP, IP_UNICAST_IF, iface.Index); err != nil {
  42. return newError("failed to set IP_UNICAST_IF", err)
  43. }
  44. if err := windows.SetsockoptInt(windows.Handle(fd), windows.IPPROTO_IPV6, IPV6_UNICAST_IF, iface.Index); err != nil {
  45. return newError("failed to set IPV6_UNICAST_IF", err)
  46. }
  47. }
  48. if config.TxBufSize != 0 {
  49. if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_SNDBUF, int(config.TxBufSize)); err != nil {
  50. return newError("failed to set SO_SNDBUF").Base(err)
  51. }
  52. }
  53. if config.RxBufSize != 0 {
  54. if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_RCVBUF, int(config.TxBufSize)); err != nil {
  55. return newError("failed to set SO_RCVBUF").Base(err)
  56. }
  57. }
  58. return nil
  59. }
  60. func applyInboundSocketOptions(network string, fd uintptr, config *SocketConfig) error {
  61. if isTCPSocket(network) {
  62. if err := setTFO(syscall.Handle(fd), config.Tfo); err != nil {
  63. return err
  64. }
  65. if config.TcpKeepAliveIdle > 0 {
  66. if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_KEEPALIVE, 1); err != nil {
  67. return newError("failed to set SO_KEEPALIVE", err)
  68. }
  69. }
  70. }
  71. if config.TxBufSize != 0 {
  72. if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_SNDBUF, int(config.TxBufSize)); err != nil {
  73. return newError("failed to set SO_SNDBUF").Base(err)
  74. }
  75. }
  76. if config.RxBufSize != 0 {
  77. if err := windows.SetsockoptInt(windows.Handle(fd), windows.SOL_SOCKET, windows.SO_RCVBUF, int(config.TxBufSize)); err != nil {
  78. return newError("failed to set SO_RCVBUF").Base(err)
  79. }
  80. }
  81. return nil
  82. }
  83. func bindAddr(fd uintptr, ip []byte, port uint32) error {
  84. return nil
  85. }
  86. func setReuseAddr(fd uintptr) error {
  87. return nil
  88. }
  89. func setReusePort(fd uintptr) error {
  90. return nil
  91. }