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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
use zkvms_host_io::{PublicInput, PrivateInput, foreach_public_input_field, foreach_private_input_field, read_args, RunType::{Execute, Prove, Verify}};
use std::io::{self, Write};
use std::process::{Command, Stdio};
use regex::Regex;
fn build_public_input(input: &PublicInput) -> String {
let numreg: Regex = Regex::new("(?:^|[^A-Za-z])([0-9]+)").unwrap();
let mut ret = String::new();
foreach_public_input_field!{
let flat = format!("{:?}", input.yield)
.replace("false", "0")
.replace("true", "1");
let numbers: Vec<&str> = numreg
.captures_iter(&flat)
.map(|cap| cap.get(1).unwrap().as_str())
.collect();
for num in numbers {
ret.push_str(num);
ret.push_str(":i64,");
}
}
ret.pop(); // removes trailing comma
ret
}
fn build_private_input(input: &PrivateInput) -> String {
let numreg: Regex = Regex::new("(?:^|[^A-Za-z])([0-9]+)").unwrap();
let mut ret = String::new();
foreach_private_input_field!{
let flat = format!("{:?}", input.yield)
.replace("false", "0")
.replace("true", "1");
let numbers: Vec<&str> = numreg
.captures_iter(&flat)
.map(|cap| cap.get(1).unwrap().as_str())
.collect();
for num in numbers {
ret.push_str(num);
ret.push_str(":i64,");
}
}
ret.pop(); // removes trailing comma
ret
}
fn zkwasm_command(subcmd: &str) -> Command {
let mut command = Command::new("zkwasm-cli");
command
.arg("--params").arg("./params")
.arg("prog").arg(subcmd)
.arg("--wasm").arg(env!("GUEST_PATH"));
command
}
fn run(cmd: &mut Command) {
assert!(cmd.status().expect("couldn't execute command!").success());
}
fn main() {
let run_info = read_args();
let k = run_info
.env_or(
"ZKWASM_K",
"19",
);
let scheme = run_info
.env_or(
"ZKWASM_SCHEME",
"shplonk",
);
run(zkwasm_command("setup")
.arg("-k").arg(k)
.arg("--scheme").arg(scheme));
let public_input = build_public_input(&run_info.public_input);
let private_input = build_private_input(&run_info.private_input);
let output = run_info
.env_or(
"ZKWASM_OUTPUT",
"./output",
);
let params = run_info
.env_or(
"ZKWASM_PARAMS",
"./params",
);
match run_info.run_type {
Execute => {
run(zkwasm_command("dry-run")
.arg("--public").arg(public_input)
.arg("--private").arg(private_input)
.arg("--output").arg(output));
},
Prove => {
run(zkwasm_command("prove")
.arg("--public").arg(public_input)
.arg("--private").arg(private_input)
.arg("--output").arg(output));
},
Verify => {
run(zkwasm_command("prove")
.arg("--public").arg(public_input)
.arg("--private").arg(private_input)
.arg("--output").arg(output.clone()));
run(Command::new("zkwasm-cli")
.arg("--params").arg(params)
.arg("prog").arg("verify")
.arg("--output").arg(output));
},
}
}
|
