From 74ac452eebd4d3520c1f610260d55b7e7233eca9 Mon Sep 17 00:00:00 2001
From: Stefan Kreutz <mail@skreutz.com>
Date: Mon, 3 Jan 2022 15:49:53 +0100
Subject: Add initial implementation

---
 src/main.rs | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 src/main.rs

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..2a8c555
--- /dev/null
+++ b/src/main.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::into_app().debug_assert()
+}
-- 
cgit v1.2.3