zack3d/pylingual
1
0
Fork 0
mirror of https://github.com/syssec-utd/pylingual.git synced 2026-05-10 18:39:03 -07:00
Code Issues Packages Projects Releases Wiki Activity

Bandage fix for CDG errors

Browse Source
This commit is contained in:
Xinlong Hu
2025-08-04 23:20:55 -05:00
parent 04afaa7d6d
commit deb0408285
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pylingual/control_flow_reconstruction/cfg.py
+13 -1
View File
@@ -187,6 +187,17 @@ class CFG(DiGraph_CFT):
self.iteration_graphs[-1].append(dot.to_string())
def cdg(self) -> CFG:
if self.start not in self or self.end not in self:
start_disc_node = [n for n in self.nodes if self.out_degree(n) == 0 and n != self.end]
end_disc_node = [n for n in self.nodes if self.in_degree(n) == 0 and n != self.start]
if len(end_disc_node) != 0 and len(start_disc_node) != 0:
self.add_edge(start_disc_node[0], end_disc_node[0])
elif len(start_disc_node) != 0:
self.add_edge(start_disc_node[0], self.end)
elif len(end_disc_node) != 0:
self.add_edge(self.start, end_disc_node[0])
else:
self.add_edge(self.start, self.end)
pdt = nx.create_empty_copy(self)
pdt.add_edges_from((B, A) for A, B in nx.immediate_dominators(self.reverse(), self.end).items())
pdt.remove_edge(self.end, self.end)
@@ -196,5 +207,6 @@ class CFG(DiGraph_CFT):
cdg = nx.create_empty_copy(self)
cdg.add_edges_from((B, A, {"kind": EdgeKind.Fall}) for A, B in itertools.product(self.nodes, self.nodes) if A != B and control_dependent(A, B))
cdg.remove_node(self.end)
cdg.add_edges_from(((self.start, n) for n in cdg.nodes if cdg.in_degree(n) == 0 and n != self.start), kind=EdgeKind.Fall)
start_nodes = [n for n in cdg.nodes if cdg.in_degree(n) == 0 and n != self.start]
cdg.add_edges_from(((self.start, n) for n in start_nodes), kind=EdgeKind.Fall)
return cdg