feat: support CNAME record

Digs now able to query CNAME record
BiznetGIO · Feb 22, 2021 · 4ce3318 · 4ce3318
1 parent 2b39738
commit 4ce3318
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/dns.rs b/src/dns.rs
@@ -5,11 +5,12 @@ use trust_dns_client::op::DnsResponse;
 use trust_dns_client::rr::{DNSClass, Name, RecordType};
 use trust_dns_client::udp::UdpClientConnection;
 
-// Record Types
+/// Record Types
 #[derive(Debug, Clone, Copy)]
 pub enum RTypes {
     A,
     AAAA,
+    CNAME,
     MX,
     NS,
     SOA,
@@ -23,6 +24,7 @@ impl FromStr for RTypes {
         match s {
             "A" => Ok(RTypes::A),
             "AAAA" => Ok(RTypes::AAAA),
+            "CNAME" => Ok(RTypes::CNAME),
             "MX" => Ok(RTypes::MX),
             "NS" => Ok(RTypes::NS),
             "SOA" => Ok(RTypes::SOA),
@@ -36,13 +38,15 @@ fn get_rtype(rtype: RTypes) -> RecordType {
     match rtype {
         RTypes::A => RecordType::A,
         RTypes::AAAA => RecordType::AAAA,
+        RTypes::CNAME => RecordType::CNAME,
         RTypes::MX => RecordType::MX,
         RTypes::NS => RecordType::NS,
         RTypes::SOA => RecordType::SOA,
         RTypes::TXT => RecordType::TXT,
     }
 }
-// Parse address string
+
+/// Parse address string
 fn get_address(nameserver: &str) -> std::net::SocketAddr {
     let address = format!("{}:53", nameserver).parse::<std::net::SocketAddr>();
     match address {

diff --git a/src/main.rs b/src/main.rs
@@ -25,7 +25,7 @@ fn main() {
         )
         .arg(
             Arg::new("rtype")
-                .possible_values(&["A", "AAAA", "MX", "NS", "SOA", "TXT"])
+                .possible_values(&["A", "AAAA", "CNAME", "MX", "NS", "SOA", "TXT"])
                 .default_value("A"),
         )
         .arg(