rust/src/libstd/sys/wasi/time.rs

use crate::time::Duration;
use crate::mem;
use crate::sys::cvt_wasi;
use libc;

#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Debug, Hash)]
pub struct Instant(Duration);

#[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Debug, Hash)]
pub struct SystemTime(Duration);

pub const UNIX_EPOCH: SystemTime = SystemTime(Duration::from_secs(0));

fn current_time(clock: u32) -> Duration {
    unsafe {
        let mut ts = mem::zeroed();
        cvt_wasi(libc::__wasi_clock_time_get(
            clock,
            1, // precision... seems ignored though?
            &mut ts,
        )).unwrap();
        Duration::new(
            (ts / 1_000_000_000) as u64,
            (ts % 1_000_000_000) as u32,
        )
    }
}

impl Instant {
    pub fn now() -> Instant {
        Instant(current_time(libc::__WASI_CLOCK_MONOTONIC))
    }

    pub const fn zero() -> Instant {
        Instant(Duration::from_secs(0))
    }

    pub fn actually_monotonic() -> bool {
        true
    }

    pub fn checked_sub_instant(&self, other: &Instant) -> Option<Duration> {
        self.0.checked_sub(other.0)
    }

    pub fn checked_add_duration(&self, other: &Duration) -> Option<Instant> {
        Some(Instant(self.0.checked_add(*other)?))
    }

    pub fn checked_sub_duration(&self, other: &Duration) -> Option<Instant> {
        Some(Instant(self.0.checked_sub(*other)?))
    }
}

impl SystemTime {
    pub fn now() -> SystemTime {
        SystemTime(current_time(libc::__WASI_CLOCK_REALTIME))
    }

    pub fn from_wasi_timestamp(ts: libc::__wasi_timestamp_t) -> SystemTime {
        SystemTime(Duration::from_nanos(ts))
    }

    pub fn sub_time(&self, other: &SystemTime)
                    -> Result<Duration, Duration> {
        self.0.checked_sub(other.0).ok_or_else(|| other.0 - self.0)
    }

    pub fn checked_add_duration(&self, other: &Duration) -> Option<SystemTime> {
        Some(SystemTime(self.0.checked_add(*other)?))
    }

    pub fn checked_sub_duration(&self, other: &Duration) -> Option<SystemTime> {
        Some(SystemTime(self.0.checked_sub(*other)?))
    }
}