test_environment_state.hpp

Go to the documentation of this file.

 
#ifndef SIMPLEKERNEL_TESTS_UNIT_TEST_MOCKS_TEST_ENVIRONMENT_STATE_HPP_
#define SIMPLEKERNEL_TESTS_UNIT_TEST_MOCKS_TEST_ENVIRONMENT_STATE_HPP_
 
#include <cstddef>
#include <cstdint>
#include <mutex>
#include <thread>
#include <unordered_map>
#include <vector>
 
// 前向声明，防止循环依赖
struct TaskControlBlock;
struct CpuSchedData;
 
namespace test_env {
 
struct CoreEnvironment {
  // 核心标识
  size_t core_id = 0;
 
  // 中断使能状态
  bool interrupt_enabled = true;
 
  // 当前页表基地址
  uint64_t page_directory = 0;
  // 分页是否启用
  bool paging_enabled = false;
 
  // 上下文切换历史
  struct SwitchEvent {
    // 切换发生的时刻（从 sched_data->local_tick）
    uint64_t timestamp;
    // 源任务
    TaskControlBlock* from;
    // 目标任务
    TaskControlBlock* to;
    // 执行切换的核心
    size_t core_id;
  };
  std::vector<SwitchEvent> switch_history;
 
  // 关联的 std::thread ID
  std::thread::id thread_id;
 
  // 构造函数
  explicit CoreEnvironment(size_t id = 0) : core_id(id) {}
};
 
class TestEnvironmentState {
 public:
  TestEnvironmentState() = default;
  ~TestEnvironmentState() = default;
 
  TestEnvironmentState(const TestEnvironmentState&) = delete;
  TestEnvironmentState(TestEnvironmentState&&) = delete;
  auto operator=(const TestEnvironmentState&) -> TestEnvironmentState& = delete;
  auto operator=(TestEnvironmentState&&) -> TestEnvironmentState& = delete;
 
  void InitializeCores(size_t num_cores);
 
  void ResetAllCores();
 
  auto GetCore(size_t core_id) -> CoreEnvironment&;
 
  auto GetCoreCount() const -> size_t;
 
  void BindThreadToCore(std::thread::id tid, size_t core_id);
 
  auto GetCoreIdForThread(std::thread::id tid) -> size_t;
 
  auto GetCurrentCoreEnv() -> CoreEnvironment&;
 
  void RegisterTaskContext(void* context_ptr, TaskControlBlock* task);
 
  void UnregisterTaskContext(void* context_ptr);
 
  auto FindTaskByContext(void* context_ptr) -> TaskControlBlock*;
 
  void DumpAllCoreStates() const;
 
  auto GetAllSwitchHistory() const -> std::vector<CoreEnvironment::SwitchEvent>;
 
  void ClearSwitchHistory();
 
  void SetCurrentThreadEnvironment();
 
  void ClearCurrentThreadEnvironment();
 
  static auto GetCurrentThreadEnvironment() -> TestEnvironmentState*;
 
 private:
  // 核心环境列表（每个核心一个）
  std::vector<CoreEnvironment> cores_;
 
  // 线程 ID 到核心 ID 的映射
  std::unordered_map<std::thread::id, size_t> thread_to_core_map_;
 
  // 上下文指针到任务的映射
  std::unordered_map<void*, TaskControlBlock*> context_to_task_map_;
 
  // 保护以上数据结构的互斥锁
  mutable std::mutex map_mutex_;
 
  // 下一个自动分配的核心 ID
  size_t next_core_id_ = 0;
};
 
}  // namespace test_env
 
#endif /* SIMPLEKERNEL_TESTS_UNIT_TEST_MOCKS_TEST_ENVIRONMENT_STATE_HPP_ */