Refactoring ctypes module to a PyPy-like file tree

RustPython · rodrigocam · Aug 7, 2021 · Oct 29, 2020 · Oct 31, 2020 · Oct 31, 2020
commit b9ada0a0e16a0f21e99ad24adc18680fa5d7e94e
diff --git a/vm/src/stdlib/ctypes.rs b/vm/src/stdlib/ctypes.rs
diff --git a/vm/src/stdlib/ctypes/array.rs b/vm/src/stdlib/ctypes/array.rs
diff --git a/vm/src/stdlib/ctypes/basics.rs b/vm/src/stdlib/ctypes/basics.rs
diff --git a/vm/src/stdlib/ctypes/common.rs b/vm/src/stdlib/ctypes/common.rs
diff --git a/vm/src/stdlib/ctypes/ctypes.rs b/vm/src/stdlib/ctypes/ctypes.rs
diff --git a/vm/src/stdlib/ctypes/dll.rs b/vm/src/stdlib/ctypes/dll.rs
@@ -0,0 +1,36 @@
+extern crate libloading;
+use crate::builtins::pystr::PyStrRef;
+use crate::builtins::pytype::PyTypeRef;
+use crate::pyobject::{PyObjectRef, PyResult, PyValue};
+use crate::VirtualMachine;
+
+// #[pyclass()]
+#[derive(Debug)]
+struct SharedLibrary {
+    lib: libloading::Library,
+}
+
+impl PyValue for SharedLibrary {
+    fn class(vm: &VirtualMachine) -> &PyTypeRef {
+        &vm.ctx.types.object_type
+    }
+}
+
+pub fn dlopen(lib_path: PyStrRef, vm: &VirtualMachine) -> PyResult {
+    let shared_lib = SharedLibrary {
+        lib: libloading::Library::new(lib_path.as_ref()).expect("Failed to load library"),
+    };
+    Ok(vm.new_pyobj(shared_lib))
+}
+
+pub fn dlsym(handle: PyObjectRef, func_name: PyStrRef) {
+    if let Some(slib) = handle.payload::<SharedLibrary>() {
+        unsafe {
+            let func: libloading::Symbol<extern "C" fn()> = slib
+                .lib
+                .get(func_name.as_ref().as_bytes())
+                .expect("Failed to get func");
+            func();
+        }
+    }
+}
diff --git a/vm/src/stdlib/ctypes/function.rs b/vm/src/stdlib/ctypes/function.rs
diff --git a/vm/src/stdlib/ctypes/mod.rs b/vm/src/stdlib/ctypes/mod.rs
@@ -0,0 +1,15 @@
+use crate::pyobject::PyObjectRef;
+use crate::VirtualMachine;
+
+mod dll;
+
+use crate::stdlib::ctypes::dll::*;
+
+pub(crate) fn make_module(vm: &VirtualMachine) -> PyObjectRef {
+    let ctx = &vm.ctx;
+
+    py_module!(vm, "_ctypes", {
+        "dlopen" => ctx.new_function(dlopen),
+        "dlsym" => ctx.new_function(dlsym),
+    })
+}
diff --git a/vm/src/stdlib/ctypes/pointer.rs b/vm/src/stdlib/ctypes/pointer.rs
diff --git a/vm/src/stdlib/ctypes/primitive.rs b/vm/src/stdlib/ctypes/primitive.rs
diff --git a/vm/src/stdlib/ctypes/structure.rs b/vm/src/stdlib/ctypes/structure.rs
diff --git a/vm/src/stdlib/ctypes/union.rs b/vm/src/stdlib/ctypes/union.rs