efi: Extend install() to support usr/lib/efi

Author: HuijingHei

src/efi.rs

@@ -349,8 +349,6 @@ impl Component for Efi {
             anyhow::bail!("No update metadata for component {} found", self.name());
         };
         log::debug!("Found metadata {}", meta.version);
-        let srcdir_name = component_updatedirname(self);
-        let ft = crate::filetree::FileTree::new_from_dir(&src_dir.sub_dir(&srcdir_name)?)?;
 
         // Let's attempt to use an already mounted ESP at the target
         // dest_root if one is already mounted there in a known ESP location.
@@ -371,16 +369,35 @@ impl Component for Efi {
             .with_context(|| format!("opening dest dir {}", destpath.display()))?;
         validate_esp_fstype(destd)?;
 
-        // TODO - add some sort of API that allows directly setting the working
-        // directory to a file descriptor.
-        std::process::Command::new("cp")
-            .args(["-rp", "--reflink=auto"])
-            .arg(&srcdir_name)
-            .arg(destpath)
-            .current_dir(format!("/proc/self/fd/{}", src_dir.as_raw_fd()))
-            .run()?;
+        let sysroot_path = Utf8Path::new(src_root);
+        let efilib_path = sysroot_path.join(EFILIB);
+        let efi_comps = if efilib_path.exists() {
+            get_efi_component_from_usr(&sysroot_path, EFILIB)?
+        } else {
+            None
+        };
+
+        let target_vendor_path;
+
+        // Copy EFI components from usr/lib/efi if existing
+        let ft = if let Some(efi_components) = efi_comps {
+            target_vendor_path = efilib_path.to_string();
+            for efi in efi_components {
+                log::trace!("Copy {} to {}", efi.path, destpath.display());
+                util::copy_in_fd(&src_dir, &efi.path.as_ref(), destpath.as_path())?;
+            }
+            crate::filetree::FileTree::new_from_dir(&src_dir.sub_dir(EFILIB)?)?
+        } else {
+            let updatesdir_name = Utf8PathBuf::from_path_buf(component_updatedirname(self))
+                .expect("Invalide UTF-8 path");
+            target_vendor_path = updatesdir_name.to_string();
+
+            util::copy_in_fd(&src_dir, &target_vendor_path, destpath.as_path())?;
+            crate::filetree::FileTree::new_from_dir(&src_dir.sub_dir(&target_vendor_path)?)?
+        };
+
         if update_firmware {
-            if let Some(vendordir) = self.get_efi_vendor(&src_root)? {
+            if let Some(vendordir) = self.get_efi_vendor(&target_vendor_path)? {
                 self.update_firmware(device, destd, &vendordir)?
             }
         }
@@ -449,12 +466,11 @@ impl Component for Efi {
             let mut packages = Vec::new();
             let sysroot_dir = Dir::open_ambient_dir(sysroot_path, cap_std::ambient_authority())?;
             for efi in efi_components {
-                Command::new("cp")
-                    .args(["-rp", "--reflink=auto"])
-                    .arg(&efi.path)
-                    .arg(crate::model::BOOTUPD_UPDATES_DIR)
-                    .current_dir(format!("/proc/self/fd/{}", sysroot_dir.as_raw_fd()))
-                    .run()?;
+                util::copy_in_fd(
+                    &sysroot_dir,
+                    &efi.path.as_ref(),
+                    crate::model::BOOTUPD_UPDATES_DIR,
+                )?;
                 packages.push(format!("{}-{}", efi.name, efi.version));
             }