implemented with winapi

jiang1997 · jiang1997 · commit aee97968ef2e · 2022-08-18T16:58:40.000+08:00
diff --git a/stdlib/src/lib.rs b/stdlib/src/lib.rs
@@ -12,6 +12,8 @@ mod cmath;
 mod contextvars;
 mod csv;
 mod dis;
+#[cfg(windows)]
+mod win32api;
 mod gc;
 mod hashlib;
 mod json;
@@ -133,6 +135,11 @@ 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
@@ -0,0 +1,53 @@
+pub(crate) use win32api::make_module;
+
+#[pymodule]
+mod win32api {
+    use crate::vm::{
+        builtins::{PyStrRef},
+        function::{},
+        PyResult,
+    };
+
+    use winapi::shared::minwindef::HINSTANCE__;
+    use winapi::um::libloaderapi::{LoadLibraryW, GetModuleFileNameW};
+
+    const MAX_PATH: usize = 260;
+
+    #[pyfunction]
+    fn LoadLibrary(path: PyStrRef) -> PyResult<usize> {
+        let lp_caption = to_wstring(path.as_str());
+        let handle = unsafe {
+            LoadLibraryW(lp_caption.as_ptr()) 
+        }  as usize;
+        
+        Ok(handle)
+    }
+
+    #[pyfunction]
+    fn GetModuleFileName(handle: usize) -> String {
+        let mut path: Vec<u16> = vec![0; MAX_PATH];
+
+        let length = unsafe {
+            GetModuleFileNameW(handle as *mut HINSTANCE__, path.as_mut_ptr(), MAX_PATH as u32)
+        };
+
+        if length == 0  {
+            // error
+        }
+
+        let (path, _) = path.split_at(length as usize);
+        String::from_utf16(&path).unwrap()
+    }
+
+    #[cfg(windows)]
+    // Get win32 lpstr from &str, converting u8 to u16 and appending '\0'
+    // See retep998's traits for a more general solution: https://users.rust-lang.org/t/tidy-pattern-to-work-with-lpstr-mutable-char-array/2976/2
+    fn to_wstring(value: &str) -> Vec<u16> {
+        use std::os::windows::ffi::OsStrExt;
+    
+        std::ffi::OsStr::new(value)
+            .encode_wide()
+            .chain(std::iter::once(0))
+            .collect()
+    }
+}
