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/macros/config.h"
  14 #include "src/__support/threads/thread.h" // For STACK_ALIGNMENT
  15 #include "src/errno/libc_errno.h"
  16
  17 #include <pthread.h>
  18 #include <stdint.h>
  19
  20 namespace LIBC_NAMESPACE_DECL {
  21
  22 LLVM_LIBC_FUNCTION(int, pthread_attr_setstack,
  23                    (pthread_attr_t *__restrict attr, void *stack,
  24                     size_t stacksize)) {
  25   uintptr_t stackaddr = reinterpret_cast<uintptr_t>(stack);
  26   // TODO: Do we need to check for overflow on stackaddr + stacksize?
  27   if ((stackaddr % STACK_ALIGNMENT != 0) ||
  28       ((stackaddr + stacksize) % STACK_ALIGNMENT != 0))
  29     return EINVAL;
  30
  31   if (stacksize < PTHREAD_STACK_MIN)
  32     return EINVAL;
  33
  34   attr->__stack = stack;
  35   attr->__stacksize = stacksize;
  36   return 0;
  37 }
  38
  39 } // namespace LIBC_NAMESPACE_DECL