xref: /aosp_15_r20/external/llvm-libc/src/pthread/pthread_attr_setstack.cpp (revision 71db0c75aadcf003ffe3238005f61d7618a3fead)
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
40