Add factorize_pq_pair function

Closes #2.
cher-nov · Sep 26, 2021 · c274f4a · c274f4a
1 parent b105ae7
commit c274f4a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -40,9 +40,17 @@ fn decrypt_ige(cipher: &[u8], key: &[u8], iv: &[u8]) -> PyResult<Py<PyBytes>> {
     Python::with_gil(|py| Ok(PyBytes::new(py, &plain).into()))
 }
 
+/// Factorizes the pair of primes ``pq`` into ``(p, q)``.
+#[pyfunction]
+#[pyo3(text_signature = "(pq)")]
+fn factorize_pq_pair(pq: u64) -> (u64, u64) {
+    grammers_crypto::factorize::factorize(pq)
+}
+
 #[pymodule]
 fn cryptg(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
     m.add_wrapped(wrap_pyfunction!(encrypt_ige))?;
     m.add_wrapped(wrap_pyfunction!(decrypt_ige))?;
+    m.add_wrapped(wrap_pyfunction!(factorize_pq_pair))?;
     Ok(())
 }