Make the comment make sense ;<

[user/henk/code/inspircd.git] / src / dns.cpp

diff --git a/src/dns.cpp b/src/dns.cpp
index 123901f8ace3e09f621c590876495c189ddd48fc..dc25fe8fee417c6d0be5be4837369ef077f1bc23 100644 (file)
--- a/src/dns.cpp
+++ b/src/dns.cpp
@@ -162,7 +162,8 @@ inline void dns_empty_header(unsigned char *output, const s_header *header, cons
 void dns_close(int fd)
 {
 #ifndef THREADED_DNS
-        ServerInstance->SE->DelFd(fd);
+       if (ServerInstance && ServerInstance->SE)
+               ServerInstance->SE->DelFd(fd);
 #endif
        log(DEBUG,"DNS: dns_close on fd %d",fd);
        if (fd == lastcreate)
@@ -669,7 +670,6 @@ char* DNS::dns_getresult_s(const int cfd, char *res) { /* retrieve result of DNS
                        res[rr.rdlength] = '\0';
                        break;
                default:
-               defaultcase:
                        memcpy(res,&h.payload[i],rr.rdlength);
                        res[rr.rdlength] = '\0';
                        break;
@@ -684,13 +684,13 @@ DNS::DNS()
        log(DEBUG,"Create blank DNS");
 }
 
-DNS::DNS(std::string dnsserver)
+DNS::DNS(const std::string &dnsserver)
 {
        dns_init_2(dnsserver.c_str());
-       log(DEBUG,"Create DNS");
+       log(DEBUG,"Create DNS with server '%s'",dnsserver.c_str());
 }
 
-void DNS::SetNS(std::string dnsserver)
+void DNS::SetNS(const std::string &dnsserver)
 {
        dns_init_2(dnsserver.c_str());
        log(DEBUG,"Set NS");
@@ -700,9 +700,10 @@ DNS::~DNS()
 {
 }
 
-bool DNS::ReverseLookup(std::string ip)
+bool DNS::ReverseLookup(const std::string &ip)
 {
-       ServerInstance->stats->statsDns++;
+       if (ServerInstance && ServerInstance->stats)
+               ServerInstance->stats->statsDns++;
         binip = dns_aton4(ip.c_str());
         if (binip == NULL)
        {
@@ -716,14 +717,16 @@ bool DNS::ReverseLookup(std::string ip)
        }
        log(DEBUG,"DNS: ReverseLookup, fd=%d",this->myfd);
 #ifndef THREADED_DNS
-       ServerInstance->SE->AddFd(this->myfd,true,X_ESTAB_DNS);
+       if (ServerInstance && ServerInstance->SE)
+               ServerInstance->SE->AddFd(this->myfd,true,X_ESTAB_DNS);
 #endif
        return true;
 }
 
-bool DNS::ForwardLookup(std::string host)
+bool DNS::ForwardLookup(const std::string &host)
 {
-       ServerInstance->stats->statsDns++;
+       if (ServerInstance && ServerInstance->stats)
+               ServerInstance->stats->statsDns++;
        this->myfd = dns_getip4(host.c_str());
        if (this->myfd == -1)
        {
@@ -731,11 +734,28 @@ bool DNS::ForwardLookup(std::string host)
        }
        log(DEBUG,"DNS: ForwardLookup, fd=%d",this->myfd);
 #ifndef THREADED_DNS
-       ServerInstance->SE->AddFd(this->myfd,true,X_ESTAB_DNS);
+       if (ServerInstance && ServerInstance->SE)
+               ServerInstance->SE->AddFd(this->myfd,true,X_ESTAB_DNS);
 #endif
        return true;
 }
 
+bool DNS::ForwardLookupWithFD(const std::string &host, int &fd)
+{
+       if (ServerInstance && ServerInstance->stats)
+               ServerInstance->stats->statsDns++;
+       this->myfd = dns_getip4(host.c_str());
+       fd = this->myfd;
+       if (this->myfd == -1)
+       {
+               
+       }
+       log(DEBUG,"DNS: ForwardLookupWithFD, fd=%d",this->myfd);
+       if (ServerInstance && ServerInstance->SE)
+               ServerInstance->SE->AddFd(this->myfd,true,X_ESTAB_MODULE);
+       return true;
+}
+
 bool DNS::HasResult(int fd)
 {
        return (fd == this->myfd);
@@ -767,13 +787,15 @@ std::string DNS::GetResult()
         result = dns_getresult(this->myfd);
         if (result)
        {
-               ServerInstance->stats->statsDnsGood++;
+               if (ServerInstance && ServerInstance->stats)
+                       ServerInstance->stats->statsDnsGood++;
                dns_close(this->myfd);
                return result;
         }
        else
        {
-               ServerInstance->stats->statsDnsBad++;
+               if (ServerInstance && ServerInstance->stats)
+                       ServerInstance->stats->statsDnsBad++;
                if (this->myfd != -1)
                {
                        dns_close(this->myfd);
@@ -793,7 +815,8 @@ std::string DNS::GetResultIP()
        }
        if (result)
        {
-               ServerInstance->stats->statsDnsGood++;
+               if (ServerInstance && ServerInstance->stats)
+                       ServerInstance->stats->statsDnsGood++;
                unsigned char a = (unsigned)result[0];
                unsigned char b = (unsigned)result[1];
                unsigned char c = (unsigned)result[2];
@@ -803,7 +826,8 @@ std::string DNS::GetResultIP()
        }
        else
        {
-               ServerInstance->stats->statsDnsBad++;
+               if (ServerInstance && ServerInstance->stats)
+                       ServerInstance->stats->statsDnsBad++;
                log(DEBUG,"DANGER WILL ROBINSON! NXDOMAIN for forward lookup, but we got a reverse lookup!");
                return "";
        }
@@ -820,7 +844,7 @@ void* dns_task(void* arg)
         DNS dns2;
         std::string host;
         std::string ip;
-        if (dns1.ReverseLookup(u->ip))
+        if (dns1.ReverseLookup((char*)inet_ntoa(u->ip4)))
         {
                 while (!dns1.HasResult())
                 {
@@ -836,7 +860,7 @@ void* dns_task(void* arg)
                                         usleep(100);
                                 }
                                 ip = dns2.GetResultIP();
-                                if (ip == std::string(u->ip))
+                                if (ip == std::string((char*)inet_ntoa(u->ip4)))
                                 {
                                         if (host.length() < 160)
                                         {