#include <NetInfo/syslock.h>
#include <string.h>
#ifdef _THREAD_TYPE_PTHREAD_
#include <stdlib.h>
#endif
static thread_id_t
_thread_id(void)
{
#ifdef _THREAD_TYPE_PTHREAD_
return (unsigned int)pthread_self();
#else
return (unsigned int)cthread_self();
#endif
}
static void
_internal_lock(syslock *s)
{
#ifdef _THREAD_TYPE_PTHREAD_
pthread_mutex_lock(s->internal);
#else
spin_lock(&(s->internal));
#endif
}
static void
_internal_unlock(syslock *s)
{
#ifdef _THREAD_TYPE_PTHREAD_
pthread_mutex_unlock(s->internal);
#else
spin_unlock(&(s->internal));
#endif
}
static void
_main_lock(syslock *s)
{
#ifdef _THREAD_TYPE_PTHREAD_
pthread_mutex_lock(s->mutex);
#else
mutex_lock(s->mutex);
#endif
}
static void
_main_unlock(syslock *s)
{
#ifdef _THREAD_TYPE_PTHREAD_
pthread_mutex_unlock(s->mutex);
#else
mutex_unlock(s->mutex);
#endif
}
syslock *
syslock_new(bool_t recursive)
{
syslock *s;
s = (syslock *)malloc(sizeof(syslock));
memset(s, 0, sizeof(syslock));
#ifdef _THREAD_TYPE_PTHREAD_
s->mutex = (pthread_mutex_t *)malloc(sizeof(pthread_mutex_t));
pthread_mutex_init(s->mutex, NULL);
s->internal = (pthread_mutex_t *)malloc(sizeof(pthread_mutex_t));
pthread_mutex_init(s->internal, NULL);
s->condition = (pthread_cond_t *)malloc(sizeof(pthread_cond_t));
pthread_cond_init(s->condition, NULL);
#else
s->mutex = mutex_alloc();
mutex_init(s->mutex);
s->internal = 0;
s->condition = condition_alloc();
condition_init(s->condition);
#endif
s->locked = 0;
s->thread = NO_THREAD;
s->recursive = recursive;
return s;
}
void
syslock_free(syslock *s)
{
if (s == NULL) return;
#ifdef _THREAD_TYPE_PTHREAD_
pthread_mutex_destroy(s->mutex);
free(s->mutex);
pthread_mutex_destroy(s->internal);
free(s->internal);
pthread_cond_destroy(s->condition);
free(s->condition);
#else
mutex_free(s->mutex);
condition_free(s->condition);
#endif
free(s);
}
void
syslock_lock(syslock *s)
{
thread_id_t t;
if (s == NULL) return;
t = _thread_id();
_internal_lock(s);
if ((s->locked > 0) && s->recursive && (s->thread == t))
{
s->locked++;
_internal_unlock(s);
return;
}
_internal_unlock(s);
_main_lock(s);
_internal_lock(s);
s->locked = 1;
s->thread = _thread_id();
_internal_unlock(s);
}
void
syslock_unlock(syslock *s)
{
int unlock_me;
if (s == NULL) return;
unlock_me = 0;
_internal_lock(s);
if (s->locked > 0) s->locked--;
if (s->locked == 0)
{
s->thread = NO_THREAD;
unlock_me = 1;
}
_internal_unlock(s);
if (unlock_me == 1) _main_unlock(s);
}
bool_t
syslock_trylock(syslock *s)
{
int t;
if (s == NULL) return FALSE;
_internal_lock(s);
if ((s->locked > 0) && s->recursive && (s->thread == _thread_id()))
{
s->locked++;
_internal_unlock(s);
return TRUE;
}
_internal_unlock(s);
#ifdef _THREAD_TYPE_PTHREAD_
t = pthread_mutex_trylock(s->mutex);
#else
t = mutex_try_lock(s->mutex);
#endif
if (t != 0) return FALSE;
_internal_lock(s);
s->locked = 1;
s->thread = _thread_id();
_internal_unlock(s);
return TRUE;
}
bool_t
syslock_is_locked(syslock *s)
{
bool_t ret;
if (s == NULL) return FALSE;
ret = FALSE;
_internal_lock(s);
if (s->locked > 0) ret = TRUE;
_internal_unlock(s);
return ret;
}
void
syslock_signal_wait(syslock *s)
{
if (s == NULL) return;
#ifdef _THREAD_TYPE_PTHREAD_
pthread_cond_wait(s->condition, s->mutex);
#else
condition_wait(s->condition, s->mutex);
#endif
}
void
syslock_signal_send(syslock *s)
{
if (s == NULL) return;
#ifdef _THREAD_TYPE_PTHREAD_
pthread_cond_signal(s->condition);
#else
condition_signal(s->condition);
#endif
}
void
syslock_signal_broadcast(syslock *s)
{
if (s == NULL) return;
#ifdef _THREAD_TYPE_PTHREAD_
pthread_cond_broadcast(s->condition);
#else
condition_broadcast(s->condition);
#endif
}