You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
125 lines
3.3 KiB
C
125 lines
3.3 KiB
C
#include "../kernel.h"
|
|
#include "sched.h"
|
|
#include "csr.h"
|
|
|
|
|
|
// scheduling data:
|
|
ProcessControlBlock processes[PROCESS_COUNT];
|
|
int current_process_index = 1;
|
|
|
|
void scheduler_run_next ()
|
|
{
|
|
current_process_index = scheduler_select_free();
|
|
// set up timer interrupt
|
|
unsigned long long int mtimecmp = read_time() + TIME_SLICE_LEN;
|
|
write_mtimecmp(mtimecmp);
|
|
scheduler_switch_to(current_process_index);
|
|
}
|
|
|
|
int scheduler_select_free()
|
|
{
|
|
long long int mtime;
|
|
int i;
|
|
int timeout_available = false; // note if a timeout is available
|
|
|
|
while (true) {
|
|
mtime = read_time();
|
|
|
|
for (i=1; i < PROCESS_COUNT; i++) {
|
|
ProcessControlBlock *pcb = processes + ((current_process_index + i) % PROCESS_COUNT);
|
|
|
|
if (pcb->status == PROC_RDY)
|
|
return (current_process_index + i) % PROCESS_COUNT;
|
|
|
|
if (pcb->status == PROC_WAIT_SLEEP) {
|
|
if (pcb->asleep_until < mtime) {
|
|
return (current_process_index + i) % PROCESS_COUNT;
|
|
}
|
|
timeout_available = true;
|
|
}
|
|
|
|
if (pcb->status == PROC_WAIT_PROC) {
|
|
if (pcb->asleep_until != 0) {
|
|
if (pcb->asleep_until < mtime) {
|
|
// set process return args!
|
|
return (current_process_index + i) % PROCESS_COUNT;
|
|
}
|
|
timeout_available = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (timeout_available == false) {
|
|
// either process deadlock or no processes alive.
|
|
//TODO: handle missing executable thread
|
|
HALT(22);
|
|
}
|
|
}
|
|
}
|
|
|
|
void scheduler_switch_to(int proc_index)
|
|
{
|
|
ProcessControlBlock *pcb = processes + proc_index;
|
|
|
|
CSR_WRITE(CSR_MEPC, pcb->pc);
|
|
|
|
// set up registers
|
|
__asm__(
|
|
"mv x31, %0\n"
|
|
"csrrw zero, %1, x31\n"
|
|
"lw x1, 0(x31)\n"
|
|
"lw x2, 4(x31)\n"
|
|
"lw x3, 8(x31)\n"
|
|
"lw x4, 12(x31)\n"
|
|
"lw x5, 16(x31)\n"
|
|
"lw x6, 20(x31)\n"
|
|
"lw x7, 24(x31)\n"
|
|
"lw x8, 28(x31)\n"
|
|
"lw x9, 32(x31)\n"
|
|
"lw x10, 36(x31)\n"
|
|
"lw x11, 40(x31)\n"
|
|
"lw x12, 44(x31)\n"
|
|
"lw x13, 48(x31)\n"
|
|
"lw x14, 52(x31)\n"
|
|
"lw x15, 56(x31)\n"
|
|
"lw x16, 60(x31)\n"
|
|
"lw x17, 64(x31)\n"
|
|
"lw x18, 68(x31)\n"
|
|
"lw x19, 72(x31)\n"
|
|
"lw x20, 76(x31)\n"
|
|
"lw x21, 80(x31)\n"
|
|
"lw x22, 84(x31)\n"
|
|
"lw x23, 88(x31)\n"
|
|
"lw x24, 92(x31)\n"
|
|
"lw x25, 96(x31)\n"
|
|
"lw x26, 100(x31)\n"
|
|
"lw x27, 104(x31)\n"
|
|
"lw x28, 108(x31)\n"
|
|
"lw x29, 112(x31)\n"
|
|
"lw x30, 116(x31)\n"
|
|
"lw x31, 120(x31)\n"
|
|
"mret \n"
|
|
:: "r"(pcb->regs), "I"(CSR_MSCRATCH)
|
|
);
|
|
__builtin_unreachable();
|
|
}
|
|
|
|
int scheduler_index_from_pid(int pid)
|
|
{
|
|
for (int i = 0; i < PROCESS_COUNT; i++) {
|
|
if (processes[i].pid == pid)
|
|
return i;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
int* get_current_process_registers()
|
|
{
|
|
return processes[current_process_index].regs;
|
|
}
|
|
|
|
ProcessControlBlock* get_current_process()
|
|
{
|
|
return &processes[current_process_index];
|
|
}
|