fix pointertypes heuritics

trolldbois · Jan 25, 2016 · de6032c · de6032c
1 parent f645fcf
commit de6032c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/haystack/allocators/heapwalker.py b/haystack/allocators/heapwalker.py
@@ -80,6 +80,9 @@ def get_heap_walker(self, mapping):
             raise TypeError('Feed me a IMemoryMapping object')
         if not self._heap_walkers_dict:
             self.list_heap_walkers()
+        # BUG FIXME reverse
+        if mapping.start not in self._heap_walkers_dict:
+            raise ValueError('mapping not used as a heap')
         walker = self._heap_walkers_dict[mapping.start]
         return walker