Vendor things

This commit is contained in:
John Doty 2024-03-08 11:03:01 -08:00
parent 5deceec006
commit 977e3c17e5
19434 changed files with 10682014 additions and 0 deletions

View file

@ -0,0 +1,63 @@
use crate::FileTime;
use std::ffi::CString;
use std::fs::File;
use std::io;
use std::os::unix::prelude::*;
use std::path::Path;
pub fn set_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), false)
}
pub fn set_file_mtime(p: &Path, mtime: FileTime) -> io::Result<()> {
set_times(p, None, Some(mtime), false)
}
pub fn set_file_atime(p: &Path, atime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), None, false)
}
pub fn set_file_handle_times(
f: &File,
atime: Option<FileTime>,
mtime: Option<FileTime>,
) -> io::Result<()> {
let times = [super::to_timespec(&atime), super::to_timespec(&mtime)];
// On Android NDK before version 19, `futimens` is not available.
//
// For better compatibility, we reimplement `futimens` using `utimensat`,
// the same way as bionic libc uses it to implement `futimens`.
let rc = unsafe { libc::utimensat(f.as_raw_fd(), core::ptr::null(), times.as_ptr(), 0) };
if rc == 0 {
Ok(())
} else {
Err(io::Error::last_os_error())
}
}
pub fn set_symlink_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), true)
}
fn set_times(
p: &Path,
atime: Option<FileTime>,
mtime: Option<FileTime>,
symlink: bool,
) -> io::Result<()> {
let flags = if symlink {
libc::AT_SYMLINK_NOFOLLOW
} else {
0
};
let p = CString::new(p.as_os_str().as_bytes())?;
let times = [super::to_timespec(&atime), super::to_timespec(&mtime)];
let rc = unsafe { libc::utimensat(libc::AT_FDCWD, p.as_ptr(), times.as_ptr(), flags) };
if rc == 0 {
Ok(())
} else {
Err(io::Error::last_os_error())
}
}

View file

@ -0,0 +1,117 @@
//! On Linux we try to use the more accurate `utimensat` syscall but this isn't
//! always available so we also fall back to `utimes` if we couldn't find
//! `utimensat` at runtime.
use crate::FileTime;
use std::ffi::CString;
use std::fs;
use std::io;
use std::os::unix::prelude::*;
use std::path::Path;
use std::ptr;
use std::sync::atomic::AtomicBool;
use std::sync::atomic::Ordering::SeqCst;
pub fn set_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), false)
}
pub fn set_file_mtime(p: &Path, mtime: FileTime) -> io::Result<()> {
set_times(p, None, Some(mtime), false)
}
pub fn set_file_atime(p: &Path, atime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), None, false)
}
pub fn set_file_handle_times(
f: &fs::File,
atime: Option<FileTime>,
mtime: Option<FileTime>,
) -> io::Result<()> {
// Attempt to use the `utimensat` syscall, but if it's not supported by the
// current kernel then fall back to an older syscall.
static INVALID: AtomicBool = AtomicBool::new(false);
if !INVALID.load(SeqCst) {
let times = [super::to_timespec(&atime), super::to_timespec(&mtime)];
// We normally use a syscall because the `utimensat` function is documented
// as not accepting a file descriptor in the first argument (even though, on
// Linux, the syscall itself can accept a file descriptor there).
#[cfg(not(target_env = "musl"))]
let rc = unsafe {
libc::syscall(
libc::SYS_utimensat,
f.as_raw_fd(),
ptr::null::<libc::c_char>(),
times.as_ptr(),
0,
)
};
// However, on musl, we call the musl libc function instead. This is because
// on newer musl versions starting with musl 1.2, `timespec` is always a 64-bit
// value even on 32-bit targets. As a result, musl internally converts their
// `timespec` values to the correct ABI before invoking the syscall. Since we
// use `timespec` from the libc crate, it matches musl's definition and not
// the Linux kernel's version (for some platforms) so we must use musl's
// `utimensat` function to properly convert the value. musl's `utimensat`
// function allows file descriptors in the path argument so this is fine.
#[cfg(target_env = "musl")]
let rc = unsafe {
libc::utimensat(
f.as_raw_fd(),
ptr::null::<libc::c_char>(),
times.as_ptr(),
0,
)
};
if rc == 0 {
return Ok(());
}
let err = io::Error::last_os_error();
if err.raw_os_error() == Some(libc::ENOSYS) {
INVALID.store(true, SeqCst);
} else {
return Err(err);
}
}
super::utimes::set_file_handle_times(f, atime, mtime)
}
pub fn set_symlink_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), true)
}
fn set_times(
p: &Path,
atime: Option<FileTime>,
mtime: Option<FileTime>,
symlink: bool,
) -> io::Result<()> {
let flags = if symlink {
libc::AT_SYMLINK_NOFOLLOW
} else {
0
};
// Same as the `if` statement above.
static INVALID: AtomicBool = AtomicBool::new(false);
if !INVALID.load(SeqCst) {
let p = CString::new(p.as_os_str().as_bytes())?;
let times = [super::to_timespec(&atime), super::to_timespec(&mtime)];
let rc = unsafe { libc::utimensat(libc::AT_FDCWD, p.as_ptr(), times.as_ptr(), flags) };
if rc == 0 {
return Ok(());
}
let err = io::Error::last_os_error();
if err.raw_os_error() == Some(libc::ENOSYS) {
INVALID.store(true, SeqCst);
} else {
return Err(err);
}
}
super::utimes::set_times(p, atime, mtime, symlink)
}

View file

@ -0,0 +1,108 @@
//! Beginning with macOS 10.13, `utimensat` is supported by the OS, so here, we check if the symbol exists
//! and if not, we fallback to `utimes`.
use crate::FileTime;
use libc::{c_char, c_int, timespec};
use std::ffi::{CStr, CString};
use std::fs::File;
use std::os::unix::prelude::*;
use std::path::Path;
use std::sync::atomic::AtomicUsize;
use std::sync::atomic::Ordering::SeqCst;
use std::{io, mem};
pub fn set_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), false)
}
pub fn set_file_mtime(p: &Path, mtime: FileTime) -> io::Result<()> {
set_times(p, None, Some(mtime), false)
}
pub fn set_file_atime(p: &Path, atime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), None, false)
}
pub fn set_file_handle_times(
f: &File,
atime: Option<FileTime>,
mtime: Option<FileTime>,
) -> io::Result<()> {
// Attempt to use the `futimens` syscall, but if it's not supported by the
// current kernel then fall back to an older syscall.
if let Some(func) = futimens() {
let times = [super::to_timespec(&atime), super::to_timespec(&mtime)];
let rc = unsafe { func(f.as_raw_fd(), times.as_ptr()) };
if rc == 0 {
return Ok(());
} else {
return Err(io::Error::last_os_error());
}
}
super::utimes::set_file_handle_times(f, atime, mtime)
}
pub fn set_symlink_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), true)
}
fn set_times(
p: &Path,
atime: Option<FileTime>,
mtime: Option<FileTime>,
symlink: bool,
) -> io::Result<()> {
// Attempt to use the `utimensat` syscall, but if it's not supported by the
// current kernel then fall back to an older syscall.
if let Some(func) = utimensat() {
let flags = if symlink {
libc::AT_SYMLINK_NOFOLLOW
} else {
0
};
let p = CString::new(p.as_os_str().as_bytes())?;
let times = [super::to_timespec(&atime), super::to_timespec(&mtime)];
let rc = unsafe { func(libc::AT_FDCWD, p.as_ptr(), times.as_ptr(), flags) };
if rc == 0 {
return Ok(());
} else {
return Err(io::Error::last_os_error());
}
}
super::utimes::set_times(p, atime, mtime, symlink)
}
fn utimensat() -> Option<unsafe extern "C" fn(c_int, *const c_char, *const timespec, c_int) -> c_int>
{
static ADDR: AtomicUsize = AtomicUsize::new(0);
unsafe {
fetch(&ADDR, CStr::from_bytes_with_nul_unchecked(b"utimensat\0"))
.map(|sym| mem::transmute(sym))
}
}
fn futimens() -> Option<unsafe extern "C" fn(c_int, *const timespec) -> c_int> {
static ADDR: AtomicUsize = AtomicUsize::new(0);
unsafe {
fetch(&ADDR, CStr::from_bytes_with_nul_unchecked(b"futimens\0"))
.map(|sym| mem::transmute(sym))
}
}
fn fetch(cache: &AtomicUsize, name: &CStr) -> Option<usize> {
match cache.load(SeqCst) {
0 => {}
1 => return None,
n => return Some(n),
}
let sym = unsafe { libc::dlsym(libc::RTLD_DEFAULT, name.as_ptr() as *const _) };
let (val, ret) = if sym.is_null() {
(1, None)
} else {
(sym as usize, Some(sym as usize))
};
cache.store(val, SeqCst);
return ret;
}

View file

@ -0,0 +1,101 @@
use crate::FileTime;
use libc::{time_t, timespec};
use std::fs;
use std::os::unix::prelude::*;
cfg_if::cfg_if! {
if #[cfg(target_os = "linux")] {
mod utimes;
mod linux;
pub use self::linux::*;
} else if #[cfg(target_os = "android")] {
mod android;
pub use self::android::*;
} else if #[cfg(target_os = "macos")] {
mod utimes;
mod macos;
pub use self::macos::*;
} else if #[cfg(any(target_os = "aix",
target_os = "solaris",
target_os = "illumos",
target_os = "emscripten",
target_os = "freebsd",
target_os = "netbsd",
target_os = "openbsd",
target_os = "haiku"))] {
mod utimensat;
pub use self::utimensat::*;
} else {
mod utimes;
pub use self::utimes::*;
}
}
#[allow(dead_code)]
fn to_timespec(ft: &Option<FileTime>) -> timespec {
cfg_if::cfg_if! {
if #[cfg(any(target_os = "macos",
target_os = "illumos",
target_os = "freebsd"))] {
// https://github.com/apple/darwin-xnu/blob/a449c6a3b8014d9406c2ddbdc81795da24aa7443/bsd/sys/stat.h#L541
// https://github.com/illumos/illumos-gate/blob/master/usr/src/boot/sys/sys/stat.h#L312
// https://svnweb.freebsd.org/base/head/sys/sys/stat.h?view=markup#l359
const UTIME_OMIT: i64 = -2;
} else if #[cfg(target_os = "openbsd")] {
// https://github.com/openbsd/src/blob/master/sys/sys/stat.h#L189
const UTIME_OMIT: i64 = -1;
} else if #[cfg(target_os = "haiku")] {
// https://git.haiku-os.org/haiku/tree/headers/posix/sys/stat.h?#n106
const UTIME_OMIT: i64 = 1000000001;
} else if #[cfg(target_os = "aix")] {
// AIX hasn't disclosed system header files yet.
// https://github.com/golang/go/blob/master/src/cmd/vendor/golang.org/x/sys/unix/zerrors_aix_ppc64.go#L1007
const UTIME_OMIT: i64 = -3;
} else {
// http://cvsweb.netbsd.org/bsdweb.cgi/src/sys/sys/stat.h?annotate=1.68.30.1
// https://github.com/emscripten-core/emscripten/blob/master/system/include/libc/sys/stat.h#L71
const UTIME_OMIT: i64 = 1_073_741_822;
}
}
let mut ts: timespec = unsafe { std::mem::zeroed() };
if let &Some(ft) = ft {
ts.tv_sec = ft.seconds() as time_t;
ts.tv_nsec = ft.nanoseconds() as _;
} else {
ts.tv_sec = 0;
ts.tv_nsec = UTIME_OMIT as _;
}
ts
}
pub fn from_last_modification_time(meta: &fs::Metadata) -> FileTime {
FileTime {
seconds: meta.mtime(),
nanos: meta.mtime_nsec() as u32,
}
}
pub fn from_last_access_time(meta: &fs::Metadata) -> FileTime {
FileTime {
seconds: meta.atime(),
nanos: meta.atime_nsec() as u32,
}
}
pub fn from_creation_time(meta: &fs::Metadata) -> Option<FileTime> {
#[cfg(target_os = "bitrig")]
{
use std::os::bitrig::fs::MetadataExt;
Some(FileTime {
seconds: meta.st_birthtime(),
nanos: meta.st_birthtime_nsec() as u32,
})
}
#[cfg(not(target_os = "bitrig"))]
{
meta.created().map(|i| i.into()).ok()
}
}

View file

@ -0,0 +1,64 @@
use crate::FileTime;
use std::ffi::CString;
use std::fs::File;
use std::io;
use std::os::unix::prelude::*;
use std::path::Path;
pub fn set_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), false)
}
pub fn set_file_mtime(p: &Path, mtime: FileTime) -> io::Result<()> {
set_times(p, None, Some(mtime), false)
}
pub fn set_file_atime(p: &Path, atime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), None, false)
}
pub fn set_file_handle_times(
f: &File,
atime: Option<FileTime>,
mtime: Option<FileTime>,
) -> io::Result<()> {
let times = [super::to_timespec(&atime), super::to_timespec(&mtime)];
let rc = unsafe { libc::futimens(f.as_raw_fd(), times.as_ptr()) };
if rc == 0 {
Ok(())
} else {
Err(io::Error::last_os_error())
}
}
pub fn set_symlink_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), true)
}
fn set_times(
p: &Path,
atime: Option<FileTime>,
mtime: Option<FileTime>,
symlink: bool,
) -> io::Result<()> {
let flags = if symlink {
if cfg!(target_os = "emscripten") {
return Err(io::Error::new(
io::ErrorKind::Other,
"emscripten does not support utimensat for symlinks",
));
}
libc::AT_SYMLINK_NOFOLLOW
} else {
0
};
let p = CString::new(p.as_os_str().as_bytes())?;
let times = [super::to_timespec(&atime), super::to_timespec(&mtime)];
let rc = unsafe { libc::utimensat(libc::AT_FDCWD, p.as_ptr(), times.as_ptr(), flags) };
if rc == 0 {
Ok(())
} else {
Err(io::Error::last_os_error())
}
}

View file

@ -0,0 +1,130 @@
use crate::FileTime;
use std::ffi::CString;
use std::fs;
use std::io;
use std::os::unix::prelude::*;
use std::path::Path;
#[allow(dead_code)]
pub fn set_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), false)
}
#[allow(dead_code)]
pub fn set_file_mtime(p: &Path, mtime: FileTime) -> io::Result<()> {
set_times(p, None, Some(mtime), false)
}
#[allow(dead_code)]
pub fn set_file_atime(p: &Path, atime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), None, false)
}
#[cfg(not(target_env = "uclibc"))]
#[allow(dead_code)]
pub fn set_file_handle_times(
f: &fs::File,
atime: Option<FileTime>,
mtime: Option<FileTime>,
) -> io::Result<()> {
let (atime, mtime) = match get_times(atime, mtime, || f.metadata())? {
Some(pair) => pair,
None => return Ok(()),
};
let times = [to_timeval(&atime), to_timeval(&mtime)];
let rc = unsafe { libc::futimes(f.as_raw_fd(), times.as_ptr()) };
return if rc == 0 {
Ok(())
} else {
Err(io::Error::last_os_error())
};
}
#[cfg(target_env = "uclibc")]
#[allow(dead_code)]
pub fn set_file_handle_times(
f: &fs::File,
atime: Option<FileTime>,
mtime: Option<FileTime>,
) -> io::Result<()> {
let (atime, mtime) = match get_times(atime, mtime, || f.metadata())? {
Some(pair) => pair,
None => return Ok(()),
};
let times = [to_timespec(&atime), to_timespec(&mtime)];
let rc = unsafe { libc::futimens(f.as_raw_fd(), times.as_ptr()) };
return if rc == 0 {
Ok(())
} else {
Err(io::Error::last_os_error())
};
}
fn get_times(
atime: Option<FileTime>,
mtime: Option<FileTime>,
current: impl FnOnce() -> io::Result<fs::Metadata>,
) -> io::Result<Option<(FileTime, FileTime)>> {
let pair = match (atime, mtime) {
(Some(a), Some(b)) => (a, b),
(None, None) => return Ok(None),
(Some(a), None) => {
let meta = current()?;
(a, FileTime::from_last_modification_time(&meta))
}
(None, Some(b)) => {
let meta = current()?;
(FileTime::from_last_access_time(&meta), b)
}
};
Ok(Some(pair))
}
#[allow(dead_code)]
pub fn set_symlink_file_times(p: &Path, atime: FileTime, mtime: FileTime) -> io::Result<()> {
set_times(p, Some(atime), Some(mtime), true)
}
pub fn set_times(
p: &Path,
atime: Option<FileTime>,
mtime: Option<FileTime>,
symlink: bool,
) -> io::Result<()> {
let (atime, mtime) = match get_times(atime, mtime, || p.metadata())? {
Some(pair) => pair,
None => return Ok(()),
};
let p = CString::new(p.as_os_str().as_bytes())?;
let times = [to_timeval(&atime), to_timeval(&mtime)];
let rc = unsafe {
if symlink {
libc::lutimes(p.as_ptr(), times.as_ptr())
} else {
libc::utimes(p.as_ptr(), times.as_ptr())
}
};
return if rc == 0 {
Ok(())
} else {
Err(io::Error::last_os_error())
};
}
fn to_timeval(ft: &FileTime) -> libc::timeval {
libc::timeval {
tv_sec: ft.seconds() as libc::time_t,
tv_usec: (ft.nanoseconds() / 1000) as libc::suseconds_t,
}
}
#[cfg(target_env = "uclibc")]
fn to_timespec(ft: &FileTime) -> libc::timespec {
libc::timespec {
tv_sec: ft.seconds() as libc::time_t,
#[cfg(all(target_arch = "x86_64", target_pointer_width = "32"))]
tv_nsec: (ft.nanoseconds()) as i64,
#[cfg(not(all(target_arch = "x86_64", target_pointer_width = "32")))]
tv_nsec: (ft.nanoseconds()) as libc::c_long,
}
}