From adf2231edbc4097444da3aa5d5fb07c158be56de Mon Sep 17 00:00:00 2001
From: Miroslav Stampar <miroslav.stampar@gmail.com>
Date: Wed, 6 Oct 2010 13:38:03 +0000
Subject: [PATCH] minor update

---
 lib/core/option.py     | 3 ++-
 lib/core/target.py     | 2 +-
 lib/parse/html.py      | 2 +-
 lib/request/connect.py | 2 ++
 4 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/lib/core/option.py b/lib/core/option.py
index 9f4414665..4ba0fe0fd 100644
--- 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
index b3d832a5d..e27c64e02 100644
--- 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
index effe5d3b4..1269d173d 100644
--- a/lib/parse/html.py
+++ b/lib/parse/html.py
@@ -57,7 +57,7 @@ class htmlHandler(ContentHandler):
             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
index 7d61ddad8..95604a65b 100644
--- a/lib/request/connect.py
+++ b/lib/request/connect.py
@@ -89,6 +89,8 @@ class Connect:
         requestHeaders  = ""
         responseHeaders = ""
 
+        kb.requestUID  += 1
+
         try:
             if silent:
                 socket.setdefaulttimeout(3)