mirror of
https://github.com/arnaucube/ark-r1cs-std.git
synced 2026-01-12 00:41:32 +01:00
Initial commit to extract crypto-primitives to new crate
This commit is contained in:
Pratyush Mishra
650
crypto-primitives/src/prf/blake2s/constraints.rs
Normal file
650
crypto-primitives/src/prf/blake2s/constraints.rs
Normal file
@@ -0,0 +1,650 @@
|
||||
use algebra::PrimeField;
|
||||
use r1cs_core::{ConstraintSystem, SynthesisError};
|
||||
|
||||
use crate::prf::PRFGadget;
|
||||
use r1cs_std::prelude::*;
|
||||
|
||||
use std::borrow::Borrow;
|
||||
|
||||
// 2.1. Parameters
|
||||
// The following table summarizes various parameters and their ranges:
|
||||
// | BLAKE2b | BLAKE2s |
|
||||
// --------------+------------------+------------------+
|
||||
// Bits in word | w = 64 | w = 32 |
|
||||
// Rounds in F | r = 12 | r = 10 |
|
||||
// Block bytes | bb = 128 | bb = 64 |
|
||||
// Hash bytes | 1 <= nn <= 64 | 1 <= nn <= 32 |
|
||||
// Key bytes | 0 <= kk <= 64 | 0 <= kk <= 32 |
|
||||
// Input bytes | 0 <= ll < 2**128 | 0 <= ll < 2**64 |
|
||||
// --------------+------------------+------------------+
|
||||
// G Rotation | (R1, R2, R3, R4) | (R1, R2, R3, R4) |
|
||||
// constants = | (32, 24, 16, 63) | (16, 12, 8, 7) |
|
||||
// --------------+------------------+------------------+
|
||||
//
|
||||
|
||||
const R1: usize = 16;
|
||||
const R2: usize = 12;
|
||||
const R3: usize = 8;
|
||||
const R4: usize = 7;
|
||||
|
||||
// Round | 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
|
||||
// ----------+-------------------------------------------------+
|
||||
// SIGMA[0] | 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
|
||||
// SIGMA[1] | 14 10 4 8 9 15 13 6 1 12 0 2 11 7 5 3 |
|
||||
// SIGMA[2] | 11 8 12 0 5 2 15 13 10 14 3 6 7 1 9 4 |
|
||||
// SIGMA[3] | 7 9 3 1 13 12 11 14 2 6 5 10 4 0 15 8 |
|
||||
// SIGMA[4] | 9 0 5 7 2 4 10 15 14 1 11 12 6 8 3 13 |
|
||||
// SIGMA[5] | 2 12 6 10 0 11 8 3 4 13 7 5 15 14 1 9 |
|
||||
// SIGMA[6] | 12 5 1 15 14 13 4 10 0 7 6 3 9 2 8 11 |
|
||||
// SIGMA[7] | 13 11 7 14 12 1 3 9 5 0 15 4 8 6 2 10 |
|
||||
// SIGMA[8] | 6 15 14 9 11 3 0 8 12 2 13 7 1 4 10 5 |
|
||||
// SIGMA[9] | 10 2 8 4 7 6 1 5 15 11 9 14 3 12 13 0 |
|
||||
// ----------+-------------------------------------------------+
|
||||
//
|
||||
|
||||
const SIGMA: [[usize; 16]; 10] = [
|
||||
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15],
|
||||
[14, 10, 4, 8, 9, 15, 13, 6, 1, 12, 0, 2, 11, 7, 5, 3],
|
||||
[11, 8, 12, 0, 5, 2, 15, 13, 10, 14, 3, 6, 7, 1, 9, 4],
|
||||
[7, 9, 3, 1, 13, 12, 11, 14, 2, 6, 5, 10, 4, 0, 15, 8],
|
||||
[9, 0, 5, 7, 2, 4, 10, 15, 14, 1, 11, 12, 6, 8, 3, 13],
|
||||
[2, 12, 6, 10, 0, 11, 8, 3, 4, 13, 7, 5, 15, 14, 1, 9],
|
||||
[12, 5, 1, 15, 14, 13, 4, 10, 0, 7, 6, 3, 9, 2, 8, 11],
|
||||
[13, 11, 7, 14, 12, 1, 3, 9, 5, 0, 15, 4, 8, 6, 2, 10],
|
||||
[6, 15, 14, 9, 11, 3, 0, 8, 12, 2, 13, 7, 1, 4, 10, 5],
|
||||
[10, 2, 8, 4, 7, 6, 1, 5, 15, 11, 9, 14, 3, 12, 13, 0],
|
||||
];
|
||||
|
||||
// 3.1. Mixing Function G
|
||||
// The G primitive function mixes two input words, "x" and "y", into
|
||||
// four words indexed by "a", "b", "c", and "d" in the working vector
|
||||
// v[0..15]. The full modified vector is returned. The rotation
|
||||
// constants (R1, R2, R3, R4) are given in Section 2.1.
|
||||
// FUNCTION G( v[0..15], a, b, c, d, x, y )
|
||||
// |
|
||||
// | v[a] := (v[a] + v[b] + x) mod 2**w
|
||||
// | v[d] := (v[d] ^ v[a]) >>> R1
|
||||
// | v[c] := (v[c] + v[d]) mod 2**w
|
||||
// | v[b] := (v[b] ^ v[c]) >>> R2
|
||||
// | v[a] := (v[a] + v[b] + y) mod 2**w
|
||||
// | v[d] := (v[d] ^ v[a]) >>> R3
|
||||
// | v[c] := (v[c] + v[d]) mod 2**w
|
||||
// | v[b] := (v[b] ^ v[c]) >>> R4
|
||||
// |
|
||||
// | RETURN v[0..15]
|
||||
// |
|
||||
// END FUNCTION.
|
||||
//
|
||||
|
||||
fn mixing_g<ConstraintF: PrimeField, CS: ConstraintSystem<ConstraintF>>(
|
||||
mut cs: CS,
|
||||
v: &mut [UInt32],
|
||||
a: usize,
|
||||
b: usize,
|
||||
c: usize,
|
||||
d: usize,
|
||||
x: &UInt32,
|
||||
y: &UInt32,
|
||||
) -> Result<(), SynthesisError> {
|
||||
v[a] = UInt32::addmany(
|
||||
cs.ns(|| "mixing step 1"),
|
||||
&[v[a].clone(), v[b].clone(), x.clone()],
|
||||
)?;
|
||||
v[d] = v[d].xor(cs.ns(|| "mixing step 2"), &v[a])?.rotr(R1);
|
||||
v[c] = UInt32::addmany(cs.ns(|| "mixing step 3"), &[v[c].clone(), v[d].clone()])?;
|
||||
v[b] = v[b].xor(cs.ns(|| "mixing step 4"), &v[c])?.rotr(R2);
|
||||
v[a] = UInt32::addmany(
|
||||
cs.ns(|| "mixing step 5"),
|
||||
&[v[a].clone(), v[b].clone(), y.clone()],
|
||||
)?;
|
||||
v[d] = v[d].xor(cs.ns(|| "mixing step 6"), &v[a])?.rotr(R3);
|
||||
v[c] = UInt32::addmany(cs.ns(|| "mixing step 7"), &[v[c].clone(), v[d].clone()])?;
|
||||
v[b] = v[b].xor(cs.ns(|| "mixing step 8"), &v[c])?.rotr(R4);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// 3.2. Compression Function F
|
||||
// Compression function F takes as an argument the state vector "h",
|
||||
// message block vector "m" (last block is padded with zeros to full
|
||||
// block size, if required), 2w-bit_gadget offset counter "t", and final block
|
||||
// indicator flag "f". Local vector v[0..15] is used in processing. F
|
||||
// returns a new state vector. The number of rounds, "r", is 12 for
|
||||
// BLAKE2b and 10 for BLAKE2s. Rounds are numbered from 0 to r - 1.
|
||||
// FUNCTION F( h[0..7], m[0..15], t, f )
|
||||
// |
|
||||
// | // Initialize local work vector v[0..15]
|
||||
// | v[0..7] := h[0..7] // First half from state.
|
||||
// | v[8..15] := IV[0..7] // Second half from IV.
|
||||
// |
|
||||
// | v[12] := v[12] ^ (t mod 2**w) // Low word of the offset.
|
||||
// | v[13] := v[13] ^ (t >> w) // High word.
|
||||
// |
|
||||
// | IF f = TRUE THEN // last block flag?
|
||||
// | | v[14] := v[14] ^ 0xFF..FF // Invert all bits.
|
||||
// | END IF.
|
||||
// |
|
||||
// | // Cryptographic mixing
|
||||
// | FOR i = 0 TO r - 1 DO // Ten or twelve rounds.
|
||||
// | |
|
||||
// | | // Message word selection permutation for this round.
|
||||
// | | s[0..15] := SIGMA[i mod 10][0..15]
|
||||
// | |
|
||||
// | | v := G( v, 0, 4, 8, 12, m[s[ 0]], m[s[ 1]] )
|
||||
// | | v := G( v, 1, 5, 9, 13, m[s[ 2]], m[s[ 3]] )
|
||||
// | | v := G( v, 2, 6, 10, 14, m[s[ 4]], m[s[ 5]] )
|
||||
// | | v := G( v, 3, 7, 11, 15, m[s[ 6]], m[s[ 7]] )
|
||||
// | |
|
||||
// | | v := G( v, 0, 5, 10, 15, m[s[ 8]], m[s[ 9]] )
|
||||
// | | v := G( v, 1, 6, 11, 12, m[s[10]], m[s[11]] )
|
||||
// | | v := G( v, 2, 7, 8, 13, m[s[12]], m[s[13]] )
|
||||
// | | v := G( v, 3, 4, 9, 14, m[s[14]], m[s[15]] )
|
||||
// | |
|
||||
// | END FOR
|
||||
// |
|
||||
// | FOR i = 0 TO 7 DO // XOR the two halves.
|
||||
// | | h[i] := h[i] ^ v[i] ^ v[i + 8]
|
||||
// | END FOR.
|
||||
// |
|
||||
// | RETURN h[0..7] // New state.
|
||||
// |
|
||||
// END FUNCTION.
|
||||
//
|
||||
|
||||
fn blake2s_compression<ConstraintF: PrimeField, CS: ConstraintSystem<ConstraintF>>(
|
||||
mut cs: CS,
|
||||
h: &mut [UInt32],
|
||||
m: &[UInt32],
|
||||
t: u64,
|
||||
f: bool,
|
||||
) -> Result<(), SynthesisError> {
|
||||
assert_eq!(h.len(), 8);
|
||||
assert_eq!(m.len(), 16);
|
||||
|
||||
// static const uint32_t blake2s_iv[8] =
|
||||
// {
|
||||
// 0x6A09E667, 0xBB67AE85, 0x3C6EF372, 0xA54FF53A,
|
||||
// 0x510E527F, 0x9B05688C, 0x1F83D9AB, 0x5BE0CD19
|
||||
// };
|
||||
//
|
||||
|
||||
let mut v = Vec::with_capacity(16);
|
||||
v.extend_from_slice(h);
|
||||
v.push(UInt32::constant(0x6A09E667));
|
||||
v.push(UInt32::constant(0xBB67AE85));
|
||||
v.push(UInt32::constant(0x3C6EF372));
|
||||
v.push(UInt32::constant(0xA54FF53A));
|
||||
v.push(UInt32::constant(0x510E527F));
|
||||
v.push(UInt32::constant(0x9B05688C));
|
||||
v.push(UInt32::constant(0x1F83D9AB));
|
||||
v.push(UInt32::constant(0x5BE0CD19));
|
||||
|
||||
assert_eq!(v.len(), 16);
|
||||
|
||||
v[12] = v[12].xor(cs.ns(|| "first xor"), &UInt32::constant(t as u32))?;
|
||||
v[13] = v[13].xor(cs.ns(|| "second xor"), &UInt32::constant((t >> 32) as u32))?;
|
||||
|
||||
if f {
|
||||
v[14] = v[14].xor(cs.ns(|| "third xor"), &UInt32::constant(u32::max_value()))?;
|
||||
}
|
||||
|
||||
for i in 0..10 {
|
||||
let mut cs = cs.ns(|| format!("round {}", i));
|
||||
|
||||
let s = SIGMA[i % 10];
|
||||
|
||||
mixing_g(
|
||||
cs.ns(|| "mixing invocation 1"),
|
||||
&mut v,
|
||||
0,
|
||||
4,
|
||||
8,
|
||||
12,
|
||||
&m[s[0]],
|
||||
&m[s[1]],
|
||||
)?;
|
||||
mixing_g(
|
||||
cs.ns(|| "mixing invocation 2"),
|
||||
&mut v,
|
||||
1,
|
||||
5,
|
||||
9,
|
||||
13,
|
||||
&m[s[2]],
|
||||
&m[s[3]],
|
||||
)?;
|
||||
mixing_g(
|
||||
cs.ns(|| "mixing invocation 3"),
|
||||
&mut v,
|
||||
2,
|
||||
6,
|
||||
10,
|
||||
14,
|
||||
&m[s[4]],
|
||||
&m[s[5]],
|
||||
)?;
|
||||
mixing_g(
|
||||
cs.ns(|| "mixing invocation 4"),
|
||||
&mut v,
|
||||
3,
|
||||
7,
|
||||
11,
|
||||
15,
|
||||
&m[s[6]],
|
||||
&m[s[7]],
|
||||
)?;
|
||||
|
||||
mixing_g(
|
||||
cs.ns(|| "mixing invocation 5"),
|
||||
&mut v,
|
||||
0,
|
||||
5,
|
||||
10,
|
||||
15,
|
||||
&m[s[8]],
|
||||
&m[s[9]],
|
||||
)?;
|
||||
mixing_g(
|
||||
cs.ns(|| "mixing invocation 6"),
|
||||
&mut v,
|
||||
1,
|
||||
6,
|
||||
11,
|
||||
12,
|
||||
&m[s[10]],
|
||||
&m[s[11]],
|
||||
)?;
|
||||
mixing_g(
|
||||
cs.ns(|| "mixing invocation 7"),
|
||||
&mut v,
|
||||
2,
|
||||
7,
|
||||
8,
|
||||
13,
|
||||
&m[s[12]],
|
||||
&m[s[13]],
|
||||
)?;
|
||||
mixing_g(
|
||||
cs.ns(|| "mixing invocation 8"),
|
||||
&mut v,
|
||||
3,
|
||||
4,
|
||||
9,
|
||||
14,
|
||||
&m[s[14]],
|
||||
&m[s[15]],
|
||||
)?;
|
||||
}
|
||||
|
||||
for i in 0..8 {
|
||||
let mut cs = cs.ns(|| format!("h[{i}] ^ v[{i}] ^ v[{i} + 8]", i = i));
|
||||
|
||||
h[i] = h[i].xor(cs.ns(|| "first xor"), &v[i])?;
|
||||
h[i] = h[i].xor(cs.ns(|| "second xor"), &v[i + 8])?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// FUNCTION BLAKE2( d[0..dd-1], ll, kk, nn )
|
||||
// |
|
||||
// | h[0..7] := IV[0..7] // Initialization Vector.
|
||||
// |
|
||||
// | // Parameter block p[0]
|
||||
// | h[0] := h[0] ^ 0x01010000 ^ (kk << 8) ^ nn
|
||||
// |
|
||||
// | // Process padded key and data blocks
|
||||
// | IF dd > 1 THEN
|
||||
// | | FOR i = 0 TO dd - 2 DO
|
||||
// | | | h := F( h, d[i], (i + 1) * bb, FALSE )
|
||||
// | | END FOR.
|
||||
// | END IF.
|
||||
// |
|
||||
// | // Final block.
|
||||
// | IF kk = 0 THEN
|
||||
// | | h := F( h, d[dd - 1], ll, TRUE )
|
||||
// | ELSE
|
||||
// | | h := F( h, d[dd - 1], ll + bb, TRUE )
|
||||
// | END IF.
|
||||
// |
|
||||
// | RETURN first "nn" bytes from little-endian word array h[].
|
||||
// |
|
||||
// END FUNCTION.
|
||||
//
|
||||
|
||||
pub fn blake2s_gadget<ConstraintF: PrimeField, CS: ConstraintSystem<ConstraintF>>(
|
||||
mut cs: CS,
|
||||
input: &[Boolean],
|
||||
) -> Result<Vec<UInt32>, SynthesisError> {
|
||||
assert!(input.len() % 8 == 0);
|
||||
|
||||
let mut h = Vec::with_capacity(8);
|
||||
h.push(UInt32::constant(0x6A09E667 ^ 0x01010000 ^ 32));
|
||||
h.push(UInt32::constant(0xBB67AE85));
|
||||
h.push(UInt32::constant(0x3C6EF372));
|
||||
h.push(UInt32::constant(0xA54FF53A));
|
||||
h.push(UInt32::constant(0x510E527F));
|
||||
h.push(UInt32::constant(0x9B05688C));
|
||||
h.push(UInt32::constant(0x1F83D9AB));
|
||||
h.push(UInt32::constant(0x5BE0CD19));
|
||||
|
||||
let mut blocks: Vec<Vec<UInt32>> = vec![];
|
||||
|
||||
for block in input.chunks(512) {
|
||||
let mut this_block = Vec::with_capacity(16);
|
||||
for word in block.chunks(32) {
|
||||
let mut tmp = word.to_vec();
|
||||
while tmp.len() < 32 {
|
||||
tmp.push(Boolean::constant(false));
|
||||
}
|
||||
this_block.push(UInt32::from_bits_le(&tmp));
|
||||
}
|
||||
while this_block.len() < 16 {
|
||||
this_block.push(UInt32::constant(0));
|
||||
}
|
||||
blocks.push(this_block);
|
||||
}
|
||||
|
||||
if blocks.is_empty() {
|
||||
blocks.push((0..16).map(|_| UInt32::constant(0)).collect());
|
||||
}
|
||||
|
||||
for (i, block) in blocks[0..blocks.len() - 1].iter().enumerate() {
|
||||
let cs = cs.ns(|| format!("block {}", i));
|
||||
|
||||
blake2s_compression(cs, &mut h, block, ((i as u64) + 1) * 64, false)?;
|
||||
}
|
||||
|
||||
{
|
||||
let cs = cs.ns(|| "final block");
|
||||
|
||||
blake2s_compression(
|
||||
cs,
|
||||
&mut h,
|
||||
&blocks[blocks.len() - 1],
|
||||
(input.len() / 8) as u64,
|
||||
true,
|
||||
)?;
|
||||
}
|
||||
|
||||
Ok(h)
|
||||
}
|
||||
|
||||
use crate::prf::Blake2s;
|
||||
|
||||
pub struct Blake2sGadget;
|
||||
#[derive(Clone, Debug)]
|
||||
pub struct Blake2sOutputGadget(pub Vec<UInt8>);
|
||||
|
||||
impl PartialEq for Blake2sOutputGadget {
|
||||
fn eq(&self, other: &Self) -> bool {
|
||||
self.0 == other.0
|
||||
}
|
||||
}
|
||||
|
||||
impl Eq for Blake2sOutputGadget {}
|
||||
|
||||
impl<ConstraintF: PrimeField> EqGadget<ConstraintF> for Blake2sOutputGadget {}
|
||||
|
||||
impl<ConstraintF: PrimeField> ConditionalEqGadget<ConstraintF> for Blake2sOutputGadget {
|
||||
#[inline]
|
||||
fn conditional_enforce_equal<CS: ConstraintSystem<ConstraintF>>(
|
||||
&self,
|
||||
mut cs: CS,
|
||||
other: &Self,
|
||||
condition: &Boolean,
|
||||
) -> Result<(), SynthesisError> {
|
||||
for (i, (a, b)) in self.0.iter().zip(other.0.iter()).enumerate() {
|
||||
a.conditional_enforce_equal(
|
||||
&mut cs.ns(|| format!("blake2s_equal_{}", i)),
|
||||
b,
|
||||
condition,
|
||||
)?;
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn cost() -> usize {
|
||||
32 * <UInt8 as ConditionalEqGadget<ConstraintF>>::cost()
|
||||
}
|
||||
}
|
||||
|
||||
impl<ConstraintF: PrimeField> ToBytesGadget<ConstraintF> for Blake2sOutputGadget {
|
||||
#[inline]
|
||||
fn to_bytes<CS: ConstraintSystem<ConstraintF>>(&self, _cs: CS) -> Result<Vec<UInt8>, SynthesisError> {
|
||||
Ok(self.0.clone())
|
||||
}
|
||||
|
||||
#[inline]
|
||||
fn to_bytes_strict<CS: ConstraintSystem<ConstraintF>>(
|
||||
&self,
|
||||
cs: CS,
|
||||
) -> Result<Vec<UInt8>, SynthesisError> {
|
||||
self.to_bytes(cs)
|
||||
}
|
||||
}
|
||||
|
||||
impl<ConstraintF: PrimeField> AllocGadget<[u8; 32], ConstraintF> for Blake2sOutputGadget {
|
||||
#[inline]
|
||||
fn alloc<F, T, CS: ConstraintSystem<ConstraintF>>(cs: CS, value_gen: F) -> Result<Self, SynthesisError>
|
||||
where
|
||||
F: FnOnce() -> Result<T, SynthesisError>,
|
||||
T: Borrow<[u8; 32]>,
|
||||
{
|
||||
let zeros = [0u8; 32];
|
||||
let value = match value_gen() {
|
||||
Ok(val) => *(val.borrow()),
|
||||
Err(_) => zeros,
|
||||
};
|
||||
let bytes = <UInt8>::alloc_vec(cs, &value)?;
|
||||
|
||||
Ok(Blake2sOutputGadget(bytes))
|
||||
}
|
||||
|
||||
#[inline]
|
||||
fn alloc_input<F, T, CS: ConstraintSystem<ConstraintF>>(
|
||||
cs: CS,
|
||||
value_gen: F,
|
||||
) -> Result<Self, SynthesisError>
|
||||
where
|
||||
F: FnOnce() -> Result<T, SynthesisError>,
|
||||
T: Borrow<[u8; 32]>,
|
||||
{
|
||||
let zeros = [0u8; 32];
|
||||
let value = match value_gen() {
|
||||
Ok(val) => *(val.borrow()),
|
||||
Err(_) => zeros,
|
||||
};
|
||||
let bytes = <UInt8>::alloc_input_vec(cs, &value)?;
|
||||
|
||||
Ok(Blake2sOutputGadget(bytes))
|
||||
}
|
||||
}
|
||||
|
||||
impl<ConstraintF: PrimeField> PRFGadget<Blake2s, ConstraintF> for Blake2sGadget {
|
||||
type OutputGadget = Blake2sOutputGadget;
|
||||
|
||||
fn new_seed<CS: ConstraintSystem<ConstraintF>>(mut cs: CS, seed: &[u8; 32]) -> Vec<UInt8> {
|
||||
UInt8::alloc_vec(&mut cs.ns(|| "alloc_seed"), seed).unwrap()
|
||||
}
|
||||
|
||||
fn check_evaluation_gadget<CS: ConstraintSystem<ConstraintF>>(
|
||||
mut cs: CS,
|
||||
seed: &[UInt8],
|
||||
input: &[UInt8],
|
||||
) -> Result<Self::OutputGadget, SynthesisError> {
|
||||
assert_eq!(seed.len(), 32);
|
||||
// assert_eq!(input.len(), 32);
|
||||
let mut gadget_input = Vec::with_capacity(512);
|
||||
for byte in seed.into_iter().chain(input) {
|
||||
gadget_input.extend_from_slice(&byte.into_bits_le());
|
||||
}
|
||||
let mut result = Vec::new();
|
||||
for (i, int) in blake2s_gadget(cs.ns(|| "Blake2s Eval"), &gadget_input)?
|
||||
.into_iter()
|
||||
.enumerate()
|
||||
{
|
||||
let chunk = int.to_bytes(&mut cs.ns(|| format!("Result ToBytes {}", i)))?;
|
||||
result.extend_from_slice(&chunk);
|
||||
}
|
||||
Ok(Blake2sOutputGadget(result))
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
use algebra::fields::bls12_377::fr::Fr;
|
||||
use digest::{FixedOutput, Input};
|
||||
use rand::{Rng, SeedableRng};
|
||||
use rand_xorshift::XorShiftRng;
|
||||
|
||||
use crate::prf::blake2s::{Blake2s as B2SPRF, constraints::blake2s_gadget};
|
||||
use blake2::Blake2s;
|
||||
use r1cs_core::ConstraintSystem;
|
||||
|
||||
use super::Blake2sGadget;
|
||||
use r1cs_std::{prelude::*, boolean::AllocatedBit, test_constraint_system::TestConstraintSystem};
|
||||
|
||||
#[test]
|
||||
fn test_blake2s_constraints() {
|
||||
let mut cs = TestConstraintSystem::<Fr>::new();
|
||||
let input_bits: Vec<_> = (0..512)
|
||||
.map(|i| {
|
||||
AllocatedBit::alloc(cs.ns(|| format!("input bit_gadget {}", i)), || Ok(true))
|
||||
.unwrap()
|
||||
.into()
|
||||
})
|
||||
.collect();
|
||||
blake2s_gadget(&mut cs, &input_bits).unwrap();
|
||||
assert!(cs.is_satisfied());
|
||||
assert_eq!(cs.num_constraints(), 21792);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_blake2s_prf() {
|
||||
use crate::prf::{PRF, PRFGadget};
|
||||
use rand::Rng;
|
||||
|
||||
let mut rng = XorShiftRng::seed_from_u64(1231275789u64);
|
||||
let mut cs = TestConstraintSystem::<Fr>::new();
|
||||
|
||||
let mut seed = [0u8; 32];
|
||||
rng.fill(&mut seed);
|
||||
|
||||
let mut input = [0u8; 32];
|
||||
rng.fill(&mut input);
|
||||
|
||||
let seed_gadget = Blake2sGadget::new_seed(&mut cs.ns(|| "declare_seed"), &seed);
|
||||
let input_gadget = UInt8::alloc_vec(&mut cs.ns(|| "declare_input"), &input).unwrap();
|
||||
let out = B2SPRF::evaluate(&seed, &input).unwrap();
|
||||
let actual_out_gadget = <Blake2sGadget as PRFGadget<_, Fr>>::OutputGadget::alloc(
|
||||
&mut cs.ns(|| "declare_output"),
|
||||
|| Ok(out),
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
let output_gadget = Blake2sGadget::check_evaluation_gadget(
|
||||
&mut cs.ns(|| "eval_blake2s"),
|
||||
&seed_gadget,
|
||||
&input_gadget,
|
||||
)
|
||||
.unwrap();
|
||||
output_gadget
|
||||
.enforce_equal(&mut cs, &actual_out_gadget)
|
||||
.unwrap();
|
||||
|
||||
if !cs.is_satisfied() {
|
||||
println!(
|
||||
"which is unsatisfied: {:?}",
|
||||
cs.which_is_unsatisfied().unwrap()
|
||||
);
|
||||
}
|
||||
assert!(cs.is_satisfied());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_blake2s_precomp_constraints() {
|
||||
// Test that 512 fixed leading bits (constants)
|
||||
// doesn't result in more constraints.
|
||||
|
||||
let mut cs = TestConstraintSystem::<Fr>::new();
|
||||
let mut rng = XorShiftRng::seed_from_u64(1231275789u64);
|
||||
let input_bits: Vec<_> = (0..512)
|
||||
.map(|_| Boolean::constant(rng.gen()))
|
||||
.chain((0..512).map(|i| {
|
||||
AllocatedBit::alloc(cs.ns(|| format!("input bit_gadget {}", i)), || Ok(true))
|
||||
.unwrap()
|
||||
.into()
|
||||
}))
|
||||
.collect();
|
||||
blake2s_gadget(&mut cs, &input_bits).unwrap();
|
||||
assert!(cs.is_satisfied());
|
||||
assert_eq!(cs.num_constraints(), 21792);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_blake2s_constant_constraints() {
|
||||
let mut cs = TestConstraintSystem::<Fr>::new();
|
||||
let mut rng = XorShiftRng::seed_from_u64(1231275789u64);
|
||||
let input_bits: Vec<_> = (0..512).map(|_| Boolean::constant(rng.gen())).collect();
|
||||
blake2s_gadget(&mut cs, &input_bits).unwrap();
|
||||
assert_eq!(cs.num_constraints(), 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_blake2s() {
|
||||
let mut rng = XorShiftRng::seed_from_u64(1231275789u64);
|
||||
|
||||
for input_len in (0..32).chain((32..256).filter(|a| a % 8 == 0)) {
|
||||
let mut h = Blake2s::new_keyed(&[], 32);
|
||||
|
||||
let data: Vec<u8> = (0..input_len).map(|_| rng.gen()).collect();
|
||||
|
||||
h.process(&data);
|
||||
|
||||
let hash_result = h.fixed_result();
|
||||
|
||||
let mut cs = TestConstraintSystem::<Fr>::new();
|
||||
|
||||
let mut input_bits = vec![];
|
||||
|
||||
for (byte_i, input_byte) in data.into_iter().enumerate() {
|
||||
for bit_i in 0..8 {
|
||||
let cs = cs.ns(|| format!("input bit_gadget {} {}", byte_i, bit_i));
|
||||
|
||||
input_bits.push(
|
||||
AllocatedBit::alloc(cs, || Ok((input_byte >> bit_i) & 1u8 == 1u8))
|
||||
.unwrap()
|
||||
.into(),
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
let r = blake2s_gadget(&mut cs, &input_bits).unwrap();
|
||||
|
||||
assert!(cs.is_satisfied());
|
||||
|
||||
let mut s = hash_result
|
||||
.as_ref()
|
||||
.iter()
|
||||
.flat_map(|&byte| (0..8).map(move |i| (byte >> i) & 1u8 == 1u8));
|
||||
|
||||
for chunk in r {
|
||||
for b in chunk.to_bits_le() {
|
||||
match b {
|
||||
Boolean::Is(b) => {
|
||||
assert!(s.next().unwrap() == b.get_value().unwrap());
|
||||
},
|
||||
Boolean::Not(b) => {
|
||||
assert!(s.next().unwrap() != b.get_value().unwrap());
|
||||
},
|
||||
Boolean::Constant(b) => {
|
||||
assert!(input_len == 0);
|
||||
assert!(s.next().unwrap() == b);
|
||||
},
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
28
crypto-primitives/src/prf/blake2s/mod.rs
Normal file
28
crypto-primitives/src/prf/blake2s/mod.rs
Normal file
@@ -0,0 +1,28 @@
|
||||
use blake2::Blake2s as b2s;
|
||||
use digest::Digest;
|
||||
|
||||
use super::PRF;
|
||||
use crate::CryptoError;
|
||||
|
||||
#[cfg(feature = "r1cs")]
|
||||
pub mod constraints;
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct Blake2s;
|
||||
|
||||
impl PRF for Blake2s {
|
||||
type Input = [u8; 32];
|
||||
type Output = [u8; 32];
|
||||
type Seed = [u8; 32];
|
||||
|
||||
fn evaluate(seed: &Self::Seed, input: &Self::Input) -> Result<Self::Output, CryptoError> {
|
||||
let eval_time = start_timer!(|| "Blake2s::Eval");
|
||||
let mut h = b2s::new();
|
||||
h.input(seed.as_ref());
|
||||
h.input(input.as_ref());
|
||||
let mut result = [0u8; 32];
|
||||
result.copy_from_slice(&h.result());
|
||||
end_timer!(eval_time);
|
||||
Ok(result)
|
||||
}
|
||||
}
|
||||
19
crypto-primitives/src/prf/constraints.rs
Normal file
19
crypto-primitives/src/prf/constraints.rs
Normal file
@@ -0,0 +1,19 @@
|
||||
use algebra::Field;
|
||||
use std::fmt::Debug;
|
||||
|
||||
use crate::prf::PRF;
|
||||
use r1cs_core::{ConstraintSystem, SynthesisError};
|
||||
|
||||
use r1cs_std::prelude::*;
|
||||
|
||||
pub trait PRFGadget<P: PRF, ConstraintF: Field> {
|
||||
type OutputGadget: EqGadget<ConstraintF> + ToBytesGadget<ConstraintF> + AllocGadget<P::Output, ConstraintF> + Clone + Debug;
|
||||
|
||||
fn new_seed<CS: ConstraintSystem<ConstraintF>>(cs: CS, output: &P::Seed) -> Vec<UInt8>;
|
||||
|
||||
fn check_evaluation_gadget<CS: ConstraintSystem<ConstraintF>>(
|
||||
cs: CS,
|
||||
seed: &[UInt8],
|
||||
input: &[UInt8],
|
||||
) -> Result<Self::OutputGadget, SynthesisError>;
|
||||
}
|
||||
20
crypto-primitives/src/prf/mod.rs
Normal file
20
crypto-primitives/src/prf/mod.rs
Normal file
@@ -0,0 +1,20 @@
|
||||
use algebra::bytes::{FromBytes, ToBytes};
|
||||
use std::{fmt::Debug, hash::Hash};
|
||||
|
||||
use crate::CryptoError;
|
||||
|
||||
#[cfg(feature = "r1cs")]
|
||||
pub mod constraints;
|
||||
#[cfg(feature = "r1cs")]
|
||||
pub use constraints::*;
|
||||
|
||||
pub mod blake2s;
|
||||
pub use self::blake2s::*;
|
||||
|
||||
pub trait PRF {
|
||||
type Input: FromBytes + Default;
|
||||
type Output: ToBytes + Eq + Clone + Default + Hash;
|
||||
type Seed: FromBytes + ToBytes + Clone + Default + Debug;
|
||||
|
||||
fn evaluate(seed: &Self::Seed, input: &Self::Input) -> Result<Self::Output, CryptoError>;
|
||||
}
|
||||
Reference in New Issue
Block a user