Render indirect edges in butterfly and scatter slices as dashed lines; Resolves #7

Author: ElykDeer

slice_types/butterfly_slice.py

@@ -21,9 +21,9 @@
 import tanto
 from tanto.tanto_view import TantoView
 
-from binaryninja import FlowGraph, FlowGraphNode
+from binaryninja import FlowGraph, FlowGraphNode, EdgeStyle
 from binaryninja.function import Function, DisassemblyTextLine
-from binaryninja.enums import BranchType
+from binaryninja.enums import BranchType, EdgePenStyle, ThemeColor
 
 
 class ButterflySlice(tanto.slices.Slice):
@@ -89,7 +89,7 @@ def add_callees(function: Function, level: int, max_level: int):
           callee_node = nodes[callee]
           edge = (function, callee)
           if edge not in added_edges:
-            node.add_outgoing_edge(BranchType.CallDestination, callee_node)
+            node.add_outgoing_edge(BranchType.UserDefinedBranch, callee_node, EdgeStyle(EdgePenStyle.DashLine, 2, ThemeColor.UnconditionalBranchColor))
             added_edges.add(edge)
 
     return new_graph

slice_types/scatter_slice.py

@@ -21,9 +21,9 @@
 import tanto
 from tanto.tanto_view import TantoView
 
-from binaryninja import FlowGraph, FlowGraphNode
+from binaryninja import FlowGraph, FlowGraphNode, EdgeStyle
 from binaryninja.function import Function, DisassemblyTextLine
-from binaryninja.enums import BranchType
+from binaryninja.enums import BranchType, EdgePenStyle, ThemeColor
 
 
 class ScatterSlice(tanto.slices.Slice):
@@ -96,7 +96,7 @@ def add_callees(function: Function, level: int, max_level: int):
           callee_node = nodes[callee]
           edge = (function, callee)
           if edge not in added_edges:
-            node.add_outgoing_edge(BranchType.CallDestination, callee_node)
+            node.add_outgoing_edge(BranchType.UserDefinedBranch, callee_node, EdgeStyle(EdgePenStyle.DashLine, 2, ThemeColor.UnconditionalBranchColor))
             added_edges.add(edge)
 
     return new_graph