trivial code cleanup

bdamele · bdamele · commit 475cc8b24bdc · 2015-02-21T13:12:30.000Z
diff --git a/lib/controller/checks.py b/lib/controller/checks.py
@@ -277,8 +277,8 @@ def checkSqlInjection(place, parameter, value):
             infoMsg = "testing '%s'" % title
             logger.info(infoMsg)
 
-            # Force back-end DBMS according to the current
-            # test value for proper payload unescaping
+            # Force back-end DBMS according to the current test DBMS value
+            # for proper payload unescaping
             Backend.forceDbms(dbms[0] if isinstance(dbms, list) else dbms)
 
             # Parse test's <request>
diff --git a/lib/core/common.py b/lib/core/common.py
@@ -462,9 +462,9 @@ def getIdentifiedDbms():
         elif Backend.getForcedDbms() is not None:
             dbms = Backend.getForcedDbms()
         elif Backend.getDbms() is not None:
-            dbms = kb.dbms
+            dbms = Backend.getDbms()
         elif conf.get("dbms"):
-            dbms = conf.dbms
+            dbms = conf.get("dbms")
         elif Backend.getErrorParsedDBMSes():
             dbms = unArrayizeValue(Backend.getErrorParsedDBMSes())
         elif kb.get("injection") and kb.injection.dbms:
