squiid/rust
1
0
Fork 0
Code Issues Pull requests Projects Activity

Rollup merge of #94724 - cuviper:rmdirall-cstr, r=Dylan-DPC

Browse source 
unix: Avoid name conversions in `remove_dir_all_recursive`

Each recursive call was creating an `OsString` for a `&Path`, only for
it to be turned into a `CString` right away. Instead we can directly
pass `.name_cstr()`, saving two allocations each time.
This commit is contained in:
Dylan DPC 2022-03-08 22:44:00 +01:00 committed by GitHub
parent ff54e34463 ef3e33bd16
commit a67b6299b4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
library/std/src/sys/unix/fs.rs
View file

@ -1604,17 +1604,15 @@ mod remove_dir_impl {
} }
} }
fn remove_dir_all_recursive(parent_fd: Option<RawFd>, p: &Path) -> io::Result<()> { fn remove_dir_all_recursive(parent_fd: Option<RawFd>, path: &CStr) -> io::Result<()> {
let pcstr = cstr(p)?;
// try opening as directory // try opening as directory
let fd = match openat_nofollow_dironly(parent_fd, &pcstr) { let fd = match openat_nofollow_dironly(parent_fd, &path) {
Err(err) if err.raw_os_error() == Some(libc::ENOTDIR) => { Err(err) if err.raw_os_error() == Some(libc::ENOTDIR) => {
// not a directory - don't traverse further // not a directory - don't traverse further
return match parent_fd { return match parent_fd {
// unlink... // unlink...
Some(parent_fd) => { Some(parent_fd) => {
cvt(unsafe { unlinkat(parent_fd, pcstr.as_ptr(), 0) }).map(drop) cvt(unsafe { unlinkat(parent_fd, path.as_ptr(), 0) }).map(drop)
} }
// ...unless this was supposed to be the deletion root directory // ...unless this was supposed to be the deletion root directory
None => Err(err), None => Err(err),
@ -1627,26 +1625,27 @@ mod remove_dir_impl {
let (dir, fd) = fdreaddir(fd)?; let (dir, fd) = fdreaddir(fd)?;
for child in dir { for child in dir {
let child = child?; let child = child?;
let child_name = child.name_cstr();
match is_dir(&child) { match is_dir(&child) {
Some(true) => { Some(true) => {
remove_dir_all_recursive(Some(fd), Path::new(&child.file_name()))?; remove_dir_all_recursive(Some(fd), child_name)?;
} }
Some(false) => { Some(false) => {
cvt(unsafe { unlinkat(fd, child.name_cstr().as_ptr(), 0) })?; cvt(unsafe { unlinkat(fd, child_name.as_ptr(), 0) })?;
} }
None => { None => {
// POSIX specifies that calling unlink()/unlinkat(..., 0) on a directory can succeed // POSIX specifies that calling unlink()/unlinkat(..., 0) on a directory can succeed
// if the process has the appropriate privileges. This however can causing orphaned // if the process has the appropriate privileges. This however can causing orphaned
// directories requiring an fsck e.g. on Solaris and Illumos. So we try recursing // directories requiring an fsck e.g. on Solaris and Illumos. So we try recursing
// into it first instead of trying to unlink() it. // into it first instead of trying to unlink() it.
remove_dir_all_recursive(Some(fd), Path::new(&child.file_name()))?; remove_dir_all_recursive(Some(fd), child_name)?;
} }
} }
} }
// unlink the directory after removing its contents // unlink the directory after removing its contents
cvt(unsafe { cvt(unsafe {
unlinkat(parent_fd.unwrap_or(libc::AT_FDCWD), pcstr.as_ptr(), libc::AT_REMOVEDIR) unlinkat(parent_fd.unwrap_or(libc::AT_FDCWD), path.as_ptr(), libc::AT_REMOVEDIR)
})?; })?;
Ok(()) Ok(())
} }
@ -1659,7 +1658,7 @@ mod remove_dir_impl {
if attr.file_type().is_symlink() { if attr.file_type().is_symlink() {
crate::fs::remove_file(p) crate::fs::remove_file(p)
} else { } else {
remove_dir_all_recursive(None, p) remove_dir_all_recursive(None, &cstr(p)?)
} }
} }