From 2a3eca3be1d10f12f696ee421632ef32ed621d42 Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sun, 3 Sep 2023 23:21:10 +0800
Subject: [PATCH] feat: update bench.rs

---
 examples/{test.rs => bench.rs} | 17 -----------------
 src/decryptor.rs               |  0
 src/lib.rs                     |  9 ++++++---
 3 files changed, 6 insertions(+), 20 deletions(-)
 rename examples/{test.rs => bench.rs} (59%)
 create mode 100644 src/decryptor.rs

diff --git a/examples/test.rs b/examples/bench.rs
similarity index 59%
rename from examples/test.rs
rename to examples/bench.rs
index e914e0a..21d65dd 100644
--- a/examples/test.rs
+++ b/examples/bench.rs
@@ -3,23 +3,6 @@ use benchmark_simple::{Bench, Options};
 use sm4_gcm::Sm4GcmStreamEncryptor;
 
 fn main() {
-    let key = [0u8; 16];
-    let nonce = [0u8; 12];
-    let mut e = Sm4GcmStreamEncryptor::new(key, &nonce);
-
-    println!("{}", hex::encode(&key));
-    println!("{}", hex::encode(&nonce));
-
-    let a = e.update(b"hello world");
-    let (b, t) = e.finalize();
-
-    let mut enc = a.clone();
-    enc.extend_from_slice(&b);
-    enc.extend_from_slice(&t);
-
-    println!("{}", hex::encode(&enc));
-
-    // ----------------------------------------------------------------------
     let bench = Bench::new();
     let mut m = vec![0xd0u8; 16384];
 
diff --git a/src/decryptor.rs b/src/decryptor.rs
new file mode 100644
index 0000000..e69de29
diff --git a/src/lib.rs b/src/lib.rs
index 1ff6401..5fe669e 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,8 +1,11 @@
+pub use encryptor::Sm4GcmStreamEncryptor;
+
+pub use crate::encryptor::sm4_gcm_aad_encrypt;
+pub use crate::encryptor::sm4_gcm_encrypt;
+
 mod util;
 mod encryptor;
-
-pub use encryptor::Sm4GcmStreamEncryptor;
-use crate::encryptor::sm4_gcm_aad_encrypt;
+mod decryptor;
 
 // Test vectors are all from BC
 #[test]