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

use std::os::unix::io::AsRawFd;
use std::mem::ManuallyDrop;
use std::marker::Unpin;

use super::{Event, Cancellation};

/// A basic write event.
pub struct Write<'a, T> {
    pub io: &'a mut T,
    pub buf: Vec<u8>,
}

impl<'a, T: AsRawFd + Unpin> Write<'a, T> {
    pub fn new(io: &'a mut T, buf: Vec<u8>) -> Write<T> {
        Write { io, buf }
    }
}

impl<'a, T: AsRawFd + Unpin> Event for Write<'a, T> {
    unsafe fn prepare(&mut self, sqe: &mut iou::SubmissionQueueEvent<'_>) {
        sqe.prep_write(self.io.as_raw_fd(), self.buf.as_ref(), 0);
    }

    fn cancellation(this: &mut ManuallyDrop<Self>) -> Cancellation {
        unsafe {
            let mut buf: ManuallyDrop<Vec<u8>> = ManuallyDrop::new(ManuallyDrop::take(this).buf);
            let cap = buf.capacity();
            Cancellation::buffer(buf.as_mut_ptr(), cap)
        }
    }
}