94 lines
2.7 KiB
JavaScript
94 lines
2.7 KiB
JavaScript
const path = require("path");
|
|
const snarkjs = require("@tornado/snarkjs");
|
|
const compiler = require("circom");
|
|
const fs = require("fs");
|
|
|
|
const bigInt = snarkjs.bigInt;
|
|
const smt = require("../src/smt.js");
|
|
|
|
const circuitSource = `
|
|
include "../circuits/smt/smtverifier.circom";
|
|
template SMT(nLevels) {
|
|
signal input root;
|
|
signal input mtp[nLevels];
|
|
signal input hi;
|
|
signal input hv;
|
|
|
|
component smtClaimExists = SMTVerifier(nLevels);
|
|
smtClaimExists.enabled <== 1;
|
|
smtClaimExists.fnc <== 0;
|
|
smtClaimExists.root <== root;
|
|
for (var i=0; i<nLevels; i++) {
|
|
smtClaimExists.siblings[i] <== mtp[i];
|
|
}
|
|
smtClaimExists.oldKey <== 0;
|
|
smtClaimExists.oldValue <== 0;
|
|
smtClaimExists.isOld0 <== 0;
|
|
|
|
smtClaimExists.key <== hi;
|
|
smtClaimExists.value <== hv;
|
|
}
|
|
component main = SMT(4);
|
|
`;
|
|
|
|
describe("smt3test", function () {
|
|
this.timeout(200000);
|
|
|
|
let circuitFileName;
|
|
|
|
before(async () => {
|
|
circuitFileName = path.join(__dirname, ".", "rawsmt3.circom");
|
|
fs.writeFileSync(circuitFileName, circuitSource);
|
|
});
|
|
|
|
const levels = 4;
|
|
async function testsmt3(e1, e2) {
|
|
let tree = await smt.newMemEmptyTrie();
|
|
|
|
// insert e1, e2
|
|
await tree.insert(e1.hi, e1.hv);
|
|
await tree.insert(e2.hi, e2.hv);
|
|
|
|
// generate proof for e1
|
|
const findInfo = await tree.find(e1.hi);
|
|
const siblings = findInfo.siblings;
|
|
while (siblings.length < levels) siblings.push(bigInt(0));
|
|
|
|
const input = {
|
|
root: tree.root,
|
|
mtp: siblings,
|
|
hi: e1.hi,
|
|
hv: e1.hv,
|
|
};
|
|
|
|
const compiledCircuit = await compiler(circuitFileName, { reduceConstraints: false });
|
|
|
|
const circuit = new snarkjs.Circuit(compiledCircuit);
|
|
const witness = circuit.calculateWitness(input);
|
|
circuit.checkWitness(witness);
|
|
}
|
|
|
|
it("TestSmts", async () => {
|
|
const e1 = {
|
|
hi: bigInt("17124152697573569611556136390143205198134245887034837071647643529178599000839"),
|
|
hv: bigInt("19650379996168153643111744440707177573540245771926102415571667548153444658179"),
|
|
};
|
|
|
|
const e2ok = {
|
|
hi: bigInt("16498254692537945203721083102154618658340563351558973077349594629411025251262"),
|
|
hv: bigInt("19650379996168153643111744440707177573540245771926102415571667548153444658179"),
|
|
};
|
|
|
|
const e2fail = {
|
|
hi: bigInt("17195092312975762537892237130737365903429674363577646686847513978084990105579"),
|
|
hv: bigInt("19650379996168153643111744440707177573540245771926102415571667548153444658179"),
|
|
};
|
|
|
|
console.log("test e1, e2ok");
|
|
await testsmt3(e1, e2ok);
|
|
|
|
console.log("test e1, e2fail");
|
|
await testsmt3(e1, e2fail);
|
|
});
|
|
});
|