use OwnedFd for all remaining custom file descriptors

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

// c_str!() from the byte-strings crate is implemented via a proc macro which seems a bit excessive
macro_rules! c_str {
    ($data:expr) => {{
        #![allow(unused_unsafe)]
        let bytes = concat!($data, "\0");
        unsafe { std::ffi::CStr::from_bytes_with_nul_unchecked(bytes.as_bytes()) }
    }};
}

macro_rules! file_descriptor_type {
    ($type:ident) => {
        #[repr(transparent)]
        pub struct $type(::std::os::unix::io::OwnedFd);

        file_descriptor_impl!($type);

        impl ::std::os::unix::io::FromRawFd for $type {
            unsafe fn from_raw_fd(fd: ::std::os::unix::io::RawFd) -> Self {
                Self(unsafe { ::std::os::unix::io::FromRawFd::from_raw_fd(fd) })
            }
        }
    };
}

macro_rules! file_descriptor_impl {
    ($type:ty) => {
        impl ::std::os::unix::io::AsFd for $type {
            fn as_fd(&self) -> ::std::os::unix::io::BorrowedFd<'_> {
                ::std::os::unix::io::AsFd::as_fd(&self.0)
            }
        }

        impl ::std::os::unix::io::AsRawFd for $type {
            fn as_raw_fd(&self) -> ::std::os::unix::io::RawFd {
                ::std::os::unix::io::AsRawFd::as_raw_fd(&self.0)
            }
        }

        impl ::std::os::unix::io::IntoRawFd for $type {
            fn into_raw_fd(self) -> ::std::os::unix::io::RawFd {
                ::std::os::unix::io::IntoRawFd::into_raw_fd(self.0)
            }
        }
    };
}

macro_rules! c_result {
    ($expr:expr) => {{
        let res = $expr;
        if res == -1 {
            Err(::std::io::Error::last_os_error())
        } else {
            Ok::<_, ::std::io::Error>(res)
        }
    }};
}

macro_rules! c_try {
    ($expr:expr) => {
        c_result!($expr)?
    };
}

macro_rules! io_format_err {
    ($($msg:tt)*) => {
        ::std::io::Error::new(::std::io::ErrorKind::Other, format!($($msg)*))
    };
}

macro_rules! io_bail {
    ($($msg:tt)*) => {
        return Err(::std::io::Error::new(::std::io::ErrorKind::Other, format!($($msg)*)));
    };
}