pep/xmpp-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

happy_eyeballs: enable parallel A/AAAA resolution

Browse source
This commit is contained in:
Astro 2018-09-08 02:12:49 +02:00
parent d3fab29d38
commit 662c61b801
2 changed files with 19 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
src/client/mod.rs
View file

@ -62,7 +62,6 @@ impl Client {
.map_err(|_| Error::Idna)
.and_then(|domain|
done(Connecter::from_lookup(&domain, Some("_xmpp-client._tcp"), 5222))
.map_err(Error::Connection)
)
.flatten()
.and_then(move |tcp_stream|

25
src/happy_eyeballs.rs
View file

@ -1,4 +1,5 @@
use std::mem;
use std::io::Error as IoError;
use std::net::SocketAddr;
use std::collections::BTreeMap;
use std::collections::VecDeque;
@ -8,6 +9,8 @@ use tokio::net::{ConnectFuture, TcpStream};
use trust_dns_resolver::{IntoName, Name, ResolverFuture, error::ResolveError};
use trust_dns_resolver::lookup::SrvLookupFuture;
use trust_dns_resolver::lookup_ip::LookupIpFuture;
use trust_dns_resolver::system_conf;
use trust_dns_resolver::config::LookupIpStrategy;
use {Error, ConnecterError};
enum State {
@ -27,8 +30,14 @@ pub struct Connecter {
error: Option<Error>,
}
fn resolver_future() -> Result<Box<Future<Item = ResolverFuture, Error = ResolveError> + Send>, IoError> {
let (conf, mut opts) = system_conf::read_system_conf()?;
opts.ip_strategy = LookupIpStrategy::Ipv4AndIpv6;
Ok(ResolverFuture::new(conf, opts))
}
impl Connecter {
pub fn from_lookup(domain: &str, srv: Option<&str>, fallback_port: u16) -> Result<Connecter, ConnecterError> {
pub fn from_lookup(domain: &str, srv: Option<&str>, fallback_port: u16) -> Result<Connecter, Error> {
if let Ok(ip) = domain.parse() {
// use specified IP address, not domain name, skip the whole dns part
let connect =
@ -36,18 +45,21 @@ impl Connecter {
return Ok(Connecter {
fallback_port,
srv_domain: None,
domain: "nohost".into_name()?,
domain: "nohost".into_name()
.map_err(ConnecterError::Dns)?,
state: State::Connecting(None, vec![connect]),
targets: VecDeque::new(),
error: None,
});
}
let resolver_future = ResolverFuture::from_system_conf()?;
let state = State::AwaitResolver(resolver_future);
let state = State::AwaitResolver(resolver_future()?);
let srv_domain = match srv {
Some(srv) =>
Some(format!("{}.{}.", srv, domain).into_name()?),
Some(format!("{}.{}.", srv, domain)
.into_name()
.map_err(ConnecterError::Dns)?
),
None =>
None,
};
@ -55,7 +67,8 @@ impl Connecter {
Ok(Connecter {
fallback_port,
srv_domain,
domain: domain.into_name()?,
domain: domain.into_name()
.map_err(ConnecterError::Dns)?,
state,
targets: VecDeque::new(),
error: None,