further updates

lib/core/common.py

@@ -44,6 +44,7 @@ from subprocess import Popen as execute
 from tempfile import NamedTemporaryFile
 from tempfile import mkstemp
 from xml.etree import ElementTree as ET
+from xml.dom import minidom
 from xml.sax import parse
 
 from extra.cloak.cloak import decloak
@@ -1218,6 +1219,7 @@ def getConsoleWidth(default=80):
     return width if width else default
 
 def parseXmlFile(xmlFile, handler):
+    checkFile(xmlFile)  
     xfile = codecs.open(xmlFile, 'rb', conf.dataEncoding)
     content = xfile.read()
     stream = StringIO(content)
@@ -1225,6 +1227,13 @@ def parseXmlFile(xmlFile, handler):
     stream.close()
     xfile.close()
 
+def readXmlFile(xmlFile):
+    checkFile(xmlFile)  
+    xfile = codecs.open(xmlFile, 'r', conf.dataEncoding)
+    retVal = minidom.parse(xfile).documentElement
+    xfile.close()
+    return retVal
+
 def calculateDeltaSeconds(start, epsilon=0.05):
     """
     Returns elapsed time from start till now (including expected

lib/core/testing.py

@@ -21,7 +21,6 @@ with sqlmap; if not, write to the Free Software Foundation, Inc., 51
 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 """
 
-import codecs
 import doctest
 import logging
 import os
@@ -31,12 +30,11 @@ import sys
 import tempfile
 import time
 
-from xml.dom import minidom
-
 from lib.controller.controller import start
 from lib.core.common import dataToStdout
 from lib.core.common import getCompiledRegex
 from lib.core.common import getConsoleWidth
+from lib.core.common import readXmlFile
 from lib.core.data import conf
 from lib.core.data import logger
 from lib.core.data import paths
@@ -112,9 +110,7 @@ def liveTest():
     count = 0
     global_ = {}
     vars_ = {}
-    xfile = codecs.open(paths.LIVE_TESTS_XML, 'r', conf.dataEncoding)
-    livetests = minidom.parse(xfile).documentElement
-    xfile.close()
+    livetests = readXmlFile(paths.LIVE_TESTS_XML)
     length = len(livetests.getElementsByTagName("case"))
 
     element = livetests.getElementsByTagName("global")