tornado-packages/websnark
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: tornado-packages:1ab309ae3e779541cbe7c5e1cd8849722104e2e6
Branches Tags
tornado-packages:master
tornado-packages:develop
..
compare: tornado-packages:671762fab73f01771d0e7ebcf6b6a3123e193fb4
Branches Tags
tornado-packages:master
tornado-packages:develop

No commits in common. "1ab309ae3e779541cbe7c5e1cd8849722104e2e6" and "671762fab73f01771d0e7ebcf6b6a3123e193fb4" have entirely different histories.
1ab309ae3e ... 671762fab7

10 changed files with 2400 additions and 5342 deletions
Show Stats Expand all files Collapse all files

1
.npmignore
View File

@ -1 +0,0 @@
test

1
.npmrc
View File

@ -1 +0,0 @@
@tornado:registry=https://git.tornado.ws/api/packages/tornado-packages/npm/

3507
build/websnark.js
View File

File diff suppressed because it is too large Load Diff

3507
example/websnark.js
View File

File diff suppressed because it is too large Load Diff

467
package-lock.json generated
View File

File diff suppressed because it is too large Load Diff

76
package.json
View File

@ -1,40 +1,40 @@
{
"name": "@tornado/websnark",
"version": "0.0.4",
"description": "big integer library to work in Zq",
"main": "index.js",
"scripts": {
"test": "mocha --experimental-worker",
"buildwasm": "node tools/buildwasm.js",
"build": "node tools/buildwasm.js && browserify main.js -o build/websnark.js --exclude worker_threads --exclude crypto && cp build/websnark.js example/websnark.js"
},
"keywords": [
"bigint",
"bignum",
"biginteger",
"zq",
"elliptic",
"curve",
"prime",
"field"
],
"author": "Jordi Baylina",
"license": "GPL-3.0",
"repository": {
"type": "git",
"url": "https://git.tornado.ws/tornado-packages/websnark"
},
"devDependencies": {
"browserify": "^16.2.3",
"eslint": "^5.14.0",
"eslint-plugin-mocha": "^5.3.0",
"eslint-plugin-webassembly": "^1.8.4",
"mocha": "^6.1.4",
"package": "^1.0.1",
"wasmbuilder": "0.0.3"
},
"dependencies": {
"@tornado/snarkjs": "^0.1.20",
"big-integer": "1.6.42"
}
"name": "websnark",
"version": "0.0.4",
"description": "big integer library to work in Zq",
"main": "index.js",
"scripts": {
"test": "mocha --experimental-worker",
"buildwasm": "node tools/buildwasm.js",
"build": "node tools/buildwasm.js; browserify main.js -o build/websnark.js --exclude worker_threads --exclude crypto; cp build/websnark.js example/websnark.js"
},
"keywords": [
"bigint",
"bignum",
"biginteger",
"zq",
"elliptic",
"curve",
"prime",
"field"
],
"author": "Jordi Baylina",
"license": "GPL-3.0",
"repository": {
"type": "git",
"url": "https://github.com/iden3/websnark.git"
},
"devDependencies": {
"browserify": "^16.2.3",
"eslint": "^5.14.0",
"eslint-plugin-mocha": "^5.3.0",
"eslint-plugin-webassembly": "^1.8.4",
"mocha": "^6.1.4",
"package": "^1.0.1",
"snarkjs": "git+https://development.tornadocash.community/tornadocash/snarkjs.git#869181cfaf7526fe8972073d31655493a04326d5",
"wasmbuilder": "0.0.3"
},
"dependencies": {
"big-integer": "1.6.42"
}
}

40
src/utils.js
View File

@ -18,20 +18,20 @@
*/
const bigInt = require("big-integer");
const Circuit = require("@tornado/snarkjs/src/circuit");
const bigInt2 = require("@tornado/snarkjs/src/bigint");
const Circuit = require("snarkjs/src/circuit");
const bigInt2 = require("snarkjs/src/bigint");
const hexifyBigInts = require("../tools/stringifybigint").hexifyBigInts;
const unhexifyBigInts = require("../tools/stringifybigint").unhexifyBigInts;
const stringifyBigInts = require("../tools/stringifybigint").stringifyBigInts;
const unstringifyBigInts = require("../tools/stringifybigint").unstringifyBigInts;
const stringifyBigInts2 = require("@tornado/snarkjs/src/stringifybigint").stringifyBigInts;
const unstringifyBigInts2 = require("@tornado/snarkjs/src/stringifybigint").unstringifyBigInts;
const stringifyBigInts2 = require("snarkjs/src/stringifybigint").stringifyBigInts;
const unstringifyBigInts2 = require("snarkjs/src/stringifybigint").unstringifyBigInts;
function bigInt2BytesLE(_a, len) {
const b = Array(len);
let v = bigInt(_a);
for (let i = 0; i < len; i++) {
b[i] = v.and(0xff).toJSNumber();
for (let i=0; i<len; i++) {
b[i] = v.and(0xFF).toJSNumber();
v = v.shiftRight(8);
}
return b;
@ -40,8 +40,8 @@ function bigInt2BytesLE(_a, len) {
function bigInt2U32LE(_a, len) {
const b = Array(len);
let v = bigInt(_a);
for (let i = 0; i < len; i++) {
b[i] = v.and(0xffffffff).toJSNumber();
for (let i=0; i<len; i++) {
b[i] = v.and(0xFFFFFFFF).toJSNumber();
v = v.shiftRight(32);
}
return b;
@ -52,9 +52,9 @@ function convertWitness(witness) {
const buff = new ArrayBuffer(buffLen);
const h = {
dataView: new DataView(buff),
offset: 0,
offset: 0
};
const mask = bigInt2(0xffffffff);
const mask = bigInt2(0xFFFFFFFF);
for (let i = 0; i < witness.length; i++) {
for (let j = 0; j < 8; j++) {
const v = Number(witness[i].shr(j * 32).and(mask));
@ -73,17 +73,13 @@ function toHex32(number) {
function toSolidityInput(proof) {
const flatProof = unstringifyBigInts([
proof.pi_a[0],
proof.pi_a[1],
proof.pi_b[0][1],
proof.pi_b[0][0],
proof.pi_b[1][1],
proof.pi_b[1][0],
proof.pi_c[0],
proof.pi_c[1],
proof.pi_a[0], proof.pi_a[1],
proof.pi_b[0][1], proof.pi_b[0][0],
proof.pi_b[1][1], proof.pi_b[1][0],
proof.pi_c[0], proof.pi_c[1],
]);
const result = {
proof: "0x" + flatProof.map((x) => toHex32(x)).join(""),
proof: "0x" + flatProof.map(x => toHex32(x)).join("")
};
if (proof.publicSignals) {
result.publicSignals = hexifyBigInts(unstringifyBigInts(proof.publicSignals));
@ -91,11 +87,11 @@ function toSolidityInput(proof) {
return result;
}
function genWitness(input, circuitJson) {
function genWitness(input, circuitJson) {
const circuit = new Circuit(unstringifyBigInts2(circuitJson));
const witness = circuit.calculateWitness(unstringifyBigInts2(input));
const publicSignals = witness.slice(1, circuit.nPubInputs + circuit.nOutputs + 1);
return { witness, publicSignals };
return {witness, publicSignals};
}
async function genWitnessAndProve(groth16, input, circuitJson, provingKey) {
@ -106,4 +102,4 @@ async function genWitnessAndProve(groth16, input, circuitJson, provingKey) {
return result;
}
module.exports = { bigInt2BytesLE, bigInt2U32LE, toSolidityInput, genWitnessAndProve };
module.exports = {bigInt2BytesLE, bigInt2U32LE, toSolidityInput, genWitnessAndProve};

20
test/bn128.js
View File

@ -1,6 +1,6 @@
const assert = require("assert");
const refBn128 = require("@tornado/snarkjs").bn128;
const refBigInt = require("@tornado/snarkjs").bigInt;
const refBn128 = require("snarkjs").bn128;
const refBigInt = require("snarkjs").bigInt;
const buildBn128 = require("../index.js").buildBn128;
@ -15,7 +15,7 @@ describe("Basic tests for g1 in bn128", () => {
bn128.g1_fromMontgomery(p1, p1);
const d = bn128.g1_getPoint(p1);
for (let i = 0; i < 3; i++) {
for (let i=0; i<3; i++) {
d[i] = refBigInt(d[i].toString());
}
@ -39,6 +39,7 @@ describe("Basic tests for g1 in bn128", () => {
d[1] = refBigInt(d[1].toString());
d[2] = refBigInt(d[2].toString());
assert(d[0].equals(refD[0]));
assert(d[1].equals(refD[1]));
assert(d[2].equals(1));
@ -58,7 +59,7 @@ describe("Basic tests for g1 in bn128", () => {
bn128.g1_fromMontgomery(p1, p1);
const d = bn128.g1_getPoint(p1);
for (let i = 0; i < 3; i++) {
for (let i=0; i<3; i++) {
d[i] = refBigInt(d[i].toString());
}
@ -95,11 +96,12 @@ describe("Basic tests for g1 in bn128", () => {
bn128.g1_fromMontgomery(p2, p2);
const d = bn128.g1_getPoint(p2);
for (let i = 0; i < 3; i++) {
for (let i=0; i<3; i++) {
d[i] = refBigInt(d[i].toString());
}
assert(refBn128.G1.equals(d, refD));
}).timeout(10000000);
it("It should do a basic point doubling in G2", async () => {
const bn128 = await buildBn128();
@ -111,8 +113,8 @@ describe("Basic tests for g1 in bn128", () => {
bn128.g2_fromMontgomery(p1, p1);
const d = bn128.g2_getPoint(p1);
for (let i = 0; i < 3; i++) {
for (let j = 0; j < 2; j++) {
for (let i=0; i<3; i++) {
for (let j=0; j<2; j++) {
d[i][j] = refBigInt(d[i][j].toString());
}
}
@ -133,8 +135,8 @@ describe("Basic tests for g1 in bn128", () => {
bn128.g2_fromMontgomery(p2, p2);
const d = bn128.g2_getPoint(p2);
for (let i = 0; i < 3; i++) {
for (let j = 0; j < 2; j++) {
for (let i=0; i<3; i++) {
for (let j=0; j<2; j++) {
d[i][j] = refBigInt(d[i][j].toString());
assert(d[i][j].equals(refD[i][j]));
}

106
test/fft.js
View File

@ -1,6 +1,6 @@
const assert = require("assert");
const refBn128 = require("@tornado/snarkjs").bn128;
const refBigInt = require("@tornado/snarkjs").bigInt;
const refBn128 = require("snarkjs").bn128;
const refBigInt = require("snarkjs").bigInt;
const buildBn128 = require("../index.js").buildBn128;
@ -8,11 +8,11 @@ describe("FFT tests", () => {
it("create a basic FFT", async () => {
const bn128 = await buildBn128();
const N = 4;
const N=4;
const p = bn128.alloc(32 * N);
for (let i = 0; i < N; i++) {
bn128.putInt(p + i * 32, i);
const p = bn128.alloc(32*N);
for (let i=0; i<N; i++) {
bn128.putInt(p+i*32, i);
}
bn128.fft_toMontgomeryN(p, p, N);
@ -20,20 +20,20 @@ describe("FFT tests", () => {
bn128.fft_ifft(p, N);
bn128.fft_fromMontgomeryN(p, p, N);
for (let i = 0; i < N; i++) {
const a = bn128.getInt(p + i * 32);
assert.equal(a, i);
for (let i=0; i<N; i++) {
const a = bn128.getInt(p+i*32);
assert.equal(a,i);
}
});
it("create a do it reverselly FFT", async () => {
const bn128 = await buildBn128();
const N = 1024;
const N=1024;
const p = bn128.alloc(32 * N);
for (let i = 0; i < N; i++) {
bn128.putInt(p + i * 32, i);
const p = bn128.alloc(32*N);
for (let i=0; i<N; i++) {
bn128.putInt(p+i*32, i);
}
bn128.fft_toMontgomeryN(p, p, N);
@ -41,19 +41,19 @@ describe("FFT tests", () => {
bn128.fft_fft(p, N, 0);
bn128.fft_fromMontgomeryN(p, p, N);
for (let i = 0; i < N; i++) {
const a = bn128.getInt(p + i * 32);
assert.equal(a, i);
for (let i=0; i<N; i++) {
const a = bn128.getInt(p+i*32);
assert.equal(a,i);
}
});
it("test with zeros", async () => {
const bn128 = await buildBn128();
const N = 1024;
const N=1024;
const p = bn128.alloc(32 * N);
for (let i = 0; i < N; i++) {
bn128.putInt(p + i * 32, i % 2 == 0 ? 0 : 1);
const p = bn128.alloc(32*N);
for (let i=0; i<N; i++) {
bn128.putInt(p+i*32, (i%2 == 0)? 0 : 1);
}
bn128.fft_toMontgomeryN(p, p, N);
@ -61,61 +61,63 @@ describe("FFT tests", () => {
bn128.fft_fft(p, N, 0);
bn128.fft_fromMontgomeryN(p, p, N);
for (let i = 0; i < N; i++) {
const a = bn128.getInt(p + i * 32);
assert.equal(a, i % 2 == 0 ? 0 : 1);
for (let i=0; i<N; i++) {
const a = bn128.getInt(p+i*32);
assert.equal(a,(i%2 == 0)? 0 : 1);
}
});
it("test interleaved", async () => {
const bn128 = await buildBn128();
const N = 1024;
const N=1024;
const p = bn128.alloc(32 * N);
const pr1 = bn128.alloc(32 * N * 2);
const pr2 = bn128.alloc(32 * N * 2);
for (let i = 0; i < N; i++) {
bn128.putInt(p + i * 32, i);
const p = bn128.alloc(32*N);
const pr1 = bn128.alloc(32*N*2);
const pr2 = bn128.alloc(32*N*2);
for (let i=0; i<N; i++) {
bn128.putInt(p+i*32, i);
}
bn128.fft_toMontgomeryN(p, p, N);
bn128.fft_fft(p, N, 0);
bn128.fft_copyNInterleaved(p, pr1, N);
for (let i = 0; i < N; i++) {
bn128.putInt(p + i * 32, i);
for (let i=0; i<N; i++) {
bn128.putInt(p+i*32, i);
}
bn128.fft_toMontgomeryN(p, p, N);
bn128.fft_fft(p, N, 1);
bn128.fft_copyNInterleaved(p, pr1 + 32, N);
bn128.fft_copyNInterleaved(p, pr1+32, N);
bn128.fft_fromMontgomeryN(pr1, pr1, N * 2);
bn128.fft_fromMontgomeryN(pr1, pr1, N*2);
for (let i = 0; i < N; i++) {
bn128.putInt(pr2 + i * 32, i);
for (let i=0; i<N; i++) {
bn128.putInt(pr2+i*32, i);
}
for (let i = N; i < N * 2; i++) {
bn128.putInt(pr2 + i * 32, 0);
for (let i=N; i<N*2; i++) {
bn128.putInt(pr2+i*32, 0);
}
bn128.fft_toMontgomeryN(pr2, pr2, N * 2);
bn128.fft_fft(pr2, N * 2, 0);
bn128.fft_fromMontgomeryN(pr2, pr2, N * 2);
bn128.fft_toMontgomeryN(pr2, pr2, N*2);
bn128.fft_fft(pr2, N*2, 0);
bn128.fft_fromMontgomeryN(pr2, pr2, N*2);
for (let i = 0; i < N * 2; i++) {
const a = bn128.getInt(pr1 + i * 32, 1);
const b = bn128.getInt(pr2 + i * 32, 1);
for (let i=0; i<N*2; i++) {
const a = bn128.getInt(pr1+i*32, 1);
const b = bn128.getInt(pr2+i*32, 1);
assert(a.equals(b));
}
bn128.fft_toMontgomeryN(pr1, pr1, N * 2);
bn128.fft_ifft(pr1, N * 2, 0);
bn128.fft_fromMontgomeryN(pr1, pr1, N * 2);
for (let i = 0; i < N; i++) {
const a = bn128.getInt(pr1 + i * 32, 1);
assert.equal(a, i);
bn128.fft_toMontgomeryN(pr1, pr1, N*2);
bn128.fft_ifft(pr1, N*2, 0);
bn128.fft_fromMontgomeryN(pr1, pr1, N*2);
for (let i=0; i<N; i++) {
const a = bn128.getInt(pr1+i*32, 1);
assert.equal(a,i);
}
for (let i = N; i < N * 2; i++) {
const a = bn128.getInt(pr1 + i * 32, 1);
assert.equal(a, 0);
for (let i=N; i<N*2; i++) {
const a = bn128.getInt(pr1+i*32, 1);
assert.equal(a,0);
}
});
});

17
test/groth16.js
View File

@ -1,7 +1,8 @@
const assert = require("assert");
const fs = require("fs");
const path = require("path");
const snarkjs = require("@tornado/snarkjs");
const snarkjs = require("snarkjs");
const buildGroth16 = require("../index.js").buildGroth16;
@ -17,8 +18,8 @@ describe("Basic tests for groth16 proof generator", () => {
const pkey32 = new Uint32Array(provingKey);
const pPointsA = pkey32[5];
const points = provingKey.slice(pPointsA, pPointsA + nSignals * 64);
const signals = signalsAll.slice(0, nSignals * 32);
const points = provingKey.slice(pPointsA, pPointsA + nSignals*64);
const signals = signalsAll.slice(0, nSignals*32);
const pr1 = groth16.alloc(96);
const pPoints = groth16.alloc(points.byteLength);
@ -41,10 +42,11 @@ describe("Basic tests for groth16 proof generator", () => {
const r2 = groth16.bin2g1(groth16.getBin(pr1, 96));
assert.equal(r1[0], r2[0]);
assert.equal(r1[1], r2[1]);
assert.equal(r1[0],r2[0]);
assert.equal(r1[1],r2[1]);
groth16.terminate();
});
it("It should do a basic point doubling G1", async () => {
@ -55,13 +57,12 @@ describe("Basic tests for groth16 proof generator", () => {
const proofS = await groth16.proof(signals.buffer, provingKey.buffer);
const proof = snarkjs.unstringifyBigInts(proofS);
const verifierKey = snarkjs.unstringifyBigInts(
JSON.parse(fs.readFileSync(path.join(__dirname, "data", "verification_key.json"), "utf8"))
);
const verifierKey = snarkjs.unstringifyBigInts(JSON.parse(fs.readFileSync(path.join(__dirname, "data", "verification_key.json"), "utf8")));
const pub = snarkjs.unstringifyBigInts(JSON.parse(fs.readFileSync(path.join(__dirname, "data", "public.json"), "utf8")));
assert(snarkjs.groth.isValid(verifierKey, proof, pub));
groth16.terminate();
}).timeout(10000000);
});