jnk/dns2.cpp

   1 // brain dead command line tool to query DNS
   2
   3 #include "DNS.hpp"
   4 #include "IP4.hpp"
   5
   6 #include <glog/logging.h>
   7
   8 void do_dotted_quad(char const* addr)
   9 {
  10   DNS::Resolver res;
  11
  12   std::string reversed{IP4::reverse(addr)};
  13
  14   constexpr const char* const rbls[]
  15       = {"zen.spamhaus.org", "b.barracudacentral.org"};
  16
  17   for (auto rbl : rbls) {
  18     std::string lookup = reversed + rbl;
  19
  20     DNS::Domain dom(lookup.c_str());
  21     DNS::Query<DNS::RR_type::A> q(res, dom);
  22
  23     if (q.get_rcode() == DNS::Pkt_rcode::NOERROR) {
  24       DNS::Rrlist<DNS::RR_type::A> rrlst(q);
  25
  26       if (!rrlst.empty()) {
  27         std::cout << "found in " << rbl << '\n';
  28         std::vector<std::string> codes = rrlst.get();
  29         for (auto code : codes) {
  30           std::cout << code << '\n';
  31         }
  32       }
  33     }
  34     else {
  35       if (q.get_rcode() == DNS::Pkt_rcode::NXDOMAIN) {
  36         std::cout << "not found in " << rbl << '\n';
  37       }
  38       else {
  39         std::cout << "Error from lookup at " << rbl << " " << q.get_rcode()
  40                   << '\n';
  41       }
  42     }
  43   }
  44
  45   std::string fcrdns;
  46
  47   auto ptrs
  48       = DNS::get_records<DNS::RR_type::PTR>(res, reversed + "in-addr.arpa");
  49
  50   for (auto ptr : ptrs) {
  51     // chop off the trailing '.'
  52     auto length = ptr.length();
  53     if (length && ('.' == ptr.at(length - 1))) {
  54       ptr.erase(length - 1, 1);
  55     }
  56     auto addrs = DNS::get_records<DNS::RR_type::A>(res, ptr);
  57     for (auto const& a : addrs) {
  58       if (a == addr) {
  59         fcrdns = ptr;
  60         goto found;
  61       }
  62     }
  63   }
  64
  65   std::cout << "no fcrdns\n";
  66   return;
  67
  68 found:
  69   std::cout << fcrdns << '\n';
  70
  71   auto txts = DNS::get_records<DNS::RR_type::TXT>(res, fcrdns);
  72   for (auto const& txt : txts) {
  73     std::cout << "\"" << txt << "\"\n";
  74   }
  75 }
  76
  77 void do_domain(char const* domain)
  78 {
  79   DNS::Resolver res;
  80
  81   auto addrs = DNS::get_records<DNS::RR_type::A>(res, domain);
  82   for (auto const& a : addrs) {
  83     std::cout << a << '\n';
  84   }
  85
  86   auto txts = DNS::get_records<DNS::RR_type::TXT>(res, domain);
  87   for (auto const& txt : txts) {
  88     std::cout << "\"" << txt << "\"\n";
  89   }
  90 }
  91
  92 int main(int argc, char const* argv[])
  93 {
  94   google::InitGoogleLogging(argv[0]);
  95
  96   for (int i = 1; i < argc; ++i) {
  97     if (IP4::is_address(argv[i])) {
  98       do_dotted_quad(argv[i]);
  99     }
 100     else {
 101       do_domain(argv[i]);
 102     }
 103   }
 104 }