From f853f8973fcae576b2353b4858c2a2fe16bc0571 Mon Sep 17 00:00:00 2001 From: Miroslav Stampar Date: Sun, 23 Nov 2014 15:41:24 +0100 Subject: [PATCH] Minor refactorign --- lib/core/common.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/lib/core/common.py b/lib/core/common.py index 10c84d93d..35a14cc75 100755 --- a/lib/core/common.py +++ b/lib/core/common.py @@ -81,6 +81,7 @@ from lib.core.exception import SqlmapNoneDataException from lib.core.exception import SqlmapMissingDependence from lib.core.exception import SqlmapSilentQuitException from lib.core.exception import SqlmapSyntaxException +from lib.core.exception import SqlmapSystemException from lib.core.exception import SqlmapUserQuitException from lib.core.log import LOGGER_HANDLER from lib.core.optiondict import optDict @@ -845,7 +846,7 @@ def dataToTrafficFile(data): except IOError, ex: errMsg = "something went wrong while trying " errMsg += "to write to the traffic file '%s' ('%s')" % (conf.trafficFile, ex) - raise SqlmapGenericException(errMsg) + raise SqlmapSystemException(errMsg) def dataToDumpFile(dumpFile, data): dumpFile.write(data) @@ -1918,7 +1919,7 @@ def getFileItems(filename, commentPrefix='#', unicode_=True, lowercase=False, un except (IOError, OSError), ex: errMsg = "something went wrong while trying " errMsg += "to read the content of file '%s' ('%s')" % (filename, ex) - raise SqlmapGenericException(errMsg) + raise SqlmapSystemException(errMsg) return retVal if not unique else retVal.keys()