update to nix 0.24

[pve-lxc-syscalld.git] / src / client.rs

use std::sync::Arc;

use anyhow::Error;
use nix::errno::Errno;

use crate::io::seq_packet::SeqPacketSocket;
use crate::lxcseccomp::ProxyMessageBuffer;
use crate::syscall::{self, Syscall, SyscallStatus};

pub struct Client {
    socket: SeqPacketSocket,
}

impl Client {
    pub fn new(socket: SeqPacketSocket) -> Arc<Self> {
        Arc::new(Self { socket })
    }

    /// Wrap futures returning a `Result` so if they fail we `shutdown()` the socket to drop the
    /// client.
    async fn wrap_error<F>(self: Arc<Self>, fut: F)
    where
        F: std::future::Future<Output = Result<(), Error>>,
    {
        if let Err(err) = fut.await {
            eprintln!("client error, dropping connection: {}", err);
            if let Err(err) = self.socket.shutdown(nix::sys::socket::Shutdown::Both) {
                eprintln!("    (error shutting down client socket: {})", err);
            }
        }
    }

    pub async fn main(self: Arc<Self>) {
        self.clone().wrap_error(self.main_do()).await
    }

    async fn main_do(self: Arc<Self>) -> Result<(), Error> {
        let mut msg = ProxyMessageBuffer::new(64);
        loop {
            if !msg.recv(&self.socket).await? {
                break Ok(());
            }

            self.handle_syscall(&mut msg).await?;
        }
    }

    async fn handle_syscall(&self, msg: &mut ProxyMessageBuffer) -> Result<(), Error> {
        let result = match Self::handle_syscall_do(msg).await {
            Ok(r) => r,
            Err(err) => {
                // handle the various kinds of errors we may get:
                if let Some(errno) = err.downcast_ref::<nix::Error>() {
                    SyscallStatus::Err(*errno as _)
                } else if let Some(ioerr) = err.downcast_ref::<std::io::Error>() {
                    if let Some(errno) = ioerr.raw_os_error() {
                        SyscallStatus::Err(errno)
                    } else {
                        return Err(err);
                    }
                } else {
                    return Err(err);
                }
            }
        };

        let resp = msg.response_mut();
        match result {
            SyscallStatus::Ok(val) => {
                resp.val = val;
                resp.error = 0;
            }
            SyscallStatus::Err(err) => {
                resp.val = -1;
                resp.error = -err;
            }
        }

        msg.respond(&self.socket).await.map_err(Error::from)
    }

    async fn handle_syscall_do(msg: &ProxyMessageBuffer) -> Result<SyscallStatus, Error> {
        let (arch, sysnr) = (msg.request().data.arch, msg.request().data.nr);

        let syscall_nr = match syscall::translate_syscall(arch, sysnr) {
            Some(nr) => nr,
            None => return Ok(Errno::ENOSYS.into()),
        };

        match syscall_nr {
            Syscall::Mknod => crate::sys_mknod::mknod(msg).await,
            Syscall::MknodAt => crate::sys_mknod::mknodat(msg).await,
            Syscall::Quotactl => crate::sys_quotactl::quotactl(msg).await,
        }
    }
}