Write slice in single command

libp2p · Nov 14, 2023 · 81ef6cc · 81ef6cc
1 parent df9f485
commit 81ef6cc
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/misc/quick-protobuf-codec/src/lib.rs b/misc/quick-protobuf-codec/src/lib.rs
@@ -182,10 +182,14 @@ impl<'a> WriterBackend for MaybeUninitWriterBackend<'a> {
             return Err(quick_protobuf::errors::Error::UnexpectedEndOfBuffer);
         }
 
-        for b in buf {
-            self.memory[*self.written].write(*b);
-            *self.written += 1;
-        }
+        // SAFETY: &[u8] and &[MaybeUninit<u8>] have the same layout
+        let uninit_src: &[MaybeUninit<u8>] = unsafe { std::mem::transmute(buf) };
+
+        let start = *self.written;
+        let end = *self.written + buf.len();
+
+        self.memory[start..end].copy_from_slice(uninit_src);
+        *self.written += buf.len();
 
         Ok(())
     }