@aztec/bb-prover 0.0.1-commit.b33fc05d0 → 0.0.1-commit.b3d3157a

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (58) hide show
  1. package/dest/avm_proving_tests/avm_proving_tester.d.ts +9 -5
  2. package/dest/avm_proving_tests/avm_proving_tester.d.ts.map +1 -1
  3. package/dest/avm_proving_tests/avm_proving_tester.js +146 -104
  4. package/dest/bb/bb_js_backend.d.ts +196 -0
  5. package/dest/bb/bb_js_backend.d.ts.map +1 -0
  6. package/dest/bb/bb_js_backend.js +379 -0
  7. package/dest/bb/bb_js_debug.d.ts +52 -0
  8. package/dest/bb/bb_js_debug.d.ts.map +1 -0
  9. package/dest/bb/bb_js_debug.js +176 -0
  10. package/dest/bb/file_names.d.ts +4 -0
  11. package/dest/bb/file_names.d.ts.map +1 -0
  12. package/dest/bb/file_names.js +5 -0
  13. package/dest/config.d.ts +17 -1
  14. package/dest/config.d.ts.map +1 -1
  15. package/dest/index.d.ts +3 -2
  16. package/dest/index.d.ts.map +1 -1
  17. package/dest/index.js +2 -1
  18. package/dest/prover/client/bb_private_kernel_prover.d.ts +10 -2
  19. package/dest/prover/client/bb_private_kernel_prover.d.ts.map +1 -1
  20. package/dest/prover/client/bb_private_kernel_prover.js +38 -4
  21. package/dest/prover/proof_utils.d.ts +11 -1
  22. package/dest/prover/proof_utils.d.ts.map +1 -1
  23. package/dest/prover/proof_utils.js +24 -1
  24. package/dest/prover/server/bb_prover.d.ts +4 -5
  25. package/dest/prover/server/bb_prover.d.ts.map +1 -1
  26. package/dest/prover/server/bb_prover.js +207 -78
  27. package/dest/verification_key/verification_key_data.js +1 -1
  28. package/dest/verifier/batch_chonk_verifier.d.ts +56 -0
  29. package/dest/verifier/batch_chonk_verifier.d.ts.map +1 -0
  30. package/dest/verifier/batch_chonk_verifier.js +384 -0
  31. package/dest/verifier/bb_verifier.d.ts +4 -1
  32. package/dest/verifier/bb_verifier.d.ts.map +1 -1
  33. package/dest/verifier/bb_verifier.js +134 -45
  34. package/dest/verifier/index.d.ts +2 -1
  35. package/dest/verifier/index.d.ts.map +1 -1
  36. package/dest/verifier/index.js +1 -0
  37. package/dest/verifier/queued_chonk_verifier.d.ts +2 -3
  38. package/dest/verifier/queued_chonk_verifier.d.ts.map +1 -1
  39. package/dest/verifier/queued_chonk_verifier.js +6 -5
  40. package/package.json +19 -17
  41. package/src/avm_proving_tests/avm_proving_tester.ts +45 -126
  42. package/src/bb/bb_js_backend.ts +435 -0
  43. package/src/bb/bb_js_debug.ts +227 -0
  44. package/src/bb/file_names.ts +6 -0
  45. package/src/config.ts +16 -0
  46. package/src/index.ts +2 -1
  47. package/src/prover/client/bb_private_kernel_prover.ts +115 -3
  48. package/src/prover/proof_utils.ts +41 -1
  49. package/src/prover/server/bb_prover.ts +132 -137
  50. package/src/verification_key/verification_key_data.ts +1 -1
  51. package/src/verifier/batch_chonk_verifier.ts +415 -0
  52. package/src/verifier/bb_verifier.ts +66 -76
  53. package/src/verifier/index.ts +1 -0
  54. package/src/verifier/queued_chonk_verifier.ts +6 -7
  55. package/dest/bb/execute.d.ts +0 -108
  56. package/dest/bb/execute.d.ts.map +0 -1
  57. package/dest/bb/execute.js +0 -652
  58. package/src/bb/execute.ts +0 -687
@@ -1,15 +1,14 @@
1
1
  import type { ClientProtocolCircuitVerifier, IVCProofVerificationResult } from '@aztec/stdlib/interfaces/server';
2
2
  import type { Tx } from '@aztec/stdlib/tx';
3
3
  import { type TelemetryClient } from '@aztec/telemetry-client';
4
- import type { BBConfig } from '../config.js';
5
4
  export declare class QueuedIVCVerifier implements ClientProtocolCircuitVerifier {
6
5
  private verifier;
7
6
  private telemetry;
8
7
  private logger;
9
8
  private queue;
10
9
  private metrics;
11
- constructor(config: BBConfig, verifier: ClientProtocolCircuitVerifier, telemetry?: TelemetryClient, logger?: import("@aztec/foundation/log").Logger);
10
+ constructor(verifier: ClientProtocolCircuitVerifier, concurrency: number, telemetry?: TelemetryClient, logger?: import("@aztec/foundation/log").Logger);
12
11
  verifyProof(tx: Tx): Promise<IVCProofVerificationResult>;
13
12
  stop(): Promise<void>;
14
13
  }
15
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicXVldWVkX2Nob25rX3ZlcmlmaWVyLmQudHMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvdmVyaWZpZXIvcXVldWVkX2Nob25rX3ZlcmlmaWVyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUVBLE9BQU8sS0FBSyxFQUFFLDZCQUE2QixFQUFFLDBCQUEwQixFQUFFLE1BQU0saUNBQWlDLENBQUM7QUFDakgsT0FBTyxLQUFLLEVBQUUsRUFBRSxFQUFFLE1BQU0sa0JBQWtCLENBQUM7QUFDM0MsT0FBTyxFQU1MLEtBQUssZUFBZSxFQUlyQixNQUFNLHlCQUF5QixDQUFDO0FBSWpDLE9BQU8sS0FBSyxFQUFFLFFBQVEsRUFBRSxNQUFNLGNBQWMsQ0FBQztBQWlFN0MscUJBQWEsaUJBQWtCLFlBQVcsNkJBQTZCO0lBTW5FLE9BQU8sQ0FBQyxRQUFRO0lBQ2hCLE9BQU8sQ0FBQyxTQUFTO0lBQ2pCLE9BQU8sQ0FBQyxNQUFNO0lBUGhCLE9BQU8sQ0FBQyxLQUFLLENBQWM7SUFDM0IsT0FBTyxDQUFDLE9BQU8sQ0FBcUI7SUFFcEMsWUFDRSxNQUFNLEVBQUUsUUFBUSxFQUNSLFFBQVEsRUFBRSw2QkFBNkIsRUFDdkMsU0FBUyxHQUFFLGVBQXNDLEVBQ2pELE1BQU0seUNBQWtELEVBTWpFO0lBRVksV0FBVyxDQUFDLEVBQUUsRUFBRSxFQUFFLEdBQUcsT0FBTyxDQUFDLDBCQUEwQixDQUFDLENBSXBFO0lBRUQsSUFBSSxJQUFJLE9BQU8sQ0FBQyxJQUFJLENBQUMsQ0FFcEI7Q0FDRiJ9
14
+ //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoicXVldWVkX2Nob25rX3ZlcmlmaWVyLmQudHMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvdmVyaWZpZXIvcXVldWVkX2Nob25rX3ZlcmlmaWVyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUVBLE9BQU8sS0FBSyxFQUFFLDZCQUE2QixFQUFFLDBCQUEwQixFQUFFLE1BQU0saUNBQWlDLENBQUM7QUFDakgsT0FBTyxLQUFLLEVBQUUsRUFBRSxFQUFFLE1BQU0sa0JBQWtCLENBQUM7QUFDM0MsT0FBTyxFQU1MLEtBQUssZUFBZSxFQUlyQixNQUFNLHlCQUF5QixDQUFDO0FBbUVqQyxxQkFBYSxpQkFBa0IsWUFBVyw2QkFBNkI7SUFLbkUsT0FBTyxDQUFDLFFBQVE7SUFFaEIsT0FBTyxDQUFDLFNBQVM7SUFDakIsT0FBTyxDQUFDLE1BQU07SUFQaEIsT0FBTyxDQUFDLEtBQUssQ0FBYztJQUMzQixPQUFPLENBQUMsT0FBTyxDQUFxQjtJQUVwQyxZQUNVLFFBQVEsRUFBRSw2QkFBNkIsRUFDL0MsV0FBVyxFQUFFLE1BQU0sRUFDWCxTQUFTLEdBQUUsZUFBc0MsRUFDakQsTUFBTSx5Q0FBa0QsRUFNakU7SUFFWSxXQUFXLENBQUMsRUFBRSxFQUFFLEVBQUUsR0FBRyxPQUFPLENBQUMsMEJBQTBCLENBQUMsQ0FJcEU7SUFFSyxJQUFJLElBQUksT0FBTyxDQUFDLElBQUksQ0FBQyxDQUcxQjtDQUNGIn0=
@@ -1 +1 @@
1
- {"version":3,"file":"queued_chonk_verifier.d.ts","sourceRoot":"","sources":["../../src/verifier/queued_chonk_verifier.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,6BAA6B,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AACjH,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AAC3C,OAAO,EAML,KAAK,eAAe,EAIrB,MAAM,yBAAyB,CAAC;AAIjC,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAiE7C,qBAAa,iBAAkB,YAAW,6BAA6B;IAMnE,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,MAAM;IAPhB,OAAO,CAAC,KAAK,CAAc;IAC3B,OAAO,CAAC,OAAO,CAAqB;IAEpC,YACE,MAAM,EAAE,QAAQ,EACR,QAAQ,EAAE,6BAA6B,EACvC,SAAS,GAAE,eAAsC,EACjD,MAAM,yCAAkD,EAMjE;IAEY,WAAW,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,CAAC,0BAA0B,CAAC,CAIpE;IAED,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAEpB;CACF"}
1
+ {"version":3,"file":"queued_chonk_verifier.d.ts","sourceRoot":"","sources":["../../src/verifier/queued_chonk_verifier.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,6BAA6B,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AACjH,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AAC3C,OAAO,EAML,KAAK,eAAe,EAIrB,MAAM,yBAAyB,CAAC;AAmEjC,qBAAa,iBAAkB,YAAW,6BAA6B;IAKnE,OAAO,CAAC,QAAQ;IAEhB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,MAAM;IAPhB,OAAO,CAAC,KAAK,CAAc;IAC3B,OAAO,CAAC,OAAO,CAAqB;IAEpC,YACU,QAAQ,EAAE,6BAA6B,EAC/C,WAAW,EAAE,MAAM,EACX,SAAS,GAAE,eAAsC,EACjD,MAAM,yCAAkD,EAMjE;IAEY,WAAW,CAAC,EAAE,EAAE,EAAE,GAAG,OAAO,CAAC,0BAA0B,CAAC,CAIpE;IAEK,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAG1B;CACF"}
@@ -81,21 +81,22 @@ export class QueuedIVCVerifier {
81
81
  logger;
82
82
  queue;
83
83
  metrics;
84
- constructor(config, verifier, telemetry = getTelemetryClient(), logger = createLogger('bb-prover:queued_chonk_verifier')){
84
+ constructor(verifier, concurrency, telemetry = getTelemetryClient(), logger = createLogger('bb-prover:queued_chonk_verifier')){
85
85
  this.verifier = verifier;
86
86
  this.telemetry = telemetry;
87
87
  this.logger = logger;
88
88
  this.metrics = new IVCVerifierMetrics(this.telemetry, 'QueuedIVCVerifier');
89
89
  this.queue = new SerialQueue();
90
- this.logger.info(`Starting QueuedIVCVerifier with ${config.numConcurrentIVCVerifiers} concurrent verifiers`);
91
- this.queue.start(config.numConcurrentIVCVerifiers);
90
+ this.logger.info(`Starting QueuedIVCVerifier with ${concurrency} concurrent verifiers`);
91
+ this.queue.start(concurrency);
92
92
  }
93
93
  async verifyProof(tx) {
94
94
  const result = await this.queue.put(()=>this.verifier.verifyProof(tx));
95
95
  this.metrics.recordIVCVerification(result);
96
96
  return result;
97
97
  }
98
- stop() {
99
- return this.queue.end();
98
+ async stop() {
99
+ await this.queue.end();
100
+ await this.verifier.stop();
100
101
  }
101
102
  }
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@aztec/bb-prover",
3
- "version": "0.0.1-commit.b33fc05d0",
3
+ "version": "0.0.1-commit.b3d3157a",
4
4
  "type": "module",
5
5
  "exports": {
6
6
  ".": "./dest/index.js",
@@ -9,7 +9,8 @@
9
9
  "./client": "./dest/prover/client/bb_private_kernel_prover.js",
10
10
  "./verifier": "./dest/verifier/index.js",
11
11
  "./test": "./dest/test/index.js",
12
- "./config": "./dest/config.js"
12
+ "./config": "./dest/config.js",
13
+ "./debug": "./dest/bb/bb_js_debug.js"
13
14
  },
14
15
  "bin": {
15
16
  "bb-cli": "./dest/bb/index.js"
@@ -69,27 +70,28 @@
69
70
  ]
70
71
  },
71
72
  "dependencies": {
72
- "@aztec/bb.js": "0.0.1-commit.b33fc05d0",
73
- "@aztec/constants": "0.0.1-commit.b33fc05d0",
74
- "@aztec/foundation": "0.0.1-commit.b33fc05d0",
75
- "@aztec/noir-noirc_abi": "0.0.1-commit.b33fc05d0",
76
- "@aztec/noir-protocol-circuits-types": "0.0.1-commit.b33fc05d0",
77
- "@aztec/noir-types": "0.0.1-commit.b33fc05d0",
78
- "@aztec/simulator": "0.0.1-commit.b33fc05d0",
79
- "@aztec/stdlib": "0.0.1-commit.b33fc05d0",
80
- "@aztec/telemetry-client": "0.0.1-commit.b33fc05d0",
81
- "@aztec/world-state": "0.0.1-commit.b33fc05d0",
73
+ "@aztec/bb.js": "0.0.1-commit.b3d3157a",
74
+ "@aztec/constants": "0.0.1-commit.b3d3157a",
75
+ "@aztec/foundation": "0.0.1-commit.b3d3157a",
76
+ "@aztec/noir-noirc_abi": "0.0.1-commit.b3d3157a",
77
+ "@aztec/noir-protocol-circuits-types": "0.0.1-commit.b3d3157a",
78
+ "@aztec/noir-types": "0.0.1-commit.b3d3157a",
79
+ "@aztec/simulator": "0.0.1-commit.b3d3157a",
80
+ "@aztec/stdlib": "0.0.1-commit.b3d3157a",
81
+ "@aztec/telemetry-client": "0.0.1-commit.b3d3157a",
82
+ "@aztec/world-state": "0.0.1-commit.b3d3157a",
82
83
  "commander": "^12.1.0",
84
+ "msgpackr": "^1.11.2",
83
85
  "pako": "^2.1.0",
84
86
  "source-map-support": "^0.5.21",
85
87
  "tslib": "^2.4.0"
86
88
  },
87
89
  "devDependencies": {
88
- "@aztec/ethereum": "0.0.1-commit.b33fc05d0",
89
- "@aztec/kv-store": "0.0.1-commit.b33fc05d0",
90
- "@aztec/noir-contracts.js": "0.0.1-commit.b33fc05d0",
91
- "@aztec/noir-test-contracts.js": "0.0.1-commit.b33fc05d0",
92
- "@aztec/protocol-contracts": "0.0.1-commit.b33fc05d0",
90
+ "@aztec/ethereum": "0.0.1-commit.b3d3157a",
91
+ "@aztec/kv-store": "0.0.1-commit.b3d3157a",
92
+ "@aztec/noir-contracts.js": "0.0.1-commit.b3d3157a",
93
+ "@aztec/noir-test-contracts.js": "0.0.1-commit.b3d3157a",
94
+ "@aztec/protocol-contracts": "0.0.1-commit.b3d3157a",
93
95
  "@jest/globals": "^30.0.0",
94
96
  "@types/jest": "^30.0.0",
95
97
  "@types/node": "^22.15.17",
@@ -1,4 +1,4 @@
1
- import type { LogFn, LogLevel, Logger } from '@aztec/foundation/log';
1
+ import type { AvmStat } from '@aztec/bb.js';
2
2
  import { Timer } from '@aztec/foundation/timer';
3
3
  import {
4
4
  PublicTxSimulationTester,
@@ -15,77 +15,12 @@ import type { MerkleTreeWriteOperations } from '@aztec/stdlib/interfaces/server'
15
15
  import type { GlobalVariables } from '@aztec/stdlib/tx';
16
16
  import { NativeWorldStateService } from '@aztec/world-state';
17
17
 
18
- import fs from 'node:fs/promises';
19
- import { tmpdir } from 'node:os';
20
18
  import path from 'path';
21
19
 
22
- import { type BBResult, type BBSuccess, BB_RESULT, generateAvmProof, verifyAvmProof } from '../bb/execute.js';
20
+ import { BBJsFactory } from '../bb/bb_js_backend.js';
23
21
 
24
22
  const BB_PATH = path.resolve('../../barretenberg/cpp/build/bin/bb-avm');
25
23
 
26
- // An InterceptingLogger that records all log messages and forwards them to a wrapped logger.
27
- class InterceptingLogger implements Logger {
28
- public readonly logs: string[] = [];
29
- public level: LogLevel;
30
- public module: string;
31
-
32
- private logger: Logger;
33
-
34
- constructor(logger: Logger) {
35
- this.logger = logger;
36
- this.level = logger.level;
37
- this.module = logger.module;
38
- }
39
-
40
- isLevelEnabled(level: LogLevel): boolean {
41
- return this.logger.isLevelEnabled(level);
42
- }
43
-
44
- createChild(_childModule: string): Logger {
45
- throw new Error('Not implemented');
46
- }
47
-
48
- getBindings() {
49
- return this.logger.getBindings();
50
- }
51
-
52
- private intercept(level: LogLevel, msg: string, ...args: any[]) {
53
- this.logs.push(...msg.split('\n'));
54
- // Forward to the wrapped logger
55
- (this.logger[level] as LogFn)(msg, ...args);
56
- }
57
-
58
- // Log methods for each level
59
- silent(msg: string, ...args: any[]) {
60
- this.intercept('silent', msg, ...args);
61
- }
62
- fatal(msg: string, ...args: any[]) {
63
- this.intercept('fatal', msg, ...args);
64
- }
65
- warn(msg: string, ...args: any[]) {
66
- this.intercept('warn', msg, ...args);
67
- }
68
- info(msg: string, ...args: any[]) {
69
- this.intercept('info', msg, ...args);
70
- }
71
- verbose(msg: string, ...args: any[]) {
72
- this.intercept('verbose', msg, ...args);
73
- }
74
- debug(msg: string, ...args: any[]) {
75
- this.intercept('debug', msg, ...args);
76
- }
77
- trace(msg: string, ...args: any[]) {
78
- this.intercept('trace', msg, ...args);
79
- }
80
-
81
- // Error log function can be string or Error
82
- error(err: Error | string, ...args: any[]) {
83
- const msg = typeof err === 'string' ? err : err.message;
84
- this.logs.push(msg);
85
- this.logger.error(msg, err, ...args);
86
- }
87
- }
88
-
89
24
  // Config with collectHints enabled for proving tests
90
25
  const provingConfig: PublicSimulatorConfig = PublicSimulatorConfig.from({
91
26
  skipFeeEnforcement: false,
@@ -97,7 +32,7 @@ const provingConfig: PublicSimulatorConfig = PublicSimulatorConfig.from({
97
32
  });
98
33
 
99
34
  export class AvmProvingTester extends PublicTxSimulationTester {
100
- private bbWorkingDirectory: string = '';
35
+ private readonly bbJsFactory = new BBJsFactory(BB_PATH);
101
36
 
102
37
  constructor(
103
38
  private checkCircuitOnly: boolean,
@@ -121,53 +56,57 @@ export class AvmProvingTester extends PublicTxSimulationTester {
121
56
  return new AvmProvingTester(checkCircuitOnly, contractDataSource, merkleTrees, globals, metrics);
122
57
  }
123
58
 
124
- async prove(avmCircuitInputs: AvmCircuitInputs, txLabel: string = 'unlabeledTx'): Promise<BBResult> {
125
- // We use a new working directory for each proof.
126
- this.bbWorkingDirectory = await fs.mkdtemp(path.join(tmpdir(), 'bb-'));
59
+ /**
60
+ * Generate an AVM proof (or run check-circuit if configured). Records per-stage timings in the test metrics.
61
+ * Returns the in-memory proof fields on success; throws via jest expect() on failure.
62
+ */
63
+ async prove(avmCircuitInputs: AvmCircuitInputs, txLabel: string = 'unlabeledTx'): Promise<Uint8Array[]> {
64
+ const inputsBuffer = avmCircuitInputs.serializeWithMessagePack();
127
65
 
128
- const interceptingLogger = new InterceptingLogger(this.logger);
66
+ if (this.checkCircuitOnly) {
67
+ await using instance = await this.bbJsFactory.getInstance();
68
+ const { passed, stats } = await instance.checkAvmCircuit(inputsBuffer);
69
+ this.recordProverMetrics(stats, txLabel);
70
+ expect(passed).toBe(true);
71
+ return [];
72
+ }
129
73
 
130
- // Then we prove.
131
- const proofRes = await generateAvmProof(
132
- BB_PATH,
133
- this.bbWorkingDirectory,
134
- avmCircuitInputs,
135
- interceptingLogger,
136
- this.checkCircuitOnly,
137
- );
138
- if (proofRes.status === BB_RESULT.FAILURE) {
139
- this.logger.error(`Proof generation failed: ${proofRes.reason}`);
74
+ await using instance = await this.bbJsFactory.getInstance();
75
+ const { proof, stats } = await instance.generateAvmProof(inputsBuffer);
76
+ this.recordProverMetrics(stats, txLabel);
77
+ return proof;
78
+ }
79
+
80
+ async verify(proof: Uint8Array[], publicInputs: AvmCircuitPublicInputs): Promise<void> {
81
+ if (this.checkCircuitOnly) {
82
+ // Check-circuit did not generate a proof; nothing to verify.
83
+ return;
140
84
  }
141
- expect(proofRes.status).toEqual(BB_RESULT.SUCCESS);
85
+ const piBuffer = publicInputs.serializeWithMessagePack();
86
+ await using instance = await this.bbJsFactory.getInstance();
87
+ const { verified } = await instance.verifyAvmProof(proof, piBuffer);
88
+ expect(verified).toBe(true);
89
+ }
142
90
 
143
- // Parse the logs into a structured format.
144
- const logs = interceptingLogger.logs;
145
- // const traceSizes: { name: string; size: number }[] = [];
146
- // logs.forEach(log => {
147
- // const match = log.match(/\b(\w+): (\d+) \(~2/);
148
- // if (match) {
149
- // traceSizes.push({
150
- // name: match[1],
151
- // size: parseInt(match[2]),
152
- // });
153
- // }
154
- // });
155
- const times: { [key: string]: number } = {};
156
- logs.forEach(log => {
157
- const match = log.match(/\b([\w/]+)_ms: (\d+)/);
158
- if (match) {
159
- times[match[1]] = parseInt(match[2]);
160
- }
161
- });
91
+ public async proveVerify(avmCircuitInputs: AvmCircuitInputs, txLabel: string = 'unlabeledTx') {
92
+ const proof = await this.prove(avmCircuitInputs, txLabel);
93
+ await this.verify(proof, avmCircuitInputs.publicInputs);
94
+ }
162
95
 
163
- // Throw if logs did not contain any times.
96
+ private recordProverMetrics(stats: AvmStat[], txLabel: string) {
97
+ // Build a lookup keyed on the stage name with the `_ms` suffix stripped, matching the legacy
98
+ // stdout-scraped shape. bb::avm2::Stats::time() stores keys with `_ms` appended.
99
+ const times: { [key: string]: number } = {};
100
+ for (const { name, valueMs } of stats) {
101
+ const key = name.endsWith('_ms') ? name.slice(0, -'_ms'.length) : name;
102
+ times[key] = valueMs;
103
+ }
164
104
  if (Object.keys(times).length === 0) {
165
- throw new Error('AVM stdout did not contain any proving times in the stats!');
105
+ throw new Error('AVM response did not contain any proving-stage timings!');
166
106
  }
167
-
168
107
  // Hack to make labels match.
169
108
  const txLabelWithCount = `${txLabel}/${this.txCount - 1}`;
170
- // I need to cast because TS doesnt realize metrics is protected not private.
109
+ // Cast because TS doesn't realize `metrics` is protected, not private on the parent class.
171
110
  (this as any).metrics?.recordProverMetrics(txLabelWithCount, {
172
111
  proverSimulationStepMs: times['simulation/all'],
173
112
  proverProvingStepMs: times['proving/all'],
@@ -180,26 +119,6 @@ export class AvmProvingTester extends PublicTxSimulationTester {
180
119
  provingLogDerivativeInverseCommitmentsMs: times['prove/log_derivative_inverse_commitments_round'],
181
120
  provingWireCommitmentsMs: times['prove/wire_commitments_round'],
182
121
  });
183
-
184
- return proofRes as BBSuccess;
185
- }
186
-
187
- async verify(proofRes: BBSuccess, publicInputs: AvmCircuitPublicInputs): Promise<BBResult> {
188
- if (this.checkCircuitOnly) {
189
- // Skip verification if we are only checking the circuit.
190
- // Check-circuit does not generate a proof to verify.
191
- return proofRes;
192
- }
193
-
194
- return await verifyAvmProof(BB_PATH, this.bbWorkingDirectory, proofRes.proofPath!, publicInputs, this.logger);
195
- }
196
-
197
- public async proveVerify(avmCircuitInputs: AvmCircuitInputs, txLabel: string = 'unlabeledTx') {
198
- const provingRes = await this.prove(avmCircuitInputs, txLabel);
199
- expect(provingRes.status).toEqual(BB_RESULT.SUCCESS);
200
-
201
- const verificationRes = await this.verify(provingRes as BBSuccess, avmCircuitInputs.publicInputs);
202
- expect(verificationRes.status).toBe(BB_RESULT.SUCCESS);
203
122
  }
204
123
 
205
124
  public async simProveVerify(