Introduce aligned_malloc; drop more YTAlloc dependencies

1 files changed, 18 insertions, 0 deletions

diff --git a/library/cpp/yt/malloc/malloc.cpp b/library/cpp/yt/malloc/malloc.cpp
index 4a35758cab..9a055eaaa0 100644
--- a/library/cpp/yt/malloc/malloc.cpp
+++ b/library/cpp/yt/malloc/malloc.cpp
@@ -1,9 +1,13 @@
 #include "malloc.h"
 
 #include <util/system/compiler.h>
+#include <util/system/platform.h>
+
+#include <stdlib.h>
 
 ////////////////////////////////////////////////////////////////////////////////
 
+#ifndef _win_
 Y_WEAK extern "C" size_t nallocx(size_t size, int /*flags*/) noexcept
 {
     return size;
@@ -13,5 +17,19 @@ Y_WEAK extern "C" size_t malloc_usable_size(void* /*ptr*/) noexcept
 {
     return 0;
 }
+#endif
+
+void* aligned_malloc(size_t size, size_t alignment)
+{
+#if defined(_win_)
+    return _aligned_malloc(size, alignment);
+#elif defined(_darwin_) || defined(_linux_)
+    void* ptr = nullptr;
+    ::posix_memalign(&ptr, alignment, size);
+    return ptr;
+#else
+#   error Unsupported platform
+#endif
+}
 
 ////////////////////////////////////////////////////////////////////////////////