[pbs-devel] [PATCH proxmox] sys: file: use renameat2() from `nix` crate
Christoph Heiss
c.heiss at proxmox.com
Thu Jul 11 19:07:09 CEST 2024
Saves us from converting the paths to raw C strings ourselves, thus
simplifying it quite a bit.
No functional changes.
Signed-off-by: Christoph Heiss <c.heiss at proxmox.com>
---
proxmox-sys/src/fs/file.rs | 51 ++++++++++++++++----------------------
1 file changed, 22 insertions(+), 29 deletions(-)
diff --git a/proxmox-sys/src/fs/file.rs b/proxmox-sys/src/fs/file.rs
index ac513891..3b20c4ea 100644
--- a/proxmox-sys/src/fs/file.rs
+++ b/proxmox-sys/src/fs/file.rs
@@ -10,7 +10,6 @@ use nix::errno::Errno;
use nix::fcntl::OFlag;
use nix::sys::stat;
use nix::unistd;
-use nix::NixPath;
use serde_json::Value;
use proxmox_lang::try_block;
@@ -266,30 +265,32 @@ pub fn atomic_open_or_create_file<P: AsRef<Path>>(
// rotate the file into place, but use `RENAME_NOREPLACE`, so in case 2 processes race against
// the initialization, the first one wins!
- let rename_result = temp_file_name.with_nix_path(|c_file_name| {
- path.with_nix_path(|new_path| unsafe {
- // This also works on file systems which don't support hardlinks (eg. vfat)
- match Errno::result(libc::renameat2(
- libc::AT_FDCWD,
- c_file_name.as_ptr(),
- libc::AT_FDCWD,
- new_path.as_ptr(),
- libc::RENAME_NOREPLACE,
- )) {
- Err(Errno::EINVAL) => (), // dumb file system, try `link`+`unlink`
- other => return other,
- };
- // but some file systems don't support `RENAME_NOREPLACE`
+ let rename_result = match nix::fcntl::renameat2(
+ None, // AT_FDCWD
+ &temp_file_name,
+ None, // AT_FDCWD
+ path,
+ nix::fcntl::RenameFlags::RENAME_NOREPLACE,
+ ) {
+ Err(Errno::EINVAL) => {
+ // some file systems don't support `RENAME_NOREPLACE`
// so we just use `link` + `unlink` instead
- let result = Errno::result(libc::link(c_file_name.as_ptr(), new_path.as_ptr()));
- let _ = libc::unlink(c_file_name.as_ptr());
+ let result = nix::unistd::linkat(
+ None,
+ &temp_file_name,
+ None,
+ &path.to_path_buf(),
+ nix::unistd::LinkatFlags::NoSymlinkFollow,
+ );
+ let _ = nix::unistd::unlink(&temp_file_name);
result
- })
- });
+ }
+ other => other,
+ };
match rename_result {
- Ok(Ok(Ok(_))) => Ok(file),
- Ok(Ok(Err(err))) => {
+ Ok(_) => Ok(file),
+ Err(err) => {
// if another process has already raced ahead and created
// the file, let's just open theirs instead:
let _ = nix::unistd::unlink(&temp_file_name);
@@ -308,14 +309,6 @@ pub fn atomic_open_or_create_file<P: AsRef<Path>>(
);
}
}
- Ok(Err(err)) => {
- let _ = nix::unistd::unlink(&temp_file_name);
- bail!("with_nix_path {:?} failed - {}", path, err);
- }
- Err(err) => {
- let _ = nix::unistd::unlink(&temp_file_name);
- bail!("with_nix_path {:?} failed - {}", temp_file_name, err);
- }
}
}
--
2.45.1
More information about the pbs-devel
mailing list