composefs/status: Check if deployment is soft rebootable

Signed-off-by: Pragyan Poudyal <[email protected]>

Author: Johan-Liebert1

crates/lib/src/bootc_composefs/status.rs

@@ -258,6 +258,12 @@ pub(crate) async fn get_composefs_status(
     composefs_deployment_status_from(&storage, booted_cfs.cmdline).await
 }
 
+fn set_reboot_capable(deployment: &mut BootEntry, booted_boot_digest: &String) -> Result<()> {
+    let boot_digest = deployment.composefs_boot_digest()?;
+    deployment.soft_reboot_capable = boot_digest == booted_boot_digest;
+    Ok(())
+}
+
 #[context("Getting composefs deployment status")]
 pub(crate) async fn composefs_deployment_status_from(
     storage: &Storage,
@@ -350,9 +356,9 @@ pub(crate) async fn composefs_deployment_status_from(
         anyhow::bail!("Could not determine boot type");
     };
 
-    let booted = host.require_composefs_booted()?;
+    let booted_cfs = host.require_composefs_booted()?;
 
-    let is_rollback_queued = match booted.bootloader {
+    let is_rollback_queued = match booted_cfs.bootloader {
         Bootloader::Grub => match boot_type {
             BootType::Bls => {
                 let bls_config = get_sorted_type1_boot_entries(boot_dir, false)?;
@@ -413,6 +419,29 @@ pub(crate) async fn composefs_deployment_status_from(
         host.spec.boot_order = BootOrder::Rollback
     };
 
+    // Can only soft reboot non UKI boot entries
+    if !matches!(boot_type, BootType::Uki) {
+        let booted_mut = host
+            .status
+            .booted
+            .as_mut()
+            .ok_or_else(|| anyhow::anyhow!("Failed to find booted entry"))?;
+
+        let booted_boot_digest = booted_mut.composefs_boot_digest()?;
+
+        if let Some(staged) = host.status.staged.as_mut() {
+            set_reboot_capable(staged, booted_boot_digest)?;
+        }
+
+        if let Some(rollback) = host.status.rollback.as_mut() {
+            set_reboot_capable(rollback, booted_boot_digest)?;
+        }
+
+        for deployment in &mut host.status.other_deployments {
+            set_reboot_capable(deployment, booted_boot_digest)?;
+        }
+    }
+
     Ok(host)
 }
 

crates/lib/src/cli.rs

@@ -11,8 +11,8 @@ use std::sync::Arc;
 
 use anyhow::{anyhow, ensure, Context, Result};
 use camino::{Utf8Path, Utf8PathBuf};
-use cap_std_ext::cap_std;
 use cap_std_ext::cap_std::fs::Dir;
+use cap_std_ext::cap_std;
 use clap::Parser;
 use clap::ValueEnum;
 use composefs::dumpfile;

crates/lib/src/status.rs

@@ -307,6 +307,13 @@ impl BootEntry {
             "BootEntry is not a composefs native boot entry"
         ))
     }
+
+    pub(crate) fn composefs_boot_digest(&self) -> Result<&String> {
+        self.require_composefs()?
+            .boot_digest
+            .as_ref()
+            .ok_or_else(|| anyhow::anyhow!("Could not find boot digest for deployment"))
+    }
 }
 
 /// A variant of [`get_status`] that requires a booted deployment.