chris/ruwuma
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

serde: Improve error handling in time modules

Browse Source
This commit is contained in:
Jonas Platte 2021-04-05 17:27:13 +02:00
parent 2f1b9f0979
commit 2805dd733b
No known key found for this signature in database
GPG Key ID: CC154DE0E30B7C67
4 changed files with 36 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
ruma-serde/src/time/ms_since_unix_epoch.rs
View File

@ -2,14 +2,14 @@
//! since the UNIX epoch. Delegates to `js_int::UInt` to ensure integer size is within bounds.
use std::{
convert::TryFrom,
convert::TryInto,
time::{Duration, SystemTime, UNIX_EPOCH},
};
use js_int::UInt;
use serde::{
de::{Deserialize, Deserializer},
ser::{Error, Serialize, Serializer},
de::{self, Deserialize, Deserializer},
ser::{self, Serialize, Serializer},
};
/// Serialize a SystemTime.
@ -20,12 +20,10 @@ pub fn serialize<S>(time: &SystemTime, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
// If this unwrap fails, the system this is executed is completely broken.
let time_since_epoch = time.duration_since(UNIX_EPOCH).unwrap();
match UInt::try_from(time_since_epoch.as_millis()) {
Ok(uint) => uint.serialize(serializer),
Err(err) => Err(S::Error::custom(err)),
}
let time_since_epoch = time.duration_since(UNIX_EPOCH).map_err(ser::Error::custom)?;
let uint: UInt = time_since_epoch.as_millis().try_into().map_err(ser::Error::custom)?;
uint.serialize(serializer)
}
/// Deserializes a SystemTime.
@ -37,7 +35,9 @@ where
D: Deserializer<'de>,
{
let millis = UInt::deserialize(deserializer)?;
Ok(UNIX_EPOCH + Duration::from_millis(millis.into()))
UNIX_EPOCH
.checked_add(Duration::from_millis(millis.into()))
.ok_or_else(|| de::Error::custom("input too large for SystemTime"))
}
#[cfg(test)]

11
ruma-serde/src/time/opt_ms_since_unix_epoch.rs
View File

@ -6,7 +6,7 @@ use std::time::{Duration, SystemTime, UNIX_EPOCH};
use js_int::UInt;
use serde::{
de::{Deserialize, Deserializer},
de::{self, Deserialize, Deserializer},
ser::{Serialize, Serializer},
};
@ -32,8 +32,13 @@ pub fn deserialize<'de, D>(deserializer: D) -> Result<Option<SystemTime>, D::Err
where
D: Deserializer<'de>,
{
Ok(Option::<UInt>::deserialize(deserializer)?
.map(|millis| UNIX_EPOCH + Duration::from_millis(millis.into())))
Option::<UInt>::deserialize(deserializer)?
.map(|millis| {
UNIX_EPOCH
.checked_add(Duration::from_millis(millis.into()))
.ok_or_else(|| de::Error::custom("input too large for SystemTime"))
})
.transpose()
}
#[cfg(test)]

11
ruma-serde/src/time/opt_s_since_unix_epoch.rs
View File

@ -6,7 +6,7 @@ use std::time::{Duration, SystemTime, UNIX_EPOCH};
use js_int::UInt;
use serde::{
de::{Deserialize, Deserializer},
de::{self, Deserialize, Deserializer},
ser::{Serialize, Serializer},
};
@ -32,8 +32,13 @@ pub fn deserialize<'de, D>(deserializer: D) -> Result<Option<SystemTime>, D::Err
where
D: Deserializer<'de>,
{
Ok(Option::<UInt>::deserialize(deserializer)?
.map(|secs| UNIX_EPOCH + Duration::from_secs(secs.into())))
Option::<UInt>::deserialize(deserializer)?
.map(|secs| {
UNIX_EPOCH
.checked_add(Duration::from_secs(secs.into()))
.ok_or_else(|| de::Error::custom("input too large for SystemTime"))
})
.transpose()
}
#[cfg(test)]

20
ruma-serde/src/time/s_since_unix_epoch.rs
View File

@ -2,14 +2,14 @@
//! the UNIX epoch. Delegates to `js_int::UInt` to ensure integer size is within bounds.
use std::{
convert::TryFrom,
convert::TryInto,
time::{Duration, SystemTime, UNIX_EPOCH},
};
use js_int::UInt;
use serde::{
de::{Deserialize, Deserializer},
ser::{Error, Serialize, Serializer},
de::{self, Deserialize, Deserializer},
ser::{self, Serialize, Serializer},
};
/// Serialize a SystemTime.
@ -20,12 +20,10 @@ pub fn serialize<S>(time: &SystemTime, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
// If this unwrap fails, the system this is executed is completely broken.
let time_since_epoch = time.duration_since(UNIX_EPOCH).unwrap();
match UInt::try_from(time_since_epoch.as_secs()) {
Ok(uint) => uint.serialize(serializer),
Err(err) => Err(S::Error::custom(err)),
}
let time_since_epoch = time.duration_since(UNIX_EPOCH).map_err(ser::Error::custom)?;
let uint: UInt = time_since_epoch.as_secs().try_into().map_err(ser::Error::custom)?;
uint.serialize(serializer)
}
/// Deserializes a SystemTime.
@ -37,7 +35,9 @@ where
D: Deserializer<'de>,
{
let secs = UInt::deserialize(deserializer)?;
Ok(UNIX_EPOCH + Duration::from_secs(secs.into()))
UNIX_EPOCH
.checked_add(Duration::from_secs(secs.into()))
.ok_or_else(|| de::Error::custom("input too large for SystemTime"))
}
#[cfg(test)]