2019-06-14 15:14:06 -07:00
|
|
|
/*
|
|
|
|
Implementation of the ZK Range Proof scheme, based on:
|
|
|
|
Efficient Protocols for Set Membership and Range Proofs
|
|
|
|
Jan Camenisch, Rafik Chaabouni, and abhi shelat
|
|
|
|
Asiacrypt 2008
|
|
|
|
*/
|
|
|
|
extern crate pairing;
|
|
|
|
extern crate rand;
|
|
|
|
|
2019-06-16 13:29:48 -07:00
|
|
|
use rand::{thread_rng, Rng};
|
2019-06-14 15:14:06 -07:00
|
|
|
use super::*;
|
2019-07-01 18:51:24 -07:00
|
|
|
use cl::{KeyPair, Signature, PublicParams, setup, BlindKeyPair};
|
2019-06-18 14:49:58 -07:00
|
|
|
use ped92::{CSParams, Commitment};
|
2019-06-16 09:27:35 -07:00
|
|
|
use pairing::{Engine, CurveProjective};
|
2019-06-14 15:14:06 -07:00
|
|
|
use ff::PrimeField;
|
2019-06-16 09:27:35 -07:00
|
|
|
use std::collections::HashMap;
|
|
|
|
use std::fmt::Display;
|
2019-06-16 12:38:07 -07:00
|
|
|
use std::mem::transmute;
|
2019-06-14 15:14:06 -07:00
|
|
|
|
|
|
|
/*
|
|
|
|
paramsUL contains elements generated by the verifier, which are necessary for the prover.
|
|
|
|
This must be computed in a trusted setup.
|
|
|
|
*/
|
|
|
|
#[derive(Clone)]
|
|
|
|
struct ParamsUL<E: Engine> {
|
|
|
|
pub mpk: PublicParams<E>,
|
2019-06-16 09:27:35 -07:00
|
|
|
pub signatures: HashMap<String, Signature<E>>,
|
2019-06-18 14:49:58 -07:00
|
|
|
pub com: CSParams<E>,
|
2019-07-01 18:51:24 -07:00
|
|
|
kp: BlindKeyPair<E>,
|
2019-06-14 15:14:06 -07:00
|
|
|
// u determines the amount of signatures we need in the public params.
|
|
|
|
// Each signature can be compressed to just 1 field element of 256 bits.
|
|
|
|
// Then the parameters have minimum size equal to 256*u bits.
|
|
|
|
u: i64,
|
|
|
|
// l determines how many pairings we need to compute, then in order to improve
|
|
|
|
// verifier`s performance we want to minize it.
|
|
|
|
// Namely, we have 2*l pairings for the prover and 3*l for the verifier.
|
|
|
|
l: i64,
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
proofUL contains the necessary elements for the ZK range proof.
|
|
|
|
*/
|
|
|
|
#[derive(Clone)]
|
|
|
|
struct ProofUL<E: Engine> {
|
2019-06-18 13:20:16 -07:00
|
|
|
V: Vec<(E::G1, E::G1)>,
|
2019-06-17 12:01:31 -07:00
|
|
|
D: E::G2,
|
2019-06-16 09:27:35 -07:00
|
|
|
comm: Commitment<E>,
|
2019-06-14 15:14:06 -07:00
|
|
|
a: Vec<E::Fqk>,
|
2019-06-18 13:20:16 -07:00
|
|
|
zx: Vec<E::Fr>,
|
2019-07-02 14:28:49 -07:00
|
|
|
zsig: Vec<Vec<E::Fr>>,
|
2019-06-14 15:14:06 -07:00
|
|
|
zv: Vec<E::Fr>,
|
|
|
|
ch: E::Fr,
|
|
|
|
zr: E::Fr,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone)]
|
|
|
|
pub struct RangeProof<E: Engine> {
|
|
|
|
p1: ProofUL<E>,
|
|
|
|
p2: ProofUL<E>,
|
|
|
|
}
|
|
|
|
/*
|
|
|
|
params contains elements generated by the verifier, which are necessary for the prover.
|
|
|
|
This must be computed in a trusted setup.
|
|
|
|
*/
|
|
|
|
#[derive(Clone)]
|
|
|
|
pub struct RPPublicParams<E: Engine> {
|
|
|
|
p: ParamsUL<E>,
|
|
|
|
a: i64,
|
|
|
|
b: i64,
|
|
|
|
}
|
|
|
|
|
2019-06-16 13:29:48 -07:00
|
|
|
impl<E: Engine> ParamsUL<E> {
|
|
|
|
/*
|
2019-06-14 15:14:06 -07:00
|
|
|
setup_ul generates the signature for the interval [0,u^l).
|
|
|
|
The value of u should be roughly b/log(b), but we can choose smaller values in
|
|
|
|
order to get smaller parameters, at the cost of having worse performance.
|
|
|
|
*/
|
2019-06-17 12:01:31 -07:00
|
|
|
pub fn setup_ul<R: Rng>(rng: &mut R, u: i64, l: i64) -> Self {
|
2019-06-16 13:29:48 -07:00
|
|
|
let mpk = setup(rng);
|
2019-07-01 18:51:24 -07:00
|
|
|
let kp = BlindKeyPair::<E>::generate(rng, &mpk, 1);
|
2019-06-14 15:14:06 -07:00
|
|
|
|
2019-06-16 13:29:48 -07:00
|
|
|
let mut signatures: HashMap<String, Signature<E>> = HashMap::new();
|
|
|
|
for i in 0..u {
|
|
|
|
let sig_i = kp.sign(rng, &vec! {E::Fr::from_str(i.to_string().as_str()).unwrap()});
|
|
|
|
signatures.insert(i.to_string(), sig_i);
|
|
|
|
}
|
2019-06-14 15:14:06 -07:00
|
|
|
|
2019-06-18 14:49:58 -07:00
|
|
|
let com = CSParams::setup(rng);
|
2019-06-16 13:29:48 -07:00
|
|
|
return ParamsUL { mpk, signatures, com, kp, u, l };
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
|
|
|
|
2019-06-16 13:29:48 -07:00
|
|
|
/*
|
2019-06-16 09:27:35 -07:00
|
|
|
prove_ul method is used to produce the ZKRP proof that secret x belongs to the interval [0,U^L].
|
|
|
|
*/
|
2019-06-16 13:29:48 -07:00
|
|
|
pub fn prove_ul<R: Rng>(&self, rng: &mut R, x: i64, r: E::Fr) -> ProofUL<E> {
|
2019-06-28 14:44:14 -07:00
|
|
|
if x > self.u.pow(self.l as u32) || x < 0 {
|
|
|
|
panic!("x is not within the range.");
|
|
|
|
}
|
2019-06-17 12:01:31 -07:00
|
|
|
let decx = decompose(x, self.u, self.l);
|
2019-06-16 13:29:48 -07:00
|
|
|
let modx = E::Fr::from_str(&(x.to_string())).unwrap();
|
2019-06-16 09:27:35 -07:00
|
|
|
|
|
|
|
// Initialize variables
|
2019-06-16 13:29:48 -07:00
|
|
|
let mut v = Vec::<E::Fr>::with_capacity(self.l as usize);
|
2019-06-18 13:20:16 -07:00
|
|
|
let mut V = Vec::<(E::G1, E::G1)>::with_capacity(self.l as usize);
|
2019-06-16 13:29:48 -07:00
|
|
|
let mut a = Vec::<E::Fqk>::with_capacity(self.l as usize);
|
|
|
|
let mut s = Vec::<E::Fr>::with_capacity(self.l as usize);
|
2019-07-02 14:28:49 -07:00
|
|
|
let mut t = Vec::<Vec<E::Fr>>::with_capacity(self.l as usize);
|
2019-06-18 13:20:16 -07:00
|
|
|
let mut tt = Vec::<E::Fr>::with_capacity(self.l as usize);
|
|
|
|
let mut zx = Vec::<E::Fr>::with_capacity(self.l as usize);
|
2019-07-02 14:28:49 -07:00
|
|
|
let mut zsig = Vec::<Vec<E::Fr>>::with_capacity(self.l as usize);
|
2019-06-16 13:29:48 -07:00
|
|
|
let mut zv = Vec::<E::Fr>::with_capacity(self.l as usize);
|
|
|
|
let mut D = E::G2::zero();
|
2019-06-26 10:33:08 -07:00
|
|
|
let m = E::Fr::rand(rng);
|
2019-06-16 09:27:35 -07:00
|
|
|
|
2019-06-18 13:20:16 -07:00
|
|
|
// D = H^m
|
2019-06-17 12:01:31 -07:00
|
|
|
let mut hm = self.com.h.clone();
|
|
|
|
hm.mul_assign(m);
|
2019-06-26 10:33:08 -07:00
|
|
|
for i in 0..self.l as usize {
|
2019-07-01 18:51:24 -07:00
|
|
|
let signature = self.signatures.get(&decx[i].to_string()).unwrap();
|
2019-07-02 14:28:49 -07:00
|
|
|
let (v1, s1, t1, tt1, gx, V1) = self.proof_sig_commitment(rng, &signature);
|
2019-07-01 18:51:24 -07:00
|
|
|
|
2019-07-02 14:28:49 -07:00
|
|
|
s.push(s1);
|
|
|
|
t.push(t1);
|
|
|
|
v.push(v1);
|
|
|
|
tt.push(tt1);
|
|
|
|
V.push(V1);
|
2019-07-01 18:51:24 -07:00
|
|
|
a.push(gx);
|
2019-06-16 09:27:35 -07:00
|
|
|
|
2019-06-16 13:29:48 -07:00
|
|
|
let ui = self.u.pow(i as u32);
|
2019-06-17 12:01:31 -07:00
|
|
|
let mut aux = self.com.g.clone();
|
2019-07-02 14:28:49 -07:00
|
|
|
for j in 0..self.kp.public.Y2.len() {
|
|
|
|
let mut muiti = t[i][j].clone();
|
|
|
|
muiti.mul_assign(&E::Fr::from_str(&ui.to_string()).unwrap());
|
|
|
|
aux.mul_assign(muiti);
|
|
|
|
}
|
2019-06-16 13:29:48 -07:00
|
|
|
D.add_assign(&aux);
|
2019-06-26 10:33:08 -07:00
|
|
|
}
|
2019-06-17 12:01:31 -07:00
|
|
|
D.add_assign(&hm);
|
2019-06-16 13:29:48 -07:00
|
|
|
|
2019-06-17 12:01:31 -07:00
|
|
|
let C = self.com.commit(rng, modx, Some(r));
|
2019-06-18 13:20:16 -07:00
|
|
|
// Fiat-Shamir heuristic
|
2019-06-16 13:29:48 -07:00
|
|
|
let c = Hash::<E>(a.clone(), D.clone());
|
|
|
|
|
|
|
|
let mut zr = m.clone();
|
2019-06-17 12:01:31 -07:00
|
|
|
let mut rc = r.clone();
|
|
|
|
rc.mul_assign(&c);
|
2019-06-18 13:20:16 -07:00
|
|
|
zr.add_assign(&rc);
|
2019-06-16 13:29:48 -07:00
|
|
|
for i in 0..self.l as usize {
|
|
|
|
let mut dx = E::Fr::from_str(&decx[i].to_string()).unwrap();
|
2019-07-01 18:51:24 -07:00
|
|
|
|
2019-07-02 14:28:49 -07:00
|
|
|
let (zsig1, zx1, zv1) = <ParamsUL<E>>::proof_sig_response(v[i], s[i], &mut t[i], tt[i], c, &mut dx);
|
2019-07-01 18:51:24 -07:00
|
|
|
|
|
|
|
zv.push(zv1);
|
|
|
|
zx.push(zx1);
|
|
|
|
zsig.push(zsig1);
|
2019-06-17 12:01:31 -07:00
|
|
|
}
|
|
|
|
|
2019-06-18 13:20:16 -07:00
|
|
|
return ProofUL { V, D, comm: C, a, zx, zsig, zv, ch: c, zr };
|
2019-06-17 12:01:31 -07:00
|
|
|
}
|
|
|
|
|
2019-07-02 14:28:49 -07:00
|
|
|
fn proof_sig_commitment<R: Rng>(&self, rng: &mut R, signature: &Signature<E>) -> (E::Fr, E::Fr, Vec<E::Fr>, E::Fr, E::Fqk, (E::G1, E::G1)) {
|
|
|
|
let v1 = E::Fr::rand(rng);
|
|
|
|
let blindSig = self.kp.blind(rng, &v1, signature);
|
|
|
|
let s1 = E::Fr::rand(rng);
|
|
|
|
let mut t1 = Vec::<E::Fr>::with_capacity(self.kp.public.Y2.len());
|
|
|
|
let tt1 = E::Fr::rand(rng);
|
|
|
|
let mut gx = E::pairing(blindSig.h, self.kp.public.X);
|
|
|
|
gx = gx.pow(s1.into_repr());
|
|
|
|
for j in 0..self.kp.public.Y2.len() {
|
|
|
|
t1.push(E::Fr::rand(rng));
|
|
|
|
let mut gy = E::pairing(blindSig.h, self.kp.public.Y2[j]);
|
|
|
|
gy = gy.pow(t1[j].into_repr());
|
|
|
|
gx.mul_assign(&gy);
|
|
|
|
}
|
|
|
|
let mut h = E::pairing(blindSig.h, self.mpk.g2);
|
|
|
|
h = h.pow(tt1.into_repr());
|
|
|
|
gx.mul_assign(&h);
|
|
|
|
(v1, s1, t1, tt1, gx, (blindSig.h, blindSig.H))
|
|
|
|
}
|
|
|
|
|
|
|
|
fn proof_sig_response(v: E::Fr, s: E::Fr, t: &mut Vec<E::Fr>, tt: E::Fr, c: E::Fr, message: &mut E::Fr) -> (Vec<E::Fr>, E::Fr, E::Fr) {
|
2019-07-01 18:51:24 -07:00
|
|
|
let mut zsig1 = t.clone();
|
2019-07-02 14:28:49 -07:00
|
|
|
for zsig in &mut zsig1 {
|
|
|
|
message.mul_assign(&c);
|
|
|
|
zsig.add_assign(&message);
|
|
|
|
}
|
2019-07-01 18:51:24 -07:00
|
|
|
let mut zx1 = s.clone();
|
|
|
|
zx1.add_assign(&c);
|
|
|
|
let mut zv1 = tt.clone();
|
|
|
|
let mut vic = v.clone();
|
|
|
|
vic.mul_assign(&c);
|
|
|
|
zv1.add_assign(&vic);
|
|
|
|
(zsig1, zx1, zv1)
|
|
|
|
}
|
|
|
|
|
2019-06-17 12:01:31 -07:00
|
|
|
/*
|
|
|
|
verify_ul is used to validate the ZKRP proof. It returns true iff the proof is valid.
|
|
|
|
*/
|
|
|
|
pub fn verify_ul(&self, proof: &ProofUL<E>) -> bool {
|
|
|
|
// D == C^c.h^ zr.g^zsig ?
|
|
|
|
let r1 = self.verify_part1(&proof);
|
|
|
|
let r2 = self.verify_part2(&proof);
|
|
|
|
return r1 && r2;
|
|
|
|
}
|
|
|
|
|
|
|
|
fn verify_part2(&self, proof: &ProofUL<E>) -> bool {
|
|
|
|
let mut r2 = true;
|
|
|
|
for i in 0..self.l as usize {
|
2019-07-01 18:51:24 -07:00
|
|
|
let blindSig = proof.V[i];
|
|
|
|
let zx = proof.zx[i];
|
2019-07-02 14:28:49 -07:00
|
|
|
let zsig = proof.zsig[i].clone();
|
2019-07-01 18:51:24 -07:00
|
|
|
let zv = proof.zv[i];
|
|
|
|
let challenge = proof.ch;
|
|
|
|
let a = proof.a[i];
|
2019-06-18 13:20:16 -07:00
|
|
|
|
2019-07-01 18:51:24 -07:00
|
|
|
let subResult = self.verify_sig(blindSig, zx, zsig, zv, challenge, &a);
|
2019-06-18 13:20:16 -07:00
|
|
|
|
2019-07-01 18:51:24 -07:00
|
|
|
r2 = r2 && subResult;
|
|
|
|
}
|
|
|
|
r2
|
|
|
|
}
|
2019-06-18 13:20:16 -07:00
|
|
|
|
2019-07-02 14:28:49 -07:00
|
|
|
fn verify_sig(&self, blindSig: (E::G1, E::G1), zx: E::Fr, zsig: Vec<E::Fr>, zv: E::Fr, challenge: E::Fr, a: &E::Fqk) -> bool {
|
2019-07-01 18:51:24 -07:00
|
|
|
let mut gx = E::pairing(blindSig.0, self.kp.public.X);
|
|
|
|
gx = gx.pow(zx.into_repr());
|
2019-07-02 14:28:49 -07:00
|
|
|
for j in 0..self.kp.public.Y2.len() {
|
|
|
|
let mut gy = E::pairing(blindSig.0, self.kp.public.Y2[j]);
|
|
|
|
gy = gy.pow(zsig[j].into_repr());
|
2019-07-01 18:51:24 -07:00
|
|
|
gx.mul_assign(&gy);
|
2019-06-17 12:01:31 -07:00
|
|
|
}
|
2019-07-01 18:51:24 -07:00
|
|
|
let mut h = E::pairing(blindSig.0, self.mpk.g2);
|
|
|
|
h = h.pow(zv.into_repr());
|
|
|
|
gx.mul_assign(&h);
|
|
|
|
let mut g = E::pairing(blindSig.1, self.mpk.g2);
|
|
|
|
g = g.pow(challenge.into_repr());
|
|
|
|
g.mul_assign(&a);
|
|
|
|
gx == g
|
2019-06-17 12:01:31 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
fn verify_part1(&self, proof: &ProofUL<E>) -> bool {
|
|
|
|
let mut D = proof.comm.c.clone();
|
|
|
|
D.mul_assign(proof.ch);
|
2019-06-18 13:20:16 -07:00
|
|
|
D.negate();
|
2019-06-17 12:01:31 -07:00
|
|
|
let mut hzr = self.com.h.clone();
|
|
|
|
hzr.mul_assign(proof.zr);
|
|
|
|
D.add_assign(&hzr);
|
2019-07-02 14:28:49 -07:00
|
|
|
for i in 0..self.l as usize {
|
2019-06-17 12:01:31 -07:00
|
|
|
let ui = self.u.pow(i as u32);
|
|
|
|
let mut aux = self.com.g.clone();
|
2019-07-02 14:28:49 -07:00
|
|
|
for j in 0..self.kp.public.Y2.len() {
|
|
|
|
let mut muizsigi = proof.zsig[i][j];
|
|
|
|
muizsigi.mul_assign(&E::Fr::from_str(&ui.to_string()).unwrap());
|
|
|
|
aux.mul_assign(muizsigi);
|
|
|
|
}
|
2019-06-17 12:01:31 -07:00
|
|
|
D.add_assign(&aux);
|
2019-06-16 13:29:48 -07:00
|
|
|
}
|
2019-06-17 12:01:31 -07:00
|
|
|
return D == proof.D;
|
2019-06-16 09:27:35 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn Hash<E: Engine>(a: Vec<E::Fqk>, D: E::G2) -> E::Fr {
|
|
|
|
// create a Sha256 object
|
|
|
|
let mut a_vec: Vec<u8> = Vec::new();
|
|
|
|
for a_el in a {
|
|
|
|
a_vec.extend(format!("{}", a_el).bytes());
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut x_vec: Vec<u8> = Vec::new();
|
|
|
|
x_vec.extend(format!("{}", D).bytes());
|
|
|
|
a_vec.extend(x_vec);
|
|
|
|
let sha2_digest = sha512::hash(a_vec.as_slice());
|
|
|
|
|
|
|
|
let mut hash_buf: [u8; 64] = [0; 64];
|
|
|
|
hash_buf.copy_from_slice(&sha2_digest[0..64]);
|
2019-06-26 10:33:08 -07:00
|
|
|
let hexresult = fmt_bytes_to_int(hash_buf);
|
2019-06-16 12:38:07 -07:00
|
|
|
let result = E::Fr::from_str(&hexresult);
|
|
|
|
return result.unwrap();
|
2019-06-16 09:27:35 -07:00
|
|
|
}
|
2019-06-14 15:14:06 -07:00
|
|
|
|
2019-06-16 13:29:48 -07:00
|
|
|
/*
|
|
|
|
Decompose receives as input an integer x and outputs an array of integers such that
|
|
|
|
x = sum(xi.u^i), i.e. it returns the decomposition of x into base u.
|
|
|
|
*/
|
2019-06-17 12:01:31 -07:00
|
|
|
fn decompose(x: i64, u: i64, l: i64) -> Vec<i64> {
|
2019-06-16 13:29:48 -07:00
|
|
|
let mut result = Vec::with_capacity(l as usize);
|
|
|
|
let mut decomposer = x.clone();
|
2019-06-26 10:33:08 -07:00
|
|
|
for _i in 0..l {
|
2019-06-16 13:29:48 -07:00
|
|
|
result.push(decomposer % u);
|
|
|
|
decomposer = decomposer / u;
|
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2019-06-16 12:38:07 -07:00
|
|
|
fn fmt_bytes_to_int(bytearray: [u8; 64]) -> String {
|
|
|
|
let mut result: String = "".to_string();
|
|
|
|
for byte in bytearray.iter() {
|
|
|
|
// Decide if you want upper- or lowercase results,
|
|
|
|
// padding the values to two characters, spaces
|
|
|
|
// between bytes, etc.
|
|
|
|
result = result + &format!("{}", *byte as u8);
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
2019-06-16 12:38:07 -07:00
|
|
|
result.to_string()
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<E: Engine> RPPublicParams<E> {
|
|
|
|
/*
|
|
|
|
Setup receives integers a and b, and configures the parameters for the rangeproof scheme.
|
|
|
|
*/
|
2019-06-17 12:01:31 -07:00
|
|
|
pub fn setup<R: Rng>(rng: &mut R, a: i64, b: i64) -> Self {
|
2019-06-16 12:38:07 -07:00
|
|
|
// Compute optimal values for u and l
|
|
|
|
if a > b {
|
|
|
|
panic!("a must be less than or equal to b");
|
|
|
|
}
|
2019-06-28 14:44:14 -07:00
|
|
|
//TODO: optimize u?
|
|
|
|
let logb = (b as f64).log2();
|
2019-07-01 18:51:24 -07:00
|
|
|
let loglogb = logb.log2();
|
|
|
|
if loglogb > 0.0 {
|
|
|
|
let mut u = (logb / loglogb) as i64;
|
|
|
|
if u < 2 {
|
|
|
|
u = 2;
|
2019-06-16 12:38:07 -07:00
|
|
|
}
|
2019-07-01 18:51:24 -07:00
|
|
|
let l = (b as f64).log(u as f64).ceil() as i64;
|
|
|
|
let params_out: ParamsUL<E> = ParamsUL::<E>::setup_ul(rng, u, l);
|
|
|
|
return RPPublicParams { p: params_out, a, b };
|
2019-06-14 15:14:06 -07:00
|
|
|
} else {
|
2019-07-01 18:51:24 -07:00
|
|
|
panic!("log(log(b)) is zero");
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
|
|
|
}
|
2019-06-18 15:17:51 -07:00
|
|
|
|
|
|
|
/*
|
|
|
|
Prove method is responsible for generating the zero knowledge proof.
|
|
|
|
*/
|
|
|
|
pub fn prove<R: Rng>(&self, rng: &mut R, x: i64) -> RangeProof<E> {
|
2019-06-28 14:44:14 -07:00
|
|
|
if x > self.b || x < self.a {
|
|
|
|
panic!("x is not within the range.");
|
|
|
|
}
|
2019-06-18 15:17:51 -07:00
|
|
|
let ul = self.p.u.pow(self.p.l as u32);
|
|
|
|
let r = E::Fr::rand(rng);
|
|
|
|
|
|
|
|
// x - b + ul
|
2019-06-26 09:37:27 -07:00
|
|
|
let xb = x - self.b + ul;
|
2019-06-18 15:17:51 -07:00
|
|
|
let first = self.p.prove_ul(rng, xb, r);
|
|
|
|
|
|
|
|
// x - a
|
|
|
|
let xa = x - self.a;
|
|
|
|
let second = self.p.prove_ul(rng, xa, r);
|
|
|
|
|
|
|
|
return RangeProof { p1: first, p2: second };
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
Verify is responsible for validating the proof.
|
|
|
|
*/
|
|
|
|
pub fn verify(&self, proof: RangeProof<E>) -> bool {
|
|
|
|
let first = self.p.verify_ul(&proof.p1);
|
|
|
|
let second = self.p.verify_ul(&proof.p2);
|
|
|
|
return first && second;
|
|
|
|
}
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::*;
|
2019-06-18 13:20:16 -07:00
|
|
|
use pairing::bls12_381::{Bls12, G1, G2, Fq12, Fr};
|
2019-06-28 14:44:14 -07:00
|
|
|
use time::PreciseTime;
|
|
|
|
use std::ops::Add;
|
|
|
|
use core::mem;
|
2019-06-14 15:14:06 -07:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn setup_ul_works() {
|
2019-06-16 13:29:48 -07:00
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let params = ParamsUL::<Bls12>::setup_ul(rng, 2, 3);
|
2019-06-17 12:01:31 -07:00
|
|
|
assert_eq!(params.signatures.len(), 2);
|
2019-06-16 13:29:48 -07:00
|
|
|
for (m, s) in params.signatures {
|
2019-07-02 14:28:49 -07:00
|
|
|
assert_eq!(params.kp.verify(¶ms.mpk, &vec! {Fr::from_str(m.to_string().as_str()).unwrap()}, &Fr::zero(), &s), true);
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-06-16 13:29:48 -07:00
|
|
|
#[test]
|
|
|
|
fn prove_ul_works() {
|
|
|
|
let rng = &mut rand::thread_rng();
|
2019-06-28 14:44:14 -07:00
|
|
|
let params = ParamsUL::<Bls12>::setup_ul(rng, 2, 4);
|
2019-06-16 13:29:48 -07:00
|
|
|
let fr = Fr::rand(rng);
|
|
|
|
let proof = params.prove_ul(rng, 10, fr);
|
2019-06-28 14:44:14 -07:00
|
|
|
assert_eq!(proof.a.len(), 4);
|
|
|
|
assert_eq!(proof.V.len(), 4);
|
|
|
|
assert_eq!(proof.zsig.len(), 4);
|
|
|
|
assert_eq!(proof.zv.len(), 4);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_panic(expected = "x is not within the range")]
|
|
|
|
fn prove_ul_not_in_range() {
|
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let params = ParamsUL::<Bls12>::setup_ul(rng, 2, 3);
|
|
|
|
let fr = Fr::rand(rng);
|
|
|
|
params.prove_ul(rng, 100, fr);
|
2019-06-17 12:01:31 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn prove_and_verify_part1_ul_works() {
|
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let params = ParamsUL::<Bls12>::setup_ul(rng, 2, 4);
|
|
|
|
let fr = Fr::rand(rng);
|
|
|
|
let proof = params.prove_ul(rng, 10, fr);
|
|
|
|
assert_eq!(params.verify_part1(&proof), true);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn prove_and_verify_part2_ul_works() {
|
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let params = ParamsUL::<Bls12>::setup_ul(rng, 2, 4);
|
|
|
|
let fr = Fr::rand(rng);
|
|
|
|
let proof = params.prove_ul(rng, 10, fr);
|
|
|
|
assert_eq!(params.verify_part2(&proof), true);
|
|
|
|
}
|
2019-06-16 13:29:48 -07:00
|
|
|
|
2019-06-17 12:01:31 -07:00
|
|
|
#[test]
|
|
|
|
fn prove_and_verify_ul_works() {
|
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let params = ParamsUL::<Bls12>::setup_ul(rng, 2, 4);
|
|
|
|
let fr = Fr::rand(rng);
|
|
|
|
let proof = params.prove_ul(rng, 10, fr);
|
|
|
|
assert_eq!(params.verify_ul(&proof), true);
|
2019-06-16 13:29:48 -07:00
|
|
|
}
|
|
|
|
|
2019-06-18 15:17:51 -07:00
|
|
|
#[test]
|
|
|
|
fn prove_and_verify_works() {
|
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let params = RPPublicParams::<Bls12>::setup(rng, 2, 25);
|
|
|
|
let proof = params.prove(rng, 10);
|
|
|
|
assert_eq!(params.verify(proof), true);
|
|
|
|
}
|
|
|
|
|
2019-06-28 14:44:14 -07:00
|
|
|
#[test]
|
|
|
|
#[should_panic(expected = "x is not within the range")]
|
|
|
|
fn prove_not_in_range() {
|
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let params = RPPublicParams::<Bls12>::setup(rng, 2, 25);
|
|
|
|
let proof = params.prove(rng, 26);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[ignore]
|
|
|
|
fn prove_and_verify_performance() {
|
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let mut averageSetup = time::Duration::nanoseconds(0);
|
|
|
|
let mut averageSetupSize = 0;
|
|
|
|
let mut averageProve = time::Duration::nanoseconds(0);
|
|
|
|
let mut averageProofSize = 0;
|
|
|
|
let mut averageVerify = time::Duration::nanoseconds(0);
|
|
|
|
let iter = 5;
|
|
|
|
for i in 0..iter {
|
|
|
|
let a = rng.gen_range(0, 1000000);
|
|
|
|
let b = rng.gen_range(a, 1000000);
|
|
|
|
let x = rng.gen_range(a, b);
|
|
|
|
|
|
|
|
let sSetup = PreciseTime::now();
|
|
|
|
let params = RPPublicParams::<Bls12>::setup(rng, a, b);
|
|
|
|
averageSetup = averageSetup.add(sSetup.to(PreciseTime::now()));
|
|
|
|
averageSetupSize += mem::size_of_val(¶ms);
|
|
|
|
|
|
|
|
let sProve = PreciseTime::now();
|
|
|
|
let proof = params.prove(rng, x);
|
|
|
|
averageProve = averageProve.add(sProve.to(PreciseTime::now()));
|
|
|
|
averageProofSize += mem::size_of_val(&proof);
|
|
|
|
|
|
|
|
let sVerify = PreciseTime::now();
|
|
|
|
params.verify(proof);
|
|
|
|
averageVerify = averageVerify.add(sVerify.to(PreciseTime::now()));
|
|
|
|
}
|
|
|
|
print!("Setup: {}\n", averageSetup.num_milliseconds() / iter);
|
|
|
|
print!("Setup size: {}\n", averageSetupSize / iter as usize);
|
|
|
|
print!("Prove: {}\n", averageProve.num_milliseconds() / iter);
|
|
|
|
print!("Proof size: {}\n", averageProofSize / iter as usize);
|
|
|
|
print!("Verify: {}\n", averageVerify.num_milliseconds() / iter);
|
|
|
|
}
|
|
|
|
|
2019-06-14 15:14:06 -07:00
|
|
|
#[test]
|
|
|
|
fn decompose_works() {
|
2019-06-17 12:01:31 -07:00
|
|
|
assert_eq!(decompose(25, 3, 3), vec! {1, 2, 2});
|
|
|
|
assert_eq!(decompose(336, 7, 3), vec! {0, 6, 6});
|
|
|
|
assert_eq!(decompose(285, 8, 3), vec! {5, 3, 4});
|
|
|
|
assert_eq!(decompose(125, 13, 2), vec! {8, 9});
|
|
|
|
assert_eq!(decompose(143225, 6, 7), vec! {5, 2, 0, 3, 2, 0, 3});
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn decompose_recompose_works() {
|
|
|
|
let vec1 = decompose(25, 3, 5);
|
|
|
|
let mut result = 0;
|
|
|
|
for i in 0..5 {
|
|
|
|
result += vec1[i] * 3i64.pow(i as u32);
|
|
|
|
}
|
|
|
|
assert_eq!(result, 25);
|
|
|
|
|
|
|
|
let vec1 = decompose(143225, 6, 7);
|
|
|
|
let mut result = 0;
|
|
|
|
for i in 0..7 {
|
|
|
|
result += vec1[i] * 6i64.pow(i as u32);
|
|
|
|
}
|
|
|
|
assert_eq!(result, 143225);
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn setup_works() {
|
2019-06-16 13:29:48 -07:00
|
|
|
let rng = &mut rand::thread_rng();
|
|
|
|
let public_params = RPPublicParams::<Bls12>::setup(rng, 2, 10);
|
2019-06-17 12:01:31 -07:00
|
|
|
assert_eq!(public_params.a, 2);
|
|
|
|
assert_eq!(public_params.b, 10);
|
2019-07-01 18:51:24 -07:00
|
|
|
assert_eq!(public_params.p.signatures.len(), 2);
|
|
|
|
assert_eq!(public_params.p.u, 2);
|
|
|
|
assert_eq!(public_params.p.l, 4);
|
2019-06-14 15:14:06 -07:00
|
|
|
for (m, s) in public_params.p.signatures {
|
2019-07-02 14:28:49 -07:00
|
|
|
assert_eq!(public_params.p.kp.verify(&public_params.p.mpk, &vec! {Fr::from_str(m.to_string().as_str()).unwrap()}, &Fr::zero(), &s), true);
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
#[should_panic(expected = "a must be less than or equal to b")]
|
|
|
|
fn setup_wrong_a_and_b() {
|
2019-06-16 13:29:48 -07:00
|
|
|
let rng = &mut rand::thread_rng();
|
2019-06-26 10:33:08 -07:00
|
|
|
RPPublicParams::<Bls12>::setup(rng, 10, 2);
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2019-07-01 18:51:24 -07:00
|
|
|
#[should_panic(expected = "log(log(b)) is zero")]
|
2019-06-14 15:14:06 -07:00
|
|
|
fn setup_wrong_logb() {
|
2019-06-16 13:29:48 -07:00
|
|
|
let rng = &mut rand::thread_rng();
|
2019-07-01 18:51:24 -07:00
|
|
|
RPPublicParams::<Bls12>::setup(rng, -2, -1);
|
2019-06-16 12:38:07 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn fmt_byte_to_int_works() {
|
2019-06-17 12:01:31 -07:00
|
|
|
assert_eq!(fmt_bytes_to_int([12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123, 13, 43, 12, 235, 23, 123]),
|
|
|
|
"122352312313431223523123134312235231231343122352312313431223523123134312235231231343122352312313431223523123134312235231231343122352312313431223523123");
|
2019-06-16 12:38:07 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn hash_works() {
|
2019-06-16 13:29:48 -07:00
|
|
|
let rng = &mut rand::thread_rng();
|
2019-06-16 12:38:07 -07:00
|
|
|
let D = G2::rand(rng);
|
|
|
|
let D2 = G2::rand(rng);
|
|
|
|
let a = vec! {Fq12::rand(rng), Fq12::rand(rng), Fq12::rand(rng)};
|
|
|
|
let a2 = vec! {Fq12::rand(rng), Fq12::rand(rng), Fq12::rand(rng)};
|
2019-06-17 12:01:31 -07:00
|
|
|
assert_eq!(Hash::<Bls12>(a.clone(), D.clone()).is_zero(), false);
|
2019-06-16 12:38:07 -07:00
|
|
|
assert_ne!(Hash::<Bls12>(a2.clone(), D.clone()), Hash::<Bls12>(a.clone(), D.clone()));
|
|
|
|
assert_ne!(Hash::<Bls12>(a.clone(), D2.clone()), Hash::<Bls12>(a.clone(), D.clone()));
|
|
|
|
assert_ne!(Hash::<Bls12>(a2.clone(), D2.clone()), Hash::<Bls12>(a.clone(), D.clone()));
|
2019-06-14 15:14:06 -07:00
|
|
|
}
|
|
|
|
}
|