void
MemTime(int ns)
{
Ns_Time start, end, diff;
int i;
Ns_Thread *tids;
tids = ns_malloc(sizeof(Ns_Thread *) * nthreads);
Ns_MutexLock(&lock);
nrunning = 0;
memstart = 0;
Ns_MutexUnlock(&lock);
printf("starting %d %smalloc threads...", nthreads, ns ? "ns_" : "");
fflush(stdout);
Ns_GetTime(&start);
for (i = 0; i < nthreads; ++i) {
Ns_ThreadCreate(MemThread, (void *) ns, 0, &tids[i]);
}
Ns_MutexLock(&lock);
while (nrunning < nthreads) {
Ns_CondWait(&cond, &lock);
}
printf("waiting....");
fflush(stdout);
memstart = 1;
Ns_CondBroadcast(&cond);
Ns_MutexUnlock(&lock);
for (i = 0; i < nthreads; ++i) {
Ns_ThreadJoin(&tids[i], NULL);
}
Ns_GetTime(&end);
Ns_DiffTime(&end, &start, &diff);
printf("done: %d seconds, %d usec\n", (int) diff.sec, (int) diff.usec);
}
int
Ns_WaitForThread(Ns_Thread *thrPtr)
{
Ns_ThreadJoin(thrPtr, NULL);
return NS_OK;
}
int
Ns_WaitThread(Ns_Thread *thrPtr, int *exitCodePtr)
{
void *arg;
Ns_ThreadJoin(thrPtr, &arg);
if (exitCodePtr != NULL) {
*exitCodePtr = (int) arg;
}
return NS_OK;
}
void
NsJoinConnThreads(void)
{
ConnData *firstPtr;
void *arg;
Ns_MutexLock(&joinlock);
firstPtr = joinPtr;
joinPtr = NULL;
Ns_MutexUnlock(&joinlock);
while (firstPtr != NULL) {
Ns_ThreadJoin(&firstPtr->thread, &arg);
firstPtr = firstPtr->nextPtr;
ns_free(arg);
}
}
int main(int argc, char *argv[])
{
int i, code;
Ns_Thread threads[10];
Ns_Thread self, dumper;
void *arg;
char *p;
#if PTHREAD_TEST
pthread_t tids[10];
#endif
NsThreads_LibInit();
Ns_ThreadSetName("-main-");
/*
* Jump directly to memory test if requested.
*/
for (i = 1; i < argc; ++i) {
p = argv[i];
switch (*p) {
case 'n':
break;
case 'm':
nthreads = atoi(p + 1);
goto mem;
break;
}
}
Ns_ThreadCreate(DetachedThread, NULL, 0, NULL);
Ns_ThreadCreate(DumperThread, NULL, 0, &dumper);
Ns_MutexSetName(&lock, "startlock");
Ns_MutexSetName(&dlock, "dumplock");
Ns_MutexSetName(&mlock, "msglock");
Ns_MutexSetName(&block, "busylock");
Ns_ThreadStackSize(81920);
Ns_SemaInit(&sema, 3);
Msg("sema initialized to 3");
atexit(AtExit);
Msg("pid = %d", getpid());
Ns_TlsAlloc(&key, TlsLogArg);
for (i = 0; i < 10; ++i) {
Msg("starting work thread %d", i);
Ns_ThreadCreate(WorkThread, (void *) i, 0, &threads[i]);
}
sleep(1);
/* Ns_CondSignal(&cond); */
Ns_SemaPost(&sema, 10);
Msg("sema post 10");
Ns_RWLockWrLock(&rwlock);
Msg("rwlock write locked (main thread)");
sleep(1);
Ns_RWLockUnlock(&rwlock);
Msg("rwlock write unlocked (main thread)");
for (i = 0; i < 10; ++i) {
Msg("waiting for thread %d to exit", i);
Ns_ThreadJoin(&threads[i], (void **) &code);
Msg("thread %d exited - code: %d", i, code);
}
#if PTHREAD_TEST
for (i = 0; i < 10; ++i) {
pthread_create(&tids[i], NULL, Pthread, (void *) i);
printf("pthread: create %d = %d\n", i, (int) tids[i]);
Ns_ThreadYield();
}
Ns_MutexLock(&plock);
pgo = 1;
Ns_MutexUnlock(&plock);
Ns_CondBroadcast(&pcond);
for (i = 0; i < 10; ++i) {
pthread_join(tids[i], &arg);
printf("pthread: join %d = %d\n", i, (int) arg);
}
#endif
Ns_ThreadSelf(&self);
Ns_MutexLock(&dlock);
dstop = 1;
Ns_CondSignal(&dcond);
Ns_MutexUnlock(&dlock);
Ns_ThreadJoin(&dumper, NULL);
Msg("threads joined");
for (i = 0; i < 10; ++i) {
Ns_ThreadCreate(CheckStackThread, NULL, 8192*(i+1), &threads[i]);
}
for (i = 0; i < 10; ++i) {
Ns_ThreadJoin(&threads[i], &arg);
printf("check stack %d = %d\n", i, (int) arg);
}
/*Ns_ThreadEnum(DumpThreads, NULL);*/
/*Ns_MutexEnum(DumpLocks, NULL);*/
mem:
MemTime(0);
MemTime(1);
return 0;
}
