mirror of
https://github.com/MetaCubeX/mihomo.git
synced 2024-08-20 17:34:06 +00:00
42 lines
982 B
Go
42 lines
982 B
Go
//go:build linux
|
|
|
|
package dialer
|
|
|
|
import (
|
|
"net"
|
|
"net/netip"
|
|
"syscall"
|
|
)
|
|
|
|
func bindMarkToDialer(mark int, dialer *net.Dialer, _ string, _ netip.Addr) {
|
|
dialer.Control = bindMarkToControl(mark, dialer.Control)
|
|
}
|
|
|
|
func bindMarkToListenConfig(mark int, lc *net.ListenConfig, _, _ string) {
|
|
lc.Control = bindMarkToControl(mark, lc.Control)
|
|
}
|
|
|
|
func bindMarkToControl(mark int, chain controlFn) controlFn {
|
|
return func(network, address string, c syscall.RawConn) (err error) {
|
|
defer func() {
|
|
if err == nil && chain != nil {
|
|
err = chain(network, address, c)
|
|
}
|
|
}()
|
|
|
|
addrPort, err := netip.ParseAddrPort(address)
|
|
if err == nil && !addrPort.Addr().IsGlobalUnicast() {
|
|
return
|
|
}
|
|
|
|
return c.Control(func(fd uintptr) {
|
|
switch network {
|
|
case "tcp4", "udp4":
|
|
_ = syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_MARK, mark)
|
|
case "tcp6", "udp6":
|
|
_ = syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, syscall.SO_MARK, mark)
|
|
}
|
|
})
|
|
}
|
|
}
|