initial _ctypes implementation with _CData, get_errno, and set_errno

arihant2math · arihant2math · commit efc5a4bccb89 · 2025-02-12T20:48:46.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,35 @@
+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
@@ -47,6 +47,8 @@ pub mod sys;
 mod winapi;
 #[cfg(windows)]
 mod winreg;
+#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]
+mod ctypes;
 
 use crate::{builtins::PyModule, PyRef, VirtualMachine};
 use std::{borrow::Cow, collections::HashMap};
@@ -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,
+        }
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -47,6 +47,8 @@ pub mod sys;`
`47`	`47`	`mod winapi;`
`48`	`48`	`#[cfg(windows)]`
`49`	`49`	`mod winreg;`
	`50`	`+#[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]`
	`51`	`+mod ctypes;`
`50`	`52`
`51`	`53`	`use crate::{builtins::PyModule, PyRef, VirtualMachine};`
`52`	`54`	`use std::{borrow::Cow, collections::HashMap};`
`@@ -124,5 +126,9 @@ pub fn get_module_inits() -> StdlibMap {`
`124`	`126`	`"_winapi" => winapi::make_module,`
`125`	`127`	`"winreg" => winreg::make_module,`
`126`	`128`	`}`
	`129`	`+ #[cfg(any(not(target_arch = "wasm32"), target_os = "wasi"))]`
	`130`	`+ {`
	`131`	`+ "_ctypes" => ctypes::make_module,`
	`132`	`+ }`
`127`	`133`	`}`
`128`	`134`	`}`