fn print_usage() {
println!("Usage:");
- println!(" {} [--help] [--encrypt <message>]", get_exe_name());
+ println!(" {} [--help] [--encrypt <plain-text>|--decrypt <cipher-text>]", get_exe_name());
}
fn read_key() -> String {
- use url::percent_encoding::percent_decode;
let mut key = String::new();
File::open(consts::FILE_KEY)
.expect(&format!("Failed to open key file: {}", consts::FILE_KEY))
.expect(&format!("Failed to read key file: {}", consts::FILE_KEY));
String::from(
- percent_decode(key.as_bytes())
+ url::percent_encoding::percent_decode(key.as_bytes())
.decode_utf8()
.expect(&format!("Failed to decode key file: {}", consts::FILE_KEY))
)
} else if let Some((position_arg_encrypt, _)) = args.iter().find_position(|arg| arg == &"--encrypt") {
match args.iter().nth(position_arg_encrypt + 1) {
Some(mess_to_encrypt) => {
+
let encrypted_mess = crypto::encrypt(&key, mess_to_encrypt);
- println!("Encrypted message: {}", encrypted_mess);
+ let encrypted_mess_encoded = url::percent_encoding::utf8_percent_encode(&encrypted_mess, url::percent_encoding::DEFAULT_ENCODE_SET).to_string();
+ println!("Encrypted message percent-encoded: {}", encrypted_mess_encoded);
+ }
+ None => print_usage()
+ }
+
+ return Ok(());
+ } else if let Some((position_arg_decrypt, _)) = args.iter().find_position(|arg| arg == &"--decrypt") {
+ match args.iter().nth(position_arg_decrypt + 1) {
+ Some(cipher_text) => {
+ let cipher_text_decoded = url::percent_encoding::percent_decode(cipher_text.as_bytes()).decode_utf8().expect("Unable to decode encoded cipher text");
+ let plain_text = crypto::decrypt(&key, &cipher_text_decoded).unwrap();
+ println!("Decrypted message: {}", plain_text);
}
None => print_usage()
}