Skip to content

Commit

Permalink
Improve size usage by removing use of expect
Browse files Browse the repository at this point in the history
  • Loading branch information
sosthene-nitrokey committed Oct 9, 2024
1 parent c31b9ca commit cda9178
Showing 1 changed file with 20 additions and 5 deletions.
25 changes: 20 additions & 5 deletions src/hpke.rs
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,10 @@ fn extract_and_expand(dh: x25519::SharedSecret, kem_context: &[u8]) -> [u8; 32]
kem_context,
&mut shr,
)
.expect("Length of shr is known to be OK");
.map_err(|_err| {
error!("Length of shr is known to be OK: {_err:?}");
})
.unwrap();
shr
}

Expand Down Expand Up @@ -170,7 +173,10 @@ fn key_schedule<T: Aead>(_role: Role, shared_secret: [u8; 32], info: &[u8]) -> C
&key_schedule_context,
&mut key,
)
.expect("KEY is not too large");
.map_err(|_err| {
error!("KEY is not too large: {_err:?}");
})
.unwrap();
let mut base_nonce = [0; NN];
labeled_expand(
T::X25519_HKDF_SHA256_SELF_HPKE_SUITE_ID,
Expand All @@ -179,7 +185,10 @@ fn key_schedule<T: Aead>(_role: Role, shared_secret: [u8; 32], info: &[u8]) -> C
&key_schedule_context,
&mut base_nonce,
)
.expect("NONCE is not too large");
.map_err(|_err| {
error!("NONCE is not too large: {_err:?}");
})
.unwrap();
let mut exporter_secret = [0; NH];
labeled_expand(
T::X25519_HKDF_SHA256_SELF_HPKE_SUITE_ID,
Expand All @@ -188,7 +197,10 @@ fn key_schedule<T: Aead>(_role: Role, shared_secret: [u8; 32], info: &[u8]) -> C
&key_schedule_context,
&mut exporter_secret,
)
.expect("EXP is not too large");
.map_err(|_err| {
error!("EXP is not too large: {_err:?}");
})
.unwrap();
Context {
key,
base_nonce,
Expand Down Expand Up @@ -224,7 +236,10 @@ impl Context {
let mut aead = T::new((&self.key).into());
let tag = aead
.encrypt_in_place_detached(nonce, aad, plaintext)
.expect("Not used to encrypt data too large");
.map_err(|_err| {
error!("Not used to encrypt data too large: {_err:?}");
})
.unwrap();

tag.into()
}
Expand Down

0 comments on commit cda9178

Please sign in to comment.