Эх сурвалжийг харах

[zk/circuit/lead_contract] removed redundant serial number

mohab 4 жил өмнө
parent
commit
762d80ac8b

+ 111 - 72
example/lead.rs

@@ -30,9 +30,11 @@ use darkfi::{
     },
     },
 };
 };
 
 
+use pasta_curves::group::Curve;
+use pasta_curves::arithmetic::CurveAffine;
+//use halo2_proofs::arithmetic::CurveAffine;
 
 
-
-#[derive(Debug,Default,Clone)]
+#[derive(Debug,Default,Clone,Copy)]
 pub struct Coin
 pub struct Coin
 {
 {
     value : Option<pallas::Base>, //stake
     value : Option<pallas::Base>, //stake
@@ -42,9 +44,9 @@ pub struct Coin
     sl : Option<pallas::Base>, //slot id
     sl : Option<pallas::Base>, //slot id
     tau : Option<pallas::Base>,
     tau : Option<pallas::Base>,
     nonce : Option<pallas::Base>,
     nonce : Option<pallas::Base>,
-    sn : Option<pallas::Base>, // coin's serial number
-    sk : Option<SecretKey>,
-    pk : Option<PublicKey>,
+    sn : Option<pallas::Point>, // coin's serial number
+    //sk : Option<SecretKey>,
+    pk : Option<pallas::Point>,
     root_cm : Option<pallas::Scalar>,
     root_cm : Option<pallas::Scalar>,
     root_sk : Option<pallas::Scalar>,
     root_sk : Option<pallas::Scalar>,
     path: Option<[MerkleNode; MERKLE_DEPTH_ORCHARD]>,
     path: Option<[MerkleNode; MERKLE_DEPTH_ORCHARD]>,
@@ -62,34 +64,36 @@ fn main()
     //TODO construct a tree of multiple coins
     //TODO construct a tree of multiple coins
     const LEN : usize = 10;
     const LEN : usize = 10;
     let mut rng = thread_rng();
     let mut rng = thread_rng();
-    let sks : Vec<u64> = vec![];
-    let root_sks : Vec<MerkleNode> = vec![];
-    let path_sks : Vec<[MerkleNode;MERKLE_DEPTH_ORCHARD]> = vec![];
-    let tree  =  BridgeTree::<MerkleNode, 32>::new(LEN);
+    let mut sks : Vec<u64> = vec![];
+    let mut root_sks : Vec<MerkleNode> = vec![];
+    let mut path_sks : Vec<[MerkleNode;MERKLE_DEPTH_ORCHARD]> = vec![];
+    let mut tree  =  BridgeTree::<MerkleNode, 32>::new(LEN);
     for i in 0..LEN {
     for i in 0..LEN {
         let tmp : u64 = rng.gen();
         let tmp : u64 = rng.gen();
-        let sk : u64 = tmp;
-        sks.push(sk);
+        let mut sk : u64 = tmp;
+        sks.push(sk.clone());
         let node = MerkleNode(pallas::Base::from(sk));
         let node = MerkleNode(pallas::Base::from(sk));
-        tree.append(&node);
+        tree.append(&node.clone());
         let (leaf_pos, path) = tree.authentication_path(&node).unwrap();
         let (leaf_pos, path) = tree.authentication_path(&node).unwrap();
-        root_sks.push(tree.root());
-        path_sks.push(path);
+        root_sks.push(tree.root().clone());
+        path_sks.push(path.as_slice().try_into().unwrap());
     }
     }
-    let seeds : Vec<u64> = vec![];
+    let mut seeds : Vec<u64> = vec![];
     for i in 0..LEN {
     for i in 0..LEN {
         let rho : u64 = rng.gen();
         let rho : u64 = rng.gen();
-        seeds.push(rho);
+        seeds.push(rho.clone());
     }
     }
     //
     //
-    let mau_y : pallas::Scalar = pallas::Scalar::from(rng.gen());
-    let mau_rho : pallas::Scalar = pallas::Scalar::from(rng.gen());
+    let yu64 : u64 = rng.gen();
+    let rhou64 : u64 = rng.gen();
+    let mau_y : pallas::Scalar = pallas::Scalar::from(yu64);
+    let mau_rho : pallas::Scalar = pallas::Scalar::from(rhou64);
 
 
     //
     //
-    let coins : Vec<Coin> = vec![];
+    let mut coins : Vec<Coin> = vec![];
 
 
     //
     //
-    let tree_cm = BridgeTree::<MerkleNode, 32>::new(LEN);
+    let mut tree_cm = BridgeTree::<MerkleNode, 32>::new(LEN);
     let zerou64 : u64 = 0;
     let zerou64 : u64 = 0;
     for i in 0..LEN {
     for i in 0..LEN {
         let c_v = pallas::Base::from(u64::try_from(i*2).unwrap());
         let c_v = pallas::Base::from(u64::try_from(i*2).unwrap());
@@ -99,35 +103,46 @@ fn main()
         let iu64 : u64 = u64::try_from(i).unwrap();
         let iu64 : u64 = u64::try_from(i).unwrap();
         let c_sl  = pallas::Base::from(iu64);
         let c_sl  = pallas::Base::from(iu64);
 
 
+        let c_tau  = pallas::Base::from(u64::try_from(i).unwrap()); // let's assume it's sl for simplicity
+        let c_root_sk : MerkleNode  = root_sks[i];
+        // =========================
         //TODO 512 secret-key/public-key to cop with pallas curves
         //TODO 512 secret-key/public-key to cop with pallas curves
+        // =========================
         //note! sk is used in MerkleNode takes pallas::Base as input
         //note! sk is used in MerkleNode takes pallas::Base as input
         //while the pallas::base is 512, the SecretKey is  of size 256, a larger keyring is needed
         //while the pallas::base is 512, the SecretKey is  of size 256, a larger keyring is needed
         //TODO what is the endianess of this keyring
         //TODO what is the endianess of this keyring
-        let sk_bits = vec![];
-        sk_bits.append(&mut c_sk.to_le_bytes().to_vec());
-        sk_bits.append(&mut zerou64.to_le_bytes().to_vec());
-        sk_bits.append(&mut zerou64.to_le_bytes().to_vec());
-        sk_bits.append(&mut zerou64.to_le_bytes().to_vec());
-        let c_pk = PublicKey::from_secret(SecretKey::from_bytes(sk_bits.as_slice().try_into().unwrap()).unwrap());
-        let c_tau  = pallas::Base::from(u64::try_from(i).unwrap()); // let's assume it's sl for simplicity
-        let c_root_sk : MerkleNode  = root_sks[i];
+        //let sk_bits = vec![];
+        //sk_bits.append(&mut c_sk.to_le_bytes().to_vec());
+        //sk_bits.append(&mut zerou64.to_le_bytes().to_vec());
+        //sk_bits.append(&mut zerou64.to_le_bytes().to_vec());
+        //sk_bits.append(&mut zerou64.to_le_bytes().to_vec());
+        //let c_pk = PublicKey::from_secret(SecretKey::from_bytes(sk_bits.as_slice().try_into().unwrap()).unwrap());
+        let c_pk = pedersen_commitment_scalar(mod_r_p(c_tau), mod_r_p(c_root_sk.inner()));
+        //
+        // TODO (fix) no use random value for the secret key as random pallas base
+        // =======================
+
+
         let c_seed  = pallas::Base::from(seeds[i]);
         let c_seed  = pallas::Base::from(seeds[i]);
-        let c_sn  = pedersen_commitment_base(c_seed, c_root_sk);
-        let c_cm_message = [c_pk.clone(), c_v.clone(), c_seed.clone()];
-        let c_cm_v = poseidon::Hash::<_,P128Pow5T3, ConstantLength<6>, 3, 2>::init().hash(c_cm_message);
+        let c_sn  = pedersen_commitment_scalar(mod_r_p(c_seed), mod_r_p(c_root_sk.inner()));
+        let c_pk_pt = c_pk.to_affine().coordinates().unwrap();
+        let c_cm_message = [*c_pk_pt.x(), *c_pk_pt.y(), c_v.clone(), c_seed.clone()];
+        let c_cm_v = poseidon::Hash::<_,P128Pow5T3, ConstantLength<4>, 3, 2>::init().hash(c_cm_message);
         let c_cm1_blind = pallas::Base::from(0); //tmp val
         let c_cm1_blind = pallas::Base::from(0); //tmp val
         let c_cm2_blind = pallas::Base::from(0); //tmp val
         let c_cm2_blind = pallas::Base::from(0); //tmp val
-        let c_cm : pallas::Point  = pedersen_commitment_scalar(c_cm_v, c_cm1_blind);
-        let c_cm_node = MerkleNode(c_cm);
-        tree_cm.append(&c_cm_node);
+        let c_cm : pallas::Point  = pedersen_commitment_scalar(mod_r_p(c_cm_v), mod_r_p(c_cm1_blind));
+        //TODO (fix) which affine coefficient point to be used a/b ?
+        let c_cm_node = MerkleNode(*c_cm.to_affine().coordinates().unwrap().x()); //CurveAffine::a()
+        tree_cm.append(&c_cm_node.clone());
         let (leaf_pos, c_cm_path) = tree_cm.authentication_path(&c_cm_node).unwrap();
         let (leaf_pos, c_cm_path) = tree_cm.authentication_path(&c_cm_node).unwrap();
         let c_root_cm = tree_cm.root();
         let c_root_cm = tree_cm.root();
         // lead coin commitment
         // lead coin commitment
         //TODO this c_v can be
         //TODO this c_v can be
-        let c_seed2 = pedersen_commitment_u64(c_seed, c_root_sk);
-        let lead_coin_msg = [c_pk, c_v, c_seed2];
-        poseidon::Hash::<_,P128Pow5T3, ConstantLength<6>, 3, 2>::init().hash(lead_coin_msg);
-        let c_cm2 = pedersen_commitment_u64(lead_coin_msg, c_seed2);
+        let c_seed2 = pedersen_commitment_scalar(mod_r_p(c_seed), mod_r_p(c_root_sk.inner()));
+        let c_seed2_pt = c_seed2.to_affine().coordinates().unwrap();
+        let lead_coin_msg = [*c_pk_pt.x(), *c_pk_pt.y(), c_v, *c_seed2_pt.x(), *c_seed2_pt.y()];
+        let lead_coin_msg_hash = poseidon::Hash::<_,P128Pow5T3, ConstantLength<5>, 3, 2>::init().hash(lead_coin_msg);
+        let c_cm2 = pedersen_commitment_scalar(mod_r_p(lead_coin_msg_hash), mod_r_p(c_cm2_blind));
         let c_root_sk = root_sks[i];
         let c_root_sk = root_sks[i];
         let c_path_sk = path_sks[i];
         let c_path_sk = path_sks[i];
         let coin  = Coin {
         let coin  = Coin {
@@ -139,11 +154,11 @@ fn main()
             tau: Some(c_tau),
             tau: Some(c_tau),
             nonce: Some(c_seed),
             nonce: Some(c_seed),
             sn:  Some(c_sn),
             sn:  Some(c_sn),
-            sk: Some(c_sk),
+            //sk: Some(c_sk),
             pk: Some(c_pk),
             pk: Some(c_pk),
-            root_cm: Some(c_root_cm),
-            root_sk: Some(c_root_sk),
-            path: Some(c_cm_path),
+            root_cm: Some(mod_r_p(c_root_cm.inner())),
+            root_sk: Some(mod_r_p(c_root_sk.inner())),
+            path: Some(c_cm_path.as_slice().try_into().unwrap()),
             path_sk: Some(c_path_sk),
             path_sk: Some(c_path_sk),
             opening1: Some(c_cm1_blind),
             opening1: Some(c_cm1_blind),
             opening2: Some(c_cm2_blind),
             opening2: Some(c_cm2_blind),
@@ -151,47 +166,71 @@ fn main()
         coins.push(coin);
         coins.push(coin);
     }
     }
 
 
-    let coin_idx = 0;
+    // ================
+    // public inputs
+    // ================
+    let coin_idx  = 0;
     let coin = coins[coin_idx];
     let coin = coins[coin_idx];
+    let c0 = pedersen_commitment_scalar(mod_r_p(coin.nonce.unwrap()), coin.root_cm.unwrap())
+        .to_affine()
+        .coordinates()
+        .unwrap();
+    let c1 = pedersen_commitment_scalar(mod_r_p(coin.tau.unwrap()), coin.root_cm.unwrap())
+        .to_affine()
+        .coordinates()
+        .unwrap();
+    //TODO root_cm need to be converted to Fp
+    let c2 = pedersen_commitment_scalar(mod_r_p(coin.nonce.unwrap()), coin.root_cm.unwrap())
+        .to_affine()
+        .coordinates()
+        .unwrap();
+   //
+    let c3 = coin.cm.unwrap().to_affine().coordinates().unwrap();
+    let c4 = coin.cm2.unwrap().to_affine().coordinates().unwrap();
+
+    //TODO (fix) public key is a commitemnet of the root of secret key, and the timestamp
+    let c7 = coin.pk.unwrap().to_affine().coordinates().unwrap();
+    let c8 = coin.sn.unwrap().to_affine().coordinates().unwrap();
+    //TODO (fix) this need to be replaced by computed final path as pallas::Base
+    let c5 = coin.path.unwrap();
+    let c6 = pallas::Base::from(0);
+    // ===============
+
     let path_sk = path_sks[coin_idx];
     let path_sk = path_sks[coin_idx];
     let contract = LeadContract {
     let contract = LeadContract {
-        path: Some(coin.path),
-        root_sk: Some(coin.root_sk), //TODO where doesn' this come from?
+        path: coin.path,
+        root_sk: coin.root_sk,
         path_sk: Some(path_sk),
         path_sk: Some(path_sk),
-        coin_timestamp: Some(pallas::Base::from(coin.tau)), //
-        coin_nonce: Some(pallas::Base::from(coin.nonce)),
-        coin_opening_1: Some(coin.opening1),
-        value: Some(coin.value),
-        coin_opening_2: Some(coin.opening2),
-        cm_c1_x: Some(coin.cm.0.x),
-        cm_c1_y: Some(coin.cm.0.y),
-        cm_c2_x: Some(coin.cm2.0.x),
-        cm_c2_y: Some(coin.cm2.0.y),
-        cm_pos : Some(coin_idx.unwrap()),
-        sn_c1: Some(coin.sn.unwrap()),
+        coin_timestamp: coin.tau, //
+        coin_nonce: coin.nonce,
+        coin_opening_1: Some(mod_r_p(coin.opening1.unwrap())),
+        value: coin.value,
+        coin_opening_2: Some(mod_r_p(coin.opening2.unwrap())),
+        cm_c1_x: Some(*c3.x()),
+        cm_c1_y: Some(*c3.y()),
+        cm_c2_x: Some(*c4.x()),
+        cm_c2_y: Some(*c4.y()),
+        cm_pos : Some(u32::try_from(coin_idx).unwrap()),
+        //sn_c1: Some(coin.sn.unwrap()),
         slot: Some(coin.sl.unwrap()),
         slot: Some(coin.sl.unwrap()),
         mau_rho: Some(mau_rho.clone()),
         mau_rho: Some(mau_rho.clone()),
         mau_y: Some(mau_y.clone()),
         mau_y: Some(mau_y.clone()),
         root_cm: Some(coin.root_cm.unwrap()),
         root_cm: Some(coin.root_cm.unwrap()),
     };
     };
-    //public inputs
-    let c0 = pedersen_commitment_scalar(mod_r_p(coin.nonce.unwrap()), coin.root_cm.unwrap());
-    let c1 = pedersen_commitment_scalar(mod_r_p(coin.tau.unwrap()), coin.root_cm.unwrap());
-    //TODO root_cm need to be converted to Fp
-    let c2 = pedersen_commitment_scalar(mod_r_p(coin.nonce.unwrap()), coin.root_cm.unwrap());
-    let c3 = coin.cm.unwrap();
-    let c4 = coin.cm2.unwrap();
-    let c5 = coin.path.unwrap();
-    let c6 = pallas::Base::from(0);
-    let mut public_inputs = vec![c0.x, c0.y,
-                                 c1.x, c1.y,
-                                 c2.x, c2.y,
-                                 c3.x, c3.y,
-                                 c4.x(), c4.y(),
-                                 c5,
+
+    let mut public_inputs = vec![*c0.x(), *c0.y(),
+                                 *c1.x(), *c1.y(),
+                                 *c2.x(), *c2.y(),
+                                 *c7.x(), *c7.y(),
+                                 *c8.x(), *c8.y(),
+                                 *c3.x(), *c3.y(),
+                                 *c4.x(), *c4.y(),
+                                 c5[31].inner(), //TODO (res) how the path is structured assumed root is last node in the path.
                                  c6,
                                  c6,
     ];
     ];
+
+    let mut vec_inputs = vec![public_inputs];
     //TODO
     //TODO
-    let prover = MockProver::run(k, &contract, vec![public_inputs]).unwrap();
+    let prover = MockProver::run(k, &contract, vec_inputs).unwrap();
     assert_eq!(prover.verify(), Ok(()));
     assert_eq!(prover.verify(), Ok(()));
 }
 }

+ 3 - 2
src/zk/circuit/lead_contract.rs

@@ -143,7 +143,7 @@ pub struct LeadContract {
     //
     //
     pub cm_pos : Option<u32>,
     pub cm_pos : Option<u32>,
     //
     //
-    pub sn_c1 : Option<pallas::Base>,
+    //pub sn_c1 : Option<pallas::Base>,
     pub slot : Option<pallas::Base>,
     pub slot : Option<pallas::Base>,
     pub mau_rho: Option<pallas::Scalar>,
     pub mau_rho: Option<pallas::Scalar>,
     pub mau_y: Option<pallas::Scalar>,
     pub mau_y: Option<pallas::Scalar>,
@@ -367,12 +367,13 @@ impl Circuit<pallas::Base> for LeadContract {
             config.advices[0],
             config.advices[0],
             self.cm_pos
             self.cm_pos
         )?;
         )?;
-         */
+
         let sn_c1 = self.load_private(
         let sn_c1 = self.load_private(
             layouter.namespace(|| ""),
             layouter.namespace(|| ""),
             config.advices[0],
             config.advices[0],
             self.sn_c1,
             self.sn_c1,
         )?;
         )?;
+        */
 
 
         /*
         /*
         let eta = self.load_private(
         let eta = self.load_private(