feat(fs): add set_permissions for unix

Author: Berrysoft

compio-driver/src/iour/op.rs

@@ -67,7 +67,7 @@ impl FileStat {
 }
 
 impl OpCode for FileStat {
-    fn create_entry(mut self: Pin<&mut Self>) -> io_uring::squeue::Entry {
+    fn create_entry(mut self: Pin<&mut Self>) -> OpEntry {
         static EMPTY_NAME: &[u8] = b"\0";
         opcode::Statx::new(
             Fd(self.fd),
@@ -76,6 +76,7 @@ impl OpCode for FileStat {
         )
         .flags(libc::AT_EMPTY_PATH)
         .build()
+        .into()
     }
 }
 
@@ -106,7 +107,7 @@ impl PathStat {
 }
 
 impl OpCode for PathStat {
-    fn create_entry(mut self: Pin<&mut Self>) -> io_uring::squeue::Entry {
+    fn create_entry(mut self: Pin<&mut Self>) -> OpEntry {
         let mut flags = libc::AT_EMPTY_PATH;
         if !self.follow_symlink {
             flags |= libc::AT_SYMLINK_NOFOLLOW;
@@ -118,6 +119,7 @@ impl OpCode for PathStat {
         )
         .flags(flags)
         .build()
+        .into()
     }
 }
 

compio-fs/src/file.rs

@@ -112,6 +112,18 @@ impl File {
             .await
     }
 
+    /// Changes the permissions on the underlying file.
+    #[cfg(unix)]
+    pub async fn set_permissions(&self, perm: Permissions) -> io::Result<()> {
+        let fd = self.try_as_raw_fd()? as _;
+        Runtime::current()
+            .spawn_blocking(move || {
+                syscall!(libc::fchmod(fd, perm.0))?;
+                Ok(())
+            })
+            .await
+    }
+
     async fn sync_impl(&self, datasync: bool) -> io::Result<()> {
         let op = Sync::new(self.try_as_raw_fd()?, datasync);
         Runtime::current().submit(op).await.0?;

compio-fs/src/lib.rs

@@ -36,9 +36,11 @@ pub(crate) fn path_string(
 
 #[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,
+    use std::os::unix::ffi::OsStrExt;
+
+    std::ffi::CString::new(path.as_ref().as_os_str().as_bytes().to_vec()).map_err(|_| {
+        std::io::Error::new(
+            std::io::ErrorKind::InvalidInput,
             "file name contained an unexpected NUL byte",
         )
     })

compio-fs/src/metadata/unix.rs

@@ -1,22 +1,18 @@
 use std::{
-    ffi::CString,
     io,
-    os::unix::prelude::{FileTypeExt, MetadataExt, OsStrExt, PermissionsExt},
+    os::unix::prelude::{FileTypeExt, MetadataExt, PermissionsExt},
     path::Path,
     time::{Duration, SystemTime},
 };
 
 use compio_buf::{BufResult, IntoInner};
-use compio_driver::op::PathStat;
+use compio_driver::{op::PathStat, syscall};
 use compio_runtime::Runtime;
 
+use crate::path_string;
+
 async fn metadata_impl(path: impl AsRef<Path>, follow_symlink: bool) -> io::Result<Metadata> {
-    let path = 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",
-        )
-    })?;
+    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_stat(op.into_inner()))
@@ -33,6 +29,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!(libc::chmod(path.as_ptr(), perm.0))?;
+            Ok(())
+        })
+        .await
+}
+
 /// Metadata information about a file.
 #[derive(Clone)]
 pub struct Metadata(pub(crate) libc::stat);

compio-fs/src/open_options/unix.rs

@@ -1,9 +1,9 @@
-use std::{ffi::CString, io, os::unix::prelude::OsStrExt, path::Path};
+use std::{io, path::Path};
 
 use compio_driver::{op::OpenFile, FromRawFd, RawFd};
 use compio_runtime::Runtime;
 
-use crate::File;
+use crate::{path_string, File};
 
 #[derive(Clone, Debug)]
 pub struct OpenOptions {
@@ -89,12 +89,7 @@ impl OpenOptions {
         if cfg!(not(any(target_os = "linux", target_os = "android"))) {
             flags |= libc::O_NONBLOCK;
         }
-        let p = CString::new(p.as_ref().as_os_str().as_bytes().to_vec()).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, flags, self.mode);
         let fd = Runtime::current().submit(op).await.0? as RawFd;
         Ok(unsafe { File::from_raw_fd(fd) })