Minor code adjustments

lib/parse/headers.py

@@ -26,23 +26,20 @@ Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
 import re
 
-from xml.sax import parse
-from xml.sax.handler import ContentHandler
-
 from lib.core.common import checkFile
-from lib.core.common import sanitizeStr
 from lib.core.data import kb
 from lib.core.data import paths
 from lib.parse.banner import BannerHandler
 
+
 def headersParser(headers):
     """
     This function calls a class that parses the input HTTP headers to
     fingerprint the back-end database management system operating system
-    and web application technology
+    and the web application technology
     """
 
-    topHeaders = {
+    topHeaders = (
                    "cookie",
                    "microsoftsharepointteamservices",
                    "server",
@@ -50,8 +47,9 @@ def headersParser(headers):
                    "www-authenticate",
                    "x-aspnet-version",
                    "x-powered-by",
-                 }
+                 )
 
     for header in headers:
         if header in topHeaders:
+            # TODO: fill me
             pass

lib/parse/html.py

@@ -63,15 +63,13 @@ class htmlHandler(ContentHandler):
                 self.__match = None
 
 
-def htmlParser(page, xmlfile=None):
+def htmlParser(page):
     """
     This function calls a class that parses the input HTML page to
     fingerprint the back-end database management system
     """
 
-    if not xmlfile:
-        xmlfile = paths.ERRORS_XML
-
+    xmlfile = paths.ERRORS_XML
     checkFile(xmlfile)
     page = sanitizeStr(page)
     handler = htmlHandler(page)