starry-kernel 0.10.1

use alloc::{borrow::Cow, sync::Arc};
use core::{
    mem,
    sync::atomic::{AtomicBool, Ordering},
};

use axpoll::{IoEvents, Pollable};
use axpoll_set::PollSet;
use starry_signal::{SignalInfo, SignalSet};
use zerocopy::{Immutable, IntoBytes};

use crate::{
    StarryError, StarryResult,
    file::{FileLike, IoDst, IoSrc},
    sync::IrqMutex,
    task::{
        current_user_task,
        future::{block_on_user, poll_io},
    },
};

/// The size of signalfd_siginfo structure (128 bytes as per Linux
/// specification)
const SIGNALFD_SIGINFO_SIZE: usize = 128;

/// signalfd_siginfo structure layout
/// This matches the Linux signalfd_siginfo structure (128 bytes)
#[repr(C)]
#[derive(Immutable, IntoBytes)]
struct SignalfdSiginfo {
    ssi_signo: u32,    // Signal number
    ssi_errno: i32,    // Error number (unused)
    ssi_code: i32,     // Signal code
    ssi_pid: u32,      // PID of sender
    ssi_uid: u32,      // Real UID of sender
    ssi_fd: i32,       // File descriptor (SIGIO)
    ssi_tid: u32,      // Kernel timer ID (POSIX timers)
    ssi_band: u32,     // Band event (SIGIO)
    ssi_overrun: u32,  // POSIX timer overrun count
    ssi_trapno: u32,   // Trap number that caused signal
    ssi_status: i32,   // Exit status or signal (SIGCHLD)
    ssi_int: i32,      // Integer sent by sigqueue(2)
    ssi_ptr: u64,      // Pointer sent by sigqueue(2)
    ssi_utime: u64,    // User CPU time consumed (SIGCHLD)
    ssi_stime: u64,    // System CPU time consumed (SIGCHLD)
    ssi_addr: u64,     // Address that generated signal
    ssi_addr_lsb: u16, // Least significant bit of address
    _pad: [u8; 46],    // Padding to make it 128 bytes
}

const _: [(); SIGNALFD_SIGINFO_SIZE] = [(); mem::size_of::<SignalfdSiginfo>()];

impl SignalfdSiginfo {
    /// Convert from SignalInfo to signalfd_siginfo
    fn from_signal_info(sig_info: &SignalInfo) -> Self {
        let errno = sig_info.errno();

        SignalfdSiginfo {
            ssi_signo: sig_info.signo() as u32,
            ssi_errno: errno,
            ssi_code: sig_info.code(),
            ssi_pid: sig_info.pid(),
            ssi_uid: sig_info.uid(),
            ssi_fd: -1,
            ssi_tid: 0,
            ssi_band: 0,
            ssi_overrun: sig_info.timer_overrun(),
            ssi_trapno: 0,
            ssi_status: 0,
            ssi_int: 0,
            ssi_ptr: 0,
            ssi_utime: 0,
            ssi_stime: 0,
            ssi_addr: 0,
            ssi_addr_lsb: 0,
            _pad: [0u8; 46],
        }
    }
}

pub struct Signalfd {
    // SignalSet is a single Copy bitset, so a short project-visible spin lock
    // is enough for now. Revisit this when a lockdep-aware project RwLock exists.
    mask: IrqMutex<SignalSet>,
    non_blocking: AtomicBool,
    poll_rx: PollSet,
}

impl Signalfd {
    pub fn new(mask: SignalSet) -> Arc<Self> {
        Arc::new(Self {
            mask: IrqMutex::new(mask),
            non_blocking: AtomicBool::new(false),
            poll_rx: PollSet::new(),
        })
    }

    pub fn update_mask(&self, mask: SignalSet) {
        {
            *self.mask.lock() = mask;
        }
        // The signal mask update is visible before waking readers.
        unsafe { self.poll_rx.wake(IoEvents::IN) };
    }

    fn mask(&self) -> SignalSet {
        *self.mask.lock()
    }

    /// Check if there are any pending signals matching the mask
    fn has_pending_signals(&self) -> bool {
        let mask = self.mask();
        let curr = current_user_task();
        let signal = curr.as_thread().signal();
        let pending = signal.pending();
        !(pending & mask).is_empty()
    }

    /// Dequeue a signal matching the mask
    fn dequeue_signal(&self) -> Option<SignalInfo> {
        let mask = self.mask();
        let curr = current_user_task();
        let signal = curr.as_thread().signal();
        signal.dequeue_signal(&mask)
    }
}

impl FileLike for Signalfd {
    fn read(&self, dst: &mut IoDst) -> StarryResult<usize> {
        if dst.remaining_mut() < SIGNALFD_SIGINFO_SIZE {
            return Err(StarryError::InvalidInput);
        }

        let task = current_user_task();
        block_on_user(
            &task,
            poll_io(self, IoEvents::IN, self.nonblocking(), || {
                if let Some(sig_info) = self.dequeue_signal() {
                    // Convert SignalInfo to SignalfdSiginfo
                    let sfd_info = SignalfdSiginfo::from_signal_info(&sig_info);

                    // Write the structure to the destination buffer
                    let bytes = sfd_info.as_bytes();
                    dst.write(bytes)?;

                    // Wake up other waiters if there are more signals pending
                    if self.has_pending_signals() {
                        // Remaining pending signals are visible before re-wake.
                        unsafe { self.poll_rx.wake(IoEvents::IN) };
                    }

                    Ok(SIGNALFD_SIGINFO_SIZE)
                } else {
                    Err(crate::StarryError::WouldBlock)
                }
            }),
        )
        .into_result()?
    }

    fn write(&self, _src: &mut IoSrc) -> StarryResult<usize> {
        // signalfd is read-only
        Err(StarryError::BadFileDescriptor)
    }

    fn nonblocking(&self) -> bool {
        self.non_blocking.load(Ordering::Acquire)
    }

    fn set_nonblocking(&self, non_blocking: bool) -> StarryResult {
        self.non_blocking.store(non_blocking, Ordering::Release);
        Ok(())
    }

    fn path(&self) -> Cow<'_, str> {
        "anon_inode:[signalfd]".into()
    }
}

impl Pollable for Signalfd {
    fn poll(&self) -> IoEvents {
        let mut events = IoEvents::empty();
        events.set(IoEvents::IN, self.has_pending_signals());
        events
    }

    unsafe fn register_shared(
        &self,
        sink: &mut dyn axpoll::SharedRegistrationSink,
        events: IoEvents,
    ) {
        if events.contains(IoEvents::IN) {
            // The private poll set covers mask updates and additional queued
            // signals. New signal delivery wakes the current thread's shared
            // signalfd poll set, so an already-blocked epoll waiter must be
            // registered with both sources.
            unsafe {
                sink.register_shared(&self.poll_rx, IoEvents::IN);
                sink.register_shared(
                    current_user_task().as_thread().signalfd_poll_source(),
                    IoEvents::IN,
                );
            }
        }
    }

    unsafe fn register_exclusive(
        &self,
        sink: &mut dyn axpoll::ExclusiveRegistrationSink,
        events: IoEvents,
    ) {
        if events.contains(IoEvents::IN) {
            unsafe {
                sink.register_exclusive(&self.poll_rx, IoEvents::IN);
                sink.register_exclusive(
                    current_user_task().as_thread().signalfd_poll_source(),
                    IoEvents::IN,
                );
            }
        }
    }
}