jemalloc_ctl 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 9ab59d11b2bfb3adb1e40476ed0902532eae1436edf9b0e31f64be2dd4f3d8da
+  data.tar.gz: d771262a02cf483a43a85c3d0d643a499e5c5f1cde09bf4be70a5678995010bd
+SHA512:
+  metadata.gz: 86e68a6ed91e73633fb5cbaa416152d374f266de91ff564e49145e62aa4cd766478b3d626640a4c53a55f5e30d8e47e73acc06d77af641e2fe205d32d5b0e418
+  data.tar.gz: 865191340c23cfb157aa6f74800cf78ee25a21c611d3999813eaec291419cc6ccb9f4de9f6ff52b8d9bffc44ccc0c6aab4e5d71f91174a674d4493a68c32c860

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Luke Gruber
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/README

@@ -0,0 +1 @@
+Configure jemalloc at runtime from Ruby.

data/ext/jemalloc_ctl/extconf.rb

@@ -0,0 +1,6 @@
+require "mkmf"
+
+have_header("dlfcn.h")
+have_library("dl", "dlsym")
+
+create_makefile("jemalloc_ctl/jemalloc_ctl")

data/ext/jemalloc_ctl/jemalloc_ctl.c

@@ -0,0 +1,116 @@
+#include <ruby.h>
+#include <dlfcn.h>
+#include <stdbool.h>
+
+typedef int (*mallctl_fn)(const char *, void *, size_t *, void *, size_t);
+
+static mallctl_fn my_mallctl = NULL;
+
+#define JEMALLOC_CTL_DLOPEN_FALLBACK 0
+
+#if JEMALLOC_CTL_DLOPEN_FALLBACK > 0
+    // used for testing
+    static const char *jemalloc_lib_names[] = {
+#ifdef __APPLE__
+        "libjemalloc.2.dylib",
+        "libjemalloc.dylib",
+        "/usr/local/lib/libjemalloc.2.dylib",
+        "/usr/local/lib/libjemalloc.dylib",
+        "/opt/homebrew/lib/libjemalloc.2.dylib",
+        "/opt/homebrew/lib/libjemalloc.dylib",
+#else
+        "libjemalloc.so.2",
+        "libjemalloc.so",
+#endif
+        NULL
+    };
+#endif
+
+static VALUE rb_mJemallocCtl;
+
+static VALUE rb_jemalloc_ctl_enabled_p(VALUE self)
+{
+    return my_mallctl != NULL ? Qtrue : Qfalse;
+}
+
+static VALUE rb_jemalloc_ctl_version(VALUE self)
+{
+    if (my_mallctl == NULL)
+        rb_raise(rb_eRuntimeError, "jemalloc is not available");
+
+    const char *version;
+    size_t sz = sizeof(version);
+    int err = my_mallctl("version", &version, &sz, NULL, 0);
+
+    if (err != 0)
+        rb_raise(rb_eRuntimeError, "mallctl(\"version\") failed with error %d", err);
+
+    return rb_str_new_cstr(version);
+}
+
+static VALUE rb_jemalloc_ctl_enable_background_thread(VALUE self)
+{
+    if (my_mallctl == NULL)
+        rb_raise(rb_eRuntimeError, "jemalloc is not available");
+
+    bool val = true;
+    int err = my_mallctl("background_thread", NULL, NULL, &val, sizeof(bool));
+
+    if (err != 0)
+        rb_raise(rb_eRuntimeError, "mallctl(\"background_thread\") failed with error %d", err);
+
+    return Qtrue;
+}
+
+static VALUE rb_jemalloc_ctl_disable_background_thread(VALUE self)
+{
+    if (my_mallctl == NULL)
+        rb_raise(rb_eRuntimeError, "jemalloc is not available");
+
+    bool val = false;
+    int err = my_mallctl("background_thread", NULL, NULL, &val, sizeof(bool));
+
+    if (err != 0)
+        rb_raise(rb_eRuntimeError, "mallctl(\"background_thread\") failed with error %d", err);
+
+    return Qfalse;
+}
+
+static mallctl_fn find_mallctl_in(void *handle)
+{
+    // Try unprefixed (jemalloc as default allocator) then je_-prefixed
+    mallctl_fn fn = (mallctl_fn)dlsym(handle, "mallctl");
+    if (fn == NULL)
+        fn = (mallctl_fn)dlsym(handle, "je_mallctl");
+    return fn;
+}
+
+void Init_jemalloc_ctl(void)
+{
+    // First: check if mallctl is already in the process
+    // (jemalloc compiled into Ruby or LD_PRELOADed)
+    my_mallctl = find_mallctl_in(RTLD_DEFAULT);
+
+#if JEMALLOC_CTL_DLOPEN_FALLBACK > 0
+    // Fallback: try to dlopen jemalloc from common paths
+    if (my_mallctl == NULL) {
+        void *jemalloc_handle;
+        for (const char **name = jemalloc_lib_names; *name != NULL; name++) {
+            jemalloc_handle = dlopen(*name, RTLD_LAZY);
+            if (jemalloc_handle != NULL) {
+                my_mallctl = find_mallctl_in(jemalloc_handle);
+                if (my_mallctl != NULL)
+                    break;
+                dlclose(jemalloc_handle);
+                jemalloc_handle = NULL;
+            }
+        }
+    }
+#endif
+
+    rb_mJemallocCtl = rb_define_module("JemallocCtl");
+    rb_define_singleton_method(rb_mJemallocCtl, "enabled?", rb_jemalloc_ctl_enabled_p, 0);
+    rb_define_singleton_method(rb_mJemallocCtl, "version", rb_jemalloc_ctl_version, 0);
+    rb_define_singleton_method(rb_mJemallocCtl, "enable_background_thread", rb_jemalloc_ctl_enable_background_thread, 0);
+    rb_define_singleton_method(rb_mJemallocCtl, "disable_background_thread", rb_jemalloc_ctl_disable_background_thread, 0);
+}

data/lib/jemalloc_ctl.rb

@@ -0,0 +1,3 @@
+require "jemalloc_ctl/jemalloc_ctl"
+module JemallocCtl
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: jemalloc_ctl
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Luke Gruber
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies: []
+description: Ruby interface to jemalloc's mallctl, detected at runtime via dlsym.
+  Works with jemalloc compiled into Ruby or loaded via LD_PRELOAD.
+executables: []
+extensions:
+- ext/jemalloc_ctl/extconf.rb
+extra_rdoc_files: []
+files:
+- LICENSE
+- README
+- ext/jemalloc_ctl/extconf.rb
+- ext/jemalloc_ctl/jemalloc_ctl.c
+- lib/jemalloc_ctl.rb
+licenses:
+- MIT
+metadata: {}
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.7.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 4.0.6
+specification_version: 4
+summary: Configure jemalloc via mallctl
+test_files: []