options.go 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. package dialer
  2. import (
  3. "context"
  4. "net"
  5. "github.com/metacubex/mihomo/common/atomic"
  6. "github.com/metacubex/mihomo/component/resolver"
  7. )
  8. var (
  9. DefaultOptions []Option
  10. DefaultInterface = atomic.NewTypedValue[string]("")
  11. DefaultRoutingMark = atomic.NewInt32(0)
  12. )
  13. type NetDialer interface {
  14. DialContext(ctx context.Context, network, address string) (net.Conn, error)
  15. }
  16. type option struct {
  17. interfaceName string
  18. fallbackBind bool
  19. addrReuse bool
  20. routingMark int
  21. network int
  22. prefer int
  23. tfo bool
  24. mpTcp bool
  25. resolver resolver.Resolver
  26. netDialer NetDialer
  27. }
  28. type Option func(opt *option)
  29. func WithInterface(name string) Option {
  30. return func(opt *option) {
  31. opt.interfaceName = name
  32. }
  33. }
  34. func WithFallbackBind(fallback bool) Option {
  35. return func(opt *option) {
  36. opt.fallbackBind = fallback
  37. }
  38. }
  39. func WithAddrReuse(reuse bool) Option {
  40. return func(opt *option) {
  41. opt.addrReuse = reuse
  42. }
  43. }
  44. func WithRoutingMark(mark int) Option {
  45. return func(opt *option) {
  46. opt.routingMark = mark
  47. }
  48. }
  49. func WithResolver(r resolver.Resolver) Option {
  50. return func(opt *option) {
  51. opt.resolver = r
  52. }
  53. }
  54. func WithPreferIPv4() Option {
  55. return func(opt *option) {
  56. opt.prefer = 4
  57. }
  58. }
  59. func WithPreferIPv6() Option {
  60. return func(opt *option) {
  61. opt.prefer = 6
  62. }
  63. }
  64. func WithOnlySingleStack(isIPv4 bool) Option {
  65. return func(opt *option) {
  66. if isIPv4 {
  67. opt.network = 4
  68. } else {
  69. opt.network = 6
  70. }
  71. }
  72. }
  73. func WithTFO(tfo bool) Option {
  74. return func(opt *option) {
  75. opt.tfo = tfo
  76. }
  77. }
  78. func WithMPTCP(mpTcp bool) Option {
  79. return func(opt *option) {
  80. opt.mpTcp = mpTcp
  81. }
  82. }
  83. func WithNetDialer(netDialer NetDialer) Option {
  84. return func(opt *option) {
  85. opt.netDialer = netDialer
  86. }
  87. }
  88. func WithOption(o option) Option {
  89. return func(opt *option) {
  90. *opt = o
  91. }
  92. }