-
Notifications
You must be signed in to change notification settings - Fork 19
/
close_preopen.rs
48 lines (40 loc) · 1.3 KB
/
close_preopen.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
use std::{env, process};
use wasi_tests::{assert_errno, open_scratch_directory};
unsafe fn test_close_preopen(dir_fd: wasi::Fd) {
let pre_fd: wasi::Fd = (libc::STDERR_FILENO + 1) as wasi::Fd;
assert!(dir_fd > pre_fd, "dir_fd number");
// Try to close a preopened directory handle.
wasi::fd_close(pre_fd).expect("closing a preopened file descriptor");
// Ensure that dir_fd is still open.
let dir_fdstat = wasi::fd_fdstat_get(dir_fd).expect("failed fd_fdstat_get");
assert_eq!(
dir_fdstat.fs_filetype,
wasi::FILETYPE_DIRECTORY,
"expected the scratch directory to be a directory",
);
// Ensure that pre_fd is closed.
assert_errno!(
wasi::fd_fdstat_get(pre_fd).expect_err("failed fd_fdstat_get"),
wasi::ERRNO_BADF
);
}
fn main() {
let mut args = env::args();
let prog = args.next().unwrap();
let arg = if let Some(arg) = args.next() {
arg
} else {
eprintln!("usage: {} <scratch directory>", prog);
process::exit(1);
};
// Open scratch directory
let dir_fd = match open_scratch_directory(&arg) {
Ok(dir_fd) => dir_fd,
Err(err) => {
eprintln!("{}", err);
process::exit(1)
}
};
// Run the tests.
unsafe { test_close_preopen(dir_fd) }
}