Fix memory leak when not using guard pages

STEllAR-GROUP · Jul 30, 2023 · 498c7c8 · 498c7c8
1 parent d3f50ae
commit 498c7c8
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/libs/core/coroutines/include/hpx/coroutines/detail/posix_utility.hpp b/libs/core/coroutines/include/hpx/coroutines/detail/posix_utility.hpp
@@ -83,16 +83,21 @@ namespace hpx::threads::coroutines::detail::posix {
 
     inline void* alloc_stack(std::size_t size)
     {
-        void* real_stack =
-            ::mmap(nullptr, size + EXEC_PAGESIZE, PROT_READ | PROT_WRITE,
+#if defined(HPX_HAVE_THREAD_GUARD_PAGE)
+        if (use_guard_pages)
+        {
+            size += EXEC_PAGESIZE;
+        }
+#endif
+        void* real_stack = ::mmap(nullptr, size, PROT_READ | PROT_WRITE,
 #if defined(__APPLE__)
-                MAP_PRIVATE | MAP_ANON | MAP_NORESERVE,
+            MAP_PRIVATE | MAP_ANON | MAP_NORESERVE,
 #elif defined(__FreeBSD__)
-                MAP_PRIVATE | MAP_ANON,
+            MAP_PRIVATE | MAP_ANON,
 #else
-                MAP_PRIVATE | MAP_ANONYMOUS | MAP_NORESERVE,
+            MAP_PRIVATE | MAP_ANONYMOUS | MAP_NORESERVE,
 #endif
-                -1, 0);
+            -1, 0);
 
         if (real_stack == MAP_FAILED)
         {
@@ -112,7 +117,7 @@ namespace hpx::threads::coroutines::detail::posix {
 #if defined(HPX_HAVE_THREAD_GUARD_PAGE)
         if (use_guard_pages)
         {
-            // Add a guard page.
+            // Set the guard page.
             ::mprotect(real_stack, EXEC_PAGESIZE, PROT_NONE);
 
             void** stack = static_cast<void**>(real_stack) +