1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
//! NT Time routines.

use ::shared::{SYSTEMTIME};

#[cfg(target_arch = "x86_64")]
use ::shared::{KUSER_SHARED_DATA};


extern "system"
{
	// The following exports exists only on x86 kernels.
	// x64 drivers must use KUSER_SHARED_DATA to obtain these values.

	#[cfg(target_arch = "x86")]
	fn KeQuerySystemTime(CurrentTime: *mut SYSTEMTIME);
	#[cfg(target_arch = "x86")]
	fn KeQueryTickCount(TickCount: *mut i64);

	/// Converts a GMT system time value to the local system time for the current time zone.
	pub fn ExSystemTimeToLocalTime(SystemTime: *const SYSTEMTIME, LocalTime: *mut SYSTEMTIME);
}


/// Obtains the current system time.
#[cfg(target_arch = "x86")]
pub fn QuerySystemTime() -> SYSTEMTIME {
	let mut t = 0i64;
	unsafe { KeQuerySystemTime(&mut t) };
	return t;
}

/// Obtains the current system time.
#[cfg(target_arch = "x86_64")]
pub fn QuerySystemTime() -> SYSTEMTIME {
	let shared = KUSER_SHARED_DATA::get();
	SYSTEMTIME::from(shared.SystemTime)
}


/// A count of the interval timer interrupts that have occurred since the system was booted.
#[cfg(target_arch = "x86")]
pub fn QueryTickCount() -> i64 {
	let mut t = 0i64;
	unsafe { KeQueryTickCount(&mut t) };
	return t;
}


/// A count of the interval timer interrupts that have occurred since the system was booted.
#[cfg(target_arch = "x86_64")]
pub fn QueryTickCount() -> i64 {
	let shared = KUSER_SHARED_DATA::get();
	SYSTEMTIME::from(shared.TickCount)
}