Make the comment make sense ;<

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

diff --git a/src/dns.cpp b/src/dns.cpp
index 67c309f4108663beca2d50078170e11f2033a5e0..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)
@@ -683,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");
@@ -699,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)
        {
@@ -715,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)
        {
@@ -730,14 +734,16 @@ 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(std::string host, int &fd)
+bool DNS::ForwardLookupWithFD(const std::string &host, int &fd)
 {
-       ServerInstance->stats->statsDns++;
+       if (ServerInstance && ServerInstance->stats)
+               ServerInstance->stats->statsDns++;
        this->myfd = dns_getip4(host.c_str());
        fd = this->myfd;
        if (this->myfd == -1)
@@ -745,7 +751,8 @@ bool DNS::ForwardLookupWithFD(std::string host, int &fd)
                
        }
        log(DEBUG,"DNS: ForwardLookupWithFD, fd=%d",this->myfd);
-       ServerInstance->SE->AddFd(this->myfd,true,X_ESTAB_MODULE);
+       if (ServerInstance && ServerInstance->SE)
+               ServerInstance->SE->AddFd(this->myfd,true,X_ESTAB_MODULE);
        return true;
 }
 
@@ -780,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);
@@ -806,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];
@@ -816,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 "";
        }
@@ -833,7 +844,7 @@ void* dns_task(void* arg)
         DNS dns2;
         std::string host;
         std::string ip;
-        if (dns1.ReverseLookup((char*)inet_ntoa(dest->ip4)))
+        if (dns1.ReverseLookup((char*)inet_ntoa(u->ip4)))
         {
                 while (!dns1.HasResult())
                 {
@@ -849,7 +860,7 @@ void* dns_task(void* arg)
                                         usleep(100);
                                 }
                                 ip = dns2.GetResultIP();
-                                if (ip == std::string((char*)inet_ntoa(dest->ip4)))
+                                if (ip == std::string((char*)inet_ntoa(u->ip4)))
                                 {
                                         if (host.length() < 160)
                                         {