feat: optimize decrypt

This commit is contained in:
2023-10-17 00:06:23 +08:00
parent e0b6df7c46
commit 17fae72d91

View File

@@ -124,16 +124,17 @@ pub fn decrypt_single(config: &Option<TinyEncryptConfig>,
if cmd_decrypt.direct_print {
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 _ = decrypt_file(
&mut file_in, meta.file_length, &mut output, &key, &nonce, meta.compress,
)?;
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),
}
}
} else {
let mut file_out = File::create(path_out)?;
let _ = decrypt_file(