111 lines
3.5 KiB
Rust
111 lines
3.5 KiB
Rust
use crate::cli_constants::MPC_DEFAULT_PATH;
|
|
use crate::common::get_seeded_rng;
|
|
use clap::{App, Arg, ArgMatches, SubCommand};
|
|
use rand_0_8::{rngs::StdRng, SeedableRng};
|
|
use std::fs::File;
|
|
use std::io::{BufReader, BufWriter};
|
|
use std::path::Path;
|
|
use zokrates_bellman::Bellman;
|
|
use zokrates_common::constants::{BLS12_381, BN128};
|
|
use zokrates_field::{BellmanFieldExtensions, Bls12_381Field, Bn128Field, Field};
|
|
use zokrates_proof_systems::{MpcBackend, MpcScheme, G16};
|
|
|
|
pub fn subcommand() -> App<'static, 'static> {
|
|
SubCommand::with_name("contribute")
|
|
.about("Contributes to a ceremony")
|
|
.arg(
|
|
Arg::with_name("input")
|
|
.short("i")
|
|
.long("input")
|
|
.help("Path of the MPC parameters")
|
|
.value_name("FILE")
|
|
.takes_value(true)
|
|
.required(false)
|
|
.default_value(MPC_DEFAULT_PATH),
|
|
)
|
|
.arg(
|
|
Arg::with_name("entropy")
|
|
.short("e")
|
|
.long("entropy")
|
|
.help("User provided randomness")
|
|
.takes_value(true)
|
|
.required(false),
|
|
)
|
|
.arg(
|
|
Arg::with_name("curve")
|
|
.short("c")
|
|
.long("curve")
|
|
.help("Curve used in the ceremony")
|
|
.takes_value(true)
|
|
.required(false)
|
|
.possible_values(&[BN128, BLS12_381])
|
|
.default_value(BN128),
|
|
)
|
|
.arg(
|
|
Arg::with_name("output")
|
|
.short("o")
|
|
.long("output")
|
|
.help("Path of the output file")
|
|
.value_name("FILE")
|
|
.takes_value(true)
|
|
.required(false)
|
|
.default_value(MPC_DEFAULT_PATH),
|
|
)
|
|
}
|
|
|
|
pub fn exec(sub_matches: &ArgMatches) -> Result<(), String> {
|
|
match sub_matches.value_of("curve").unwrap() {
|
|
BN128 => cli_mpc_contribute::<Bn128Field, G16, Bellman>(sub_matches),
|
|
BLS12_381 => cli_mpc_contribute::<Bls12_381Field, G16, Bellman>(sub_matches),
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
pub fn cli_mpc_contribute<
|
|
T: Field + BellmanFieldExtensions,
|
|
S: MpcScheme<T>,
|
|
B: MpcBackend<T, S>,
|
|
>(
|
|
sub_matches: &ArgMatches,
|
|
) -> Result<(), String> {
|
|
let path = Path::new(sub_matches.value_of("input").unwrap());
|
|
let file =
|
|
File::open(&path).map_err(|why| format!("Could not open `{}`: {}", path.display(), why))?;
|
|
|
|
let mut reader = BufReader::new(file);
|
|
|
|
let output_path = Path::new(sub_matches.value_of("output").unwrap());
|
|
let output_file = File::create(&output_path)
|
|
.map_err(|why| format!("Could not create `{}`: {}", output_path.display(), why))?;
|
|
|
|
let mut writer = BufWriter::new(output_file);
|
|
|
|
println!("Contributing to `{}`...", path.display());
|
|
|
|
let mut rng = sub_matches
|
|
.value_of("entropy")
|
|
.map(get_seeded_rng)
|
|
.unwrap_or_else(StdRng::from_entropy);
|
|
|
|
let hash = B::contribute(&mut reader, &mut rng, &mut writer)
|
|
.map_err(|e| format!("Failed to contribute: {}", e))?;
|
|
println!("The BLAKE2b hash of your contribution is:\n");
|
|
|
|
for line in hash.chunks(16) {
|
|
print!("\t");
|
|
for section in line.chunks(4) {
|
|
for b in section {
|
|
print!("{:02x}", b);
|
|
}
|
|
print!(" ");
|
|
}
|
|
println!();
|
|
}
|
|
|
|
println!(
|
|
"\nYour contribution has been written to `{}`",
|
|
output_path.display()
|
|
);
|
|
|
|
Ok(())
|
|
}
|