50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
//go:build windows
|
||
|
||
package netforward
|
||
|
||
import (
|
||
"net"
|
||
"os"
|
||
"runtime"
|
||
"syscall"
|
||
"unsafe"
|
||
)
|
||
|
||
func SetTcpInfo(conn *net.TCPConn, usingKeepAlive bool, keepAliveIdel, keepAlivePeriod, keepAliveCount, userTimeout int) error {
|
||
//windows上，这两个值是毫秒，linux上则是秒
|
||
keepAlivePeriod *= 1000
|
||
keepAliveIdel *= 1000
|
||
if usingKeepAlive {
|
||
rawConn, err := conn.SyscallConn()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
err = rawConn.Control(func(fd uintptr) {
|
||
ka := syscall.TCPKeepalive{
|
||
OnOff: 1,
|
||
Time: uint32(keepAliveIdel),
|
||
Interval: uint32(keepAlivePeriod),
|
||
}
|
||
ret := uint32(0)
|
||
size := uint32(unsafe.Sizeof(ka))
|
||
err = syscall.WSAIoctl(syscall.Handle(fd), syscall.SIO_KEEPALIVE_VALS, (*byte)(unsafe.Pointer(&ka)), size, nil, 0, &ret, nil, 0)
|
||
runtime.KeepAlive(fd)
|
||
})
|
||
return os.NewSyscallError("wsaioctl", err)
|
||
}
|
||
return conn.SetKeepAlive(false)
|
||
}
|
||
|
||
func SetReUseAddr(fd uintptr) {
|
||
syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_REUSEADDR, 1)
|
||
}
|
||
|
||
func ControlSetReUseAddr(network, address string, c syscall.RawConn) (err error) {
|
||
if err := c.Control(func(fd uintptr) {
|
||
err = syscall.SetsockoptInt(syscall.Handle(fd), syscall.SOL_SOCKET, syscall.SO_REUSEADDR, 1)
|
||
}); err != nil {
|
||
return err
|
||
}
|
||
return
|
||
}
|