63 lines
1.9 KiB
JavaScript
63 lines
1.9 KiB
JavaScript
const { wtns, groth16 } = require('snarkjs')
|
|
const { utils } = require('ffjavascript')
|
|
|
|
const fs = require('fs')
|
|
const tmp = require('tmp-promise')
|
|
const util = require('util')
|
|
const exec = util.promisify(require('child_process').exec)
|
|
|
|
const { toFixedHex } = require('./utils')
|
|
|
|
async function prove(input, keyBasePath) {
|
|
const { proof } = await groth16.fullProve(
|
|
utils.stringifyBigInts(input),
|
|
`${keyBasePath}.wasm`,
|
|
`${keyBasePath}.zkey`,
|
|
)
|
|
return (
|
|
'0x' +
|
|
toFixedHex(proof.pi_a[0]).slice(2) +
|
|
toFixedHex(proof.pi_a[1]).slice(2) +
|
|
toFixedHex(proof.pi_b[0][1]).slice(2) +
|
|
toFixedHex(proof.pi_b[0][0]).slice(2) +
|
|
toFixedHex(proof.pi_b[1][1]).slice(2) +
|
|
toFixedHex(proof.pi_b[1][0]).slice(2) +
|
|
toFixedHex(proof.pi_c[0]).slice(2) +
|
|
toFixedHex(proof.pi_c[1]).slice(2)
|
|
)
|
|
}
|
|
|
|
function proveZkutil(input, keyBasePath) {
|
|
input = utils.stringifyBigInts(input)
|
|
// console.log('input', input)
|
|
return tmp.dir().then(async (dir) => {
|
|
dir = dir.path
|
|
let out
|
|
|
|
try {
|
|
await wtns.debug(
|
|
utils.unstringifyBigInts(input),
|
|
`${keyBasePath}.wasm`,
|
|
`${dir}/witness.wtns`,
|
|
`${keyBasePath}.sym`,
|
|
{},
|
|
console,
|
|
)
|
|
const witness = utils.stringifyBigInts(await wtns.exportJson(`${dir}/witness.wtns`))
|
|
fs.writeFileSync(`${dir}/witness.json`, JSON.stringify(witness, null, 2))
|
|
|
|
out = await exec(
|
|
`zkutil prove -c ${keyBasePath}.r1cs -p ${keyBasePath}.params -w ${dir}/witness.json -r ${dir}/proof.json -o ${dir}/public.json`,
|
|
)
|
|
// todo catch inconsistent input during witness generation
|
|
await exec(`zkutil verify -p ${keyBasePath}.params -r ${dir}/proof.json -i ${dir}/public.json`)
|
|
} catch (e) {
|
|
console.log(out, e)
|
|
throw e
|
|
}
|
|
return '0x' + JSON.parse(fs.readFileSync(`${dir}/proof.json`).toString()).proof
|
|
})
|
|
}
|
|
|
|
module.exports = { prove, proveZkutil }
|