libc/test/src/pthread/pthread_attr_test.cpp

   1 //===-- Unittests for pthread_attr_t --------------------------------------===//
   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 "src/pthread/pthread_attr_destroy.h"
  10 #include "src/pthread/pthread_attr_getdetachstate.h"
  11 #include "src/pthread/pthread_attr_getguardsize.h"
  12 #include "src/pthread/pthread_attr_getstack.h"
  13 #include "src/pthread/pthread_attr_getstacksize.h"
  14 #include "src/pthread/pthread_attr_init.h"
  15 #include "src/pthread/pthread_attr_setdetachstate.h"
  16 #include "src/pthread/pthread_attr_setguardsize.h"
  17 #include "src/pthread/pthread_attr_setstack.h"
  18 #include "src/pthread/pthread_attr_setstacksize.h"
  19
  20 #include "test/UnitTest/Test.h"
  21
  22 #include <errno.h>
  23 #include <linux/param.h> // For EXEC_PAGESIZE.
  24 #include <pthread.h>
  25
  26 TEST(LlvmLibcPThreadAttrTest, InitAndDestroy) {
  27   pthread_attr_t attr;
  28   ASSERT_EQ(__llvm_libc::pthread_attr_init(&attr), 0);
  29
  30   int detachstate;
  31   ASSERT_EQ(__llvm_libc::pthread_attr_getdetachstate(&attr, &detachstate), 0);
  32   ASSERT_EQ(detachstate, int(PTHREAD_CREATE_JOINABLE));
  33
  34   size_t guardsize;
  35   ASSERT_EQ(__llvm_libc::pthread_attr_getguardsize(&attr, &guardsize), 0);
  36   ASSERT_EQ(guardsize, size_t(EXEC_PAGESIZE));
  37
  38   ASSERT_EQ(__llvm_libc::pthread_attr_destroy(&attr), 0);
  39 }
  40
  41 TEST(LlvmLibcPThreadattrTest, SetAndGetDetachState) {
  42   pthread_attr_t attr;
  43   ASSERT_EQ(__llvm_libc::pthread_attr_init(&attr), 0);
  44
  45   int detachstate;
  46   __llvm_libc::pthread_attr_getdetachstate(&attr, &detachstate);
  47   ASSERT_EQ(detachstate, int(PTHREAD_CREATE_JOINABLE));
  48   ASSERT_EQ(
  49       __llvm_libc::pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED),
  50       0);
  51   ASSERT_EQ(__llvm_libc::pthread_attr_getdetachstate(&attr, &detachstate), 0);
  52   ASSERT_EQ(detachstate, int(PTHREAD_CREATE_DETACHED));
  53
  54   ASSERT_EQ(__llvm_libc::pthread_attr_setdetachstate(&attr, 0xBAD), EINVAL);
  55
  56   ASSERT_EQ(__llvm_libc::pthread_attr_destroy(&attr), 0);
  57 }
  58
  59 TEST(LlvmLibcPThreadattrTest, SetAndGetGuardSize) {
  60   pthread_attr_t attr;
  61   ASSERT_EQ(__llvm_libc::pthread_attr_init(&attr), 0);
  62
  63   size_t guardsize;
  64   ASSERT_EQ(__llvm_libc::pthread_attr_getguardsize(&attr, &guardsize), 0);
  65   ASSERT_EQ(guardsize, size_t(EXEC_PAGESIZE));
  66   ASSERT_EQ(__llvm_libc::pthread_attr_setguardsize(&attr, 2 * EXEC_PAGESIZE),
  67             0);
  68   ASSERT_EQ(__llvm_libc::pthread_attr_getguardsize(&attr, &guardsize), 0);
  69   ASSERT_EQ(guardsize, size_t(2 * EXEC_PAGESIZE));
  70
  71   ASSERT_EQ(__llvm_libc::pthread_attr_setguardsize(&attr, EXEC_PAGESIZE / 2),
  72             EINVAL);
  73
  74   ASSERT_EQ(__llvm_libc::pthread_attr_destroy(&attr), 0);
  75 }
  76
  77 TEST(LlvmLibcPThreadattrTest, SetAndGetStackSize) {
  78   pthread_attr_t attr;
  79   ASSERT_EQ(__llvm_libc::pthread_attr_init(&attr), 0);
  80
  81   size_t stacksize;
  82   ASSERT_EQ(
  83       __llvm_libc::pthread_attr_setstacksize(&attr, PTHREAD_STACK_MIN << 2), 0);
  84   ASSERT_EQ(__llvm_libc::pthread_attr_getstacksize(&attr, &stacksize), 0);
  85   ASSERT_EQ(stacksize, size_t(PTHREAD_STACK_MIN << 2));
  86
  87   ASSERT_EQ(
  88       __llvm_libc::pthread_attr_setstacksize(&attr, PTHREAD_STACK_MIN / 2),
  89       EINVAL);
  90
  91   ASSERT_EQ(__llvm_libc::pthread_attr_destroy(&attr), 0);
  92 }
  93
  94 TEST(LlvmLibcPThreadattrTest, SetAndGetStack) {
  95   pthread_attr_t attr;
  96   ASSERT_EQ(__llvm_libc::pthread_attr_init(&attr), 0);
  97
  98   void *stack;
  99   size_t stacksize;
 100   ASSERT_EQ(
 101       __llvm_libc::pthread_attr_setstack(&attr, 0, PTHREAD_STACK_MIN << 2), 0);
 102   ASSERT_EQ(__llvm_libc::pthread_attr_getstack(&attr, &stack, &stacksize), 0);
 103   ASSERT_EQ(stacksize, size_t(PTHREAD_STACK_MIN << 2));
 104   ASSERT_EQ(reinterpret_cast<uintptr_t>(stack), uintptr_t(0));
 105
 106   ASSERT_EQ(__llvm_libc::pthread_attr_setstack(
 107                 &attr, reinterpret_cast<void *>(1), PTHREAD_STACK_MIN << 2),
 108             EINVAL);
 109   ASSERT_EQ(__llvm_libc::pthread_attr_setstack(&attr, 0, PTHREAD_STACK_MIN / 2),
 110             EINVAL);
 111
 112   ASSERT_EQ(__llvm_libc::pthread_attr_destroy(&attr), 0);
 113 }