1 // Copyright 2015-2018 Benjamin Fry <benjaminfry@me.com> 2 // 3 // Licensed under the Apache License, Version 2.0, <LICENSE-APACHE or 4 // http://apache.org/licenses/LICENSE-2.0> or the MIT license <LICENSE-MIT or 5 // http://opensource.org/licenses/MIT>, at your option. This file may not be 6 // copied, modified, or distributed except according to those terms. 7 8 #![cfg(feature = "dns-over-tls")] 9 10 mod dns_over_native_tls; 11 mod dns_over_openssl; 12 mod dns_over_rustls; 13 14 cfg_if! { 15 if #[cfg(feature = "dns-over-rustls")] { 16 pub(crate) use self::dns_over_rustls::{new_tls_stream, CLIENT_CONFIG}; 17 } else if #[cfg(feature = "dns-over-native-tls")] { 18 pub(crate) use self::dns_over_native_tls::new_tls_stream; 19 } else if #[cfg(feature = "dns-over-openssl")] { 20 pub(crate) use self::dns_over_openssl::new_tls_stream; 21 } else { 22 compile_error!("One of the dns-over-rustls, dns-over-native-tls, or dns-over-openssl must be enabled for dns-over-tls features"); 23 } 24 } 25 26 #[cfg(any(feature = "dns-over-native-tls", feature = "dns-over-rustls"))] 27 #[cfg(test)] 28 mod tests { 29 extern crate env_logger; 30 use std::net::{IpAddr, Ipv4Addr, Ipv6Addr}; 31 32 use tokio::runtime::Runtime; 33 34 use crate::config::{ResolverConfig, ResolverOpts}; 35 use crate::AsyncResolver; 36 tls_test(config: ResolverConfig)37 fn tls_test(config: ResolverConfig) { 38 //env_logger::try_init().ok(); 39 let mut io_loop = Runtime::new().unwrap(); 40 41 let (resolver, bg) = AsyncResolver::new(config, ResolverOpts::default()); 42 io_loop.spawn(bg); 43 44 let response = io_loop 45 .block_on(resolver.lookup_ip("www.example.com.")) 46 .expect("failed to run lookup"); 47 48 assert_eq!(response.iter().count(), 1); 49 for address in response.iter() { 50 if address.is_ipv4() { 51 assert_eq!(address, IpAddr::V4(Ipv4Addr::new(93, 184, 216, 34))); 52 } else { 53 assert_eq!( 54 address, 55 IpAddr::V6(Ipv6Addr::new( 56 0x2606, 0x2800, 0x220, 0x1, 0x248, 0x1893, 0x25c8, 0x1946, 57 )) 58 ); 59 } 60 } 61 } 62 63 #[test] 64 #[cfg(not(windows))] // flakes on AppVeyor... test_cloudflare_tls()65 fn test_cloudflare_tls() { 66 tls_test(ResolverConfig::cloudflare_tls()) 67 } 68 69 #[test] 70 #[cfg(not(windows))] // flakes on AppVeyor... test_quad9_tls()71 fn test_quad9_tls() { 72 tls_test(ResolverConfig::quad9_tls()) 73 } 74 } 75