mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-25 11:03:47 +03:00
Minor refactoring
This commit is contained in:
parent
1363f26367
commit
ec4e49d771
|
@ -1011,8 +1011,8 @@ def parseTargetDirect():
|
||||||
conf.hostname = "localhost"
|
conf.hostname = "localhost"
|
||||||
conf.port = 0
|
conf.port = 0
|
||||||
elif not remote:
|
elif not remote:
|
||||||
errMsg = "missing remote connection details"
|
errMsg = "missing remote connection details"
|
||||||
raise SqlmapSyntaxException(errMsg)
|
raise SqlmapSyntaxException(errMsg)
|
||||||
|
|
||||||
if dbmsName in (DBMS.MSSQL, DBMS.SYBASE):
|
if dbmsName in (DBMS.MSSQL, DBMS.SYBASE):
|
||||||
import _mssql
|
import _mssql
|
||||||
|
|
|
@ -750,7 +750,7 @@ def _setOS():
|
||||||
|
|
||||||
def _setTechnique():
|
def _setTechnique():
|
||||||
validTechniques = sorted(getPublicTypeMembers(PAYLOAD.TECHNIQUE), key=lambda x: x[1])
|
validTechniques = sorted(getPublicTypeMembers(PAYLOAD.TECHNIQUE), key=lambda x: x[1])
|
||||||
validLetters = map(lambda x: x[0][0].upper(), validTechniques)
|
validLetters = (_[0][0].upper() for _ in validTechniques)
|
||||||
|
|
||||||
if conf.tech and isinstance(conf.tech, basestring):
|
if conf.tech and isinstance(conf.tech, basestring):
|
||||||
_ = []
|
_ = []
|
||||||
|
|
|
@ -130,7 +130,7 @@ def _setRequestParams():
|
||||||
|
|
||||||
kb.processUserMarks = True if kb.postHint else kb.processUserMarks
|
kb.processUserMarks = True if kb.postHint else kb.processUserMarks
|
||||||
|
|
||||||
if re.search(URI_INJECTABLE_REGEX, conf.url, re.I) and not any(map(lambda place: place in conf.parameters, [PLACE.GET, PLACE.POST])):
|
if re.search(URI_INJECTABLE_REGEX, conf.url, re.I) and not any(place in conf.parameters for place in (PLACE.GET, PLACE.POST)):
|
||||||
warnMsg = "you've provided target url without any GET "
|
warnMsg = "you've provided target url without any GET "
|
||||||
warnMsg += "parameters (e.g. www.site.com/article.php?id=1) "
|
warnMsg += "parameters (e.g. www.site.com/article.php?id=1) "
|
||||||
warnMsg += "and without providing any POST parameters "
|
warnMsg += "and without providing any POST parameters "
|
||||||
|
|
|
@ -104,7 +104,7 @@ def runThreads(numThreads, threadFunction, cleanupFunction=None, forwardExceptio
|
||||||
kb.threadContinue = True
|
kb.threadContinue = True
|
||||||
kb.threadException = False
|
kb.threadException = False
|
||||||
|
|
||||||
if threadChoice and numThreads == 1 and any(map(lambda _: _ in kb.injection.data, (PAYLOAD.TECHNIQUE.BOOLEAN, PAYLOAD.TECHNIQUE.ERROR, PAYLOAD.TECHNIQUE.QUERY, PAYLOAD.TECHNIQUE.UNION))):
|
if threadChoice and numThreads == 1 and any(_ in kb.injection.data for _ in (PAYLOAD.TECHNIQUE.BOOLEAN, PAYLOAD.TECHNIQUE.ERROR, PAYLOAD.TECHNIQUE.QUERY, PAYLOAD.TECHNIQUE.UNION)):
|
||||||
while True:
|
while True:
|
||||||
message = "please enter number of threads? [Enter for %d (current)] " % numThreads
|
message = "please enter number of threads? [Enter for %d (current)] " % numThreads
|
||||||
choice = readInput(message, default=str(numThreads))
|
choice = readInput(message, default=str(numThreads))
|
||||||
|
|
|
@ -423,7 +423,7 @@ def bisection(payload, expression, length=None, charsetType=None, firstChar=None
|
||||||
abortedFlag = True
|
abortedFlag = True
|
||||||
|
|
||||||
finally:
|
finally:
|
||||||
value = map(lambda _: partialValue[_] if _ < len(partialValue) else threadData.shared.value[_], xrange(length))
|
value = (partialValue[_] if _ < len(partialValue) else threadData.shared.value[_] for _ in xrange(length))
|
||||||
|
|
||||||
infoMsg = None
|
infoMsg = None
|
||||||
|
|
||||||
|
|
|
@ -127,7 +127,7 @@ def tableExists(tableFile, regex=None):
|
||||||
else:
|
else:
|
||||||
kb.data.cachedTables[conf.db].append(item)
|
kb.data.cachedTables[conf.db].append(item)
|
||||||
|
|
||||||
for _ in map(lambda x: (conf.db, x), threadData.shared.value):
|
for _ in ((conf.db, item) for item in threadData.shared.value):
|
||||||
if _ not in kb.brute.tables:
|
if _ not in kb.brute.tables:
|
||||||
kb.brute.tables.append(_)
|
kb.brute.tables.append(_)
|
||||||
|
|
||||||
|
|
|
@ -221,7 +221,7 @@ class Enumeration(GenericEnumeration):
|
||||||
|
|
||||||
if colList:
|
if colList:
|
||||||
table = {}
|
table = {}
|
||||||
table[safeSQLIdentificatorNaming(tbl)] = dict(map(lambda x: (x, None), colList))
|
table[safeSQLIdentificatorNaming(tbl)] = dict((_, None) for _ in colList)
|
||||||
kb.data.cachedColumns[safeSQLIdentificatorNaming(conf.db)] = table
|
kb.data.cachedColumns[safeSQLIdentificatorNaming(conf.db)] = table
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
|
|
@ -275,7 +275,7 @@ class Databases:
|
||||||
values = filter(None, arrayizeValue(values))
|
values = filter(None, arrayizeValue(values))
|
||||||
|
|
||||||
if len(values) > 0 and not isListLike(values[0]):
|
if len(values) > 0 and not isListLike(values[0]):
|
||||||
values = map(lambda x: (dbs[0], x), values)
|
values = ((dbs[0], _) for _ in values)
|
||||||
|
|
||||||
for db, table in filterPairValues(values):
|
for db, table in filterPairValues(values):
|
||||||
db = safeSQLIdentificatorNaming(db)
|
db = safeSQLIdentificatorNaming(db)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user