Patching profile capabilities

lib/core/profiling.py

@@ -14,86 +14,19 @@ from lib.core.data import logger
 from lib.core.data import paths
 from lib.core.settings import UNICODE_ENCODING
 
-def profile(profileOutputFile=None, dotOutputFile=None, imageOutputFile=None):
+def profile(profileOutputFile=None):
     """
     This will run the program and present profiling data in a nice looking graph
     """
 
-    try:
-        __import__("gobject")
-        from thirdparty.gprof2dot import gprof2dot
-        from thirdparty.xdot import xdot
-        import gtk
-        import pydot
-    except ImportError as ex:
-        errMsg = "profiling requires third-party libraries ('%s') " % getSafeExString(ex)
-        errMsg += "(Hint: 'sudo apt install python-pydot python-pyparsing python-profiler graphviz')"
-        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
+    infoMsg = "execution profiled and stored into file '%s' (e.g. 'gprof2dot -f pstats %s | dot -Tpng -o /tmp/sqlmap_profile.png')" % (profileOutputFile, profileOutputFile)
     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', UNICODE_ENCODING)
-    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)
-
-    # Reference: http://stackoverflow.com/questions/38176472/graph-write-pdfiris-pdf-attributeerror-list-object-has-no-attribute-writ
-    if isinstance(pydotGraph, list):
-        pydotGraph = pydotGraph[0]
-
-    try:
-        pydotGraph.write_png(imageOutputFile)
-    except OSError:
-        errMsg = "profiling requires graphviz installed "
-        errMsg += "(Hint: 'sudo apt install graphviz')"
-        logger.error(errMsg)
-    else:
-        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)
-        gtk.main()