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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

#![allow(unused_imports)]

#[macro_use]
extern crate serde;
extern crate bellman_ce;
extern crate rand;
extern crate byteorder;
extern crate blake2_rfc;
extern crate num_cpus;
extern crate crossbeam;
extern crate num_bigint;
extern crate num_traits;
extern crate cfg_if;
extern crate itertools;
extern crate blake2;

use cfg_if::cfg_if;

pub mod keypair;
pub mod keypair_assembly;
pub mod hash_writer;
pub mod parameters;
pub mod utils;
pub mod circom_circuit;

cfg_if! {
    if #[cfg(feature = "wasm")] {
        extern crate serde_json;
        extern crate js_sys;
        extern crate web_sys;
        extern crate wasm_bindgen;
        extern crate console_error_panic_hook;

        use wasm_bindgen::prelude::*;
        use itertools::Itertools;
        use parameters::MPCParameters;
        use std::io::{
            Read,
            Write,
        };

        macro_rules! log {
            ($($t:tt)*) => (web_sys::console::log_1(&format_args!($($t)*).to_string().into()))
        }

        #[wasm_bindgen]
        pub fn contribute(params: Vec<u8>, entropy: Vec<u8>) -> Result<Vec<u8>, JsValue> {
            console_error_panic_hook::set_once();
            let disallow_points_at_infinity = false;

            log!("Initializing phase2");
            // Create an RNG based on provided randomness
            let mut rng = {
                use byteorder::{BigEndian, ReadBytesExt};
                use blake2::{Blake2b, Digest};
                use rand::{SeedableRng};
                use rand::chacha::ChaChaRng;
                
                let h = {
                    let mut h = Blake2b::default();
                    h.input(&*entropy);
                    h.result()
                };
                let mut digest = &h[..];
                
                // Interpret the first 32 bytes of the digest as 8 32-bit words
                let mut seed = [0u32; 8];
                for i in 0..8 {
                    seed[i] = digest.read_u32::<BigEndian>().expect("digest is large enough for this to work");
                }
                
                ChaChaRng::from_seed(&seed)
            };
        
            let mut params = MPCParameters::read(&*params, disallow_points_at_infinity, true).expect("unable to read params");

            log!("Contributing...");
            let zero: u32 = 0;
            let hash = params.contribute(&mut rng, &zero);
            log!("Contribution hash: 0x{:02x}", hash.iter().format(""));

            let mut output: Vec<u8> = vec![];
            params.write(&mut output).expect("failed to write updated parameters");
            log!("Returning parameters");
            Ok(output)
        }
    }
}