/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright 2025 The FreeBSD Foundation
 *
 * This software was developed by Konstantin Belousov <kib@FreeBSD.org>
 * under sponsorship from the FreeBSD Foundation.
 */

#include <atf-c.h>
#include <err.h>
#include <errno.h>
#include <pthread.h>
#include <pthread_np.h>
#include <stdatomic.h>
#include <stdio.h>
#include <unistd.h>

static atomic_int finish;

static void *
thr_fun(void *arg)
{
        while (atomic_load_explicit(&finish, memory_order_relaxed) != 1)
                sleep(1);
        atomic_store_explicit(&finish, 2, memory_order_relaxed);
        return (arg);
}

ATF_TC(pthread_tryjoin);
ATF_TC_HEAD(pthread_tryjoin, tc)
{
        atf_tc_set_md_var(tc, "descr",
            "Checks pthread_tryjoin(3)");
}

ATF_TC_BODY(pthread_tryjoin, tc)
{
        pthread_t thr;
        void *retval;
        int error, x;

        error = pthread_create(&thr, NULL, thr_fun, &x);
        ATF_REQUIRE_EQ(error, 0);

        error = pthread_tryjoin_np(thr, &retval);
        ATF_REQUIRE_EQ(error, EBUSY);

        atomic_store_explicit(&finish, 1, memory_order_relaxed);
        while (atomic_load_explicit(&finish, memory_order_relaxed) != 2)
                sleep(1);

        error = pthread_tryjoin_np(thr, &retval);
        ATF_REQUIRE_EQ(error, 0);
        ATF_REQUIRE_EQ(retval, &x);
}

ATF_TP_ADD_TCS(tp)
{
        ATF_TP_ADD_TC(tp, pthread_tryjoin);
        return (atf_no_error());
}