implemented with windows

jiang1997 · jiang1997 · commit fae255b7ad8c · 2022-08-18T22:26:23.000+08:00
diff --git a/stdlib/src/lib.rs b/stdlib/src/lib.rs
@@ -12,8 +12,6 @@ mod cmath;
 mod contextvars;
 mod csv;
 mod dis;
-#[cfg(windows)]
-mod win32api;
 mod gc;
 mod hashlib;
 mod json;
@@ -135,11 +133,6 @@ pub fn get_module_inits() -> impl Iterator<Item = (Cow<'static, str>, StdlibInit
         {
             "_bz2" => bz2::make_module,
         }
-        // Windoes-only
-        #[cfg(windows)]
-        {
-            "win32api" => win32api::make_module,
-        }
         // Unix-only
         #[cfg(unix)]
         {
diff --git a/stdlib/src/win32api.rs b/stdlib/src/win32api.rs
diff --git a/vm/Cargo.toml b/vm/Cargo.toml
@@ -113,9 +113,9 @@ schannel = "0.1.19"
 widestring = "0.5.1"
 
 [target.'cfg(windows)'.dependencies.windows]
-version = "0.39"
+version = "0.39.0"
 features = [
-  "Win32_UI_Shell",
+  "Win32_UI_Shell", "Win32_System_LibraryLoader", "Win32_Foundation"
 ]
 
 [target.'cfg(windows)'.dependencies.winapi]
diff --git a/vm/src/stdlib/winapi.rs b/vm/src/stdlib/winapi.rs
@@ -16,6 +16,14 @@ mod _winapi {
         fileapi, handleapi, namedpipeapi, processenv, processthreadsapi, synchapi, winbase,
         winnt::HANDLE,
     };
+    // windows
+    use windows::Win32::Foundation::{
+        HINSTANCE, MAX_PATH,
+    };
+    use windows::Win32::System::LibraryLoader::{
+        LoadLibraryW, GetModuleFileNameW,
+    };
+    use windows::core::PCWSTR;
 
     #[pyattr]
     use winapi::{
@@ -402,4 +410,57 @@ mod _winapi {
         })
         .map(drop)
     }
+
+    use std::ffi::{OsStr, OsString};
+    use std::os::windows::prelude::*;
+    pub trait ToWide {
+        fn to_wide(&self) -> Vec<u16>;
+        fn to_wide_null(&self) -> Vec<u16>;
+    }
+    impl<T> ToWide for T where T: AsRef<OsStr> {
+        fn to_wide(&self) -> Vec<u16> {
+            self.as_ref().encode_wide().collect()
+        }
+        fn to_wide_null(&self) -> Vec<u16> {
+            self.as_ref().encode_wide().chain(Some(0)).collect()
+        }
+    }
+    pub trait FromWide where Self: Sized {
+        fn from_wide_null(wide: &[u16]) -> Self;
+    }
+    impl FromWide for OsString {
+        fn from_wide_null(wide: &[u16]) -> OsString {
+            let len = wide.iter().take_while(|&&c| c != 0).count();
+            OsString::from_wide(&wide[..len])
+        }
+    }
+
+
+    #[pyfunction]
+    fn LoadLibrary(path: PyStrRef, vm: &VirtualMachine) -> PyResult<isize> {
+        let path = path.as_str().to_wide_null();
+        let handle = unsafe { LoadLibraryW(PCWSTR::from_raw(path.as_ptr())).unwrap() };
+        if handle.is_invalid() {
+            return Err(
+                vm.new_runtime_error("LoadLibrary failed".to_owned())
+            );
+        }
+        Ok(handle.0)
+    }
+
+    #[pyfunction]
+    fn GetModuleFileName(handle: isize, vm: &VirtualMachine) -> PyResult<String> {
+        let mut path: Vec<u16> = vec![0; MAX_PATH as usize];
+        let handle = HINSTANCE(handle);
+
+        let length = unsafe { GetModuleFileNameW(handle, &mut path) };
+        if length == 0  {
+            return Err(
+                vm.new_runtime_error("GetModuleFileName failed".to_owned())
+            );
+        }
+
+        let (path, _) = path.split_at(length as usize);
+        Ok(String::from_utf16(&path).unwrap())
+    }
 }
