use crate::{FromInner, Inner, IntoInner};
use uv::uv_udp_send_t;
callbacks! {
pub UdpSendCB(req: UdpSendReq, status: crate::Result<u32>);
}
pub(crate) struct UdpSendDataFields<'a> {
bufs_ptr: *mut uv::uv_buf_t,
bufs_len: usize,
bufs_capacity: usize,
udp_send_cb: UdpSendCB<'a>,
}
pub(crate) extern "C" fn uv_udp_send_cb(req: *mut uv_udp_send_t, status: std::os::raw::c_int) {
let dataptr = crate::Req::get_data(uv_handle!(req));
if !dataptr.is_null() {
unsafe {
if let super::UdpSendData(d) = &mut *dataptr {
let status = if status < 0 {
Err(crate::Error::from_inner(status as uv::uv_errno_t))
} else {
Ok(status as _)
};
d.udp_send_cb.call(req.into_inner(), status);
}
}
}
let mut req = UdpSendReq::from_inner(req);
req.destroy();
}
#[derive(Clone, Copy)]
pub struct UdpSendReq {
req: *mut uv_udp_send_t,
pub(crate) bufs_ptr: *const uv::uv_buf_t,
}
impl UdpSendReq {
pub fn new<CB: Into<UdpSendCB<'static>>>(
bufs: &[impl crate::BufTrait],
cb: CB,
) -> crate::Result<UdpSendReq> {
let layout = std::alloc::Layout::new::<uv_udp_send_t>();
let req = unsafe { std::alloc::alloc(layout) as *mut uv_udp_send_t };
if req.is_null() {
return Err(crate::Error::ENOMEM);
}
let (bufs_ptr, bufs_len, bufs_capacity) = bufs.into_inner();
let udp_send_cb = cb.into();
crate::Req::initialize_data(
uv_handle!(req),
super::UdpSendData(UdpSendDataFields {
bufs_ptr,
bufs_len,
bufs_capacity,
udp_send_cb,
}),
);
Ok(UdpSendReq { req, bufs_ptr })
}
pub fn handle(&self) -> crate::UdpHandle {
unsafe { (*self.req).handle }.into_inner()
}
pub fn destroy(&mut self) {
let dataptr = crate::Req::get_data(uv_handle!(self.req));
if !dataptr.is_null() {
if let super::UdpSendData(d) = unsafe { &mut *dataptr } {
if !d.bufs_ptr.is_null() {
unsafe {
std::mem::drop(Vec::from_raw_parts(d.bufs_ptr, d.bufs_len, d.bufs_capacity))
};
}
}
}
crate::Req::free_data(uv_handle!(self.req));
let layout = std::alloc::Layout::new::<uv_udp_send_t>();
unsafe { std::alloc::dealloc(self.req as _, layout) }
}
}
impl FromInner<*mut uv_udp_send_t> for UdpSendReq {
fn from_inner(req: *mut uv_udp_send_t) -> UdpSendReq {
UdpSendReq {
req,
bufs_ptr: std::ptr::null(),
}
}
}
impl Inner<*mut uv_udp_send_t> for UdpSendReq {
fn inner(&self) -> *mut uv_udp_send_t {
self.req
}
}
impl Inner<*mut uv::uv_req_t> for UdpSendReq {
fn inner(&self) -> *mut uv::uv_req_t {
uv_handle!(self.req)
}
}
impl From<UdpSendReq> for crate::Req {
fn from(udp_send: UdpSendReq) -> crate::Req {
crate::Req::from_inner(Inner::<*mut uv::uv_req_t>::inner(&udp_send))
}
}
impl crate::ToReq for UdpSendReq {
fn to_req(&self) -> crate::Req {
crate::Req::from_inner(Inner::<*mut uv::uv_req_t>::inner(self))
}
}
impl crate::ReqTrait for UdpSendReq {}