More overlapped implementation

RustPython · youknowone · Apr 27, 2025 · Apr 27, 2025 · Apr 27, 2025 · ad8caa3317924548ae609f584ae4019ccb76400e
commit ad8caa3317924548ae609f584ae4019ccb76400e
diff --git a/stdlib/src/overlapped.rs b/stdlib/src/overlapped.rs
@@ -297,6 +297,10 @@ mod _overlapped {
        }
    }

+    unsafe fn u64_to_handle(raw_ptr_value: u64) -> HANDLE {
+        raw_ptr_value as HANDLE
+    }
+
    #[pyfunction]
    fn CreateIoCompletionPort(
        handle: isize,
@@ -354,4 +358,56 @@ mod _overlapped {
        ]);
        Ok(value.into())
    }
+
+    #[pyfunction]
+    fn CreateEvent(
+        event_attributes: PyObjectRef,
+        manual_reset: bool,
+        initial_state: bool,
+        name: Option<String>,
+        vm: &VirtualMachine,
+    ) -> PyResult<isize> {
+        if !vm.is_none(&event_attributes) {
+            return Err(vm.new_value_error("EventAttributes must be None".to_owned()));
+        }
+
+        let name = match name {
+            Some(name) => {
+                let name = widestring::WideCString::from_str(&name).unwrap();
+                name.as_ptr()
+            }
+            None => std::ptr::null(),
+        };
+        let event = unsafe {
+            windows_sys::Win32::System::Threading::CreateEventW(
+                std::ptr::null(),
+                manual_reset as _,
+                initial_state as _,
+                name,
+            ) as isize
+        };
+        if event == NULL {
+            return Err(errno_err(vm));
+        }
+        Ok(event)
+    }
+
+    #[pyfunction]
+    fn SetEvent(handle: u64, vm: &VirtualMachine) -> PyResult<()> {
+        let ret = unsafe { windows_sys::Win32::System::Threading::SetEvent(u64_to_handle(handle)) };
+        if ret == 0 {
+            return Err(errno_err(vm));
+        }
+        Ok(())
+    }
+
+    #[pyfunction]
+    fn ResetEvent(handle: u64, vm: &VirtualMachine) -> PyResult<()> {
+        let ret =
+            unsafe { windows_sys::Win32::System::Threading::ResetEvent(u64_to_handle(handle)) };
+        if ret == 0 {
+            return Err(errno_err(vm));
+        }
+        Ok(())
+    }
 }