/* Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "apr_thread_proc.h"
#include "apr_errno.h"
#include "apr_general.h"
#include "apr_time.h"
#include "testutil.h"

#if APR_HAS_THREADS

static apr_thread_mutex_t *thread_lock;
static apr_thread_once_t *control = NULL;
static int x = 0;
static int value = 0;

static apr_thread_t *t1;
static apr_thread_t *t2;
static apr_thread_t *t3;
static apr_thread_t *t4;

/* just some made up number to check on later */
static apr_status_t exit_ret_val = (APR_OS_START_USERERR + 123);

static void init_func(void)
{
    value++;
}

static void * APR_THREAD_FUNC thread_func1(apr_thread_t *thd, void *data)
{
    apr_status_t rv;
    int i;

    apr_thread_once(control, init_func);

    for (i = 0; i < 10000; i++) {
        rv = apr_thread_mutex_lock(thread_lock);
        if (rv != APR_SUCCESS) apr_thread_exit(thd, rv);
        x++;
        rv = apr_thread_mutex_unlock(thread_lock);
        if (rv != APR_SUCCESS) apr_thread_exit(thd, rv);
    }
    apr_thread_exit(thd, exit_ret_val);
    return NULL;
}

static void thread_init(abts_case *tc, void *data)
{
    apr_status_t rv;

    rv = apr_thread_once_init(&control, p);
    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);

    rv = apr_thread_mutex_create(&thread_lock, APR_THREAD_MUTEX_DEFAULT, p);
    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);
}

static void create_threads(abts_case *tc, void *data)
{
    apr_status_t rv;

    rv = apr_thread_create(&t1, NULL, thread_func1, NULL, p);
    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);
    rv = apr_thread_create(&t2, NULL, thread_func1, NULL, p);
    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);
    rv = apr_thread_create(&t3, NULL, thread_func1, NULL, p);
    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);
    rv = apr_thread_create(&t4, NULL, thread_func1, NULL, p);
    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);
}

static void join_threads(abts_case *tc, void *data)
{
    apr_status_t s;

    apr_thread_join(&s, t1);
    ABTS_INT_EQUAL(tc, exit_ret_val, s);
    apr_thread_join(&s, t2);
    ABTS_INT_EQUAL(tc, exit_ret_val, s);
    apr_thread_join(&s, t3);
    ABTS_INT_EQUAL(tc, exit_ret_val, s);
    apr_thread_join(&s, t4);
    ABTS_INT_EQUAL(tc, exit_ret_val, s);
}

static void check_locks(abts_case *tc, void *data)
{
    ABTS_INT_EQUAL(tc, 40000, x);
}

static void check_thread_once(abts_case *tc, void *data)
{
    ABTS_INT_EQUAL(tc, 1, value);
}

static void thread_name(abts_case *tc, void *data)
{
    apr_status_t rv;
    char *name;

    rv = apr_thread_name_set("thread-1", NULL, p);
    if (rv == APR_ENOTIMPL) {
        ABTS_NOT_IMPL(tc, "apr_thread_name_set is not implemented.")
        return ;
    }

    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);

    rv = apr_thread_name_get(&name, NULL, p);
    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);
    ABTS_STR_EQUAL(tc, "thread-1", name);

    rv = apr_thread_name_set("thread-1", NULL, p);
    if (rv == APR_ENOTIMPL) {
        ABTS_NOT_IMPL(tc, "apr_thread_name_set is not implemented.")
        return ;
    }

    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);

    rv = apr_thread_name_get(&name, NULL, p);
    ABTS_INT_EQUAL(tc, APR_SUCCESS, rv);
    ABTS_STR_EQUAL(tc, "thread-1", name);
}

#else

static void threads_not_impl(abts_case *tc, void *data)
{
    ABTS_NOT_IMPL(tc, "Threads not implemented on this platform");
}

#endif

abts_suite *testthread(abts_suite *suite)
{
    suite = ADD_SUITE(suite)

#if !APR_HAS_THREADS
    abts_run_test(suite, threads_not_impl, NULL);
#else
    abts_run_test(suite, thread_init, NULL);
    abts_run_test(suite, create_threads, NULL);
    abts_run_test(suite, join_threads, NULL);
    abts_run_test(suite, check_locks, NULL);
    abts_run_test(suite, check_thread_once, NULL);
    abts_run_test(suite, thread_name, NULL);
#endif

    return suite;
}