feat: bind GCM intial bindings

AshGw · May 31, 2024 · 375247a · 375247a
1 parent 597db6c
commit 375247a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/python/gcm_rs/__init__.py b/python/gcm_rs/__init__.py
@@ -1,3 +1,4 @@
 from gcm_rs import *
 
-print(gen_key())
+print(gen_key())
+a = GCM()
diff --git a/rust/lib.rs b/rust/lib.rs
@@ -5,6 +5,7 @@ pub mod gcm;
 pub mod random;
 pub mod types;
 
+use gcm::Aes256Gcm;
 use random::{gen_key as gk, gen_nonce as gn};
 
 use pyo3::prelude::*;
@@ -19,10 +20,14 @@ fn gen_key() -> PyResult<Vec<u8>> {
     Ok(gk())
 }
 
+#[pyclass]
+struct Gcm(Aes256Gcm);
+
 #[pymodule]
 #[pyo3(name = "_lib_name")]
 fn pyrust(_py: Python, m: &PyModule) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(gen_nonce, m)?)?;
     m.add_function(wrap_pyfunction!(gen_key, m)?)?;
+    m.add_class::<Gcm>()?;
     Ok(())
 }