Add `getMapping` primitive to allocator directly

1 files changed, 10 insertions, 0 deletions

diff --git a/include/klee/KDAlloc/allocator.h b/include/klee/KDAlloc/allocator.h
index 2d8a07ac..45d90536 100644
--- a/include/klee/KDAlloc/allocator.h
+++ b/include/klee/KDAlloc/allocator.h
@@ -117,6 +117,16 @@ public:
 
   explicit operator bool() const noexcept { return !control.isNull(); }
 
+  Mapping &getMapping() noexcept {
+    assert(!!*this && "Cannot get mapping of uninitialized allocator.");
+    return control->mapping;
+  }
+
+  Mapping const &getMapping() const noexcept {
+    assert(!!*this && "Cannot get mapping of uninitialized allocator.");
+    return control->mapping;
+  }
+
   [[nodiscard]] void *allocate(std::size_t size) {
     assert(*this && "Invalid allocator");