extern/SSH-Snake
1
0
Fork 1
mirror of https://github.com/MegaManSec/SSH-Snake.git synced 2024-12-12 16:10:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix issue in generating DOT files with nodes containing colons.

Browse Source 
Issue documented in https://github.com/pydot/pydot/issues/258#issuecomment-795798099

Fixes #5.
This commit is contained in:
Joshua Rogers 2024-01-11 22:14:39 +07:00
parent 730963eb57
commit b30ee12a27
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tools/generate-graph.py
View File

@ -11,7 +11,7 @@ def create_graph_from_edges(lookup_table):
return graph return graph
def build_lookup_table(input_lines, ignore_dest_user): def build_lookup_table(input_lines, ignore_dest_user, dot):
lookup_table = set() lookup_table = set()
@ -28,6 +28,13 @@ def build_lookup_table(input_lines, ignore_dest_user):
for match in matches: for match in matches:
user, host, _, dest_user, dest_host = match.groups() user, host, _, dest_user, dest_host = match.groups()
if dot:
user = f'"{user}'
host = f'{host}"'
dest_user = f'"{dest_user}'
dest_host = f'{dest_host}"'
if host == "(127.0.0.1)" or host == "127.0.0.1": if host == "(127.0.0.1)" or host == "127.0.0.1":
if prev_dest_host is not None: if prev_dest_host is not None:
host = prev_dest_host host = prev_dest_host
@ -71,7 +78,7 @@ if __name__ == "__main__":
if args.with_users: if args.with_users:
ignore_dest_user = False ignore_dest_user = False
lookup_table = build_lookup_table(input_lines, ignore_dest_user) lookup_table = build_lookup_table(input_lines, ignore_dest_user, args.format == "dot")
graph = create_graph_from_edges(lookup_table) graph = create_graph_from_edges(lookup_table)
if len(lookup_table) > 500 and args.format == "dot": if len(lookup_table) > 500 and args.format == "dot":