gecko-dev/third_party/rust/argon2rs/examples/cli.rs

56 строки
1.5 KiB
Rust

extern crate argon2rs;
use argon2rs::{Argon2, Variant};
use std::string::String;
use std::env;
use std::io::{Read, stdin};
const CLI_TOOL_SALT_LEN: usize = 16;
fn that_cli_tool(msg: &[u8], salt: &[u8], passes: u32, lanes: u32, logkib: u32)
-> [u8; argon2rs::defaults::LENGTH] {
assert!(salt.len() <= CLI_TOOL_SALT_LEN && passes > 0 && logkib > 0 &&
lanes > 0);
let a = Argon2::new(passes, lanes, 1 << logkib, Variant::Argon2i)
.ok()
.unwrap();
let mut s = [0; CLI_TOOL_SALT_LEN];
for (&v, mut k) in salt.iter().zip(s.iter_mut()) {
*k = v;
}
let mut out = [0 as u8; argon2rs::defaults::LENGTH];
a.hash(&mut out, msg, &s, &[], &[]);
out
}
fn to_string(bs: &[u8]) -> String {
let mut rv = String::new();
for b in bs.iter() {
rv.push_str(&format!("{:02x}", b));
}
rv
}
fn main() {
let args: Vec<String> = env::args().collect();
if args.len() != 5 {
println!("Usage: {} passes lanes logkib salt", args[0]);
println!("where salt.len() <= {}, memory usage is 2^logkib, and \
plaintext is read from stdin.", CLI_TOOL_SALT_LEN);
return;
}
let t: u32 = args[1].parse().unwrap();
let l: u32 = args[2].parse().unwrap();
let logm: u32 = args[3].parse().unwrap();
let salt = args[4].as_ref();
let mut msg = String::new();
stdin().read_to_string(&mut msg).unwrap();
let p = msg.as_bytes();
println!("Hash: {}", to_string(&that_cli_tool(p, salt, t, l, logm)));
}