/*
- * lwp.c -- lightweight process creation, destruction and manipulation.
- * Copyright (C) 1991-3 Stephen Crane.
+ * Empire - A multi-player, client/server Internet based war game.
+ * Copyright (C) 1994-2008, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ * Ken Stevens, Steve McClure
+ * Copyright (C) 1991-3 Stephen Crane
*
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Library General Public
- * License as published by the Free Software Foundation; either
- * version 2 of the License, or (at your option) any later version.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Library General Public License for more details.
- *
- * You should have received a copy of the GNU Library General Public
- * License along with this library; if not, write to the Free
- * Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
- * author: Stephen Crane, (jsc@doc.ic.ac.uk), Department of Computing,
- * Imperial College of Science, Technology and Medicine, 180 Queen's
- * Gate, London SW7 2BZ, England.
+ * ---
+ *
+ * See files README, COPYING and CREDITS in the root of the source
+ * tree for related information and legal notices. It is expected
+ * that future projects/authors will amend these files as needed.
+ *
+ * ---
+ *
+ * lwp.c: lightweight process creation, destruction and manipulation
+ *
+ * Known contributors to this file:
+ * Markus Armbruster, 2004-2007
*/
-#include <stdio.h>
-#include <signal.h>
-#include <stdlib.h>
+#include <config.h>
+
#include <string.h>
#include "lwp.h"
#include "lwpint.h"
#include "prototypes.h"
-#if defined(_EMPTH_LWP)
-
-#ifdef BOUNDS_CHECK
-#include <bounds/fix-args.h>
-#include <bounds/unchecked.h>
-#endif
-
struct lwpQueue LwpSchedQ[LWP_MAX_PRIO], LwpDeadQ;
struct lwpProc *LwpCurrent = NULL;
-char **LwpContextPtr;
+void **LwpContextPtr;
int LwpMaxpri = 0; /* maximum priority so far */
+int LwpStackGrowsDown;
-#ifdef POSIXSIGNALS
-static sigset_t oldmask;
-#else /* POSIXSIGNALS */
-static int oldmask;
-#endif /* POSIXSIGNALS */
-
-/* for systems without strdup */
-#ifdef NOSTRDUP
-extern char *strdup();
-#endif /* NOSTRDUP */
-
+static void lwpDestroy(struct lwpProc *proc);
static void lwpStackCheckInit(struct lwpProc *newp);
-static int lwpStackCheck(struct lwpProc *newp);
+static void lwpStackCheck(struct lwpProc *newp);
static void lwpStackCheckUsed(struct lwpProc *newp);
/* check stack direction */
growsdown(void *x)
{
int y;
-
-#ifdef BOUNDS_CHECK
- BOUNDS_CHECKING_OFF;
-#endif
y = (x > (void *)&y);
-
-#ifdef BOUNDS_CHECK
- BOUNDS_CHECKING_ON;
-#endif
-
return y;
}
void
lwpReschedule(void)
{
- static int lcount = LCOUNT;
static struct lwpProc *nextp;
static int i;
-#ifdef POSIXSIGNALS
- static sigset_t tmask;
-#endif /* POSIXSIGNALS */
if (LwpCurrent && (LwpCurrent->flags & LWP_STACKCHECK)) {
lwpStackCheck(LwpCurrent);
}
- if (!--lcount) {
- int p = lwpSetPriority(LWP_MAX_PRIO - 1);
- lcount = LCOUNT;
-#ifdef POSIXSIGNALS
- sigprocmask(SIG_SETMASK, &oldmask, &tmask);
- sigprocmask(SIG_SETMASK, &tmask, &oldmask);
-#else /* POSIXSIGNALS */
- sigsetmask(sigsetmask(oldmask));
-#endif /* POSIXSIGNALS */
- LwpCurrent->pri = p;
- }
+
+ lwpSigWakeup();
+ lwpWakeupSleep();
/* destroy dead threads */
lwpStatus(LwpCurrent, "Cleaning dead queue");
while (NULL != (nextp = lwpGetFirst(&LwpDeadQ))) {
- if (nextp == LwpCurrent) {
- lwpStatus(nextp, "OOOPS, we are running already dead thread");
- exit(1);
- }
+ if (CANT_HAPPEN(nextp == LwpCurrent))
+ abort();
lwpDestroy(nextp);
lwpStatus(LwpCurrent, "Destroying done");
}
lwpAddTail(&LwpDeadQ, nextp);
} else {
lwpDestroy(nextp);
-/* fprintf(stderr, "Destroying done\n"); */
}
nextp = 0;
}
if (nextp)
break;
}
- if (LwpCurrent == 0 && nextp == 0) {
- fprintf(stderr, "No processes to run!\n");
- exit(1);
- }
- if (LwpCurrent)
- lwpStatus(LwpCurrent, "switch out");
- /* do context switch */
-#ifdef BOUNDS_CHECK
- BOUNDS_CHECKING_OFF;
-#endif
-
-#if defined(hpc)
- {
- int endpoint;
-
- endpoint = &endpoint;
- if (initcontext == NULL || endpoint < startpoint) {
- i = lwpSave(LwpCurrent->context);
- } else {
- LwpCurrent->size = endpoint - startpoint;
- LwpCurrent->sbtm = realloc(LwpCurrent->sbtm, LwpCurrent->size);
- memcpy(LwpCurrent->sbtm, startpoint, LwpCurrent->size);
- if (i = lwpSave(LwpCurrent->context)) {
- memcpy(startpoint, LwpCurrent->sbtm, LwpCurrent->size);
- i = 1;
- }
- }
- }
-#else
- i = lwpSave(LwpCurrent->context);
-#endif
-#ifdef BOUNDS_CHECK
- BOUNDS_CHECKING_ON;
-#endif
-
- if (LwpCurrent != nextp && !(LwpCurrent && i)) {
- /* restore previous context */
- lwpStatus(nextp, "switch in", nextp->pri);
+ if (CANT_HAPPEN(LwpCurrent == 0 && nextp == 0))
+ abort();
+ if (LwpCurrent != nextp) {
+ struct lwpProc *oldp = LwpCurrent;
+ if (oldp)
+ lwpStatus(oldp, "switch out");
LwpCurrent = nextp;
- *LwpContextPtr = LwpCurrent->ud;
-#ifdef BOUNDS_CHECK
- BOUNDS_CHECKING_OFF;
-#endif
- lwpRestore(LwpCurrent->context);
-
-#ifdef BOUNDS_CHECK
- BOUNDS_CHECKING_ON;
-#endif
+ *LwpContextPtr = nextp->ud;
+ lwpSwitchContext(oldp, nextp);
+ lwpStatus(nextp, "switch in %d", nextp->pri);
}
}
void
lwpEntryPoint(void)
{
-#ifdef POSIXSIGNALS
- sigset_t set;
-#endif /* POSIXSIGNALS */
-
-#ifdef BOUNDS_CHECK
- BOUNDS_CHECKING_OFF;
-#endif
-#ifdef POSIXSIGNALS
- sigemptyset(&set);
- sigaddset(&set, SIGALRM);
- sigprocmask(SIG_SETMASK, &set, &oldmask);
-#else /* POSIXSIGNALS */
- sigsetmask(SIGNALS);
-#endif /* POSIXSIGNALS */
- *LwpContextPtr = LwpCurrent->ud;
-
lwpStatus(LwpCurrent, "starting at entry point");
(*LwpCurrent->entry)(LwpCurrent->ud);
lwpExit();
-#ifdef BOUNDS_CHECK
- BOUNDS_CHECKING_ON;
-#endif
-
-
}
/*
* lwpCreate -- create a process.
*/
struct lwpProc *
-lwpCreate(int priority, void (*entry)(void *), int size, int flags, char *name, char *desc, int argc, char **argv, void *ud)
+lwpCreate(int priority, void (*entry)(void *), int stacksz,
+ int flags, char *name, int argc, char **argv, void *ud)
{
struct lwpProc *newp;
- int *s, x;
-#ifdef UCONTEXT
- stack_t sp;
-#else /* UCONTEXT */
- void *sp;
-#endif /* UCONTEXT */
- unsigned long stackp;
-
- if (!(newp = (struct lwpProc *)malloc(sizeof(struct lwpProc))))
- return (0);
- if (flags & LWP_STACKCHECK) {
- /* Add a 1K buffer on each side of the stack */
- size += 2 * LWP_REDZONE;
- }
- size += LWP_EXTRASTACK;
- size += sizeof(stkalign_t);
- if (!(s = (int *)malloc(size)))
- return (0);
+
+ if (!(newp = malloc(sizeof(struct lwpProc))))
+ return NULL;
newp->flags = flags;
newp->name = strdup(name);
- newp->desc = strdup(desc);
newp->entry = entry;
newp->argc = argc;
newp->argv = argv;
newp->ud = ud;
- if ((newp->flags & LWP_STACKCHECK) == 0) {
- stackp = growsdown((void *)&x) ?
- (((long)s) + size - sizeof(stkalign_t) - LWP_EXTRASTACK) :
- (long)s + LWP_EXTRASTACK;
-#ifdef UCONTEXT
- sp.ss_sp = (void *)(stackp & -sizeof(stkalign_t));
- sp.ss_size = size;
- sp.ss_flags = 0;
-#else /* UCONTEXT */
- sp = (void *)(stackp & -sizeof(stkalign_t));
-#endif /* UCONTEXT */
- } else {
- if (growsdown(&x)) {
- /* round address off to stkalign_t */
- stackp = ((long)s) + size - LWP_REDZONE -
- LWP_EXTRASTACK - sizeof(stkalign_t);
-#ifdef UCONTEXT
- sp.ss_sp = (void *)(stackp & -sizeof(stkalign_t));
- sp.ss_size = size;
- sp.ss_flags = 0;
- newp->lowmark = (void *)(((long)sp.ss_sp) + LWP_EXTRASTACK);
-#else /* UCONTEXT */
- sp = (void *)(stackp & -sizeof(stkalign_t));
- newp->lowmark = (void *)(((long)sp) + LWP_EXTRASTACK);
-#endif /* UCONTEXT */
- newp->himark = s;
- } else {
- stackp = ((long)s) + LWP_REDZONE + LWP_EXTRASTACK;
-#ifdef UCONTEXT
- sp.ss_sp = (void *)(((long)stackp) & -sizeof(stkalign_t));
- sp.ss_size = size;
- sp.ss_flags = 0;
-#else /* UCONTEXT */
- sp = (void *)(((long)stackp) & -sizeof(stkalign_t));
-#endif /* UCONTEXT */
- newp->lowmark = (void *)s;
- newp->himark = (void *)(((long)s) + size - LWP_REDZONE);
- }
- }
+ newp->dead = 0;
+ newp->runtime = (time_t)-1;
+ newp->fd = -1;
if (LWP_MAX_PRIO <= priority)
priority = LWP_MAX_PRIO - 1;
if (LwpMaxpri < (newp->pri = priority))
LwpMaxpri = priority;
- newp->sbtm = (void *)s;
- newp->size = size;
- newp->dead = 0;
+ if (lwpNewContext(newp, stacksz) < 0) {
+ free(newp->name);
+ free(newp);
+ return NULL;
+ }
+ lwpStatus(newp, "creating process structure %p (sbtm %p)",
+ newp, newp->sbtm);
if (flags & LWP_STACKCHECK)
lwpStackCheckInit(newp);
- lwpStatus(newp, "creating process structure sbtm: %d",
- (int)newp->sbtm);
lwpReady(newp);
lwpReady(LwpCurrent);
-#ifdef UCONTEXT
- lwpInitContext(newp, &sp); /* architecture-dependent: from arch.c */
-#else /* UCONTEXT */
- lwpInitContext(newp, sp); /* architecture-dependent: from arch.c */
-#endif /* UCONTEXT */
lwpReschedule();
- return (newp);
+ return newp;
}
-void
+static void
lwpDestroy(struct lwpProc *proc)
{
if (proc->flags & LWP_STACKCHECK) {
lwpStackCheckUsed(proc);
lwpStackCheck(proc);
}
- lwpStatus(proc, "destroying sbtm: %d", (int)proc->sbtm);
- proc->entry = 0;
- proc->ud = 0;
- proc->argv = 0;
- free((char *)proc->sbtm);
+ lwpStatus(proc, "destroying sbtm: %p", proc->sbtm);
+ free(proc->sbtm);
free(proc->name);
- free(proc->desc);
- proc->name = 0;
- proc->desc = 0;
- proc->sbtm = 0;
- proc->lowmark = 0;
- proc->himark = 0;
- free((char *)proc);
+ free(proc);
}
/*
{
if (!p)
p = LwpCurrent;
- return (p->ud);
+ return p->ud;
}
/*
p->ud = ud;
}
-/*
- * set name & desc
- */
-void
-lwpSetDesc(struct lwpProc *p, char *name, char *desc)
-{
- if (!p)
- p = LwpCurrent;
- free(p->name);
- free(p->desc);
- p->name = strdup(name);
- p->desc = strdup(desc);
-}
-
/*
* lwpYield -- yield the processor to another thread.
*/
{
lwpStatus(p, "terminating process");
p->dead = 1;
- if (p->fd >= 0)
- lwpWakeupFd(p);
+ lwpWakeup(p);
}
/*
lwpStatus(LwpCurrent, "resetting priority (%d -> %d)", old, new);
if (new < old)
lwpYield();
- return (old);
+ return old;
}
/*
* initialise the coroutine structures
*/
struct lwpProc *
-lwpInitSystem(int pri, char **ctxptr, int flags)
+lwpInitSystem(int pri, void **ctxptr, int flags, sigset_t *waitset)
{
struct lwpQueue *q;
- int i, *stack;
+ int i, *stack, marker;
struct lwpProc *sel;
LwpContextPtr = ctxptr;
if (pri < 1)
pri = 1;
- /* *LwpContextPtr = 0; */
+ LwpStackGrowsDown = growsdown(&marker);
if (!(LwpCurrent = calloc(1, sizeof(struct lwpProc))))
- return (0);
- if (!(stack = (int *)malloc(64)))
- return (0);
+ return NULL;
+ if (!(stack = malloc(64)))
+ return NULL;
if (LWP_MAX_PRIO <= pri)
pri = LWP_MAX_PRIO - 1;
if (LwpMaxpri < pri)
LwpCurrent->sbtm = stack; /* dummy stack for "main" */
LwpCurrent->pri = pri;
LwpCurrent->dead = 0;
- LwpCurrent->flags = flags;
+ LwpCurrent->flags = flags & ~LWP_STACKCHECK;
LwpCurrent->name = "Main";
for (i = LWP_MAX_PRIO, q = LwpSchedQ; i--; q++)
q->head = q->tail = 0;
LwpDeadQ.head = LwpDeadQ.tail = 0;
+ lwpInitSigWait(waitset);
/* must be lower in priority than us for this to work right */
- sel = lwpCreate(0, lwpSelect, 16384, flags, "EventHandler",
- "Select (main loop) Event Handler", 0, 0, 0);
+ sel = lwpCreate(0, lwpSelect, 16384, flags, "EventHandler", 0, 0, 0);
lwpInitSelect(sel);
- return (LwpCurrent);
+ return LwpCurrent;
}
/* lwpStackCheckInit
static void
lwpStackCheckInit(struct lwpProc *newp)
{
- register int i;
- register long *lp;
-
- int lim = newp->size / sizeof(long);
- if (!newp || !newp->sbtm)
- return;
- for (lp = newp->sbtm, i = 0; i < lim; i++, lp++) {
- *lp = LWP_CHECKMARK;
- }
+ int *p;
+ int *lim = (int *)((char *)newp->sbtm + newp->size);
+
+ for (p = newp->sbtm; p < lim; p++)
+ *p = LWP_CHECKMARK;
}
/* lwpStackCheck
*
* Check if the thread has overflowed/underflowed its stack.
- * NOTE:
- * If an problem occurs, it is not corrected.
- * The buffer is not cleaned up, nor is the thread terminated.
- * Cleaning up the buffer would be a mistake, and terminating
- * the thread, well, could be done. Should more like take
- * down the entire process.
+ * Should that happen, abort the process, as we cannot recover.
*/
-static int
+static void
lwpStackCheck(struct lwpProc *newp)
{
- register int end, amt;
- register unsigned int i;
- register long *lp;
- register int growsDown;
- int marker;
-
- if (!newp || !newp->himark || !newp->lowmark)
- return (1);
- growsDown = growsdown(&marker);
- for (lp = newp->himark, i = 0; i < LWP_REDZONE / sizeof(long);
- i++, lp++) {
- if (*lp == LWP_CHECKMARK)
- continue;
- /* Stack overflow. */
- if (growsDown) {
- end = i;
- while (i < LWP_REDZONE / sizeof(long)) {
- if (*lp++ != LWP_CHECKMARK)
- end = i;
- i++;
- }
- amt = (end + 1) * sizeof(long);
- } else {
- amt = (i + 1) * sizeof(long);
- }
- lwpStatus(newp, "Thread stack overflowed %d bytes (of %u)",
- amt, newp->size - 2 * LWP_REDZONE - sizeof(stkalign_t));
- return (0);
- }
- for (lp = newp->lowmark, i = 0; i < LWP_REDZONE / sizeof(long);
- i++, lp++) {
- if (*lp == LWP_CHECKMARK)
- continue;
- /* Stack underflow. */
- if (growsDown) {
- end = i;
- while (i < LWP_REDZONE / sizeof(long)) {
- if (*lp++ != LWP_CHECKMARK)
- end = i;
- i++;
- }
- amt = (end + 1) * sizeof(long);
- } else {
- amt = (LWP_REDZONE - i + 1) * sizeof(long);
- }
- lwpStatus(newp, "Thread stack underflow %d bytes (of %u)",
- amt, newp->size - 2 * LWP_REDZONE - sizeof(stkalign_t));
- return (0);
+ int *btm = (int *)(newp->ustack - LWP_REDZONE);
+ int *top = (int *)(newp->ustack + newp->usize + LWP_REDZONE);
+ int n = LWP_REDZONE / sizeof(int);
+ int i, lo_clean, hi_clean, overflow, underflow;
+
+ for (i = 0; i < n && btm[i] == LWP_CHECKMARK; i++) ;
+ lo_clean = i;
+
+ for (i = 1; i <= n && top[-i] == LWP_CHECKMARK; i++) ;
+ hi_clean = i - 1;
+
+ if (LwpStackGrowsDown) {
+ overflow = n - lo_clean;
+ underflow = n - hi_clean;
+ } else {
+ overflow = n - hi_clean;
+ underflow = n - lo_clean;
}
- return (1);
+ if (overflow)
+ logerror("Thread %s stack overflow %d bytes",
+ newp->name, overflow * (int)sizeof(int));
+ if (underflow)
+ logerror("Thread %s stack underflow %d bytes",
+ newp->name, underflow * (int)sizeof(int));
+ if (overflow || underflow)
+ abort();
}
/* lwpStackCheckUsed
static void
lwpStackCheckUsed(struct lwpProc *newp)
{
- register int i;
- register long *lp;
- register int lim;
- int marker;
-
- if (!newp || !newp->sbtm)
- return;
- lim = newp->size / sizeof(long);
- if (growsdown(&marker)) {
- /* Start at the bottom and find first non checkmark. */
- for (lp = newp->sbtm, i = 0; i < lim; i++, lp++) {
- if (*lp != LWP_CHECKMARK) {
- break;
- }
- }
+ int *base = (int *)newp->ustack;
+ int *lim = (int *)(newp->ustack + newp->usize);
+ int total = (lim + 1 - base) * sizeof(int);
+ int used, *p;
+
+ if (LwpStackGrowsDown) {
+ for (p = base; p < lim && *p == LWP_CHECKMARK; ++p) ;
+ used = (lim - p) * sizeof(int);
} else {
- /* Start at the top and find first non checkmark. */
- lp = newp->sbtm;
- lp += newp->size / sizeof(long);
- lp--;
- for (i = 0; i < lim; i++, lp--) {
- if (*lp != LWP_CHECKMARK) {
- break;
- }
- }
+ for (p = lim - 1; p >= base && *p == LWP_CHECKMARK; --p) ;
+ used = (p - base + 1) * sizeof(int);
}
- lwpStatus(newp, "stack use: %u bytes (of %u total)",
- (i * sizeof(long)) - LWP_REDZONE,
- newp->size - 2 * LWP_REDZONE - sizeof(stkalign_t));
+ lwpStatus(newp, "Thread stack %d used, %d left, %d total",
+ used, total - used, total);
}
-
-#endif