feat: patch

sha2/src/sha256.rs

@@ -1,7 +1,10 @@
 use digest::{generic_array::GenericArray, typenum::U64};
 
 cfg_if::cfg_if! {
-    if #[cfg(feature = "force-soft")] {
+    if #[cfg(all(target_os = "zkvm", target_vendor = "succinct", target_arch = "riscv32"))] {
+        mod succinct;
+        use succinct::compress;
+    } else if #[cfg(feature = "force-soft")] {
         mod soft;
         use soft::compress;
     } else if #[cfg(any(target_arch = "x86", target_arch = "x86_64"))] {

sha2/src/sha256/succinct.rs

@@ -0,0 +1,23 @@
+extern "C" {
+    fn syscall_sha256_extend(w: *mut u32);
+    fn syscall_sha256_compress(w: *mut u32, state: *mut u32);
+}
+
+#[inline]
+pub fn compress(state: &mut [u32; 8], blocks: &[[u8; 64]]) {
+    unsafe {
+        for i in 0..blocks.len() {
+            let mut w = [0u32; 64];
+            for j in 0..16 {
+                w[j] = u32::from_be_bytes([
+                    blocks[i][j * 4],
+                    blocks[i][j * 4 + 1],
+                    blocks[i][j * 4 + 2],
+                    blocks[i][j * 4 + 3],
+                ]);
+            }
+            syscall_sha256_extend(w.as_mut_ptr());
+            syscall_sha256_compress(w.as_mut_ptr(), state.as_mut_ptr());
+        }
+    }
+}