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

minor adjustments

Browse Source
This commit is contained in:
Miroslav Stampar 2010-05-21 13:15:21 +00:00
parent 5a5b31ad53
commit 050015d2bb
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lib/core/common.py
View File

@ -1109,14 +1109,16 @@ def profile(profileOutputFile=None, imageOutputFile=None):
cProfile.run("start()", profileOutputFile)
infoMsg = "converting profile data into a graph image, %s" % imageOutputFile
infoMsg = "converting profile data into a graph image '%s'" % imageOutputFile
logger.info(infoMsg)
graphScriptPath = os.path.join(paths.SQLMAP_EXTRAS_PATH, 'gprof2dot', 'gprof2dot.py')
stderr = execute('%s %s -f pstats %s | dot -Tpng -o %s' % (sys.executable, graphScriptPath, profileOutputFile, imageOutputFile), shell=True, stderr=PIPE).stderr.read()
process = execute('%s %s -f pstats %s | dot -Tpng -o %s' % (sys.executable, graphScriptPath, profileOutputFile, imageOutputFile), shell=True, stdout=None, stderr=PIPE)
pollProcess(process)
payloadStderr = process.communicate()[1]
if stderr or not os.path.exists(imageOutputFile):
errMsg = "there was an error while converting ('%s')" % stderr.strip()
if payloadStderr or not os.path.exists(imageOutputFile):
errMsg = "there was an error while converting ('%s'), " % payloadStderr.strip()
errMsg += "but you can still find raw profile data "
errMsg += "inside file '%s'" % profileOutputFile
logger.error(errMsg)