apic_timer.h 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. #pragma once
  2. #include <common/unistd.h>
  3. #include "apic.h"
  4. extern uint64_t apic_timer_ticks_result;
  5. // 5ms产生一次中断
  6. #define APIC_TIMER_INTERVAL 5
  7. #define APIC_TIMER_DIVISOR 3
  8. #define APIC_TIMER_IRQ_NUM 151
  9. #pragma GCC push_options
  10. #pragma GCC optimize("O0")
  11. /**
  12. * @brief 设置apic定时器的分频计数
  13. *
  14. * @param divider 分频除数
  15. */
  16. static __always_inline void apic_timer_set_div(uint64_t divider)
  17. {
  18. if (CURRENT_APIC_STATE == APIC_X2APIC_ENABLED)
  19. wrmsr(0x83e, divider);
  20. else
  21. __write4b(APIC_LOCAL_APIC_VIRT_BASE_ADDR + LOCAL_APIC_OFFSET_Local_APIC_CLKDIV, divider);
  22. }
  23. /**
  24. * @brief 设置apic定时器的初始计数值
  25. *
  26. * @param init_cnt 初始计数值
  27. */
  28. static __always_inline void apic_timer_set_init_cnt(uint32_t init_cnt)
  29. {
  30. if (CURRENT_APIC_STATE == APIC_X2APIC_ENABLED)
  31. wrmsr(0x838, init_cnt);
  32. else
  33. __write4b(APIC_LOCAL_APIC_VIRT_BASE_ADDR + LOCAL_APIC_OFFSET_Local_APIC_INITIAL_COUNT_REG, init_cnt);
  34. }
  35. /**
  36. * @brief 设置apic定时器的lvt,并启动定时器
  37. *
  38. * @param vector 中断向量号
  39. * @param mask 是否屏蔽(1:屏蔽, 0:不屏蔽)
  40. * @param mode 计时模式
  41. */
  42. static __always_inline void apic_timer_set_LVT(uint32_t vector, uint32_t mask, uint32_t mode)
  43. {
  44. register uint32_t val = (mode << 17) | vector | (mask ? (APIC_LVT_INT_MASKED) : 0);
  45. if (CURRENT_APIC_STATE == APIC_X2APIC_ENABLED)
  46. wrmsr(0x832, val);
  47. else
  48. __write4b(APIC_LOCAL_APIC_VIRT_BASE_ADDR + LOCAL_APIC_OFFSET_Local_APIC_LVT_TIMER, val);
  49. }
  50. static __always_inline void apic_timer_write_LVT(uint32_t value)
  51. {
  52. if (CURRENT_APIC_STATE == APIC_X2APIC_ENABLED)
  53. wrmsr(0x832, value);
  54. else
  55. __write4b(APIC_LOCAL_APIC_VIRT_BASE_ADDR + LOCAL_APIC_OFFSET_Local_APIC_LVT_TIMER, value);
  56. }
  57. /**
  58. * @brief 获取apic定时器的LVT的值
  59. *
  60. */
  61. static __always_inline uint32_t apic_timer_get_LVT()
  62. {
  63. if (CURRENT_APIC_STATE == APIC_X2APIC_ENABLED)
  64. return rdmsr(0x832);
  65. else
  66. return __read4b(APIC_LOCAL_APIC_VIRT_BASE_ADDR + LOCAL_APIC_OFFSET_Local_APIC_LVT_TIMER);
  67. }
  68. /**
  69. * @brief 获取apic定时器当前计数值
  70. *
  71. */
  72. static __always_inline uint32_t apic_timer_get_current()
  73. {
  74. if (CURRENT_APIC_STATE == APIC_X2APIC_ENABLED)
  75. return (uint32_t)rdmsr(0x839);
  76. else
  77. return __read4b(APIC_LOCAL_APIC_VIRT_BASE_ADDR + LOCAL_APIC_OFFSET_Local_APIC_CURRENT_COUNT_REG);
  78. }
  79. /**
  80. * @brief 停止apic定时器
  81. *
  82. */
  83. #define apic_timer_stop() \
  84. do \
  85. { \
  86. uint32_t val = apic_timer_get_LVT(); \
  87. val |= APIC_LVT_INT_MASKED; \
  88. apic_timer_write_LVT(val); \
  89. } while (0)
  90. /**
  91. * @brief 初始化local APIC定时器
  92. *
  93. */
  94. void apic_timer_init();
  95. #pragma GCC pop_options