add support for chmod/fchmod/fchmodat

Author: Mic92

src/sys/stat.rs

@@ -147,3 +147,35 @@ pub fn fstatat<P: ?Sized + NixPath>(dirfd: RawFd, pathname: &P, f: AtFlags) -> R
     Ok(dst)
 }
 
+/// Change permissions of a file 
+/// ([see chmod(2)](http://man7.org/linux/man-pages/man2/chmod.2.html)).
+pub fn chmod<P: ?Sized + NixPath>(pathname: &P, mode: Mode) -> Result<()> {
+    let res = try!(pathname.with_nix_path(|cstr| {
+        unsafe { libc::chmod(cstr.as_ptr(), mode.bits()) }
+    }));
+
+    Errno::result(res).map(drop)
+}
+
+/// Change permissions of a file 
+/// ([see fchmod(2)](http://man7.org/linux/man-pages/man2/fchmod.2.html)).
+pub fn fchmod(fd: RawFd, mode: Mode) -> Result<()> {
+    let res = unsafe { libc::fchmod(fd, mode.bits()) };
+
+    Errno::result(res).map(drop)
+}
+
+/// Change permissions of a file 
+/// ([see fchmodat(2)](http://man7.org/linux/man-pages/man2/fchmodat.2.html)).
+pub fn fchmodat<P: ?Sized + NixPath>(dirfd: RawFd, pathname: &P, mode: Mode, flags: AtFlags) -> Result<()> {
+    let res = try!(pathname.with_nix_path(|cstr| {
+        unsafe {
+            libc::fchmodat(dirfd,
+                           cstr.as_ptr(),
+                           mode.bits(),
+                           flags.bits())
+        }
+    }));
+
+    Errno::result(res).map(drop)
+}

test/test_stat.rs

@@ -1,12 +1,12 @@
 use std::fs::File;
-use std::os::unix::fs::symlink;
+use std::os::unix::fs::{symlink, PermissionsExt};
 use std::os::unix::prelude::AsRawFd;
+use tempfile::NamedTempFile;
 
 use libc::{S_IFMT, S_IFLNK};
 
 use nix::fcntl;
-use nix::sys::stat::{self, stat, fstat, lstat};
-use nix::sys::stat::FileStat;
+use nix::sys::stat::*;
 use nix::Result;
 use tempdir::TempDir;
 
@@ -81,11 +81,11 @@ fn test_fstatat() {
     File::create(&filename).unwrap();
     let dirfd = fcntl::open(tempdir.path(),
                             fcntl::OFlag::empty(),
-                            stat::Mode::empty());
+                            Mode::empty());
 
-    let result = stat::fstatat(dirfd.unwrap(),
-                               &filename,
-                               fcntl::AtFlags::empty());
+    let result = fstatat(dirfd.unwrap(),
+                         &filename,
+                         fcntl::AtFlags::empty());
     assert_stat_results(result);
 }
 
@@ -110,3 +110,30 @@ fn test_stat_fstat_lstat() {
     let fstat_result = fstat(link.as_raw_fd());
     assert_stat_results(fstat_result);
 }
+
+fn assert_mode(f: &NamedTempFile, mode: u32) {
+    assert_eq!(f.metadata().unwrap().permissions().mode(), 
+               mode);
+}
+
+#[test]
+fn test_chmod() {
+    let tempfile = NamedTempFile::new().unwrap();
+    chmod(tempfile.path(),
+          Mode::from_bits(0o755).unwrap()).unwrap();
+    assert_mode(&tempfile, 0o755);
+
+    fchmod(tempfile.as_raw_fd(),
+          Mode::from_bits(0o644).unwrap()).unwrap();
+    assert_mode(&tempfile, 0o644);
+
+    let parent_dir = tempfile.path().parent().unwrap();
+    let dirfd = fcntl::open(parent_dir,
+                            fcntl::OFlag::empty(),
+                            Mode::empty()).unwrap();
+    fchmodat(dirfd,
+             tempfile.path().file_name().unwrap(),
+             Mode::from_bits(0o600).unwrap(),
+             fcntl::AtFlags::empty()).unwrap();
+    assert_mode(&tempfile, 0o600);
+}