tsndr/sync-ssh-keys
1
1
Fork 0
Code Pull Requests Activity

fix exception if host isn't resolvable

Browse Source
This commit is contained in:
Toby
2022-10-26 16:25:37 +02:00
parent 09096d2769
commit ee1d6346ba
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/main.rs
View File

@@ -25,14 +25,16 @@ fn upload_config(h: &Host) -> Result<String, String> {
let content = h.content.lock().unwrap();
let connection_string = format!("{}@{}", username, host);
let addr: SocketAddr = format!("{}:{}", host, port)
.to_socket_addrs()
.unwrap().nth(0)
.expect(format!("Invalid host/port in `{}:{}`", host, port).as_str());
let tcp: TcpStream = match TcpStream::connect_timeout(&addr, Duration::from_secs(1)) {
let addr: SocketAddr = match format!("{}:{}", host, port).to_socket_addrs() {
Ok(mut addrs) => addrs.nth(0).expect(format!("Invalid host/port in `{}:{}`", host, port).as_str()),
_ => return Err(connection_string)
};
let tcp: TcpStream = match TcpStream::connect_timeout(&addr, Duration::from_secs(3)) {
Ok(s) => s,
Err(_) => return Err(connection_string)
};
let mut sess = Session::new().unwrap();
sess.set_tcp_stream(tcp);