file-crypto 0.3.0

A high performace command line tool for encrypting / decrypting any file
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

pub mod crypto;
pub mod file;
pub mod ctrl;

use file_crypto::crypto::*;
use file_crypto::ctrl::*;
use file_crypto::*;

extern crate clap;
use clap::{App, Arg};
fn main() {
     let matches = App::new("File Crypto")
          .version("0.2.0")
          .author("William Huang <william.hng@outlook.com>")
          .about("file-crypto is a platform-cross command line tool for fastly encrypting / decrypting any file with AES-256-GCM")
          .arg(Arg::with_name("encrypt")
               .short("e")
               .long("encrypt")
               .help("To encrypt the file"))
          .arg(Arg::with_name("decrypt")
               .short("d")
               .long("decrypt")
               .help("To decrypt the file"))
          .arg(Arg::with_name("key")
               .short("k")
               .long("key")
               .value_name("KEY")
               .help("The key for encrypt/decrypt the file")
               .takes_value(true))
          .arg(Arg::with_name("FILE")
               .help("The path for the target file")
               .required(true)
               .index(1))
          .get_matches();

     let path = matches.value_of("FILE").unwrap();
     let encrypt_mode = (matches.is_present("encrypt") as u8) << 1;
     let decrypt_mode = matches.is_present("decrypt") as u8;
     let meta = match encrypt_mode | decrypt_mode {
          0b01 => CipherCtrl::init_with_type(path, ProcessType::Decrypt),
          0b10 => CipherCtrl::init_with_type(path, ProcessType::Encrypt),
          0b11 => panic!("Cannot set encrypt-mode and decrypt-mode at the same time"),
          _ => CipherCtrl::init(path),
     };

     let key = match matches.value_of("key") {
          Some(s) => {
               Key::from(s)
          }
          None => {
               let k = Key::new();
               println!(
                    "The key is: {}\n (Please keep the key in the safe way.)",
                    k.base64()
               );
               k
          }
     };
     match meta.proc_type {
          ProcessType::Encrypt => println!("The encrypted file is at: {}", encrypt(&key, &meta)),
          ProcessType::Decrypt => println!("The decrypted file is at {}", decrypt(&key, &meta)),
     };
}