diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/connection.rs | 29 |
1 files changed, 22 insertions, 7 deletions
diff --git a/src/connection.rs b/src/connection.rs index 24f7745..ccc2ae7 100644 --- a/src/connection.rs +++ b/src/connection.rs @@ -6,6 +6,7 @@ use tokio::net::TcpStream; use tokio_native_tls::native_tls::TlsConnector; // TODO: use rustls use tokio_native_tls::TlsStream; +use tracing::{debug, info, instrument, trace}; use crate::Jabber; use crate::JabberError; @@ -34,12 +35,16 @@ impl Connection { // Self::connect(&server, jid.try_into()?, Some(auth)).await // } + #[instrument] async fn connect(server: &str) -> Result<Self> { + info!("connecting to {}", server); let sockets = Self::get_sockets(&server).await; + debug!("discovered sockets: {:?}", sockets); for (socket_addr, tls) in sockets { match tls { true => { if let Ok(connection) = Self::connect_tls(socket_addr, &server).await { + info!("connected via encrypted stream to {}", socket_addr); let (readhalf, writehalf) = tokio::io::split(connection); return Ok(Self::Encrypted(Jabber::new( readhalf, @@ -52,6 +57,7 @@ impl Connection { } false => { if let Ok(connection) = Self::connect_unencrypted(socket_addr).await { + info!("connected via unencrypted stream to {}", socket_addr); let (readhalf, writehalf) = tokio::io::split(connection); return Ok(Self::Unencrypted(Jabber::new( readhalf, @@ -67,13 +73,16 @@ impl Connection { Err(JabberError::Connection) } - async fn get_sockets(domain: &str) -> Vec<(SocketAddr, bool)> { + #[instrument] + async fn get_sockets(address: &str) -> Vec<(SocketAddr, bool)> { let mut socket_addrs = Vec::new(); // if it's a socket/ip then just return that // socket - if let Ok(socket_addr) = SocketAddr::from_str(domain) { + trace!("checking if address is a socket address"); + if let Ok(socket_addr) = SocketAddr::from_str(address) { + debug!("{} is a socket address", address); match socket_addr.port() { 5223 => socket_addrs.push((socket_addr, true)), _ => socket_addrs.push((socket_addr, false)), @@ -82,16 +91,19 @@ impl Connection { return socket_addrs; } // ip - if let Ok(ip) = IpAddr::from_str(domain) { + trace!("checking if address is an ip"); + if let Ok(ip) = IpAddr::from_str(address) { + debug!("{} is an ip", address); socket_addrs.push((SocketAddr::new(ip, 5222), false)); socket_addrs.push((SocketAddr::new(ip, 5223), true)); return socket_addrs; } // otherwise resolve + debug!("resolving {}", address); if let Ok(resolver) = trust_dns_resolver::AsyncResolver::tokio_from_system_conf() { if let Ok(lookup) = resolver - .srv_lookup(format!("_xmpp-client._tcp.{}", domain)) + .srv_lookup(format!("_xmpp-client._tcp.{}", address)) .await { for srv in lookup { @@ -106,7 +118,7 @@ impl Connection { } } if let Ok(lookup) = resolver - .srv_lookup(format!("_xmpps-client._tcp.{}", domain)) + .srv_lookup(format!("_xmpps-client._tcp.{}", address)) .await { for srv in lookup { @@ -122,7 +134,7 @@ impl Connection { } // in case cannot connect through SRV records - resolver.lookup_ip(domain).await.map(|ips| { + resolver.lookup_ip(address).await.map(|ips| { for ip in ips { socket_addrs.push((SocketAddr::new(ip, 5222), false)); socket_addrs.push((SocketAddr::new(ip, 5223), true)); @@ -133,6 +145,7 @@ impl Connection { } /// establishes a connection to the server + #[instrument] pub async fn connect_tls(socket_addr: SocketAddr, domain_name: &str) -> Result<Tls> { let socket = TcpStream::connect(socket_addr) .await @@ -144,6 +157,7 @@ impl Connection { .map_err(|_| JabberError::Connection) } + #[instrument] pub async fn connect_unencrypted(socket_addr: SocketAddr) -> Result<Unencrypted> { TcpStream::connect(socket_addr) .await @@ -154,8 +168,9 @@ impl Connection { #[cfg(test)] mod tests { use super::*; + use test_log::test; - #[tokio::test] + #[test(tokio::test)] async fn connect() { Connection::connect("blos.sm").await.unwrap(); } |