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 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791
//! Performant cross-platform timing with goodies.
//!
//! `quanta` provides a simple and fast API for measuring the current time and the duration between
//! events. It does this by providing a thin layer on top of native OS timing functions, or, if
//! available, using the Time Stamp Counter feature found on modern CPUs.
//!
//! # Design
//! Internally, `quanta` maintains the concept of two potential clock sources: a reference clock and
//! a source clock.
//!
//! The reference clock is provided by the OS, and always available. It is equivalent to what is
//! provided by the standard library in terms of the underlying system calls being made. As it
//! uses the native timing facilities provided by the operating system, we ultimately depend on the
//! OS itself to give us a stable and correct value.
//!
//! The source clock is a potential clock source based on the Time Stamp Counter feature found on
//! modern CPUs. If the TSC feature is not present or is not reliable enough, `quanta` will
//! transparently utilize the reference clock instead.
//!
//! Depending on the underlying processor(s) in the system, `quanta` will figure out the most
//! accurate/efficient way to calibrate the source clock to the reference clock in order to provide
//! measurements scaled to wall clock time.
//!
//! Details on TSC support, and calibration, are detailed below.
//!
//! # Features
//! Beyond simply taking measurements of the current time, `quanta` provides features for more easily
//! working with clocks, as well as being able to enhance performance further:
//! - `Clock` can be mocked for testing
//! - globally accessible "recent" time with amortized overhead
//!
//! For any code that uses a `Clock`, a mocked version can be substituted. This allows for
//! application authors to control the time in tests, which allows simulating not only the normal
//! passage of time but provides the ability to warp time forwards and backwards in order to test
//! corner cases in logic, etc. Creating a mocked clock can be acheived with [`Clock::mock`], and
//! [`Mock`] contains more details on mock usage.
//!
//! `quanta` also provides a "recent" time feature, which allows a slightly-delayed version of time
//! to be provided to callers, trading accuracy for speed of access. An upkeep thread is spawned,
//! which is responsible for taking measurements and updating the global recent time. Callers then
//! can access the cached value by calling `Clock::recent`. This interface can be 4-10x faster
//! than directly calling `Clock::now`, even when TSC support is available. As the upkeep thread
//! is the only code updating the recent time, the accuracy of the value given to callers is
//! limited by how often the upkeep thread updates the time, thus the trade off between accuracy
//! and speed of access.
//!
//! # Feature Flags
//! `quanta` comes with feature flags that enable convenient conversions to time types in
//! other popular crates, such as:
//! - `prost` - provides an implementation into [`Timestamp`][prost_types_timestamp] from
//! `prost_types`
//!
//! # Platform Support
//! At a high level, `quanta` carries support for most major operating systems out of the box:
//! - Windows ([QueryPerformanceCounter])
//! - macOS/OS X/iOS ([mach_continuous_time])
//! - Linux/*BSD/Solaris ([clock_gettime])
//!
//! These platforms are supported in the "reference" clock sense, and support for using the Time
//! Stamp Counter as a clocksource is more subtle, and explained below.
//!
//! # Time Stamp Counter support
//! Accessing the TSC requires being on the x86_64 architecture, with access to SSE2. Additionally,
//! the processor must support either constant or nonstop/invariant TSC. This ensures that the TSC
//! ticks at a constant rate which can be easily scaled.
//!
//! A caveat is that "constant" TSC doesn't account for all possible power states (levels of power
//! down or sleep that a CPU can enter to save power under light load, etc) and so a constant TSC
//! can lead to drift in measurements over time, after they've been scaled to reference time.
//!
//! This is a limitation of the TSC mode, as well as the nature of `quanta` not being able to know,
//! as the OS would, when a power state transition has happened, and thus compensate with a
//! recalibration. Nonstop/invariant TSC does not have this limitation and is stable over long
//! periods of time.
//!
//! Roughly speaking, the following list contains the beginning model/generation of processors
//! where you should be able to expect having invariant TSC support:
//! - Intel Nehalem and newer for server-grade
//! - Intel Skylake and newer for desktop-grade
//! - VIA Centaur Nano and newer (circumstantial evidence here)
//! - AMD Phenom and newer
//!
//! Ultimately, `quanta` will query CPUID information to determine if the processor has the
//! required features to use the TSC.
//!
//! # Calibration
//! As the TSC doesn't necessarily tick at reference scale -- i.e. one tick isn't always one
//! nanosecond -- we have to apply a scaling factor when converting from source to reference time
//! scale. We acquire this scaling factor by querying the processor or calibrating our source
//! clock to the reference clock.
//!
//! In some cases, on newer processors, the frequency of the TSC can be queried directly, providing
//! a fixed scaling factor with no further calibration necessary. In other cases, `quanta` will
//! have to run its own calibration before the clock is ready to be used: repeatedly taking
//! measurements from both the reference and source clocks until a stable scaling factor has been
//! established.
//!
//! This calibration is stored globally and reused. However, the first `Clock` that is created in
//! an application will block for a small period of time as it runs this calibration loop. The
//! time spent in the calibration loop is limited to 200ms overall. In practice, `quanta` will
//! reach a stable calibration quickly (usually 10-20ms, if not less) and so this deadline is
//! unlikely to be reached.
//!
//! # Caveats
//! Utilizing the TSC can be a tricky affair, and so here is a list of caveats that may or may not
//! apply, and is in no way exhaustive:
//! - CPU hotplug behavior is undefined
//! - raw values may time warp
//! - measurements from the TSC may drift past or behind the comparable reference clock
//!
//! # WASM support
//!
//! This library can be built for WASM targets, but in this case the resolution
//! and accuracy of measurements can be limited by the WASM environment. In
//! particular, when running on the `wasm32-unknown-unknown` target in browsers,
//! `quanta` will use [windows.performance.now] as a clock. This mean the
//! accuracy is limited to milliseconds instead of the usual nanoseconds on
//! other targets. When running within a WASI environment (target
//! `wasm32-wasi`), the accuracy of the clock depends on the VM implementation.
//!
//! [QueryPerformanceCounter]: https://msdn.microsoft.com/en-us/library/ms644904(v=VS.85).aspx
//! [mach_continuous_time]: https://developer.apple.com/documentation/kernel/1646199-mach_continuous_time
//! [clock_gettime]: https://linux.die.net/man/3/clock_gettime
//! [prost_types_timestamp]: https://docs.rs/prost-types/0.7.0/prost_types/struct.Timestamp.html
//! [windows.performance.now]: https://developer.mozilla.org/en-US/docs/Web/API/Performance/now
use crossbeam_utils::atomic::AtomicCell;
use std::time::Duration;
use std::{cell::RefCell, sync::Arc};
use once_cell::sync::OnceCell;
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
use raw_cpuid::CpuId;
mod monotonic;
use self::monotonic::Monotonic;
mod counter;
use self::counter::Counter;
mod mock;
pub use self::mock::{IntoNanoseconds, Mock};
mod instant;
pub use self::instant::Instant;
mod upkeep;
pub use self::upkeep::{Error, Handle, Upkeep};
mod stats;
use self::stats::Variance;
static GLOBAL_CLOCK: OnceCell<Clock> = OnceCell::new();
static GLOBAL_RECENT: AtomicCell<u64> = AtomicCell::new(0);
static GLOBAL_CALIBRATION: OnceCell<Calibration> = OnceCell::new();
thread_local! {
static CLOCK_OVERRIDE: RefCell<Option<Clock>> = RefCell::new(None);
}
// Run 500 rounds of calibration before we start actually seeing what the numbers look like.
const MINIMUM_CAL_ROUNDS: u64 = 500;
// We want our maximum error to be 10 nanoseconds.
const MAXIMUM_CAL_ERROR_NS: u64 = 10;
// Don't run the calibration loop for longer than 200ms of wall time.
const MAXIMUM_CAL_TIME: Duration = Duration::from_millis(200);
#[allow(dead_code)]
#[derive(Debug)]
enum ClockType {
Monotonic(Monotonic),
Counter(AtomicCell<u64>, Monotonic, Counter, Calibration),
Mock(Arc<Mock>),
}
#[derive(Debug, Copy, Clone)]
pub(crate) struct Calibration {
ref_time: u64,
src_time: u64,
scale_factor: u64,
scale_shift: u32,
}
impl Calibration {
pub fn new() -> Calibration {
Calibration {
ref_time: 0,
src_time: 0,
scale_factor: 1,
scale_shift: 1,
}
}
fn scale_src_to_ref(&self, src_raw: u64) -> u64 {
let delta = src_raw.saturating_sub(self.src_time);
let scaled = mul_div_po2_u64(delta, self.scale_factor, self.scale_shift);
scaled + self.ref_time
}
fn calibrate(&mut self, reference: &Monotonic, source: &Counter) {
// future improvement: read the TSC frequency directly with something like the
// code in this PR: https://github.com/hermitcore/uhyve/pull/24
let mut variance = Variance::default();
let deadline = reference.now() + MAXIMUM_CAL_TIME.as_nanos() as u64;
self.ref_time = reference.now();
self.src_time = source.now();
let loop_delta = 1000;
loop {
// Busy loop to burn some time.
let mut last = reference.now();
let target = last + loop_delta;
while last < target {
last = reference.now();
}
// We put an upper bound on how long we run calibration before to provide a predictable
// overhead to the calibration process. In practice, even if we hit the calibration
// deadline, we should still have run a sufficient number of rounds to get an accurate
// calibration.
if last >= deadline {
break;
}
// Adjust our calibration before we take our measurement.
self.adjust_cal_ratio(reference, source);
let r_time = reference.now();
let s_raw = source.now();
let s_time = self.scale_src_to_ref(s_raw);
variance.add(s_time as f64 - r_time as f64);
// If we've collected enough samples, check what the mean and mean error are. If we're
// already within the target bounds, we can break out of the calibration loop early.
if variance.has_significant_result() {
let mean = variance.mean().abs();
let mean_error = variance.mean_error().abs();
let mwe = variance.mean_with_error();
let samples = variance.samples();
if samples > MINIMUM_CAL_ROUNDS
&& mwe < MAXIMUM_CAL_ERROR_NS
&& mean_error / mean <= 1.0
{
break;
}
}
}
}
fn adjust_cal_ratio(&mut self, reference: &Monotonic, source: &Counter) {
// Overall algorithm: measure the delta between our ref/src_time values and "now" versions
// of them, calculate the ratio between the deltas, and then find a numerator and
// denominator to express that ratio such that the denominator is always a power of two.
//
// In practice, this means we take the "source" delta, and find the next biggest number that
// is a power of two. We then figure out the ratio that describes the difference between
// _those_ two values, and multiple the "reference" delta by that much, which becomes our
// numerator while the power-of-two "source" delta becomes our denominator.
//
// Then, conversion from a raw value simply becomes a multiply and a bit shift instead of a
// multiply and full-blown divide.
let ref_end = reference.now();
let src_end = source.end();
let ref_d = ref_end.wrapping_sub(self.ref_time);
let src_d = src_end.wrapping_sub(self.src_time);
// TODO: we should almost never get a zero here because that would mean denom was greater
// than 2^63 which is already a red flag.. but i'm not 100% sure if we can prove it well
// enough to simply keep the panic around? gotta think on this
let src_d_po2 = src_d.next_power_of_two();
if src_d_po2 == 0 {
panic!("po2_denom was zero!");
}
// TODO: lossy conversion back and forth just to get an approximate value, can we do better
// with integer math? not sure
let po2_ratio = src_d_po2 as f64 / src_d as f64;
self.scale_factor = (ref_d as f64 * po2_ratio) as u64;
self.scale_shift = src_d_po2.trailing_zeros();
}
}
impl Default for Calibration {
fn default() -> Self {
Self::new()
}
}
/// Unified clock for taking measurements.
#[derive(Debug, Clone)]
pub struct Clock {
inner: ClockType,
}
impl Clock {
/// Creates a new clock with the optimal reference and source clocks.
///
/// Support for TSC, etc, are checked at the time of creation, not compile-time.
pub fn new() -> Clock {
let reference = Monotonic::new();
let inner = if has_tsc_support() {
let source = Counter::new();
let calibration = GLOBAL_CALIBRATION.get_or_init(|| {
let mut calibration = Calibration::new();
calibration.calibrate(&reference, &source);
calibration
});
ClockType::Counter(AtomicCell::new(0), reference, source, *calibration)
} else {
ClockType::Monotonic(reference)
};
Clock { inner }
}
/// Creates a new clock that is mocked for controlling the underlying time.
///
/// Returns a [`Clock`] instance and a handle to the underlying [`Mock`] source so that the
/// caller can control the passage of time.
pub fn mock() -> (Clock, Arc<Mock>) {
let mock = Arc::new(Mock::new());
let clock = Clock {
inner: ClockType::Mock(mock.clone()),
};
(clock, mock)
}
/// Gets the current time, scaled to reference time.
///
/// This method is the spiritual equivalent of [`std::time::Instant::now`]. It is guaranteed
/// to return a monotonically increasing value between calls to the same `Clock` instance.
///
/// Returns an [`Instant`].
pub fn now(&self) -> Instant {
match &self.inner {
ClockType::Monotonic(monotonic) => Instant(monotonic.now()),
ClockType::Counter(last, _, counter, _) => {
let now = counter.now();
// Update the last timestamp with `now`, if `now` is greater
// than the current value.
// TODO: replace with `AtomicCell::fetch_max` once `crossbeam-utils` implements it.
let last = last.fetch_update(|current| Some(current.max(now))).unwrap();
// `fetch_max` always returns the previous value, so we need to
// see which is *actually* the max.
let actual = std::cmp::max(now, last);
self.scaled(actual)
}
ClockType::Mock(mock) => Instant(mock.value()),
}
}
/// Gets the underlying time from the fastest available clock source.
///
/// As the clock source may or may not be the TSC, value is not guaranteed to be in nanoseconds
/// or to be monotonic. Value can be scaled to reference time by calling either [`scaled`]
/// or [`delta`].
///
/// If you need maximum accuracy in your measurements, consider using [`start`] and [`end`].
///
/// [`scaled`]: Clock::scaled
/// [`delta`]: Clock::delta
/// [`start`]: Clock::start
/// [`end`]: Clock::end
pub fn raw(&self) -> u64 {
match &self.inner {
ClockType::Monotonic(monotonic) => monotonic.now(),
ClockType::Counter(_, _, counter, _) => counter.now(),
ClockType::Mock(mock) => mock.value(),
}
}
/// Gets the underlying time from the source clock, specific to starting an operation.
///
/// Value is not guaranteed to be in nanoseconds.
///
/// Provides the same functionality as [`raw`], but tries to ensure that no extra CPU
/// instructions end up executing after the measurement is taken. Since normal processors are
/// typically out-of-order, other operations that logically come before a call to this method
/// could be reordered to come after the measurement, thereby skewing the overall time
/// measured.
///
/// [`raw`]: Clock::raw
pub fn start(&self) -> u64 {
match &self.inner {
ClockType::Monotonic(monotonic) => monotonic.now(),
ClockType::Counter(_, _, counter, _) => counter.start(),
ClockType::Mock(mock) => mock.value(),
}
}
/// Gets the underlying time from the source clock, specific to ending an operation.
///
/// Value is not guaranteed to be in nanoseconds.
///
/// Provides the same functionality as [`raw`], but tries to ensure that no extra CPU
/// instructions end up executing before the measurement is taken. Since normal processors are
/// typically out-of-order, other operations that logically come after a call to this method
/// could be reordered to come before the measurement, thereby skewing the overall time
/// measured.
///
/// [`raw`]: Clock::raw
pub fn end(&self) -> u64 {
match &self.inner {
ClockType::Monotonic(monotonic) => monotonic.now(),
ClockType::Counter(_, _, counter, _) => counter.end(),
ClockType::Mock(mock) => mock.value(),
}
}
/// Scales a raw measurement to reference time.
///
/// You must scale raw measurements to ensure your result is in nanoseconds. The raw
/// measurement is not guaranteed to be in nanoseconds and may vary. It is only OK to avoid
/// scaling raw measurements if you don't need actual nanoseconds.
///
/// Returns an [`Instant`].
pub fn scaled(&self, value: u64) -> Instant {
let scaled = match &self.inner {
ClockType::Counter(_, _, _, calibration) => scale_src_to_ref(value, calibration),
_ => value,
};
Instant(scaled)
}
/// Calculates the delta between two measurements, and scales to reference time.
///
/// This method is slightly faster when you know you need the delta between two raw
/// measurements, or a start/end measurement, than using [`scaled`] for both conversions.
///
/// [`scaled`]: Clock::scaled
pub fn delta(&self, start: u64, end: u64) -> Duration {
// Safety: we want wrapping_sub on the end/start delta calculation so that two measurements
// split across a rollover boundary still return the right result. However, we also know
// the TSC could potentially give us different values between cores/sockets, so we're just
// doing our due diligence here to make sure we're not about to create some wacky duration.
if end <= start {
return Duration::new(0, 0);
}
let delta = end.wrapping_sub(start);
let scaled = match &self.inner {
ClockType::Counter(_, _, _, calibration) => {
mul_div_po2_u64(delta, calibration.scale_factor, calibration.scale_shift)
}
_ => delta,
};
Duration::from_nanos(scaled)
}
/// Gets the most recent current time, scaled to reference time.
///
/// This method provides ultra-low-overhead access to a slightly-delayed version of the current
/// time. Instead of querying the underlying source clock directly, a shared, global value is
/// read directly without the need to scale to reference time.
///
/// The upkeep thread must be started in order to update the time. You can read the
/// documentation for [`Builder`] for more information on starting the upkeep thread, as well
/// as the details of the "current time" mechanism.
///
/// If the upkeep thread has not been started, the return value will be `0`.
///
/// Returns an [`Instant`].
pub fn recent(&self) -> Instant {
match &self.inner {
ClockType::Mock(mock) => Instant(mock.value()),
_ => Instant(GLOBAL_RECENT.load()),
}
}
/// Updates the recent current time.
///
/// Most callers should use the existing [`Builder`] machinery for spawning a background thread
/// to handle upkeep, rather than calling [`upkeep`] directly.
pub fn upkeep(value: Instant) {
GLOBAL_RECENT.store(value.0);
}
}
impl Default for Clock {
fn default() -> Clock {
Clock::new()
}
}
// A manual `Clone` impl is required because `atomic_shim`'s `AtomicU64` is not `Clone`.
impl Clone for ClockType {
fn clone(&self) -> Self {
match self {
ClockType::Mock(mock) => ClockType::Mock(mock.clone()),
ClockType::Monotonic(monotonic) => ClockType::Monotonic(monotonic.clone()),
ClockType::Counter(last, monotonic, counter, calibration) => ClockType::Counter(
AtomicCell::new(last.load()),
monotonic.clone(),
counter.clone(),
*calibration,
),
}
}
}
/// Sets this clock as the default for the duration of a closure.
///
/// This will only affect calls made against [`Instant`]. [`Clock`] is always self-contained.
pub fn with_clock<T>(clock: &Clock, f: impl FnOnce() -> T) -> T {
CLOCK_OVERRIDE.with(|current| {
let old = current.replace(Some(clock.clone()));
let result = f();
let _ = current.replace(old);
result
})
}
/// Sets the global recent time.
///
/// While callers should typically prefer to use [`Upkeep`] to establish a background thread in
/// order to drive the global recent time, this function allows callers to customize how the global
/// recent time is updated. For example, programs using an asynchronous runtime may prefer to
/// schedule a task that does the updating, avoiding an extra thread.
pub fn set_recent(instant: Instant) {
GLOBAL_RECENT.store(instant.as_u64());
}
#[inline]
pub(crate) fn get_now() -> Instant {
if let Some(instant) = CLOCK_OVERRIDE.with(|clock| clock.borrow().as_ref().map(|c| c.now())) {
instant
} else {
GLOBAL_CLOCK.get_or_init(Clock::new).now()
}
}
#[inline]
pub(crate) fn get_recent() -> Instant {
// We make a small trade-off here where if the global recent time isn't zero, we use that,
// regardless of whether or not there's a thread-specific clock override. Otherwise, we would
// blow our performance budget.
//
// Given that global recent time shouldn't ever be getting _actually_ updated in tests, this
// should be a reasonable trade-off.
let recent = GLOBAL_RECENT.load();
if recent != 0 {
Instant(recent)
} else {
get_now()
}
}
#[inline]
fn scale_src_to_ref(src_raw: u64, cal: &Calibration) -> u64 {
let delta = src_raw.saturating_sub(cal.src_time);
let scaled = mul_div_po2_u64(delta, cal.scale_factor, cal.scale_shift);
scaled + cal.ref_time
}
#[inline]
fn mul_div_po2_u64(value: u64, numer: u64, denom: u32) -> u64 {
// Modified muldiv routine where the denominator has to be a power of two. `denom` is expected
// to be the number of bits to shift, not the actual decimal value.
let mut v: u128 = value as u128;
v *= numer as u128;
v >>= denom;
v as u64
}
#[allow(dead_code)]
#[cfg(all(target_arch = "x86_64", target_feature = "sse2"))]
fn has_tsc_support() -> bool {
read_cpuid_nonstop_tsc() && read_cpuid_rdtscp_support()
}
#[allow(dead_code)]
#[cfg(not(all(target_arch = "x86_64", target_feature = "sse2")))]
fn has_tsc_support() -> bool {
false
}
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
fn read_cpuid_nonstop_tsc() -> bool {
let cpuid = CpuId::new();
cpuid
.get_advanced_power_mgmt_info()
.map_or(false, |efi| efi.has_invariant_tsc())
}
#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
fn read_cpuid_rdtscp_support() -> bool {
let cpuid = CpuId::new();
cpuid
.get_extended_processor_and_feature_identifiers()
.map_or(false, |efi| efi.has_rdtscp())
}
#[cfg(test)]
pub mod tests {
use super::{Clock, Monotonic};
use average::{Merge, Variance};
#[cfg(all(target_arch = "wasm32", target_os = "unknown"))]
mod configure_wasm_tests {
// until https://github.com/rustwasm/wasm-bindgen/issues/2571 is resolved
// these tests will only run in browsers
wasm_bindgen_test::wasm_bindgen_test_configure!(run_in_browser);
}
#[test]
#[cfg_attr(
all(target_arch = "wasm32", target_os = "unknown"),
wasm_bindgen_test::wasm_bindgen_test
)]
fn test_mock() {
let (clock, mock) = Clock::mock();
assert_eq!(clock.now().as_u64(), 0);
mock.increment(42);
assert_eq!(clock.now().as_u64(), 42);
}
#[test]
#[cfg_attr(
all(target_arch = "wasm32", target_os = "unknown"),
wasm_bindgen_test::wasm_bindgen_test
)]
fn test_now() {
let clock = Clock::new();
assert!(clock.now().as_u64() > 0);
}
#[test]
#[cfg_attr(
all(target_arch = "wasm32", target_os = "unknown"),
wasm_bindgen_test::wasm_bindgen_test
)]
fn test_raw() {
let clock = Clock::new();
assert!(clock.raw() > 0);
}
#[test]
#[cfg_attr(
all(target_arch = "wasm32", target_os = "unknown"),
wasm_bindgen_test::wasm_bindgen_test
)]
fn test_start() {
let clock = Clock::new();
assert!(clock.start() > 0);
}
#[test]
#[cfg_attr(
all(target_arch = "wasm32", target_os = "unknown"),
wasm_bindgen_test::wasm_bindgen_test
)]
fn test_end() {
let clock = Clock::new();
assert!(clock.end() > 0);
}
#[test]
#[cfg_attr(
all(target_arch = "wasm32", target_os = "unknown"),
wasm_bindgen_test::wasm_bindgen_test
)]
fn test_scaled() {
let clock = Clock::new();
let raw = clock.raw();
let scaled = clock.scaled(raw);
assert!(scaled.as_u64() > 0);
}
#[test]
#[cfg_attr(not(feature = "flaky_tests"), ignore)]
#[cfg_attr(
all(target_arch = "wasm32", target_os = "unknown"),
wasm_bindgen_test::wasm_bindgen_test
)]
fn test_reference_source_calibration() {
let clock = Clock::new();
let reference = Monotonic::new();
let loops = 10000;
let mut overall = Variance::new();
let mut src_samples = [0u64; 1024];
let mut ref_samples = [0u64; 1024];
for _ in 0..loops {
for i in 0..1024 {
src_samples[i] = clock.now().as_u64();
ref_samples[i] = reference.now();
}
let mut last = None;
let is_src_monotonic = src_samples.iter().all(|n| match last {
None => {
last = Some(n);
true
}
Some(on) => {
if n >= on {
last = Some(n);
true
} else {
false
}
}
});
assert!(is_src_monotonic);
let mut last = None;
let is_ref_monotonic = ref_samples.iter().all(|n| match last {
None => {
last = Some(n);
true
}
Some(on) => {
if n >= on {
last = Some(n);
true
} else {
false
}
}
});
assert!(is_ref_monotonic);
let local = src_samples
.iter()
.zip(ref_samples.iter())
.map(|(s, r)| (*s as f64, *r as f64))
.map(|(s, r)| (r - s).abs())
.collect::<Variance>();
overall.merge(&local);
}
println!(
"reference/source delta: mean={} error={} mean-var={} samples={}",
overall.mean(),
overall.error(),
overall.variance_of_mean(),
overall.len(),
);
// If things are out of sync more than 1000ns, something is likely scaled wrong.
assert!(overall.mean() < 1000.0);
}
#[test]
#[cfg_attr(not(feature = "flaky_tests"), ignore)]
#[cfg_attr(
all(target_arch = "wasm32", target_os = "unknown"),
wasm_bindgen_test::wasm_bindgen_test
)]
fn test_reference_self_calibration() {
let reference = Monotonic::new();
let loops = 10000;
let samples = 1024;
let mut overall = Variance::new();
let mut deltas = Vec::with_capacity(samples);
deltas.reserve(samples);
for _ in 0..loops {
deltas.clear();
for _ in 0..samples {
let rstart = reference.now();
let rend = reference.now();
deltas.push(rend - rstart);
}
let local = deltas.iter().map(|i| *i as f64).collect::<Variance>();
overall.merge(&local);
}
println!(
"reference/reference inter-call delta: mean={} error={} mean-var={}",
overall.mean(),
overall.error(),
overall.variance_of_mean()
);
// We should be able to call the reference clock, back-to-back, within 2000 nanoseconds. If
// it's slower than that, something is probably amiss on the system the test is running under.
assert!(overall.mean() < 2000.0);
}
}