Reenable graphviz graph generation.

Author: 0xbeefd00d

src/crashd/dwarf/dwarf_source_code.py

@@ -310,8 +310,9 @@ def show_tainted_source(zelos, binary_path, trace, taint_path):
     address_map = dwarf_data._address_map
     address_map._attach_src_to_external_addrs(trace, dwarf_data)
 
-    # from .render.graphviz import render_source_graph
-    # render_source_graph(taint_path, address_map, files)
+    from crashd.taint.render.graphviz import render_source_graph
+
+    render_source_graph(taint_path, address_map, files)
 
     (nodes, edges) = get_nodes_and_edges(taint_path, address_map)
     if len(file_to_syspath) > 0:

src/crashd/taint/render/__init__.py

@@ -0,0 +1,16 @@
+# Copyright (C) 2020 Zeropoint Dynamics
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+
+# This program 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 Affero General Public License for more details.
+
+# You should have received a copy of the GNU Affero General Public
+# License along with this program.  If not, see
+# <http://www.gnu.org/licenses/>.
+# ======================================================================

src/crashd/taint/taint.py

@@ -357,18 +357,16 @@ def analyze_crash(z, dataflow, inst_address, mem_address):
         logger.notice(f"Wrote file: {zcovPath}")
     except Exception:
         logger.exception("Unable to show source code")
-    if False:
-        # This is all debug-graph generation
+    if True:
+        # This is all png graph generation
         from .render.graphviz import (
             render_reduced_path_taint_graph,
             render_path_taint_graph,
-            render_source_graph,
         )
 
-        # print(taint_path)
         render_reduced_path_taint_graph(logger, taint_path)
         render_path_taint_graph(logger, taint_path, trace)
-        render_source_graph(logger, taint_path, trace)
+
         # self.get_ida_taint_overlay(
         #     z.internal_engine.main_module.EntryPoint,
         #     trace,