stack_posix.c

Go to the documentation of this file.

/* -*- C -*-
 * Serene programming language
 * Copyright (C) 2019-2026 Sameer Rahmani <[email protected]>
 *
 * This library is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#include <stdint.h>
 
#include "serene/rt/fiber.h"
#include "serene/rt/mm/interface.h"
#include "serene/utils.h"
 
// POSIX stack backend, mmap-based allocation with a guard page. On a target without mmap (Windows)
// this file compiles to nothing and a sibling backend should supply srn_fiber_stack_alloc/free
// instead. The selection is by target macro, so the whole runtime source set can be handed to a
// cross compiler and only the matching backend emits symbols.
#if defined(__unix__) || defined(__APPLE__)
 
#  include <sys/mman.h>
 
#  ifdef __APPLE__
#    define STACK_FLAGS MAP_PRIVATE | MAP_ANONYMOUS
#  else
#    define STACK_FLAGS MAP_PRIVATE | MAP_ANONYMOUS | MAP_NORESERVE | MAP_STACK
#  endif
 
// NOLINTBEGIN(hicpp-sign*, performance-no-int*)
srn_fiber_stack_t srn_fiber_stack_alloc(size_t size) {
  const size_t os_page_size = srn_mm_get_os_page_size();
  const size_t desired_size = size ? size : (size_t)SRN_FIBER_DEFAULT_STACK_SIZE;
 
  // Reject sizes the page rounding below would wrap. A wrapped size would
  // silently grant a one page stack instead of failing.
  PANIC_IF(desired_size > SIZE_MAX - (2 * os_page_size), "Requested fiber stack size is too large");
 
  // Round upto a multiple of PAGE size
  const size_t usable = (desired_size + os_page_size - 1) & ~(os_page_size - 1);
  // Add a page to act as a guard page
  const size_t total = usable + os_page_size;
 
  // Since stack grows downward, the starting point where mmap returns will be the end of the stack
  // for us. We could of used MAP_GROWSDOWN to change the behaviour, but that causes the stack to
  // grow, but we have a fix stack.
  void *end = mmap(
    nullptr, total, PROT_READ | PROT_WRITE, STACK_FLAGS,
    // according to the man page, for anonymous mode fd should be -1
    -1,
    // according to the man page, for anonymous mode offset should be 0
    0
  );
 
  PANIC_IF(end == MAP_FAILED, "Failed to allocate the stack for the fiber subsystem");
 
  if (mprotect(end, os_page_size, PROT_NONE) == -1) {
    munmap(end, total);
    PANIC("Failed to allocate the guard page for the fiber subsystem");
  }
 
  const uintptr_t start = (uintptr_t)end + (uintptr_t)total;
  const uintptr_t limit = (uintptr_t)end + (uintptr_t)os_page_size;
 
  srn_fiber_stack_t stack = {.start = (void *)start, .limit = (void *)limit, .guard = end};
  return stack;
}
 
void srn_fiber_stack_free(srn_fiber_stack_t stack) {
  munmap(stack.guard, (uintptr_t)stack.start - (uintptr_t)stack.guard);
}
// NOLINTEND(hicpp-sign*, performance-no-int*)
 
#endif // POSIX stack backend