SymbolPool: assert LocKey type in APIs

1 files changed, 2 insertions, 0 deletions

diff --git a/miasm2/core/asmblock.py b/miasm2/core/asmblock.py
index c2b7aa0a..9a0f8081 100644
--- a/miasm2/core/asmblock.py
+++ b/miasm2/core/asmblock.py
@@ -374,6 +374,7 @@ class AsmSymbolPool(object):
         Return offset of @loc_key, None otherwise.
         @loc_key: LocKey instance
         """
+        assert isinstance(loc_key, LocKey)
         return self._loc_key_to_offset.get(loc_key)
 
     def loc_key_to_name(self, loc_key):
@@ -381,6 +382,7 @@ class AsmSymbolPool(object):
         Return name of @loc_key.
         @loc_key: LocKey instance
         """
+        assert isinstance(loc_key, LocKey)
         return self._loc_key_to_name[loc_key]
 
     def add_location(self, name, offset=None):