mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-23 01:56:36 +03:00
140 lines
4.4 KiB
Python
140 lines
4.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Copyright (c) 2006-2019 sqlmap developers (http://sqlmap.org/)
|
|
See the file 'LICENSE' for copying permission
|
|
"""
|
|
|
|
from __future__ import print_function
|
|
|
|
import re
|
|
import sys
|
|
|
|
from lib.core.common import Backend
|
|
from lib.core.common import dataToStdout
|
|
from lib.core.common import getSQLSnippet
|
|
from lib.core.common import getUnicode
|
|
from lib.core.common import isStackingAvailable
|
|
from lib.core.data import conf
|
|
from lib.core.data import logger
|
|
from lib.core.dicts import SQL_STATEMENTS
|
|
from lib.core.enums import AUTOCOMPLETE_TYPE
|
|
from lib.core.exception import SqlmapNoneDataException
|
|
from lib.core.settings import NULL
|
|
from lib.core.settings import PARAMETER_SPLITTING_REGEX
|
|
from lib.core.shell import autoCompletion
|
|
from lib.request import inject
|
|
|
|
class Custom:
|
|
"""
|
|
This class defines custom enumeration functionalities for plugins.
|
|
"""
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def sqlQuery(self, query):
|
|
output = None
|
|
sqlType = None
|
|
query = query.rstrip(';')
|
|
|
|
try:
|
|
for sqlTitle, sqlStatements in SQL_STATEMENTS.items():
|
|
for sqlStatement in sqlStatements:
|
|
if query.lower().startswith(sqlStatement):
|
|
sqlType = sqlTitle
|
|
break
|
|
|
|
if not any(_ in query.upper() for _ in ("OPENROWSET", "INTO")) and (not sqlType or "SELECT" in sqlType):
|
|
infoMsg = "fetching %s query output: '%s'" % (sqlType if sqlType is not None else "SQL", query)
|
|
logger.info(infoMsg)
|
|
|
|
output = inject.getValue(query, fromUser=True)
|
|
|
|
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)
|
|
|
|
return None
|
|
else:
|
|
if sqlType:
|
|
debugMsg = "executing %s query: '%s'" % (sqlType if sqlType is not None else "SQL", query)
|
|
else:
|
|
debugMsg = "executing unknown SQL type query: '%s'" % query
|
|
logger.debug(debugMsg)
|
|
|
|
inject.goStacked(query)
|
|
|
|
debugMsg = "done"
|
|
logger.debug(debugMsg)
|
|
|
|
output = NULL
|
|
|
|
except SqlmapNoneDataException as ex:
|
|
logger.warn(ex)
|
|
|
|
return output
|
|
|
|
def sqlShell(self):
|
|
infoMsg = "calling %s shell. To quit type " % Backend.getIdentifiedDbms()
|
|
infoMsg += "'x' or 'q' and press ENTER"
|
|
logger.info(infoMsg)
|
|
|
|
autoCompletion(AUTOCOMPLETE_TYPE.SQL)
|
|
|
|
while True:
|
|
query = None
|
|
|
|
try:
|
|
query = raw_input("sql-shell> ")
|
|
query = getUnicode(query, encoding=sys.stdin.encoding)
|
|
query = query.strip("; ")
|
|
except KeyboardInterrupt:
|
|
print()
|
|
errMsg = "user aborted"
|
|
logger.error(errMsg)
|
|
except EOFError:
|
|
print()
|
|
errMsg = "exit"
|
|
logger.error(errMsg)
|
|
break
|
|
|
|
if not query:
|
|
continue
|
|
|
|
if query.lower() in ("x", "q", "exit", "quit"):
|
|
break
|
|
|
|
output = self.sqlQuery(query)
|
|
|
|
if output and output != "Quit":
|
|
conf.dumper.query(query, output)
|
|
|
|
elif not output:
|
|
pass
|
|
|
|
elif output != "Quit":
|
|
dataToStdout("No output\n")
|
|
|
|
def sqlFile(self):
|
|
infoMsg = "executing SQL statements from given file(s)"
|
|
logger.info(infoMsg)
|
|
|
|
for filename in re.split(PARAMETER_SPLITTING_REGEX, conf.sqlFile):
|
|
filename = filename.strip()
|
|
|
|
if not filename:
|
|
continue
|
|
|
|
snippet = getSQLSnippet(Backend.getDbms(), filename)
|
|
|
|
if snippet and all(query.strip().upper().startswith("SELECT") for query in (_ for _ in snippet.split(';' if ';' in snippet else '\n') if _)):
|
|
for query in (_ for _ in snippet.split(';' if ';' in snippet else '\n') if _):
|
|
query = query.strip()
|
|
if query:
|
|
conf.dumper.query(query, self.sqlQuery(query))
|
|
else:
|
|
conf.dumper.query(snippet, self.sqlQuery(snippet))
|