01dec70e0a
git-svn-id: https://yap.svn.sf.net/svnroot/yap/trunk@1010 b08c6af1-5177-4d33-ba66-4b1c6b8b522a
452 lines
11 KiB
C
452 lines
11 KiB
C
/*************************************************************************
|
|
* *
|
|
* YAP Prolog *
|
|
* *
|
|
* Yap Prolog was developed at NCCUP - Universidade do Porto *
|
|
* *
|
|
* Copyright L.Damas, V.S.Costa and Universidade do Porto 1985-1997 *
|
|
* *
|
|
**************************************************************************
|
|
* *
|
|
* File: stdpreds.c *
|
|
* Last rev: *
|
|
* mods: *
|
|
* comments: threads *
|
|
* *
|
|
*************************************************************************/
|
|
#ifdef SCCS
|
|
static char SccsId[] = "%W% %G%";
|
|
#endif
|
|
|
|
#include "Yap.h"
|
|
#include "Yatom.h"
|
|
#include "Heap.h"
|
|
#include "eval.h"
|
|
#include "yapio.h"
|
|
#include <stdio.h>
|
|
#if HAVE_STRING_H
|
|
#include <string.h>
|
|
#endif
|
|
|
|
#if THREADS
|
|
|
|
/*
|
|
* This file includes the definition of threads in Yap. Threads
|
|
* are supposed to be compatible with the SWI-Prolog thread package.
|
|
*
|
|
*/
|
|
|
|
static int
|
|
allocate_new_tid(void)
|
|
{
|
|
int new_worker_id = 0;
|
|
LOCK(ThreadHandlesLock);
|
|
while(new_worker_id < MAX_WORKERS &&
|
|
ThreadHandle[new_worker_id].in_use == TRUE)
|
|
new_worker_id++;
|
|
ThreadHandle[new_worker_id].in_use = TRUE;
|
|
UNLOCK(ThreadHandlesLock);
|
|
if (new_worker_id == MAX_WORKERS)
|
|
return -1;
|
|
return new_worker_id;
|
|
}
|
|
|
|
static void
|
|
store_specs(int new_worker_id, UInt ssize, UInt tsize, Term tgoal, Term tdetach)
|
|
{
|
|
ThreadHandle[new_worker_id].ssize = ssize;
|
|
ThreadHandle[new_worker_id].tsize = tsize;
|
|
ThreadHandle[new_worker_id].tgoal =
|
|
Yap_StoreTermInDB(tgoal,4);
|
|
ThreadHandle[new_worker_id].cmod =
|
|
CurrentModule;
|
|
if (IsVarTerm(tdetach)){
|
|
ThreadHandle[new_worker_id].tdetach =
|
|
MkAtomTerm(AtomFalse);
|
|
} else {
|
|
ThreadHandle[new_worker_id].tdetach =
|
|
tdetach;
|
|
}
|
|
}
|
|
|
|
|
|
static void
|
|
thread_die(int wid, int always_die)
|
|
{
|
|
Prop p0;
|
|
|
|
LOCK(ThreadHandlesLock);
|
|
if (ThreadHandle[wid].tdetach == MkAtomTerm(AtomTrue) ||
|
|
always_die) {
|
|
p0 = AbsPredProp(heap_regs->thread_handle[wid].local_preds);
|
|
/* kill all thread local preds */
|
|
while(p0) {
|
|
PredEntry *ap = RepPredProp(p0);
|
|
p0 = ap->NextOfPE;
|
|
Yap_Abolish(ap);
|
|
Yap_FreeCodeSpace((char *)ap);
|
|
}
|
|
Yap_KillStacks(wid);
|
|
heap_regs->wl[wid].active_signals = 0L;
|
|
free(heap_regs->wl[wid].scratchpad.ptr);
|
|
free(ThreadHandle[wid].default_yaam_regs);
|
|
ThreadHandle[wid].in_use = FALSE;
|
|
pthread_mutex_destroy(&(ThreadHandle[wid].tlock));
|
|
}
|
|
UNLOCK(ThreadHandlesLock);
|
|
}
|
|
|
|
static void *
|
|
thread_run(void *widp)
|
|
{
|
|
Term tgoal;
|
|
Term tgs[2];
|
|
int out;
|
|
REGSTORE *standard_regs = (REGSTORE *)malloc(sizeof(REGSTORE));
|
|
int myworker_id = *((int *)widp);
|
|
|
|
/* create the YAAM descriptor */
|
|
ThreadHandle[myworker_id].default_yaam_regs = standard_regs;
|
|
pthread_setspecific(Yap_yaamregs_key, (void *)standard_regs);
|
|
worker_id = myworker_id;
|
|
Yap_InitExStacks(ThreadHandle[myworker_id].ssize, ThreadHandle[myworker_id].tsize);
|
|
CurrentModule = ThreadHandle[myworker_id].cmod;
|
|
Yap_InitYaamRegs();
|
|
{
|
|
Yap_ReleasePreAllocCodeSpace(Yap_PreAllocCodeSpace());
|
|
}
|
|
tgs[0] = Yap_FetchTermFromDB(ThreadHandle[worker_id].tgoal);
|
|
tgs[1] = ThreadHandle[worker_id].tdetach;
|
|
tgoal = Yap_MkApplTerm(FunctorThreadRun, 2, tgs);
|
|
out = Yap_RunTopGoal(tgoal);
|
|
thread_die(worker_id, FALSE);
|
|
return NULL;
|
|
}
|
|
|
|
static Int
|
|
p_thread_new_tid(void)
|
|
{
|
|
return Yap_unify(MkIntegerTerm(allocate_new_tid()), ARG1);
|
|
}
|
|
|
|
static Int
|
|
p_create_thread(void)
|
|
{
|
|
UInt ssize = IntegerOfTerm(Deref(ARG2));
|
|
UInt tsize = IntegerOfTerm(Deref(ARG3));
|
|
/* UInt systemsize = IntegerOfTerm(Deref(ARG4)); */
|
|
Term tgoal = Deref(ARG1);
|
|
Term tdetach = Deref(ARG5);
|
|
int new_worker_id = IntegerOfTerm(Deref(ARG6));
|
|
|
|
if (new_worker_id == -1) {
|
|
/* YAP ERROR */
|
|
return FALSE;
|
|
}
|
|
ThreadHandle[new_worker_id].id = new_worker_id;
|
|
store_specs(new_worker_id, ssize, tsize, tgoal, tdetach);
|
|
pthread_mutex_init(&ThreadHandle[new_worker_id].tlock, NULL);
|
|
if ((ThreadHandle[new_worker_id].ret = pthread_create(&ThreadHandle[new_worker_id].handle, NULL, thread_run, (void *)(&(ThreadHandle[new_worker_id].id)))) == 0) {
|
|
return TRUE;
|
|
}
|
|
/* YAP ERROR */
|
|
return FALSE;
|
|
}
|
|
|
|
static Int
|
|
p_thread_self(void)
|
|
{
|
|
return Yap_unify(MkIntegerTerm(worker_id), ARG1);
|
|
}
|
|
|
|
static Int
|
|
p_thread_join(void)
|
|
{
|
|
Int tid = IntegerOfTerm(Deref(ARG1));
|
|
|
|
LOCK(ThreadHandlesLock);
|
|
if (!ThreadHandle[tid].in_use) {
|
|
UNLOCK(ThreadHandlesLock);
|
|
return FALSE;
|
|
}
|
|
if (!ThreadHandle[tid].tdetach == MkAtomTerm(AtomTrue)) {
|
|
UNLOCK(ThreadHandlesLock);
|
|
return FALSE;
|
|
}
|
|
UNLOCK(ThreadHandlesLock);
|
|
if (pthread_join(ThreadHandle[tid].handle, NULL) < 0) {
|
|
/* ERROR */
|
|
fprintf(stderr, "join error %d %d\n", tid, errno);
|
|
return FALSE;
|
|
}
|
|
fprintf(stderr, "join %d\n", tid);
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_thread_destroy(void)
|
|
{
|
|
Int tid = IntegerOfTerm(Deref(ARG1));
|
|
|
|
thread_die(tid, TRUE);
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_thread_detach(void)
|
|
{
|
|
if (pthread_detach(ThreadHandle[IntegerOfTerm(Deref(ARG1))].handle) < 0) {
|
|
/* ERROR */
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_thread_exit(void)
|
|
{
|
|
thread_die(worker_id, FALSE);
|
|
fprintf(stderr,"here i go %d %ld\n", worker_id, ThreadHandle[worker_id].handle);
|
|
pthread_exit(NULL);
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_thread_set_concurrency(void)
|
|
{
|
|
Term tnew = Deref(ARG2);
|
|
int newc, cur;
|
|
|
|
if (IsVarTerm(tnew)) {
|
|
newc = 0;
|
|
} else if (IsIntegerTerm(tnew)) {
|
|
newc = IntegerOfTerm(tnew);
|
|
} else {
|
|
Yap_Error(TYPE_ERROR_INTEGER,tnew,"thread_set_concurrency/2");
|
|
return(FALSE);
|
|
}
|
|
cur = MkIntegerTerm(pthread_getconcurrency());
|
|
if (pthread_setconcurrency(newc) != 0) {
|
|
return FALSE;
|
|
}
|
|
return Yap_unify(ARG1, MkIntegerTerm(cur));
|
|
}
|
|
|
|
static Int
|
|
p_valid_thread(void)
|
|
{
|
|
Int i = IntegerOfTerm(Deref(ARG1));
|
|
return ThreadHandle[i].in_use;
|
|
}
|
|
|
|
/* Mutex Support */
|
|
|
|
typedef struct swi_mutex {
|
|
UInt owners;
|
|
Int tid_own;
|
|
pthread_mutex_t m;
|
|
} SWIMutex;
|
|
|
|
static Int
|
|
p_new_mutex(void)
|
|
{
|
|
SWIMutex* mutp;
|
|
pthread_mutexattr_t mat;
|
|
|
|
mutp = (SWIMutex *)Yap_AllocCodeSpace(sizeof(SWIMutex));
|
|
if (mutp == NULL) {
|
|
return FALSE;
|
|
}
|
|
pthread_mutexattr_init(&mat);
|
|
#ifdef HAVE_PTHREAD_MUTEXATTR_SETKIND_NP
|
|
pthread_mutexattr_setkind_np(&mat, PTHREAD_MUTEX_RECURSIVE_NP);
|
|
#else
|
|
#ifdef HAVE_PTHREAD_MUTEXATTR_SETTYPE
|
|
pthread_mutexattr_settype(&mat, PTHREAD_MUTEX_RECURSIVE);
|
|
#endif
|
|
#endif
|
|
pthread_mutex_init(&mutp->m, &mat);
|
|
mutp->owners = 0;
|
|
mutp->tid_own = 0;
|
|
return Yap_unify(ARG1, MkIntegerTerm((Int)mutp));
|
|
}
|
|
|
|
static Int
|
|
p_destroy_mutex(void)
|
|
{
|
|
SWIMutex *mut = (SWIMutex*)IntegerOfTerm(Deref(ARG1));
|
|
|
|
if (pthread_mutex_destroy(&mut->m) < 0)
|
|
return FALSE;
|
|
Yap_FreeCodeSpace((void *)mut);
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_lock_mutex(void)
|
|
{
|
|
SWIMutex *mut = (SWIMutex*)IntegerOfTerm(Deref(ARG1));
|
|
|
|
if (pthread_mutex_lock(&mut->m) < 0)
|
|
return FALSE;
|
|
mut->owners++;
|
|
mut->tid_own = worker_id;
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_trylock_mutex(void)
|
|
{
|
|
SWIMutex *mut = (SWIMutex*)IntegerOfTerm(Deref(ARG1));
|
|
|
|
if (pthread_mutex_trylock(&mut->m) == EBUSY)
|
|
return FALSE;
|
|
mut->owners++;
|
|
mut->tid_own = worker_id;
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_unlock_mutex(void)
|
|
{
|
|
SWIMutex *mut = (SWIMutex*)IntegerOfTerm(Deref(ARG1));
|
|
|
|
if (pthread_mutex_unlock(&mut->m) < 0)
|
|
return FALSE;
|
|
mut->owners--;
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_info_mutex(void)
|
|
{
|
|
SWIMutex *mut = (SWIMutex*)IntegerOfTerm(Deref(ARG1));
|
|
|
|
return Yap_unify(ARG2, MkIntegerTerm(mut->owners)) &&
|
|
Yap_unify(ARG2, MkIntegerTerm(mut->tid_own));
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_cond_create(void)
|
|
{
|
|
pthread_cond_t* condp;
|
|
|
|
condp = (pthread_cond_t *)Yap_AllocCodeSpace(sizeof(pthread_cond_t));
|
|
if (condp == NULL) {
|
|
return FALSE;
|
|
}
|
|
pthread_cond_init(condp, NULL);
|
|
return Yap_unify(ARG1, MkIntegerTerm((Int)condp));
|
|
}
|
|
|
|
static Int
|
|
p_cond_destroy(void)
|
|
{
|
|
pthread_cond_t *condp = (pthread_cond_t *)IntegerOfTerm(Deref(ARG1));
|
|
|
|
if (pthread_cond_destroy(condp) < 0)
|
|
return FALSE;
|
|
Yap_FreeCodeSpace((void *)condp);
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_cond_signal(void)
|
|
{
|
|
pthread_cond_t *condp = (pthread_cond_t *)IntegerOfTerm(Deref(ARG1));
|
|
|
|
if (pthread_cond_signal(condp) < 0)
|
|
return FALSE;
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_cond_broadcast(void)
|
|
{
|
|
pthread_cond_t *condp = (pthread_cond_t *)IntegerOfTerm(Deref(ARG1));
|
|
|
|
if (pthread_cond_broadcast(condp) < 0)
|
|
return FALSE;
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_cond_wait(void)
|
|
{
|
|
pthread_cond_t *condp = (pthread_cond_t *)IntegerOfTerm(Deref(ARG1));
|
|
SWIMutex *mut = (SWIMutex*)IntegerOfTerm(Deref(ARG2));
|
|
|
|
pthread_cond_wait(condp, &mut->m);
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
static Int
|
|
p_thread_signal(void)
|
|
{ /* '$thread_signal'(+P) */
|
|
Int wid = IntegerOfTerm(Deref(ARG1));
|
|
/* make sure the lock is available */
|
|
pthread_mutex_lock(&(ThreadHandle[wid].tlock));
|
|
if (!ThreadHandle[wid].in_use) {
|
|
pthread_mutex_unlock(&(ThreadHandle[wid].tlock));
|
|
return TRUE;
|
|
}
|
|
LOCK(heap_regs->wl[wid].signal_lock);
|
|
ThreadHandle[wid].current_yaam_regs->CreepFlag_ = Unsigned(LCL0);
|
|
heap_regs->wl[wid].active_signals |= YAP_ITI_SIGNAL;
|
|
UNLOCK(heap_regs->wl[wid].signal_lock);
|
|
pthread_mutex_unlock(&(ThreadHandle[wid].tlock));
|
|
return TRUE;
|
|
}
|
|
|
|
static Int
|
|
p_no_threads(void)
|
|
{ /* '$thread_signal'(+P) */
|
|
return FALSE;
|
|
}
|
|
|
|
void Yap_InitThreadPreds(void)
|
|
{
|
|
Yap_InitCPred("$no_threads", 0, p_no_threads, 0);
|
|
Yap_InitCPred("$thread_new_tid", 1, p_thread_new_tid, 0);
|
|
Yap_InitCPred("$create_thread", 6, p_create_thread, 0);
|
|
Yap_InitCPred("$thread_self", 1, p_thread_self, SafePredFlag);
|
|
Yap_InitCPred("$thread_join", 1, p_thread_join, 0);
|
|
Yap_InitCPred("$thread_destroy", 1, p_thread_destroy, 0);
|
|
Yap_InitCPred("$detach_thread", 1, p_thread_detach, 0);
|
|
Yap_InitCPred("$thread_exit", 0, p_thread_exit, 0);
|
|
Yap_InitCPred("thread_setconcurrency", 2, p_thread_set_concurrency, 0);
|
|
Yap_InitCPred("$valid_thread", 1, p_valid_thread, 0);
|
|
Yap_InitCPred("$new_mutex", 1, p_new_mutex, SafePredFlag);
|
|
Yap_InitCPred("$destroy_mutex", 1, p_destroy_mutex, SafePredFlag);
|
|
Yap_InitCPred("$lock_mutex", 1, p_lock_mutex, SafePredFlag);
|
|
Yap_InitCPred("$trylock_mutex", 1, p_trylock_mutex, SafePredFlag);
|
|
Yap_InitCPred("$unlock_mutex", 1, p_unlock_mutex, SafePredFlag);
|
|
Yap_InitCPred("$info_mutex", 2, p_info_mutex, SafePredFlag);
|
|
Yap_InitCPred("$cond_create", 1, p_cond_create, SafePredFlag);
|
|
Yap_InitCPred("$cond_destroy", 1, p_cond_destroy, SafePredFlag);
|
|
Yap_InitCPred("$cond_signal", 1, p_cond_signal, SafePredFlag);
|
|
Yap_InitCPred("$cond_broadcast", 1, p_cond_broadcast, SafePredFlag);
|
|
Yap_InitCPred("$cond_wait", 2, p_cond_wait, SafePredFlag);
|
|
Yap_InitCPred("$signal_thread", 1, p_thread_signal, SafePredFlag);
|
|
}
|
|
|
|
#else
|
|
|
|
static Int
|
|
p_no_threads(void)
|
|
{ /* '$thread_signal'(+P) */
|
|
return TRUE;
|
|
}
|
|
|
|
void Yap_InitThreadPreds(void)
|
|
{
|
|
Yap_InitCPred("$no_threads", 0, p_no_threads, 0);
|
|
}
|
|
|
|
|
|
#endif /* THREADS */
|
|
|
|
|