Add support for setting serial using AMIDE if found

Author: jackpot51

src/app/bios.rs

@@ -9,7 +9,7 @@ use std::fs::{find, load};
 use std::vars::{get_boot_item, get_boot_order, set_boot_item, set_boot_order};
 use uefi::status::{Error, Result};
 
-use super::{FIRMWAREDIR, FIRMWARENSH, FIRMWAREROM, UEFIFLASH, shell, Component};
+use super::{AMIDE, FIRMWAREDIR, FIRMWARENSH, FIRMWAREROM, SERIAL, UEFIFLASH, shell, Component};
 
 pub struct BiosComponent {
     bios_vendor: String,
@@ -107,6 +107,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 {
@@ -427,6 +444,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

@@ -31,12 +31,14 @@ mod bios;
 mod component;
 mod ec;
 
+static AMIDE: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware\\amide.efi");
 static ECROM: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware\\ec.rom");
 static EC2ROM: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware\\ec2.rom");
 static FIRMWAREDIR: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware");
 static FIRMWARENSH: &'static str = concat!("\\", env!("BASEDIR"), "\\res\\firmware.nsh");
 static FIRMWAREROM: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware\\firmware.rom");
 static MESETTAG: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware\\meset.tag");
+static SERIAL: &'static str = concat!("\\", env!("BASEDIR"), "\\serial");
 static SHELLEFI: &'static str = concat!("\\", env!("BASEDIR"), "\\res\\shell.efi");
 static SPLASHBMP: &'static str = concat!("\\", env!("BASEDIR"), "\\res\\splash.bmp");
 static UEFIFLASH: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware\\uefiflash.efi");