|
|
|
@ -4,16 +4,16 @@ mod task;
|
|
|
|
|
|
|
|
|
|
use crate::config::MAX_APP_NUM;
|
|
|
|
|
use crate::loader::{get_num_app, init_app_cx};
|
|
|
|
|
use core::cell::RefCell;
|
|
|
|
|
use lazy_static::*;
|
|
|
|
|
use switch::__switch;
|
|
|
|
|
use task::{TaskControlBlock, TaskStatus};
|
|
|
|
|
use crate::sync::UPSafeCell;
|
|
|
|
|
|
|
|
|
|
pub use context::TaskContext;
|
|
|
|
|
|
|
|
|
|
pub struct TaskManager {
|
|
|
|
|
num_app: usize,
|
|
|
|
|
inner: RefCell<TaskManagerInner>,
|
|
|
|
|
inner: UPSafeCell<TaskManagerInner>,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
struct TaskManagerInner {
|
|
|
|
@ -21,56 +21,62 @@ struct TaskManagerInner {
|
|
|
|
|
current_task: usize,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
unsafe impl Sync for TaskManager {}
|
|
|
|
|
|
|
|
|
|
lazy_static! {
|
|
|
|
|
pub static ref TASK_MANAGER: TaskManager = {
|
|
|
|
|
let num_app = get_num_app();
|
|
|
|
|
let mut tasks = [
|
|
|
|
|
TaskControlBlock { task_cx_ptr: 0, task_status: TaskStatus::UnInit };
|
|
|
|
|
TaskControlBlock {
|
|
|
|
|
task_cx: TaskContext::zero_init(),
|
|
|
|
|
task_status: TaskStatus::UnInit
|
|
|
|
|
};
|
|
|
|
|
MAX_APP_NUM
|
|
|
|
|
];
|
|
|
|
|
for i in 0..num_app {
|
|
|
|
|
tasks[i].task_cx_ptr = init_app_cx(i) as * const _ as usize;
|
|
|
|
|
tasks[i].task_cx = TaskContext::goto_restore(init_app_cx(i));
|
|
|
|
|
tasks[i].task_status = TaskStatus::Ready;
|
|
|
|
|
}
|
|
|
|
|
TaskManager {
|
|
|
|
|
num_app,
|
|
|
|
|
inner: RefCell::new(TaskManagerInner {
|
|
|
|
|
inner: unsafe { UPSafeCell::new(TaskManagerInner {
|
|
|
|
|
tasks,
|
|
|
|
|
current_task: 0,
|
|
|
|
|
}),
|
|
|
|
|
})},
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl TaskManager {
|
|
|
|
|
fn run_first_task(&self) {
|
|
|
|
|
self.inner.borrow_mut().tasks[0].task_status = TaskStatus::Running;
|
|
|
|
|
let next_task_cx_ptr2 = self.inner.borrow().tasks[0].get_task_cx_ptr2();
|
|
|
|
|
let _unused: usize = 0;
|
|
|
|
|
fn run_first_task(&self) -> ! {
|
|
|
|
|
let mut inner = self.inner.exclusive_access();
|
|
|
|
|
let task0 = &mut inner.tasks[0];
|
|
|
|
|
task0.task_status = TaskStatus::Running;
|
|
|
|
|
let next_task_cx_ptr = &task0.task_cx as *const TaskContext;
|
|
|
|
|
drop(inner);
|
|
|
|
|
let mut _unused = TaskContext::zero_init();
|
|
|
|
|
// before this, we should drop local variables that must be dropped manually
|
|
|
|
|
unsafe {
|
|
|
|
|
__switch(
|
|
|
|
|
&_unused as *const _,
|
|
|
|
|
next_task_cx_ptr2,
|
|
|
|
|
&mut _unused as *mut TaskContext,
|
|
|
|
|
next_task_cx_ptr,
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
panic!("unreachable in run_first_task!");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn mark_current_suspended(&self) {
|
|
|
|
|
let mut inner = self.inner.borrow_mut();
|
|
|
|
|
let mut inner = self.inner.exclusive_access();
|
|
|
|
|
let current = inner.current_task;
|
|
|
|
|
inner.tasks[current].task_status = TaskStatus::Ready;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn mark_current_exited(&self) {
|
|
|
|
|
let mut inner = self.inner.borrow_mut();
|
|
|
|
|
let mut inner = self.inner.exclusive_access();
|
|
|
|
|
let current = inner.current_task;
|
|
|
|
|
inner.tasks[current].task_status = TaskStatus::Exited;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn find_next_task(&self) -> Option<usize> {
|
|
|
|
|
let inner = self.inner.borrow();
|
|
|
|
|
let inner = self.inner.exclusive_access();
|
|
|
|
|
let current = inner.current_task;
|
|
|
|
|
(current + 1..current + self.num_app + 1)
|
|
|
|
|
.map(|id| id % self.num_app)
|
|
|
|
@ -81,19 +87,21 @@ impl TaskManager {
|
|
|
|
|
|
|
|
|
|
fn run_next_task(&self) {
|
|
|
|
|
if let Some(next) = self.find_next_task() {
|
|
|
|
|
let mut inner = self.inner.borrow_mut();
|
|
|
|
|
let mut inner = self.inner.exclusive_access();
|
|
|
|
|
let current = inner.current_task;
|
|
|
|
|
inner.tasks[next].task_status = TaskStatus::Running;
|
|
|
|
|
inner.current_task = next;
|
|
|
|
|
let current_task_cx_ptr2 = inner.tasks[current].get_task_cx_ptr2();
|
|
|
|
|
let next_task_cx_ptr2 = inner.tasks[next].get_task_cx_ptr2();
|
|
|
|
|
core::mem::drop(inner);
|
|
|
|
|
let current_task_cx_ptr = &mut inner.tasks[current].task_cx as *mut TaskContext;
|
|
|
|
|
let next_task_cx_ptr = &inner.tasks[next].task_cx as *const TaskContext;
|
|
|
|
|
drop(inner);
|
|
|
|
|
// before this, we should drop local variables that must be dropped manually
|
|
|
|
|
unsafe {
|
|
|
|
|
__switch(
|
|
|
|
|
current_task_cx_ptr2,
|
|
|
|
|
next_task_cx_ptr2,
|
|
|
|
|
current_task_cx_ptr,
|
|
|
|
|
next_task_cx_ptr,
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
// go back to user mode
|
|
|
|
|
} else {
|
|
|
|
|
panic!("All applications completed!");
|
|
|
|
|
}
|
|
|
|
|