github/zk-kit
1
1
Fork 0
mirror of https://github.com/privacy-scaling-explorations/zk-kit.git synced 2026-04-22 03:00:15 -04:00
Code Issues Packages Projects Releases Wiki Activity

enable in browser proofs

Browse Source
This commit is contained in:
Andrija
2021-11-25 15:51:56 +01:00
parent e3b1aaf2d5
commit 35499fefe0
2 changed files with 24 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
packages/protocols/package.json
View File

@@ -1,6 +1,6 @@
{
"name": "@libsem/protocols",
"version": "1.0.22",
"version": "1.0.23",
"description": "Client library for generating and verifying Semaphore & Rln ZK proofs.",
"main": "dist/index.node.js",
"types": "dist/types/index.d.ts",

33
packages/protocols/src/zk-protocol.ts
View File

@@ -14,15 +14,21 @@ export class ZkProtocol {
* @param witnessFileName where to save witness
* @returns creates and saves witness to witnessFileName
*/
async buildWnts(input: any, wasmFilePath: string, witnessFileName: string) {
const buffer = fs.readFileSync(wasmFilePath);
async genWtns(input: any, wasmFilePath: string) {
let wntsBuff: ArrayBuffer;
//window exists only in browser
if(typeof window !== 'undefined') {
const resp = await fetch(wasmFilePath);
wntsBuff = await resp.arrayBuffer();
} else {
wntsBuff = fs.readFileSync(wasmFilePath);
}
return new Promise((resolve, reject) => {
builder(buffer)
builder(wntsBuff)
.then(async witnessCalculator => {
const buff= await witnessCalculator.calculateWTNSBin(input, 0);
fs.writeFileSync(witnessFileName, buff);
resolve(witnessFileName);
const buff = await witnessCalculator.calculateWTNSBin(input, 0);
resolve(buff);
}).catch((error) => {
reject(error);
});
@@ -36,10 +42,17 @@ export class ZkProtocol {
* @returns zero knowledge proof
*/
async genProof(grothInput: any, wasmFilePath: string, finalZkeyPath: string): Promise<FullProof> {
await this.buildWnts(grothInput, wasmFilePath, 'witness.wtns');
const { proof, publicSignals } = await groth16.prove(finalZkeyPath, 'witness.wtns', null);
const exists = fs.existsSync('witness.wtns');
if(exists) fs.unlinkSync('witness.wtns');
let zkeyBuff: ArrayBuffer;
const wtnsBuff = await this.genWtns(grothInput, wasmFilePath);
//window exists only in browser
if(typeof window !== 'undefined') {
const resp = await fetch(finalZkeyPath);
zkeyBuff = await resp.arrayBuffer();
} else {
zkeyBuff = fs.readFileSync(finalZkeyPath);
}
const { proof, publicSignals } = await groth16.prove(new Uint8Array(zkeyBuff), wtnsBuff, null);
return { proof, publicSignals };
}