From fa7fa92870eefb2fa15497706b4e65e6c823f60a Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Fri, 27 Apr 2018 07:39:00 +0000
Subject: mjit.c: clean so file on Windows

* mjit.c (dlclose): use FreeLibrary to manage the reference count
  on the loaded module properly.

* mjit.c (clean_so_file): clean shared object file after unloaded,
  in-use files cannot be removed on Windows.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@63273 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 mjit.c | 32 ++++++++++++++++++++++++++++++--
 1 file changed, 30 insertions(+), 2 deletions(-)

(limited to 'mjit.c')

diff --git a/mjit.c b/mjit.c
index 1ba574f919..6d91afbaae 100644
--- a/mjit.c
+++ b/mjit.c
@@ -115,7 +115,7 @@ extern int rb_thread_create_mjit_thread(void (*child_hook)(void), void (*worker_
 #define dlopen(name,flag) ((void*)LoadLibrary(name))
 #define dlerror() strerror(rb_w32_map_errno(GetLastError()))
 #define dlsym(handle,name) ((void*)GetProcAddress((handle),(name)))
-#define dlclose(handle) (CloseHandle(handle))
+#define dlclose(handle) (FreeLibrary(handle))
 #define RTLD_NOW  -1
 
 #define waitpid(pid,stat_loc,options) (WaitForSingleObject((HANDLE)(pid), INFINITE), GetExitCodeProcess((HANDLE)(pid), (LPDWORD)(stat_loc)))
@@ -148,6 +148,10 @@ struct rb_mjit_unit {
     /* Dlopen handle of the loaded object file.  */
     void *handle;
     const rb_iseq_t *iseq;
+#ifdef _WIN32
+    /* DLL cannot be removed while loaded on Windows */
+    char *so_file;
+#endif
     /* Only used by unload_units. Flag to check this unit is currently on stack or not. */
     char used_code_p;
 };
@@ -441,6 +445,23 @@ mjit_free_iseq(const rb_iseq_t *iseq)
     CRITICAL_SECTION_FINISH(4, "mjit_free_iseq");
 }
 
+static void
+clean_so_file(struct rb_mjit_unit *unit)
+{
+#ifdef _WIN32
+# undef Sleep
+    char *so_file = unit->so_file;
+    if (so_file) {
+        unit->so_file = NULL;
+        if (remove(so_file)) {
+            fprintf(stderr, "failed to remove \"%s\": %s\n",
+                    so_file, strerror(errno));
+        }
+        free(so_file);
+    }
+#endif
+}
+
 static void
 free_unit(struct rb_mjit_unit *unit)
 {
@@ -448,6 +469,7 @@ free_unit(struct rb_mjit_unit *unit)
         unit->iseq->body->jit_func = 0;
     if (unit->handle) /* handle is NULL if it's in queue */
         dlclose(unit->handle);
+    clean_so_file(unit);
     xfree(unit);
 }
 
@@ -865,8 +887,13 @@ convert_unit_to_func(struct rb_mjit_unit *unit)
     }
 
     func = load_func_from_so(so_file, funcname, unit);
-    if (!mjit_opts.save_temps)
+    if (!mjit_opts.save_temps) {
+#ifdef _WIN32
+        unit->so_file = strdup(so_file);
+#else
         remove(so_file);
+#endif
+    }
 
     if ((ptrdiff_t)func > (ptrdiff_t)LAST_JIT_ISEQ_FUNC) {
         struct rb_mjit_unit_node *node = create_list_node(unit);
@@ -1095,6 +1122,7 @@ unload_units(void)
         assert(unit->handle != NULL);
         dlclose(unit->handle);
         unit->handle = NULL;
+        clean_so_file(unit);
     }
     verbose(1, "Too many JIT code -- %d units unloaded", units_num - active_units.length);
 }
-- 
cgit v1.2.3