This commit is contained in:
Miroslav Stampar 2015-11-23 09:20:35 +01:00
parent 4d576928a7
commit b2dc443835

View File

@ -17,6 +17,7 @@ from lib.core.data import conf
from lib.core.data import logger from lib.core.data import logger
from lib.core.dicts import SQL_STATEMENTS from lib.core.dicts import SQL_STATEMENTS
from lib.core.enums import AUTOCOMPLETE_TYPE from lib.core.enums import AUTOCOMPLETE_TYPE
from lib.core.exception import SqlmapNoneDataException
from lib.core.settings import NULL from lib.core.settings import NULL
from lib.core.settings import PARAMETER_SPLITTING_REGEX from lib.core.settings import PARAMETER_SPLITTING_REGEX
from lib.core.shell import autoCompletion from lib.core.shell import autoCompletion
@ -35,6 +36,7 @@ class Custom:
sqlType = None sqlType = None
query = query.rstrip(';') query = query.rstrip(';')
try:
for sqlTitle, sqlStatements in SQL_STATEMENTS.items(): for sqlTitle, sqlStatements in SQL_STATEMENTS.items():
for sqlStatement in sqlStatements: for sqlStatement in sqlStatements:
if query.lower().startswith(sqlStatement): if query.lower().startswith(sqlStatement):
@ -68,6 +70,9 @@ class Custom:
output = NULL output = NULL
except SqlmapNoneDataException, ex:
logger.warn(ex)
return output return output
def sqlShell(self): def sqlShell(self):