idapython: fix arguments to get_func_cmt

williballenthin · Jul 30, 2018 · 9e5e3ba · 9e5e3ba
1 parent ad85663
commit 9e5e3ba
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/idb/idapython.py b/idb/idapython.py
@@ -1328,7 +1328,7 @@ def get_func(self, ea):
             else:
                 return func
 
-    def get_func_cmt(self, func, repeatable):
+    def get_func_cmt(self, ea, repeatable):
         # function comments are stored on the `$ funcs` netnode
         # tag is either `R` or `C`.
         # index is effective address of the function.
@@ -1345,6 +1345,8 @@ def get_func_cmt(self, func, repeatable):
         #
         # i think its a bug that when you set a repeatable function via the IDA UI,
         # it also sets a local function comment.
+
+        func = self.get_func(ea)
         if func is None:
             return ''