--- source-old/src/header/sys_timerfd/mod.rs 2026-05-03 20:55:05.750445686 +0100 +++ source-old/src/header/sys_timerfd/mod.rs 2026-04-29 20:43:18.559857138 +0100 @@ -0,0 +1,135 @@ +//! `sys/timerfd.h` implementation. +//! +//! Non-POSIX, see . + +use alloc::{collections::BTreeMap, format}; +use core::{mem, slice, sync::atomic::{AtomicBool, Ordering}}; + +use crate::{ + c_str::{CStr, CString}, + error::{Errno, ResultExt}, + header::{ + bits_timespec::timespec, + errno::{EBADF, EFAULT, EINVAL, EIO}, + fcntl::{O_CLOEXEC, O_NONBLOCK, O_RDWR}, + }, + out::Out, + platform::{ + Pal, Sys, + types::{c_int, clockid_t}, + }, + sync::Mutex, +}; + +pub use crate::header::time::itimerspec; + +pub const TFD_CLOEXEC: c_int = 0x80000; +pub const TFD_NONBLOCK: c_int = 0x800; +pub const TFD_TIMER_ABSTIME: c_int = 0x1; +pub const TFD_TIMER_CANCEL_ON_SET: c_int = 0x2; + +const NSEC_PER_SEC: i64 = 1_000_000_000; + +static MAP_INIT: AtomicBool = AtomicBool::new(false); +static TIMERFD_CLOCKIDS: Mutex>> = Mutex::new(None); + +fn with_map(f: impl FnOnce(&mut BTreeMap) -> R) -> R { + let mut guard = TIMERFD_CLOCKIDS.lock(); + if !MAP_INIT.load(Ordering::Acquire) { + *guard = Some(BTreeMap::new()); + MAP_INIT.store(true, Ordering::Release); + } + f(guard.as_mut().unwrap()) +} + +fn add_timespec(a: ×pec, b: ×pec) -> timespec { + let total_nsec = a.tv_nsec as i64 + b.tv_nsec as i64; + let mut sec = a.tv_sec + b.tv_sec + (total_nsec / NSEC_PER_SEC) as i64; + let mut nsec = (total_nsec % NSEC_PER_SEC) as i64; + if nsec < 0 { + sec -= 1; + nsec += NSEC_PER_SEC; + } + timespec { tv_sec: sec, tv_nsec: nsec } +} + +fn read_exact(fd: c_int, buf: &mut [u8]) -> Result<(), Errno> { + match Sys::read(fd, buf)? { + n if n == buf.len() => Ok(()), + _ => Err(Errno(EIO)), + } +} + +fn write_exact(fd: c_int, buf: &[u8]) -> Result<(), Errno> { + match Sys::write(fd, buf)? { + n if n == buf.len() => Ok(()), + _ => Err(Errno(EIO)), + } +} + +#[unsafe(no_mangle)] +pub extern "C" fn timerfd_create(clockid: clockid_t, flags: c_int) -> c_int { + let supported = TFD_CLOEXEC | TFD_NONBLOCK; + if flags & !supported != 0 { + return Err::(Errno(EINVAL)).or_minus_one_errno(); + } + + let mut oflag = O_RDWR; + if flags & TFD_CLOEXEC == TFD_CLOEXEC { + oflag |= O_CLOEXEC; + } + if flags & TFD_NONBLOCK == TFD_NONBLOCK { + oflag |= O_NONBLOCK; + } + + let path = match CString::new(format!("/scheme/time/{clockid}")) { + Ok(path) => path, + Err(_) => return Err::(Errno(EINVAL)).or_minus_one_errno(), + }; + let fd = Sys::open(CStr::borrow(&path), oflag, 0).or_minus_one_errno()?; + + // Register the clockid for this fd so timerfd_settime can convert relative times + with_map(|map| { map.insert(fd, clockid); }); + + fd +} + +#[unsafe(no_mangle)] +pub unsafe extern "C" fn timerfd_settime(fd: c_int, flags: c_int, new: *const itimerspec, old: *mut itimerspec) -> c_int { + let supported = TFD_TIMER_ABSTIME | TFD_TIMER_CANCEL_ON_SET; + if flags & !supported != 0 { + return Err::(Errno(EINVAL)).or_minus_one_errno(); + } + if new.is_null() { + return Err::(Errno(EFAULT)).or_minus_one_errno(); + } + if !old.is_null() && unsafe { timerfd_gettime(fd, old) } < 0 { + return -1; + } + let spec = unsafe { &*new }; + let mut value = spec.it_value; + + // Convert relative time to absolute if TFD_TIMER_ABSTIME is not set + if flags & TFD_TIMER_ABSTIME == 0 { + let clockid = with_map(|map| map.get(&fd).copied()).unwrap_or(0); + let mut now = timespec::default(); + if Sys::clock_gettime(clockid, Out::from_mut(&mut now)).is_err() { + return Err::(Errno(EBADF)).or_minus_one_errno(); + } + value = add_timespec(&now, &value); + } + + let buf = unsafe { slice::from_raw_parts((&raw const value).cast::(), mem::size_of::()) }; + write_exact(fd, buf).map(|()| 0).or_minus_one_errno() +} + +#[unsafe(no_mangle)] +pub unsafe extern "C" fn timerfd_gettime(fd: c_int, curr: *mut itimerspec) -> c_int { + if curr.is_null() { + return Err::(Errno(EFAULT)).or_minus_one_errno(); + } + let curr = unsafe { &mut *curr }; + curr.it_interval = timespec::default(); + let buf = unsafe { slice::from_raw_parts_mut((&raw mut curr.it_value).cast::(), mem::size_of::()) }; + read_exact(fd, buf).map(|()| 0).or_minus_one_errno() +}