mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-08-23 18:52:31 +00:00
93 lines
1.9 KiB
Go
93 lines
1.9 KiB
Go
//go:build !linux && !darwin
|
|
|
|
package dialer
|
|
|
|
import (
|
|
"net"
|
|
"net/netip"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/Dreamacro/clash/component/iface"
|
|
)
|
|
|
|
func lookupLocalAddr(ifaceName string, network string, destination netip.Addr, port int) (net.Addr, error) {
|
|
ifaceObj, err := iface.ResolveInterface(ifaceName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var addr *netip.Prefix
|
|
switch network {
|
|
case "udp4", "tcp4":
|
|
addr, err = ifaceObj.PickIPv4Addr(destination)
|
|
case "tcp6", "udp6":
|
|
addr, err = ifaceObj.PickIPv6Addr(destination)
|
|
default:
|
|
if destination.IsValid() {
|
|
if destination.Is4() {
|
|
addr, err = ifaceObj.PickIPv4Addr(destination)
|
|
} else {
|
|
addr, err = ifaceObj.PickIPv6Addr(destination)
|
|
}
|
|
} else {
|
|
addr, err = ifaceObj.PickIPv4Addr(destination)
|
|
}
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if strings.HasPrefix(network, "tcp") {
|
|
return &net.TCPAddr{
|
|
IP: addr.Addr().AsSlice(),
|
|
Port: port,
|
|
}, nil
|
|
} else if strings.HasPrefix(network, "udp") {
|
|
return &net.UDPAddr{
|
|
IP: addr.Addr().AsSlice(),
|
|
Port: port,
|
|
}, nil
|
|
}
|
|
|
|
return nil, iface.ErrAddrNotFound
|
|
}
|
|
|
|
func bindIfaceToDialer(ifaceName string, dialer *net.Dialer, network string, destination netip.Addr) error {
|
|
if !destination.IsGlobalUnicast() {
|
|
return nil
|
|
}
|
|
|
|
local := uint64(0)
|
|
if dialer.LocalAddr != nil {
|
|
_, port, err := net.SplitHostPort(dialer.LocalAddr.String())
|
|
if err == nil {
|
|
local, _ = strconv.ParseUint(port, 10, 16)
|
|
}
|
|
}
|
|
|
|
addr, err := lookupLocalAddr(ifaceName, network, destination, int(local))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dialer.LocalAddr = addr
|
|
|
|
return nil
|
|
}
|
|
|
|
func bindIfaceToListenConfig(ifaceName string, _ *net.ListenConfig, network, address string) (string, error) {
|
|
_, port, err := net.SplitHostPort(address)
|
|
if err != nil {
|
|
port = "0"
|
|
}
|
|
|
|
local, _ := strconv.ParseUint(port, 10, 16)
|
|
|
|
addr, err := lookupLocalAddr(ifaceName, network, netip.Addr{}, int(local))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return addr.String(), nil
|
|
}
|