Enable interface setting for socketopt under Windows (#2819)

* allow set interface under windows

Signed-off-by: San Zhang <zhangan@mail.com>

* polish code

Signed-off-by: San Zhang <zhangan@mail.com>

---------

Signed-off-by: San Zhang <zhangan@mail.com>
Co-authored-by: San Zhang <zhangan@mail.com>
This commit is contained in:
Zhang San 2023-12-18 06:56:15 +08:00 committed by GitHub
parent c01a30e8f4
commit 5a5e615b46
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,11 +1,16 @@
package internet package internet
import ( import (
"encoding/binary"
"net"
"syscall" "syscall"
"unsafe"
) )
const ( const (
TCP_FASTOPEN = 15 TCP_FASTOPEN = 15
IP_UNICAST_IF = 31
IPV6_UNICAST_IF = 31
) )
func setTFO(fd syscall.Handle, tfo int) error { func setTFO(fd syscall.Handle, tfo int) error {
@ -21,6 +26,26 @@ func setTFO(fd syscall.Handle, tfo int) error {
} }
func applyOutboundSocketOptions(network string, address string, fd uintptr, config *SocketConfig) error { func applyOutboundSocketOptions(network string, address string, fd uintptr, config *SocketConfig) error {
if config.Interface != "" {
inf, err := net.InterfaceByName(config.Interface)
if err != nil {
return newError("failed to find the interface").Base(err)
}
isV4 := (network == "tcp4")
if isV4 {
var bytes [4]byte
binary.BigEndian.PutUint32(bytes[:], uint32(inf.Index))
idx := *(*uint32)(unsafe.Pointer(&bytes[0]))
if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.IPPROTO_IP, IP_UNICAST_IF, int(idx)); err != nil {
return newError("failed to set IP_UNICAST_IF").Base(err)
}
} else {
if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.IPPROTO_IPV6, IPV6_UNICAST_IF, inf.Index); err != nil {
return newError("failed to set IPV6_UNICAST_IF").Base(err)
}
}
}
if isTCPSocket(network) { if isTCPSocket(network) {
if err := setTFO(syscall.Handle(fd), config.ParseTFOValue()); err != nil { if err := setTFO(syscall.Handle(fd), config.ParseTFOValue()); err != nil {
return err return err