zincio 0.2.22

A high-performance, cross-platform asynchronous runtime for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use std::io;
use std::os::fd::{AsRawFd, BorrowedFd};
use std::task::{Context, Poll};

use mio::Interest;

use crate::driver::AnyDriver;
use crate::fd_inner::InnerRawHandle;
use crate::op::io_util::poll_result_or_wait;
use crate::op::Op;

pub struct SendfileOp<'a> {
    fd_in: BorrowedFd<'a>,
    fd_out: &'a InnerRawHandle,
    len: usize,
    completion_token: Option<usize>,
}

impl<'a> SendfileOp<'a> {
    #[inline]
    pub fn new(fd_in: BorrowedFd<'a>, fd_out: &'a InnerRawHandle, len: usize) -> Self {
        Self {
            fd_in,
            fd_out,
            len,
            completion_token: None,
        }
    }
}

impl Op for SendfileOp<'_> {
    type Output = usize;

    #[inline]
    fn poll_poll(
        &mut self,
        cx: &mut Context<'_>,
        driver: &AnyDriver,
    ) -> Poll<io::Result<Self::Output>> {
        let result = {
            #[cfg(target_os = "linux")]
            let returned = unsafe {
                libc::sendfile(
                    self.fd_out.handle,
                    self.fd_in.as_raw_fd(),
                    std::ptr::null_mut(),
                    self.len,
                )
            };
            #[cfg(target_os = "freebsd")]
            let returned = {
                let mut sbytes = 0;
                // On FreeBSD, sendfile returns 0 on success, and bytes sent inside `sbytes`,
                // unlike on Linux, where it returns the number of bytes sent directly.
                let result = unsafe {
                    libc::sendfile(
                        self.fd_in.as_raw_fd(),
                        self.fd_out.handle,
                        0,
                        self.len,
                        std::ptr::null_mut(),
                        &mut sbytes,
                        0,
                    )
                };
                if result < 0 {
                    result as i64
                } else {
                    sbytes
                }
            };

            if returned == -1 {
                Err(io::Error::last_os_error())
            } else {
                Ok(returned as usize)
            }
        };

        poll_result_or_wait(result, self.fd_out, cx, driver, Interest::WRITABLE)
    }
}

impl Drop for SendfileOp<'_> {
    #[inline]
    fn drop(&mut self) {
        if let Some(completion_token) = self.completion_token {
            if let Some(driver) = crate::try_current_driver() {
                driver.ignore_completion(completion_token, Box::new(()));
            }
        }
    }
}