Skip to content

Commit

Permalink
fix security
Browse files Browse the repository at this point in the history
  • Loading branch information
changluyi committed Jul 10, 2023
1 parent 72f269b commit b8eae39
Show file tree
Hide file tree
Showing 2 changed files with 50 additions and 11 deletions.
21 changes: 18 additions & 3 deletions pkg/daemon/tproxy.go
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,12 @@ func (c *Controller) StartTProxyForwarding(stopCh <-chan struct{}) {
return
}

defer tcpListener.Close()
defer func() {
if err := tcpListener.Close(); err != nil {
klog.Errorf("Error tcpListener Close err: %v ", err)
}
}()

go listenTCP()

<-stopCh
Expand Down Expand Up @@ -298,7 +303,13 @@ func listenTCP() {
func handleRedirectFlow(conn net.Conn) {

klog.V(5).Info("Accepting TCP connection from %v with destination of %v", conn.RemoteAddr().String(), conn.LocalAddr().String())
defer conn.Close()

defer func() {
if err := conn.Close(); err != nil {
klog.Errorf("conn Close err: %v ", err)
}
}()

podIPPort := conn.LocalAddr().String()
var podIP, probePort string
if strings.HasPrefix(podIPPort, "[") {
Expand Down Expand Up @@ -358,7 +369,11 @@ func probePortInNs(podIP, probePort string, transferHTTPMessage bool, conn net.C
customVPCPodTCPProbeIPPort.Store(getIPPortString(podIP, probePort), true)
}

defer remoteConn.Close()
defer func() {
if err := remoteConn.Close(); err != nil {
klog.Errorf("remoteConn %v Close err: %v ", remoteConn, err)
}
}()

if transferHTTPMessage {
var streamWait sync.WaitGroup
Expand Down
40 changes: 32 additions & 8 deletions pkg/tproxy/tproxy_tcp.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,8 @@ import (
"os"
"strings"
"syscall"

"k8s.io/klog/v2"
)

// Listener describes a TCP Listener
Expand Down Expand Up @@ -70,7 +72,12 @@ func listenTCP(device, network string, laddr *net.TCPAddr) (net.Listener, error)
if err != nil {
return nil, &net.OpError{Op: "listen", Net: network, Source: nil, Addr: laddr, Err: fmt.Errorf("get file descriptor: %s", err)}
}
defer fileDescriptorSource.Close()

defer func() {
if err := fileDescriptorSource.Close(); err != nil {
klog.Errorf("fileDescriptorSource %v Close err: %v ", fileDescriptorSource, err)
}
}()

if device != "" {
if err = syscall.BindToDevice(int(fileDescriptorSource.Fd()), device); err != nil {
Expand Down Expand Up @@ -167,38 +174,55 @@ func dialTCP(device string, laddr, raddr *net.TCPAddr, dontAssumeRemote bool, is
}

if err = syscall.SetsockoptInt(fileDescriptor, syscall.SOL_SOCKET, syscall.SO_REUSEADDR, 1); err != nil {
syscall.Close(fileDescriptor)
if err := syscall.Close(fileDescriptor); err != nil {
klog.Errorf("fileDescriptor %v Close err: %v ", fileDescriptor, err)
}
return nil, &net.OpError{Op: "dial", Err: fmt.Errorf("set socket option: SO_REUSEADDR: %s", err)}
}

if err = syscall.SetsockoptInt(fileDescriptor, syscall.SOL_IP, syscall.IP_TRANSPARENT, 1); err != nil {
syscall.Close(fileDescriptor)
if err := syscall.Close(fileDescriptor); err != nil {
klog.Errorf("fileDescriptor %v Close err: %v ", fileDescriptor, err)
}
return nil, &net.OpError{Op: "dial", Err: fmt.Errorf("set socket option: IP_TRANSPARENT: %s", err)}
}

if err = syscall.SetNonblock(fileDescriptor, isnonblocking); err != nil {
syscall.Close(fileDescriptor)
if err := syscall.Close(fileDescriptor); err != nil {
klog.Errorf("fileDescriptor %v Close err: %v ", fileDescriptor, err)
}
return nil, &net.OpError{Op: "dial", Err: fmt.Errorf("set socket option: SO_NONBLOCK: %s", err)}
}

if !dontAssumeRemote {
if err = syscall.Bind(fileDescriptor, localSocketAddress); err != nil {
syscall.Close(fileDescriptor)
if err := syscall.Close(fileDescriptor); err != nil {
klog.Errorf("fileDescriptor %v Close err: %v ", fileDescriptor, err)
}
return nil, &net.OpError{Op: "dial", Err: fmt.Errorf("socket bind: %s", err)}
}
}

if err = syscall.Connect(fileDescriptor, remoteSocketAddress); err != nil && !strings.Contains(err.Error(), "operation now in progress") {
syscall.Close(fileDescriptor)
if err := syscall.Close(fileDescriptor); err != nil {
klog.Errorf("fileDescriptor %v Close err: %v ", fileDescriptor, err)
}
return nil, &net.OpError{Op: "dial", Err: fmt.Errorf("socket connect: %s", err)}
}

fdFile := os.NewFile(uintptr(fileDescriptor), fmt.Sprintf("net-tcp-dial-%s", raddr.String()))
defer fdFile.Close()
defer func() {
if err := fdFile.Close(); err != nil {
klog.Errorf("fdFile %v Close err: %v ", fdFile, err)
}
}()

remoteConn, err := net.FileConn(fdFile)
if err != nil {
syscall.Close(fileDescriptor)
if err := syscall.Close(fileDescriptor); err != nil {
klog.Errorf("fileDescriptor %v Close err: %v ", fileDescriptor, err)
}

return nil, &net.OpError{Op: "dial", Err: fmt.Errorf("convert file descriptor to connection: %s", err)}
}

Expand Down

0 comments on commit b8eae39

Please sign in to comment.