prometheus-client-mmap 0.22.0-x86_64-linux-musl → 1.2.1-x86_64-linux-musl

data/ext/fast_mmaped_file/file_reading.c

@@ -1,102 +0,0 @@
-#include "file_reading.h"
-
-#include <errno.h>
-#include <fcntl.h>
-
-#include "utils.h"
-
-static int file_open(file_t *source, const char *filepath) {
-    source->file = fopen(filepath, "r");
-    size_t filepath_len = strlen(filepath) + sizeof(char);
-    source->path = malloc(filepath_len);
-    memcpy(source->path, filepath, filepath_len);
-
-    if (source->file == NULL) {
-        save_exception(rb_eArgError, "Can't open %s, errno: %d", filepath, errno);
-
-        return 0;
-    }
-
-    struct stat sb;
-    if (fstat(fileno(source->file), &sb) != 0) {
-        fclose(source->file);
-        save_exception(rb_eIOError, "Can't stat file, errno: %d", errno);
-
-        return 0;
-    }
-    source->length = sb.st_size;
-
-    // go to start
-    if (fseek(source->file, 0L, SEEK_SET) != 0) {
-        fclose(source->file);
-        save_exception(rb_eIOError, "Can't fseek %zu, errno: %d", 0, errno);
-
-        return 0;
-    }
-
-    return 1;
-}
-
-int file_close(file_t *source) {
-    free(source->path);
-    if (fclose(source->file) != 0) {
-        save_exception(rb_eIOError, "Can't fclose file, errno: %d", 0, errno);
-        return 0;
-    }
-    source->file = 0;
-
-    return 1;
-}
-
-int file_open_from_params(file_t *source, VALUE params) {
-    if (RARRAY_LEN(params) != 4) {
-        save_exception(rb_eArgError, "wrong number of arguments %lu instead of 4", RARRAY_LEN(params));
-        return 0;
-    }
-
-    VALUE filepath = rb_ary_entry(params, 0);
-
-    source->multiprocess_mode = rb_sym2id(rb_ary_entry(params, 1));
-    source->type = rb_sym2id(rb_ary_entry(params, 2));
-    source->pid = rb_ary_entry(params, 3);
-
-    return file_open(source, StringValueCStr(filepath));
-}
-
-int read_from_file(const file_t *source, buffer_t *data) {
-    data->size = 0;
-    if (data->buffer == NULL) {
-        data->buffer = malloc(source->length);
-        if (data->buffer == NULL) {
-            save_exception(rb_eIOError, "Can't malloc %zu, errno: %d", source->length, errno);
-            return 0;
-        }
-
-        data->capacity = source->length;
-    } else if (data->capacity < source->length) {
-        data->buffer = realloc(data->buffer, source->length);
-        if (data->buffer == NULL) {
-            save_exception(rb_eIOError, "Can't realloc %zu, errno: %d", source->length, errno);
-            return 0;
-        }
-
-        data->capacity = source->length;
-    }
-
-    data->size = fread(data->buffer, sizeof(char), source->length, source->file);
-    if (data->size != source->length) {
-        save_exception(rb_eIOError, "Couldn't read whole file, read %zu, instead of %zu", data->size, source->length);
-        return 0;
-    }
-
-    return 1;
-}
-
-void buffer_dispose(buffer_t *buffer) {
-    if (buffer->buffer) {
-        free(buffer->buffer);
-    }
-    buffer->buffer = NULL;
-    buffer->size = 0;
-    buffer->capacity = 0;
-}

data/ext/fast_mmaped_file/file_reading.h

@@ -1,30 +0,0 @@
-#ifndef FILE_READING_H
-#define FILE_READING_H
-#include <ruby.h>
-
-typedef struct {
-    FILE *file;
-    size_t length;
-    char *path;
-
-    // Information processed from file path
-    ID multiprocess_mode;
-    ID type;
-    VALUE pid;
-} file_t;
-
-typedef struct {
-    char *buffer;
-    size_t size;
-    size_t capacity;
-} buffer_t;
-
-int file_close(file_t *file);
-
-int file_open_from_params(file_t *file, VALUE params);
-
-int read_from_file(const file_t *source, buffer_t *data);
-
-void buffer_dispose(buffer_t *buffer);
-
-#endif

data/ext/fast_mmaped_file/globals.h

@@ -1,14 +0,0 @@
-#ifndef GLOBALS_H
-#define GLOBALS_H
-#include <ruby.h>
-
-extern ID sym_min;
-extern ID sym_max;
-extern ID sym_livesum;
-extern ID sym_gauge;
-extern ID sym_pid;
-extern ID sym_samples;
-
-extern VALUE prom_eParsingError;
-
-#endif

data/ext/fast_mmaped_file/mmap.c

@@ -1,427 +0,0 @@
-#include "mmap.h"
-
-#include <errno.h>
-#include <fcntl.h>
-#include <ruby/util.h>
-#include <sys/mman.h>
-
-#include "file_format.h"
-#include "utils.h"
-
-#if 0
-#include <stdio.h>
-#define DEBUGF(format, ...) printf("%d: " format "\n", __LINE__, __VA_ARGS__)
-#else
-#define DEBUGF(format, ...)
-#endif
-
-/* This is the ID of the WeakMap used to track strings allocated that
- * are backed by a memory-mapped file.
- */
-#define WEAK_OBJ_TRACKER "@weak_obj_tracker"
-
-/**
- * Maps a given VALUE to some key for the WeakMap. For now, we just use
- * the integer value as the key since that suffices, though this does
- * require Ruby 2.7 due to https://bugs.ruby-lang.org/issues/16035.
- */
-static VALUE weak_obj_tracker_get_key(VALUE val) { return val; }
-
-/**
- * Adds a T_STRING type to the WeakMap. The WeakMap should be stored
- * as an instance variable.
- */
-static void weak_obj_tracker_add(VALUE obj, VALUE val) {
-    Check_Type(val, T_STRING);
-
-    VALUE tracker = rb_iv_get(obj, WEAK_OBJ_TRACKER);
-    VALUE key = weak_obj_tracker_get_key(val);
-
-    rb_funcall(tracker, rb_intern("[]="), 2, key, val);
-}
-
-/**
- * Iterator function for updating a single element from the WeakMap.
- */
-VALUE mm_update_obj_i(RB_BLOCK_CALL_FUNC_ARGLIST(i, self)) {
-    Check_Type(self, T_DATA);
-    Check_Type(i, T_STRING);
-    rb_check_arity(argc, 1, 1);
-
-    mm_ipc *i_mm;
-    GET_MMAP(self, i_mm, MM_MODIFY);
-
-    RSTRING(i)->as.heap.ptr = i_mm->t->addr;
-    RSTRING(i)->as.heap.len = i_mm->t->real;
-
-    return Qtrue;
-}
-
-/**
- * This iterates through the WeakMap defined on the class and updates
- * the RStrings to use the newly-allocated memory region.
- */
-void mm_update(VALUE obj) {
-    VALUE tracker = rb_iv_get(obj, WEAK_OBJ_TRACKER);
-
-    rb_block_call(tracker, rb_intern("each_value"), 0, NULL, mm_update_obj_i, obj);
-}
-
-typedef struct {
-    VALUE obj, *argv;
-    ID id;
-    int flag, argc;
-} mm_bang;
-
-static VALUE mm_protect_bang(VALUE *t) { return rb_funcall2(t[0], (ID)t[1], (int)t[2], (VALUE *)t[3]); }
-
-static VALUE mm_recycle(VALUE str) {
-    rb_gc_force_recycle(str);
-    return str;
-}
-
-static VALUE mm_vunlock(VALUE obj) {
-    mm_ipc *i_mm;
-
-    GET_MMAP(obj, i_mm, 0);
-    return Qnil;
-}
-
-static VALUE mm_str(VALUE obj, int modify) {
-    mm_ipc *i_mm;
-    VALUE ret = Qnil;
-
-    GET_MMAP(obj, i_mm, modify & ~MM_ORIGIN);
-    if (modify & MM_MODIFY) {
-        if (i_mm->t->flag & MM_FROZEN) rb_error_frozen("mmap");
-    }
-    ret = rb_obj_alloc(rb_cString);
-    RSTRING(ret)->as.heap.ptr = i_mm->t->addr;
-    RSTRING(ret)->as.heap.aux.capa = i_mm->t->len;
-    RSTRING(ret)->as.heap.len = i_mm->t->real;
-
-    weak_obj_tracker_add(obj, ret);
-
-    DEBUGF("RString capa: %d, len: %d", RSTRING(ret)->as.heap.aux.capa, RSTRING(ret)->as.heap.len);
-
-    if (modify & MM_ORIGIN) {
-#if HAVE_RB_DEFINE_ALLOC_FUNC
-        RSTRING(ret)->as.heap.aux.shared = obj;
-        FL_SET(ret, RSTRING_NOEMBED);
-        FL_SET(ret, FL_USER18);
-#else
-        RSTRING(ret)->orig = ret;
-#endif
-    }
-    if (i_mm->t->flag & MM_FROZEN) {
-        ret = rb_obj_freeze(ret);
-    }
-    return ret;
-}
-
-static VALUE mm_i_bang(bang_st) mm_bang *bang_st;
-{
-    VALUE str, res;
-    mm_ipc *i_mm;
-
-    str = mm_str(bang_st->obj, bang_st->flag);
-    if (bang_st->flag & MM_PROTECT) {
-        VALUE tmp[4];
-        tmp[0] = str;
-        tmp[1] = (VALUE)bang_st->id;
-        tmp[2] = (VALUE)bang_st->argc;
-        tmp[3] = (VALUE)bang_st->argv;
-        res = rb_ensure(mm_protect_bang, (VALUE)tmp, mm_recycle, str);
-    } else {
-        res = rb_funcall2(str, bang_st->id, bang_st->argc, bang_st->argv);
-        RB_GC_GUARD(res);
-    }
-    if (res != Qnil) {
-        GET_MMAP(bang_st->obj, i_mm, 0);
-        i_mm->t->real = RSTRING_LEN(str);
-    }
-    return res;
-}
-
-static VALUE mm_bang_i(VALUE obj, int flag, ID id, int argc, VALUE *argv) {
-    VALUE res;
-    mm_ipc *i_mm;
-    mm_bang bang_st;
-
-    GET_MMAP(obj, i_mm, 0);
-    if ((flag & MM_CHANGE) && (i_mm->t->flag & MM_FIXED)) {
-        rb_raise(rb_eTypeError, "try to change the size of a fixed map");
-    }
-    bang_st.obj = obj;
-    bang_st.flag = flag;
-    bang_st.id = id;
-    bang_st.argc = argc;
-    bang_st.argv = argv;
-    if (i_mm->t->flag & MM_IPC) {
-        res = rb_ensure(mm_i_bang, (VALUE)&bang_st, mm_vunlock, obj);
-    } else {
-        res = mm_i_bang(&bang_st);
-    }
-    if (res == Qnil) return res;
-    return (flag & MM_ORIGIN) ? res : obj;
-}
-
-static void mm_free(mm_ipc *i_mm) {
-    if (i_mm->t->path) {
-        if (munmap(i_mm->t->addr, i_mm->t->len) != 0) {
-            if (i_mm->t->path != (char *)-1 && i_mm->t->path != NULL) {
-                free(i_mm->t->path);
-            }
-            free(i_mm);
-
-            rb_raise(rb_eRuntimeError, "munmap failed at %s:%d with errno: %d", __FILE__, __LINE__, errno);
-        }
-
-        if (i_mm->t->path != (char *)-1) {
-            if (i_mm->t->real < i_mm->t->len && i_mm->t->vscope != MAP_PRIVATE &&
-                truncate(i_mm->t->path, i_mm->t->real) == -1) {
-                free(i_mm->t->path);
-                free(i_mm);
-                rb_raise(rb_eTypeError, "truncate");
-            }
-            free(i_mm->t->path);
-        }
-    }
-    free(i_mm);
-}
-
-/*
- * call-seq:
- *  new(file)
- *
- * create a new Mmap object
- *
- * * <em>file</em>
- *
- *
- *     Creates a mapping that's shared with all other processes
- *     mapping the same areas of the file.
- *
- */
-VALUE mm_s_new(int argc, VALUE *argv, VALUE obj) {
-    VALUE res = rb_funcall2(obj, rb_intern("allocate"), 0, 0);
-    rb_obj_call_init(res, argc, argv);
-    return res;
-}
-
-VALUE mm_s_alloc(VALUE obj) {
-    VALUE res;
-    mm_ipc *i_mm;
-
-    res = Data_Make_Struct(obj, mm_ipc, 0, mm_free, i_mm);
-    i_mm->t = ALLOC_N(mm_mmap, 1);
-    MEMZERO(i_mm->t, mm_mmap, 1);
-    i_mm->t->fd = -1;
-    return res;
-}
-
-size_t next_page_boundary(size_t value) {
-    size_t page_size = sysconf(_SC_PAGESIZE);
-
-    while (page_size < value) {
-        page_size *= 2;
-    }
-
-    return page_size;
-}
-
-/* Reference implementations:
- * mozilla: https://hg.mozilla.org/mozilla-central/file/3d846420a907/xpcom/glue/FileUtils.cpp#l71
- * glibc: https://github.com/lattera/glibc/blob/master/sysdeps/posix/posix_fallocate.c
- */
-int reserve_mmap_file_bytes(int fd, size_t size) {
-#if __linux__
-    /* From https://stackoverflow.com/a/22820221: The difference with
-     * ftruncate(2) is that (on file systems supporting it, e.g. Ext4)
-     * disk space is indeed reserved by posix_fallocate but ftruncate
-     * extends the file by adding holes (and without reserving disk
-     * space). */
-    return posix_fallocate(fd, 0, size);
-#else
-    /* We simplify the reference implemnetations since we generally
-     * don't need to reserve more than a page size. */
-    return ftruncate(fd, size);
-#endif
-}
-
-VALUE mm_init(VALUE obj, VALUE fname) {
-    struct stat st;
-    int fd, smode = 0, pmode = 0, vscope, perm, init;
-    MMAP_RETTYPE addr;
-    mm_ipc *i_mm;
-    char *path;
-    size_t size = 0;
-    off_t offset;
-
-    vscope = 0;
-    path = 0;
-    fd = -1;
-
-    VALUE klass = rb_eval_string("ObjectSpace::WeakMap");
-    VALUE weak_obj_tracker = rb_class_new_instance(0, NULL, klass);
-    rb_iv_set(obj, WEAK_OBJ_TRACKER, weak_obj_tracker);
-
-    fname = rb_str_to_str(fname);
-    SafeStringValue(fname);
-    path = StringValuePtr(fname);
-
-    vscope = MAP_SHARED;
-    size = 0;
-    perm = 0666;
-
-    smode = O_RDWR;
-    pmode = PROT_READ | PROT_WRITE;
-
-    if ((fd = open(path, smode, perm)) == -1) {
-        rb_raise(rb_eArgError, "Can't open %s", path);
-    }
-
-    if (fstat(fd, &st) == -1) {
-        close(fd);
-        rb_raise(rb_eArgError, "Can't stat %s", path);
-    }
-    size = st.st_size;
-
-    Data_Get_Struct(obj, mm_ipc, i_mm);
-
-    offset = 0;
-    init = 0;
-
-    if (size == 0) {
-        init = 1;
-        size = INITIAL_SIZE;
-    }
-
-    /* We need to ensure the underlying file descriptor is at least a page size.
-     * Otherwise, we could get a SIGBUS error if mmap() attempts to read or write
-     * past the file. */
-    size_t reserve_size = next_page_boundary(size);
-
-    if (reserve_mmap_file_bytes(fd, reserve_size) != 0) {
-        close(fd);
-        rb_raise(rb_eIOError, "Can't reserve %zu bytes for memory-mapped file in %s", reserve_size, path);
-    }
-
-    addr = mmap(0, size, pmode, vscope, fd, offset);
-
-    if (addr == MAP_FAILED || !addr) {
-        close(fd);
-        rb_raise(rb_eArgError, "mmap failed (%d)", errno);
-    }
-    i_mm->t->fd = fd;
-    i_mm->t->addr = addr;
-    i_mm->t->len = size;
-    if (!init) {
-        i_mm->t->real = size;
-    }
-    i_mm->t->pmode = pmode;
-    i_mm->t->vscope = vscope;
-    i_mm->t->smode = smode & ~O_TRUNC;
-    i_mm->t->path = (path) ? ruby_strdup(path) : (char *)-1;
-
-    if (smode == O_WRONLY) {
-        i_mm->t->flag |= MM_FIXED;
-    }
-    return obj;
-}
-
-/*
- * Document-method: []
- * Document-method: slice
- *
- * call-seq: [](args)
- *
- * Element reference - with the following syntax:
- *
- *   self[nth]
- *
- * retrieve the <em>nth</em> character
- *
- *   self[start..last]
- *
- * return a substring from <em>start</em> to <em>last</em>
- *
- *   self[start, length]
- *
- * return a substring of <em>lenght</em> characters from <em>start</em>
- */
-VALUE mm_aref_m(int argc, VALUE *argv, VALUE obj) { return mm_bang_i(obj, MM_ORIGIN, rb_intern("[]"), argc, argv); }
-
-/*
- * Document-method: msync
- * Document-method: sync
- * Document-method: flush
- *
- * call-seq: msync
- *
- * flush the file
- */
-VALUE mm_msync(int argc, VALUE *argv, VALUE obj) {
-    mm_ipc *i_mm;
-    GET_MMAP(obj, i_mm, MM_MODIFY);
-
-    VALUE oflag;
-    int ret;
-    int flag = MS_SYNC;
-
-    if (argc) {
-        rb_scan_args(argc, argv, "01", &oflag);
-        flag = NUM2INT(oflag);
-    }
-    if ((ret = msync(i_mm->t->addr, i_mm->t->len, flag)) != 0) {
-        rb_raise(rb_eArgError, "msync(%d)", ret);
-    }
-
-    return obj;
-}
-
-/*
- * Document-method: munmap
- * Document-method: unmap
- *
- * call-seq: munmap
- *
- * terminate the association
- */
-VALUE mm_unmap(VALUE obj) {
-    mm_ipc *i_mm;
-
-    GET_MMAP(obj, i_mm, 0);
-    if (i_mm->t->path) {
-        if (munmap(i_mm->t->addr, i_mm->t->len) != 0) {
-            if (i_mm->t->path != (char *)-1 && i_mm->t->path != NULL) {
-                free(i_mm->t->path);
-                i_mm->t->path = NULL;
-            }
-
-            rb_raise(rb_eRuntimeError, "munmap failed at %s:%d with errno: %d", __FILE__, __LINE__, errno);
-        }
-
-        if (i_mm->t->path != (char *)-1) {
-            if (i_mm->t->real < i_mm->t->len && i_mm->t->vscope != MAP_PRIVATE &&
-                truncate(i_mm->t->path, i_mm->t->real) == -1) {
-                rb_raise(rb_eTypeError, "truncate");
-            }
-            free(i_mm->t->path);
-        }
-
-        // Ensure any lingering RString values get a length of zero. We
-        // can't zero out the address since GET_MMAP() inside
-        // mm_update_obj_i() expects a non-null address and path.
-        i_mm->t->len = 0;
-        i_mm->t->real = 0;
-        mm_update(obj);
-
-        i_mm->t->addr = NULL;
-        i_mm->t->path = NULL;
-    }
-
-    close(i_mm->t->fd);
-
-    return Qnil;
-}

data/ext/fast_mmaped_file/mmap.h

@@ -1,61 +0,0 @@
-#ifndef MMAP_H
-#define MMAP_H
-
-#include <ruby.h>
-#include <unistd.h>
-
-#define MM_MODIFY 1
-#define MM_ORIGIN 2
-#define MM_CHANGE (MM_MODIFY | 4)
-#define MM_PROTECT 8
-
-#define MM_FROZEN (1 << 0)
-#define MM_FIXED (1 << 1)
-#define MM_ANON (1 << 2)
-#define MM_LOCK (1 << 3)
-#define MM_IPC (1 << 4)
-#define MM_TMP (1 << 5)
-
-#ifndef MMAP_RETTYPE
-#define MMAP_RETTYPE void *
-#endif
-
-typedef struct {
-    MMAP_RETTYPE addr;
-    int smode, pmode, vscope;
-    int advice, flag;
-    VALUE key;
-    size_t len, real;
-    off_t offset;
-    int fd;
-    char *path;
-} mm_mmap;
-
-typedef struct {
-    int count;
-    mm_mmap *t;
-} mm_ipc;
-
-#define GET_MMAP(obj, i_mm, t_modify)                                                                \
-    Data_Get_Struct(obj, mm_ipc, i_mm);                                                              \
-    if (!i_mm->t->path || i_mm->t->fd < 0 || i_mm->t->addr == NULL || i_mm->t->addr == MAP_FAILED) { \
-        rb_raise(rb_eIOError, "unmapped file");                                                      \
-    }                                                                                                \
-    if ((t_modify & MM_MODIFY) && (i_mm->t->flag & MM_FROZEN)) {                                     \
-        rb_error_frozen("mmap");                                                                     \
-    }
-
-VALUE mm_s_alloc(VALUE obj);
-VALUE mm_s_new(int argc, VALUE *argv, VALUE obj);
-VALUE mm_init(VALUE obj, VALUE fname);
-VALUE mm_aref_m(int argc, VALUE *argv, VALUE obj);
-VALUE mm_msync(int argc, VALUE *argv, VALUE obj);
-VALUE mm_unmap(VALUE obj);
-
-/* If memory is ever reallocated, any allocated Ruby strings that have not been
- * garbage collected need to be updated with the new region. If this isn't done,
- * iterating over the Ruby object space and accessing the string data will seg fault.
- */
-void mm_update(VALUE obj);
-
-#endif