Minor improvement

stamparm · stamparm · commit d3060f20d771 · 2014-12-03T13:22:55.000+01:00
diff --git a/lib/core/common.py b/lib/core/common.py
@@ -2833,7 +2833,7 @@ def showHttpErrorCodes():
           for code, count in kb.httpErrorCodes.items())
         logger.warn(warnMsg)
         if not kb.injections:
-            if any(str(_).startswith('4') or str(_).startswith('5') for _ in kb.httpErrorCodes.keys()):
+            if any((str(_).startswith('4') or str(_).startswith('5')) and _ != httplib.INTERNAL_SERVER_ERROR and _ != kb.originalCode for _ in kb.httpErrorCodes.keys()):
                 msg = "too many 4xx and/or 5xx HTTP error codes "
                 msg += "could mean that some kind of protection is involved (e.g. WAF)"
                 logger.warn(msg)
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -476,8 +476,9 @@ def getPage(**kwargs):
                 page = page if isinstance(page, unicode) else getUnicode(page)
 
             code = e.code
-            threadData.lastHTTPError = (threadData.lastRequestUID, code)
 
+            kb.originalCode = kb.originalCode or code
+            threadData.lastHTTPError = (threadData.lastRequestUID, code)
             kb.httpErrorCodes[code] = kb.httpErrorCodes.get(code, 0) + 1
 
             status = getUnicode(e.msg)
