first files

Practica_2/analisis_grafos.py

@@ -0,0 +1,49 @@
+import angr
+from angrutils import * # for plot_cfg
+import sys
+import os
+
+if len(sys.argv) != 2:
+    print("Usage: python2 analisis_grafos.py <binary_file>")
+    sys.exit(1)
+
+binary_path = sys.argv[1]
+proj = angr.Project(binary_path, load_options={'auto_load_libs': False})
+main = proj.loader.main_object.get_symbol("main")
+start_state = proj.factory.blank_state(addr=main.rebased_addr)
+cfg = proj.analyses.CFGAccurate(
+    fail_fast=True, starts=[main.rebased_addr], initial_state=start_state
+)
+
+print("This is the graph:", cfg.graph)
+print("It has %d nodes and %d edges" % (len(cfg.graph.nodes()), len(cfg.graph.edges())))
+
+# this grabs *any* node at a given location:
+entry_node = cfg.get_any_node(main.rebased_addr)
+
+# on the other hand, this grabs all of the nodes
+print(
+    "There were %d contexts for the entry block"
+    % len(cfg.get_all_nodes(main.rebased_addr))
+)
+
+# we can also look up predecessors and successors
+print("Predecessors of the entry point:", entry_node.predecessors)
+print("Successors of the entry point:", entry_node.successors)
+print(
+    "Successors (and type of jump) of the entry point:",
+    [
+        jumpkind + " to " + str(node.addr)
+        for node, jumpkind in cfg.get_successors_and_jumpkind(entry_node)
+    ],
+)
+
+# Get the filename without extension
+filename_without_extension = os.path.splitext(os.path.basename(binary_path))[0]
+plot_cfg(
+    cfg,
+    filename_without_extension,
+    asminst=True,
+    remove_imports=True,
+    remove_path_terminator=True,
+)