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

feat: update enc/dec

Browse Source
This commit is contained in:
Hatter Jiang
2023-08-26 23:51:49 +08:00
parent 877609ecbb
commit 977f0ac5d3
3 changed files with 20 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/decryptor.rs
View File

@@ -17,11 +17,11 @@ pub struct Aes128GcmStreamDecryptor {
}
impl Aes128GcmStreamDecryptor {
pub fn new(key: [u8; 16]) -> Self {
pub fn new(key: [u8; 16], nonce: &[u8]) -> Self {
let key = GenericArray::from(key);
let aes = Aes128::new(&key);
Self {
let mut s = Self {
crypto: aes,
message_buffer: vec![],
integrality_buffer: vec![],
@@ -31,14 +31,12 @@ impl Aes128GcmStreamDecryptor {
encryption_nonce: 0,
adata_len: 0,
message_len: 0,
}
}
pub fn init_nonce(&mut self, nonce: &[u8]) {
let (ghash_key, normalized_nonce) = self.normalize_nonce(nonce);
self.ghash_key = ghash_key;
self.init_nonce = normalized_nonce;
self.encryption_nonce = normalized_nonce;
};
let (ghash_key, normalized_nonce) = s.normalize_nonce(nonce);
s.ghash_key = ghash_key;
s.init_nonce = normalized_nonce;
s.encryption_nonce = normalized_nonce;
s
}
pub fn init_adata(&mut self, adata: &[u8]) {
@@ -78,7 +76,7 @@ impl Aes128GcmStreamDecryptor {
}
pub fn finalize(&mut self) -> Result<Vec<u8>, String> {
let mut plaintext_message = vec![];
let mut plaintext_message = Vec::with_capacity(16);
let message_buffer_len = self.message_buffer.len();
if message_buffer_len > 16 {
// last block and this block len is less than 128 bits