unixgram: close previous vfkit connection using cancel context

Author: vyasgun

cmd/crc/cmd/daemon_darwin.go

@@ -13,6 +13,11 @@ import (
 	"github.com/pkg/errors"
 )
 
+var (
+	ctx    context.Context
+	cancel context.CancelFunc
+)
+
 func vsockListener() (net.Listener, error) {
 	_ = os.Remove(constants.TapSocketPath)
 	ln, err := net.Listen("unix", constants.TapSocketPath)
@@ -39,14 +44,19 @@ func unixgramListener(vn *virtualnetwork.VirtualNetwork) (*net.UnixConn, error)
 	if err != nil {
 		return conn, errors.Wrap(err, "failed to listen unixgram")
 	}
-	logging.Infof("listening on %s:", constants.UnixgramSocketPath)
+	logging.Infof("listening on %s", constants.UnixgramSocketPath)
 	vfkitConn, err := transport.AcceptVfkit(conn)
 	if err != nil {
 		return conn, errors.Wrap(err, "failed to accept vfkit connection")
 	}
+	if cancel != nil {
+		logging.Warnf("new connection to %s. Closing old connection", constants.UnixgramSocketPath)
+		cancel()
+	}
+	ctx, cancel = context.WithCancel(context.Background())
 	go func() {
-		err := vn.AcceptVfkit(context.Background(), vfkitConn)
-		if err != nil {
+		defer vfkitConn.Close()
+		if err := vn.AcceptVfkit(ctx, vfkitConn); err != nil {
 			logging.Errorf("failed to accept vfkit connection: %v", err)
 			return
 		}