Fix unstable vz network issue

Signed-off-by: Balaji Vijayakumar <[email protected]>

Author: balajiv113

pkg/networks/usernet/UDPFileConn.go

@@ -0,0 +1,21 @@
+package usernet
+
+import (
+	"errors"
+	"net"
+	"time"
+)
+
+type UDPFileConn struct {
+	net.Conn
+}
+
+func (conn *UDPFileConn) Read(b []byte) (n int, err error) {
+	// Check if the connection has been closed
+	if err := conn.SetReadDeadline(time.Time{}); err != nil {
+		if opErr, ok := err.(*net.OpError); ok && opErr.Err.Error() == "use of closed network connection" {
+			return 0, errors.New("UDPFileConn connection closed")
+		}
+	}
+	return conn.Conn.Read(b)
+}

pkg/networks/usernet/gvproxy.go

@@ -163,11 +163,11 @@ func listenFD(ctx context.Context, vn *virtualnetwork.VirtualNetwork) error {
 			files[0].Close()
 
 			go func() {
-				err = vn.AcceptBess(ctx, fileConn)
+				err = vn.AcceptBess(ctx, &UDPFileConn{Conn: fileConn})
 				if err != nil {
 					logrus.Error("FD connection closed with error", err)
 				}
-				defer fileConn.Close()
+				fileConn.Close()
 			}()
 			select {
 			case <-ctx.Done():

pkg/vz/vm_darwin.go

@@ -36,6 +36,9 @@ type virtualMachineWrapper struct {
 	stopped bool
 }
 
+// Hold all *os.File created via socketpair() so that they won't get garbage collected. f.FD() gets invalid if f gets garbage collected.
+var vmNetworkFiles = make([]*os.File, 1)
+
 func startVM(ctx context.Context, driver *driver.BaseDriver) (*virtualMachineWrapper, chan error, error) {
 	usernetClient, err := startUsernet(ctx, driver)
 	if err != nil {
@@ -57,7 +60,11 @@ func startVM(ctx context.Context, driver *driver.BaseDriver) (*virtualMachineWra
 	errCh := make(chan error)
 	go func() {
 		//Handle errors via errCh and handle stop vm during context close
-
+		defer func() {
+			for i := range vmNetworkFiles {
+				vmNetworkFiles[i].Close()
+			}
+		}()
 		for {
 			select {
 			case <-ctx.Done():
@@ -700,5 +707,14 @@ func createSockPair() (*os.File, *os.File, error) {
 	if err = syscall.SetsockoptInt(clientFD, syscall.SOL_SOCKET, syscall.SO_RCVBUF, 4*1024*1024); err != nil {
 		return nil, nil, err
 	}
-	return os.NewFile(uintptr(serverFD), "server"), os.NewFile(uintptr(clientFD), "client"), nil
+	server := os.NewFile(uintptr(serverFD), "server")
+	client := os.NewFile(uintptr(clientFD), "client")
+	runtime.SetFinalizer(server, func(file *os.File) {
+		logrus.Debugf("Server network file GC'ed")
+	})
+	runtime.SetFinalizer(client, func(file *os.File) {
+		logrus.Debugf("Client network file GC'ed")
+	})
+	vmNetworkFiles = append(vmNetworkFiles, server, client)
+	return server, client, nil
 }