direct.go 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. package outbound
  2. import (
  3. "context"
  4. "errors"
  5. "os"
  6. "strconv"
  7. N "github.com/metacubex/mihomo/common/net"
  8. "github.com/metacubex/mihomo/component/dialer"
  9. "github.com/metacubex/mihomo/component/loopback"
  10. "github.com/metacubex/mihomo/component/resolver"
  11. C "github.com/metacubex/mihomo/constant"
  12. "github.com/metacubex/mihomo/constant/features"
  13. )
  14. var DisableLoopBackDetector, _ = strconv.ParseBool(os.Getenv("DISABLE_LOOPBACK_DETECTOR"))
  15. type Direct struct {
  16. *Base
  17. loopBack *loopback.Detector
  18. }
  19. type DirectOption struct {
  20. BasicOption
  21. Name string `proxy:"name"`
  22. }
  23. // DialContext implements C.ProxyAdapter
  24. func (d *Direct) DialContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.Conn, error) {
  25. if !features.Android && !DisableLoopBackDetector {
  26. if err := d.loopBack.CheckConn(metadata); err != nil {
  27. return nil, err
  28. }
  29. }
  30. opts = append(opts, dialer.WithResolver(resolver.DefaultResolver))
  31. c, err := dialer.DialContext(ctx, "tcp", metadata.RemoteAddress(), d.Base.DialOptions(opts...)...)
  32. if err != nil {
  33. return nil, err
  34. }
  35. N.TCPKeepAlive(c)
  36. return d.loopBack.NewConn(NewConn(c, d)), nil
  37. }
  38. // ListenPacketContext implements C.ProxyAdapter
  39. func (d *Direct) ListenPacketContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.PacketConn, error) {
  40. if !features.Android && !DisableLoopBackDetector {
  41. if err := d.loopBack.CheckPacketConn(metadata); err != nil {
  42. return nil, err
  43. }
  44. }
  45. // net.UDPConn.WriteTo only working with *net.UDPAddr, so we need a net.UDPAddr
  46. if !metadata.Resolved() {
  47. ip, err := resolver.ResolveIPWithResolver(ctx, metadata.Host, resolver.DefaultResolver)
  48. if err != nil {
  49. return nil, errors.New("can't resolve ip")
  50. }
  51. metadata.DstIP = ip
  52. }
  53. pc, err := dialer.NewDialer(d.Base.DialOptions(opts...)...).ListenPacket(ctx, "udp", "", metadata.AddrPort())
  54. if err != nil {
  55. return nil, err
  56. }
  57. return d.loopBack.NewPacketConn(newPacketConn(pc, d)), nil
  58. }
  59. func (d *Direct) IsL3Protocol(metadata *C.Metadata) bool {
  60. return true // tell DNSDialer don't send domain to DialContext, avoid lookback to DefaultResolver
  61. }
  62. func NewDirectWithOption(option DirectOption) *Direct {
  63. return &Direct{
  64. Base: &Base{
  65. name: option.Name,
  66. tp: C.Direct,
  67. udp: true,
  68. tfo: option.TFO,
  69. mpTcp: option.MPTCP,
  70. iface: option.Interface,
  71. rmark: option.RoutingMark,
  72. prefer: C.NewDNSPrefer(option.IPVersion),
  73. },
  74. loopBack: loopback.NewDetector(),
  75. }
  76. }
  77. func NewDirect() *Direct {
  78. return &Direct{
  79. Base: &Base{
  80. name: "DIRECT",
  81. tp: C.Direct,
  82. udp: true,
  83. prefer: C.DualStack,
  84. },
  85. loopBack: loopback.NewDetector(),
  86. }
  87. }
  88. func NewCompatible() *Direct {
  89. return &Direct{
  90. Base: &Base{
  91. name: "COMPATIBLE",
  92. tp: C.Compatible,
  93. udp: true,
  94. prefer: C.DualStack,
  95. },
  96. loopBack: loopback.NewDetector(),
  97. }
  98. }