hatter/tiny-encrypt-rs
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat: optimize decrypt

Browse Source
This commit is contained in:
Hatter Jiang
2023-10-17 00:06:23 +08:00
parent e0b6df7c46
commit 17fae72d91
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/cmd_decrypt.rs
View File

@@ -124,16 +124,17 @@ pub fn decrypt_single(config: &Option<TinyEncryptConfig>,
if cmd_decrypt.direct_print { if cmd_decrypt.direct_print {
if meta.file_length > 10 * 1024 { if meta.file_length > 10 * 1024 {
return simple_error!("File too large(more than 10K) cannot direct print on console."); warning!("File too large(more than 10K) cannot direct print on console.");
} } else {
let mut output: Vec<u8> = Vec::with_capacity(10 * 1024); let mut output: Vec<u8> = Vec::with_capacity(10 * 1024);
let _ = decrypt_file( let _ = decrypt_file(
&mut file_in, meta.file_length, &mut output, &key, &nonce, meta.compress, &mut file_in, meta.file_length, &mut output, &key, &nonce, meta.compress,
)?; )?;
match String::from_utf8(output) { match String::from_utf8(output) {
Err(_) => return simple_error!("File is not UTF-8 content."), Err(_) => warning!("File is not UTF-8 content."),
Ok(output) => println!(">>>>> BEGIN CONTENT >>>>>\n{}\n<<<<< END CONTENT <<<<<", &output), Ok(output) => println!(">>>>> BEGIN CONTENT >>>>>\n{}\n<<<<< END CONTENT <<<<<", &output),
} }
}
} else { } else {
let mut file_out = File::create(path_out)?; let mut file_out = File::create(path_out)?;
let _ = decrypt_file( let _ = decrypt_file(