@aztec/bb-prover 0.0.0-test.0

package/src/verifier/index.ts

@@ -0,0 +1 @@
+export * from './bb_verifier.js';

package/src/wasm/bb_wasm_private_kernel_prover.ts

@@ -0,0 +1,55 @@
+import { AztecClientBackend } from '@aztec/bb.js';
+import { createLogger } from '@aztec/foundation/log';
+import { Timer } from '@aztec/foundation/timer';
+import type { ArtifactProvider } from '@aztec/noir-protocol-circuits-types/types';
+import type { SimulationProvider } from '@aztec/simulator/client';
+import { ClientIvcProof } from '@aztec/stdlib/proofs';
+
+import { serializeWitness } from '@noir-lang/noirc_abi';
+import type { WitnessMap } from '@noir-lang/types';
+import { ungzip } from 'pako';
+
+import { BBPrivateKernelProver } from '../prover/bb_private_kernel_prover.js';
+
+export abstract class BBWASMPrivateKernelProver extends BBPrivateKernelProver {
+  constructor(
+    protected override artifactProvider: ArtifactProvider,
+    protected override simulationProvider: SimulationProvider,
+    private threads: number = 1,
+    protected override log = createLogger('bb-prover:wasm'),
+  ) {
+    super(artifactProvider, simulationProvider, log);
+  }
+
+  public override async createClientIvcProof(acirs: Buffer[], witnessStack: WitnessMap[]): Promise<ClientIvcProof> {
+    const timer = new Timer();
+    this.log.info(`Generating ClientIVC proof...`);
+    const backend = new AztecClientBackend(
+      acirs.map(acir => ungzip(acir)),
+      { threads: this.threads, logger: this.log.verbose, wasmPath: process.env.BB_WASM_PATH },
+    );
+
+    const [proof, vk] = await backend.prove(witnessStack.map(witnessMap => ungzip(serializeWitness(witnessMap))));
+    await backend.destroy();
+    this.log.info(`Generated ClientIVC proof`, {
+      eventName: 'client-ivc-proof-generation',
+      duration: timer.ms(),
+      proofSize: proof.length,
+      vkSize: vk.length,
+    });
+    return new ClientIvcProof(Buffer.from(proof), Buffer.from(vk));
+  }
+
+  public override async computeGateCountForCircuit(_bytecode: Buffer, _circuitName: string): Promise<number> {
+    const backend = new AztecClientBackend([ungzip(_bytecode)], {
+      threads: this.threads,
+      logger: this.log.verbose,
+      wasmPath: process.env.BB_WASM_PATH,
+    });
+
+    const gateCount = await backend.gates();
+    await backend.destroy();
+
+    return gateCount[0];
+  }
+}

package/src/wasm/bundle.ts

@@ -0,0 +1,11 @@
+import { createLogger } from '@aztec/foundation/log';
+import { BundleArtifactProvider } from '@aztec/noir-protocol-circuits-types/client/bundle';
+import type { SimulationProvider } from '@aztec/simulator/client';
+
+import { BBWASMPrivateKernelProver } from './bb_wasm_private_kernel_prover.js';
+
+export class BBWASMBundlePrivateKernelProver extends BBWASMPrivateKernelProver {
+  constructor(simulationProvider: SimulationProvider, threads = 1, log = createLogger('bb-prover:wasm:bundle')) {
+    super(new BundleArtifactProvider(), simulationProvider, threads, log);
+  }
+}

package/src/wasm/lazy.ts

@@ -0,0 +1,11 @@
+import { createLogger } from '@aztec/foundation/log';
+import { LazyArtifactProvider } from '@aztec/noir-protocol-circuits-types/client/lazy';
+import type { SimulationProvider } from '@aztec/simulator/client';
+
+import { BBWASMPrivateKernelProver } from './bb_wasm_private_kernel_prover.js';
+
+export class BBWASMLazyPrivateKernelProver extends BBWASMPrivateKernelProver {
+  constructor(simulationProvider: SimulationProvider, threads = 1, log = createLogger('bb-prover:wasm:lazy')) {
+    super(new LazyArtifactProvider(), simulationProvider, threads, log);
+  }
+}