minor update (for future implementation of more advanced error page logic)

stamparm · stamparm · commit 56dbf0038f69 · 2010-10-06T12:10:00.000Z
diff --git a/lib/core/option.py b/lib/core/option.py
@@ -1003,6 +1003,7 @@ def __setKnowledgeBaseAttributes():
     kb.dep            = None
     kb.docRoot        = None
     kb.dynamicContent = []
+    kb.errorPages     = set()
     kb.headersCount   = 0
     kb.headersFp      = {}
     kb.htmlFp         = []
diff --git a/lib/core/target.py b/lib/core/target.py
@@ -266,6 +266,7 @@ def initTargetEnv():
         kb.dbms           = None
         kb.dbmsDetected   = False
         kb.dbmsVersion    = [ "Unknown" ]
+        kb.errorPages     = set()
         kb.injParameter   = None
         kb.injPlace       = None
         kb.injType        = None
diff --git a/lib/parse/html.py b/lib/parse/html.py
@@ -57,6 +57,7 @@ def startElement(self, name, attrs):
             if self.__match:
                 self.dbms = self.__dbms
                 self.__match = None
+                kb.errorPages.add(self.__page)
 
 def htmlParser(page):
     """
