Add support for setting serial using AMIDE if found

Author: jackpot51

src/app/bios.rs

@@ -13,7 +13,7 @@ use std::vars::{get_boot_item, get_boot_order, set_boot_item, set_boot_order};
 use std::uefi::reset::ResetType;
 use std::uefi::status::{Error, Result, Status};
 
-use super::{FIRMWARECAP, FIRMWAREDIR, FIRMWARENSH, FIRMWAREROM, H2OFFT, IFLASHV, UEFIFLASH, shell, Component, pci_mcfg, UefiMapper};
+use super::{AMIDE, FIRMWARECAP, FIRMWAREDIR, FIRMWARENSH, FIRMWAREROM, H2OFFT, IFLASHV, SERIAL, UEFIFLASH, shell, Component, pci_mcfg, UefiMapper};
 
 fn copy_region(region: intelflash::RegionKind, old_data: &[u8], new_data: &mut [u8]) -> core::result::Result<bool, String> {
     let old_opt = intelflash::Rom::new(old_data)?.get_region_base_limit(region)?;
@@ -188,6 +188,23 @@ impl BiosComponent {
             println!("Failed to locate EC");
         }
     }
+
+    fn set_serial(&self, serial: &str) -> Result<()> {
+        if find(AMIDE).is_ok() {
+            let cmd = format!("{} /SS {}", AMIDE, serial);
+            let status = shell(&cmd)?;
+
+            if status == 0 {
+                Ok(())
+            } else {
+                println!("{} Set Serial Error: {}", self.name(), status);
+                Err(Error::DeviceError)
+            }
+        } else {
+            //TODO
+            Err(Error::NotFound)
+        }
+    }
 }
 
 impl Component for BiosComponent {
@@ -541,6 +558,25 @@ impl Component for BiosComponent {
             }
         }
 
+        if let Ok(serial_vec) = load(SERIAL) {
+            match String::from_utf8(serial_vec) {
+                Ok(serial_str) => {
+                    let serial = serial_str.trim();
+                    match self.set_serial(&serial) {
+                        Ok(()) => {
+                            println!("Set serial to '{}'", serial);
+                        },
+                        Err(err) => {
+                            println!("Failed to set serial to '{}': {:?}", serial, err);
+                        }
+                    }
+                },
+                Err(err) => {
+                    println!("Failed to parse serial: {:?}", err);
+                }
+            }
+        }
+
         Ok(())
     }
 }

src/app/mod.rs

@@ -38,6 +38,7 @@ mod ec;
 mod mapper;
 mod pci;
 
+static AMIDE: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\amide.efi");
 static ECROM: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\ec.rom");
 static ECTAG: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\ec.tag");
 static EC2ROM: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\ec2.rom");
@@ -50,6 +51,7 @@ static IFLASHV: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\iflashv.efi")
 static IFLASHVTAG: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\iflashv.tag");
 static IPXEEFI: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\ipxe.efi");
 static MESETTAG: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\meset.tag");
+static SERIAL: &str = concat!("\\", env!("BASEDIR"), "\\serial");
 static SHELLEFI: &str = concat!("\\", env!("BASEDIR"), "\\res\\shell.efi");
 static SPLASHBMP: &str = concat!("\\", env!("BASEDIR"), "\\res\\splash.bmp");
 static UEFIFLASH: &str = concat!("\\", env!("BASEDIR"), "\\firmware\\uefiflash.efi");