libuv/requests/
connect.rs

1use crate::{FromInner, Inner, IntoInner};
2use uv::uv_connect_t;
3
4callbacks! {
5    pub ConnectCB(req: ConnectReq, status: crate::Result<u32>);
6}
7
8/// Additional data stored on the request
9pub(crate) struct ConnectDataFields<'a> {
10    connect_cb: ConnectCB<'a>,
11}
12
13/// Callback for uv_tcp_connect
14pub(crate) extern "C" fn uv_connect_cb(req: *mut uv_connect_t, status: std::os::raw::c_int) {
15    let dataptr = crate::Req::get_data(uv_handle!(req));
16    if !dataptr.is_null() {
17        unsafe {
18            if let super::ConnectData(d) = &mut *dataptr {
19                let status = if status < 0 {
20                    Err(crate::Error::from_inner(status as uv::uv_errno_t))
21                } else {
22                    Ok(status as _)
23                };
24                d.connect_cb.call(req.into_inner(), status);
25            }
26        }
27    }
28
29    // free memory
30    let mut req = ConnectReq::from_inner(req);
31    req.destroy();
32}
33
34/// Connect request type
35#[derive(Clone, Copy)]
36pub struct ConnectReq {
37    req: *mut uv_connect_t,
38}
39
40impl ConnectReq {
41    /// Create a new connect request
42    pub fn new<CB: Into<ConnectCB<'static>>>(cb: CB) -> crate::Result<ConnectReq> {
43        let layout = std::alloc::Layout::new::<uv_connect_t>();
44        let req = unsafe { std::alloc::alloc(layout) as *mut uv_connect_t };
45        if req.is_null() {
46            return Err(crate::Error::ENOMEM);
47        }
48
49        let connect_cb = cb.into();
50        crate::Req::initialize_data(
51            uv_handle!(req),
52            super::ConnectData(ConnectDataFields { connect_cb }),
53        );
54
55        Ok(ConnectReq { req })
56    }
57
58    /// The stream where this connection request is running
59    pub fn handle(&self) -> crate::StreamHandle {
60        unsafe { (*self.req).handle }.into_inner()
61    }
62
63    pub fn destroy(&mut self) {
64        crate::Req::free_data(uv_handle!(self.req));
65
66        let layout = std::alloc::Layout::new::<uv_connect_t>();
67        unsafe { std::alloc::dealloc(self.req as _, layout) };
68    }
69}
70
71impl FromInner<*mut uv_connect_t> for ConnectReq {
72    fn from_inner(req: *mut uv_connect_t) -> ConnectReq {
73        ConnectReq { req }
74    }
75}
76
77impl Inner<*mut uv_connect_t> for ConnectReq {
78    fn inner(&self) -> *mut uv_connect_t {
79        self.req
80    }
81}
82
83impl Inner<*mut uv::uv_req_t> for ConnectReq {
84    fn inner(&self) -> *mut uv::uv_req_t {
85        uv_handle!(self.req)
86    }
87}
88
89impl From<ConnectReq> for crate::Req {
90    fn from(connect: ConnectReq) -> crate::Req {
91        crate::Req::from_inner(Inner::<*mut uv::uv_req_t>::inner(&connect))
92    }
93}
94
95impl crate::ToReq for ConnectReq {
96    fn to_req(&self) -> crate::Req {
97        crate::Req::from_inner(Inner::<*mut uv::uv_req_t>::inner(self))
98    }
99}
100
101impl crate::ReqTrait for ConnectReq {}
libuv/requests/connect.rs

libuv/requests/
connect.rs
