_N0x/0xid_crypt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Cleaned up message padding function (thx @roh)

Browse Source
This commit is contained in:
_N0x 2024-10-31 22:35:05 +01:00
parent b644455842
commit a01fd584fc
1 changed files with 12 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/sha/mod.rs
View File

@ -29,32 +29,28 @@ fn ssig1(x: u32) -> u32 {
return ret; return ret;
} }
fn pad_message(message: String) -> Vec<u32> { fn pad_message(msg: String) -> Vec<u32> {
let mut msg_bytes = message.as_bytes().to_vec(); let l = (msg.len() as u64) << 3;
let l = (msg_bytes.len() as u64) * 8; let k = ((448u64.wrapping_sub((l + 8) % 512) % 512) >> 3) as usize;
msg_bytes.push(0x80); let mut result_u8 = Vec::with_capacity(msg.len() + 1 + k + std::mem::size_of::<u64>());
let k = (448u64.wrapping_sub((l + 8) % 512)) % 512; result_u8.extend(msg.as_bytes());
let k_bytes = k / 8; result_u8.push(0x80);
result_u8.resize(result_u8.len() + k, 0u8);
result_u8.extend(&l.to_be_bytes());
let padding = vec![0u8; k_bytes as usize]; let mut result_u32 = Vec::with_capacity(result_u8.len() / 4); // Allocate the right size for u32 vector
msg_bytes.extend(padding); for chunk in result_u8.chunks(4) {
msg_bytes.extend(&l.to_be_bytes());
let mut u32_vec = Vec::with_capacity(msg_bytes.len() / 4); // Allocate the right size for u32 vector
for chunk in msg_bytes.chunks(4) {
let value = (chunk[0] as u32) << 24 let value = (chunk[0] as u32) << 24
| (chunk[1] as u32) << 16 | (chunk[1] as u32) << 16
| (chunk[2] as u32) << 8 | (chunk[2] as u32) << 8
| (chunk[3] as u32); | (chunk[3] as u32);
u32_vec.push(value); result_u32.push(value);
} }
return u32_vec; return result_u32;
} }
//////////// END HELPER FUNCTIONS ///////////// //////////// END HELPER FUNCTIONS /////////////