zisk-core 1.3.0-alpha

Core types and proving primitives for the ZisK zkVM
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use sha2::compress256;

#[allow(deprecated)]
use sha2::digest::generic_array::{typenum::U64, GenericArray};

use zisk_precomp_helpers::{blake2b_round, blake2s_f, blake3_f};

#[allow(deprecated)]
pub fn sha256f(state: &mut [u64; 4], input: &[u64; 8]) {
    let state_u32: &mut [u32; 8] = unsafe { &mut *(state.as_mut_ptr() as *mut [u32; 8]) };
    let input_u8: &[GenericArray<u8, U64>; 1] =
        unsafe { &*(input.as_ptr() as *const [GenericArray<u8, U64>; 1]) };
    compress256(state_u32, input_u8);
}

#[allow(deprecated)]
pub fn blake2br(index: u64, state: &mut [u64; 16], input: &[u64; 16]) {
    blake2b_round(state, input, index as u32);
}

/// Blake3 permutation (7 rounds of G-mixing, no feed-forward) over the 16-u32 state,
/// viewing each u64 word as two little-endian u32 words.
pub fn blake3f(state: &mut [u64; 8], input: &[u64; 8]) {
    let state_u32: &mut [u32; 16] = unsafe { &mut *(state.as_mut_ptr() as *mut [u32; 16]) };
    let input_u32: &[u32; 16] = unsafe { &*(input.as_ptr() as *const [u32; 16]) };
    blake3_f(state_u32, input_u32);
}

/// Blake2s permutation (the 10 rounds of G-mixing, no feed-forward) over the 16-u32 state,
/// viewing each u64 word as two little-endian u32 words.
pub fn blake2sf(state: &mut [u64; 8], input: &[u64; 8]) {
    let state_u32: &mut [u32; 16] = unsafe { &mut *(state.as_mut_ptr() as *mut [u32; 16]) };
    let input_u32: &[u32; 16] = unsafe { &*(input.as_ptr() as *const [u32; 16]) };
    blake2s_f(state_u32, input_u32);
}