PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-14 03:39:03 +00:00
Code Issues Projects Releases Wiki Activity

code refactoring

Browse Source
This commit is contained in:
Miroslav Stampar
2011-01-02 16:51:21 +00:00
parent 252ef7626f
commit 5f9b6b2254
13 changed files with 33 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/controller/action.py
View File

@@ -8,7 +8,7 @@ See the file 'doc/COPYING' for copying permission
"""
from lib.controller.handler import setHandler
from lib.core.common import getHtmlErrorFp
from lib.core.common import getErrorParsedDBMSesFormatted
from lib.core.common import dataToStdout
from lib.core.data import conf
from lib.core.data import kb
@@ -32,7 +32,7 @@ def action():
setHandler()
if not kb.dbmsDetected or not conf.dbmsHandler:
htmlParsed = getHtmlErrorFp()
htmlParsed = getErrorParsedDBMSesFormatted()
errMsg = "sqlmap was not able to fingerprint the "
errMsg += "back-end database management system"

11
lib/controller/checks.py
View File

@@ -16,7 +16,8 @@ from lib.core.common import beep
from lib.core.common import extractRegexResult
from lib.core.common import findDynamicContent
from lib.core.common import getCompiledRegex
from lib.core.common import getErrorParsedDBMS
from lib.core.common import getErrorParsedDBMSes
from lib.core.common import getErrorParsedDBMSesFormatted
from lib.core.common import getInjectionTests
from lib.core.common import getUnicode
from lib.core.common import popValue
@@ -140,9 +141,9 @@ def checkSqlInjection(place, parameter, value):
continue
if getErrorParsedDBMS() and getErrorParsedDBMS() != dbms\
if getErrorParsedDBMSes() and dbms not in getErrorParsedDBMSes()\
and kb.skipTests is None:
message = "parsed error message(s) showed that the back-end DBMS could be '%s'." % getErrorParsedDBMS()
message = "parsed error message(s) showed that the back-end DBMS could be '%s'." % getErrorParsedDBMSesFormatted()
message += " do you want to skip test payloads specific for other DBMSes? [Y/n]"
kb.skipTests = conf.realTest or readInput(message, default="Y") not in ("n", "N")
@@ -150,7 +151,7 @@ def checkSqlInjection(place, parameter, value):
debugMsg = "skipping test '%s' because " % title
debugMsg += "the parsed error message(s) showed "
debugMsg += "that the back-end DBMS could be "
debugMsg += "%s" % getErrorParsedDBMS()
debugMsg += "%s" % getErrorParsedDBMSesFormatted()
logger.debug(debugMsg)
continue
@@ -473,7 +474,7 @@ def heuristicCheckSqlInjection(place, parameter, value):
infoMsg += "parameter '%s' might " % parameter
if result:
infoMsg += "be injectable (possible DBMS: %s)" % (getErrorParsedDBMS() or 'Unknown')
infoMsg += "be injectable (possible DBMS: %s)" % (getErrorParsedDBMSesFormatted() or 'Unknown')
logger.info(infoMsg)
else:
infoMsg += "not be injectable"

4
lib/controller/handler.py
View File

@@ -7,7 +7,7 @@ Copyright (c) 2006-2010 sqlmap developers (http://sqlmap.sourceforge.net/)
See the file 'doc/COPYING' for copying permission
"""
from lib.core.common import getErrorParsedDBMS
from lib.core.common import getErrorParsedDBMSes
from lib.core.common import popValue
from lib.core.common import pushValue
from lib.core.data import conf
@@ -63,7 +63,7 @@ def setHandler():
( SYBASE_ALIASES, SybaseMap, SybaseConn ),
]
inferencedDbms = getErrorParsedDBMS() or kb.dbms
inferencedDbms = (getErrorParsedDBMSes()[0] if getErrorParsedDBMSes() else '') or kb.dbms
for injection in kb.injections:
if hasattr(injection, "dbms") and injection.dbms: