summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--include/dns.h3
-rw-r--r--src/dns.cpp62
-rw-r--r--src/dnsqueue.cpp25
3 files changed, 54 insertions, 36 deletions
diff --git a/include/dns.h b/include/dns.h
index 6fcabee47..f1c118909 100644
--- a/include/dns.h
+++ b/include/dns.h
@@ -37,9 +37,10 @@ class DNS
private:
in_addr *binip;
char* result;
+ char localbuf[1024];
int t;
void dns_init();
- int fd;
+ int myfd;
void dns_init_2(const char* dnsserver);
in_addr *dns_aton4(const char * const ipstring);
char *dns_ntoa4(const in_addr * const ip);
diff --git a/src/dns.cpp b/src/dns.cpp
index 7adbbfe15..fbe512b9f 100644
--- a/src/dns.cpp
+++ b/src/dns.cpp
@@ -152,6 +152,7 @@ void dns_empty_header(unsigned char *output, const s_header *header, const int l
}
void dns_close(int fd) { /* close query */
+ log(DEBUG,"DNS: dns_close on fd %d",fd);
if (fd == lastcreate) {
wantclose = 1;
return;
@@ -412,12 +413,6 @@ char* DNS::dns_ntoa4(const in_addr * const ip) { /* numeric to ascii: convert 4p
return dns_ntoa4_s(ip,r);
}
-char* DNS::dns_ntoa4_r(const in_addr *ip) { /* numeric to ascii (reentrant): convert 4part IP addr struct to new string */
- char *r;
- r = new char[256];
- return dns_ntoa4_s(ip,r);
-}
-
char* DNS::dns_ntoa4_s(const in_addr *ip, char *r) { /* numeric to ascii (buffered): convert 4part IP addr struct to given string */
unsigned char *m;
m = (unsigned char *)&ip->s_addr;
@@ -426,18 +421,8 @@ char* DNS::dns_ntoa4_s(const in_addr *ip, char *r) { /* numeric to ascii (buffer
}
char* DNS::dns_getresult(const int cfd) { /* retrieve result of DNS query */
- static char r[RESULTSIZE];
- return dns_getresult_s(cfd,r);
-}
-
-char* DNS::dns_getresult_r(const int cfd) { /* retrieve result of DNS query (reentrant) */
- char *r;
- r = new char[RESULTSIZE];
- if(dns_getresult_s(cfd,r) == NULL) {
- delete r;
- return NULL;
- }
- return r;
+ log(DEBUG,"DNS: dns_getresult with cfd=%d",cfd);
+ return dns_getresult_s(cfd,this->localbuf);
}
char* DNS::dns_getresult_s(const int cfd, char *res) { /* retrieve result of DNS query (buffered) */
@@ -462,7 +447,7 @@ char* DNS::dns_getresult_s(const int cfd, char *res) { /* retrieve result of DNS
c = c->next;
}
if (c == NULL) {
- log(DEBUG,"DNS: got a response for a query we didnt send");
+ log(DEBUG,"DNS: got a response for a query we didnt send with fd=%d",cfd);
return NULL; /* query not found */
}
/* query found-- pull from list: */
@@ -494,7 +479,7 @@ char* DNS::dns_getresult_s(const int cfd, char *res) { /* retrieve result of DNS
return NULL;
}
if ((h.flags2 & FLAGS2_MASK_RCODE) != 0) {
- log(DEBUG,"DNS: got an RCODE and didnt want one");
+ log(DEBUG,"DNS lookup failed due to SERVFAIL");
delete c;
return NULL;
}
@@ -646,20 +631,24 @@ char* DNS::dns_getresult_s(const int cfd, char *res) { /* retrieve result of DNS
DNS::DNS()
{
dns_init();
+ log(DEBUG,"Create blank DNS");
}
DNS::DNS(std::string dnsserver)
{
dns_init_2(dnsserver.c_str());
+ log(DEBUG,"Create DNS");
}
void DNS::SetNS(std::string dnsserver)
{
dns_init_2(dnsserver.c_str());
+ log(DEBUG,"Set NS");
}
DNS::~DNS()
{
+ log(DEBUG,"Delete DNS fd=%d",this->myfd);
}
bool DNS::ReverseLookup(std::string ip)
@@ -670,51 +659,56 @@ bool DNS::ReverseLookup(std::string ip)
return false;
}
- this->fd = dns_getname4(binip);
- if (this->fd == -1)
+ this->myfd = dns_getname4(binip);
+ if (this->myfd == -1)
{
return false;
}
+ log(DEBUG,"DNS: ReverseLookup, fd=%d",this->myfd);
return true;
}
bool DNS::ForwardLookup(std::string host)
{
statsDns++;
- this->fd = dns_getip4(host.c_str());
- if (this->fd == -1)
+ this->myfd = dns_getip4(host.c_str());
+ if (this->myfd == -1)
{
return false;
}
+ log(DEBUG,"DNS: ForwardLookup, fd=%d",this->myfd);
return true;
}
bool DNS::HasResult()
{
+ log(DEBUG,"DNS: HasResult, fd=%d",this->myfd);
pollfd polls;
- polls.fd = this->fd;
+ polls.fd = this->myfd;
polls.events = POLLIN;
int ret = poll(&polls,1,1);
+ log(DEBUG,"DNS: Hasresult returning %d",ret);
return (ret > 0);
}
int DNS::GetFD()
{
- return this->fd;
+ return this->myfd;
}
std::string DNS::GetResult()
{
- result = dns_getresult(this->fd);
+ log(DEBUG,"DNS: GetResult()");
+ result = dns_getresult(this->myfd);
if (result) {
statsDnsGood++;
- dns_close(this->fd);
+ dns_close(this->myfd);
return result;
} else {
statsDnsBad++;
- if (this->fd != -1)
+ if (this->myfd != -1)
{
- dns_close(this->fd);
+ dns_close(this->myfd);
}
return "";
}
@@ -723,10 +717,11 @@ std::string DNS::GetResult()
std::string DNS::GetResultIP()
{
char r[1024];
- result = dns_getresult(this->fd);
- if (this->fd != -1)
+ log(DEBUG,"DNS: GetResultIP()");
+ result = dns_getresult(this->myfd);
+ if (this->myfd != -1)
{
- dns_close(this->fd);
+ dns_close(this->myfd);
}
if (result)
{
@@ -745,3 +740,4 @@ std::string DNS::GetResultIP()
return "";
}
}
+
diff --git a/src/dnsqueue.cpp b/src/dnsqueue.cpp
index 610288673..d7b4119ad 100644
--- a/src/dnsqueue.cpp
+++ b/src/dnsqueue.cpp
@@ -99,15 +99,18 @@ public:
Lookup()
{
strcpy(u,"");
+ log(DEBUG,"Create class Lookup");
}
void Reset()
{
strcpy(u,"");
+ log(DEBUG,"Reset class Lookup");
}
~Lookup()
{
+ log(DEBUG,"Delete class Lookup");
}
bool DoLookup(std::string nick)
@@ -119,10 +122,14 @@ public:
log(DEBUG,"New Lookup class for %s with DNSServer set to '%s'",nick.c_str(),DNSServer);
resolver1.SetNS(std::string(DNSServer));
if (!resolver1.ReverseLookup(std::string(usr->host)))
+ {
+ log(DEBUG,"ReverseLookup didnt return true, we're outta here");
return false;
+ }
strlcpy(u,nick.c_str(),NICKMAX);
return true;
}
+ log(DEBUG,"We couldnt find that user");
return false;
}
@@ -130,10 +137,12 @@ public:
{
if (hostname != "")
{
+ log(DEBUG,"Doing forward lookup here with host %s",hostname.c_str());
// doing forward lookup
userrec* usr = NULL;
if (resolver2.HasResult())
{
+ log(DEBUG,"resolver2 has result");
if (resolver2.GetFD() != 0)
{
std::string ip = resolver2.GetResultIP();
@@ -144,6 +153,7 @@ public:
{
if (usr->registered > 3)
{
+ log(DEBUG,"Point 1: Returning true as usr->dns_done is true");
usr->dns_done = true;
return true;
}
@@ -168,10 +178,15 @@ public:
{
usr = Find(u);
if (usr)
+ {
+ log(DEBUG,"Point 2: Returning true");
usr->dns_done = true;
+ }
return true;
}
}
+ log(DEBUG,"Returning false in forward");
+ return false;
}
else
{
@@ -187,16 +202,22 @@ public:
hostname = resolver1.GetResult();
if (usr)
{
- if ((usr->registered > 3) && (hostname == ""))
+ if ((usr->registered > 3) || (hostname == ""))
{
+ log(DEBUG,"Hostname is blank and user->registered > 3, returning true and setting done");
usr->dns_done = true;
return true;
}
}
- resolver2.ForwardLookup(hostname);
+ log(DEBUG,"Starting forwardlookup now for host '%s'...",hostname.c_str());
+ if ((resolver2.GetFD() <= 0) && (hostname != ""))
+ {
+ resolver2.ForwardLookup(hostname);
+ }
}
}
}
+ log(DEBUG,"Returning false");
return false;
}