Add acpi-dumper

Cargo.toml

@@ -1,2 +1,2 @@
 [workspace]
-members = ["acpi", "aml"]
+members = ["acpi", "aml", "acpi-dumper"]

acpi-dumper/Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "acpi-dumper"
+version = "0.0.0"
+publish = false
+authors = ["Isaac Woods", "Matt Taylor"]
+repository = "https://github.com/rust-osdev/acpi"
+description = "ACPI table dumper"
+categories = ["hardware-support"]
+readme = "../README.md"
+license = "MIT/Apache-2.0"
+edition = "2018"
+
+[dependencies]
+pico-args = "0.2.0"
+libc = "0.2.61"

acpi-dumper/src/main.rs

@@ -0,0 +1,102 @@
+#[cfg(target_os = "linux")]
+fn main() {
+    use pico_args::Arguments;
+    use std::{
+        fs::{self, Permissions},
+        os::unix::fs::PermissionsExt,
+        path::PathBuf,
+        process::{self, Command},
+    };
+
+    // Find output folder from command line arguments
+    let mut args = Arguments::from_env();
+    let name: String = match args.value_from_str("--name") {
+        Ok(Some(name)) => {
+            println!("output directory is: dumps/{}", name);
+            name
+        }
+        _ => {
+            eprintln!(
+                "usage: acpi-dumper --name [NAME]\n\
+                 the acpi tables will be dumped to the dumps/[name] directory"
+            );
+            process::exit(-1);
+        }
+    };
+
+    // Check lshw is installed
+    match Command::new("lshw").arg("-version").output() {
+        Ok(output) => {
+            println!(
+                "found lshw version: {}",
+                String::from_utf8(output.stderr)
+                    .expect("lshw returned invalid utf-8")
+                    .rsplit_terminator(' ')
+                    .next()
+                    .unwrap_or("unknown")
+                    .trim_end()
+            );
+        }
+        Err(e) => {
+            eprintln!("error: lshw failed to execute ({}) - please check that it is installed", e);
+            process::exit(-1);
+        }
+    };
+
+    if unsafe { libc::geteuid() } != 0 {
+        eprintln!("error: please re-run with root privileges");
+        process::exit(-1);
+    }
+
+    // Create output folder
+    let mut out_path = PathBuf::from("./dumps");
+    out_path.push(name);
+    fs::create_dir(&out_path).expect("failed to create output directory");
+    fs::set_permissions(&out_path, Permissions::from_mode(0o777))
+        .expect("failed to set permissions for output directory");
+
+    println!("dumping acpi tables...");
+
+    // Loop over the ACPI tables, dump their contents
+    for file in fs::read_dir("/sys/firmware/acpi/tables")
+        .expect("failed to access acpi tables, check permissions")
+        .filter_map(|entry| entry.ok()) // Ignore failures
+        .filter(|entry| entry.metadata().unwrap().is_file())
+    {
+        let table_signature = file.file_name().into_string().expect("invalid utf-8 file name in acpi tables");
+
+        // All tables have a 4 letter signature (apart from SSDTs)
+        if table_signature.len() == 4 || &table_signature[..4] != "SSDT" {
+            // Read table
+            let data = fs::read(file.path()).expect("failed to read acpi table");
+            out_path.push(table_signature + ".bin");
+
+            // Dump table to disk
+            fs::write(&out_path, data).expect("failed to write acpi table");
+
+            // Allow anyone to read or write the file
+            fs::set_permissions(&out_path, Permissions::from_mode(0o666))
+                .expect("failed to set permissions of dumped acpi table");
+            out_path.pop();
+        }
+    }
+
+    println!("done!");
+
+    println!("capturing lshw output...");
+    let lshw_info = Command::new("lshw").output().expect("failed to get hardware information from lshw");
+
+    // Write lshw output to disk
+    let lshw_info = String::from_utf8(lshw_info.stdout).expect("lshw returned invalid utf-8");
+    out_path.push("lshw.txt");
+    fs::write(&out_path, lshw_info).expect("failed to write lshw dump");
+    fs::set_permissions(&out_path, Permissions::from_mode(0o666))
+        .expect("failed to set permissions of dumped acpi table");
+    out_path.pop();
+    println!("done!");
+}
+
+#[cfg(not(target_os = "linux"))]
+fn main() {
+    std::compilation_error!("acpi-dumper currently only supports linux");
+}