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