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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

use aes::*;
use block_modes::{BlockMode, Cbc, block_padding::{Pkcs7}};
use anyhow::*;
use wasm_bindgen::{prelude::*, throw_str};


#[wasm_bindgen]
pub struct AES {

}

impl AES {
  pub fn encrypt_128_impl(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>> {
    let cipher = Cbc::<Aes128, Pkcs7>::new_from_slices(key, iv)?;

    Ok(cipher.encrypt_vec(message))
  } 

  pub fn decrypt_128_impl(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>> {
    let cipher = Cbc::<Aes128, Pkcs7>::new_from_slices(key, iv)?;

    Ok(cipher.decrypt_vec(&message)?)
  }

  pub fn encrypt_256_impl(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>> {
    let cipher = Cbc::<Aes256, Pkcs7>::new_from_slices(key, iv)?;

    Ok(cipher.encrypt_vec(&message))
  }

  pub fn decrypt_256_impl(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>> {
    let cipher = Cbc::<Aes256, Pkcs7>::new_from_slices(key, iv)?;

    Ok(cipher.decrypt_vec(&message)?)
  }
}

#[cfg(target_arch = "wasm32")]
#[wasm_bindgen]
impl AES { 
  #[wasm_bindgen(js_name = encrypt128)]
  pub fn encrypt_128(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>, JsValue> {
    match AES::encrypt_128_impl(key, iv, message) {
      Ok(v) => Ok(v),
      Err(e) => throw_str(&e.to_string()),
    }
  }

  #[wasm_bindgen(js_name = decrypt128)]
  pub fn decrypt_128(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>, JsValue> {
    match AES::decrypt_128_impl(key, iv, message) {
      Ok(v) => Ok(v),
      Err(e) => throw_str(&e.to_string()),
    }
  }

  #[wasm_bindgen(js_name = encrypt256)]
  pub fn encrypt_256(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>, JsValue> {
    match AES::encrypt_256_impl(key, iv, message) {
      Ok(v) => Ok(v),
      Err(e) => throw_str(&e.to_string()),
    }
  }

  #[wasm_bindgen(js_name = decrypt256)]
  pub fn decrypt_256(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>, JsValue> {
    match AES::decrypt_256_impl(key, iv, message) {
      Ok(v) => Ok(v),
      Err(e) => throw_str(&e.to_string()),
    }
  }
}

#[cfg(not(target_arch = "wasm32"))]
impl AES { 
  pub fn encrypt_128(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>> {
    AES::encrypt_128_impl(key, iv, message)
  }

  pub fn decrypt_128(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>> {
    AES::decrypt_128_impl(key, iv, message)
  }
  pub fn encrypt_256(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>> {
    AES::encrypt_256_impl(key, iv, message)
  }

  pub fn decrypt_256(key: &[u8], iv: &[u8], message: &[u8]) -> Result<Vec<u8>> {
    AES::decrypt_256_impl(key, iv, message)
  }
}