shadowsocks-crypto 0.8.0

Shadowsocks Crypto
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use chacha20::{
    cipher::{IvSizeUser, Iv, KeyIvInit, KeySizeUser, StreamCipher, typenum::Unsigned},
    ChaCha20,
    Key,
};

/// ChaCha20 for IETF Protocols
///
/// https://tools.ietf.org/html/rfc8439
pub struct Chacha20 {
    cipher: ChaCha20,
}

impl Chacha20 {
    pub fn new(key: &[u8], nonce: &[u8]) -> Self {
        let key = Key::try_from(key).expect("chacha20 key");
        let nonce = Iv::<ChaCha20>::try_from(nonce).expect("chacha20 nonce");
        let cipher = ChaCha20::new(&key, &nonce);

        Self { cipher }
    }

    pub fn encryptor_update(&mut self, plaintext_in_ciphertext_out: &mut [u8]) {
        self.cipher.apply_keystream(plaintext_in_ciphertext_out);
    }

    pub fn decryptor_update(&mut self, ciphertext_in_plaintext_out: &mut [u8]) {
        self.cipher.apply_keystream(ciphertext_in_plaintext_out);
    }

    pub fn key_size() -> usize {
        <ChaCha20 as KeySizeUser>::KeySize::to_usize()
    }

    pub fn nonce_size() -> usize {
        <ChaCha20 as IvSizeUser>::IvSize::to_usize()
    }
}