Fix windows build

Author: pnv1

ydb/library/actors/interconnect/rdma/mem_pool.cpp

@@ -30,13 +30,21 @@ struct ibv_mr {
 #include <vector>
 #include <list>
 
-#include <unistd.h>
-#include <sys/syscall.h>
 #include <mutex>
 #include <thread>
 
 #if defined(_linux_)
 #include <sys/mman.h>
+#include <unistd.h>
+#include <sys/syscall.h>
+#endif
+
+#include <cstdlib>
+#include <cstring>
+#include <cerrno>
+
+#if defined(_WIN32)
+#include <malloc.h> // _aligned_malloc, _aligned_free
 #endif
 
 static constexpr size_t HPageSz = (1 << 21);
@@ -45,6 +53,10 @@ using ::NMonitoring::TDynamicCounters;
 
 namespace NInterconnect::NRdma {
 
+    // Cross-platform memory management
+    static void* allocateMemory(size_t size, size_t alignment, bool hp);
+    static void freeMemory(void* ptr) noexcept;
+
     class TChunk: public NNonCopyable::TMoveOnly, public TAtomicRefCount<TChunk> {
     public:
 
@@ -67,7 +79,7 @@ namespace NInterconnect::NRdma {
 #else
         free(MRs.front());
 #endif
-        std::free(addr);
+        freeMemory(addr);
         MRs.clear();
     }
 
@@ -204,26 +216,55 @@ namespace NInterconnect::NRdma {
         );
     }
 
-    void* allocateMemory(size_t size, size_t alignment, bool hp) {
+    static void* allocateMemory(size_t size, size_t alignment, bool hp) {
         if (size % alignment != 0) {
             return nullptr;
         }
-        void* buf = std::aligned_alloc(alignment, size);
+
+        void* buf = nullptr;
+
+    #if defined(_WIN32)
+        // Windows: use _aligned_malloc
+        buf = _aligned_malloc(size, alignment);
+        if (!buf) {
+            fprintf(stderr, "Failed to allocate aligned memory on Windows\n");
+            return nullptr;
+        }
+    #else
+        // POSIX/C++: std::aligned_alloc (C++17)
+        buf = std::aligned_alloc(alignment, size);
+        if (!buf) {
+            fprintf(stderr, "Failed to allocate aligned memory on Unix\n");
+            return nullptr;
+        }
+    #endif
+
         if (hp) {
-#if defined(_linux_)
+    #if defined(_linux_)
             if (madvise(buf, size, MADV_HUGEPAGE) < 0) {
-                fprintf(stderr, "Unable to madvice to use THP, %d (%d)",
-                    strerror(errno), errno);
+                fprintf(stderr, "Unable to madvise to use THP: %s (%d)\n", strerror(errno), errno);
             }
-#endif
+    #endif
             for (size_t i = 0; i < size; i += HPageSz) {
-                // We use THP right now. We need to touch each page to promote it to HUGE.
-                ((char*)buf)[i] = 0;
+                // touch pages to promote to huge pages
+                static_cast<char*>(buf)[i] = 0;
             }
         }
+
         return buf;
     }
 
+    static void freeMemory(void* ptr) noexcept {
+        if (!ptr) {
+            return;
+        }
+    #if defined(_WIN32)
+        _aligned_free(ptr);
+    #else
+        std::free(ptr);
+    #endif
+    }
+
     std::vector<ibv_mr*> registerMemory(void* addr, size_t size, const NInterconnect::NRdma::NLinkMgr::TCtxsMap& ctxs) {
         std::vector<ibv_mr*> res;
 #ifndef MEM_POOL_DISABLE_RDMA_SUPPORT
@@ -307,7 +348,7 @@ namespace NInterconnect::NRdma {
 
             auto mrs = registerMemory(ptr, size, Ctxs);
             if (mrs.empty()) {
-                std::free(ptr);
+                freeMemory(ptr);
                 return nullptr;
             }