natrium 0.2.0

Cryptography lib
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
39
40
41

use byteorder::{ReadBytesExt, WriteBytesExt, LE};
use std::convert::AsRef;
use std::io::Cursor;
use Result;

/// ChaCha key.
#[derive(Debug, Clone)]
pub struct Key {
  key: Vec<u32>,
}

impl Key {
  /// Create a new instance from a byte slice.
  pub fn from_bytes(bytes: impl AsRef<Vec<u8>>) -> Result<Self> {
    let bytes = bytes.as_ref();
    let max = bytes.len() / 4;

    let mut key = Vec::with_capacity(max);
    let mut reader = Cursor::new(bytes);
    for _ in 0..max {
      let byte = reader.read_u32::<LE>().expect("Error writing key");
      key.push(byte);
    }
    Ok(Self { key })
  }

  /// Convert into a byte slice.
  pub fn as_bytes(&self) -> Vec<u8> {
    let mut bytes = Vec::with_capacity(self.key.len() * 4);
    for num in &self.key {
      bytes.write_u32::<LE>(*num).expect("Error converting key");
    }
    bytes
  }

  /// Get the key.
  #[inline]
  pub fn key(&self) -> &[u32] {
    &self.key
  }
}