Let --schema work also for Sybase and MaxDB

bdamele · bdamele · commit 9927f5a7dbd5 · 2011-04-29T00:02:28.000Z
diff --git a/plugins/dbms/maxdb/enumeration.py b/plugins/dbms/maxdb/enumeration.py
@@ -38,8 +38,11 @@ def searchDb(self):
 
     def getColumns(self, onlyColNames=False):
         if not conf.tbl:
-            errMsg = "missing table parameter"
-            raise sqlmapMissingMandatoryOptionException, errMsg
+            warnMsg = "missing table parameter, sqlmap will enumerate "
+            warnMsg += "the whole database management system schema"
+            logger.warn(warnMsg)
+
+            return self.getSchema()
 
         if "." in conf.tbl:
             conf.db, conf.tbl = conf.tbl.split(".")
diff --git a/plugins/dbms/sybase/enumeration.py b/plugins/dbms/sybase/enumeration.py
@@ -48,8 +48,11 @@ def getUsers(self):
 
     def getColumns(self, onlyColNames=False):
         if not conf.tbl:
-            errMsg = "missing table parameter"
-            raise sqlmapMissingMandatoryOptionException, errMsg
+            warnMsg = "missing table parameter, sqlmap will enumerate "
+            warnMsg += "the whole database management system schema"
+            logger.warn(warnMsg)
+
+            return self.getSchema()
 
         if "." in conf.tbl:
             conf.db, conf.tbl = conf.tbl.split(".")
