Make DontSwapMemoryRAII work on windows

src/cpp-utils/CMakeLists.txt

@@ -42,7 +42,8 @@ set(SOURCES
         assert/AssertFailed.cpp
         system/get_total_memory.cpp
         system/homedir.cpp
-        system/memory.cpp
+        system/memory_nonwindows.cpp
+        system/memory_windows.cpp
 )
 
 add_library(${PROJECT_NAME} STATIC ${SOURCES})

src/cpp-utils/system/memory.h

@@ -10,11 +10,11 @@ namespace cpputils {
 // i.e. tells the operating system not to swap it.
 class DontSwapMemoryRAII final {
 public:
-    DontSwapMemoryRAII(const void* addr, size_t len);
+    DontSwapMemoryRAII(void* addr, size_t len);
     ~DontSwapMemoryRAII();
 
 private:
-    const void* addr_;
+    void* const addr_;
     const size_t len_;
 };
 

src/cpp-utils/system/memory_nonwindows.cpp

@@ -0,0 +1,30 @@
+#if !defined(_MSC_VER)
+
+#include "memory.h"
+#include <sys/mman.h>
+#include <errno.h>
+#include <stdexcept>
+#include <cpp-utils/logging/logging.h>
+
+using namespace cpputils::logging;
+
+namespace cpputils {
+
+DontSwapMemoryRAII::DontSwapMemoryRAII(void *addr, size_t len)
+: addr_(addr), len_(len) {
+    const int result = ::mlock(addr_, len_);
+    if (0 != result) {
+        throw std::runtime_error("Error calling mlock. Errno: " + std::to_string(errno));
+    }
+}
+
+DontSwapMemoryRAII::~DontSwapMemoryRAII() {
+    const int result = ::munlock(addr_, len_);
+    if (0 != result) {
+        LOG(WARN, "Error calling munlock. Errno: {}", errno);
+    }
+}
+
+}
+
+#endif

src/cpp-utils/system/memory_windows.cpp

@@ -0,0 +1,29 @@
+#if defined(_MSC_VER)
+
+#include "memory.h"
+#include <Windows.h>
+#include <stdexcept>
+#include <cpp-utils/logging/logging.h>
+
+using namespace cpputils::logging;
+
+namespace cpputils {
+
+DontSwapMemoryRAII::DontSwapMemoryRAII(void *addr, size_t len)
+: addr_(addr), len_(len) {
+	const BOOL result = ::VirtualLock(addr_, len_);
+    if (!result) {
+        throw std::runtime_error("Error calling VirtualLock. Errno: " + std::to_string(GetLastError()));
+    }
+}
+
+DontSwapMemoryRAII::~DontSwapMemoryRAII() {
+	const BOOL result = ::VirtualUnlock(addr_, len_);
+    if (!result) {
+        LOG(WARN, "Error calling VirtualUnlock. Errno: {}", GetLastError());
+    }
+}
+
+}
+
+#endif