|
1 | 1 | #![allow(dead_code)] |
2 | 2 |
|
3 | | -use core::hash::{Hash, Hasher}; |
| 3 | +use core::hash::Hash; |
4 | 4 |
|
5 | 5 | use super::hermit_abi::{self, CLOCK_MONOTONIC, CLOCK_REALTIME, timespec}; |
6 | | -use crate::cmp::Ordering; |
7 | 6 | use crate::ops::{Add, AddAssign, Sub, SubAssign}; |
| 7 | +use crate::sys::common::timespec::Timespec; |
8 | 8 | use crate::time::Duration; |
9 | 9 |
|
10 | 10 | const NSEC_PER_SEC: i32 = 1_000_000_000; |
11 | 11 |
|
12 | | -#[derive(Copy, Clone, Debug)] |
13 | | -struct Timespec { |
14 | | - t: timespec, |
15 | | -} |
16 | | - |
17 | | -impl Timespec { |
18 | | - const fn zero() -> Timespec { |
19 | | - Timespec { t: timespec { tv_sec: 0, tv_nsec: 0 } } |
20 | | - } |
21 | | - |
22 | | - const fn new(tv_sec: i64, tv_nsec: i32) -> Timespec { |
23 | | - assert!(tv_nsec >= 0 && tv_nsec < NSEC_PER_SEC); |
24 | | - // SAFETY: The assert above checks tv_nsec is within the valid range |
25 | | - Timespec { t: timespec { tv_sec, tv_nsec } } |
26 | | - } |
27 | | - |
28 | | - fn sub_timespec(&self, other: &Timespec) -> Result<Duration, Duration> { |
29 | | - fn sub_ge_to_unsigned(a: i64, b: i64) -> u64 { |
30 | | - debug_assert!(a >= b); |
31 | | - a.wrapping_sub(b).cast_unsigned() |
32 | | - } |
33 | | - |
34 | | - if self >= other { |
35 | | - // Logic here is identical to Unix version of `Timestamp::sub_timespec`, |
36 | | - // check comments there why operations do not overflow. |
37 | | - Ok(if self.t.tv_nsec >= other.t.tv_nsec { |
38 | | - Duration::new( |
39 | | - sub_ge_to_unsigned(self.t.tv_sec, other.t.tv_sec), |
40 | | - (self.t.tv_nsec - other.t.tv_nsec) as u32, |
41 | | - ) |
42 | | - } else { |
43 | | - Duration::new( |
44 | | - sub_ge_to_unsigned(self.t.tv_sec - 1, other.t.tv_sec), |
45 | | - (self.t.tv_nsec + NSEC_PER_SEC - other.t.tv_nsec) as u32, |
46 | | - ) |
47 | | - }) |
48 | | - } else { |
49 | | - match other.sub_timespec(self) { |
50 | | - Ok(d) => Err(d), |
51 | | - Err(d) => Ok(d), |
52 | | - } |
53 | | - } |
54 | | - } |
55 | | - |
56 | | - fn checked_add_duration(&self, other: &Duration) -> Option<Timespec> { |
57 | | - let mut secs = self.t.tv_sec.checked_add_unsigned(other.as_secs())?; |
58 | | - |
59 | | - // Nano calculations can't overflow because nanos are <1B which fit |
60 | | - // in a u32. |
61 | | - let mut nsec = other.subsec_nanos() + u32::try_from(self.t.tv_nsec).unwrap(); |
62 | | - if nsec >= NSEC_PER_SEC.try_into().unwrap() { |
63 | | - nsec -= u32::try_from(NSEC_PER_SEC).unwrap(); |
64 | | - secs = secs.checked_add(1)?; |
65 | | - } |
66 | | - Some(Timespec { t: timespec { tv_sec: secs, tv_nsec: nsec as _ } }) |
67 | | - } |
68 | | - |
69 | | - fn checked_sub_duration(&self, other: &Duration) -> Option<Timespec> { |
70 | | - let mut secs = self.t.tv_sec.checked_sub_unsigned(other.as_secs())?; |
71 | | - |
72 | | - // Similar to above, nanos can't overflow. |
73 | | - let mut nsec = self.t.tv_nsec as i32 - other.subsec_nanos() as i32; |
74 | | - if nsec < 0 { |
75 | | - nsec += NSEC_PER_SEC as i32; |
76 | | - secs = secs.checked_sub(1)?; |
77 | | - } |
78 | | - Some(Timespec { t: timespec { tv_sec: secs, tv_nsec: nsec as _ } }) |
79 | | - } |
80 | | -} |
81 | | - |
82 | | -impl PartialEq for Timespec { |
83 | | - fn eq(&self, other: &Timespec) -> bool { |
84 | | - self.t.tv_sec == other.t.tv_sec && self.t.tv_nsec == other.t.tv_nsec |
85 | | - } |
86 | | -} |
87 | | - |
88 | | -impl Eq for Timespec {} |
89 | | - |
90 | | -impl PartialOrd for Timespec { |
91 | | - fn partial_cmp(&self, other: &Timespec) -> Option<Ordering> { |
92 | | - Some(self.cmp(other)) |
93 | | - } |
94 | | -} |
95 | | - |
96 | | -impl Ord for Timespec { |
97 | | - fn cmp(&self, other: &Timespec) -> Ordering { |
98 | | - let me = (self.t.tv_sec, self.t.tv_nsec); |
99 | | - let other = (other.t.tv_sec, other.t.tv_nsec); |
100 | | - me.cmp(&other) |
101 | | - } |
102 | | -} |
103 | | - |
104 | | -impl Hash for Timespec { |
105 | | - fn hash<H: Hasher>(&self, state: &mut H) { |
106 | | - self.t.tv_sec.hash(state); |
107 | | - self.t.tv_nsec.hash(state); |
108 | | - } |
109 | | -} |
110 | | - |
111 | 12 | #[derive(Copy, Clone, PartialEq, Eq, PartialOrd, Ord, Debug, Hash)] |
112 | 13 | pub struct Instant(Timespec); |
113 | 14 |
|
114 | 15 | impl Instant { |
115 | 16 | pub fn now() -> Instant { |
116 | | - let mut time: Timespec = Timespec::zero(); |
117 | | - let _ = unsafe { hermit_abi::clock_gettime(CLOCK_MONOTONIC, &raw mut time.t) }; |
| 17 | + let mut time: timespec = timespec { tv_sec: 0, tv_nsec: 0 }; |
| 18 | + let _ = unsafe { hermit_abi::clock_gettime(CLOCK_MONOTONIC, &raw mut time) }; |
118 | 19 |
|
119 | | - Instant(time) |
| 20 | + Instant(Timespec::new(time.tv_sec, time.tv_nsec as i64).unwrap()) |
120 | 21 | } |
121 | 22 |
|
122 | 23 | #[stable(feature = "time2", since = "1.8.0")] |
@@ -210,14 +111,13 @@ pub const UNIX_EPOCH: SystemTime = SystemTime(Timespec::zero()); |
210 | 111 |
|
211 | 112 | impl SystemTime { |
212 | 113 | pub fn new(tv_sec: i64, tv_nsec: i32) -> SystemTime { |
213 | | - SystemTime(Timespec::new(tv_sec, tv_nsec)) |
| 114 | + SystemTime(Timespec::new(tv_sec, tv_nsec as i64).unwrap()) |
214 | 115 | } |
215 | 116 |
|
216 | 117 | pub fn now() -> SystemTime { |
217 | | - let mut time: Timespec = Timespec::zero(); |
218 | | - let _ = unsafe { hermit_abi::clock_gettime(CLOCK_REALTIME, &raw mut time.t) }; |
219 | | - |
220 | | - SystemTime(time) |
| 118 | + let mut time: timespec = timespec { tv_sec: 0, tv_nsec: 0 }; |
| 119 | + let _ = unsafe { hermit_abi::clock_gettime(CLOCK_REALTIME, &raw mut time) }; |
| 120 | + SystemTime::new(time.tv_sec, time.tv_nsec) |
221 | 121 | } |
222 | 122 |
|
223 | 123 | pub fn sub_time(&self, other: &SystemTime) -> Result<Duration, Duration> { |
|
0 commit comments