Allow upgrade in 'exited' state

Dstack-TEE · Dec 18, 2024 · 3201723 · 3201723
1 parent 5c6f092
commit 3201723
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tdxctl/src/fde_setup.rs b/tdxctl/src/fde_setup.rs
@@ -296,6 +296,7 @@ impl SetupFdeArgs {
         let hash_file = self.rootfs_dir.join(".rootfs_hash");
         let existing_rootfs_hash = fs::read(&hash_file).unwrap_or_default();
         if existing_rootfs_hash != host_shared.vm_config.rootfs_hash {
+            info!("Rootfs hash changed, upgrading the rootfs");
             fs::remove_file(&hash_file).context("Failed to remove old rootfs hash file")?;
             nc.notify_q("boot.progress", "upgrading rootfs").await;
             self.extract_rootfs(&host_shared.vm_config.rootfs_hash)

diff --git a/teepod/src/main_service.rs b/teepod/src/main_service.rs
@@ -244,7 +244,7 @@ impl TeepodRpc for RpcHandler {
             .vm_info(&request.id)
             .await?
             .context("vm not found")?;
-        if vm.status != "stopped" {
+        if !["stopped", "exited"].contains(&vm.status.as_str()) {
             return Err(anyhow!(
                 "vm should be stopped before resize: {}",
                 request.id