intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 65 insertions, 0 deletions

diff --git a/library/cpp/malloc/jemalloc/malloc-info.cpp b/library/cpp/malloc/jemalloc/malloc-info.cpp
new file mode 100644
index 0000000000..2643ca4766
--- /dev/null
+++ b/library/cpp/malloc/jemalloc/malloc-info.cpp
@@ -0,0 +1,65 @@
+#include <library/cpp/malloc/api/malloc.h>
+
+using namespace NMalloc;
+
+#if defined(_MSC_VER)
+TMallocInfo NMalloc::MallocInfo() {
+    TMallocInfo r;
+    r.Name = "jemalloc";
+    return r;
+}
+#else
+#include <strings.h>
+#include <stdlib.h>
+#include <inttypes.h>
+
+#include <contrib/libs/jemalloc/include/jemalloc/jemalloc.h>
+
+namespace {
+    bool JESetParam(const char* param, const char*) {
+        if (param) {
+            if (strcmp(param, "j:reset_epoch") == 0) {
+                uint64_t epoch = 1;
+                size_t sz = sizeof(epoch);
+
+                mallctl("epoch", &epoch, &sz, &epoch, sz);
+
+                return true;
+            }
+
+            return false;
+        }
+
+        return false;
+    }
+
+    const char* JEGetParam(const char* param) {
+        if (param) {
+            if (strcmp(param, "allocated") == 0) {
+                JESetParam("j:reset_epoch", nullptr);
+
+                size_t allocated = 0;
+                size_t sz = sizeof(allocated);
+
+                mallctl("stats.allocated", &allocated, &sz, nullptr, 0);
+
+                static_assert(sizeof(size_t) == sizeof(void*), "fix me");
+
+                return (const char*)(void*)allocated;
+            }
+
+            return nullptr;
+        }
+
+        return nullptr;
+    }
+}
+
+TMallocInfo NMalloc::MallocInfo() {
+    TMallocInfo r;
+    r.Name = "jemalloc";
+    r.SetParam = JESetParam;
+    r.GetParam = JEGetParam;
+    return r;
+}
+#endif