const chai = require("chai"); const assert = chai.assert; const fs = require("fs"); var tmp = require("tmp-promise"); const path = require("path"); const util = require("util"); const exec = util.promisify(require("child_process").exec); const loadR1cs = require("r1csfile").load; const ZqField = require("ffjavascript").ZqField; module.exports = wasm_tester; async function wasm_tester(circomInput, _options) { assert(await compiler_above_version("2.0.0"),"Wrong compiler version. Must be at least 2.0.0"); tmp.setGracefulCleanup(); const dir = await tmp.dir({prefix: "circom_", unsafeCleanup: true }); //console.log(dir.path); const baseName = path.basename(circomInput, ".circom"); const options = Object.assign({}, _options); options.wasm = true; options.sym = true; options.json = options.json || false; // costraints in json format options.r1cs = true; options.output = dir.path; await compile(circomInput, options); const utils = require("./utils"); const WitnessCalculator = require("./witness_calculator"); const wasm = await fs.promises.readFile(path.join(dir.path, baseName+"_js/"+ baseName + ".wasm")); const wc = await WitnessCalculator(wasm); return new WasmTester(dir, baseName, wc); } async function compile (fileName, options) { var flags = "--wasm "; if (options.sym) flags += "--sym "; if (options.r1cs) flags += "--r1cs "; if (options.json) flags += "--json "; if (options.output) flags += "--output " + options.output + " "; console.log(circom + flags + fileName); await exec("circom " + flags + fileName); } class WasmTester { constructor(dir, baseName, witnessCalculator) { this.dir=dir; this.baseName = baseName; this.witnessCalculator = witnessCalculator; } async release() { await this.dir.cleanup(); } async calculateWitness(input, sanityCheck) { return await this.witnessCalculator.calculateWitness(input, sanityCheck); } async loadSymbols() { if (this.symbols) return; this.symbols = {}; const symsStr = await fs.promises.readFile( path.join(this.dir.path, this.baseName + ".sym"), "utf8" ); const lines = symsStr.split("\n"); for (let i=0; i ${v}`); } return lines.join("\n"); } async checkConstraints(witness) { const self = this; if (!self.constraints) await self.loadConstraints(); for (let i=0; i v2[i]) return true; if (v1[i] < v2[i]) return false; } return true; } async function compiler_above_version(v) { let output = await exec('circom --version').toString(); let compiler_version = version_to_list(output.slice(output.search(/\d/),-1)); vlist = version_to_list(v); return check_versions ( compiler_version, vlist ); }