tests/semaphore_test.cpp

   1 /*
   2  * Copyright (C) 2014 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <semaphore.h>
  18
  19 #include <errno.h>
  20 #include <gtest/gtest.h>
  21 #include <limits.h>
  22 #include <pthread.h>
  23 #include <time.h>
  24 #include <unistd.h>
  25
  26 #include "private/bionic_constants.h"
  27
  28 TEST(semaphore, sem_init) {
  29   sem_t s;
  30
  31   // Perfectly fine initial values.
  32   ASSERT_EQ(0, sem_init(&s, 0, 0));
  33   ASSERT_EQ(0, sem_init(&s, 0, 1));
  34   ASSERT_EQ(0, sem_init(&s, 0, 123));
  35
  36   // Too small an initial value.
  37   errno = 0;
  38   ASSERT_EQ(-1, sem_init(&s, 0, -1));
  39   ASSERT_EQ(EINVAL, errno);
  40
  41   ASSERT_EQ(SEM_VALUE_MAX, sysconf(_SC_SEM_VALUE_MAX));
  42
  43   // The largest initial value.
  44   ASSERT_EQ(0, sem_init(&s, 0, SEM_VALUE_MAX));
  45
  46   // Too large an initial value.
  47   errno = 0;
  48   ASSERT_EQ(-1, sem_init(&s, 0, SEM_VALUE_MAX + 1));
  49   ASSERT_EQ(EINVAL, errno);
  50
  51   ASSERT_EQ(0, sem_destroy(&s));
  52 }
  53
  54 TEST(semaphore, sem_trywait) {
  55   sem_t s;
  56   ASSERT_EQ(0, sem_init(&s, 0, 3));
  57   ASSERT_EQ(0, sem_trywait(&s));
  58   ASSERT_EQ(0, sem_trywait(&s));
  59   ASSERT_EQ(0, sem_trywait(&s));
  60   errno = 0;
  61   ASSERT_EQ(-1, sem_trywait(&s));
  62   ASSERT_EQ(EAGAIN, errno);
  63   ASSERT_EQ(0, sem_destroy(&s));
  64 }
  65
  66 static void SemWaitThreadTestFn(sem_t& sem) {
  67   ASSERT_EQ(0, sem_wait(&sem));
  68 }
  69
  70 static void* SemWaitThreadFn(void* arg) {
  71   SemWaitThreadTestFn(*reinterpret_cast<sem_t*>(arg));
  72   return nullptr;
  73 }
  74
  75 TEST(semaphore, sem_wait__sem_post) {
  76   sem_t s;
  77   ASSERT_EQ(0, sem_init(&s, 0, 0));
  78
  79   pthread_t t1, t2, t3;
  80   ASSERT_EQ(0, pthread_create(&t1, NULL, SemWaitThreadFn, &s));
  81   ASSERT_EQ(0, pthread_create(&t2, NULL, SemWaitThreadFn, &s));
  82   ASSERT_EQ(0, pthread_create(&t3, NULL, SemWaitThreadFn, &s));
  83
  84   ASSERT_EQ(0, sem_post(&s));
  85   ASSERT_EQ(0, sem_post(&s));
  86   ASSERT_EQ(0, sem_post(&s));
  87
  88   void* result;
  89   ASSERT_EQ(0, pthread_join(t1, &result));
  90   ASSERT_EQ(0, pthread_join(t2, &result));
  91   ASSERT_EQ(0, pthread_join(t3, &result));
  92 }
  93
  94 static inline void timespec_add_ms(timespec& ts, size_t ms) {
  95   ts.tv_sec  += ms / 1000;
  96   ts.tv_nsec += (ms % 1000) * 1000000;
  97   if (ts.tv_nsec >= NS_PER_S) {
  98     ts.tv_sec++;
  99     ts.tv_nsec -= NS_PER_S;
 100   }
 101 }
 102
 103 TEST(semaphore, sem_timedwait) {
 104   sem_t s;
 105   ASSERT_EQ(0, sem_init(&s, 0, 0));
 106
 107   timespec ts;
 108   ASSERT_EQ(0, clock_gettime(CLOCK_REALTIME, &ts));
 109   timespec_add_ms(ts, 100);
 110
 111   errno = 0;
 112   ASSERT_EQ(-1, sem_timedwait(&s, &ts));
 113   ASSERT_EQ(ETIMEDOUT, errno);
 114
 115   // A negative timeout is an error.
 116   errno = 0;
 117   ts.tv_nsec = -1;
 118   ASSERT_EQ(-1, sem_timedwait(&s, &ts));
 119   ASSERT_EQ(EINVAL, errno);
 120
 121   ASSERT_EQ(0, sem_destroy(&s));
 122 }
 123
 124 TEST(semaphore, sem_getvalue) {
 125   sem_t s;
 126   ASSERT_EQ(0, sem_init(&s, 0, 0));
 127
 128   int i;
 129   ASSERT_EQ(0, sem_getvalue(&s, &i));
 130   ASSERT_EQ(0, i);
 131
 132   ASSERT_EQ(0, sem_post(&s));
 133   ASSERT_EQ(0, sem_getvalue(&s, &i));
 134   ASSERT_EQ(1, i);
 135
 136   ASSERT_EQ(0, sem_post(&s));
 137   ASSERT_EQ(0, sem_getvalue(&s, &i));
 138   ASSERT_EQ(2, i);
 139
 140   ASSERT_EQ(0, sem_wait(&s));
 141   ASSERT_EQ(0, sem_getvalue(&s, &i));
 142   ASSERT_EQ(1, i);
 143 }