2018-09-10 12:39:57 +00:00
|
|
|
package internet
|
|
|
|
|
|
|
|
import (
|
|
|
|
"strings"
|
|
|
|
"syscall"
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
TCP_FASTOPEN = 15
|
|
|
|
)
|
|
|
|
|
|
|
|
func applyOutboundSocketOptions(network string, address string, fd uintptr, config *SocketConfig) error {
|
|
|
|
if strings.HasPrefix(network, "tcp") {
|
|
|
|
switch config.Tfo {
|
|
|
|
case SocketConfig_Enable:
|
2018-09-10 18:06:58 +00:00
|
|
|
if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.IPPROTO_TCP, TCP_FASTOPEN, 1); err != nil {
|
2018-09-10 12:39:57 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
case SocketConfig_Disable:
|
2018-09-10 18:06:58 +00:00
|
|
|
if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.IPPROTO_TCP, TCP_FASTOPEN, 0); err != nil {
|
2018-09-10 12:39:57 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func applyInboundSocketOptions(network string, fd uintptr, config *SocketConfig) error {
|
|
|
|
if strings.HasPrefix(network, "tcp") {
|
|
|
|
switch config.Tfo {
|
|
|
|
case SocketConfig_Enable:
|
2018-09-10 18:06:58 +00:00
|
|
|
if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.IPPROTO_TCP, TCP_FASTOPEN, 1); err != nil {
|
2018-09-10 12:39:57 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
case SocketConfig_Disable:
|
2018-09-10 18:06:58 +00:00
|
|
|
if err := syscall.SetsockoptInt(syscall.Handle(fd), syscall.IPPROTO_TCP, TCP_FASTOPEN, 0); err != nil {
|
2018-09-10 12:39:57 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|