Implement fd_fdstat_set_flags for Windows.
This commit implements `fd_fdstat_set_flags` for Windows. Additionally, it fixes a problem where `O_APPEND` was not working correctly because `GENERIC_WRITE` was always being set; as a result, `FILE_WRITE_DATA` could not be removed from the permission set to properly enable append-only mode. It also treats `O_TRUNC` with `O_APPEND` as an invalid argument error. This is because Windows cannot support these two flags together. To support `O_TRUNC`, the `GENERIC_WRITE` bit must be set for the file access flags. Setting this bit will cause `FILE_WRITE_DATA` to be set, which will not properly treat the file as append-only (it requires `FILE_APPEND_DATA` without `FILE_WRITE_DATA`).
This commit is contained in:
165
crates/test-programs/wasi-tests/src/bin/fd_flags_set.rs
Normal file
165
crates/test-programs/wasi-tests/src/bin/fd_flags_set.rs
Normal file
@@ -0,0 +1,165 @@
|
||||
use std::{env, process};
|
||||
use wasi;
|
||||
use wasi_tests::open_scratch_directory;
|
||||
|
||||
unsafe fn test_fd_fdstat_set_flags(dir_fd: wasi::Fd) {
|
||||
const FILE_NAME: &str = "file";
|
||||
let data = &[0u8; 100];
|
||||
|
||||
let file_fd = wasi::path_open(
|
||||
dir_fd,
|
||||
0,
|
||||
FILE_NAME,
|
||||
wasi::OFLAGS_CREAT,
|
||||
wasi::RIGHTS_FD_READ
|
||||
| wasi::RIGHTS_FD_WRITE
|
||||
| wasi::RIGHTS_FD_SEEK
|
||||
| wasi::RIGHTS_FD_TELL
|
||||
| wasi::RIGHTS_FD_FDSTAT_SET_FLAGS,
|
||||
0,
|
||||
wasi::FDFLAGS_APPEND,
|
||||
)
|
||||
.expect("opening a file");
|
||||
|
||||
// Write some data and then verify the written data
|
||||
assert_eq!(
|
||||
wasi::fd_write(
|
||||
file_fd,
|
||||
&[wasi::Ciovec {
|
||||
buf: data.as_ptr(),
|
||||
buf_len: data.len(),
|
||||
}],
|
||||
)
|
||||
.expect("writing to a file"),
|
||||
data.len(),
|
||||
"should write {} bytes",
|
||||
data.len(),
|
||||
);
|
||||
|
||||
wasi::fd_seek(file_fd, 0, wasi::WHENCE_SET).expect("seeking file");
|
||||
|
||||
let buffer = &mut [0u8; 100];
|
||||
|
||||
assert_eq!(
|
||||
wasi::fd_read(
|
||||
file_fd,
|
||||
&[wasi::Iovec {
|
||||
buf: buffer.as_mut_ptr(),
|
||||
buf_len: buffer.len(),
|
||||
}]
|
||||
)
|
||||
.expect("reading file"),
|
||||
buffer.len(),
|
||||
"shoudl read {} bytes",
|
||||
buffer.len()
|
||||
);
|
||||
|
||||
assert_eq!(&data[..], &buffer[..]);
|
||||
|
||||
let data = &[1u8; 100];
|
||||
|
||||
// Seek back to the start to ensure we're in append-only mode
|
||||
wasi::fd_seek(file_fd, 0, wasi::WHENCE_SET).expect("seeking file");
|
||||
|
||||
assert_eq!(
|
||||
wasi::fd_write(
|
||||
file_fd,
|
||||
&[wasi::Ciovec {
|
||||
buf: data.as_ptr(),
|
||||
buf_len: data.len(),
|
||||
}],
|
||||
)
|
||||
.expect("writing to a file"),
|
||||
data.len(),
|
||||
"should write {} bytes",
|
||||
data.len(),
|
||||
);
|
||||
|
||||
wasi::fd_seek(file_fd, 100, wasi::WHENCE_SET).expect("seeking file");
|
||||
|
||||
assert_eq!(
|
||||
wasi::fd_read(
|
||||
file_fd,
|
||||
&[wasi::Iovec {
|
||||
buf: buffer.as_mut_ptr(),
|
||||
buf_len: buffer.len(),
|
||||
}]
|
||||
)
|
||||
.expect("reading file"),
|
||||
buffer.len(),
|
||||
"shoudl read {} bytes",
|
||||
buffer.len()
|
||||
);
|
||||
|
||||
assert_eq!(&data[..], &buffer[..]);
|
||||
|
||||
wasi::fd_fdstat_set_flags(file_fd, 0).expect("disabling flags");
|
||||
|
||||
// Overwrite some existing data to ensure the append mode is now off
|
||||
wasi::fd_seek(file_fd, 0, wasi::WHENCE_SET).expect("seeking file");
|
||||
|
||||
let data = &[2u8; 100];
|
||||
|
||||
assert_eq!(
|
||||
wasi::fd_write(
|
||||
file_fd,
|
||||
&[wasi::Ciovec {
|
||||
buf: data.as_ptr(),
|
||||
buf_len: data.len(),
|
||||
}],
|
||||
)
|
||||
.expect("writing to a file"),
|
||||
data.len(),
|
||||
"should write {} bytes",
|
||||
data.len(),
|
||||
);
|
||||
|
||||
wasi::fd_seek(file_fd, 0, wasi::WHENCE_SET).expect("seeking file");
|
||||
|
||||
assert_eq!(
|
||||
wasi::fd_read(
|
||||
file_fd,
|
||||
&[wasi::Iovec {
|
||||
buf: buffer.as_mut_ptr(),
|
||||
buf_len: buffer.len(),
|
||||
}]
|
||||
)
|
||||
.expect("reading file"),
|
||||
buffer.len(),
|
||||
"shoudl read {} bytes",
|
||||
buffer.len()
|
||||
);
|
||||
|
||||
assert_eq!(&data[..], &buffer[..]);
|
||||
|
||||
wasi::fd_close(file_fd).expect("close file");
|
||||
|
||||
let stat = wasi::path_filestat_get(dir_fd, 0, FILE_NAME).expect("stat path");
|
||||
|
||||
assert_eq!(stat.size, 200, "expected a file size of 200");
|
||||
|
||||
wasi::path_unlink_file(dir_fd, FILE_NAME).expect("unlinking file");
|
||||
}
|
||||
|
||||
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);
|
||||
};
|
||||
|
||||
let dir_fd = match open_scratch_directory(&arg) {
|
||||
Ok(dir_fd) => dir_fd,
|
||||
Err(err) => {
|
||||
eprintln!("{}", err);
|
||||
process::exit(1)
|
||||
}
|
||||
};
|
||||
|
||||
unsafe {
|
||||
test_fd_fdstat_set_flags(dir_fd);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user