include/dns.h

   1 /*
   2 dns.h - dns library declarations based on firedns Copyright (C) 2002 Ian Gulliver
   3
   4 This program is free software; you can redistribute it and/or modify
   5 it under the terms of version 2 of the GNU General Public License as
   6 published by the Free Software Foundation.
   7
   8 This program is distributed in the hope that it will be useful,
   9 but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 GNU General Public License for more details.
  12
  13 You should have received a copy of the GNU General Public License
  14 along with this program; if not, write to the Free Software
  15 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  16 */
  17
  18 #ifndef _DNS_H
  19 #define _DNS_H
  20
  21 #include <string>
  22 #include "socket.h"
  23
  24 struct dns_ip4list {
  25         in_addr ip;
  26         dns_ip4list *next;
  27 };
  28
  29
  30 /** The DNS class allows fast nonblocking resolution of hostnames
  31  * and ip addresses. It is based heavily upon firedns by Ian Gulliver.
  32  */
  33 class DNS
  34 {
  35 private:
  36         in_addr *binip;
  37         char* result;
  38         char localbuf[1024];
  39         int t;
  40         void dns_init();
  41         int myfd;
  42         void dns_init_2(const char* dnsserver);
  43         in_addr *dns_aton4(const char * const ipstring);
  44         char *dns_ntoa4(const in_addr * const ip);
  45         int dns_getip4(const char * const name);
  46         int dns_getip4list(const char * const name);
  47         int dns_getname4(const in_addr * const ip);
  48         char *dns_getresult(const int fd);
  49         in_addr *dns_aton4_s(const char * const ipstring, in_addr * const ip);
  50         char *dns_ntoa4_s(const in_addr * const ip, char * const result);
  51         char *dns_getresult_s(const int fd, char * const result);
  52         in_addr *dns_aton4_r(const char * const ipstring);
  53         char *dns_ntoa4_r(const in_addr * const ip);
  54         char *dns_getresult_r(const int fd);
  55 public:
  56         /** The default constructor uses dns addresses read from /etc/resolv.conf.
  57          * Please note that it will re-read /etc/resolv.conf for each copy of the
  58          * class you instantiate, causing disk access and slow lookups if you create
  59          * a lot of them. Consider passing the constructor a server address as a parameter
  60          * instead.
  61          */
  62         DNS();
  63         /** This constructor accepts a dns server address. The address must be in dotted
  64          * decimal form, e.g. 1.2.3.4.
  65          */
  66         DNS(const std::string &dnsserver);
  67         /** The destructor frees all used structures.
  68          */
  69         ~DNS();
  70         /** This method will start the reverse lookup of an ip given in dotted decimal
  71          * format, e.g. 1.2.3.4, and returns true if the lookup was successfully
  72          * initiated.
  73          */
  74         bool ReverseLookup(const std::string &ip);
  75         /** This method will start the forward lookup of a hostname, e.g. www.inspircd.org,
  76          * and returns true if the lookup was successfully initiated.
  77          */
  78         bool ForwardLookup(const std::string &host);
  79         /** Used by modules to perform a dns lookup but have the socket engine poll a module, instead of the dns object directly.
  80          */
  81         bool ForwardLookupWithFD(const std::string &host, int &fd);
  82         /** This method will return true when the lookup is completed. It uses poll internally
  83          * to determine the status of the socket.
  84          */
  85         bool HasResult();
  86         /** This method will return true if the lookup's fd matches the one provided
  87          */
  88         bool HasResult(int fd);
  89         /** This method returns the result of your query as a string, depending upon wether you
  90          * called DNS::ReverseLookup() or DNS::ForwardLookup.
  91          */
  92         std::string GetResult();
  93         std::string GetResultIP();
  94         /** This method returns the file handle used by the dns query socket or zero if the
  95          * query is invalid for some reason, e.g. the dns server not responding.
  96          */
  97         int GetFD();
  98         void SetNS(const std::string &dnsserver);
  99 };
 100
 101 /** This is the handler function for multi-threaded DNS.
 102  * It cannot be a class member as pthread will not let us
 103  * create a thread whos handler function is a member of
 104  * a class (ugh).
 105  */
 106 void* dns_task(void* arg);
 107
 108 #endif