bug fix

stamparm · stamparm · commit 1d4b10dbd180 · 2012-02-08T13:55:50.000Z
diff --git a/doc/THANKS b/doc/THANKS
@@ -342,6 +342,7 @@ Anton Mogilin <azarmaster81@yahoo.com>
 Anastasios Monachos <anastasiosm@gmail.com>
     for providing some useful data
     for suggesting a feature
+    for reporting a couple of bugs
 
 Kirill Morozov <l0rda@l0rda.biz>
     for reporting a bug
diff --git a/plugins/generic/enumeration.py b/plugins/generic/enumeration.py
@@ -102,10 +102,12 @@ def getBanner(self):
             infoMsg = "fetching banner"
             logger.info(infoMsg)
 
-            # Needed for IBM DB2 versions < 9
-            if Backend.isDbms(DBMS.DB2) and int(Backend.getVersion().split(".")[0]) < 9:
-                query = queries[Backend.getIdentifiedDbms()].banner.query2
-                kb.data.banner = unArrayizeValue(inject.getValue(query, safeCharEncode=False))
+            if Backend.isDbms(DBMS.DB2):
+                rootQuery = queries[DBMS.DB2].banner
+                for query in (rootQuery.query, rootQuery.query2):
+                    kb.data.banner = unArrayizeValue(inject.getValue(query, safeCharEncode=False))
+                    if kb.data.banner:
+                        break
             else:
                 query = queries[Backend.getIdentifiedDbms()].banner.query
                 kb.data.banner = unArrayizeValue(inject.getValue(query, safeCharEncode=False))
