jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use checked math for read_body (#7117)

Browse Source
This commit is contained in:
Dan Cline
2024-03-12 12:01:36 -04:00
committed by GitHub
parent 074747a419
commit f75bebd8d5
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
crates/net/ecies/src/algorithm.rs
View File

@ -691,7 +691,11 @@ impl ECIES {
}
pub fn read_body<'a>(&mut self, data: &'a mut [u8]) -> Result<&'a mut [u8], ECIESError> {
let (body, mac_bytes) = split_at_mut(data, data.len() - 16)?;
// error if the data is too small to contain the tag
// TODO: create a custom type similar to EncryptedMessage for parsing, checking MACs, and
// decrypting the body
let mac_index = data.len().checked_sub(16).ok_or(ECIESErrorImpl::EncryptedDataTooSmall)?;
let (body, mac_bytes) = split_at_mut(data, mac_index)?;
let mac = B128::from_slice(mac_bytes);
self.ingress_mac.as_mut().unwrap().update_body(body);
let check_mac = self.ingress_mac.as_mut().unwrap().digest();