@aztec/pxe 0.0.1-commit.8f9871590 → 0.0.1-commit.934299a21

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 (61) hide show
  1. package/dest/config/package_info.js +1 -1
  2. package/dest/contract_function_simulator/contract_function_simulator.d.ts +1 -1
  3. package/dest/contract_function_simulator/contract_function_simulator.d.ts.map +1 -1
  4. package/dest/contract_function_simulator/contract_function_simulator.js +8 -8
  5. package/dest/contract_function_simulator/noir-structs/event_validation_request.js +1 -1
  6. package/dest/contract_function_simulator/noir-structs/note_validation_request.d.ts +2 -2
  7. package/dest/contract_function_simulator/noir-structs/note_validation_request.d.ts.map +1 -1
  8. package/dest/contract_function_simulator/noir-structs/note_validation_request.js +1 -1
  9. package/dest/contract_function_simulator/oracle/private_execution_oracle.d.ts +1 -1
  10. package/dest/contract_function_simulator/oracle/private_execution_oracle.d.ts.map +1 -1
  11. package/dest/contract_function_simulator/oracle/private_execution_oracle.js +2 -10
  12. package/dest/contract_function_simulator/oracle/utility_execution_oracle.d.ts +1 -1
  13. package/dest/contract_function_simulator/oracle/utility_execution_oracle.d.ts.map +1 -1
  14. package/dest/contract_function_simulator/oracle/utility_execution_oracle.js +4 -7
  15. package/dest/contract_logging.d.ts +22 -0
  16. package/dest/contract_logging.d.ts.map +1 -0
  17. package/dest/contract_logging.js +23 -0
  18. package/dest/debug/pxe_debug_utils.d.ts +2 -2
  19. package/dest/debug/pxe_debug_utils.d.ts.map +1 -1
  20. package/dest/debug/pxe_debug_utils.js +4 -4
  21. package/dest/entrypoints/client/bundle/index.d.ts +2 -1
  22. package/dest/entrypoints/client/bundle/index.d.ts.map +1 -1
  23. package/dest/entrypoints/client/bundle/index.js +1 -0
  24. package/dest/entrypoints/client/lazy/index.d.ts +2 -1
  25. package/dest/entrypoints/client/lazy/index.d.ts.map +1 -1
  26. package/dest/entrypoints/client/lazy/index.js +1 -0
  27. package/dest/private_kernel/hints/private_kernel_reset_private_inputs_builder.d.ts +4 -3
  28. package/dest/private_kernel/hints/private_kernel_reset_private_inputs_builder.d.ts.map +1 -1
  29. package/dest/private_kernel/hints/private_kernel_reset_private_inputs_builder.js +129 -68
  30. package/dest/private_kernel/hints/test_utils.d.ts +122 -0
  31. package/dest/private_kernel/hints/test_utils.d.ts.map +1 -0
  32. package/dest/private_kernel/hints/test_utils.js +203 -0
  33. package/dest/private_kernel/private_kernel_execution_prover.d.ts +1 -1
  34. package/dest/private_kernel/private_kernel_execution_prover.d.ts.map +1 -1
  35. package/dest/private_kernel/private_kernel_execution_prover.js +13 -5
  36. package/dest/private_kernel/private_kernel_oracle.d.ts +6 -2
  37. package/dest/private_kernel/private_kernel_oracle.d.ts.map +1 -1
  38. package/dest/private_kernel/private_kernel_oracle.js +7 -3
  39. package/dest/pxe.d.ts +6 -6
  40. package/dest/pxe.d.ts.map +1 -1
  41. package/dest/pxe.js +27 -23
  42. package/dest/storage/contract_store/contract_store.d.ts +42 -15
  43. package/dest/storage/contract_store/contract_store.d.ts.map +1 -1
  44. package/dest/storage/contract_store/contract_store.js +140 -64
  45. package/package.json +16 -16
  46. package/src/config/package_info.ts +1 -1
  47. package/src/contract_function_simulator/contract_function_simulator.ts +15 -17
  48. package/src/contract_function_simulator/noir-structs/event_validation_request.ts +1 -1
  49. package/src/contract_function_simulator/noir-structs/note_validation_request.ts +1 -1
  50. package/src/contract_function_simulator/oracle/private_execution_oracle.ts +2 -11
  51. package/src/contract_function_simulator/oracle/utility_execution_oracle.ts +8 -7
  52. package/src/contract_logging.ts +39 -0
  53. package/src/debug/pxe_debug_utils.ts +4 -4
  54. package/src/entrypoints/client/bundle/index.ts +1 -0
  55. package/src/entrypoints/client/lazy/index.ts +1 -0
  56. package/src/private_kernel/hints/private_kernel_reset_private_inputs_builder.ts +164 -117
  57. package/src/private_kernel/hints/test_utils.ts +325 -0
  58. package/src/private_kernel/private_kernel_execution_prover.ts +13 -6
  59. package/src/private_kernel/private_kernel_oracle.ts +7 -7
  60. package/src/pxe.ts +33 -30
  61. package/src/storage/contract_store/contract_store.ts +170 -71
@@ -0,0 +1,325 @@
1
+ import {
2
+ MAX_KEY_VALIDATION_REQUESTS_PER_CALL,
3
+ MAX_KEY_VALIDATION_REQUESTS_PER_TX,
4
+ MAX_NOTE_HASHES_PER_CALL,
5
+ MAX_NOTE_HASHES_PER_TX,
6
+ MAX_NOTE_HASH_READ_REQUESTS_PER_CALL,
7
+ MAX_NOTE_HASH_READ_REQUESTS_PER_TX,
8
+ MAX_NULLIFIERS_PER_CALL,
9
+ MAX_NULLIFIERS_PER_TX,
10
+ MAX_NULLIFIER_READ_REQUESTS_PER_CALL,
11
+ MAX_NULLIFIER_READ_REQUESTS_PER_TX,
12
+ MAX_PRIVATE_LOGS_PER_CALL,
13
+ MAX_PRIVATE_LOGS_PER_TX,
14
+ } from '@aztec/constants';
15
+ import { makeTuple } from '@aztec/foundation/array';
16
+ import { Fr } from '@aztec/foundation/curves/bn254';
17
+ import { Point } from '@aztec/foundation/curves/grumpkin';
18
+ import type { Serializable } from '@aztec/foundation/serialize';
19
+ import { AztecAddress } from '@aztec/stdlib/aztec-address';
20
+ import {
21
+ ClaimedLengthArray,
22
+ KeyValidationRequest,
23
+ KeyValidationRequestAndSeparator,
24
+ NoteHash,
25
+ Nullifier,
26
+ PrivateCircuitPublicInputs,
27
+ PrivateKernelCircuitPublicInputs,
28
+ type PrivateKernelSimulateOutput,
29
+ ReadRequest,
30
+ ScopedKeyValidationRequestAndSeparator,
31
+ ScopedNoteHash,
32
+ ScopedNullifier,
33
+ ScopedReadRequest,
34
+ } from '@aztec/stdlib/kernel';
35
+ import { PrivateLogData, ScopedPrivateLogData } from '@aztec/stdlib/kernel';
36
+ import { PrivateLog } from '@aztec/stdlib/logs';
37
+ import { PrivateCallExecutionResult } from '@aztec/stdlib/tx';
38
+ import { VerificationKeyData } from '@aztec/stdlib/vks';
39
+
40
+ const DEFAULT_CONTRACT_ADDRESS = AztecAddress.fromBigInt(987654n);
41
+
42
+ /**
43
+ * Builds a ClaimedLengthArray from a list of items, padding to the required size.
44
+ */
45
+ function makeClaimed<T extends Serializable, N extends number>(items: T[], emptyFactory: { empty(): T }, maxSize: N) {
46
+ const padded = makeTuple(maxSize, i => items[i] ?? emptyFactory.empty());
47
+ return new ClaimedLengthArray<T, typeof maxSize>(padded, items.length);
48
+ }
49
+
50
+ /** Builder for PrivateKernelCircuitPublicInputs with fluent API for adding side effects. */
51
+ export class PrivateKernelCircuitPublicInputsBuilder {
52
+ private noteHashes: ScopedNoteHash[] = [];
53
+ private nullifiers: ScopedNullifier[] = [];
54
+ private noteHashReadRequests: ScopedReadRequest[] = [];
55
+ private nullifierReadRequests: ScopedReadRequest[] = [];
56
+ private keyValidationRequests: ScopedKeyValidationRequestAndSeparator[] = [];
57
+ private privateLogs: ScopedPrivateLogData[] = [];
58
+ private nextCounter: number;
59
+
60
+ constructor(
61
+ private contractAddress: AztecAddress = DEFAULT_CONTRACT_ADDRESS,
62
+ startCounter = 1,
63
+ ) {
64
+ this.nextCounter = startCounter;
65
+ }
66
+
67
+ private getCounter(sideEffectCounter?: number): number {
68
+ if (sideEffectCounter !== undefined) {
69
+ this.nextCounter = sideEffectCounter + 1;
70
+ return sideEffectCounter;
71
+ }
72
+ return this.nextCounter++;
73
+ }
74
+
75
+ /** Adds a note hash to the accumulated data. Defaults are generated randomly. */
76
+ addNoteHash(opts?: { value?: Fr; counter?: number; contractAddress?: AztecAddress }): this {
77
+ const value = opts?.value ?? Fr.random();
78
+ const counter = this.getCounter(opts?.counter);
79
+ const addr = opts?.contractAddress ?? this.contractAddress;
80
+ this.noteHashes.push(new NoteHash(value, counter).scope(addr));
81
+ return this;
82
+ }
83
+
84
+ /** Adds a nullifier to the accumulated data. Defaults are generated randomly. */
85
+ addNullifier(opts?: { value?: Fr; noteHash?: Fr; counter?: number; contractAddress?: AztecAddress }): this {
86
+ const value = opts?.value ?? Fr.random();
87
+ const noteHash = opts?.noteHash ?? Fr.ZERO;
88
+ const counter = this.getCounter(opts?.counter);
89
+ const addr = opts?.contractAddress ?? this.contractAddress;
90
+ this.nullifiers.push(new Nullifier(value, noteHash, counter).scope(addr));
91
+ return this;
92
+ }
93
+
94
+ /** Adds a pending note hash read request (non-empty contract address, can match a pending note hash). */
95
+ addPendingNoteHashReadRequest(opts?: { value?: Fr; counter?: number; contractAddress?: AztecAddress }): this {
96
+ const value = opts?.value ?? Fr.random();
97
+ const counter = this.getCounter(opts?.counter);
98
+ const addr = opts?.contractAddress ?? this.contractAddress;
99
+ this.noteHashReadRequests.push(new ScopedReadRequest(new ReadRequest(value, counter), addr));
100
+ return this;
101
+ }
102
+
103
+ /** Adds a settled note hash read request (empty contract address, resolved against the note hash tree). */
104
+ addSettledNoteHashReadRequest(opts?: { value?: Fr; counter?: number }): this {
105
+ const value = opts?.value ?? Fr.random();
106
+ const counter = this.getCounter(opts?.counter);
107
+ this.noteHashReadRequests.push(new ScopedReadRequest(new ReadRequest(value, counter), AztecAddress.ZERO));
108
+ return this;
109
+ }
110
+
111
+ /** Adds a pending nullifier read request (non-empty contract address, can match a pending nullifier). */
112
+ addPendingNullifierReadRequest(opts?: { value?: Fr; counter?: number; contractAddress?: AztecAddress }): this {
113
+ const value = opts?.value ?? Fr.random();
114
+ const counter = this.getCounter(opts?.counter);
115
+ const addr = opts?.contractAddress ?? this.contractAddress;
116
+ this.nullifierReadRequests.push(new ScopedReadRequest(new ReadRequest(value, counter), addr));
117
+ return this;
118
+ }
119
+
120
+ /** Adds a settled nullifier read request (empty contract address, resolved against the nullifier tree). */
121
+ addSettledNullifierReadRequest(opts?: { value?: Fr; counter?: number }): this {
122
+ const value = opts?.value ?? Fr.random();
123
+ const counter = this.getCounter(opts?.counter);
124
+ this.nullifierReadRequests.push(new ScopedReadRequest(new ReadRequest(value, counter), AztecAddress.ZERO));
125
+ return this;
126
+ }
127
+
128
+ /** Adds a key validation request to validation requests. */
129
+ addKeyValidationRequest(opts?: { contractAddress?: AztecAddress }): this {
130
+ const addr = opts?.contractAddress ?? this.contractAddress;
131
+ this.keyValidationRequests.push(
132
+ new ScopedKeyValidationRequestAndSeparator(
133
+ new KeyValidationRequestAndSeparator(
134
+ new KeyValidationRequest(new Point(Fr.random(), Fr.random(), false), Fr.random()),
135
+ Fr.random(),
136
+ ),
137
+ addr,
138
+ ),
139
+ );
140
+ return this;
141
+ }
142
+
143
+ /** Adds a private log to the accumulated data. Defaults are generated randomly. */
144
+ addPrivateLog(opts?: { noteHashCounter?: number; counter?: number; contractAddress?: AztecAddress }): this {
145
+ const noteHashCounter = opts?.noteHashCounter ?? 0;
146
+ const counter = this.getCounter(opts?.counter);
147
+ const addr = opts?.contractAddress ?? this.contractAddress;
148
+ this.privateLogs.push(
149
+ new ScopedPrivateLogData(new PrivateLogData(PrivateLog.empty(), noteHashCounter, counter), addr),
150
+ );
151
+ return this;
152
+ }
153
+
154
+ /** Builds the PrivateKernelCircuitPublicInputs with all added side effects. */
155
+ build(): PrivateKernelCircuitPublicInputs {
156
+ const publicInputs = PrivateKernelCircuitPublicInputs.empty();
157
+ publicInputs.end.noteHashes = makeClaimed(this.noteHashes, ScopedNoteHash, MAX_NOTE_HASHES_PER_TX);
158
+ publicInputs.end.nullifiers = makeClaimed(this.nullifiers, ScopedNullifier, MAX_NULLIFIERS_PER_TX);
159
+ publicInputs.end.privateLogs = makeClaimed(this.privateLogs, ScopedPrivateLogData, MAX_PRIVATE_LOGS_PER_TX);
160
+ publicInputs.validationRequests.noteHashReadRequests = makeClaimed(
161
+ this.noteHashReadRequests,
162
+ ScopedReadRequest,
163
+ MAX_NOTE_HASH_READ_REQUESTS_PER_TX,
164
+ );
165
+ publicInputs.validationRequests.nullifierReadRequests = makeClaimed(
166
+ this.nullifierReadRequests,
167
+ ScopedReadRequest,
168
+ MAX_NULLIFIER_READ_REQUESTS_PER_TX,
169
+ );
170
+ publicInputs.validationRequests.scopedKeyValidationRequestsAndSeparators = makeClaimed(
171
+ this.keyValidationRequests,
172
+ ScopedKeyValidationRequestAndSeparator,
173
+ MAX_KEY_VALIDATION_REQUESTS_PER_TX,
174
+ );
175
+ return publicInputs;
176
+ }
177
+ }
178
+
179
+ /** Builder for PrivateCircuitPublicInputs (call-level) with fluent API for adding side effects. */
180
+ export class PrivateCircuitPublicInputsBuilder {
181
+ private noteHashes: NoteHash[] = [];
182
+ private nullifiers: Nullifier[] = [];
183
+ private noteHashReadRequests: ScopedReadRequest[] = [];
184
+ private nullifierReadRequests: ScopedReadRequest[] = [];
185
+ private keyValidationRequests: KeyValidationRequestAndSeparator[] = [];
186
+ private privateLogs: PrivateLogData[] = [];
187
+ private nextCounter: number;
188
+
189
+ constructor(
190
+ private contractAddress: AztecAddress = DEFAULT_CONTRACT_ADDRESS,
191
+ startCounter = 1,
192
+ ) {
193
+ this.nextCounter = startCounter;
194
+ }
195
+
196
+ private getCounter(sideEffectCounter?: number): number {
197
+ if (sideEffectCounter !== undefined) {
198
+ this.nextCounter = sideEffectCounter + 1;
199
+ return sideEffectCounter;
200
+ }
201
+ return this.nextCounter++;
202
+ }
203
+
204
+ /** Adds a note hash. Defaults are generated randomly. */
205
+ addNoteHash(opts?: { value?: Fr; counter?: number }): this {
206
+ const value = opts?.value ?? Fr.random();
207
+ const counter = this.getCounter(opts?.counter);
208
+ this.noteHashes.push(new NoteHash(value, counter));
209
+ return this;
210
+ }
211
+
212
+ /** Adds a nullifier. Defaults are generated randomly. */
213
+ addNullifier(opts?: { value?: Fr; noteHash?: Fr; counter?: number }): this {
214
+ const value = opts?.value ?? Fr.random();
215
+ const noteHash = opts?.noteHash ?? Fr.ZERO;
216
+ const counter = this.getCounter(opts?.counter);
217
+ this.nullifiers.push(new Nullifier(value, noteHash, counter));
218
+ return this;
219
+ }
220
+
221
+ /** Adds a pending note hash read request (non-empty contract address, can match a pending note hash). */
222
+ addPendingNoteHashReadRequest(opts?: { value?: Fr; counter?: number }): this {
223
+ const value = opts?.value ?? Fr.random();
224
+ const counter = this.getCounter(opts?.counter);
225
+ this.noteHashReadRequests.push(new ScopedReadRequest(new ReadRequest(value, counter), this.contractAddress));
226
+ return this;
227
+ }
228
+
229
+ /** Adds a settled note hash read request (empty contract address, resolved against the note hash tree). */
230
+ addSettledNoteHashReadRequest(opts?: { value?: Fr; counter?: number }): this {
231
+ const value = opts?.value ?? Fr.random();
232
+ const counter = this.getCounter(opts?.counter);
233
+ this.noteHashReadRequests.push(new ScopedReadRequest(new ReadRequest(value, counter), AztecAddress.ZERO));
234
+ return this;
235
+ }
236
+
237
+ /** Adds a pending nullifier read request (non-empty contract address, can match a pending nullifier). */
238
+ addPendingNullifierReadRequest(opts?: { value?: Fr; counter?: number }): this {
239
+ const value = opts?.value ?? Fr.random();
240
+ const counter = this.getCounter(opts?.counter);
241
+ this.nullifierReadRequests.push(new ScopedReadRequest(new ReadRequest(value, counter), this.contractAddress));
242
+ return this;
243
+ }
244
+
245
+ /** Adds a settled nullifier read request (empty contract address, resolved against the nullifier tree). */
246
+ addSettledNullifierReadRequest(opts?: { value?: Fr; counter?: number }): this {
247
+ const value = opts?.value ?? Fr.random();
248
+ const counter = this.getCounter(opts?.counter);
249
+ this.nullifierReadRequests.push(new ScopedReadRequest(new ReadRequest(value, counter), AztecAddress.ZERO));
250
+ return this;
251
+ }
252
+
253
+ /** Adds a key validation request. */
254
+ addKeyValidationRequest(): this {
255
+ this.keyValidationRequests.push(
256
+ new KeyValidationRequestAndSeparator(
257
+ new KeyValidationRequest(new Point(Fr.random(), Fr.random(), false), Fr.random()),
258
+ Fr.random(),
259
+ ),
260
+ );
261
+ return this;
262
+ }
263
+
264
+ /** Adds a private log. Defaults are generated randomly. */
265
+ addPrivateLog(opts?: { noteHashCounter?: number; counter?: number }): this {
266
+ const noteHashCounter = opts?.noteHashCounter ?? 0;
267
+ const counter = this.getCounter(opts?.counter);
268
+ this.privateLogs.push(new PrivateLogData(PrivateLog.empty(), noteHashCounter, counter));
269
+ return this;
270
+ }
271
+
272
+ /** Builds the PrivateCircuitPublicInputs with all added side effects. */
273
+ build(): PrivateCircuitPublicInputs {
274
+ const publicInputs = PrivateCircuitPublicInputs.empty();
275
+ publicInputs.callContext.contractAddress = this.contractAddress;
276
+ publicInputs.noteHashes = makeClaimed(this.noteHashes, NoteHash, MAX_NOTE_HASHES_PER_CALL);
277
+ publicInputs.nullifiers = makeClaimed(this.nullifiers, Nullifier, MAX_NULLIFIERS_PER_CALL);
278
+ publicInputs.privateLogs = makeClaimed(this.privateLogs, PrivateLogData, MAX_PRIVATE_LOGS_PER_CALL);
279
+ publicInputs.noteHashReadRequests = makeClaimed(
280
+ this.noteHashReadRequests,
281
+ ScopedReadRequest,
282
+ MAX_NOTE_HASH_READ_REQUESTS_PER_CALL,
283
+ );
284
+ publicInputs.nullifierReadRequests = makeClaimed(
285
+ this.nullifierReadRequests,
286
+ ScopedReadRequest,
287
+ MAX_NULLIFIER_READ_REQUESTS_PER_CALL,
288
+ );
289
+ publicInputs.keyValidationRequestsAndSeparators = makeClaimed(
290
+ this.keyValidationRequests,
291
+ KeyValidationRequestAndSeparator,
292
+ MAX_KEY_VALIDATION_REQUESTS_PER_CALL,
293
+ );
294
+ return publicInputs;
295
+ }
296
+ }
297
+
298
+ /** Wraps a PrivateKernelCircuitPublicInputs in a PrivateKernelSimulateOutput. */
299
+ export function makeKernelOutput(
300
+ publicInputs?: PrivateKernelCircuitPublicInputs,
301
+ ): PrivateKernelSimulateOutput<PrivateKernelCircuitPublicInputs> {
302
+ return {
303
+ publicInputs: publicInputs ?? PrivateKernelCircuitPublicInputs.empty(),
304
+ verificationKey: VerificationKeyData.empty(),
305
+ outputWitness: new Map(),
306
+ bytecode: Buffer.from([]),
307
+ };
308
+ }
309
+
310
+ /** Wraps a PrivateCircuitPublicInputs in a PrivateCallExecutionResult. */
311
+ export function makeExecutionResult(publicInputs?: PrivateCircuitPublicInputs): PrivateCallExecutionResult {
312
+ return new PrivateCallExecutionResult(
313
+ Buffer.alloc(0),
314
+ Buffer.alloc(0),
315
+ new Map(),
316
+ publicInputs ?? PrivateCircuitPublicInputs.empty(),
317
+ [],
318
+ new Map(),
319
+ [],
320
+ [],
321
+ [],
322
+ [],
323
+ [],
324
+ );
325
+ }
@@ -116,6 +116,7 @@ export class PrivateKernelExecutionProver {
116
116
  splitCounter,
117
117
  );
118
118
  while (resetBuilder.needsReset()) {
119
+ // Inner reset: without siloing.
119
120
  const witgenTimer = new Timer();
120
121
  const privateInputs = await resetBuilder.build(this.oracle);
121
122
  output = generateWitnesses
@@ -216,16 +217,24 @@ export class PrivateKernelExecutionProver {
216
217
  firstIteration = false;
217
218
  }
218
219
 
219
- // Reset.
220
- let resetBuilder = new PrivateKernelResetPrivateInputsBuilder(
220
+ // Final reset: include siloing of note hashes, nullifiers and private logs.
221
+ const finalResetBuilder = new PrivateKernelResetPrivateInputsBuilder(
221
222
  output,
222
223
  [],
223
224
  noteHashNullifierCounterMap,
224
225
  splitCounter,
225
226
  );
226
- while (resetBuilder.needsReset()) {
227
+ if (!finalResetBuilder.needsReset()) {
228
+ // The final reset must be performed exactly once, because each tx has at least one nullifier that requires
229
+ // siloing, and siloing cannot be done multiple times.
230
+ // While, in theory, it might be possible to silo note hashes first and then run another reset to silo nullifiers
231
+ // and/or private logs, we currently don't have standalone dimensions for the arrays that require siloing. As a
232
+ // result, all necessary siloing must be done together in a single reset.
233
+ // Refer to the possible combinations of dimensions in private_kernel_reset_config.json.
234
+ throw new Error('Nothing to reset for the final reset.');
235
+ } else {
227
236
  const witgenTimer = new Timer();
228
- const privateInputs = await resetBuilder.build(this.oracle);
237
+ const privateInputs = await finalResetBuilder.build(this.oracle);
229
238
  output = generateWitnesses
230
239
  ? await this.proofCreator.generateResetOutput(privateInputs)
231
240
  : await this.proofCreator.simulateReset(privateInputs);
@@ -239,8 +248,6 @@ export class PrivateKernelExecutionProver {
239
248
  witgen: witgenTimer.ms(),
240
249
  },
241
250
  });
242
-
243
- resetBuilder = new PrivateKernelResetPrivateInputsBuilder(output, [], noteHashNullifierCounterMap, splitCounter);
244
251
  }
245
252
 
246
253
  if (output.publicInputs.feePayer.isZero() && skipFeeEnforcement) {
@@ -8,11 +8,7 @@ import { ProtocolContractAddress } from '@aztec/protocol-contracts';
8
8
  import type { FunctionSelector } from '@aztec/stdlib/abi';
9
9
  import type { AztecAddress } from '@aztec/stdlib/aztec-address';
10
10
  import { BlockHash } from '@aztec/stdlib/block';
11
- import {
12
- type ContractInstanceWithAddress,
13
- computeContractClassIdPreimage,
14
- computeSaltedInitializationHash,
15
- } from '@aztec/stdlib/contract';
11
+ import { type ContractInstanceWithAddress, computeSaltedInitializationHash } from '@aztec/stdlib/contract';
16
12
  import { DelayedPublicMutableValues, DelayedPublicMutableValuesWithHash } from '@aztec/stdlib/delayed-public-mutable';
17
13
  import { computePublicDataTreeLeafSlot } from '@aztec/stdlib/hash';
18
14
  import type { AztecNode } from '@aztec/stdlib/interfaces/client';
@@ -49,11 +45,15 @@ export class PrivateKernelOracle {
49
45
 
50
46
  /** Retrieves the preimage of a contract class id from the contract classes db. */
51
47
  public async getContractClassIdPreimage(contractClassId: Fr) {
52
- const contractClass = await this.contractStore.getContractClass(contractClassId);
48
+ const contractClass = await this.contractStore.getContractClassWithPreimage(contractClassId);
53
49
  if (!contractClass) {
54
50
  throw new Error(`Contract class not found when getting class id preimage. Class id: ${contractClassId}.`);
55
51
  }
56
- return computeContractClassIdPreimage(contractClass);
52
+ return {
53
+ artifactHash: contractClass.artifactHash,
54
+ privateFunctionsRoot: contractClass.privateFunctionsRoot,
55
+ publicBytecodeCommitment: contractClass.publicBytecodeCommitment,
56
+ };
57
57
  }
58
58
 
59
59
  /** Returns a membership witness with the sibling path and leaf index in our private functions tree. */
package/src/pxe.ts CHANGED
@@ -47,7 +47,7 @@ import {
47
47
  TxProfileResult,
48
48
  TxProvingResult,
49
49
  TxSimulationResult,
50
- UtilitySimulationResult,
50
+ UtilityExecutionResult,
51
51
  } from '@aztec/stdlib/tx';
52
52
 
53
53
  import { inspect } from 'util';
@@ -61,6 +61,7 @@ import {
61
61
  generateSimulatedProvingResult,
62
62
  } from './contract_function_simulator/contract_function_simulator.js';
63
63
  import { ProxiedContractStoreFactory } from './contract_function_simulator/proxied_contract_data_source.js';
64
+ import { displayDebugLogs } from './contract_logging.js';
64
65
  import { ContractSyncService } from './contract_sync/contract_sync_service.js';
65
66
  import { readCurrentClassId } from './contract_sync/helpers.js';
66
67
  import { PXEDebugUtils } from './debug/pxe_debug_utils.js';
@@ -111,8 +112,8 @@ export type SimulateTxOpts = {
111
112
  scopes: AccessScopes;
112
113
  };
113
114
 
114
- /** Options for PXE.simulateUtility. */
115
- export type SimulateUtilityOpts = {
115
+ /** Options for PXE.executeUtility. */
116
+ export type ExecuteUtilityOpts = {
116
117
  /** The authentication witnesses required for the function call. */
117
118
  authwits?: AuthWitness[];
118
119
  /** The accounts whose notes we can access in this call */
@@ -264,7 +265,7 @@ export class PXE {
264
265
  debugUtils.setPXEHelpers(
265
266
  pxe.#putInJobQueue.bind(pxe),
266
267
  pxe.#getSimulatorForTx.bind(pxe),
267
- pxe.#simulateUtility.bind(pxe),
268
+ pxe.#executeUtility.bind(pxe),
268
269
  );
269
270
 
270
271
  pxe.jobQueue.start();
@@ -344,9 +345,8 @@ export class PXE {
344
345
  async #registerProtocolContracts() {
345
346
  const registered: Record<string, string> = {};
346
347
  for (const name of protocolContractNames) {
347
- const { address, contractClass, instance, artifact } =
348
- await this.protocolContractsProvider.getProtocolContractArtifact(name);
349
- await this.contractStore.addContractArtifact(contractClass.id, artifact);
348
+ const { address, instance, artifact } = await this.protocolContractsProvider.getProtocolContractArtifact(name);
349
+ await this.contractStore.addContractArtifact(artifact);
350
350
  await this.contractStore.addContractInstance(instance);
351
351
  registered[name] = address.toString();
352
352
  }
@@ -370,7 +370,7 @@ export class PXE {
370
370
  contractAddress,
371
371
  functionSelector,
372
372
  (privateSyncCall, execScopes) =>
373
- this.#simulateUtility(contractFunctionSimulator, privateSyncCall, [], execScopes, jobId),
373
+ this.#executeUtility(contractFunctionSimulator, privateSyncCall, [], execScopes, jobId),
374
374
  anchorBlockHeader,
375
375
  jobId,
376
376
  scopes,
@@ -394,16 +394,16 @@ export class PXE {
394
394
  }
395
395
 
396
396
  /**
397
- * Simulate a utility function call on the given contract.
397
+ * Execute a utility function call on the given contract.
398
398
  * @param contractFunctionSimulator - The simulator to use for the function call.
399
399
  * @param call - The function call to execute.
400
400
  * @param authWitnesses - Authentication witnesses required for the function call.
401
401
  * @param scopes - Optional array of account addresses whose notes can be accessed in this call. Defaults to all
402
402
  * accounts if not specified.
403
403
  * @param jobId - The job ID for staged writes.
404
- * @returns The simulation result containing the outputs of the utility function.
404
+ * @returns The execution result containing the outputs of the utility function.
405
405
  */
406
- async #simulateUtility(
406
+ async #executeUtility(
407
407
  contractFunctionSimulator: ContractFunctionSimulator,
408
408
  call: FunctionCall,
409
409
  authWitnesses: AuthWitness[] | undefined,
@@ -601,8 +601,7 @@ export class PXE {
601
601
  * @param artifact - The build artifact for the contract class.
602
602
  */
603
603
  public async registerContractClass(artifact: ContractArtifact): Promise<void> {
604
- const { id: contractClassId } = await getContractClassFromArtifact(artifact);
605
- await this.contractStore.addContractArtifact(contractClassId, artifact);
604
+ const contractClassId = await this.contractStore.addContractArtifact(artifact);
606
605
  this.log.info(`Added contract class ${artifact.name} with id ${contractClassId}`);
607
606
  }
608
607
 
@@ -621,17 +620,17 @@ export class PXE {
621
620
  if (artifact) {
622
621
  // If the user provides an artifact, validate it against the expected class id and register it
623
622
  const contractClass = await getContractClassFromArtifact(artifact);
624
- const contractClassId = contractClass.id;
625
- if (!contractClassId.equals(instance.currentContractClassId)) {
623
+ if (!contractClass.id.equals(instance.currentContractClassId)) {
626
624
  throw new Error(
627
- `Artifact does not match expected class id (computed ${contractClassId} but instance refers to ${instance.currentContractClassId})`,
625
+ `Artifact does not match expected class id (computed ${contractClass.id} but instance refers to ${instance.currentContractClassId})`,
628
626
  );
629
627
  }
630
628
  const computedAddress = await computeContractAddressFromInstance(instance);
631
629
  if (!computedAddress.equals(instance.address)) {
632
630
  throw new Error('Added a contract in which the address does not match the contract instance.');
633
631
  }
634
- await this.contractStore.addContractArtifact(contractClass.id, artifact);
632
+
633
+ await this.contractStore.addContractArtifact(artifact, contractClass);
635
634
 
636
635
  const publicFunctionSignatures = artifact.functions
637
636
  .filter(fn => fn.functionType === FunctionType.PUBLIC)
@@ -680,15 +679,16 @@ export class PXE {
680
679
  throw new Error('Could not update contract to a class different from the current one.');
681
680
  }
682
681
 
683
- await this.contractStore.addContractArtifact(contractClass.id, artifact);
684
-
685
682
  const publicFunctionSignatures = artifact.functions
686
683
  .filter(fn => fn.functionType === FunctionType.PUBLIC)
687
684
  .map(fn => decodeFunctionSignature(fn.name, fn.parameters));
688
685
  await this.node.registerContractFunctionSignatures(publicFunctionSignatures);
689
686
 
690
687
  currentInstance.currentContractClassId = contractClass.id;
691
- await this.contractStore.addContractInstance(currentInstance);
688
+ await Promise.all([
689
+ this.contractStore.addContractArtifact(artifact, contractClass),
690
+ this.contractStore.addContractInstance(currentInstance),
691
+ ]);
692
692
  this.log.info(`Updated contract ${artifact.name} at ${contractAddress.toString()} to class ${contractClass.id}`);
693
693
  });
694
694
  }
@@ -947,6 +947,9 @@ export class PXE {
947
947
  const publicSimulationTimer = new Timer();
948
948
  publicOutput = await this.#simulatePublicCalls(simulatedTx, skipFeeEnforcement);
949
949
  publicSimulationTime = publicSimulationTimer.ms();
950
+ if (publicOutput?.debugLogs?.length) {
951
+ await displayDebugLogs(publicOutput.debugLogs, addr => this.contractStore.getDebugContractName(addr));
952
+ }
950
953
  }
951
954
 
952
955
  let validationTime: number | undefined;
@@ -1013,16 +1016,16 @@ export class PXE {
1013
1016
  }
1014
1017
 
1015
1018
  /**
1016
- * Simulates the execution of a contract utility function.
1019
+ * Executes a contract utility function.
1017
1020
  * @param call - The function call containing the function details, arguments, and target contract address.
1018
1021
  */
1019
- public simulateUtility(
1022
+ public executeUtility(
1020
1023
  call: FunctionCall,
1021
- { authwits, scopes }: SimulateUtilityOpts = { scopes: 'ALL_SCOPES' },
1022
- ): Promise<UtilitySimulationResult> {
1023
- // We disable concurrent simulations since those might execute oracles which read and write to the PXE stores (e.g.
1024
+ { authwits, scopes }: ExecuteUtilityOpts = { scopes: 'ALL_SCOPES' },
1025
+ ): Promise<UtilityExecutionResult> {
1026
+ // We disable concurrent executions since those might execute oracles which read and write to the PXE stores (e.g.
1024
1027
  // to the capsules), and we need to prevent concurrent runs from interfering with one another (e.g. attempting to
1025
- // delete the same read value, or reading values that another simulation is currently modifying).
1028
+ // delete the same read value, or reading values that another execution is currently modifying).
1026
1029
  return this.#putInJobQueue(async jobId => {
1027
1030
  try {
1028
1031
  const totalTimer = new Timer();
@@ -1037,13 +1040,13 @@ export class PXE {
1037
1040
  call.to,
1038
1041
  call.selector,
1039
1042
  (privateSyncCall, execScopes) =>
1040
- this.#simulateUtility(contractFunctionSimulator, privateSyncCall, [], execScopes, jobId),
1043
+ this.#executeUtility(contractFunctionSimulator, privateSyncCall, [], execScopes, jobId),
1041
1044
  anchorBlockHeader,
1042
1045
  jobId,
1043
1046
  scopes,
1044
1047
  );
1045
1048
 
1046
- const executionResult = await this.#simulateUtility(
1049
+ const executionResult = await this.#executeUtility(
1047
1050
  contractFunctionSimulator,
1048
1051
  call,
1049
1052
  authwits ?? [],
@@ -1070,7 +1073,7 @@ export class PXE {
1070
1073
  const stringifiedArgs = args.map(arg => arg.toString()).join(', ');
1071
1074
  throw this.#contextualizeError(
1072
1075
  err,
1073
- `simulateUtility ${to}:${name}(${stringifiedArgs})`,
1076
+ `executeUtility ${to}:${name}(${stringifiedArgs})`,
1074
1077
  `scopes=${scopes === 'ALL_SCOPES' ? scopes : scopes.map(s => s.toString()).join(', ')}`,
1075
1078
  );
1076
1079
  }
@@ -1108,7 +1111,7 @@ export class PXE {
1108
1111
  filter.contractAddress,
1109
1112
  null,
1110
1113
  async (privateSyncCall, execScopes) =>
1111
- await this.#simulateUtility(contractFunctionSimulator, privateSyncCall, [], execScopes, jobId),
1114
+ await this.#executeUtility(contractFunctionSimulator, privateSyncCall, [], execScopes, jobId),
1112
1115
  anchorBlockHeader,
1113
1116
  jobId,
1114
1117
  filter.scopes,