DNS.hpp

   1 #ifndef DNS_DOT_HPP
   2 #define DNS_DOT_HPP
   3
   4 #include <memory>
   5
   6 #include "DNS-message.hpp"
   7 #include "DNS-rrs.hpp"
   8 #include "Sock.hpp"
   9 #include "iobuffer.hpp"
  10
  11 #include <glog/logging.h>
  12
  13 namespace DNS {
  14
  15 class Resolver {
  16 public:
  17   Resolver(Resolver const&) = delete;
  18   Resolver& operator=(Resolver const&) = delete;
  19
  20   Resolver(fs::path config_path);
  21
  22   RR_collection get_records(RR_type typ, char const* name);
  23   RR_collection get_records(RR_type typ, std::string const& name)
  24   {
  25     return get_records(typ, name.c_str());
  26   }
  27
  28   std::vector<std::string> get_strings(RR_type typ, char const* name);
  29   std::vector<std::string> get_strings(RR_type typ, std::string const& name)
  30   {
  31     return get_strings(typ, name.c_str());
  32   }
  33
  34   message xchg(message const& q);
  35
  36 private:
  37   std::unique_ptr<Sock> ns_sock_;
  38   int                   ns_;
  39   int                   ns_fd_;
  40 };
  41
  42 class Query {
  43 public:
  44   Query(Query const&) = delete;
  45   Query& operator=(Query const&) = delete;
  46
  47   Query(Resolver& res, RR_type type, char const* name);
  48   Query(Resolver& res, RR_type type, std::string const& name)
  49     : Query(res, type, name.c_str())
  50   {
  51   }
  52
  53   bool authentic_data() const { return authentic_data_; }
  54   bool bogus_or_indeterminate() const { return bogus_or_indeterminate_; }
  55   bool nx_domain() const { return nx_domain_; }
  56
  57   bool                     has_record() const { return has_record_; }
  58   RR_collection            get_records();
  59   std::vector<std::string> get_strings();
  60
  61   uint16_t rcode() const { return rcode_; }
  62   uint16_t extended_rcode() const { return extended_rcode_; }
  63
  64 private:
  65   bool xchg_(Resolver& res, uint16_t id);
  66   void check_answer_(Resolver& res, RR_type type, char const* name);
  67
  68   uint16_t rcode_{0};
  69   uint16_t extended_rcode_{0};
  70
  71   RR_type type_;
  72
  73   message q_;
  74   message a_;
  75
  76   bool authentic_data_{false};
  77   bool bogus_or_indeterminate_{false};
  78   bool nx_domain_{false};
  79   bool has_record_{false};
  80 };
  81
  82 inline std::vector<std::string>
  83 get_strings(Resolver& res, RR_type type, char const* name)
  84 {
  85   return res.get_strings(type, name);
  86 }
  87
  88 inline std::vector<std::string>
  89 get_strings(Resolver& res, RR_type type, std::string const& name)
  90 {
  91   return res.get_strings(type, name.c_str());
  92 }
  93
  94 inline bool has_record(Resolver& res, RR_type type, char const* name)
  95 {
  96   Query q(res, type, name);
  97   return q.has_record();
  98 }
  99
 100 inline bool has_record(Resolver& res, RR_type type, std::string const& name)
 101 {
 102   return has_record(res, type, name.c_str());
 103 }
 104 } // namespace DNS
 105
 106 #endif // DNS_DOT_HPP