Extract wpa-psk-cli cratewpa-psk-cli-0.1.0wpa-psk-0.2.0

1 files changed, 48 insertions, 0 deletions

diff --git a/wpa-psk-cli/src/bin/wpa-psk.rs b/wpa-psk-cli/src/bin/wpa-psk.rs
new file mode 100644
index 0000000..181585c
--- /dev/null
+++ b/wpa-psk-cli/src/bin/wpa-psk.rs
@@ -0,0 +1,48 @@
+use std::process::exit;
+
+use clap::Parser;
+use wpa_psk::{bytes_to_hex, wpa_psk, wpa_psk_unchecked, Passphrase, Ssid};
+
+/// Compute the WPA-PSK of a Wi-Fi SSID and passphrase.
+#[derive(Debug, Parser)]
+#[clap(about, version, author)]
+struct Args {
+    /// An SSID consisting of 1 up to 32 arbitrary bytes.
+    ssid: String,
+
+    /// A passphrase consisting of 8 up to 63 printable ASCII characters.
+    passphrase: String,
+
+    /// Don't check the given SSID and passphrase.
+    #[clap(short, long)]
+    force: bool,
+}
+
+fn main() {
+    exit(match run() {
+        Ok(_) => 0,
+        Err(err) => {
+            eprintln!("{}", err);
+            1
+        }
+    })
+}
+
+fn run() -> Result<(), Box<dyn std::error::Error>> {
+    let args = Args::try_parse()?;
+    let psk = if args.force {
+        wpa_psk_unchecked(args.ssid.as_bytes(), args.passphrase.as_bytes())
+    } else {
+        let ssid = Ssid::try_from(args.ssid.as_bytes())?;
+        let passphrase = Passphrase::try_from(args.passphrase.as_bytes())?;
+        wpa_psk(&ssid, &passphrase)
+    };
+    println!("0x{}", bytes_to_hex(&psk));
+    Ok(())
+}
+
+#[test]
+fn verify_clap_app() {
+    use clap::IntoApp;
+    Args::command().debug_assert()
+}