diff options
| author | Christian Cunningham <cc@localhost> | 2022-03-24 09:38:08 -0700 | 
|---|---|---|
| committer | Christian Cunningham <cc@localhost> | 2022-03-24 09:38:08 -0700 | 
| commit | 93bf62580a68533dc8252b9a2a055c02f34ecb67 (patch) | |
| tree | 1b1ca92ebbe107a998136a1442c0dba5be885e13 /src/sys/schedule.c | |
| parent | 3e64dda5d5c350cc325650133f7e64967f1efe84 (diff) | |
Modularized
Diffstat (limited to 'src/sys/schedule.c')
| -rw-r--r-- | src/sys/schedule.c | 468 | 
1 files changed, 0 insertions, 468 deletions
| diff --git a/src/sys/schedule.c b/src/sys/schedule.c deleted file mode 100644 index 9b6d46e..0000000 --- a/src/sys/schedule.c +++ /dev/null @@ -1,468 +0,0 @@ -#include <cpu.h> -#include <globals.h> -#include <graphics/lfb.h> -#include <drivers/uart.h> -#include <lib/kmem.h> -#include <sys/schedule.h> -#include <util/mutex.h> - -extern void kernel_usr_task_loop(void); - -void init_scheduler(void) -{ -	// Set rthread to usrloopthread - an infinitely running thread so that the pointer will never be null -	usrloopthread.pc = (void*)kernel_usr_task_loop; -	usrloopthread.sp = (void*)0x5FC8; -	*(unsigned long**)usrloopthread.sp = (unsigned long*)kernel_usr_task_loop; -	usrloopthread.sp_base = -1; -	usrloopthread.mptr = 0; -	usrloopthread.pid = -1; -	usrloopthread.priority = -1; -	usrloopthread.old_priority = -1; -	usrloopthread.status = THREAD_READY; -	usrloopthread.offset = -1; -	scheduler.rthread = &usrloopthread; - -	// Initialize Scheduling Queues -	for (unsigned long p = 0; p < PRIORITIES; p++) { -		// Ready Init -		scheduler.ready[p].start.value = 0; -		scheduler.ready[p].start.next = &scheduler.ready[p].end; -		scheduler.ready[p].start.entry_type = START_ENTRY; -		scheduler.ready[p].end.value = 0; -		scheduler.ready[p].end.next = &scheduler.ready[p].start; -		scheduler.ready[p].end.entry_type = END_ENTRY; -		// Mutex Wait Init -		scheduler.mwait[p].start.value = 0; -		scheduler.mwait[p].start.next = &scheduler.mwait[p].end; -		scheduler.mwait[p].start.entry_type = START_ENTRY; -		scheduler.mwait[p].end.value = 0; -		scheduler.mwait[p].end.next = &scheduler.mwait[p].start; -		scheduler.mwait[p].end.entry_type = END_ENTRY; -		// Signal Wait Init -		scheduler.swait[p].start.value = 0; -		scheduler.swait[p].start.next = &scheduler.swait[p].end; -		scheduler.swait[p].start.entry_type = START_ENTRY; -		scheduler.swait[p].end.value = 0; -		scheduler.swait[p].end.next = &scheduler.swait[p].start; -		scheduler.swait[p].end.entry_type = END_ENTRY; -	} - -	// Initialize nextpid -	nextpid = FIRST_AVAIL_PID; - -	// Initialize Threads - Stack Base and Offsets -	for (unsigned long i = 0; i < MAX_THREADS; i++) { -		struct Thread* t = &threads[i]; -		t->offset = i; -		t->sp_base = 0x20000000 - STACK_SIZE*i; -		thread_entries[i].value = t; -		thread_entries[i].next = &thread_entries[(i+1)]; -		thread_entries[i].entry_type = VALUE_ENTRY; -	} -	// Initialize the free queue -	scheduler.free_threads.start.value = 0; -	scheduler.free_threads.start.entry_type = START_ENTRY; -	scheduler.free_threads.end.value = 0; -	scheduler.free_threads.end.entry_type = END_ENTRY; -	scheduler.free_threads.start.next = &thread_entries[0]; -	scheduler.free_threads.end.next = &thread_entries[MAX_THREADS-1]; -	thread_entries[MAX_THREADS-1].next = &scheduler.free_threads.end; -} - -void push_thread_to_queue(struct Thread* t, unsigned char type, unsigned char priority) -{ -	struct Entry* entry = &thread_entries[t->offset]; -	struct Queue* queue; -	if (type == THREAD_READY) { -		queue = &scheduler.ready[priority]; -	} else if (type == THREAD_MWAIT) { -		queue = &scheduler.mwait[priority]; -	} else if (type == THREAD_SWAIT) { -		queue = &scheduler.swait[priority]; -	} else { -		return; -	} -	push_to_queue(entry, queue); -	//queue->end.next->next = entry; -	//queue->end.next = entry; -	//entry->next = &queue->end; -} - -void prepend_thread_to_queue(struct Thread* t, unsigned char type, unsigned char priority) -{ -	struct Entry* entry = &thread_entries[t->offset]; -	struct Queue* queue; -	if (type == THREAD_READY) { -		queue = &scheduler.ready[priority]; -	} else if (type == THREAD_MWAIT) { -		queue = &scheduler.mwait[priority]; -	} else if (type == THREAD_SWAIT) { -		queue = &scheduler.swait[priority]; -	} else { -		return; -	} -	prepend_to_queue(entry, queue); -} - -struct Entry* pop_thread_from_queue(unsigned char type, unsigned char priority) -{ -	struct Entry* entry = 0; -	struct Queue* queue; -	if (type == THREAD_READY) { -		queue = &scheduler.ready[priority]; -	} else if (type == THREAD_MWAIT) { -		queue = &scheduler.mwait[priority]; -	} else if (type == THREAD_SWAIT) { -		queue = &scheduler.swait[priority]; -	} else { -		return entry; -	} -	return pop_from_queue(queue); -} - -struct Entry* find_pid(unsigned long pid) -{ -	for (unsigned char p = 0; p < PRIORITIES; p++) { -		struct Queue* queue; -		struct Entry* prev; -		struct Entry* entry; - -		queue = &scheduler.ready[p]; -		prev = &queue->start; -		entry = prev->next; -		while (entry->entry_type != END_ENTRY) { -			if (((struct Thread*)entry->value)->pid == pid) -				return prev; -			prev = entry; -			entry = entry->next; -		} - -		queue = &scheduler.mwait[p]; -		prev = &queue->start; -		entry = prev->next; -		while (entry->entry_type != END_ENTRY) { -			if (((struct Thread*)entry->value)->pid == pid) -				return prev; -			prev = entry; -			entry = entry->next; -		} - -		queue = &scheduler.swait[p]; -		prev = &queue->start; -		entry = prev->next; -		while (entry->entry_type != END_ENTRY) { -			if (((struct Thread*)entry->value)->pid == pid) -				return prev; -			prev = entry; -			entry = entry->next; -		} -	} -	return 0; -} - -struct Entry* find_mutex_wait_next(void* m) -{ -	for (unsigned char p = 0; p < PRIORITIES; p++) { -		struct Queue* queue = &scheduler.mwait[p]; -		struct Entry* prev = &queue->start; -		struct Entry* entry = prev->next; -		while (entry->entry_type != END_ENTRY) { -			if (((struct Thread*)entry->value)->mptr == m) -				return prev; -			prev = entry; -			entry = entry->next; -		} -	} -	return 0; -} - -struct Entry* find_signal_wait_next(void* s) -{ -	for (unsigned char p = 0; p < PRIORITIES; p++) { -		struct Queue* queue = &scheduler.swait[p]; -		struct Entry* prev = &queue->start; -		struct Entry* entry = prev->next; -		while (entry->entry_type != END_ENTRY) { -			if (((struct Thread*)entry->value)->mptr == s) -				return prev; -			prev = entry; -			entry = entry->next; -		} -	} -	return 0; -} - -struct Entry* get_unused_thread(void) -{ -	struct Queue* q = &scheduler.free_threads; -	// If we have no available free threads -	//  return null pointer -	if (q->start.next->entry_type == END_ENTRY) -		return 0; -	// Otherwise, get the next thread -	return pop_from_queue(q); -} - -unsigned char find_duplicate(void* pc) -{ -	for (unsigned char p = 0; p < PRIORITIES; p++) { -		struct Queue* queue = &scheduler.ready[p]; -		struct Entry* entry = queue->start.next; -		while (entry->entry_type == VALUE_ENTRY) { -			if (((struct Thread*)entry->value)->pc == pc) { -				return 1; -			} -		} -	} -	return 0; -} - -unsigned char add_thread_without_duplicate(void* pc, void* arg, unsigned char priority) -{ -	if (!find_duplicate(pc)) { -		return add_thread(pc, arg, priority); -	} -	return 1; -} - -unsigned char svc_add_thread(void* pc, void* arg, unsigned char priority) -{ -	struct Entry* thread_entry = get_unused_thread(); -	// The only point-of-failure is not having a thread available -	if (thread_entry == 0) -		return 1; -	struct Thread* thread = thread_entry->value; -	/// Thread Setup -	thread->pc = pc; -	unsigned long* argp = (void*)thread->sp_base; -	argp -= 13; -	*argp = (unsigned long)arg; // Set r0 to the argument -	argp -= 1; -	*(unsigned long**)argp = (unsigned long*)cleanup; // Set lr to the cleanup function -	thread->sp = argp; -	thread->status = THREAD_READY; -	thread->mptr = (void*)0; -	thread->pid = nextpid++; -	// Reset next pid on overflow -	if (nextpid < FIRST_AVAIL_PID) { -		nextpid = FIRST_AVAIL_PID; -	} -	// Cap Priority Level -	if (priority >= PRIORITIES) -		thread->priority = PRIORITIES - 1; -	else -		thread->priority = priority; -	// This thread is new -	thread->old_priority = -1; -	// Reserved for non-preemptible tasking -	thread->preempt = 0; -	/// Add Thread to Scheduler -	push_thread_to_queue(thread, THREAD_READY, thread->priority); -	return 0; -} - -void uart_scheduler(void) -{ -	uart_string("Scheduler Info\n==============\nCurrent\n"); -	uart_hex((unsigned long)scheduler.rthread); -	uart_char(' '); -	kmemshow32((void*)scheduler.rthread, 9); -	unsigned long length; -	for(int p = 0; p < PRIORITIES; p++) { -		uart_string("Priority "); -		uart_10(p); -		uart_char('\n'); -		struct Queue* queue; -		struct Entry* entry; - -		queue = &scheduler.ready[p]; -		uart_string("Ready Queue\n"); -		entry = queue->start.next; -		length = 0; -		while (entry->entry_type != END_ENTRY) { -			uart_hex((unsigned long)entry->value); -			uart_char(' '); -			kmemshow32((void*)entry->value, 9); -			entry = entry->next; -			length++; -		} -		uart_hexn(length); - -		queue = &scheduler.mwait[p]; -		uart_string("Mutex Wait Queue\n"); -		entry = queue->start.next; -		length = 0; -		while (entry->entry_type != END_ENTRY) { -			uart_hex((unsigned long)entry->value); -			uart_char(' '); -			kmemshow32((void*)entry->value, 9); -			entry = entry->next; -			length++; -		} -		uart_hexn(length); - -		queue = &scheduler.swait[p]; -		uart_string("Signal Wait Queue\n"); -		entry = queue->start.next; -		length = 0; -		while (entry->entry_type != END_ENTRY) { -			uart_hex((unsigned long)entry->value); -			uart_char(' '); -			kmemshow32((void*)entry->value, 9); -			entry = entry->next; -			length++; -		} -		uart_hexn(length); -	} -	// Count number of free threads -	struct Queue* queue = &scheduler.free_threads; -	struct Entry* entry = queue->start.next; -	while (entry->entry_type != END_ENTRY) { -		entry = entry->next; -		length++; -	} -	uart_hexn(length); -	uart_string("==============\n"); -} - -struct Thread* next_thread(void) -{ -	// Recurse through all priorities to try to find a ready thread -	for (int p = 0; p < PRIORITIES; p++) { -		struct Queue* rq = &scheduler.ready[p]; -		if (rq->start.next->entry_type == END_ENTRY) -			continue; -		return rq->start.next->value; -	} -	// No thread found, use basic usrloopthread while waiting for new thread -	return &usrloopthread; -} - -void c_cleanup(void) -{ -	struct Thread* rt = scheduler.rthread; -	struct Entry* e = pop_thread_from_queue(THREAD_READY, rt->priority); -	// Add to free threads -	push_to_queue(e, &scheduler.free_threads); -} - -void yield(void) -{ -	struct Thread* rthread = scheduler.rthread; -	// usrloopthread should not be yielded -	if (rthread == &usrloopthread) -		return; -	// Put current thread at the end of its ready queue, -	//  thus any threads of the same priority can be run first -	unsigned char priority = rthread->priority; -	struct Entry* tq; -	// Remove from top of queue -	tq = pop_thread_from_queue(THREAD_READY, priority); -	if (tq != 0) { -		// Add to bottom of queue -		push_thread_to_queue(tq->value, THREAD_READY, priority); -	} -} - -void sched_mutex_yield(void* m) -{ -	struct Thread* rthread = scheduler.rthread; -	// usrloopthread should not be yielded -	if (rthread == &usrloopthread) -		return; -	unsigned char priority = rthread->priority; -	// Signify which lock this thread is waiting for -	rthread->mptr = m; -	struct Entry* rt; -	// Remove from top of running queue -	rt = pop_thread_from_queue(THREAD_READY, priority); -	if (rt != 0) -		// Push to bottom of wait queue -		push_thread_to_queue(rt->value, THREAD_MWAIT, priority); -	// Find the thread that has the mutex locked -	struct Mutex* mtex = m; -	struct Entry* mutex_next = find_pid(mtex->pid); -	if (mutex_next == 0) -		return; -	// The next thread is the one with the lock -	struct Entry* mutex_thread_entry = mutex_next->next; -	// Check if it is lower priority -	if (((struct Thread*)mutex_thread_entry->value)->priority > priority) { -		// Remove it from the old priority queue -		remove_next_from_queue(mutex_next); -		struct Thread* t = mutex_thread_entry->value; -		// Preserve the old priority -		if (t->old_priority == 0xFF) -			t->old_priority = t->priority; -		t->priority = priority; -		// Add it to the higher priority queue -		push_thread_to_queue(t, THREAD_READY, priority); -	} -} - -void sched_semaphore_yield(void* s) -{ -	struct Thread* rthread = scheduler.rthread; -	// usrloopthread should not be yielded -	if (rthread == &usrloopthread) -		return; -	unsigned char priority = rthread->priority; -	// Signify which lock this thread is waiting for -	rthread->mptr = s; -	struct Entry* rt; -	// Remove from top of running queue -	rt = pop_thread_from_queue(THREAD_READY, priority); -	if (rt != 0) -		// Push to bottom of wait queue -		push_thread_to_queue(rt->value, THREAD_SWAIT, priority); -} - -void sched_mutex_resurrect(void* m) -{ -	// Find any mutex to resurrect -	struct Entry* prev = find_mutex_wait_next(m); -	if (prev == 0) -		return; -	struct Entry* entry = prev->next; -	struct Thread* thread = entry->value; -	// Resurrect the thread -	thread->mptr = 0; -	// Remove from wait queue -	entry = remove_next_from_queue(prev); -	if (entry == 0) -		return; -	// Add to ready queue -	push_thread_to_queue(entry->value, THREAD_READY, ((struct Thread*)entry->value)->priority); -	// Demote current thread -	struct Thread* rthread = scheduler.rthread; -	unsigned long p = rthread->priority; -	unsigned long op = rthread->old_priority; -	// Restore the original priority level -	if (op != 0xFF) { -		struct Entry* tentry = pop_thread_from_queue(THREAD_READY, p); -		((struct Thread*)tentry->value)->priority = op; -		((struct Thread*)tentry->value)->old_priority = 0xFF; -		prepend_thread_to_queue(tentry->value, THREAD_READY, op); -	} -} - -void sched_semaphore_resurrect(void* s, unsigned long count) -{ -	while (count--) { -		// Find any signal/ semaphore to resurrect -		struct Entry* prev = find_signal_wait_next(s); -		if (prev == 0) -			return; -		struct Entry* entry = prev->next; -		struct Thread* thread = entry->value; -		// Resurrect the thread -		thread->mptr = 0; -		// Remove from wait queue -		entry = remove_next_from_queue(prev); -		if (entry == 0) -			return; -		// Add to ready queue -		push_thread_to_queue(entry->value, THREAD_READY, ((struct Thread*)entry->value)->priority); -	} -} | 
