Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly
Sign up
Appearance settings

Commit c619035

Browse files
Auto merge of #139514 - Qelxiros:120426-dirfd, r=<try>
dirfd: preliminary unix and windows implementations Tracking issue: #120426 As per [this comment](#120426 (comment)), this issue needs someone to start work on an implementation, so I've implemented a couple functions for UNIX. There's a lot more work to be done here (most of the feature), so I'd love some guidance on what needs fixing in this PR and any notes on how to proceed. Thanks! try-job: `x86_64-msvc*` try-job: `test-various*` try-job: `dist-various*`
2 parents 7f2065a + f62937b commit c619035

File tree

13 files changed

+1918
-94
lines changed

13 files changed

+1918
-94
lines changed

‎library/std/src/fs.rs

Lines changed: 502 additions & 0 deletions
Large diffs are not rendered by default.

‎library/std/src/fs/tests.rs

Lines changed: 118 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use rand::RngCore;
22

3+
use super::Dir;
34
#[cfg(any(
45
windows,
56
target_os = "freebsd",
@@ -17,7 +18,7 @@ use crate::char::MAX_LEN_UTF8;
1718
target_vendor = "apple",
1819
))]
1920
use crate::fs::TryLockError;
20-
use crate::fs::{self, File, FileTimes, OpenOptions};
21+
use crate::fs::{self, File, FileTimes, OpenOptions, create_dir, exists};
2122
use crate::io::prelude::*;
2223
use crate::io::{BorrowedBuf, ErrorKind, SeekFrom};
2324
use crate::mem::MaybeUninit;
@@ -2084,3 +2085,119 @@ fn test_rename_junction() {
20842085
// Junction links are always absolute so we just check the file name is correct.
20852086
assert_eq!(fs::read_link(&dest).unwrap().file_name(), Some(not_exist.as_os_str()));
20862087
}
2088+
2089+
#[test]
2090+
fn test_dir_smoke_test() {
2091+
let tmpdir = tmpdir();
2092+
let dir = Dir::new(tmpdir.path());
2093+
println!("{dir:?}");
2094+
check!(dir);
2095+
}
2096+
2097+
#[test]
2098+
fn test_dir_read_file() {
2099+
let tmpdir = tmpdir();
2100+
let mut f = check!(File::create(tmpdir.join("foo.txt")));
2101+
check!(f.write(b"bar"));
2102+
check!(f.flush());
2103+
drop(f);
2104+
let dir = check!(Dir::new(tmpdir.path()));
2105+
let mut f = check!(dir.open("foo.txt"));
2106+
let mut buf = [0u8; 3];
2107+
check!(f.read_exact(&mut buf));
2108+
assert_eq!(b"bar", &buf);
2109+
}
2110+
2111+
#[test]
2112+
fn test_dir_write_file() {
2113+
let tmpdir = tmpdir();
2114+
let dir = check!(Dir::new(tmpdir.path()));
2115+
let mut f = check!(dir.open_with("foo.txt", &OpenOptions::new().write(true).create(true)));
2116+
check!(f.write(b"bar"));
2117+
check!(f.flush());
2118+
drop(f);
2119+
let mut f = check!(File::open(tmpdir.join("foo.txt")));
2120+
let mut buf = [0u8; 3];
2121+
check!(f.read_exact(&mut buf));
2122+
assert_eq!(b"bar", &buf);
2123+
}
2124+
2125+
#[test]
2126+
fn test_dir_remove_file() {
2127+
let tmpdir = tmpdir();
2128+
let mut f = check!(File::create(tmpdir.join("foo.txt")));
2129+
check!(f.write(b"bar"));
2130+
check!(f.flush());
2131+
drop(f);
2132+
let dir = check!(Dir::new(tmpdir.path()));
2133+
check!(dir.remove_file("foo.txt"));
2134+
assert!(!matches!(exists(tmpdir.join("foo.txt")), Ok(true)));
2135+
}
2136+
2137+
#[test]
2138+
fn test_dir_remove_dir() {
2139+
let tmpdir = tmpdir();
2140+
check!(create_dir(tmpdir.join("foo")));
2141+
let dir = check!(Dir::new(tmpdir.path()));
2142+
check!(dir.remove_dir("foo"));
2143+
assert!(!matches!(exists(tmpdir.join("foo")), Ok(true)));
2144+
}
2145+
2146+
#[test]
2147+
fn test_dir_rename_file() {
2148+
let tmpdir = tmpdir();
2149+
let mut f = check!(File::create(tmpdir.join("foo.txt")));
2150+
check!(f.write(b"bar"));
2151+
check!(f.flush());
2152+
drop(f);
2153+
let dir = check!(Dir::new(tmpdir.path()));
2154+
check!(dir.rename("foo.txt", &dir, "baz.txt"));
2155+
let mut f = check!(File::open(tmpdir.join("baz.txt")));
2156+
let mut buf = [0u8; 3];
2157+
check!(f.read_exact(&mut buf));
2158+
assert_eq!(b"bar", &buf);
2159+
}
2160+
2161+
#[test]
2162+
fn test_dir_create_dir() {
2163+
let tmpdir = tmpdir();
2164+
let dir = check!(Dir::new(tmpdir.path()));
2165+
check!(dir.create_dir("foo"));
2166+
check!(Dir::new(tmpdir.join("foo")));
2167+
}
2168+
2169+
#[test]
2170+
fn test_dir_open_dir() {
2171+
let tmpdir = tmpdir();
2172+
let dir1 = check!(Dir::new(tmpdir.path()));
2173+
check!(dir1.create_dir("foo"));
2174+
let dir2 = check!(Dir::new(tmpdir.path().join("foo")));
2175+
let mut f = check!(dir2.open_with("bar.txt", &OpenOptions::new().create(true).write(true)));
2176+
check!(f.write(b"baz"));
2177+
check!(f.flush());
2178+
drop(f);
2179+
let dir3 = check!(dir1.open_dir("foo"));
2180+
let mut f = check!(dir3.open("bar.txt"));
2181+
let mut buf = [0u8; 3];
2182+
check!(f.read_exact(&mut buf));
2183+
assert_eq!(b"baz", &buf);
2184+
}
2185+
2186+
#[test]
2187+
fn test_dir_symlink() {
2188+
let tmpdir = tmpdir();
2189+
if !got_symlink_permission(&tmpdir) {
2190+
return;
2191+
};
2192+
2193+
let dir = check!(Dir::new(tmpdir.path()));
2194+
let mut f = check!(dir.open_with("foo.txt", &OpenOptions::new().write(true).create(true)));
2195+
check!(f.write(b"quux"));
2196+
check!(f.flush());
2197+
drop(f);
2198+
check!(dir.symlink("foo.txt", "bar.txt"));
2199+
let mut f = check!(dir.open("bar.txt"));
2200+
let mut buf = [0u8; 4];
2201+
check!(f.read_exact(&mut buf));
2202+
assert_eq!(b"quux", &buf);
2203+
}

‎library/std/src/sys/fs/common.rs

Lines changed: 71 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
#![allow(dead_code)] // not used on all platforms
22

3-
use crate::fs;
3+
use crate::fmt;
4+
use crate::fs::{self, create_dir, remove_dir, remove_file, rename};
45
use crate::io::{self, Error, ErrorKind};
5-
use crate::path::Path;
6+
use crate::path::{Path, PathBuf};
7+
use crate::sys::fs::{File, OpenOptions, symlink};
68
use crate::sys_common::ignore_notfound;
79

810
pub(crate) const NOT_FILE_ERROR: Error = io::const_error!(
@@ -58,3 +60,70 @@ pub fn exists(path: &Path) -> io::Result<bool> {
5860
Err(error) => Err(error),
5961
}
6062
}
63+
64+
pub struct Dir {
65+
path: PathBuf,
66+
}
67+
68+
impl Dir {
69+
pub fn new<P: AsRef<Path>>(path: P) -> io::Result<Self> {
70+
Ok(Self { path: path.as_ref().to_path_buf() })
71+
}
72+
73+
pub fn new_with<P: AsRef<Path>>(path: P, _opts: &OpenOptions) -> io::Result<Self> {
74+
Ok(Self { path: path.as_ref().to_path_buf() })
75+
}
76+
77+
pub fn new_for_traversal<P: AsRef<Path>>(path: P) -> io::Result<Self> {
78+
Ok(Self { path: path.as_ref().to_path_buf() })
79+
}
80+
81+
pub fn open<P: AsRef<Path>>(&self, path: P) -> io::Result<File> {
82+
let mut opts = OpenOptions::new();
83+
opts.read(true);
84+
File::open(&self.path.join(path), &opts)
85+
}
86+
87+
pub fn open_with<P: AsRef<Path>>(&self, path: P, opts: &OpenOptions) -> io::Result<File> {
88+
File::open(&self.path.join(path), opts)
89+
}
90+
91+
pub fn create_dir<P: AsRef<Path>>(&self, path: P) -> io::Result<()> {
92+
create_dir(self.path.join(path))
93+
}
94+
95+
pub fn open_dir<P: AsRef<Path>>(&self, path: P) -> io::Result<Self> {
96+
Self::new(self.path.join(path))
97+
}
98+
99+
pub fn open_dir_with<P: AsRef<Path>>(&self, path: P, opts: &OpenOptions) -> io::Result<Self> {
100+
Self::new_with(self.path.join(path), opts)
101+
}
102+
103+
pub fn remove_file<P: AsRef<Path>>(&self, path: P) -> io::Result<()> {
104+
remove_file(self.path.join(path))
105+
}
106+
107+
pub fn remove_dir<P: AsRef<Path>>(&self, path: P) -> io::Result<()> {
108+
remove_dir(self.path.join(path))
109+
}
110+
111+
pub fn rename<P: AsRef<Path>, Q: AsRef<Path>>(
112+
&self,
113+
from: P,
114+
to_dir: &Self,
115+
to: Q,
116+
) -> io::Result<()> {
117+
rename(self.path.join(from), to_dir.path.join(to))
118+
}
119+
120+
pub fn symlink<P: AsRef<Path>, Q: AsRef<Path>>(&self, original: P, link: Q) -> io::Result<()> {
121+
symlink(original.as_ref(), link.as_ref())
122+
}
123+
}
124+
125+
impl fmt::Debug for Dir {
126+
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
127+
f.debug_struct("Dir").field("path", &self.path).finish()
128+
}
129+
}

‎library/std/src/sys/fs/hermit.rs

Lines changed: 29 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,14 +11,16 @@ use crate::path::{Path, PathBuf};
1111
use crate::sync::Arc;
1212
use crate::sys::common::small_c_string::run_path_with_cstr;
1313
use crate::sys::fd::FileDesc;
14-
pub use crate::sys::fs::common::{copy, exists};
14+
pub use crate::sys::fs::common::{Dir, copy, exists};
15+
use crate::sys::fs::{remove_dir, remove_file};
1516
use crate::sys::time::SystemTime;
1617
use crate::sys::{cvt, unsupported, unsupported_err};
1718
use crate::sys_common::{AsInner, AsInnerMut, FromInner, IntoInner};
1819
use crate::{fmt, mem};
1920

2021
#[derive(Debug)]
2122
pub struct File(FileDesc);
23+
2224
#[derive(Clone)]
2325
pub struct FileAttr {
2426
stat_val: stat_struct,
@@ -251,6 +253,32 @@ impl DirEntry {
251253
pub fn file_name_os_str(&self) -> &OsStr {
252254
self.name.as_os_str()
253255
}
256+
257+
pub fn open_file(&self) -> io::Result<File> {
258+
let mut opts = OpenOptions::new();
259+
opts.read(true);
260+
File::open(&self.path(), &opts)
261+
}
262+
263+
pub fn open_file_with(&self, opts: &OpenOptions) -> io::Result<File> {
264+
File::open(&self.path(), opts)
265+
}
266+
267+
pub fn open_dir(&self) -> io::Result<Dir> {
268+
Dir::new(self.path())
269+
}
270+
271+
pub fn open_dir_with(&self, opts: &OpenOptions) -> io::Result<Dir> {
272+
Dir::new_with(self.path(), opts)
273+
}
274+
275+
pub fn remove_file(&self) -> io::Result<()> {
276+
remove_file(&self.path())
277+
}
278+
279+
pub fn remove_dir(&self) -> io::Result<()> {
280+
remove_dir(&self.path())
281+
}
254282
}
255283

256284
impl OpenOptions {

‎library/std/src/sys/fs/mod.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ pub fn with_native_path<T>(path: &Path, f: &dyn Fn(&Path) -> io::Result<T>) -> i
4747
}
4848

4949
pub use imp::{
50-
DirBuilder, DirEntry, File, FileAttr, FilePermissions, FileTimes, FileType, OpenOptions,
50+
Dir,DirBuilder, DirEntry, File, FileAttr, FilePermissions, FileTimes, FileType, OpenOptions,
5151
ReadDir,
5252
};
5353

‎library/std/src/sys/fs/solid.rs

Lines changed: 28 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@ use crate::os::raw::{c_int, c_short};
99
use crate::os::solid::ffi::OsStrExt;
1010
use crate::path::{Path, PathBuf};
1111
use crate::sync::Arc;
12-
pub use crate::sys::fs::common::exists;
12+
pub use crate::sys::fs::common::{Dir, exists};
13+
use crate::sys::fs::{remove_dir, remove_file};
1314
use crate::sys::pal::{abi, error};
1415
use crate::sys::time::SystemTime;
1516
use crate::sys::{unsupported, unsupported_err};
@@ -219,6 +220,32 @@ impl DirEntry {
219220
_ => lstat(&self.path()).map(|m| m.file_type()),
220221
}
221222
}
223+
224+
pub fn open_file(&self) -> io::Result<File> {
225+
let mut opts = OpenOptions::new();
226+
opts.read(true);
227+
File::open(&self.path(), &opts)
228+
}
229+
230+
pub fn open_file_with(&self, opts: &OpenOptions) -> io::Result<File> {
231+
File::open(&self.path(), opts)
232+
}
233+
234+
pub fn open_dir(&self) -> io::Result<Dir> {
235+
Dir::new(self.path())
236+
}
237+
238+
pub fn open_dir_with(&self, opts: &OpenOptions) -> io::Result<Dir> {
239+
Dir::new_with(self.path(), opts)
240+
}
241+
242+
pub fn remove_file(&self) -> io::Result<()> {
243+
remove_file(&self.path())
244+
}
245+
246+
pub fn remove_dir(&self) -> io::Result<()> {
247+
remove_dir(&self.path())
248+
}
222249
}
223250

224251
impl OpenOptions {

0 commit comments

Comments
(0)

AltStyle によって変換されたページ (->オリジナル) /