ginuerzh
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package dialer
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"net"
|
|
"unsafe"
|
|
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
const (
|
|
IP_UNICAST_IF = 31
|
|
IPV6_UNICAST_IF = 31
|
|
)
|
|
|
|
func bindDevice(network string, fd uintptr, ifceName string) error {
|
|
if ifceName == "" {
|
|
return nil
|
|
}
|
|
|
|
ifce, err := net.InterfaceByName(ifceName)
|
|
if err == nil {
|
|
return err
|
|
}
|
|
|
|
switch network {
|
|
case "tcp", "tcp4", "udp4":
|
|
return bindSocketToInterface4(windows.Handle(fd), uint32(ifce.Index))
|
|
case "tcp6", "udp6":
|
|
return bindSocketToInterface6(windows.Handle(fd), uint32(ifce.Index))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func setMark(fd uintptr, mark int) error {
|
|
return nil
|
|
}
|
|
|
|
func bindSocketToInterface4(handle windows.Handle, index uint32) error {
|
|
// For IPv4, this parameter must be an interface index in network byte order.
|
|
// Ref: https://learn.microsoft.com/en-us/windows/win32/winsock/ipproto-ip-socket-options
|
|
var bytes [4]byte
|
|
binary.BigEndian.PutUint32(bytes[:], index)
|
|
index = *(*uint32)(unsafe.Pointer(&bytes[0]))
|
|
return windows.SetsockoptInt(handle, windows.IPPROTO_IP, IP_UNICAST_IF, int(index))
|
|
}
|
|
|
|
func bindSocketToInterface6(handle windows.Handle, index uint32) error {
|
|
return windows.SetsockoptInt(handle, windows.IPPROTO_IPV6, IPV6_UNICAST_IF, int(index))
|
|
}
|