minor update

stamparm · stamparm · commit adf2231edbc4 · 2010-10-06T13:38:03.000Z
diff --git a/lib/core/option.py b/lib/core/option.py
@@ -1003,7 +1003,7 @@ def __setKnowledgeBaseAttributes():
     kb.dep            = None
     kb.docRoot        = None
     kb.dynamicContent = []
-    kb.errorPages     = set()
+    kb.lastErrorPage  = None
     kb.headersCount   = 0
     kb.headersFp      = {}
     kb.htmlFp         = []
@@ -1021,6 +1021,7 @@ def __setKnowledgeBaseAttributes():
 
     kb.parenthesis    = None
     kb.partRun        = None
+    kb.requestUID     = 0
     kb.queryCounter   = 0
     kb.resumedQueries = {}
     kb.stackedTest    = None
diff --git a/lib/core/target.py b/lib/core/target.py
@@ -266,7 +266,7 @@ def initTargetEnv():
         kb.dbms           = None
         kb.dbmsDetected   = False
         kb.dbmsVersion    = [ "Unknown" ]
-        kb.errorPages     = set()
+        kb.lastErrorPage  = None
         kb.injParameter   = None
         kb.injPlace       = None
         kb.injType        = None
diff --git a/lib/parse/html.py b/lib/parse/html.py
@@ -57,7 +57,7 @@ def startElement(self, name, attrs):
             if self.__match:
                 self.dbms = self.__dbms
                 self.__match = None
-                kb.errorPages.add(self.__page)
+                kb.lastErrorPage = (kb.requestUID, self.__page)
 
 def htmlParser(page):
     """
diff --git a/lib/request/connect.py b/lib/request/connect.py
@@ -89,6 +89,8 @@ def getPage(**kwargs):
         requestHeaders  = ""
         responseHeaders = ""
 
+        kb.requestUID  += 1
+
         try:
             if silent:
                 socket.setdefaulttimeout(3)
