about summary refs log tree commit diff stats
path: root/miasm2/analysis/mem.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--miasm2/analysis/mem.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/miasm2/analysis/mem.py b/miasm2/analysis/mem.py
index 057f7a37..a42df7ba 100644
--- a/miasm2/analysis/mem.py
+++ b/miasm2/analysis/mem.py
@@ -10,6 +10,11 @@ log.setLevel(logging.WARN)
 # allocator is a function(vm, size) -> allocated_address
 allocator = None
 
+def set_allocator(alloc_func):
+    global allocator
+    allocator = alloc_func
+
+
 # Helpers
 
 def indent(s, size=4):