rt: Call maybe_gc on mallocs

1 files changed, 25 insertions, 11 deletions

diff --git a/src/rt/rust_gc.cpp b/src/rt/rust_gc.cpp
index 257df4c8134..9b8743dd03e 100644
--- a/src/rt/rust_gc.cpp
+++ b/src/rt/rust_gc.cpp
@@ -3,6 +3,7 @@
 #include <utility>
 #include <stdint.h>
 
+#include "rust_gc.h"
 #include "rust_internal.h"
 
 #ifdef __WIN32__
@@ -31,38 +32,51 @@ class safe_point_map {
 
 public:
     safe_point_map() {
-        const uintptr_t *data;
-#ifdef __WIN32__
-        data = (const uintptr_t *)GetProcAddress(GetModuleHandle(NULL),
-                                                 "rust_gc_safe_points");
-#else
-        data = (const uintptr_t *)dlsym(RTLD_DEFAULT, "rust_gc_safe_points");
-#endif
+        const uintptr_t *data = get_safe_point_data();
         n_safe_points = *data++;
         index = (const std::pair<void *,const safe_point *> *)data;
         data += n_safe_points * 2;
         safe_points = (const safe_point *)data;
     }
+
+    static const uintptr_t *get_safe_point_data() {
+        static bool init = false;
+        static const uintptr_t *data;
+        if (!init) {
+#ifdef __WIN32__
+            data = (const uintptr_t *)GetProcAddress(GetModuleHandle(NULL),
+                                                     "rust_gc_safe_points");
+#else
+            data = (const uintptr_t *)dlsym(RTLD_DEFAULT,
+                                            "rust_gc_safe_points");
+#endif
+            init = true;
+        }
+        return data;
+    }
 };
 
 void
-gc() {
+gc(rust_task *task) {
     safe_point_map map;
 
     // TODO
 }
 
 void
-maybe_gc() {
+maybe_gc(rust_task *task) {
+    if (safe_point_map::get_safe_point_data() == NULL)
+        return;
+
     // FIXME: We ought to lock this.
     static int zeal = -1;
     if (zeal == -1) {
         char *ev = getenv("RUST_GC_ZEAL");
-        zeal = ev[0] != '\0' && ev[0] != '0';
+        zeal = ev && ev[0] != '\0' && ev[0] != '0';
     }
 
     if (zeal)
-        gc();
+        gc(task);
 }
 
 }