feat: optimize decrypt
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user