Update loadsnapshot params

Signed-off-by: char-1ee <[email protected]>

Author: char-1ee

bin/containerd-shim-aws-firecracker

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9595863ca1a4903d7cd0715a5da046e1ce4d88c76bd58c9deef586b583bed79a
-size 36354528
+oid sha256:593f7fff0ae8512859e08bd129375da10c13088d836cbaeb1608847b72c1cf28
+size 36382664

bin/default-rootfs.img

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7a2e7504d702db942e88158547c1076db786f861fb0b69ec96eb27716e1e37b1
-size 73318400
+oid sha256:f042d5db9797c16255db5a32d9644e8cad838a2ef46da9c08a6ae38a24111f97
+size 73342976

bin/firecracker-containerd

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c4450b3c8e9cb2db1a193cd2b9594eb90054cdbdae1d837a7f426b4b0d83950f
-size 72445304
+oid sha256:257ee96d4b2cd4a04f61833eafc31e2d814bb923e42ee0386077873f082aaa07
+size 72481176

bin/firecracker-ctr

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37487275ed6a08f4e759d9847435537bf52f20c207087e3e4226e599de8bae72
-size 35276648
+oid sha256:2f60c7f0984dfdba666c9159761fe93357d6558dbbe7c3326bc4959b3bdf72b6
+size 35333240

ctriface/iface.go

@@ -458,7 +458,7 @@ func (o *Orchestrator) LoadSnapshot(ctx context.Context, originVmID string, vmID
 		loadSnapshotMetric   *metrics.Metric = metrics.NewMetric()
 		tStart               time.Time
 		loadErr, activateErr error
-		loadDone             = make(chan int)
+		// loadDone             = make(chan int)
 	)
 
 	logger := log.WithFields(log.Fields{"vmID": vmID})
@@ -505,9 +505,12 @@ func (o *Orchestrator) LoadSnapshot(ctx context.Context, originVmID string, vmID
 		BackendType: fileBackend,
 		BackendPath: snap.GetMemFilePath(),
 	}
+	conf.ResumeVM = true
+	conf.EnableDiffSnapshots = false
 
 	var sendfdConn *net.UnixConn
-	uffdListenerCh := make(chan struct{}, 1)
+	// uffdListenerCh := make(chan struct{}, 1)
+	var listener net.Listener
 
 	if o.GetUPFEnabled() {
 		logger.Debug("TEST: UPF is enabled")
@@ -527,73 +530,76 @@ func (o *Orchestrator) LoadSnapshot(ctx context.Context, originVmID string, vmID
 			os.Remove(conf.MemBackend.BackendPath)
 		}
 
-		go func() {
-			listener, err := net.Listen("unix", conf.MemBackend.BackendPath)
-			if err != nil {
-				logger.Error("failed to listen to uffd socket")
-				return
-			}
-			defer listener.Close()
+		// ===============================================================
+		listener, err = net.Listen("unix", conf.MemBackend.BackendPath)
+		if err != nil {
+			logger.Error("failed to listen to uffd socket")
+			return
+		}
+		// defer listener.Close()
 
-			logger.Debug("Listening ...")
-			conn, err := listener.Accept()
-			if err != nil {
-				logger.Error("failed to accept connection to uffd socket")
-				return
-			}
+		// logger.Debug("Listening ...")
+		// conn, err := listener.Accept()
+		// if err != nil {
+		// 	logger.Error("failed to accept connection to uffd socket")
+		// 	return
+		// }
 
-			sendfdConn, _ = conn.(*net.UnixConn)
-			close(uffdListenerCh)
-		}()
+		// sendfdConn, _ = conn.(*net.UnixConn)
+		// close(uffdListenerCh)
 
-		time.Sleep(10 * time.Second) // TODO: sleep for 10 seconds to wait for the uffd socket to be ready
+		// time.Sleep(10 * time.Second) // TODO: sleep for 10 seconds to wait for the uffd socket to be ready
 	}
 
 	tStart = time.Now()
 
-	go func() {
-		defer close(loadDone)
-
-		confStr, _ := json.Marshal(conf)
-		logger.Debugf("TEST: CreateVM request: %s", confStr)
+	confStr, _ := json.Marshal(conf)
+	logger.Debugf("TEST: CreateVM request: %s", confStr)
 
-		if _, loadErr := o.fcClient.CreateVM(ctx, conf); loadErr != nil {
-			logger.Error("Failed to load snapshot of the VM: ", loadErr)
-			logger.Errorf("snapFilePath: %s, memFilePath: %s, newSnapshotPath: %s", snap.GetSnapshotFilePath(), snap.GetMemFilePath(), containerSnap.GetDevicePath())
-			files, err := os.ReadDir(filepath.Dir(snap.GetSnapshotFilePath()))
-			if err != nil {
-				logger.Error(err)
-			}
-
-			snapFiles := ""
-			for _, f := range files {
-				snapFiles += f.Name() + ", "
-			}
+	if _, loadErr := o.fcClient.CreateVM(ctx, conf); loadErr != nil {
+		logger.Error("Failed to load snapshot of the VM: ", loadErr)
+		logger.Errorf("snapFilePath: %s, memFilePath: %s, newSnapshotPath: %s", snap.GetSnapshotFilePath(), snap.GetMemFilePath(), containerSnap.GetDevicePath())
+		files, err := os.ReadDir(filepath.Dir(snap.GetSnapshotFilePath()))
+		if err != nil {
+			logger.Error(err)
+		}
 
-			logger.Error(snapFiles)
+		snapFiles := ""
+		for _, f := range files {
+			snapFiles += f.Name() + ", "
+		}
 
-			files, _ = os.ReadDir(filepath.Dir(containerSnap.GetDevicePath()))
-			if err != nil {
-				logger.Error(err)
-			}
+		logger.Error(snapFiles)
 
-			snapFiles = ""
-			for _, f := range files {
-				snapFiles += f.Name() + ", "
-			}
-			logger.Error(snapFiles)
+		files, _ = os.ReadDir(filepath.Dir(containerSnap.GetDevicePath()))
+		if err != nil {
+			logger.Error(err)
 		}
-	}()
 
+		snapFiles = ""
+		for _, f := range files {
+			snapFiles += f.Name() + ", "
+		}
+		logger.Error(snapFiles)
+	}
 	logger.Debug("TEST: CreatVM request sent")
 
-	<-loadDone
+	// <-loadDone
 
 	if o.GetUPFEnabled() {
+		logger.Debug("Listening ...")
+		conn, err := listener.Accept() // TODO: a question, must accept() first before connect()?
+		if err != nil {
+			logger.Error("failed to accept connection to uffd socket")
+			return
+		}
+		sendfdConn, _ = conn.(*net.UnixConn)
+		listener.Close()
+		// close(uffdListenerCh)
 
 		logger.Debug("TEST: Registering VM with snap with the memory manager")
 
-		<-uffdListenerCh
+		// <-uffdListenerCh
 
 		stateCfg := manager.SnapshotStateCfg{
 			VMID:             vmID,
@@ -613,6 +619,8 @@ func (o *Orchestrator) LoadSnapshot(ctx context.Context, originVmID string, vmID
 		if activateErr = o.memoryManager.Activate(originVmID, sendfdConn); activateErr != nil {
 			logger.Warn("Failed to activate VM in the memory manager", activateErr)
 		}
+
+		// time.Sleep(30 * time.Minute) // pause to see fc logs
 	}
 
 	// <-loadDone