sqlmapproject/sqlmap
1
1
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-02-18 12:30:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

conf->kb

Browse Source
This commit is contained in:
Bernardo Damele 2012-07-11 17:32:12 +01:00
parent 1d2c87e24e
commit 77b275f1a6
2 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
lib/core/common.py
View File

@ -696,43 +696,43 @@ def singleTimeLogMessage(message, level=logging.INFO, flag=None):
def setCurrentMessage(message): def setCurrentMessage(message):
if "[CRITICAL]" in message: if "[CRITICAL]" in message:
conf.currentMessage = "CRITICAL" kb.currentMessage = "CRITICAL"
elif "[ERROR]" in message: elif "[ERROR]" in message:
conf.currentMessage = "ERROR" kb.currentMessage = "ERROR"
elif "[WARNING]" in message: elif "[WARNING]" in message:
conf.currentMessage = "WARNING" kb.currentMessage = "WARNING"
elif "[INFO]" in message: elif "[INFO]" in message:
conf.currentMessage = "INFO" kb.currentMessage = "INFO"
elif "[DEBUG]" in message: elif "[DEBUG]" in message:
conf.currentMessage = "DEBUG" kb.currentMessage = "DEBUG"
elif "[PAYLOAD]" in message: elif "[PAYLOAD]" in message:
conf.currentMessage = "PAYLOAD" kb.currentMessage = "PAYLOAD"
elif "[TRAFFIC OUT]" in message: elif "[TRAFFIC OUT]" in message:
conf.currentMessage = "TRAFFIC OUT" kb.currentMessage = "TRAFFIC OUT"
elif "[TRAFFIC IN]" in message: elif "[TRAFFIC IN]" in message:
conf.currentMessage = "TRAFFIC IN" kb.currentMessage = "TRAFFIC IN"
def setColour(message): def setColour(message):
setCurrentMessage(message) setCurrentMessage(message)
if not hasattr(conf, "currentMessage"): if not hasattr(kb, "currentMessage"):
return message return message
if conf.currentMessage == "CRITICAL": if kb.currentMessage == "CRITICAL":
return colored(message, 'white', on_color='on_red', attrs=['bold']) return colored(message, 'white', on_color='on_red', attrs=['bold'])
elif conf.currentMessage == "ERROR": elif kb.currentMessage == "ERROR":
return colored(message, 'red', attrs=['bold']) return colored(message, 'red', attrs=['bold'])
elif conf.currentMessage == "WARNING": elif kb.currentMessage == "WARNING":
return colored(message, 'yellow') return colored(message, 'yellow')
elif conf.currentMessage == "INFO": elif kb.currentMessage == "INFO":
return colored(message, 'green') return colored(message, 'green')
elif conf.currentMessage == "DEBUG": elif kb.currentMessage == "DEBUG":
return colored(message, 'blue') return colored(message, 'blue')
elif conf.currentMessage == "PAYLOAD": elif kb.currentMessage == "PAYLOAD":
return colored(message, 'magenta') return colored(message, 'magenta')
elif conf.currentMessage == "TRAFFIC OUT": elif kb.currentMessage == "TRAFFIC OUT":
return colored(message, 'cyan') return colored(message, 'cyan')
elif conf.currentMessage == "TRAFFIC IN": elif kb.currentMessage == "TRAFFIC IN":
return colored(message, 'grey') return colored(message, 'grey')
else: else:
return message return message

2
lib/core/option.py
View File

@ -1380,7 +1380,6 @@ def __setConfAttributes():
conf.authPassword = None conf.authPassword = None
conf.boundaries = [] conf.boundaries = []
conf.cj = None conf.cj = None
conf.currentMessage = None
conf.dbmsConnector = None conf.dbmsConnector = None
conf.dbmsHandler = None conf.dbmsHandler = None
conf.dnsServer = None conf.dnsServer = None
@ -1437,6 +1436,7 @@ def __setKnowledgeBaseAttributes(flushAll=True):
kb.commonOutputs = None kb.commonOutputs = None
kb.counters = {} kb.counters = {}
kb.currentMessage = None
kb.data = AttribDict() kb.data = AttribDict()
kb.dataOutputFlag = False kb.dataOutputFlag = False