From 14a6bc6e3dad6511d12f65b40132e5c38e37e25e Mon Sep 17 00:00:00 2001 From: Toby Date: Mon, 19 Aug 2024 00:50:29 +0200 Subject: [PATCH] catch connect error --- src/main.rs | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/main.rs b/src/main.rs index 7b45e24..663a557 100644 --- a/src/main.rs +++ b/src/main.rs @@ -40,7 +40,10 @@ fn upload_config(h: &Host) -> Result { sess.set_tcp_stream(tcp); sess.handshake().unwrap(); - sess.userauth_pubkey_file(&username, None, &private_key, None).unwrap(); + match sess.userauth_pubkey_file(&username, None, &private_key, None) { + Err(_) => return Err(connection_string), + _ => {} + } let mut remote_file = match sess.scp_send(Path::new(".ssh").join("authorized_keys2").as_path(), 0o644, content.len() as u64, None) { Ok(rf) => rf,