test: fix st

Signed-off-by: Niu Zhihong <zhihong@nzhnb.com>

Author: MRNIU

src/main.cpp

@@ -163,14 +163,13 @@ auto main(int argc, const char** argv) -> int {
   TaskManagerSingleton::create();
   TaskManagerSingleton::instance().InitCurrentCore();
 
+  TimerInit();
+
   // 唤醒其余 core
   WakeUpOtherCores();
 
   DumpStack();
 
-  // 初始化定时器
-  TimerInit();
-
   klog::Info("Hello SimpleKernel");
 
   klog::Info("Initializing test tasks...");

src/task/include/task_manager.hpp

@@ -62,6 +62,9 @@ struct CpuSchedData {
   /// 本核心的总调度次数
   uint64_t total_schedules{0};
 
+  /// Schedule() 是否已被显式调用
+  bool scheduler_started{false};
+
   /// @name 构造/析构函数
   /// @{
   CpuSchedData() = default;

src/task/schedule.cpp

@@ -25,6 +25,7 @@
 
 auto TaskManager::Schedule() -> void {
   auto& cpu_sched = GetCurrentCpuSched();
+  cpu_sched.scheduler_started = true;
   cpu_sched.lock.Lock().or_else([](auto&& err) {
     klog::Err("Schedule: Failed to acquire lock: {}", err.message());
     while (true) {

src/task/tick_update.cpp

@@ -67,8 +67,7 @@ auto TaskManager::TickUpdate() -> void {
     }
   }
 
-  // 如果需要抢占，触发调度
-  if (need_preempt) {
+  if (need_preempt && cpu_sched.scheduler_started) {
     Schedule();
   }
 }

tests/system_test/main.cpp

@@ -260,12 +260,12 @@ auto main(int argc, const char** argv) -> int {
   TaskManagerSingleton::create();
   TaskManagerSingleton::instance().InitCurrentCore();
 
+  TimerInit();
+
   WakeUpOtherCores();
 
   DumpStack();
 
-  TimerInit();
-
   klog::Info("Hello SimpleKernel");
 
   for (size_t i = 0; i < kTestCount; ++i) {