feat(domain): add ProcessTable with create, get, and destroy

This commit is contained in:
2026-10-17 23:27:28 +02:00
parent fe0fde7de1
commit 42ed5ade29
3 changed files with 109 additions and 3 deletions

View File

@@ -9,4 +9,5 @@ pub enum KernelError {
InvalidProcess(ProcessId), InvalidProcess(ProcessId),
WouldBlock, WouldBlock,
TooManyThreadsInProcess, TooManyThreadsInProcess,
TooManyProcesses,
} }

View File

@@ -3,13 +3,14 @@ use types::{
id::{ProcessId, ThreadId}, id::{ProcessId, ThreadId},
}; };
use crate::capability::CapabilityTable; use crate::{capability::CapabilityTable, error::KernelError};
#[cfg(test)] #[cfg(test)]
#[path = "tests.rs"] #[path = "tests.rs"]
mod tests; mod tests;
pub const MAX_THREADS_PER_PROCESS: usize = 4; pub const MAX_THREADS_PER_PROCESS: usize = 4;
pub const MAX_PROCESSES: usize = 64;
pub struct Process { pub struct Process {
id: ProcessId, id: ProcessId,
@@ -48,12 +49,12 @@ impl Process {
self.fault_handler self.fault_handler
} }
pub fn add_thread(&mut self, thread: ThreadId) -> Result<(), crate::error::KernelError> { pub fn add_thread(&mut self, thread: ThreadId) -> Result<(), KernelError> {
let slot = self let slot = self
.threads .threads
.iter_mut() .iter_mut()
.find(|s| s.is_none()) .find(|s| s.is_none())
.ok_or(crate::error::KernelError::TooManyThreadsInProcess)?; .ok_or(KernelError::TooManyThreadsInProcess)?;
*slot = Some(thread); *slot = Some(thread);
Ok(()) Ok(())
} }
@@ -71,3 +72,55 @@ impl Process {
self.threads.iter().filter_map(|s| *s) self.threads.iter().filter_map(|s| *s)
} }
} }
pub struct ProcessTable {
processes: [Option<Process>; MAX_PROCESSES],
}
impl ProcessTable {
pub fn new() -> Self {
Self {
processes: [const { None }; MAX_PROCESSES],
}
}
pub fn create(
&mut self,
capabilities: CapabilityTable,
fault_handler: Option<CapSlot>,
) -> Result<ProcessId, KernelError> {
let idx = self
.processes
.iter()
.position(|s| s.is_none())
.ok_or(KernelError::TooManyProcesses)?;
let id = ProcessId::from_raw(idx as u16);
self.processes[idx] = Some(Process::new(id, capabilities, fault_handler));
Ok(id)
}
pub fn get(&self, id: ProcessId) -> Result<&Process, KernelError> {
let idx = id.raw() as usize;
self.processes
.get(idx)
.and_then(|s| s.as_ref())
.ok_or(KernelError::InvalidProcess(id))
}
pub fn get_mut(&mut self, id: ProcessId) -> Result<&mut Process, KernelError> {
let idx = id.raw() as usize;
self.processes
.get_mut(idx)
.and_then(|s| s.as_mut())
.ok_or(KernelError::InvalidProcess(id))
}
pub fn destroy(&mut self, id: ProcessId) -> Result<Process, KernelError> {
let idx = id.raw() as usize;
self.processes
.get_mut(idx)
.and_then(|s| s.take())
.ok_or(KernelError::InvalidProcess(id))
}
}

View File

@@ -76,3 +76,55 @@ fn capabilities_mut_accessible() {
let mut p = test_process(0); let mut p = test_process(0);
let _caps = p.capabilities_mut(); let _caps = p.capabilities_mut();
} }
// --- ProcessTable tests ---
#[test]
fn table_create_and_lookup() {
let mut table = ProcessTable::new();
let id = table.create(CapabilityTable::new(), None).unwrap();
assert_eq!(table.get(id).unwrap().id(), id);
}
#[test]
fn table_create_returns_unique_ids() {
let mut table = ProcessTable::new();
let a = table.create(CapabilityTable::new(), None).unwrap();
let b = table.create(CapabilityTable::new(), None).unwrap();
assert_ne!(a, b);
}
#[test]
fn table_destroy_then_get_fails() {
let mut table = ProcessTable::new();
let id = table.create(CapabilityTable::new(), None).unwrap();
table.destroy(id).unwrap();
assert!(table.get(id).is_err());
}
#[test]
fn table_full_returns_error() {
let mut table = ProcessTable::new();
for _ in 0..MAX_PROCESSES {
table.create(CapabilityTable::new(), None).unwrap();
}
assert!(table.create(CapabilityTable::new(), None).is_err());
}
#[test]
fn table_destroy_reuses_slot() {
let mut table = ProcessTable::new();
let id = table.create(CapabilityTable::new(), None).unwrap();
table.destroy(id).unwrap();
let new_id = table.create(CapabilityTable::new(), None).unwrap();
assert_eq!(id, new_id);
}
#[test]
fn table_get_mut_allows_modification() {
let mut table = ProcessTable::new();
let id = table.create(CapabilityTable::new(), None).unwrap();
let p = table.get_mut(id).unwrap();
p.add_thread(thread(0)).unwrap();
assert_eq!(table.get(id).unwrap().thread_ids().count(), 1);
}