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"
 
#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;
 
  // 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*)