fix(driver,unix): mark set_msg as unsafe

Author: AsakuraMizu

compio-driver/src/iour/op.rs

@@ -559,7 +559,7 @@ impl<T: IoVectoredBuf, S> IntoInner for SendToVectored<T, S> {
 impl<T: IoVectoredBufMut, C: IoBufMut, S: AsRawFd> OpCode for RecvMsg<T, C, S> {
     fn create_entry(self: Pin<&mut Self>) -> OpEntry {
         let this = unsafe { self.get_unchecked_mut() };
-        this.set_msg();
+        unsafe { this.set_msg() };
         opcode::RecvMsg::new(Fd(this.fd.as_raw_fd()), &mut this.msg)
             .build()
             .into()
@@ -569,7 +569,7 @@ impl<T: IoVectoredBufMut, C: IoBufMut, S: AsRawFd> OpCode for RecvMsg<T, C, S> {
 impl<T: IoVectoredBuf, C: IoBuf, S: AsRawFd> OpCode for SendMsg<T, C, S> {
     fn create_entry(self: Pin<&mut Self>) -> OpEntry {
         let this = unsafe { self.get_unchecked_mut() };
-        this.set_msg();
+        unsafe { this.set_msg() };
         opcode::SendMsg::new(Fd(this.fd.as_raw_fd()), &this.msg)
             .build()
             .into()

compio-driver/src/poll/op.rs

@@ -758,7 +758,7 @@ impl<T: IoVectoredBufMut, C: IoBufMut, S: AsRawFd> RecvMsg<T, C, S> {
 impl<T: IoVectoredBufMut, C: IoBufMut, S: AsRawFd> OpCode for RecvMsg<T, C, S> {
     fn pre_submit(self: Pin<&mut Self>) -> io::Result<Decision> {
         let this = unsafe { self.get_unchecked_mut() };
-        this.set_msg();
+        unsafe { this.set_msg() };
         syscall!(this.call(), wait_readable(this.fd.as_raw_fd()))
     }
 
@@ -779,7 +779,7 @@ impl<T: IoVectoredBuf, C: IoBuf, S: AsRawFd> SendMsg<T, C, S> {
 impl<T: IoVectoredBuf, C: IoBuf, S: AsRawFd> OpCode for SendMsg<T, C, S> {
     fn pre_submit(self: Pin<&mut Self>) -> io::Result<Decision> {
         let this = unsafe { self.get_unchecked_mut() };
-        this.set_msg();
+        unsafe { this.set_msg() };
         syscall!(this.call(), wait_writable(this.fd.as_raw_fd()))
     }
 

compio-driver/src/unix/op.rs

@@ -403,8 +403,8 @@ impl<T: IoVectoredBufMut, C: IoBufMut, S> RecvMsg<T, C, S> {
         }
     }
 
-    pub(crate) fn set_msg(&mut self) {
-        self.slices = unsafe { self.buffer.as_io_slices_mut() };
+    pub(crate) unsafe fn set_msg(&mut self) {
+        self.slices = self.buffer.as_io_slices_mut();
 
         self.msg.msg_name = std::ptr::addr_of_mut!(self.addr) as _;
         self.msg.msg_namelen = std::mem::size_of_val(&self.addr) as _;
@@ -457,8 +457,8 @@ impl<T: IoVectoredBuf, C: IoBuf, S> SendMsg<T, C, S> {
         }
     }
 
-    pub(crate) fn set_msg(&mut self) {
-        self.slices = unsafe { self.buffer.as_io_slices() };
+    pub(crate) unsafe fn set_msg(&mut self) {
+        self.slices = self.buffer.as_io_slices();
 
         self.msg.msg_name = self.addr.as_ptr() as _;
         self.msg.msg_namelen = self.addr.len();