2020-10-22 00:11:49 +08:00
|
|
|
package dialer
|
|
|
|
|
|
|
|
import (
|
2023-02-13 11:14:19 +08:00
|
|
|
"context"
|
2020-10-22 00:11:49 +08:00
|
|
|
"net"
|
2022-04-20 01:52:51 +08:00
|
|
|
"net/netip"
|
2020-10-22 00:11:49 +08:00
|
|
|
"syscall"
|
2021-09-06 23:07:34 +08:00
|
|
|
|
2023-11-03 21:01:45 +08:00
|
|
|
"github.com/metacubex/mihomo/component/iface"
|
2022-02-19 00:08:51 +08:00
|
|
|
|
|
|
|
"golang.org/x/sys/unix"
|
2020-10-22 00:11:49 +08:00
|
|
|
)
|
|
|
|
|
2023-02-13 11:14:19 +08:00
|
|
|
func bindControl(ifaceIdx int) controlFn {
|
|
|
|
return func(ctx context.Context, network, address string, c syscall.RawConn) (err error) {
|
2022-04-20 01:52:51 +08:00
|
|
|
addrPort, err := netip.ParseAddrPort(address)
|
|
|
|
if err == nil && !addrPort.Addr().IsGlobalUnicast() {
|
|
|
|
return
|
2020-10-25 20:31:01 +08:00
|
|
|
}
|
2020-10-22 00:11:49 +08:00
|
|
|
|
2021-11-10 22:19:11 +08:00
|
|
|
var innerErr error
|
|
|
|
err = c.Control(func(fd uintptr) {
|
2020-10-22 00:11:49 +08:00
|
|
|
switch network {
|
|
|
|
case "tcp4", "udp4":
|
2021-11-10 22:19:11 +08:00
|
|
|
innerErr = unix.SetsockoptInt(int(fd), unix.IPPROTO_IP, unix.IP_BOUND_IF, ifaceIdx)
|
2020-10-22 00:11:49 +08:00
|
|
|
case "tcp6", "udp6":
|
2021-11-10 22:19:11 +08:00
|
|
|
innerErr = unix.SetsockoptInt(int(fd), unix.IPPROTO_IPV6, unix.IPV6_BOUND_IF, ifaceIdx)
|
2020-10-22 00:11:49 +08:00
|
|
|
}
|
|
|
|
})
|
2021-11-10 22:19:11 +08:00
|
|
|
|
|
|
|
if innerErr != nil {
|
|
|
|
err = innerErr
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
2020-10-22 00:11:49 +08:00
|
|
|
}
|
2020-10-25 20:31:01 +08:00
|
|
|
}
|
|
|
|
|
2022-04-20 01:52:51 +08:00
|
|
|
func bindIfaceToDialer(ifaceName string, dialer *net.Dialer, _ string, _ netip.Addr) error {
|
2021-09-06 23:07:34 +08:00
|
|
|
ifaceObj, err := iface.ResolveInterface(ifaceName)
|
2020-10-25 20:31:01 +08:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-02-13 11:14:19 +08:00
|
|
|
addControlToDialer(dialer, bindControl(ifaceObj.Index))
|
2020-10-22 00:11:49 +08:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2024-04-10 15:34:28 +08:00
|
|
|
func bindIfaceToListenConfig(ifaceName string, lc *net.ListenConfig, _, address string, rAddrPort netip.AddrPort) (string, error) {
|
2021-09-06 23:07:34 +08:00
|
|
|
ifaceObj, err := iface.ResolveInterface(ifaceName)
|
2020-10-22 00:11:49 +08:00
|
|
|
if err != nil {
|
2021-09-06 23:07:34 +08:00
|
|
|
return "", err
|
2020-10-22 00:11:49 +08:00
|
|
|
}
|
|
|
|
|
2023-02-13 11:14:19 +08:00
|
|
|
addControlToListenConfig(lc, bindControl(ifaceObj.Index))
|
2021-09-06 23:07:34 +08:00
|
|
|
return address, nil
|
2020-10-22 00:11:49 +08:00
|
|
|
}
|
2023-02-13 10:14:59 +08:00
|
|
|
|
|
|
|
func ParseNetwork(network string, addr netip.Addr) string {
|
|
|
|
return network
|
|
|
|
}
|