tempfile-fast 0.3.6

extern crate libc;

use std::ffi::CString;
use std::fs;
use std::io;
use std::os::unix::ffi::OsStrExt;
use std::os::unix::io::AsRawFd;
use std::os::unix::io::FromRawFd;
use std::path::Path;

use self::libc::AT_FDCWD;
use self::libc::AT_SYMLINK_FOLLOW;
use self::libc::O_CLOEXEC;
use self::libc::O_RDWR;
use self::libc::O_TMPFILE;
use self::libc::c_char;
use self::libc::linkat;
use self::libc::open64 as open;

pub fn link_at<P: AsRef<Path>>(what: &fs::File, dest: P) -> io::Result<()> {
    let old_path: CString = CString::new(format!("/proc/self/fd/{}", what.as_raw_fd())).unwrap();
    let new_path = cstr(dest.as_ref())?;

    unsafe { link_symlink_fd_at(&old_path, &new_path) }
}

// Stolen from tempfile / std < 1.6.0.
pub fn cstr(path: &Path) -> io::Result<CString> {
    CString::new(path.as_os_str().as_bytes())
        .map_err(|_| io::Error::new(io::ErrorKind::InvalidInput, "path contained a null"))
}

pub fn create_nonexclusive_tempfile_in(dir: impl AsRef<Path>) -> io::Result<fs::File> {
    let path: &Path = dir.as_ref();
    match unsafe {
        let path = cstr(path)?;
        open(path.as_ptr(), O_CLOEXEC | O_TMPFILE | O_RDWR, 0o600)
    } {
        -1 => Err(io::ErrorKind::InvalidInput.into()),
        fd => Ok(unsafe { FromRawFd::from_raw_fd(fd) }),
    }
}

/// Attempt to link an old symlink to a file back into the filesystem.
unsafe fn link_symlink_fd_at(old_path: &CString, new_path: &CString) -> io::Result<()> {
    if unsafe {
        linkat(
            AT_FDCWD,
            old_path.as_ptr() as *const c_char,
            AT_FDCWD,
            new_path.as_ptr() as *const c_char,
            AT_SYMLINK_FOLLOW,
        )
    } != 0
    {
        Err(io::Error::last_os_error())
    } else {
        Ok(())
    }
}