initial _ctypes implementation with _CData, get_errno, and set_errno

arihant2math · arihant2math · commit dd5e46003369 · 2025-02-12T21:24:52.000-08:00
Signed-off-by: Ashwin Naren &lt;arihant2math@gmail.com&gt;
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/vm/Cargo.toml b/vm/Cargo.toml
@@ -99,6 +99,7 @@ uname = "0.1.1"
 [target.'cfg(not(target_arch = "wasm32"))'.dependencies]
 rustyline = { workspace = true }
 which = "6"
+errno = "0.3"
 
 [target.'cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))'.dependencies]
 num_cpus = "1.13.1"
diff --git a/vm/src/stdlib/ctypes.rs b/vm/src/stdlib/ctypes.rs
@@ -0,0 +1,32 @@
+pub(crate) use _ctypes::make_module;
+
+#[pymodule]
+mod _ctypes {
+    use crate::{common::lock::PyRwLock, PyObjectRef};
+    use crossbeam_utils::atomic::AtomicCell;
+
+    pub struct RawBuffer {
+        pub inner: Box<[u8]>,
+        pub size: usize,
+    }
+
+    #[pyattr]
+    #[pyclass(name = "_CData")]
+    pub struct PyCData {
+        _objects: AtomicCell<Vec<PyObjectRef>>,
+        _buffer: PyRwLock<RawBuffer>,
+    }
+
+    #[pyclass]
+    impl PyCData {}
+
+    #[pyfunction]
+    fn get_errno() -> i32 {
+        errno::errno().0
+    }
+
+    #[pyfunction]
+    fn set_errno(value: i32) {
+        errno::set_errno(errno::Errno(value));
+    }
+}
diff --git a/vm/src/stdlib/mod.rs b/vm/src/stdlib/mod.rs
@@ -25,7 +25,7 @@ pub mod typing;
 pub mod warnings;
 mod weakref;
 
-#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
+#[cfg(not(target_arch = "wasm32"))]
 #[macro_use]
 pub mod os;
 #[cfg(windows)]
@@ -37,6 +37,8 @@ pub mod posix;
 #[path = "posix_compat.rs"]
 pub mod posix;
 
+#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
+mod ctypes;
 #[cfg(windows)]
 pub(crate) mod msvcrt;
 #[cfg(all(unix, not(any(target_os = "android", target_os = "redox"))))]
@@ -124,5 +126,9 @@ pub fn get_module_inits() -> StdlibMap {
             "_winapi" => winapi::make_module,
             "winreg" => winreg::make_module,
         }
+        #[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
+        {
+            "_ctypes" => ctypes::make_module,
+        }
     }
 }
