Home | History | Annotate | Download | only in tests
      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 }
    144