full page read is not needed in DNS exfiltration mode

stamparm · stamparm · commit 09f214448515 · 2012-05-26T21:28:43.000Z
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -124,16 +124,19 @@ def __retryProxy(**kwargs):
     @staticmethod
     def __connReadProxy(conn):
         retVal = ""
-        while True:
-            _ = conn.read(MAX_CONNECTION_CHUNK_SIZE)
-            if len(_) == MAX_CONNECTION_CHUNK_SIZE:
-                warnMsg = "large response detected. This could take a while"
-                singleTimeWarnMessage(warnMsg)
-                _ = re.sub(r"(?si)%s.+?%s" % (kb.chars.stop, kb.chars.start), "%s%s%s" % (kb.chars.stop, LARGE_CHUNK_TRIM_MARKER, kb.chars.start), _)
-                retVal += _
-            else:
-                retVal += _
-                break
+
+        if not kb.dnsMode:
+            while True:
+                _ = conn.read(MAX_CONNECTION_CHUNK_SIZE)
+                if len(_) == MAX_CONNECTION_CHUNK_SIZE:
+                    warnMsg = "large response detected. This could take a while"
+                    singleTimeWarnMessage(warnMsg)
+                    _ = re.sub(r"(?si)%s.+?%s" % (kb.chars.stop, kb.chars.start), "%s%s%s" % (kb.chars.stop, LARGE_CHUNK_TRIM_MARKER, kb.chars.start), _)
+                    retVal += _
+                else:
+                    retVal += _
+                    break
+
         return retVal
 
     @staticmethod
