sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-03-03 11:45:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

proper fix related to the last commit

Browse Source
This commit is contained in:
Miroslav Stampar 2011-06-01 23:00:18 +00:00
parent fd57aae779
commit 8aa5625cd0
3 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/controller/checks.py
View File

@ -440,7 +440,7 @@ def checkSqlInjection(place, parameter, value):
if not isinstance(dValue, list):
injection.dbms = Backend.setDbms(dValue)
else:
Backend.forceDbms(dValue[0])
Backend.forceDbms(dValue[0], True)
elif dKey == "dbms_version" and injection.dbms_version is None:
injection.dbms_version = Backend.setVersion(dValue)
elif dKey == "os" and injection.os is None:
@ -497,6 +497,8 @@ def checkSqlInjection(place, parameter, value):
# Reset forced back-end DBMS value
Backend.flushForcedDbms()
Backend.flushForcedDbms(True)
# Return the injection object
if injection.place is not None and injection.parameter is not None:
injection = checkFalsePositives(injection)

11
lib/core/common.py
View File

@ -297,12 +297,15 @@ class Backend:
logger.error("invalid format of versionsList")
@staticmethod
def forceDbms(dbms):
kb.misc.forcedDbms = aliasToDbmsEnum(dbms)
def forceDbms(dbms, sticky=False):
if not kb.misc.stickyFlag:
kb.misc.forcedDbms = aliasToDbmsEnum(dbms)
kb.misc.stickyFlag = sticky
@staticmethod
def flushForcedDbms():
kb.misc.forcedDbms = None
def flushForcedDbms(force=False):
if not kb.misc.stickyFlag or force:
kb.misc.forcedDbms = None
@staticmethod
def setOs(os):

1
lib/core/option.py
View File

@ -1403,6 +1403,7 @@ def __setKnowledgeBaseAttributes(flushAll=True):
kb.misc.space = ":%s:" % randomStr(length=1, lowercase=True)
kb.misc.dollar = ":%s:" % randomStr(length=1, lowercase=True)
kb.misc.forcedDbms = None
kb.misc.stickyFlag = False
if flushAll:
kb.keywords = set(getFileItems(paths.SQL_KEYWORDS))