hatter/aes-gcm-stream
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: v0.2.0-alpha-1, performance improved from 36MB/s to ~450MB/s

Browse Source
This commit is contained in:
Hatter Jiang
2023-08-31 00:20:23 +08:00
parent f4b5f68cd9
commit adb0383133
6 changed files with 180 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/encryptor.rs
View File

@@ -66,11 +66,21 @@ impl $module {
}
let blocks_count = message_buffer_len / 16;
let mut encrypted_message = Vec::with_capacity(blocks_count * 16);
for i in 0..blocks_count {
let mut blocks = Vec::with_capacity(blocks_count);
for _ in 0..blocks_count {
self.encryption_nonce = inc_32(self.encryption_nonce);
let mut ctr = self.encryption_nonce.to_be_bytes();
let block = Block::<$aesn>::from_mut_slice(&mut ctr);
self.crypto.encrypt_block(block);
let ctr = self.encryption_nonce.to_be_bytes();
let block = Block::<$aesn>::clone_from_slice(&ctr);
blocks.push(block);
}
// println!("block site: {}", blocks.len());
self.crypto.encrypt_blocks(&mut blocks);
for i in 0..blocks_count {
// self.encryption_nonce = inc_32(self.encryption_nonce);
// let mut ctr = self.encryption_nonce.to_be_bytes();
// let block = Block::<$aesn>::from_mut_slice(&mut ctr);
// self.crypto.encrypt_block(block);
let block = &blocks[i];
let chunk = &message_buffer_slice[i * 16..(i + 1) * 16];
let y = u8to128(chunk) ^ u8to128(&block.as_slice());
encrypted_message.extend_from_slice(&y.to_be_bytes());
@@ -125,6 +135,7 @@ impl $module {
}
fn update_integrality_buffer(&mut self) {
// self.integrality_buffer.clear();
let integrality_buffer_slice = self.integrality_buffer.as_slice();
let integrality_buffer_slice_len = integrality_buffer_slice.len();
if integrality_buffer_slice_len >= 16 {