/*      $NetBSD: sem.c,v 1.11 2017/01/13 21:30:42 christos Exp $        */

/*
 * Common code for semaphore tests.  This can be included both into
 * programs using librt and libpthread.
 */

#include <sys/types.h>

#include <rump/rump.h>
#include <rump/rump_syscalls.h>

#include <atf-c.h>
#include <errno.h>
#include <fcntl.h>
#include <pthread.h>
#include <semaphore.h>
#include <sched.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include "h_macros.h"

ATF_TC(postwait);
ATF_TC_HEAD(postwait, tc)
{

        atf_tc_set_md_var(tc, "descr", "tests post and wait from a "
            "single thread (%s)", LIBNAME);
}

ATF_TC_BODY(postwait, tc)
{
        sem_t sem;
        int rv;

        rump_init();

        ATF_REQUIRE_EQ(sem_init(&sem, 1, 0), 0);

        sem_post(&sem);
        sem_post(&sem);

        sem_wait(&sem);
        sem_wait(&sem);
        rv = sem_trywait(&sem);
        ATF_REQUIRE(errno == EAGAIN);
        ATF_REQUIRE(rv == -1);
}

ATF_TC(initvalue);
ATF_TC_HEAD(initvalue, tc)
{

        atf_tc_set_md_var(tc, "descr", "tests initialization with a non-zero "
            "value (%s)", LIBNAME);
}

ATF_TC_BODY(initvalue, tc)
{
        sem_t sem;

        rump_init();
        sem_init(&sem, 1, 4);

        ATF_REQUIRE_EQ(sem_trywait(&sem), 0);
        ATF_REQUIRE_EQ(sem_trywait(&sem), 0);
        ATF_REQUIRE_EQ(sem_trywait(&sem), 0);
        ATF_REQUIRE_EQ(sem_trywait(&sem), 0);
        ATF_REQUIRE_EQ(sem_trywait(&sem), -1);
}

ATF_TC(destroy);
ATF_TC_HEAD(destroy, tc)
{

        atf_tc_set_md_var(tc, "descr", "tests sem_destroy works (%s)", LIBNAME);
}

ATF_TC_BODY(destroy, tc)
{
        sem_t sem;
        int rv, i;

        rump_init();
        for (i = 0; i < 2; i++) {
                sem_init(&sem, 1, 1);

                ATF_REQUIRE_EQ(sem_trywait(&sem), 0);
                ATF_REQUIRE_EQ(sem_trywait(&sem), -1);
                ATF_REQUIRE_EQ(sem_destroy(&sem), 0);
                rv = sem_trywait(&sem);
                ATF_REQUIRE_EQ(errno, EINVAL);
                ATF_REQUIRE_EQ(rv, -1);
        }
}

ATF_TC(busydestroy);
ATF_TC_HEAD(busydestroy, tc)
{

        atf_tc_set_md_var(tc, "descr", "tests sem_destroy report EBUSY for "
            "a busy semaphore (%s)", LIBNAME);
}

static void *
hthread(void *arg)
{
        sem_t *semmarit = arg;

        for (;;) {
                sem_post(&semmarit[2]);
                sem_wait(&semmarit[1]);
                sem_wait(&semmarit[0]);
        }

        return NULL;
}

ATF_TC_BODY(busydestroy, tc)
{
        sem_t semmarit[3];
        pthread_t pt;
        int i;

        /* use a unicpu rump kernel.  this means less chance for race */
        setenv("RUMP_NCPU", "1", 1);

        rump_init();
        sem_init(&semmarit[0], 1, 0);
        sem_init(&semmarit[1], 1, 0);
        sem_init(&semmarit[2], 1, 0);

        pthread_create(&pt, NULL, hthread, semmarit);

        /*
         * Make a best-effort to catch the other thread with its pants down.
         * We can't do this for sure, can we?  Although, we could reach
         * inside the rump kernel and inquire about the thread's sleep
         * status.
         */
        for (i = 0; i < 1000; i++) {
                sem_wait(&semmarit[2]);
                usleep(1);
                if (sem_destroy(&semmarit[1]) == -1)
                        if (errno == EBUSY)
                                break;

                /*
                 * Didn't catch it?  ok, recreate and post to make the
                 * other thread run
                 */
                sem_init(&semmarit[1], 1, 0);
                sem_post(&semmarit[0]);
                sem_post(&semmarit[1]);

        }
        if (i == 1000)
                atf_tc_fail("sem destroy not reporting EBUSY");

        pthread_cancel(pt);
        pthread_join(pt, NULL);
}

ATF_TC(blockwait);
ATF_TC_HEAD(blockwait, tc)
{

        atf_tc_set_md_var(tc, "descr", "tests sem_wait can handle blocking "
            "(%s)", LIBNAME);
        atf_tc_set_md_var(tc, "timeout", "2");
}

ATF_TC_BODY(blockwait, tc)
{
        sem_t semmarit[3];
        pthread_t pt;
        int i;

        rump_init();
        sem_init(&semmarit[0], 1, 0);
        sem_init(&semmarit[1], 1, 0);
        sem_init(&semmarit[2], 1, 0);

        pthread_create(&pt, NULL, hthread, semmarit);

        /*
         * Make a best-effort.  Unless we're extremely unlucky, we should
         * at least one blocking wait.
         */
        for (i = 0; i < 10; i++) {
                sem_wait(&semmarit[2]);
                usleep(1);
                sem_post(&semmarit[0]);
                sem_post(&semmarit[1]);

        }

        pthread_cancel(pt);
        pthread_join(pt, NULL);
}

ATF_TC(blocktimedwait);
ATF_TC_HEAD(blocktimedwait, tc)
{

        atf_tc_set_md_var(tc, "descr", "tests sem_timedwait can handle blocking"
            " (%s)", LIBNAME);
        atf_tc_set_md_var(tc, "timeout", "2");
}

ATF_TC_BODY(blocktimedwait, tc)
{
        sem_t semid;
        struct timespec tp;

        rump_init();

        clock_gettime(CLOCK_REALTIME, &tp);
        tp.tv_nsec += 50000000;
        tp.tv_sec += tp.tv_nsec / 1000000000;
        tp.tv_nsec %= 1000000000;

        ATF_REQUIRE_EQ(sem_init(&semid, 1, 0), 0);
        ATF_REQUIRE_ERRNO(ETIMEDOUT, sem_timedwait(&semid, &tp) == -1);
}

ATF_TC(named);
ATF_TC_HEAD(named, tc)
{

        atf_tc_set_md_var(tc, "descr", "tests named semaphores (%s)", LIBNAME);
}

/*
 * Wow, easy naming rules.  it's these times i'm really happy i can
 * single-step into the kernel.
 */
#define SEM1 "/precious_sem"
#define SEM2 "/justsem"
ATF_TC_BODY(named, tc)
{
        sem_t *sem1, *sem2;
        void *rv;

        rump_init();
        sem1 = sem_open(SEM1, 0);
        ATF_REQUIRE_EQ(errno, ENOENT);
        ATF_REQUIRE_EQ(sem1, NULL);

        sem1 = sem_open(SEM1, O_CREAT, 0444, 1);
        if (sem1 == NULL)
                atf_tc_fail_errno("sem_open O_CREAT");

        rv = sem_open(SEM1, O_CREAT | O_EXCL);
        ATF_REQUIRE_EQ(errno, EEXIST);
        ATF_REQUIRE_EQ(rv, NULL);

        sem2 = sem_open(SEM2, O_CREAT, 0444, 0);
        if (sem2 == NULL)
                atf_tc_fail_errno("sem_open O_CREAT");

        /* check that semaphores are independent */
        ATF_REQUIRE_EQ(sem_trywait(sem2), -1);
        ATF_REQUIRE_EQ(sem_trywait(sem1), 0);
        ATF_REQUIRE_EQ(sem_trywait(sem1), -1);

        /* check that unlinked remains valid */
        sem_unlink(SEM2);
        ATF_REQUIRE_EQ(sem_post(sem2), 0);
        ATF_REQUIRE_EQ(sem_trywait(sem2), 0);
        ATF_REQUIRE_EQ(sem_trywait(sem2), -1);
        ATF_REQUIRE_EQ(errno, EAGAIN);

#if 0 /* see unlink */
        /* close it and check that it's gone */
        if (sem_close(sem2) != 0)
                atf_tc_fail_errno("sem close");
        ATF_REQUIRE_EQ(sem_trywait(sem2), -1);
        ATF_REQUIRE_EQ(errno, EINVAL);
#endif

        /* check that we still have sem1 */
        sem_post(sem1);
        ATF_REQUIRE_EQ(sem_trywait(sem1), 0);
        ATF_REQUIRE_EQ(sem_trywait(sem1), -1);
        ATF_REQUIRE_EQ(errno, EAGAIN);
}

ATF_TC(unlink);
ATF_TC_HEAD(unlink, tc)
{

        /* this is currently broken.  i'll append the PR number soon */
        atf_tc_set_md_var(tc, "descr", "tests unlinked semaphores can be "
            "closed (%s)", LIBNAME);
}

#define SEM "/thesem"
ATF_TC_BODY(unlink, tc)
{
        sem_t *sem;

        rump_init();
        sem = sem_open(SEM, O_CREAT, 0444, 0);
        ATF_REQUIRE(sem);

        if (sem_unlink(SEM) == -1)
                atf_tc_fail_errno("unlink");
        if (sem_close(sem) == -1)
                atf_tc_fail_errno("close unlinked semaphore");
}

/* use rump calls for libpthread _ksem_foo() calls */
#define F1(name, a) int _ksem_##name(a); \
int _ksem_##name(a v1) {return rump_sys__ksem_##name(v1);}
#define F2(name, a, b) int _ksem_##name(a, b); \
int _ksem_##name(a v1, b v2) {return rump_sys__ksem_##name(v1, v2);}
F2(init, unsigned int, intptr_t *);
F1(close, intptr_t);
F1(destroy, intptr_t);
F1(post, intptr_t);
F1(unlink, const char *);
F1(trywait, intptr_t);
F1(wait, intptr_t);
F2(getvalue, intptr_t, unsigned int *);
F2(timedwait, intptr_t, const struct timespec *);
int _ksem_open(const char *, int, mode_t, unsigned int, intptr_t *);
int _ksem_open(const char *a, int b, mode_t c, unsigned int d, intptr_t *e)
    {return rump_sys__ksem_open(a,b,c,d,e);}