diff options
| author | Paul Dicker <pitdicker@gmail.com> | 2016-02-07 19:31:14 +0100 |
|---|---|---|
| committer | Paul Dicker <pitdicker@gmail.com> | 2016-02-07 19:31:14 +0100 |
| commit | d47036cbd1e181da6eca7f4f125e092bedeb35e0 (patch) | |
| tree | 7303638170fdc3a3d75466037c2914f01ad5bc17 /src/libstd/sys | |
| parent | fb172b676e5ab951e58b98cede795ab1a7557a58 (diff) | |
| download | rust-d47036cbd1e181da6eca7f4f125e092bedeb35e0.tar.gz rust-d47036cbd1e181da6eca7f4f125e092bedeb35e0.zip | |
Don't let `remove_dir_all` recursively remove a symlink
See #29412
Diffstat (limited to 'src/libstd/sys')
| -rw-r--r-- | src/libstd/sys/unix/fs.rs | 11 | ||||
| -rw-r--r-- | src/libstd/sys/windows/fs.rs | 13 |
2 files changed, 22 insertions, 2 deletions
diff --git a/src/libstd/sys/unix/fs.rs b/src/libstd/sys/unix/fs.rs index a2e6f467b17..b9385ebe5c5 100644 --- a/src/libstd/sys/unix/fs.rs +++ b/src/libstd/sys/unix/fs.rs @@ -511,10 +511,19 @@ pub fn rmdir(p: &Path) -> io::Result<()> { } pub fn remove_dir_all(path: &Path) -> io::Result<()> { + let filetype = try!(lstat(path)).file_type(); + if filetype.is_symlink() { + unlink(path) + } else { + remove_dir_all_recursive(path) + } +} + +fn remove_dir_all_recursive(path: &Path) -> io::Result<()> { for child in try!(readdir(path)) { let child = try!(child); if try!(child.file_type()).is_dir() { - try!(remove_dir_all(&child.path())); + try!(remove_dir_all_recursive(&child.path())); } else { try!(unlink(&child.path())); } diff --git a/src/libstd/sys/windows/fs.rs b/src/libstd/sys/windows/fs.rs index 9f9290b751d..b0decfa463f 100644 --- a/src/libstd/sys/windows/fs.rs +++ b/src/libstd/sys/windows/fs.rs @@ -525,11 +525,22 @@ pub fn rmdir(p: &Path) -> io::Result<()> { } pub fn remove_dir_all(path: &Path) -> io::Result<()> { + let filetype = try!(lstat(path)).file_type(); + if filetype.is_symlink() { + // On Windows symlinks to files and directories are removed differently. + // rmdir only deletes dir symlinks and junctions, not file symlinks. + rmdir(path) + } else { + remove_dir_all_recursive(path) + } +} + +fn remove_dir_all_recursive(path: &Path) -> io::Result<()> { for child in try!(readdir(path)) { let child = try!(child); let child_type = try!(child.file_type()); if child_type.is_dir() { - try!(remove_dir_all(&child.path())); + try!(remove_dir_all_recursive(&child.path())); } else if child_type.is_symlink_dir() { try!(rmdir(&child.path())); } else { |
