Minor refactoring

This commit is contained in:
Miroslav Stampar 2019-05-30 22:55:54 +02:00
parent f8e9f9c87d
commit 257fa3e9e4
14 changed files with 19 additions and 19 deletions

View File

@ -18,7 +18,7 @@ from lib.core.enums import OS
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)

View File

@ -23,7 +23,7 @@ class Enumeration(GenericEnumeration):
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 @@ class Enumeration(GenericEnumeration):
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)

View File

@ -29,7 +29,7 @@ class Enumeration(GenericEnumeration):
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)

View File

@ -29,7 +29,7 @@ class Enumeration(GenericEnumeration):
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)

View File

@ -217,7 +217,7 @@ class Enumeration(GenericEnumeration):
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)

View File

@ -33,7 +33,7 @@ from plugins.generic.enumeration import Enumeration as GenericEnumeration
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"

View File

@ -18,7 +18,7 @@ class Enumeration(GenericEnumeration):
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 @@ class Enumeration(GenericEnumeration):
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)

View File

@ -53,7 +53,7 @@ class Enumeration(GenericEnumeration):
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"

View File

@ -54,11 +54,11 @@ class Custom(object):
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)

View File

@ -43,7 +43,7 @@ def tamper(payload, **kwargs):
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")

View File

@ -49,7 +49,7 @@ def tamper(payload, **kwargs):
retVal = payload
if payload:
retVal = re.sub(r"(?<=\W)(?P<word>[A-Za-z_]+)(?=\W|\Z)", lambda match: process(match), retVal)
retVal = re.sub(r"(?<=\W)(?P<word>[A-Za-z_]+)(?=\W|\Z)", process, retVal)
retVal = retVal.replace(" /*!0", "/*!0")
return retVal

View File

@ -54,7 +54,7 @@ def tamper(payload, **kwargs):
retVal = ""
if payload:
payload = re.sub(r"(?<=\W)(?P<word>[A-Za-z_]+)(?=\W|\Z)", lambda match: process(match), payload)
payload = re.sub(r"(?<=\W)(?P<word>[A-Za-z_]+)(?=\W|\Z)", process, payload)
for i in xrange(len(payload)):
if payload[i].isspace():

View File

@ -46,7 +46,7 @@ def tamper(payload, **kwargs):
retVal = payload
if payload:
retVal = re.sub(r"(?<=\W)(?P<word>[A-Za-z_]+)(?=[^\w(]|\Z)", lambda match: process(match), retVal)
retVal = re.sub(r"(?<=\W)(?P<word>[A-Za-z_]+)(?=[^\w(]|\Z)", process, retVal)
retVal = retVal.replace(" /*!", "/*!").replace("*/ ", "*/")
return retVal

View File

@ -47,7 +47,7 @@ def tamper(payload, **kwargs):
retVal = payload
if payload:
retVal = re.sub(r"(?<=\W)(?P<word>[A-Za-z_]+)(?=\W|\Z)", lambda match: process(match), retVal)
retVal = re.sub(r"(?<=\W)(?P<word>[A-Za-z_]+)(?=\W|\Z)", process, retVal)
retVal = retVal.replace(" /*!", "/*!").replace("*/ ", "*/")
return retVal