wolfSSL/wolfssl
1
0
Fork 1
mirror of https://github.com/wolfSSL/wolfssl.git synced 2026-07-05 15:30:49 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add comments explaining ct.len() and ss.len() checks

Browse Source
This commit is contained in:
Josh Holtrop
2026-02-26 15:09:12 -05:00
parent 67c500ec87
commit 4235fa59b6
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wrapper/rust/wolfssl-wolfcrypt/src/mlkem.rs
+7
View File
@@ -470,10 +470,13 @@ impl MlKem {
ss: &mut [u8],
rng: &mut RNG,
) -> Result<(), i32> {
// Verify the cipher text length is as expected based on the parameter
// set (key type) in use.
let expected_ct_size = self.cipher_text_size()?;
if ct.len() != expected_ct_size {
return Err(sys::wolfCrypt_ErrorCodes_BUFFER_E);
}
// Verify the shared secret length is as expected.
if ss.len() != Self::SHARED_SECRET_SIZE {
return Err(sys::wolfCrypt_ErrorCodes_BUFFER_E);
}
@@ -535,10 +538,13 @@ impl MlKem {
if rand.len() != Self::ENC_RAND_SIZE {
return Err(sys::wolfCrypt_ErrorCodes_BUFFER_E);
}
// Verify the cipher text length is as expected based on the parameter
// set (key type) in use.
let expected_ct_size = self.cipher_text_size()?;
if ct.len() != expected_ct_size {
return Err(sys::wolfCrypt_ErrorCodes_BUFFER_E);
}
// Verify the shared secret length is as expected.
if ss.len() != Self::SHARED_SECRET_SIZE {
return Err(sys::wolfCrypt_ErrorCodes_BUFFER_E);
}
@@ -598,6 +604,7 @@ impl MlKem {
/// }
/// ```
pub fn decapsulate(&mut self, ss: &mut [u8], ct: &[u8]) -> Result<(), i32> {
// Verify the shared secret length is as expected.
if ss.len() != Self::SHARED_SECRET_SIZE {
return Err(sys::wolfCrypt_ErrorCodes_BUFFER_E);
}