From 56dbf0038f69f1ded99e945b90a1dd8d48897901 Mon Sep 17 00:00:00 2001
From: Miroslav Stampar <miroslav.stampar@gmail.com>
Date: Wed, 6 Oct 2010 12:10:00 +0000
Subject: [PATCH] minor update (for future implementation of more advanced
 error page logic)

---
 lib/core/option.py | 1 +
 lib/core/target.py | 1 +
 lib/parse/html.py  | 1 +
 3 files changed, 3 insertions(+)

diff --git a/lib/core/option.py b/lib/core/option.py
index 4b3e0574d..9f4414665 100644
--- 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
index 67e47cabf..b3d832a5d 100644
--- 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
index dda044219..effe5d3b4 100644
--- a/lib/parse/html.py
+++ b/lib/parse/html.py
@@ -57,6 +57,7 @@ class htmlHandler(ContentHandler):
             if self.__match:
                 self.dbms = self.__dbms
                 self.__match = None
+                kb.errorPages.add(self.__page)
 
 def htmlParser(page):
     """