Properly detect overflow in Instance +/- Duration.
Avoid unchecked cast from `u64` to `i64`. Use `try_into()` for checked cast. (On Unix, cast to `time_t` instead of `i64`.)
This commit is contained in:
@@ -12,6 +12,7 @@ use cmp::Ordering;
|
||||
use fmt;
|
||||
use sys::{cvt, syscall};
|
||||
use time::Duration;
|
||||
use convert::TryInto;
|
||||
|
||||
const NSEC_PER_SEC: u64 = 1_000_000_000;
|
||||
|
||||
@@ -40,8 +41,12 @@ impl Timespec {
|
||||
}
|
||||
|
||||
fn add_duration(&self, other: &Duration) -> Timespec {
|
||||
let secs = (self.t.tv_sec as i64).checked_add(other.as_secs() as i64);
|
||||
let mut secs = secs.expect("overflow when adding duration to time");
|
||||
let mut secs = other
|
||||
.as_secs()
|
||||
.try_into() // <- target type would be `i64`
|
||||
.ok()
|
||||
.and_then(|secs| self.t.tv_sec.checked_add(secs))
|
||||
.expect("overflow when adding duration to time");
|
||||
|
||||
// Nano calculations can't overflow because nanos are <1B which fit
|
||||
// in a u32.
|
||||
@@ -53,16 +58,19 @@ impl Timespec {
|
||||
}
|
||||
Timespec {
|
||||
t: syscall::TimeSpec {
|
||||
tv_sec: secs as i64,
|
||||
tv_sec: secs,
|
||||
tv_nsec: nsec as i32,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
fn sub_duration(&self, other: &Duration) -> Timespec {
|
||||
let secs = (self.t.tv_sec as i64).checked_sub(other.as_secs() as i64);
|
||||
let mut secs = secs.expect("overflow when subtracting duration \
|
||||
from time");
|
||||
let mut secs = other
|
||||
.as_secs()
|
||||
.try_into() // <- target type would be `i64`
|
||||
.ok()
|
||||
.and_then(|secs| self.t.tv_sec.checked_sub(secs))
|
||||
.expect("overflow when subtracting duration from time");
|
||||
|
||||
// Similar to above, nanos can't overflow.
|
||||
let mut nsec = self.t.tv_nsec as i32 - other.subsec_nanos() as i32;
|
||||
@@ -73,7 +81,7 @@ impl Timespec {
|
||||
}
|
||||
Timespec {
|
||||
t: syscall::TimeSpec {
|
||||
tv_sec: secs as i64,
|
||||
tv_sec: secs,
|
||||
tv_nsec: nsec as i32,
|
||||
},
|
||||
}
|
||||
|
||||
@@ -13,6 +13,7 @@ use libc;
|
||||
use time::Duration;
|
||||
|
||||
pub use self::inner::{Instant, SystemTime, UNIX_EPOCH};
|
||||
use convert::TryInto;
|
||||
|
||||
const NSEC_PER_SEC: u64 = 1_000_000_000;
|
||||
|
||||
@@ -41,8 +42,12 @@ impl Timespec {
|
||||
}
|
||||
|
||||
fn add_duration(&self, other: &Duration) -> Timespec {
|
||||
let secs = (self.t.tv_sec as i64).checked_add(other.as_secs() as i64);
|
||||
let mut secs = secs.expect("overflow when adding duration to time");
|
||||
let mut secs = other
|
||||
.as_secs()
|
||||
.try_into() // <- target type would be `libc::time_t`
|
||||
.ok()
|
||||
.and_then(|secs| self.t.tv_sec.checked_add(secs))
|
||||
.expect("overflow when adding duration to time");
|
||||
|
||||
// Nano calculations can't overflow because nanos are <1B which fit
|
||||
// in a u32.
|
||||
@@ -54,16 +59,19 @@ impl Timespec {
|
||||
}
|
||||
Timespec {
|
||||
t: libc::timespec {
|
||||
tv_sec: secs as libc::time_t,
|
||||
tv_sec: secs,
|
||||
tv_nsec: nsec as libc::c_long,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
fn sub_duration(&self, other: &Duration) -> Timespec {
|
||||
let secs = (self.t.tv_sec as i64).checked_sub(other.as_secs() as i64);
|
||||
let mut secs = secs.expect("overflow when subtracting duration \
|
||||
from time");
|
||||
let mut secs = other
|
||||
.as_secs()
|
||||
.try_into() // <- target type would be `libc::time_t`
|
||||
.ok()
|
||||
.and_then(|secs| self.t.tv_sec.checked_sub(secs))
|
||||
.expect("overflow when subtracting duration from time");
|
||||
|
||||
// Similar to above, nanos can't overflow.
|
||||
let mut nsec = self.t.tv_nsec as i32 - other.subsec_nanos() as i32;
|
||||
@@ -74,7 +82,7 @@ impl Timespec {
|
||||
}
|
||||
Timespec {
|
||||
t: libc::timespec {
|
||||
tv_sec: secs as libc::time_t,
|
||||
tv_sec: secs,
|
||||
tv_nsec: nsec as libc::c_long,
|
||||
},
|
||||
}
|
||||
|
||||
@@ -16,6 +16,7 @@ use sys::c;
|
||||
use sys::cvt;
|
||||
use sys_common::mul_div_u64;
|
||||
use time::Duration;
|
||||
use convert::TryInto;
|
||||
|
||||
const NANOS_PER_SEC: u64 = 1_000_000_000;
|
||||
const INTERVALS_PER_SEC: u64 = NANOS_PER_SEC / 100;
|
||||
@@ -173,9 +174,11 @@ impl From<c::FILETIME> for SystemTime {
|
||||
}
|
||||
|
||||
fn dur2intervals(d: &Duration) -> i64 {
|
||||
d.as_secs().checked_mul(INTERVALS_PER_SEC).and_then(|i| {
|
||||
i.checked_add(d.subsec_nanos() as u64 / 100)
|
||||
}).expect("overflow when converting duration to intervals") as i64
|
||||
d.as_secs()
|
||||
.checked_mul(INTERVALS_PER_SEC)
|
||||
.and_then(|i| i.checked_add(d.subsec_nanos() as u64 / 100))
|
||||
.and_then(|i| i.try_into().ok())
|
||||
.expect("overflow when converting duration to intervals")
|
||||
}
|
||||
|
||||
fn intervals2dur(intervals: u64) -> Duration {
|
||||
|
||||
Reference in New Issue
Block a user