增加提示

This commit is contained in:
lubeilin
2023-09-20 16:03:25 +08:00
parent 4422f9f8b7
commit 3429ee8bd6
2 changed files with 12 additions and 5 deletions
+11 -4
View File
@@ -214,10 +214,16 @@ fn main() {
return; return;
} }
let cipher_model = matches let cipher_model = match matches
.opt_get::<CipherModel>("model") .opt_get::<CipherModel>("model") {
.unwrap() Ok(model) => {
.unwrap_or(CipherModel::AesGcm); model.unwrap_or(CipherModel::AesGcm)
}
Err(e) => {
println!("'--model ' invalid,{}", e);
return;
}
};
let finger = matches.opt_present("finger"); let finger = matches.opt_present("finger");
let punch_model = matches let punch_model = matches
@@ -250,6 +256,7 @@ fn main() {
main0(config, !unused_cmd); main0(config, !unused_cmd);
std::process::exit(0); std::process::exit(0);
} }
#[tokio::main] #[tokio::main]
async fn main0(config: Config, show_cmd: bool) { async fn main0(config: Config, show_cmd: bool) {
let server_encrypt = config.server_encrypt; let server_encrypt = config.server_encrypt;
+1 -1
View File
@@ -28,7 +28,7 @@ impl FromStr for CipherModel {
"aes_gcm" => Ok(CipherModel::AesGcm), "aes_gcm" => Ok(CipherModel::AesGcm),
"aes_cbc" => Ok(CipherModel::AesCbc), "aes_cbc" => Ok(CipherModel::AesCbc),
"aes_ecb" => Ok(CipherModel::AesEcb), "aes_ecb" => Ok(CipherModel::AesEcb),
_ => Err(format!("not match '{}'", s)), _ => Err(format!("not match '{}', enum:aes_gcm/aes_cbc/aes_ecb", s)),
} }
} }
} }