fix: destination shape in nat and pep

wikicensorship · Jul 24, 2022 · 72a0981 · 72a0981
1 parent 23f1c66
commit 72a0981
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/utils/vis.py b/utils/vis.py
@@ -346,12 +346,14 @@ def vis(measurement_path, attach_jscss, edge_lable: str = "none"):
                                         device_name = PEP_NAME
                                         current_node_shape = "star"
                                         already_detected[repeat_steps]["is_pep"] = True
-                                        current_node_id = "pep" + current_node_id + "x"
+                                        if current_node_id != dst_addr_id:
+                                            current_node_id = "pep" + current_node_id + "x"
                                     elif is_nat and not already_detected[repeat_steps]["is_nat"]:
                                         device_color = NAT_COLOR
                                         device_name = NAT_NAME
                                         already_detected[repeat_steps]["is_nat"] = True
-                                        current_node_id = "nat" + current_node_id + "x"
+                                        if current_node_id != dst_addr_id:
+                                            current_node_id = "nat" + current_node_id + "x"
                                     append_lines = tooltips_append_lines(
                                         is_nat, is_middlebox, is_pep, packet_type, tcpflag)
                                     if (is_middlebox_ttl or is_middlebox):