diff --git a/include/time.h b/include/time.h index a5763c71..dc5b8020 100644 --- a/include/time.h +++ b/include/time.h @@ -25,11 +25,19 @@ #ifndef KTF_TIME_H #define KTF_TIME_H +#include #include #include typedef uint64_t time_t; -static inline void sleep(time_t ms) { pit_sleep(ms); } +static inline void sleep(time_t ms) { + if (is_apic_timer_enabled()) { + apic_timer_sleep(ms); + } + else { + pit_sleep(ms); + } +} #endif