Files
jsix/src/kernel/scheduler.h
Justin C. Miller 8817766469 [kernel] Keep other threads out of idle priority
Split out different constants for scheduler::idle_priority and
scheduler::max_priority, so that threads never fall to the same priority
level as the idle threads.
2023-02-18 14:17:57 -08:00

112 lines
2.9 KiB
C++

#pragma once
/// \file scheduler.h
/// The task scheduler and related definitions
#include <stdint.h>
#include <util/vector.h>
extern cpu_data** g_cpu_data;
namespace kernel {
namespace args {
struct program;
}}
struct cpu_data;
class lapic;
struct page_table;
struct run_queue;
namespace obj {
class process;
class thread;
}
/// The task scheduler
class scheduler
{
public:
/// Total number of priority levels
static const uint8_t num_priorities = 8;
/// Idle task priority, less urgent than anything
static const uint8_t idle_priority = num_priorities - 1;
/// Maximum (least urgent/interactive) priority
static const uint8_t max_priority = idle_priority - 1;
/// Default priority on process creation
static const uint8_t default_priority = 1;
/// Lowest (most urgent) priority achieved via promotion
static const uint8_t promote_limit = 1;
/// How long the base timer quantum is, in us
static const uint64_t quantum_micros = 500;
/// How many quanta a process gets before being rescheduled
static const uint16_t process_quanta = 10;
/// Constructor.
/// \arg cpus The number of CPUs to schedule for
scheduler(unsigned cpus);
~scheduler();
/// Create a new process from a program image in memory.
/// \arg program The descriptor of the pogram in memory
/// \returns The main thread of the loaded process
obj::thread * load_process(kernel::args::program &program);
/// Create a new kernel task
/// \arg proc Function to run as a kernel task
/// \arg priority Priority to start the process with
/// \arg constant True if this task cannot be promoted/demoted
void create_kernel_task(
void (*task)(),
uint8_t priority,
bool constant = false);
/// Get the quantum for a given priority.
static uint32_t quantum(int priority);
/// Start the scheduler working. This may involve starting
/// timer interrupts or other preemption methods.
void start();
/// Run the scheduler, possibly switching to a new task
void schedule();
/// Check if the CPU is running a more important task. If not,
/// run the scheduler.
void maybe_schedule(TCB *t);
/// Start scheduling a new thread.
/// \arg t The new thread's TCB
void add_thread(TCB *t);
/// Get a reference to the scheduler
/// \returns A reference to the global system scheduler
static scheduler & get() { return *s_instance; }
private:
friend class obj::process;
static constexpr uint64_t promote_frequency = 100;
static constexpr uint64_t steal_frequency = 10;
void prune(run_queue &queue, uint64_t now);
void check_promotions(run_queue &queue, uint64_t now);
void steal_work(cpu_data &cpu);
uint32_t m_add_index;
uint32_t m_tick_count;
obj::process *m_kernel_process;
util::vector<run_queue> m_run_queues;
unsigned m_steal_turn = 0;
static scheduler *s_instance;
};