test_3.c

#include "types.h"
#include "stat.h"
#include "param.h"
#include "memlayout.h"
#include "mmu.h"
#include "x86.h"
#include "spinlock.h"
#include "proc.h"
#include "user.h"
#include "synch.h"

#define N 8

void *stack[NTHREAD];
int tid[NTHREAD];
void *retval[NTHREAD];

struct mutex_t m;
struct cond_t c;
int buffer[10];
int idx=0;
int condition = 0;

void *thread(void *arg){
//	printf(1, "Holding thread: %d\n", m.current);
	mutex_lock(&m);
	while(condition == 0){
		cond_wait(&c, &m);
//	printf(1, "cond wait successed\n");
	}

	condition--;
	buffer[idx++] = (int)arg;
	mutex_unlock(&m);
	thread_exit(0);
}

int
main(int argc, char **argv)
{
	int i;

	printf(1, "TEST: ");

	for(i=0;i<NTHREAD;i++)
		stack[i] = malloc(4096);
	printf(1, "MUTEX: %d\nCV: %d\n", &m, &c);
	mutex_init(&m);
	cond_init(&c);

	for(i=0;i<NTHREAD-1;i++){
		tid[i] = thread_create(thread, (void *)i, stack[i]);
		printf(1, "thread %d\n", i);
		if(tid[i] == -1){
			printf(1, "thread create failed\n");
			exit();
		}
	}

	for(i=0;i<NTHREAD-1;i++){
		mutex_lock(&m);
		condition++;
		cond_signal(&c);
		mutex_unlock(&m);
	}

	for(i=0;i<NTHREAD-1;i++){
		if(thread_join(tid[i], &retval[i]) == -1){
			printf(1, "thread join failed\n");
			exit();
		}
	}

	for(i=0;i<NTHREAD-1;i++){
		if(buffer[i] != i){
			printf(1, "WRONG\n");
			exit();
		}
	}

	for(i=0;i<NTHREAD;i++)
		free(stack[i]);

	printf(1, "OK\n");

	exit();
}