feat(fs): add set_permissions for Windows

Author: Berrysoft

compio-fs/src/lib.rs

@@ -21,3 +21,25 @@ pub mod named_pipe;
 
 #[cfg(unix)]
 pub mod pipe;
+
+#[cfg(windows)]
+pub(crate) fn path_string(
+    path: impl AsRef<std::path::Path>,
+) -> std::io::Result<widestring::U16CString> {
+    widestring::U16CString::from_os_str(path.as_ref().as_os_str()).map_err(|_| {
+        std::io::Error::new(
+            std::io::ErrorKind::InvalidInput,
+            "file name contained an unexpected NUL byte",
+        )
+    })
+}
+
+#[cfg(unix)]
+pub(crate) fn path_string(path: impl AsRef<std::path::Path>) -> std::io::Result<std::ffi::CString> {
+    CString::new(path.as_ref().as_os_str().as_bytes().to_vec()).map_err(|_| {
+        io::Error::new(
+            io::ErrorKind::InvalidInput,
+            "file name contained an unexpected NUL byte",
+        )
+    })
+}

compio-fs/src/metadata/windows.rs

@@ -6,24 +6,21 @@ use std::{
 };
 
 use compio_buf::{BufResult, IntoInner};
-use compio_driver::op::PathStat;
+use compio_driver::{op::PathStat, syscall};
 use compio_runtime::Runtime;
 use widestring::U16CString;
 use windows_sys::Win32::{
     Foundation::FILETIME,
     Storage::FileSystem::{
-        BY_HANDLE_FILE_INFORMATION, FILE_ATTRIBUTE_DIRECTORY, FILE_ATTRIBUTE_READONLY,
-        FILE_ATTRIBUTE_REPARSE_POINT,
+        SetFileAttributesW, BY_HANDLE_FILE_INFORMATION, FILE_ATTRIBUTE_DIRECTORY,
+        FILE_ATTRIBUTE_READONLY, FILE_ATTRIBUTE_REPARSE_POINT,
     },
 };
 
+use crate::path_string;
+
 async fn metadata_impl(path: impl AsRef<Path>, follow_symlink: bool) -> io::Result<Metadata> {
-    let path = U16CString::from_os_str(path.as_ref().as_os_str()).map_err(|_| {
-        io::Error::new(
-            io::ErrorKind::InvalidInput,
-            "file name contained an unexpected NUL byte",
-        )
-    })?;
+    let path = path_string(path)?;
     let op = PathStat::new(path, follow_symlink);
     let BufResult(res, op) = Runtime::current().submit(op).await;
     res.map(|_| Metadata::from_path_stat(op.into_inner()))
@@ -40,6 +37,17 @@ pub async fn symlink_metadata(path: impl AsRef<Path>) -> io::Result<Metadata> {
     metadata_impl(path, false).await
 }
 
+/// Changes the permissions found on a file or a directory.
+pub async fn set_permissions(path: impl AsRef<Path>, perm: Permissions) -> io::Result<()> {
+    let path = path_string(path)?;
+    Runtime::current()
+        .spawn_blocking(move || {
+            syscall!(BOOL, SetFileAttributesW(path.as_ptr(), perm.attrs))?;
+            Ok(())
+        })
+        .await
+}
+
 const fn create_u64(high: u32, low: u32) -> u64 {
     ((high as u64) << 32) | (low as u64)
 }

compio-fs/src/open_options/windows.rs

@@ -2,7 +2,6 @@ use std::{io, path::Path, ptr::null};
 
 use compio_driver::{op::OpenFile, FromRawFd, RawFd};
 use compio_runtime::Runtime;
-use widestring::U16CString;
 use windows_sys::Win32::{
     Foundation::{ERROR_INVALID_PARAMETER, GENERIC_READ, GENERIC_WRITE},
     Security::SECURITY_ATTRIBUTES,
@@ -13,7 +12,7 @@ use windows_sys::Win32::{
     },
 };
 
-use crate::File;
+use crate::{path_string, File};
 
 #[derive(Clone, Debug)]
 pub struct OpenOptions {
@@ -131,12 +130,7 @@ impl OpenOptions {
     }
 
     pub async fn open(&self, p: impl AsRef<Path>) -> io::Result<File> {
-        let p = U16CString::from_os_str(p.as_ref().as_os_str()).map_err(|_| {
-            io::Error::new(
-                io::ErrorKind::InvalidInput,
-                "file name contained an unexpected NUL byte",
-            )
-        })?;
+        let p = path_string(p)?;
         let op = OpenFile::new(
             p,
             self.get_access_mode()?,