线程调度测试

我们终于可以来测试一下这一章的代码实现的有没有问题了！

  1. // src/process/mod.rs
  3. use scheduler::RRScheduler;
  4. use thread_pool::ThreadPool;
  5. use alloc::boxed::Box;
  7. pub fn init() {
  8.     // 使用 Round Robin Scheduler
  9.     let scheduler = RRScheduler::new(1);
  10.     // 新建线程池
  11.     let thread_pool = ThreadPool::new(100, Box::new(scheduler));
  12.     // 新建内核线程 idle ，其入口为 Processor::idle_main
  13.     let idle = Thread::new_kernel(Processor::idle_main as usize);
  14.     // 我们需要传入 CPU 的地址作为参数
  15.     idle.append_initial_arguments([&CPU as *const Processor as usize, 0, 0]);
  16.     // 初始化 CPU
  17.     CPU.init(idle, Box::new(thread_pool));
  19.     // 依次新建 5 个内核线程并加入调度单元
  20.     for i in 0..5 {
  21.         CPU.add_thread({
  22.             let thread = Thread::new_kernel(hello_thread as usize);
  23.             // 传入一个编号作为参数
  24.             thread.append_initial_arguments([i, 0, 0]);
  25.             thread
  26.         });
  27.     }
  28.     println!("++++ setup process!   ++++");
  29. }
  31. pub fn run() {
  32.     CPU.run();
  33. }
  35. // src/process/processor.rs
  37. impl Processor {
  38.     pub fn run(&self) {
  39.         // 运行，也就是从启动线程切换到调度线程 idle
  40.         Thread::get_boot_thread().switch_to(&mut self.inner().idle);
  41.     }
  42. }

内核线程的入口点是：

  1. // src/process/mod.rs
  3. #[no_mangle]
  4. pub extern "C" fn hello_thread(arg: usize) -> ! {
  5.     println!("begin of thread {}", arg);
  6.     for i in 0..800 {
  7.         print!("{}", arg);
  8.     }
  9.     println!("\nend  of thread {}", arg);
  10.     // 通知 CPU 自身已经退出
  11.     CPU.exit(0);
  12.     loop {}
  13. }

随后我们在rust_main主函数里添加调用crate::process::init()函数和crate::process::run()函数：

  1. // src/init.rs
  3. #[no_mangle]
  4. pub extern "C" fn rust_main() -> ! {
  5.     crate::interrupt::init();
  7.     extern "C" {
  8.         fn end();
  9.     }
  10.     crate::memory::init(
  11.         ((end as usize - KERNEL_BEGIN_VADDR + KERNEL_BEGIN_PADDR) >> 12) + 1,
  12.         PHYSICAL_MEMORY_END >> 12
  13.     );
  14.     crate::process::init();
  15.     crate::timer::init();
  16.     crate::process::run();
  17.     loop {}
  18. }

make run 一下，终于可以看到结果了！

这里开始就已经没有确定性的运行显示结果了，一个参考结果如下：

线程调度成功

  1. ++++ setup interrupt! ++++
  2. switch satp from 0x8000000000080221 to 0x8000000000080a37
  3. ++++ setup memory!    ++++
  4. ++++ setup process!   ++++
  5. ++++ setup timer!     ++++
  7. >>>> will switch_to thread 0 in idie_main!
  8. begin of thread 0
  9. 0000000000000000000000000000000000000000000000000000000000000000000000000
  10. 0000000000000000000000000000000000000000000000000000000000000000000000000
  11. 0000000000000000000000000000000000000000000000000000000000000000000000000
  12. 0000000000000000000000000000000000000000000000000000000000000000000000000
  13. 0000000000000000000000000000000000000000000000000000000000000000000000000
  14. 000000000000
  15. <<<< switch_back to idle in idle_main!
  17. >>>> will switch_to thread 1 in idie_main!
  18. begin of thread 1
  19. 1111111111111111111111111111111111111111111111111111111111111111111111111
  20. 1111111111111111111111111111111111111111111111111111111111111111111111111
  21. 1111111111111111111111111111111111111111111111111111111111111111111111111
  22. 1111111111111111111111111111111111111111111111111111111111111111111111111
  23. 1111111111111111111111111111111111111111111111111111111111111111111111111
  24. 1111111111111111111111111111111111111111111111111111111111111111111111111
  25. 11111111111111111111111
  26. <<<< switch_back to idle in idle_main!
  27. ......

我们可以清楚的看到在每一个时间片内每个线程所做的事情。

如果结果不对的话，这里可以看到至今的所有代码。