Blame SOURCES/nmap-6.40-trancated_dns.patch

41d651
diff -up nmap-5.51/nmap_dns.cc.tcpdns nmap-5.51/nmap_dns.cc
41d651
--- a/nmap_dns.cc	2011-01-21 01:04:16.000000000 +0100
41d651
+++ b/nmap_dns.cc	2016-10-26 11:04:49.057723356 +0200
41d651
@@ -677,9 +677,9 @@ static void read_evt_handler(nsock_pool
41d651
 
41d651
   packet_id = buf[1] + (buf[0] << 8);
41d651
 
41d651
-  // Check that this is a response, standard query, and that no truncation was performed
41d651
+  // Check that this is a response, standard query
41d651
   // 0xFA == 11111010 (we're not concerned with AA or RD bits)
41d651
-  if ((buf[2] & 0xFA) != 0x80) return;
41d651
+  if ((buf[2] & 0xF8) != 0x80) return;
41d651
 
41d651
   // Check that the zero field is all zeros and there is no error condition.
41d651
   // We don't care if recursion is available or not since we might be querying
41d651
@@ -715,7 +715,8 @@ static void read_evt_handler(nsock_pool
41d651
     return;
41d651
   }
41d651
 
41d651
-  if (queries <= 0 || answers <= 0) return;
41d651
+  bool processing_successful = false;
41d651
+//   if (queries <= 0 || answers <= 0) return;
41d651
 
41d651
   curbuf = 12;
41d651
 
41d651
@@ -732,7 +733,7 @@ static void read_evt_handler(nsock_pool
41d651
 
41d651
   // We're now at the ANSWER section
41d651
 
41d651
-  for (i=0; i
41d651
+  for (i=0; i
41d651
     curbuf = advance_past_dns_name(buf, buflen, curbuf, &nameloc);
41d651
     if (curbuf == -1) return;
41d651
 
41d651
@@ -758,7 +759,7 @@ static void read_evt_handler(nsock_pool
41d651
 
41d651
       if (encoded_name_to_normal(buf+nameloc, outbuf, sizeof(outbuf)) == -1) return;
41d651
 
41d651
-      if (process_result(ia.s_addr, outbuf, ACTION_FINISHED, packet_id)) {
41d651
+      if ((processing_successful = process_result(ia.s_addr, outbuf, ACTION_FINISHED, packet_id))) {
41d651
         if (o.debugging >= TRACE_DEBUG_LEVEL) log_write(LOG_STDOUT, "mass_rdns: OK MATCHED <%s> to <%s>\n", inet_ntoa(ia), outbuf);
41d651
         output_summary();
41d651
         stat_ok++;
41d651
@@ -771,15 +772,26 @@ static void read_evt_handler(nsock_pool
41d651
       if (ia.s_addr == 0) return;
41d651
 
41d651
       if (o.debugging >= TRACE_DEBUG_LEVEL) log_write(LOG_STDOUT, "mass_rdns: CNAME found for <%s>\n", inet_ntoa(ia));
41d651
-      process_result(ia.s_addr, NULL, ACTION_CNAME_LIST, packet_id);
41d651
+      processing_successful = process_result(ia.s_addr, NULL, ACTION_SYSTEM_RESOLVE, packet_id);
41d651
     } else {
41d651
       if (rdlen < 0 || rdlen + curbuf >= buflen) return;
41d651
       curbuf += rdlen;
41d651
     }
41d651
 
41d651
-    if (curbuf >= buflen) return;
41d651
+    if (curbuf >= buflen) break;//return;
41d651
   }
41d651
 
41d651
+  if (!processing_successful) {
41d651
+    if ((buf[2] & 0x02)) { //truncated
41d651
+      u32 discard = 0;
41d651
+      process_result(discard, NULL, ACTION_SYSTEM_RESOLVE, packet_id);
41d651
+    }
41d651
+    else {
41d651
+      if (o.debugging >= TRACE_DEBUG_LEVEL) {
41d651
+        log_write(LOG_STDOUT, "mass_rdns: Unable to process the response\n");
41d651
+      }
41d651
+    }
41d651
+  }
41d651
 }
41d651
 
41d651
 
41d651
@@ -1225,20 +1237,20 @@ static void nmap_mass_rdns_core(Target *
41d651
 
41d651
   nsp_delete(dnspool);
41d651
 
41d651
-  if (cname_reqs.size() && o.debugging)
41d651
-    log_write(LOG_STDOUT, "Performing system-dns for %d domain names that use CNAMEs\n", (int) cname_reqs.size());
41d651
+  if (deferred_reqs.size() && o.debugging)
41d651
+    log_write(LOG_STDOUT, "Performing system-dns for %d domain names that were deferred\n", (int) deferred_reqs.size());
41d651
 
41d651
-  if (cname_reqs.size()) {
41d651
-    Snprintf(spmobuf, sizeof(spmobuf), "System CNAME DNS resolution of %u host%s.", (unsigned) cname_reqs.size(), cname_reqs.size()-1 ? "s" : "");
41d651
+  if (deferred_reqs.size()) {
41d651
+    Snprintf(spmobuf, sizeof(spmobuf), "System DNS resolution of %u host%s.", (unsigned) deferred_reqs.size(), deferred_reqs.size()-1 ? "s" : "");
41d651
     SPM = new ScanProgressMeter(spmobuf);
41d651
 
41d651
-    for(i=0, reqI = cname_reqs.begin(); reqI != cname_reqs.end(); reqI++, i++) {
41d651
+    for(i=0, reqI = deferred_reqs.begin(); reqI != deferred_reqs.end(); reqI++, i++) {
41d651
       struct sockaddr_storage ss;
41d651
       size_t sslen;
41d651
       char hostname[MAXHOSTNAMELEN + 1] = "";
41d651
 
41d651
       if (keyWasPressed())
41d651
-        SPM->printStats((double) i / cname_reqs.size(), NULL);
41d651
+        SPM->printStats((double) i / deferred_reqs.size(), NULL);
41d651
 
41d651
       tpreq = *reqI;
41d651
 
41d651
diff --git a/nmap_dns.cc b/nmap_dns.cc
41d651
index 9a72336..e01230c 100644
41d651
--- a/nmap_dns.cc
41d651
+++ b/nmap_dns.cc
41d651
@@ -309,7 +309,7 @@ struct host_elem {
41d651
 
41d651
 static std::list<dns_server> servs;
41d651
 static std::list<request *> new_reqs;
41d651
-static std::list<request *> cname_reqs;
41d651
+static std::list<request *> deferred_reqs;
41d651
 static int total_reqs;
41d651
 static nsock_pool dnspool=NULL;
41d651
 
41d651
diff --git a/nmap_dns.cc b/nmap_dns.cc
41d651
index e01230c..540d39d 100644
41d651
--- a/nmap_dns.cc
41d651
+++ b/nmap_dns.cc
41d651
@@ -327,12 +327,13 @@ static ScanProgressMeter *SPM;
41d651
 
41d651
 //------------------- Prototypes and macros ---------------------
41d651
 
41d651
+static void read_evt_handler(nsock_pool, nsock_event, void *);
41d651
 static void put_dns_packet_on_wire(request *req);
41d651
 static const char *lookup_etchosts(u32 ip);
41d651
 static void addto_etchosts(u32 ip, const char *hname);
41d651
 
41d651
 #define ACTION_FINISHED 0
41d651
-#define ACTION_CNAME_LIST 1
41d651
+#define ACTION_SYSTEM_RESOLVE 1
41d651
 #define ACTION_TIMEOUT 2
41d651
 
41d651
 //------------------- Misc code ---------------------
41d651
diff --git a/nmap_dns.cc b/nmap_dns.cc
41d651
index 540d39d..e0c6f6c 100644
41d651
--- a/nmap_dns.cc
41d651
+++ b/nmap_dns.cc
41d651
@@ -1240,7 +1240,7 @@ static void nmap_mass_rdns_core(Target **targets, int num_targets) {
41d651
 
41d651
   connect_dns_servers();
41d651
 
41d651
-  cname_reqs.clear();
41d651
+  deferred_reqs.clear();
41d651
 
41d651
   read_timeout_index = MIN(sizeof(read_timeouts)/sizeof(read_timeouts[0]), servs.size()) - 1;
41d651
 
41d651
diff --git a/nmap_dns.cc b/nmap_dns.cc
41d651
index 9363614..815b3a4 100644
41d651
--- a/nmap_dns.cc
41d651
+++ b/nmap_dns.cc
41d651
@@ -558,7 +558,7 @@ static int process_result(u32 ia, char *result, int action, u16 id) {
41d651
         if (ia != 0 && tpreq->targ->v4host().s_addr != ia)
41d651
           continue;
41d651
 
41d651
-        if (action == ACTION_CNAME_LIST || action == ACTION_FINISHED) {
41d651
+        if (action == ACTION_SYSTEM_RESOLVE || action == ACTION_FINISHED) {
41d651
         servI->capacity += CAPACITY_UP_STEP;
41d651
         check_capacities(&*servI);
41d651
 
41d651
@@ -572,7 +572,7 @@ static int process_result(u32 ia, char *result, int action, u16 id) {
41d651
 
41d651
         total_reqs--;
41d651
 
41d651
-          if (action == ACTION_CNAME_LIST) cname_reqs.push_back(tpreq);
41d651
+          if (action == ACTION_SYSTEM_RESOLVE) deferred_reqs.push_back(tpreq);
41d651
           if (action == ACTION_FINISHED) delete tpreq;
41d651
         } else {
41d651
           memcpy(&tpreq->timeout, nsock_gettimeofday(), sizeof(struct timeval));
41d651
diff --git a/nmap_dns.cc b/nmap_dns.cc
41d651
index 815b3a4..f00ac18 100644
41d651
--- a/nmap_dns.cc
41d651
+++ b/nmap_dns.cc
41d651
@@ -1302,7 +1302,7 @@ static void nmap_mass_rdns_core(Target **targets, int num_targets) {
41d651
     delete SPM;
41d651
   }
41d651
 
41d651
-  cname_reqs.clear();
41d651
+  deferred_reqs.clear();
41d651
 
41d651
 }
41d651