about summary refs log tree commit diff stats
path: root/example/ida/graph_ir.py
diff options
context:
space:
mode:
Diffstat (limited to 'example/ida/graph_ir.py')
-rw-r--r--example/ida/graph_ir.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/example/ida/graph_ir.py b/example/ida/graph_ir.py
index 955dee62..50895b88 100644
--- a/example/ida/graph_ir.py
+++ b/example/ida/graph_ir.py
@@ -36,7 +36,7 @@ def label_str(self):
 
 def color_irblock(irblock, ir_arch):
     out = []
-    lbl = idaapi.COLSTR(ir_arch.loc_db.str_loc_key(irblock.loc_key), idaapi.SCOLOR_INSN)
+    lbl = idaapi.COLSTR(ir_arch.loc_db.pretty_str(irblock.loc_key), idaapi.SCOLOR_INSN)
     out.append(lbl)
     for assignblk in irblock:
         for dst, src in sorted(assignblk.iteritems()):
@@ -118,8 +118,8 @@ def build_graph(verbose=False, simplify=False):
     for addr, name in idautils.Names():
         if name is None:
             continue
-        if (mdis.loc_db.getby_offset(addr) or
-            mdis.loc_db.getby_name(name)):
+        if (mdis.loc_db.get_offset_location(addr) or
+            mdis.loc_db.get_name_location(name)):
             # Symbol alias
             continue
         mdis.loc_db.add_location(name, addr)