Minor refactoring

Author: stamparm

lib/core/settings.py

@@ -18,7 +18,7 @@
 from thirdparty.six import unichr as _unichr
 
 # sqlmap version (<major>.<minor>.<month>.<monthly commit>)
-VERSION = "1.3.5.155"
+VERSION = "1.3.5.156"
 TYPE = "dev" if VERSION.count('.') > 2 and VERSION.split('.')[-1] != '0' else "stable"
 TYPE_COLORS = {"dev": 33, "stable": 90, "pip": 34}
 VERSION_STRING = "sqlmap/%s#%s" % ('.'.join(VERSION.split('.')[:-1]) if VERSION.count('.') > 2 and VERSION.split('.')[-1] == '0' else VERSION, TYPE)

plugins/dbms/access/enumeration.py

@@ -23,7 +23,7 @@ def getCurrentDb(self):
         warnMsg = "on Microsoft Access it is not possible to get name of the current database"
         logger.warn(warnMsg)
 
-    def isDba(self):
+    def isDba(self, *args, **kwargs):
         warnMsg = "on Microsoft Access it is not possible to test if current user is DBA"
         logger.warn(warnMsg)
 
@@ -39,7 +39,7 @@ def getPasswordHashes(self):
 
         return {}
 
-    def getPrivileges(self, *args):
+    def getPrivileges(self, *args, **kwargs):
         warnMsg = "on Microsoft Access it is not possible to enumerate the user privileges"
         logger.warn(warnMsg)
 

plugins/dbms/h2/enumeration.py

@@ -29,7 +29,7 @@ def getBanner(self):
 
         return kb.data.banner
 
-    def getPrivileges(self, *args):
+    def getPrivileges(self, *args, **kwargs):
         warnMsg = "on H2 it is not possible to enumerate the user privileges"
         logger.warn(warnMsg)
 

plugins/dbms/hsqldb/enumeration.py

@@ -29,7 +29,7 @@ def getBanner(self):
 
         return kb.data.banner
 
-    def getPrivileges(self, *args):
+    def getPrivileges(self, *args, **kwargs):
         warnMsg = "on HSQLDB it is not possible to enumerate the user privileges"
         logger.warn(warnMsg)
 

plugins/dbms/maxdb/enumeration.py

@@ -217,7 +217,7 @@ def getColumns(self, onlyColNames=False, colTuple=None, bruteForce=None, dumpMod
 
         return kb.data.cachedColumns
 
-    def getPrivileges(self, *args):
+    def getPrivileges(self, *args, **kwargs):
         warnMsg = "on SAP MaxDB it is not possible to enumerate the user privileges"
         logger.warn(warnMsg)
 

plugins/dbms/mssqlserver/enumeration.py

@@ -33,7 +33,7 @@
 from thirdparty import six
 
 class Enumeration(GenericEnumeration):
-    def getPrivileges(self, *args):
+    def getPrivileges(self, *args, **kwargs):
         warnMsg = "on Microsoft SQL Server it is not possible to fetch "
         warnMsg += "database users privileges, sqlmap will check whether "
         warnMsg += "or not the database users are database administrators"

plugins/dbms/sqlite/enumeration.py

@@ -18,7 +18,7 @@ def getCurrentDb(self):
         warnMsg = "on SQLite it is not possible to get name of the current database"
         logger.warn(warnMsg)
 
-    def isDba(self):
+    def isDba(self, *args, **kwargs):
         warnMsg = "on SQLite the current user has all privileges"
         logger.warn(warnMsg)
 
@@ -36,7 +36,7 @@ def getPasswordHashes(self):
 
         return {}
 
-    def getPrivileges(self, *args):
+    def getPrivileges(self, *args, **kwargs):
         warnMsg = "on SQLite it is not possible to enumerate the user privileges"
         logger.warn(warnMsg)
 

plugins/dbms/sybase/enumeration.py

@@ -53,7 +53,7 @@ def getUsers(self):
 
         return kb.data.cachedUsers
 
-    def getPrivileges(self, *args):
+    def getPrivileges(self, *args, **kwargs):
         warnMsg = "on Sybase it is not possible to fetch "
         warnMsg += "database users privileges, sqlmap will check whether "
         warnMsg += "or not the database users are database administrators"

plugins/generic/custom.py

@@ -54,11 +54,11 @@ def sqlQuery(self, query):
 
                 return output
             elif not isStackingAvailable() and not conf.direct:
-                    warnMsg = "execution of non-query SQL statements is only "
-                    warnMsg += "available when stacked queries are supported"
-                    logger.warn(warnMsg)
+                warnMsg = "execution of non-query SQL statements is only "
+                warnMsg += "available when stacked queries are supported"
+                logger.warn(warnMsg)
 
-                    return None
+                return None
             else:
                 if sqlType:
                     debugMsg = "executing %s query: '%s'" % (sqlType if sqlType is not None else "SQL", query)

tamper/bluecoat.py

@@ -43,7 +43,7 @@ def process(match):
     retVal = payload
 
     if payload:
-        retVal = re.sub(r"\b(?P<word>[A-Z_]+)(?=[^\w(]|\Z)", lambda match: process(match), retVal)
+        retVal = re.sub(r"\b(?P<word>[A-Z_]+)(?=[^\w(]|\Z)", process, retVal)
         retVal = re.sub(r"\s*=\s*", " LIKE ", retVal)
         retVal = retVal.replace("%09 ", "%09")