arkorty/rustcm-cli
1
0
Fork 0
mirror of https://github.com/arkorty/rustcm-cli.git synced 2026-03-18 00:57:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update src/main.rs

Browse Source
This commit is contained in:
Arkaprabha Chakraborty
2023-01-30 06:53:04 +05:30
parent fd76099d8d
commit df1b25740a
1 changed files with 28 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
src/main.rs
View File

@@ -20,7 +20,7 @@ pub fn get_password(prompt: &str) -> String {
match stdout().flush() { match stdout().flush() {
Ok(_) => (), Ok(_) => (),
Err(_) => { Err(_) => {
eprintln!("Error: Could not flush stdout"); eprintln!("Error: Could not flush date to stdout");
exit(0); exit(0);
} }
}; };
@@ -54,14 +54,14 @@ pub fn get_secret_key(salt_bytes: [u8; 32], password: String) -> orion::kdf::Sec
let password = match kdf::Password::from_slice(password.as_bytes()) { let password = match kdf::Password::from_slice(password.as_bytes()) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not generate the password"); eprintln!("Error: Could not create the password");
exit(0) exit(0)
} }
}; };
let salt = match kdf::Salt::from_slice(&salt_bytes) { let salt = match kdf::Salt::from_slice(&salt_bytes) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not generate the salt"); eprintln!("Error: Could not create the salt");
exit(0) exit(0)
} }
}; };
@@ -81,7 +81,7 @@ pub fn get_salt_bytes() -> [u8; 32] {
match orion::util::secure_rand_bytes(&mut salt_bytes) { match orion::util::secure_rand_bytes(&mut salt_bytes) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not generate the salt"); eprintln!("Error: Could not generate the random bytes for the salt");
exit(0) exit(0)
} }
}; };
@@ -102,10 +102,10 @@ pub fn get_salt(salt_bytes: [u8; 32]) -> orion::kdf::Salt {
} }
pub fn write_plain(path: String, plaintext: String) { pub fn write_plain(path: String, plaintext: String) {
let mut file = match File::create(path) { let mut file = match File::create(&path) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not create the file"); eprintln!("Error: Could not create {path}");
exit(0); exit(0);
} }
}; };
@@ -113,7 +113,7 @@ pub fn write_plain(path: String, plaintext: String) {
match file.write(&plaintext.into_bytes()) { match file.write(&plaintext.into_bytes()) {
Ok(_) => (), Ok(_) => (),
Err(_) => { Err(_) => {
eprintln!("Error: Could not write the data to the file"); eprintln!("Error: Could not write the data to {path}");
exit(0); exit(0);
} }
}; };
@@ -121,17 +121,17 @@ pub fn write_plain(path: String, plaintext: String) {
match file.flush() { match file.flush() {
Ok(_) => (), Ok(_) => (),
Err(_) => { Err(_) => {
eprintln!("Error: Could not flush the file"); eprintln!("Error: Could not flush data to {path}");
exit(0); exit(0);
} }
}; };
} }
pub fn write_cipher(path: String, salt_bytes: [u8; 32], ciphertext: Vec<u8>) { pub fn write_cipher(path: String, salt_bytes: [u8; 32], ciphertext: Vec<u8>) {
let mut file = match File::create(path) { let mut file = match File::create(&path) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not create the file"); eprintln!("Error: Could not create {path}");
exit(0); exit(0);
} }
}; };
@@ -139,7 +139,7 @@ pub fn write_cipher(path: String, salt_bytes: [u8; 32], ciphertext: Vec<u8>) {
match file.write(&salt_bytes) { match file.write(&salt_bytes) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could write the salt_bytes to the file"); eprintln!("Error: Could write the salt to {path}");
exit(0); exit(0);
} }
}; };
@@ -147,7 +147,7 @@ pub fn write_cipher(path: String, salt_bytes: [u8; 32], ciphertext: Vec<u8>) {
match file.write(&ciphertext) { match file.write(&ciphertext) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not write the ciphertext to the file"); eprintln!("Error: Could not write the ciphertext to {path}");
exit(0); exit(0);
} }
}; };
@@ -155,17 +155,17 @@ pub fn write_cipher(path: String, salt_bytes: [u8; 32], ciphertext: Vec<u8>) {
match file.flush() { match file.flush() {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not flush the file"); eprintln!("Error: Could not flush data to {path}");
exit(0); exit(0);
} }
}; };
} }
pub fn read_plain(path: String) -> String { pub fn read_plain(path: String) -> String {
let file_str: String = match read_to_string(path) { let file_str: String = match read_to_string(&path) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not read the file"); eprintln!("Error: Could not read the {path}");
exit(0); exit(0);
} }
}; };
@@ -174,10 +174,10 @@ pub fn read_plain(path: String) -> String {
} }
pub fn read_cipher(path: String) -> (Vec<u8>, Vec<u8>) { pub fn read_cipher(path: String) -> (Vec<u8>, Vec<u8>) {
let mut file = match File::open(path) { let mut file = match File::open(&path) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not open the file"); eprintln!("Error: Could not open {path}");
exit(0); exit(0);
} }
}; };
@@ -185,7 +185,7 @@ pub fn read_cipher(path: String) -> (Vec<u8>, Vec<u8>) {
let metadata = match File::metadata(&file) { let metadata = match File::metadata(&file) {
Ok(temp) => temp, Ok(temp) => temp,
Err(_) => { Err(_) => {
eprintln!("Error: Could not read the metadata off the file"); eprintln!("Error: Could not read the metadata off of {path}");
exit(0); exit(0);
} }
}; };
@@ -194,7 +194,7 @@ pub fn read_cipher(path: String) -> (Vec<u8>, Vec<u8>) {
match file.read(&mut data) { match file.read(&mut data) {
Ok(_) => (), Ok(_) => (),
Err(_) => { Err(_) => {
eprintln!("Error: Could not read the file"); eprintln!("Error: Could not read {path}");
exit(0); exit(0);
} }
}; };
@@ -210,7 +210,10 @@ pub fn read_cipher(path: String) -> (Vec<u8>, Vec<u8>) {
pub fn decrypt(ciphertext: Vec<u8>, secret_key: orion::kdf::SecretKey) -> String { pub fn decrypt(ciphertext: Vec<u8>, secret_key: orion::kdf::SecretKey) -> String {
let plaintext = match aead::open(&secret_key, &ciphertext) { let plaintext = match aead::open(&secret_key, &ciphertext) {
Ok(text) => text, Ok(text) => text,
Err(_) => panic!("Error: Could not decrypt the file"), Err(_) => {
eprintln!("Error: Failed to decrypt the file, please check the password");
exit(0);
}
}; };
let plaintext = String::from_utf8(plaintext).expect("Error: Could not convert to String"); let plaintext = String::from_utf8(plaintext).expect("Error: Could not convert to String");
@@ -218,12 +221,9 @@ pub fn decrypt(ciphertext: Vec<u8>, secret_key: orion::kdf::SecretKey) -> String
plaintext plaintext
} }
pub fn convert_to_array<T>(v: Vec<T>) -> [T; 32] pub fn to_array(v: Vec<u8>) -> [u8; 32] {
where
T: Copy,
{
let slice = v.as_slice(); let slice = v.as_slice();
let array: [T; 32] = match slice.try_into() { let array: [u8; 32] = match slice.try_into() {
Ok(bytes) => bytes, Ok(bytes) => bytes,
Err(_) => { Err(_) => {
eprintln!( eprintln!(
@@ -300,9 +300,9 @@ Written by Arkaprabha Chakraborty"
} }
}; };
let password: String = get_password("Password: ");
let plaintext: String = read_plain(path); let plaintext: String = read_plain(path);
let salt_bytes = get_salt_bytes(); let salt_bytes = get_salt_bytes();
let password: String = get_password("Password: ");
let secret_key = get_secret_key(salt_bytes, password); let secret_key = get_secret_key(salt_bytes, password);
let ciphertext = encrypt(plaintext, secret_key); let ciphertext = encrypt(plaintext, secret_key);
@@ -327,9 +327,9 @@ Written by Arkaprabha Chakraborty"
} }
}; };
let password: String = get_password("Password: ");
let (salt_bytes, ciphertext) = read_cipher(path); let (salt_bytes, ciphertext) = read_cipher(path);
let salt_bytes: [u8; 32] = convert_to_array(salt_bytes); let salt_bytes: [u8; 32] = to_array(salt_bytes);
let password: String = get_password("Password: ");
let secret_key = get_secret_key(salt_bytes, password); let secret_key = get_secret_key(salt_bytes, password);
let plaintext = decrypt(ciphertext, secret_key); let plaintext = decrypt(ciphertext, secret_key);