use sodiumoxide for mlock

radumarias · Jun 23, 2024 · 4660b40 · 4660b40
1 parent ba3fab1
commit 4660b40
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -25,10 +25,16 @@ fn zeroize1(arr: &Bound<'_, PyAny>, py: Python<'_>) -> PyResult<()> {
 #[pyfunction]
 fn mlock(arr: &Bound<'_, PyAny>, py: Python<'_>) -> PyResult<()> {
     let arr = as_array_mut(arr, py)?;
-    // if !_mlock(arr.as_mut_ptr(), arr.len()) {
-    sodiumoxide::utils::mlock(arr).map_err(|_|PyErr::new::<pyo3::exceptions::PyTypeError, _>(
-        "mlock failed",
-    ))?;
+    unsafe {
+        if !_mlock(arr.as_mut_ptr(), arr.len()) {
+            return Err(PyErr::new::<pyo3::exceptions::PyTypeError, _>(
+                "mlock failed",
+            ));
+        }
+    }
+    // sodiumoxide::utils::mlock(arr).map_err(|_|PyErr::new::<pyo3::exceptions::PyTypeError, _>(
+    //     "mlock failed",
+    // ))?;
     Ok(())
 }