/* 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.h"

#include "apr_strings.h"

#include "apr_arch_global_mutex.h"

#include "apr_proc_mutex.h"

#include "apr_thread_mutex.h"

#include "apr_portable.h"

static apr_status_t global_mutex_cleanup(void *data)

{

    apr_global_mutex_t *m = (apr_global_mutex_t *)data;

    apr_status_t rv;

    rv = apr_proc_mutex_destroy(m->proc_mutex);

#if APR_HAS_THREADS

    if (m->thread_mutex) {

        if (rv != APR_SUCCESS) {

            (void)apr_thread_mutex_destroy(m->thread_mutex);

        }

        else {

            rv = apr_thread_mutex_destroy(m->thread_mutex);

        }

    }

#endif /* APR_HAS_THREADS */

    return rv;

}

APR_DECLARE(apr_status_t) apr_global_mutex_create(apr_global_mutex_t **mutex,

                                                  const char *fname,

                                                  apr_lockmech_e mech,

                                                  apr_pool_t *pool)

{

    apr_status_t rv;

    apr_global_mutex_t *m;

    m = (apr_global_mutex_t *)apr_palloc(pool, sizeof(*m));

    m->pool = pool;

    rv = apr_proc_mutex_create(&m->proc_mutex, fname, mech, m->pool);

    if (rv != APR_SUCCESS) {

        return rv;

    }

#if APR_HAS_THREADS

    if (m->proc_mutex->meth->flags & APR_PROCESS_LOCK_MECH_IS_GLOBAL) {

        m->thread_mutex = NULL; /* We don't need a thread lock. */

    }

    else {

        rv = apr_thread_mutex_create(&m->thread_mutex,

                                     APR_THREAD_MUTEX_DEFAULT, m->pool);

        if (rv != APR_SUCCESS) {

            rv = apr_proc_mutex_destroy(m->proc_mutex);

            return rv;

        }

    }

#endif /* APR_HAS_THREADS */

    apr_pool_cleanup_register(m->pool, (void *)m,

                              global_mutex_cleanup, apr_pool_cleanup_null);

    *mutex = m;

    return APR_SUCCESS;

}

APR_DECLARE(apr_status_t) apr_global_mutex_child_init(

                              apr_global_mutex_t **mutex,

                              const char *fname,

                              apr_pool_t *pool)

{

    apr_status_t rv;

    rv = apr_proc_mutex_child_init(&((*mutex)->proc_mutex), fname, pool);

    return rv;

}

APR_DECLARE(apr_status_t) apr_global_mutex_lock(apr_global_mutex_t *mutex)

{

    apr_status_t rv;

#if APR_HAS_THREADS

    if (mutex->thread_mutex) {

        rv = apr_thread_mutex_lock(mutex->thread_mutex);

        if (rv != APR_SUCCESS) {

            return rv;

        }

    }

#endif /* APR_HAS_THREADS */

    rv = apr_proc_mutex_lock(mutex->proc_mutex);

#if APR_HAS_THREADS

    if (rv != APR_SUCCESS) {

        if (mutex->thread_mutex) {

            (void)apr_thread_mutex_unlock(mutex->thread_mutex);

        }

    }

#endif /* APR_HAS_THREADS */

    return rv;

}

APR_DECLARE(apr_status_t) apr_global_mutex_trylock(apr_global_mutex_t *mutex)

{

    apr_status_t rv;

#if APR_HAS_THREADS

    if (mutex->thread_mutex) {

        rv = apr_thread_mutex_trylock(mutex->thread_mutex);

        if (rv != APR_SUCCESS) {

            return rv;

        }

    }

#endif /* APR_HAS_THREADS */

    rv = apr_proc_mutex_trylock(mutex->proc_mutex);

#if APR_HAS_THREADS

    if (rv != APR_SUCCESS) {

        if (mutex->thread_mutex) {

            (void)apr_thread_mutex_unlock(mutex->thread_mutex);

        }

    }

#endif /* APR_HAS_THREADS */

    return rv;

}

APR_DECLARE(apr_status_t) apr_global_mutex_timedlock(apr_global_mutex_t *mutex,

                                                 apr_interval_time_t timeout)

{

    apr_status_t rv;

#if APR_HAS_THREADS

    if (mutex->thread_mutex) {

        apr_time_t expiry = 0;

        if (timeout > 0) {

            expiry = apr_time_now() + timeout;

        }

        rv = apr_thread_mutex_timedlock(mutex->thread_mutex, timeout);

        if (rv != APR_SUCCESS) {

            return rv;

        }

        if (expiry) {

            timeout = expiry - apr_time_now();

            if (timeout < 0) {

                timeout = 0;

            }

        }

    }

#endif /* APR_HAS_THREADS */

    rv = apr_proc_mutex_timedlock(mutex->proc_mutex, timeout);

#if APR_HAS_THREADS

    if (rv != APR_SUCCESS) {

        if (mutex->thread_mutex) {

            (void)apr_thread_mutex_unlock(mutex->thread_mutex);

        }

    }

#endif /* APR_HAS_THREADS */

    return rv;

}

APR_DECLARE(apr_status_t) apr_global_mutex_unlock(apr_global_mutex_t *mutex)

{

    apr_status_t rv;

    rv = apr_proc_mutex_unlock(mutex->proc_mutex);

#if APR_HAS_THREADS

    if (mutex->thread_mutex) {

        if (rv != APR_SUCCESS) {

            (void)apr_thread_mutex_unlock(mutex->thread_mutex);

        }

        else {

            rv = apr_thread_mutex_unlock(mutex->thread_mutex);

        }

    }

#endif /* APR_HAS_THREADS */

    return rv;

}

APR_DECLARE(apr_status_t) apr_os_global_mutex_get(apr_os_global_mutex_t *ospmutex,

                                                apr_global_mutex_t *pmutex)

{

    ospmutex->pool = pmutex->pool;

    ospmutex->proc_mutex = pmutex->proc_mutex;

#if APR_HAS_THREADS

    ospmutex->thread_mutex = pmutex->thread_mutex;

#endif

    return APR_SUCCESS;

}

APR_DECLARE(apr_status_t) apr_global_mutex_destroy(apr_global_mutex_t *mutex)

{

    return apr_pool_cleanup_run(mutex->pool, mutex, global_mutex_cleanup);

}

APR_DECLARE(const char *) apr_global_mutex_lockfile(apr_global_mutex_t *mutex)

{

    return apr_proc_mutex_lockfile(mutex->proc_mutex);

}

APR_DECLARE(apr_lockmech_e) apr_global_mutex_mech(apr_global_mutex_t *mutex)

{

    return apr_proc_mutex_mech(mutex->proc_mutex);

}

APR_DECLARE(const char *) apr_global_mutex_name(apr_global_mutex_t *mutex)

{

    return apr_proc_mutex_name(mutex->proc_mutex);

}

APR_PERMS_SET_IMPLEMENT(global_mutex)

{

    apr_status_t rv;

    apr_global_mutex_t *mutex = (apr_global_mutex_t *)theglobal_mutex;

    rv = APR_PERMS_SET_FN(proc_mutex)(mutex->proc_mutex, perms, uid, gid);

    return rv;

}

APR_POOL_IMPLEMENT_ACCESSOR(global_mutex)