Reboot in order to set serial

Author: jackpot51

res/firmware.nsh

@@ -115,5 +115,23 @@ if "%2" == "ec2" then
     exit 1
 endif
 
+if "%2" == "serial" then
+    if exist amide.efi then
+        if exist amide.tag then
+            rm amide.tag
+            amide.efi /SS "%3"
+            exit %lasterror%
+        else
+            echo > amide.tag
+            if not exist amide.tag then
+                echo "failed to create amide.tag"
+                exit 1
+            endif
+
+            reset
+        endif
+    endif
+endif
+
 echo "unknown command '%2'"
 exit 1

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::{AMIDE, FIRMWAREDIR, FIRMWARENSH, FIRMWAREROM, SERIAL, UEFIFLASH, shell, Component};
+use super::{FIRMWAREDIR, FIRMWARENSH, FIRMWAREROM, UEFIFLASH, shell, Component};
 
 pub struct BiosComponent {
     bios_vendor: String,
@@ -107,23 +107,6 @@ 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 {
@@ -444,25 +427,6 @@ 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,7 +31,7 @@ mod bios;
 mod component;
 mod ec;
 
-static AMIDE: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware\\amide.efi");
+static AMIDETAG: &'static str = concat!("\\", env!("BASEDIR"), "\\firmware\\amide.tag");
 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");
@@ -66,6 +66,18 @@ fn ac_connected() -> bool {
     }
 }
 
+fn set_serial(serial: &str) -> Result<()> {
+    find(FIRMWARENSH)?;
+    let cmd = format!("{} {} serial {}", FIRMWARENSH, FIRMWAREDIR, serial);
+    let status = shell(&cmd)?;
+    if status == 0 {
+        Ok(())
+    } else {
+        println!("Set Serial Error: {}", status);
+        Err(Error::DeviceError)
+    }
+}
+
 #[derive(Clone, Copy, Debug, Eq, PartialEq)]
 enum ValidateKind {
     Found,
@@ -218,6 +230,11 @@ fn inner() -> Result<()> {
         let c = if find(MESETTAG).is_ok() {
             // Skip enter if in manufacturing mode
             '\n'
+        } else if find(AMIDETAG).is_ok() {
+            // Skip enter if writing serial
+            components.clear();
+            validations.clear();
+            '\n'
         } else if find(UEFIFLASH).is_ok() {
             // Skip enter if flashing a meerkat
             if find(UEFIFLASHTAG).is_ok() {
@@ -278,6 +295,25 @@ fn inner() -> Result<()> {
         }
     }
 
+    if let Ok(serial_vec) = load(SERIAL) {
+        match String::from_utf8(serial_vec) {
+            Ok(serial_str) => {
+                let serial = serial_str.trim();
+                match 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);
+            }
+        }
+    }
+
     remove_override(option)?;
 
     if shutdown {