mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-11-16 19:56:51 +08:00
98 lines
2.5 KiB
Go
98 lines
2.5 KiB
Go
package outbound
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/netip"
|
|
|
|
N "github.com/metacubex/mihomo/common/net"
|
|
"github.com/metacubex/mihomo/component/dialer"
|
|
"github.com/metacubex/mihomo/component/resolver"
|
|
C "github.com/metacubex/mihomo/constant"
|
|
)
|
|
|
|
type Direct struct {
|
|
*Base
|
|
loopBack *loopBackDetector
|
|
}
|
|
|
|
type DirectOption struct {
|
|
BasicOption
|
|
Name string `proxy:"name"`
|
|
}
|
|
|
|
// DialContext implements C.ProxyAdapter
|
|
func (d *Direct) DialContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.Conn, error) {
|
|
if d.loopBack.CheckConn(metadata.SourceAddrPort()) {
|
|
return nil, fmt.Errorf("reject loopback connection to: %s", metadata.RemoteAddress())
|
|
}
|
|
opts = append(opts, dialer.WithResolver(resolver.DefaultResolver))
|
|
c, err := dialer.DialContext(ctx, "tcp", metadata.RemoteAddress(), d.Base.DialOptions(opts...)...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
N.TCPKeepAlive(c)
|
|
return d.loopBack.NewConn(NewConn(c, d)), nil
|
|
}
|
|
|
|
// ListenPacketContext implements C.ProxyAdapter
|
|
func (d *Direct) ListenPacketContext(ctx context.Context, metadata *C.Metadata, opts ...dialer.Option) (C.PacketConn, error) {
|
|
if d.loopBack.CheckPacketConn(metadata.SourceAddrPort()) {
|
|
return nil, fmt.Errorf("reject loopback connection to: %s", metadata.RemoteAddress())
|
|
}
|
|
// net.UDPConn.WriteTo only working with *net.UDPAddr, so we need a net.UDPAddr
|
|
if !metadata.Resolved() {
|
|
ip, err := resolver.ResolveIPWithResolver(ctx, metadata.Host, resolver.DefaultResolver)
|
|
if err != nil {
|
|
return nil, errors.New("can't resolve ip")
|
|
}
|
|
metadata.DstIP = ip
|
|
}
|
|
pc, err := dialer.NewDialer(d.Base.DialOptions(opts...)...).ListenPacket(ctx, "udp", "", netip.AddrPortFrom(metadata.DstIP, metadata.DstPort))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return d.loopBack.NewPacketConn(newPacketConn(pc, d)), nil
|
|
}
|
|
|
|
func NewDirectWithOption(option DirectOption) *Direct {
|
|
return &Direct{
|
|
Base: &Base{
|
|
name: option.Name,
|
|
tp: C.Direct,
|
|
udp: true,
|
|
tfo: option.TFO,
|
|
mpTcp: option.MPTCP,
|
|
iface: option.Interface,
|
|
rmark: option.RoutingMark,
|
|
prefer: C.NewDNSPrefer(option.IPVersion),
|
|
},
|
|
loopBack: newLoopBackDetector(),
|
|
}
|
|
}
|
|
|
|
func NewDirect() *Direct {
|
|
return &Direct{
|
|
Base: &Base{
|
|
name: "DIRECT",
|
|
tp: C.Direct,
|
|
udp: true,
|
|
prefer: C.DualStack,
|
|
},
|
|
loopBack: newLoopBackDetector(),
|
|
}
|
|
}
|
|
|
|
func NewCompatible() *Direct {
|
|
return &Direct{
|
|
Base: &Base{
|
|
name: "COMPATIBLE",
|
|
tp: C.Compatible,
|
|
udp: true,
|
|
prefer: C.DualStack,
|
|
},
|
|
loopBack: newLoopBackDetector(),
|
|
}
|
|
}
|