Merge PR #11081: Native compute: cleanup temporary files on program exit

Reviewed-by: JasonGross
Reviewed-by: Zimmi48
Reviewed-by: maximedenes
Reviewed-by: ppedrot

2 files changed, 19 insertions, 0 deletions

diff --git a/clib/cUnix.ml b/clib/cUnix.ml
index c5f6bebb8e..6e3ad59b1f 100644
--- a/clib/cUnix.ml
+++ b/clib/cUnix.ml
@@ -140,3 +140,20 @@ let same_file f1 =
           Unix.Unix_error _ -> false)
   with
       Unix.Unix_error _ -> (fun _ -> false)
+
+(* Copied from ocaml filename.ml *)
+let prng = lazy(Random.State.make_self_init ())
+
+let temp_file_name temp_dir prefix suffix =
+  let rnd = (Random.State.bits (Lazy.force prng)) land 0xFFFFFF in
+  Filename.concat temp_dir (Printf.sprintf "%s%06x%s" prefix rnd suffix)
+
+let mktemp_dir ?(temp_dir=Filename.get_temp_dir_name()) prefix suffix =
+  let rec try_name counter =
+    let name = temp_file_name temp_dir prefix suffix in
+    match Unix.mkdir name 0o700 with
+    | () -> name
+    | exception (Sys_error _ as e) ->
+      if counter >= 1000 then raise e else try_name (counter + 1)
+  in
+  try_name 0
diff --git a/clib/cUnix.mli b/clib/cUnix.mli
index 17574b3c42..55d307c724 100644
--- a/clib/cUnix.mli
+++ b/clib/cUnix.mli
@@ -65,3 +65,5 @@ val waitpid_non_intr : int -> Unix.process_status
 (** Check if two file names refer to the same (existing) file *)
 val same_file : string -> string -> bool
 
+(** Like [Stdlib.Filename.temp_file] but producing a directory. *)
+val mktemp_dir : ?temp_dir:string -> string -> string -> string