source/shadowsocks-rust
1
0
Fork 0
mirror of https://github.com/shadowsocks/shadowsocks-rust.git synced 2026-02-09 01:59:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

build with nightly

Browse Source
This commit is contained in:
Y. T. Chung
2015-07-09 16:40:44 +08:00
parent c1f0291728
commit 7b4d662fa3
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/crypto/sodium.rs
View File

@@ -62,15 +62,19 @@ impl SodiumCipher {
impl Cipher for SodiumCipher {
fn update(&mut self, data: &[u8], out: &mut Vec<u8>) -> CipherResult<()> {
let padding_len = self.counter % BLOCK_SIZE;
let mut pad;
let padded_data =
let pad =
if padding_len == 0 {
data
None
} else {
pad = Some(repeat(0u8).take(padding_len).chain(data.iter().map(|&x| x)).collect::<Vec<u8>>());
&pad.as_ref().unwrap()[..]
let pad = repeat(0u8).take(padding_len).chain(data.iter().map(|&x| x)).collect::<Vec<u8>>();
Some(pad)
};
let padded_data = match &pad {
&Some(ref p) => &p[..],
&None => data,
};
if self.buf.len() < padding_len + data.len() {
self.buf.resize(padding_len + data.len(), 0u8);
}