anon
/
rtp
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor AES decription code

dtls-srtp
Takeru Ohta 2017-03-27 10:45:20 -07:00
parent e766d719b7
commit 88b02115fc
1 changed files with 6 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/rfc3711.rs
View File

@ -105,13 +105,9 @@ impl SrtpContext {
track_try!(header.write_to(&mut decrypted)); track_try!(header.write_to(&mut decrypted));
for block in encrypted_portion.chunks(block_size) { for block in encrypted_portion.chunks(block_size) {
let input = [0; 16]; let old_len = decrypted.len();
let mut output = [0; 16]; decrypted.resize(old_len + block.len(), 0);
ctr.process(&input[..], &mut output[..]); ctr.process(block, &mut decrypted[old_len..]);
for (a, b) in block.iter().zip(output.iter()) {
decrypted.push(*a ^ *b);
}
} }
Ok(decrypted) Ok(decrypted)
@ -197,15 +193,10 @@ impl SrtcpContext {
let block_size = self.session_encr_key.len(); let block_size = self.session_encr_key.len();
let mut decrypted = Vec::from(&packet[..8]); let mut decrypted = Vec::from(&packet[..8]);
for block in encrypted_portion.chunks(block_size) { for block in encrypted_portion.chunks(block_size) {
let input = [0; 16]; let old_len = decrypted.len();
let mut output = [0; 16]; decrypted.resize(old_len + block.len(), 0);
ctr.process(&input[..], &mut output[..]); ctr.process(block, &mut decrypted[old_len..]);
for (a, b) in block.iter().zip(output.iter()) {
decrypted.push(*a ^ *b);
}
} }
Ok(decrypted) Ok(decrypted)