Merge pull request #39 from gitoleg/fix-loader

fixes loader  IDA-7 compatibility

Author: gitoleg

plugins/bap/utils/ida.py

@@ -7,6 +7,10 @@
 from ._comment_handler import CommentHandlers
 from ._ctyperewriter import Rewriter
 
+try:
+    from idc import get_segm_name
+except ImportError:
+    from idaapi import get_segm_name
 
 service = Service()
 comment = CommentHandlers()
@@ -30,7 +34,7 @@ def output_segments(out):
     out.writelines(('(', info.get_proc_name()[1], ' ', size, ' ('))
     for seg in idautils.Segments():
         out.write("\n({} {} {:d} ({:#x} {:d}))".format(
-            idaapi.get_segm_name(seg),
+            get_segm_name(seg),
             "code" if idaapi.segtype(seg) == idaapi.SEG_CODE else "data",
             idaapi.get_fileregion_offset(seg),
             seg, idaapi.getseg(seg).size()))

tests/mockidaapi.py

@@ -6,11 +6,12 @@
 PLUGIN_DRAW = NotImplemented
 PLUGIN_HIDE = NotImplemented
 PLUGIN_KEEP = NotImplemented
-class plugin_t(object): NotImplemented
-class text_sink_t(object): NotImplemented
-class Choose2(object): NotImplemented
-def idadir(sub): NotImplemented
-def get_cmt(ea, off): NotImplemented
-def set_cmt(ea, off): NotImplemented
-def askyn_c(dflt, title): NotImplemented
-def get_input_file_path() : NotImplemented
+class plugin_t(object): pass
+class text_sink_t(object): pass
+class Choose2(object): pass
+def idadir(sub): return NotImplemented
+def get_cmt(ea, off): return NotImplemented
+def set_cmt(ea, off): return NotImplemented
+def askyn_c(dflt, title): return NotImplemented
+def get_input_file_path() : return NotImplemented
+def get_segm_name(ea): return NotImplemented