added neon safety comment

RustCrypto · Dec 17, 2023 · 29d8bc0 · 29d8bc0
1 parent b4a408c
commit 29d8bc0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/chacha20/src/backends/avx2.rs b/chacha20/src/backends/avx2.rs
@@ -115,7 +115,7 @@ impl<R: Rounds> StreamBackend for Backend<R> {
 
     #[inline(always)]
     fn gen_par_ks_blocks(&mut self, blocks: &mut ParBlocks<Self>) {
-        // SAFETY: ParBlocks is a 256-byte 2D array
+        // SAFETY: `ParBlocks` is a 256-byte 2D array.
         unsafe {
             self.write_par_ks_blocks(blocks.as_mut_ptr() as *mut u8);
         }

diff --git a/chacha20/src/backends/neon.rs b/chacha20/src/backends/neon.rs
@@ -102,7 +102,10 @@ impl<R: Rounds> StreamBackend for Backend<R> {
 
     #[inline(always)]
     fn gen_par_ks_blocks(&mut self, blocks: &mut ParBlocks<Self>) {
-        self.write_par_ks_blocks(blocks.as_mut_ptr() as *mut u8);
+        // SAFETY: `ParBlocks` is a 256-byte 2D array.
+        unsafe {
+            self.write_par_ks_blocks(blocks.as_mut_ptr() as *mut u8);
+        }
     }
 }