Merge pull request #47 from arvindskumar99/snpguestok-arvind

Adding snpguest ok
virtee · Mar 28, 2024 · 9579368 · 9579368
2 parents a4b9ca0 + a01a360
commit 9579368
Show file tree

Hide file tree

Showing 5 changed files with 395 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,3 @@
 /target
-Cargo.lock
+Cargo.lock
+.vscode
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -33,3 +33,6 @@ x509-parser = { version="^0.14", features=["verify"] }
 asn1-rs = "0.5.2"
 rand = "0.8.5"
 tss-esapi = { version = "7.2", optional=true }
+msru = "0.2.0"
+colorful = "0.2.2"
+bitfield = "0.13.2"
diff --git a/src/main.rs b/src/main.rs
@@ -5,6 +5,7 @@ mod certs;
 mod display;
 mod fetch;
 mod key;
+mod ok;
 mod report;
 mod verify;
 
@@ -54,6 +55,9 @@ enum SnpGuestCmd {
 
     #[structopt(about = "Key command to generate derived key.")]
     Key(KeyArgs),
+
+    #[structopt(about = "Probe system for SEV-SNP support")]
+    Ok,
 }
 
 fn main() -> Result<()> {
@@ -74,6 +78,7 @@ fn main() -> Result<()> {
         SnpGuestCmd::Verify(subcmd) => verify::cmd(subcmd, snpguest.quiet),
         SnpGuestCmd::Display(subcmd) => display::cmd(subcmd, snpguest.quiet),
         SnpGuestCmd::Key(args) => key::get_derived_key(args),
+        SnpGuestCmd::Ok => ok::cmd(snpguest.quiet),
     };
 
     if let Err(ref e) = status {