fix

setup.py

@@ -9,7 +9,7 @@
 from setuptools import setup, find_packages, Extension
 from setuptools.command.build_ext import build_ext
 
-__version__ = "0.3.26"
+__version__ = "0.3.27"
 HERE = Path(__file__).resolve().parent
 
 

src/utils/memory.cpp

@@ -6,8 +6,10 @@
 namespace dlplan::utils {
 static char *extra_memory_padding = nullptr;
 
-// Save standard out-of-memory handler.
-static void (*standard_out_of_memory_handler)() = nullptr;
+// Save default out-of-memory handler.
+static void (*default_out_of_memory_handler)() = nullptr;
+// Save current out-of-memory handler.
+static void (*current_out_of_memory_handler)() = nullptr;
 
 static void continuing_out_of_memory_handler() {
     release_extra_memory_padding();
@@ -16,19 +18,25 @@ static void continuing_out_of_memory_handler() {
 
 void reserve_extra_memory_padding(int memory_in_mb) {
     assert(!extra_memory_padding);
+    assert(!current_out_of_memory_handler);
+
     extra_memory_padding = new char[memory_in_mb * 1024 * 1024];
-    standard_out_of_memory_handler = std::set_new_handler(continuing_out_of_memory_handler);
+    std::set_new_handler(continuing_out_of_memory_handler);
+    current_out_of_memory_handler = continuing_out_of_memory_handler;
 }
 
 void release_extra_memory_padding() {
     assert(extra_memory_padding);
+    assert(current_out_of_memory_handler);
+
     delete[] extra_memory_padding;
     extra_memory_padding = nullptr;
-    assert(standard_out_of_memory_handler);
-    std::set_new_handler(standard_out_of_memory_handler);
+    std::set_new_handler(default_out_of_memory_handler);
+    current_out_of_memory_handler = nullptr;
 }
 
 bool extra_memory_padding_is_reserved() {
     return extra_memory_padding;
 }
+
 }

src/utils/memory.h

@@ -3,6 +3,7 @@
 
 #include <memory>
 #include <utility>
+#include <vector>
 
 namespace dlplan::utils {
 
@@ -21,6 +22,7 @@ namespace dlplan::utils {
 extern void reserve_extra_memory_padding(int memory_in_mb);
 extern void release_extra_memory_padding();
 extern bool extra_memory_padding_is_reserved();
+
 }
 
 #endif