Use split_init

Signed-off-by: Alex Saveau <[email protected]>

Author: SUPERCILEX

src/buffer.rs

@@ -5,17 +5,17 @@
 use core::mem::MaybeUninit;
 use core::slice;
 
-/// Split an uninitialized byte slice into initialized and uninitialized parts.
+/// Split an uninitialized slice into initialized and uninitialized parts.
 ///
 /// # Safety
 ///
-/// At least `init` bytes must be initialized.
+/// At least `init` items must be initialized.
 #[inline]
-pub(super) unsafe fn split_init(
-    buf: &mut [MaybeUninit<u8>],
+pub(super) unsafe fn split_init<T>(
+    buf: &mut [MaybeUninit<T>],
     init: usize,
-) -> (&mut [u8], &mut [MaybeUninit<u8>]) {
+) -> (&mut [T], &mut [MaybeUninit<T>]) {
     let (init, uninit) = buf.split_at_mut(init);
-    let init = slice::from_raw_parts_mut(init.as_mut_ptr() as *mut u8, init.len());
+    let init = slice::from_raw_parts_mut(init.as_mut_ptr().cast::<T>(), init.len());
     (init, uninit)
 }

src/event/epoll.rs

@@ -437,6 +437,7 @@ fn test_epoll_layouts() {
 
 mod buf {
     use super::Event;
+    use crate::buffer::split_init;
     #[cfg(feature = "alloc")]
     use alloc::vec::Vec;
     use core::mem::MaybeUninit;
@@ -501,8 +502,7 @@ mod buf {
         }
 
         unsafe fn filled(self, count: usize) -> Self::Out {
-            let filled = &mut self[..count];
-            unsafe { core::mem::transmute::<&mut [MaybeUninit<Event>], &mut [Event]>(filled) }
+            unsafe { split_init(self, count) }.0
         }
     }
 }