Protect my leaks

formalsec · Jun 25, 2024 · 7ddbab5 · 7ddbab5
1 parent d6d1ca5
commit 7ddbab5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/lib/bitwuzla_mappings.default.ml b/lib/bitwuzla_mappings.default.ml
@@ -20,7 +20,13 @@ include Mappings_intf
 
 let _debug = false
 
-let leaky_list = ref []
+let leak =
+  let leaky_list = ref [] in
+  let mutex = Mutex.create () in
+  fun module_ ->
+    Mutex.lock mutex;
+    leaky_list := module_ :: !leaky_list;
+    Mutex.unlock mutex
 
 module Fresh_bitwuzla (B : Bitwuzla_cxx.S) : M = struct
   open B
@@ -416,7 +422,7 @@ include (
     module Make () = struct
       let bitwuzla = (module Bitwuzla_cxx.Make () : Bitwuzla_cxx.S)
 
-      let () = leaky_list := bitwuzla :: !leaky_list
+      let () = leak bitwuzla
 
       include Fresh_bitwuzla ((val bitwuzla))
     end

diff --git a/lib/dune b/lib/dune
@@ -61,7 +61,7 @@
   (select
    bitwuzla_mappings.ml
    from
-   (bitwuzla-cxx -> bitwuzla_mappings.default.ml)
+   (threads bitwuzla-cxx -> bitwuzla_mappings.default.ml)
    (-> bitwuzla_mappings.nop.ml))
   (select
    z3_mappings.ml