Fix get_func_name and test_

Author: b1llow

idb/idapython.py

@@ -1821,12 +1821,11 @@ def get_func_name(self, ea):
         if func is None:
             return ""
 
-        ea = func.startEA
-
         # shouldn't be a chunk
-        if is_flag_set(func.flags, func.FUNC_TAIL):
+        if is_flag_set(func.flags, func.FUNC_TAIL) or ea != func.startEA:
             raise KeyError(ea)
 
+        ea = func.startEA
         nn = self.api.ida_netnode.netnode(ea)
         try:
             return nn.name()

tests/test_idaapi.py

@@ -583,11 +583,10 @@ def test_function_names(kernel32_idb, version, bitness, expected):
         else "__BaseDllInitialize@12"
     )
 
-    if version >= 720:
-        assert api.idc.GetFunctionName(0x689018E5) == "_InternalFindAtom@12"
-    elif version > 500:
-        assert api.idc.GetFunctionName(0x689018E5) == "sub_689017d4"
-    elif version == 500:
+    if version > 500:
+        with pytest.raises(KeyError):
+            api.idc.GetFunctionName(0x689018E5)
+    else:
         assert api.idc.GetFunctionName(0x689018E5) == "sub_689018e5"