jay-compositor 1.15.0

use crate::async_engine::SpawnedFuture;
use crate::env::XDG_RUNTIME_DIR;
use crate::state::State;
use crate::utils::errorfmt::ErrorFmt;
use jay_algorithms::oserror::OsError;
use jay_algorithms::oserror::OsErrorExt;
use jay_algorithms::oserror::OsErrorExt2;
use std::rc::Rc;
use thiserror::Error;
use uapi::OwnedFd;
use uapi::Ustring;
use uapi::c;
use uapi::format_ustr;

#[derive(Debug, Error)]
pub enum EiAcceptorError {
    #[error("XDG_RUNTIME_DIR is not set")]
    XrdNotSet,
    #[error("XDG_RUNTIME_DIR ({0:?}) is too long to form a unix socket address")]
    XrdTooLong(String),
    #[error("Could not create a libei socket")]
    SocketFailed(#[source] OsError),
    #[error("Could not stat the existing socket")]
    SocketStat(#[source] OsError),
    #[error("Could not start listening for incoming connections")]
    ListenFailed(#[source] OsError),
    #[error("Could not open the lock file")]
    OpenLockFile(#[source] OsError),
    #[error("Could not lock the lock file")]
    LockLockFile(#[source] OsError),
    #[error("Could not bind the socket to an address")]
    BindFailed(#[source] OsError),
    #[error("All libei addresses in the range 0..1000 are already in use")]
    AddressesInUse,
}

pub struct EiAcceptor {
    socket: EiAllocatedSocket,
}

struct EiAllocatedSocket {
    // eis-x
    name: String,
    // /run/user/1000/eis-x
    path: Ustring,
    insecure: Rc<OwnedFd>,
    // /run/user/1000/eis-x.lock
    lock_path: Ustring,
    _lock_fd: OwnedFd,
}

impl Drop for EiAllocatedSocket {
    fn drop(&mut self) {
        let _ = uapi::unlink(&self.path);
        let _ = uapi::unlink(&self.lock_path);
    }
}

fn bind_socket(
    insecure: &Rc<OwnedFd>,
    xrd: &str,
    id: u32,
) -> Result<EiAllocatedSocket, EiAcceptorError> {
    let mut addr: c::sockaddr_un = uapi::pod_zeroed();
    addr.sun_family = c::AF_UNIX as _;
    let name = format!("eis-{id}");
    let path = format_ustr!("{}/{}", xrd, name);
    let lock_path = format_ustr!("{}.lock", path.display());
    if path.len() + 1 > addr.sun_path.len() {
        return Err(EiAcceptorError::XrdTooLong(xrd.to_string()));
    }
    let lock_fd = uapi::open(&*lock_path, c::O_CREAT | c::O_CLOEXEC | c::O_RDWR, 0o644)
        .map_os_err(EiAcceptorError::OpenLockFile)?;
    uapi::flock(lock_fd.raw(), c::LOCK_EX | c::LOCK_NB)
        .map_os_err(EiAcceptorError::LockLockFile)?;
    match uapi::lstat(&path).to_os_error() {
        Ok(_) => {
            log::info!("Unlinking {}", path.display());
            let _ = uapi::unlink(&path);
        }
        Err(OsError(c::ENOENT)) => {}
        Err(e) => return Err(EiAcceptorError::SocketStat(e)),
    }
    let sun_path = uapi::as_bytes_mut(&mut addr.sun_path[..]);
    sun_path[..path.len()].copy_from_slice(path.as_bytes());
    sun_path[path.len()] = 0;
    uapi::bind(insecure.raw(), &addr).map_os_err(EiAcceptorError::BindFailed)?;
    Ok(EiAllocatedSocket {
        name,
        path,
        insecure: insecure.clone(),
        lock_path,
        _lock_fd: lock_fd,
    })
}

fn allocate_socket() -> Result<EiAllocatedSocket, EiAcceptorError> {
    let Some(xrd) = *XDG_RUNTIME_DIR else {
        return Err(EiAcceptorError::XrdNotSet);
    };
    let socket = uapi::socket(c::AF_UNIX, c::SOCK_STREAM | c::SOCK_CLOEXEC, 0)
        .map(Rc::new)
        .map_os_err(EiAcceptorError::SocketFailed)?;
    for i in 1..1000 {
        match bind_socket(&socket, &xrd, i) {
            Ok(s) => return Ok(s),
            Err(e) => {
                log::warn!("Cannot use the eis-{} socket: {}", i, ErrorFmt(e));
            }
        }
    }
    Err(EiAcceptorError::AddressesInUse)
}

impl EiAcceptor {
    pub fn spawn(
        state: &Rc<State>,
    ) -> Result<(Rc<EiAcceptor>, SpawnedFuture<()>), EiAcceptorError> {
        let socket = allocate_socket()?;
        log::info!("bound to socket {}", socket.path.display());
        uapi::listen(socket.insecure.raw(), 4096).map_os_err(EiAcceptorError::ListenFailed)?;
        let acc = Rc::new(EiAcceptor { socket });
        let future = state.eng.spawn(
            "ei accept",
            accept(acc.socket.insecure.clone(), state.clone()),
        );
        Ok((acc, future))
    }

    pub fn socket_name(&self) -> &str {
        &self.socket.name
    }
}

async fn accept(fd: Rc<OwnedFd>, state: Rc<State>) {
    loop {
        let fd = match state.ring.accept(&fd, c::SOCK_CLOEXEC).await {
            Ok(fd) => fd,
            Err(e) => {
                log::error!("Could not accept a client: {}", ErrorFmt(e));
                break;
            }
        };
        state.ei_clients.spawn(&state, fd);
    }
    state.ring.stop();
}