51 lines
1.3 KiB
Rust
51 lines
1.3 KiB
Rust
|
use failure::*;
|
||
|
|
use nix::unistd;
|
||
|
|
use nix::sys::stat;
|
||
|
|
|
||
|
|
use std::fs::File;
|
||
|
|
use std::io::Write;
|
||
|
|
use std::path::Path;
|
||
|
|
|
||
|
|
pub fn file_set_contents<P: AsRef<Path>>(
|
||
|
|
path: P,
|
||
|
|
data: &[u8],
|
||
|
|
perm: Option<stat::Mode>,
|
||
|
|
) -> Result<(), Error> {
|
||
|
|
|
||
|
|
let path = path.as_ref();
|
||
|
|
|
||
|
|
let mut template = path.to_owned();
|
||
|
|
template.set_extension("tmp_XXXXXX");
|
||
|
|
let (fd, tmp_path) = match unistd::mkstemp(&template) {
|
||
|
|
Ok((fd, path)) => (fd, path),
|
||
|
|
Err(err) => bail!("mkstemp {:?} failed: {}", template, err),
|
||
|
|
};
|
||
|
|
|
||
|
|
let tmp_path = tmp_path.as_path();
|
||
|
|
|
||
|
|
let mode : stat::Mode = stat::Mode::from(
|
||
|
|
stat::Mode::S_IRUSR | stat::Mode::S_IWUSR |
|
||
|
|
stat::Mode::S_IRGRP | stat::Mode::S_IROTH
|
||
|
|
);
|
||
|
|
|
||
|
|
if let Err(err) = stat::fchmod(fd, mode) {
|
||
|
|
let _ = unistd::unlink(tmp_path);
|
||
|
|
bail!("fchmod {:?} failed: {}", tmp_path, err);
|
||
|
|
}
|
||
|
|
|
||
|
|
use std::os::unix::io::FromRawFd;
|
||
|
|
let mut file = unsafe { File::from_raw_fd(fd) };
|
||
|
|
|
||
|
|
if let Err(err) = file.write_all(data) {
|
||
|
|
let _ = unistd::unlink(tmp_path);
|
||
|
|
bail!("write failed: {}", err);
|
||
|
|
}
|
||
|
|
|
||
|
|
if let Err(err) = std::fs::rename(tmp_path, path) {
|
||
|
|
let _ = unistd::unlink(tmp_path);
|
||
|
|
bail!("Atomic rename failed for file {:?} - {}", path, err);
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|