78 lines
2.1 KiB
Rust
78 lines
2.1 KiB
Rust
use alloc::sync::{Arc, Weak};
|
|
use crate::{mm::PhysPageNum, sync::UPSafeCell};
|
|
use crate::trap::TrapContext;
|
|
use super::id::TaskUserRes;
|
|
use super::{KernelStack, ProcessControlBlock, TaskContext, kstack_alloc};
|
|
use core::cell::RefMut;
|
|
|
|
pub struct TaskControlBlock {
|
|
// immutable
|
|
pub process: Weak<ProcessControlBlock>,
|
|
pub kstack: KernelStack,
|
|
// mutable
|
|
inner: UPSafeCell<TaskControlBlockInner>,
|
|
}
|
|
|
|
impl TaskControlBlock {
|
|
pub fn inner_exclusive_access(&self) -> RefMut<'_, TaskControlBlockInner> {
|
|
self.inner.exclusive_access()
|
|
}
|
|
|
|
pub fn get_user_token(&self) -> usize {
|
|
let process = self.process.upgrade().unwrap();
|
|
let inner = process.inner_exclusive_access();
|
|
inner.memory_set.token()
|
|
}
|
|
}
|
|
|
|
pub struct TaskControlBlockInner {
|
|
pub res: Option<TaskUserRes>,
|
|
pub trap_cx_ppn: PhysPageNum,
|
|
pub task_cx: TaskContext,
|
|
pub task_status: TaskStatus,
|
|
pub exit_code: Option<i32>,
|
|
}
|
|
|
|
impl TaskControlBlockInner {
|
|
pub fn get_trap_cx(&self) -> &'static mut TrapContext {
|
|
self.trap_cx_ppn.get_mut()
|
|
}
|
|
|
|
#[allow(unused)]
|
|
fn get_status(&self) -> TaskStatus {
|
|
self.task_status
|
|
}
|
|
}
|
|
|
|
impl TaskControlBlock {
|
|
pub fn new(
|
|
process: Arc<ProcessControlBlock>,
|
|
ustack_base: usize,
|
|
alloc_user_res: bool
|
|
) -> Self {
|
|
let res = TaskUserRes::new(Arc::clone(&process), ustack_base, alloc_user_res);
|
|
let trap_cx_ppn = res.trap_cx_ppn();
|
|
let kstack = kstack_alloc();
|
|
let kstack_top = kstack.get_top();
|
|
Self {
|
|
process: Arc::downgrade(&process),
|
|
kstack,
|
|
inner: unsafe { UPSafeCell::new(
|
|
TaskControlBlockInner {
|
|
res: Some(res),
|
|
trap_cx_ppn,
|
|
task_cx: TaskContext::goto_trap_return(kstack_top),
|
|
task_status: TaskStatus::Ready,
|
|
exit_code: None,
|
|
}
|
|
)},
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Copy, Clone, PartialEq)]
|
|
pub enum TaskStatus {
|
|
Ready,
|
|
Running,
|
|
Blocking,
|
|
}
|