mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-22 01:26:42 +03:00
Minor refactoring
This commit is contained in:
parent
1363f26367
commit
ec4e49d771
|
@ -1011,8 +1011,8 @@ def parseTargetDirect():
|
|||
conf.hostname = "localhost"
|
||||
conf.port = 0
|
||||
elif not remote:
|
||||
errMsg = "missing remote connection details"
|
||||
raise SqlmapSyntaxException(errMsg)
|
||||
errMsg = "missing remote connection details"
|
||||
raise SqlmapSyntaxException(errMsg)
|
||||
|
||||
if dbmsName in (DBMS.MSSQL, DBMS.SYBASE):
|
||||
import _mssql
|
||||
|
|
|
@ -750,7 +750,7 @@ def _setOS():
|
|||
|
||||
def _setTechnique():
|
||||
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):
|
||||
_ = []
|
||||
|
|
|
@ -130,7 +130,7 @@ def _setRequestParams():
|
|||
|
||||
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 += "parameters (e.g. www.site.com/article.php?id=1) "
|
||||
warnMsg += "and without providing any POST parameters "
|
||||
|
|
|
@ -104,7 +104,7 @@ def runThreads(numThreads, threadFunction, cleanupFunction=None, forwardExceptio
|
|||
kb.threadContinue = True
|
||||
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:
|
||||
message = "please enter number of threads? [Enter for %d (current)] " % numThreads
|
||||
choice = readInput(message, default=str(numThreads))
|
||||
|
|
|
@ -423,7 +423,7 @@ def bisection(payload, expression, length=None, charsetType=None, firstChar=None
|
|||
abortedFlag = True
|
||||
|
||||
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
|
||||
|
||||
|
|
|
@ -127,7 +127,7 @@ def tableExists(tableFile, regex=None):
|
|||
else:
|
||||
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:
|
||||
kb.brute.tables.append(_)
|
||||
|
||||
|
|
|
@ -221,7 +221,7 @@ class Enumeration(GenericEnumeration):
|
|||
|
||||
if colList:
|
||||
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
|
||||
continue
|
||||
|
||||
|
|
|
@ -275,7 +275,7 @@ class Databases:
|
|||
values = filter(None, arrayizeValue(values))
|
||||
|
||||
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):
|
||||
db = safeSQLIdentificatorNaming(db)
|
||||
|
|
Loading…
Reference in New Issue
Block a user