Cosmetics

This commit is contained in:
Bernardo Damele 2011-06-08 15:31:27 +00:00
parent 64bef644c3
commit 70cac24909
2 changed files with 10 additions and 10 deletions

View File

@ -136,7 +136,7 @@ def __showInjections():
conf.dumper.technic(header, data) conf.dumper.technic(header, data)
if inj.place in (HTTPMETHOD.GET, HTTPMETHOD.POST): if inj.place in (HTTPMETHOD.GET, HTTPMETHOD.POST):
infoMsg = "manual usage of %s payloads requires url encoding" % inj.place infoMsg = "usage of injected %s payloads requires manual url-encoding" % inj.place
logger.info(infoMsg) logger.info(infoMsg)
def __randomFillBlankFields(value): def __randomFillBlankFields(value):
@ -166,11 +166,10 @@ def __saveToResultsFile():
if not conf.resultsFP: if not conf.resultsFP:
return return
found = False
results = {}
techniques = dict(map(lambda x: (x[1], x[0]), getPublicTypeMembers(PAYLOAD.TECHNIQUE))) techniques = dict(map(lambda x: (x[1], x[0]), getPublicTypeMembers(PAYLOAD.TECHNIQUE)))
found = False
results = {}
for inj in kb.injections: for inj in kb.injections:
if inj.place is None or inj.parameter is None: if inj.place is None or inj.parameter is None:
continue continue
@ -399,10 +398,8 @@ def start():
break break
kb.vainRun = False kb.vainRun = False
paramKey = (conf.hostname, conf.path, place, parameter)
testSqlInj = True testSqlInj = True
paramKey = (conf.hostname, conf.path, place, parameter)
if paramKey in kb.testedParams: if paramKey in kb.testedParams:
testSqlInj = False testSqlInj = False
@ -578,8 +575,8 @@ def start():
logger.info("Fetched data logged to text files under '%s'" % conf.outputPath) logger.info("Fetched data logged to text files under '%s'" % conf.outputPath)
if conf.multipleTargets: if conf.multipleTargets:
infoMsg = "you can find results of scanning in multiple targets mode " infoMsg = "you can find results of scanning in multiple targets "
infoMsg += "inside the CSV file '%s'" % conf.resultsFilename infoMsg += "mode inside the CSV file '%s'" % conf.resultsFilename
logger.info(infoMsg) logger.info(infoMsg)
return True return True

View File

@ -20,6 +20,7 @@ def profile(profileOutputFile=None, dotOutputFile=None, imageOutputFile=None):
""" """
This will run the program and present profiling data in a nice looking graph This will run the program and present profiling data in a nice looking graph
""" """
try: try:
from extra.gprof2dot import gprof2dot from extra.gprof2dot import gprof2dot
from extra.xdot import xdot from extra.xdot import xdot
@ -28,7 +29,9 @@ def profile(profileOutputFile=None, dotOutputFile=None, imageOutputFile=None):
import pydot import pydot
except ImportError, e: except ImportError, e:
errMsg = "profiling requires third-party libraries (%s). " % getUnicode(e, UNICODE_ENCODING) errMsg = "profiling requires third-party libraries (%s). " % getUnicode(e, UNICODE_ENCODING)
errMsg += "quick steps: 1) install http://code.google.com/p/pydot/ 2) sudo apt-get install python-profiler graphviz" errMsg += "Quick steps:%s" % os.linesep
errMsg += "1) Install http://code.google.com/p/pydot/%s" % os.linesep
errMsg += "2) sudo apt-get install python-profiler graphviz"
logger.error(errMsg) logger.error(errMsg)
return return