Add support for adding a ramdisk to the disk image

Author: jasoncouture

src/lib.rs

@@ -17,19 +17,29 @@ mod mbr;
 mod pxe;
 
 const KERNEL_FILE_NAME: &str = "kernel-x86_64";
+const RAMDISK_FILE_NAME: &str = "ramdisk-x86_64";
 const BIOS_STAGE_3: &str = "boot-stage-3";
 const BIOS_STAGE_4: &str = "boot-stage-4";
 
 /// Create disk images for booting on legacy BIOS systems.
 pub struct BiosBoot {
     kernel: PathBuf,
+    ramdisk: Option<PathBuf>,
 }
 
 impl BiosBoot {
     /// Start creating a disk image for the given bootloader ELF executable.
     pub fn new(kernel_path: &Path) -> Self {
         Self {
             kernel: kernel_path.to_owned(),
+            ramdisk: None,
+        }
+    }
+
+    pub fn with_ramdisk(&self, ramdisk_path: &Path) -> Self {
+        Self {
+            kernel: self.kernel,
+            ramdisk: Some(ramdisk_path.to_owned())
         }
     }
 
@@ -66,7 +76,9 @@ impl BiosBoot {
         files.insert(KERNEL_FILE_NAME, self.kernel.as_path());
         files.insert(BIOS_STAGE_3, stage_3_path);
         files.insert(BIOS_STAGE_4, stage_4_path);
-
+        if self.ramdisk.is_some() {
+            files.insert(RAMDISK_FILE_NAME, self.ramdisk.unwrap().as_path())
+        }
         let out_file = NamedTempFile::new().context("failed to create temp file")?;
         fat::create_fat_filesystem(files, out_file.path())
             .context("failed to create BIOS FAT filesystem")?;
@@ -78,13 +90,22 @@ impl BiosBoot {
 /// Create disk images for booting on UEFI systems.
 pub struct UefiBoot {
     kernel: PathBuf,
+    ramdisk: Option<PathBuf>,
 }
 
 impl UefiBoot {
     /// Start creating a disk image for the given bootloader ELF executable.
     pub fn new(kernel_path: &Path) -> Self {
         Self {
             kernel: kernel_path.to_owned(),
+            ramdisk: None,
+        }
+    }
+
+    pub fn with_ramdisk(&self, ramdisk_path: &Path) -> Self {
+        Self {
+            kernel: self.kernel,
+            ramdisk: Some(ramdisk_path.to_owned())
         }
     }
 
@@ -125,6 +146,9 @@ impl UefiBoot {
         let mut files = BTreeMap::new();
         files.insert("efi/boot/bootx64.efi", bootloader_path);
         files.insert(KERNEL_FILE_NAME, self.kernel.as_path());
+        if self.ramdisk.is_some() {
+            files.insert(RAMDISK_FILE_NAME, self.ramdisk.unwrap().as_path())
+        }
 
         let out_file = NamedTempFile::new().context("failed to create temp file")?;
         fat::create_fat_filesystem(files, out_file.path())