diff --git a/lib/core/common.py b/lib/core/common.py index 2b736d120..55ccd2756 100644 --- a/lib/core/common.py +++ b/lib/core/common.py @@ -23,7 +23,6 @@ Franklin St, Fifth Floor, Boston, MA 02110-1301 USA """ import codecs -import cProfile import inspect import os import random @@ -1191,75 +1190,6 @@ def isHexEncodedString(subject): """ return re.match(r"\A[0-9a-fA-F]+\Z", subject) is not None -def profile(profileOutputFile=None, dotOutputFile=None, imageOutputFile=None): - try: - from extra.gprof2dot import gprof2dot - from extra.xdot import xdot - import gobject - import gtk - import pydot - except ImportError, e: - errMsg = "profiling requires third-party libraries (%s)" % getUnicode(e) - logger.error(errMsg) - return - - if profileOutputFile is None: - profileOutputFile = os.path.join(paths.SQLMAP_OUTPUT_PATH, "sqlmap_profile.raw") - - if dotOutputFile is None: - dotOutputFile = os.path.join(paths.SQLMAP_OUTPUT_PATH, "sqlmap_profile.dot") - - if imageOutputFile is None: - imageOutputFile = os.path.join(paths.SQLMAP_OUTPUT_PATH, "sqlmap_profile.png") - - if os.path.exists(profileOutputFile): - os.remove(profileOutputFile) - - if os.path.exists(dotOutputFile): - os.remove(dotOutputFile) - - if os.path.exists(imageOutputFile): - os.remove(imageOutputFile) - - infoMsg = "profiling the execution into file %s" % profileOutputFile - logger.info(infoMsg) - - # Start sqlmap main function and generate a raw profile file - cProfile.run("start()", profileOutputFile) - - infoMsg = "converting profile data into a dot file '%s'" % dotOutputFile - logger.info(infoMsg) - - # Create dot file by using extra/gprof2dot/gprof2dot.py - # http://code.google.com/p/jrfonseca/wiki/Gprof2Dot - dotFilePointer = codecs.open(dotOutputFile, 'wt', conf.dataEncoding) - parser = gprof2dot.PstatsParser(profileOutputFile) - profile = parser.parse() - profile.prune(0.5/100.0, 0.1/100.0) - dot = gprof2dot.DotWriter(dotFilePointer) - dot.graph(profile, gprof2dot.TEMPERATURE_COLORMAP) - dotFilePointer.close() - - infoMsg = "converting dot file into a graph image '%s'" % imageOutputFile - logger.info(infoMsg) - - # Create graph image (png) by using pydot (python-pydot) - # http://code.google.com/p/pydot/ - pydotGraph = pydot.graph_from_dot_file(dotOutputFile) - pydotGraph.write_png(imageOutputFile) - - infoMsg = "displaying interactive graph with xdot library" - logger.info(infoMsg) - - # Display interactive Graphviz dot file by using extra/xdot/xdot.py - # http://code.google.com/p/jrfonseca/wiki/XDot - win = xdot.DotWindow() - win.connect('destroy', gtk.main_quit) - win.set_filter("dot") - win.open_file(dotOutputFile) - gobject.timeout_add(1000, win.update, dotOutputFile) - gtk.main() - def getConsoleWidth(default=80): width = None @@ -1453,12 +1383,6 @@ def getUnicode(value, encoding=None): else: return unicode(value) #encoding ignored for non-basestring instances -def getBruteUnicode(string): - retVal = unicode() - for char in string: - retVal += unichr(ord(char)) - return retVal - # http://boredzo.org/blog/archives/2007-01-06/longest-common-prefix-in-python-2 def longestCommonPrefix(*sequences): if len(sequences) == 1: diff --git a/lib/core/profiling.py b/lib/core/profiling.py new file mode 100644 index 000000000..0dce76aca --- /dev/null +++ b/lib/core/profiling.py @@ -0,0 +1,99 @@ +#!/usr/bin/env python + +""" +$Id$ + +This file is part of the sqlmap project, http://sqlmap.sourceforge.net. + +Copyright (c) 2010 Miroslav Stampar + +sqlmap is free software; you can redistribute it and/or modify it under +the terms of the GNU General Public License as published by the Free +Software Foundation version 2 of the License. + +sqlmap is distributed in the hope that it will be useful, but WITHOUT ANY +WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS +FOR A PARTICULAR PURPOSE. See the GNU General Public License for more +details. + +You should have received a copy of the GNU General Public License along +with sqlmap; if not, write to the Free Software Foundation, Inc., 51 +Franklin St, Fifth Floor, Boston, MA 02110-1301 USA +""" + +import codecs +import os +import cProfile + +from lib.core.data import conf +from lib.core.data import logger +from lib.core.data import paths + +def profile(profileOutputFile=None, dotOutputFile=None, imageOutputFile=None): + try: + from extra.gprof2dot import gprof2dot + from extra.xdot import xdot + import gobject + import gtk + import pydot + except ImportError, e: + errMsg = "profiling requires third-party libraries (%s)" % getUnicode(e) + logger.error(errMsg) + return + + if profileOutputFile is None: + profileOutputFile = os.path.join(paths.SQLMAP_OUTPUT_PATH, "sqlmap_profile.raw") + + if dotOutputFile is None: + dotOutputFile = os.path.join(paths.SQLMAP_OUTPUT_PATH, "sqlmap_profile.dot") + + if imageOutputFile is None: + imageOutputFile = os.path.join(paths.SQLMAP_OUTPUT_PATH, "sqlmap_profile.png") + + if os.path.exists(profileOutputFile): + os.remove(profileOutputFile) + + if os.path.exists(dotOutputFile): + os.remove(dotOutputFile) + + if os.path.exists(imageOutputFile): + os.remove(imageOutputFile) + + infoMsg = "profiling the execution into file %s" % profileOutputFile + logger.info(infoMsg) + + # Start sqlmap main function and generate a raw profile file + cProfile.run("start()", profileOutputFile) + + infoMsg = "converting profile data into a dot file '%s'" % dotOutputFile + logger.info(infoMsg) + + # Create dot file by using extra/gprof2dot/gprof2dot.py + # http://code.google.com/p/jrfonseca/wiki/Gprof2Dot + dotFilePointer = codecs.open(dotOutputFile, 'wt', conf.dataEncoding) + parser = gprof2dot.PstatsParser(profileOutputFile) + profile = parser.parse() + profile.prune(0.5/100.0, 0.1/100.0) + dot = gprof2dot.DotWriter(dotFilePointer) + dot.graph(profile, gprof2dot.TEMPERATURE_COLORMAP) + dotFilePointer.close() + + infoMsg = "converting dot file into a graph image '%s'" % imageOutputFile + logger.info(infoMsg) + + # Create graph image (png) by using pydot (python-pydot) + # http://code.google.com/p/pydot/ + pydotGraph = pydot.graph_from_dot_file(dotOutputFile) + pydotGraph.write_png(imageOutputFile) + + infoMsg = "displaying interactive graph with xdot library" + logger.info(infoMsg) + + # Display interactive Graphviz dot file by using extra/xdot/xdot.py + # http://code.google.com/p/jrfonseca/wiki/XDot + win = xdot.DotWindow() + win.connect('destroy', gtk.main_quit) + win.set_filter("dot") + win.open_file(dotOutputFile) + gobject.timeout_add(1000, win.update, dotOutputFile) + gtk.main() diff --git a/sqlmap.py b/sqlmap.py index 8f52b8f2a..6c33e2028 100755 --- a/sqlmap.py +++ b/sqlmap.py @@ -45,7 +45,6 @@ except ImportError, _: from lib.controller.controller import start from lib.core.common import banner from lib.core.common import getUnicode -from lib.core.common import profile from lib.core.common import setPaths from lib.core.common import weAreFrozen from lib.core.data import conf @@ -54,6 +53,7 @@ from lib.core.data import paths from lib.core.exception import exceptionsTuple from lib.core.exception import unhandledException from lib.core.option import init +from lib.core.profiling import profile from lib.core.xmldump import closeDumper from lib.parse.cmdline import cmdLineParser