intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 41 insertions, 0 deletions

diff --git a/library/cpp/balloc/malloc-info.cpp b/library/cpp/balloc/malloc-info.cpp
new file mode 100644
index 0000000000..604b1fb145
--- /dev/null
+++ b/library/cpp/balloc/malloc-info.cpp
@@ -0,0 +1,41 @@
+#include <library/cpp/malloc/api/malloc.h>
+
+#include <string.h>
+
+using namespace NMalloc;
+
+extern "C" void DisableBalloc();
+extern "C" void EnableBalloc();
+extern "C" bool BallocDisabled();
+
+namespace {
+    bool SetAllocParam(const char* name, const char* value) {
+        if (strcmp(name, "disable") == 0) {
+            if (value == nullptr || strcmp(value, "false") != 0) {
+                // all values other than "false" are considred to be "true" for compatibility
+                DisableBalloc();
+            } else {
+                EnableBalloc();
+            }
+            return true;
+        }
+        return false;
+    }
+
+    bool CheckAllocParam(const char* name, bool defaultValue) {
+        if (strcmp(name, "disable") == 0) {
+            return BallocDisabled();
+        }
+        return defaultValue;
+    }
+}
+
+TMallocInfo NMalloc::MallocInfo() {
+    TMallocInfo r;
+
+    r.Name = "balloc";
+    r.SetParam = SetAllocParam;
+    r.CheckParam = CheckAllocParam;
+
+    return r;
+}