libc/src/pthread/pthread_attr_setstack.cpp

   1 //===-- Implementation of the pthread_attr_setstack -----------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "pthread_attr_setstack.h"
  10 #include "pthread_attr_setstacksize.h"
  11
  12 #include "src/__support/common.h"
  13 #include "src/__support/threads/thread.h" // For STACK_ALIGNMENT
  14
  15 #include <errno.h>
  16 #include <pthread.h>
  17 #include <stdint.h>
  18
  19 namespace LIBC_NAMESPACE {
  20
  21 LLVM_LIBC_FUNCTION(int, pthread_attr_setstack,
  22                    (pthread_attr_t *__restrict attr, void *stack,
  23                     size_t stacksize)) {
  24   uintptr_t stackaddr = reinterpret_cast<uintptr_t>(stack);
  25   // TODO: Do we need to check for overflow on stackaddr + stacksize?
  26   if ((stackaddr % STACK_ALIGNMENT != 0) ||
  27       ((stackaddr + stacksize) % STACK_ALIGNMENT != 0))
  28     return EINVAL;
  29
  30   if (stacksize < PTHREAD_STACK_MIN)
  31     return EINVAL;
  32
  33   attr->__stack = stack;
  34   attr->__stacksize = stacksize;
  35   return 0;
  36 }
  37
  38 } // namespace LIBC_NAMESPACE