Dealing with rare cases when getIdentifiedDbms is needed prior to DBMS isfingerprinted and there are multiples of dbmses inside details

stamparm · stamparm · commit bcdba7b7bb1d · 2012-10-28T01:11:50.000+02:00
diff --git a/lib/controller/checks.py b/lib/controller/checks.py
@@ -491,8 +491,9 @@ def genCmpPayload():
                         if hasattr(test, "details"):
                             for dKey, dValue in test.details.items():
                                 if dKey == "dbms":
+                                    injection.dbms = dValue
                                     if not isinstance(dValue, list):
-                                        injection.dbms = Backend.setDbms(dValue)
+                                        Backend.setDbms(dValue)
                                     else:
                                         Backend.forceDbms(dValue[0], True)
                                 elif dKey == "dbms_version" and injection.dbms_version is None and not conf.testFilter:
diff --git a/lib/core/common.py b/lib/core/common.py
@@ -427,6 +427,8 @@ def getIdentifiedDbms():
             dbms = conf.dbms
         elif len(Backend.getErrorParsedDBMSes()) > 0:
             dbms = Backend.getErrorParsedDBMSes()[0]
+        elif kb.injection.dbms:
+            dbms = unArrayizeValue(kb.injection.dbms)
 
         return aliasToDbmsEnum(dbms)
 
