@aztec/txe 0.0.1-commit.7d4e6cd → 0.0.1-commit.7ffbba4

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 (64) hide show
  1. package/dest/constants.d.ts +1 -2
  2. package/dest/constants.d.ts.map +1 -1
  3. package/dest/constants.js +0 -1
  4. package/dest/index.d.ts +1 -1
  5. package/dest/index.d.ts.map +1 -1
  6. package/dest/index.js +88 -54
  7. package/dest/oracle/interfaces.d.ts +29 -28
  8. package/dest/oracle/interfaces.d.ts.map +1 -1
  9. package/dest/oracle/txe_oracle_public_context.d.ts +13 -13
  10. package/dest/oracle/txe_oracle_public_context.d.ts.map +1 -1
  11. package/dest/oracle/txe_oracle_public_context.js +16 -16
  12. package/dest/oracle/txe_oracle_top_level_context.d.ts +22 -22
  13. package/dest/oracle/txe_oracle_top_level_context.d.ts.map +1 -1
  14. package/dest/oracle/txe_oracle_top_level_context.js +138 -61
  15. package/dest/rpc_translator.d.ts +86 -81
  16. package/dest/rpc_translator.d.ts.map +1 -1
  17. package/dest/rpc_translator.js +291 -175
  18. package/dest/state_machine/archiver.d.ts +20 -69
  19. package/dest/state_machine/archiver.d.ts.map +1 -1
  20. package/dest/state_machine/archiver.js +36 -178
  21. package/dest/state_machine/dummy_p2p_client.d.ts +16 -12
  22. package/dest/state_machine/dummy_p2p_client.d.ts.map +1 -1
  23. package/dest/state_machine/dummy_p2p_client.js +28 -16
  24. package/dest/state_machine/index.d.ts +5 -5
  25. package/dest/state_machine/index.d.ts.map +1 -1
  26. package/dest/state_machine/index.js +35 -12
  27. package/dest/state_machine/mock_epoch_cache.d.ts +8 -6
  28. package/dest/state_machine/mock_epoch_cache.d.ts.map +1 -1
  29. package/dest/state_machine/mock_epoch_cache.js +11 -7
  30. package/dest/state_machine/synchronizer.d.ts +3 -3
  31. package/dest/state_machine/synchronizer.d.ts.map +1 -1
  32. package/dest/txe_session.d.ts +11 -6
  33. package/dest/txe_session.d.ts.map +1 -1
  34. package/dest/txe_session.js +101 -26
  35. package/dest/util/encoding.d.ts +17 -17
  36. package/dest/util/txe_public_contract_data_source.d.ts +2 -3
  37. package/dest/util/txe_public_contract_data_source.d.ts.map +1 -1
  38. package/dest/util/txe_public_contract_data_source.js +5 -22
  39. package/dest/utils/block_creation.d.ts +4 -4
  40. package/dest/utils/block_creation.d.ts.map +1 -1
  41. package/dest/utils/block_creation.js +18 -4
  42. package/dest/utils/tx_effect_creation.d.ts +2 -3
  43. package/dest/utils/tx_effect_creation.d.ts.map +1 -1
  44. package/dest/utils/tx_effect_creation.js +3 -6
  45. package/package.json +16 -16
  46. package/src/constants.ts +0 -1
  47. package/src/index.ts +89 -52
  48. package/src/oracle/interfaces.ts +32 -31
  49. package/src/oracle/txe_oracle_public_context.ts +16 -18
  50. package/src/oracle/txe_oracle_top_level_context.ts +173 -111
  51. package/src/rpc_translator.ts +313 -198
  52. package/src/state_machine/archiver.ts +37 -234
  53. package/src/state_machine/dummy_p2p_client.ts +40 -22
  54. package/src/state_machine/index.ts +49 -11
  55. package/src/state_machine/mock_epoch_cache.ts +11 -11
  56. package/src/state_machine/synchronizer.ts +2 -2
  57. package/src/txe_session.ts +115 -87
  58. package/src/util/txe_public_contract_data_source.ts +10 -36
  59. package/src/utils/block_creation.ts +19 -16
  60. package/src/utils/tx_effect_creation.ts +3 -11
  61. package/dest/util/txe_contract_store.d.ts +0 -12
  62. package/dest/util/txe_contract_store.d.ts.map +0 -1
  63. package/dest/util/txe_contract_store.js +0 -22
  64. package/src/util/txe_contract_store.ts +0 -36
@@ -3,10 +3,13 @@ import { Fr } from '@aztec/foundation/curves/bn254';
3
3
  import { type Logger, createLogger } from '@aztec/foundation/log';
4
4
  import { KeyStore } from '@aztec/key-store';
5
5
  import { openTmpStore } from '@aztec/kv-store/lmdb-v2';
6
- import type { ProtocolContract } from '@aztec/protocol-contracts';
6
+ import type { AccessScopes } from '@aztec/pxe/client/lazy';
7
7
  import {
8
8
  AddressStore,
9
+ AnchorBlockStore,
9
10
  CapsuleStore,
11
+ ContractStore,
12
+ JobCoordinator,
10
13
  NoteService,
11
14
  NoteStore,
12
15
  PrivateEventStore,
@@ -43,15 +46,15 @@ import { CallContext, GlobalVariables, TxContext } from '@aztec/stdlib/tx';
43
46
 
44
47
  import { z } from 'zod';
45
48
 
46
- import { DEFAULT_ADDRESS, TXE_JOB_ID } from './constants.js';
49
+ import { DEFAULT_ADDRESS } from './constants.js';
47
50
  import type { IAvmExecutionOracle, ITxeExecutionOracle } from './oracle/interfaces.js';
48
51
  import { TXEOraclePublicContext } from './oracle/txe_oracle_public_context.js';
49
52
  import { TXEOracleTopLevelContext } from './oracle/txe_oracle_top_level_context.js';
50
53
  import { RPCTranslator } from './rpc_translator.js';
54
+ import { TXEArchiver } from './state_machine/archiver.js';
51
55
  import { TXEStateMachine } from './state_machine/index.js';
52
56
  import type { ForeignCallArgs, ForeignCallResult } from './util/encoding.js';
53
57
  import { TXEAccountStore } from './util/txe_account_store.js';
54
- import { TXEContractStore } from './util/txe_contract_store.js';
55
58
  import { getSingleTxBlockRequestHash, insertTxEffectIntoWorldTrees, makeTXEBlock } from './utils/block_creation.js';
56
59
  import { makeTxEffect } from './utils/tx_effect_creation.js';
57
60
 
@@ -74,7 +77,6 @@ type SessionState =
74
77
  | {
75
78
  name: 'PRIVATE';
76
79
  nextBlockGlobalVariables: GlobalVariables;
77
- protocolNullifier: Fr;
78
80
  noteCache: ExecutionNoteCache;
79
81
  taggingIndexCache: ExecutionTaggingIndexCache;
80
82
  }
@@ -111,6 +113,10 @@ export interface TXESessionStateHandler {
111
113
  enterPublicState(contractAddress?: AztecAddress): Promise<void>;
112
114
  enterPrivateState(contractAddress?: AztecAddress, anchorBlockNumber?: BlockNumber): Promise<PrivateContextInputs>;
113
115
  enterUtilityState(contractAddress?: AztecAddress): Promise<void>;
116
+
117
+ // TODO(F-335): Exposing the job info is abstraction breakage - drop the following 2 functions.
118
+ cycleJob(): Promise<string>;
119
+ getCurrentJob(): string;
114
120
  }
115
121
 
116
122
  /**
@@ -129,7 +135,7 @@ export class TXESession implements TXESessionStateHandler {
129
135
  | IPrivateExecutionOracle
130
136
  | IAvmExecutionOracle
131
137
  | ITxeExecutionOracle,
132
- private contractStore: TXEContractStore,
138
+ private contractStore: ContractStore,
133
139
  private noteStore: NoteStore,
134
140
  private keyStore: KeyStore,
135
141
  private addressStore: AddressStore,
@@ -139,18 +145,19 @@ export class TXESession implements TXESessionStateHandler {
139
145
  private senderAddressBookStore: SenderAddressBookStore,
140
146
  private capsuleStore: CapsuleStore,
141
147
  private privateEventStore: PrivateEventStore,
148
+ private jobCoordinator: JobCoordinator,
149
+ private currentJobId: string,
142
150
  private chainId: Fr,
143
151
  private version: Fr,
144
152
  private nextBlockTimestamp: bigint,
145
153
  ) {}
146
154
 
147
- static async init(protocolContracts: ProtocolContract[]) {
155
+ static async init(contractStore: ContractStore) {
148
156
  const store = await openTmpStore('txe-session');
149
157
 
150
158
  const addressStore = new AddressStore(store);
151
159
  const privateEventStore = new PrivateEventStore(store);
152
- const contractStore = new TXEContractStore(store);
153
- const noteStore = await NoteStore.create(store);
160
+ const noteStore = new NoteStore(store);
154
161
  const senderTaggingStore = new SenderTaggingStore(store);
155
162
  const recipientTaggingStore = new RecipientTaggingStore(store);
156
163
  const senderAddressBookStore = new SenderAddressBookStore(store);
@@ -158,18 +165,26 @@ export class TXESession implements TXESessionStateHandler {
158
165
  const keyStore = new KeyStore(store);
159
166
  const accountStore = new TXEAccountStore(store);
160
167
 
161
- // Register protocol contracts.
162
- for (const { contractClass, instance, artifact } of protocolContracts) {
163
- await contractStore.addContractArtifact(contractClass.id, artifact);
164
- await contractStore.addContractInstance(instance);
165
- }
168
+ // Create job coordinator and register staged stores
169
+ const jobCoordinator = new JobCoordinator(store);
170
+ jobCoordinator.registerStores([
171
+ capsuleStore,
172
+ senderTaggingStore,
173
+ recipientTaggingStore,
174
+ privateEventStore,
175
+ noteStore,
176
+ ]);
166
177
 
167
- const stateMachine = await TXEStateMachine.create(store);
178
+ const archiver = new TXEArchiver(store);
179
+ const anchorBlockStore = new AnchorBlockStore(store);
180
+ const stateMachine = await TXEStateMachine.create(archiver, anchorBlockStore, contractStore, noteStore);
168
181
 
169
182
  const nextBlockTimestamp = BigInt(Math.floor(new Date().getTime() / 1000));
170
183
  const version = new Fr(await stateMachine.node.getVersion());
171
184
  const chainId = new Fr(await stateMachine.node.getChainId());
172
185
 
186
+ const initialJobId = jobCoordinator.beginJob();
187
+
173
188
  const topLevelOracleHandler = new TXEOracleTopLevelContext(
174
189
  stateMachine,
175
190
  contractStore,
@@ -187,7 +202,7 @@ export class TXESession implements TXESessionStateHandler {
187
202
  chainId,
188
203
  new Map(),
189
204
  );
190
- await topLevelOracleHandler.txeAdvanceBlocksBy(1);
205
+ await topLevelOracleHandler.advanceBlocksBy(1);
191
206
 
192
207
  return new TXESession(
193
208
  createLogger('txe:session'),
@@ -203,6 +218,8 @@ export class TXESession implements TXESessionStateHandler {
203
218
  senderAddressBookStore,
204
219
  capsuleStore,
205
220
  privateEventStore,
221
+ jobCoordinator,
222
+ initialJobId,
206
223
  version,
207
224
  chainId,
208
225
  nextBlockTimestamp,
@@ -240,6 +257,17 @@ export class TXESession implements TXESessionStateHandler {
240
257
  }
241
258
  }
242
259
 
260
+ getCurrentJob(): string {
261
+ return this.currentJobId;
262
+ }
263
+
264
+ /** Commits the current job and begins a new one. Returns the new job ID. */
265
+ async cycleJob(): Promise<string> {
266
+ await this.jobCoordinator.commitJob(this.currentJobId);
267
+ this.currentJobId = this.jobCoordinator.beginJob();
268
+ return this.currentJobId;
269
+ }
270
+
243
271
  async enterTopLevelState() {
244
272
  switch (this.state.name) {
245
273
  case 'PRIVATE': {
@@ -262,6 +290,9 @@ export class TXESession implements TXESessionStateHandler {
262
290
  }
263
291
  }
264
292
 
293
+ // Commit all staged stores from the job that was just completed, then begin a new job
294
+ await this.cycleJob();
295
+
265
296
  this.oracleHandler = new TXEOracleTopLevelContext(
266
297
  this.stateMachine,
267
298
  this.contractStore,
@@ -290,16 +321,15 @@ export class TXESession implements TXESessionStateHandler {
290
321
  ): Promise<PrivateContextInputs> {
291
322
  this.exitTopLevelState();
292
323
 
293
- await new NoteService(
294
- this.noteStore,
295
- this.stateMachine.node,
296
- this.stateMachine.anchorBlockStore,
297
- ).syncNoteNullifiers(contractAddress);
298
-
299
324
  // Private execution has two associated block numbers: the anchor block (i.e. the historical block that is used to
300
325
  // build the proof), and the *next* block, i.e. the one we'll create once the execution ends, and which will contain
301
326
  // a single transaction with the effects of what was done in the test.
302
327
  const anchorBlock = await this.stateMachine.node.getBlockHeader(anchorBlockNumber ?? 'latest');
328
+
329
+ await new NoteService(this.noteStore, this.stateMachine.node, anchorBlock!, this.currentJobId).syncNoteNullifiers(
330
+ contractAddress,
331
+ 'ALL_SCOPES',
332
+ );
303
333
  const latestBlock = await this.stateMachine.node.getBlockHeader('latest');
304
334
 
305
335
  const nextBlockGlobalVariables = makeGlobalVariables(undefined, {
@@ -315,37 +345,38 @@ export class TXESession implements TXESessionStateHandler {
315
345
  const taggingIndexCache = new ExecutionTaggingIndexCache();
316
346
 
317
347
  const utilityExecutor = this.utilityExecutorForContractSync(anchorBlock);
318
- this.oracleHandler = new PrivateExecutionOracle(
319
- Fr.ZERO,
320
- new TxContext(this.chainId, this.version, GasSettings.empty()),
321
- new CallContext(AztecAddress.ZERO, contractAddress, FunctionSelector.empty(), false),
322
- anchorBlock!,
348
+ this.oracleHandler = new PrivateExecutionOracle({
349
+ argsHash: Fr.ZERO,
350
+ txContext: new TxContext(this.chainId, this.version, GasSettings.empty()),
351
+ callContext: new CallContext(AztecAddress.ZERO, contractAddress, FunctionSelector.empty(), false),
352
+ anchorBlockHeader: anchorBlock!,
323
353
  utilityExecutor,
324
- [],
325
- [],
326
- new HashedValuesCache(),
354
+ authWitnesses: [],
355
+ capsules: [],
356
+ executionCache: new HashedValuesCache(),
327
357
  noteCache,
328
358
  taggingIndexCache,
329
- this.contractStore,
330
- this.noteStore,
331
- this.keyStore,
332
- this.addressStore,
333
- this.stateMachine.node,
334
- this.stateMachine.anchorBlockStore,
335
- this.senderTaggingStore,
336
- this.recipientTaggingStore,
337
- this.senderAddressBookStore,
338
- this.capsuleStore,
339
- this.privateEventStore,
340
- TXE_JOB_ID,
341
- );
359
+ contractStore: this.contractStore,
360
+ noteStore: this.noteStore,
361
+ keyStore: this.keyStore,
362
+ addressStore: this.addressStore,
363
+ aztecNode: this.stateMachine.node,
364
+ senderTaggingStore: this.senderTaggingStore,
365
+ recipientTaggingStore: this.recipientTaggingStore,
366
+ senderAddressBookStore: this.senderAddressBookStore,
367
+ capsuleStore: this.capsuleStore,
368
+ privateEventStore: this.privateEventStore,
369
+ contractSyncService: this.stateMachine.contractSyncService,
370
+ jobId: this.currentJobId,
371
+ scopes: 'ALL_SCOPES',
372
+ });
342
373
 
343
374
  // We store the note and tagging index caches fed into the PrivateExecutionOracle (along with some other auxiliary
344
375
  // data) in order to refer to it later, mimicking the way this object is used by the ContractFunctionSimulator. The
345
376
  // difference resides in that the simulator has all information needed in order to run the simulation, while ours
346
377
  // will be ongoing as the different oracles will be invoked from the Noir test, until eventually the private
347
378
  // execution finishes.
348
- this.state = { name: 'PRIVATE', nextBlockGlobalVariables, protocolNullifier, noteCache, taggingIndexCache };
379
+ this.state = { name: 'PRIVATE', nextBlockGlobalVariables, noteCache, taggingIndexCache };
349
380
  this.logger.debug(`Entered state ${this.state.name}`);
350
381
 
351
382
  return (this.oracleHandler as PrivateExecutionOracle).getPrivateContextInputs();
@@ -378,6 +409,8 @@ export class TXESession implements TXESessionStateHandler {
378
409
  async enterUtilityState(contractAddress: AztecAddress = DEFAULT_ADDRESS) {
379
410
  this.exitTopLevelState();
380
411
 
412
+ const anchorBlockHeader = await this.stateMachine.anchorBlockStore.getBlockHeader();
413
+
381
414
  // There is no automatic message discovery and contract-driven syncing process in inlined private or utility
382
415
  // contexts, which means that known nullifiers are also not searched for, since it is during the tagging sync that
383
416
  // we perform this. We therefore search for known nullifiers now, as otherwise notes that were nullified would not
@@ -386,28 +419,27 @@ export class TXESession implements TXESessionStateHandler {
386
419
  await new NoteService(
387
420
  this.noteStore,
388
421
  this.stateMachine.node,
389
- this.stateMachine.anchorBlockStore,
390
- ).syncNoteNullifiers(contractAddress);
391
-
392
- const anchorBlockHeader = await this.stateMachine.anchorBlockStore.getBlockHeader();
422
+ anchorBlockHeader,
423
+ this.currentJobId,
424
+ ).syncNoteNullifiers(contractAddress, 'ALL_SCOPES');
393
425
 
394
- this.oracleHandler = new UtilityExecutionOracle(
426
+ this.oracleHandler = new UtilityExecutionOracle({
395
427
  contractAddress,
396
- [],
397
- [],
428
+ authWitnesses: [],
429
+ capsules: [],
398
430
  anchorBlockHeader,
399
- this.contractStore,
400
- this.noteStore,
401
- this.keyStore,
402
- this.addressStore,
403
- this.stateMachine.node,
404
- this.stateMachine.anchorBlockStore,
405
- this.recipientTaggingStore,
406
- this.senderAddressBookStore,
407
- this.capsuleStore,
408
- this.privateEventStore,
409
- TXE_JOB_ID,
410
- );
431
+ contractStore: this.contractStore,
432
+ noteStore: this.noteStore,
433
+ keyStore: this.keyStore,
434
+ addressStore: this.addressStore,
435
+ aztecNode: this.stateMachine.node,
436
+ recipientTaggingStore: this.recipientTaggingStore,
437
+ senderAddressBookStore: this.senderAddressBookStore,
438
+ capsuleStore: this.capsuleStore,
439
+ privateEventStore: this.privateEventStore,
440
+ jobId: this.currentJobId,
441
+ scopes: 'ALL_SCOPES',
442
+ });
411
443
 
412
444
  this.state = { name: 'UTILITY' };
413
445
  this.logger.debug(`Entered state ${this.state.name}`);
@@ -420,8 +452,8 @@ export class TXESession implements TXESessionStateHandler {
420
452
 
421
453
  // Note that while all public and private contexts do is build a single block that we then process when exiting
422
454
  // those, the top level context performs a large number of actions not captured in the following 'close' call. Among
423
- // others, it will create empty blocks (via `txeAdvanceBlocksBy` and `deploy`), create blocks with transactions via
424
- // `txePrivateCallNewFlow` and `txePublicCallNewFlow`, add accounts to PXE via `txeAddAccount`, etc. This is a
455
+ // others, it will create empty blocks (via `advanceBlocksBy` and `deploy`), create blocks with transactions via
456
+ // `privateCallNewFlow` and `publicCallNewFlow`, add accounts to PXE via `addAccount`, etc. This is a
425
457
  // slight inconsistency in the working model of this class, but is not too bad.
426
458
  // TODO: it's quite unfortunate that we need to capture the authwits created to later pass them again when the top
427
459
  // level context is re-created. This is because authwits create a temporary utility context that'd otherwise reset
@@ -442,11 +474,7 @@ export class TXESession implements TXESessionStateHandler {
442
474
  // We rely on the note cache to determine the effects of the transaction. This is incomplete as it doesn't private
443
475
  // logs (other effects like enqueued public calls don't need to be considered since those are not allowed).
444
476
 
445
- const txEffect = await makeTxEffect(
446
- this.state.noteCache,
447
- this.state.protocolNullifier,
448
- this.state.nextBlockGlobalVariables.blockNumber,
449
- );
477
+ const txEffect = await makeTxEffect(this.state.noteCache, this.state.nextBlockGlobalVariables.blockNumber);
450
478
 
451
479
  // We build a block holding just this transaction
452
480
  const forkedWorldTrees = await this.stateMachine.synchronizer.nativeWorldStateService.fork();
@@ -479,30 +507,30 @@ export class TXESession implements TXESessionStateHandler {
479
507
  }
480
508
 
481
509
  private utilityExecutorForContractSync(anchorBlock: any) {
482
- return async (call: FunctionCall) => {
510
+ return async (call: FunctionCall, scopes: AccessScopes) => {
483
511
  const entryPointArtifact = await this.contractStore.getFunctionArtifactWithDebugMetadata(call.to, call.selector);
484
512
  if (entryPointArtifact.functionType !== FunctionType.UTILITY) {
485
513
  throw new Error(`Cannot run ${entryPointArtifact.functionType} function as utility`);
486
514
  }
487
515
 
488
516
  try {
489
- const oracle = new UtilityExecutionOracle(
490
- call.to,
491
- [],
492
- [],
493
- anchorBlock!,
494
- this.contractStore,
495
- this.noteStore,
496
- this.keyStore,
497
- this.addressStore,
498
- this.stateMachine.node,
499
- this.stateMachine.anchorBlockStore,
500
- this.recipientTaggingStore,
501
- this.senderAddressBookStore,
502
- this.capsuleStore,
503
- this.privateEventStore,
504
- TXE_JOB_ID,
505
- );
517
+ const oracle = new UtilityExecutionOracle({
518
+ contractAddress: call.to,
519
+ authWitnesses: [],
520
+ capsules: [],
521
+ anchorBlockHeader: anchorBlock!,
522
+ contractStore: this.contractStore,
523
+ noteStore: this.noteStore,
524
+ keyStore: this.keyStore,
525
+ addressStore: this.addressStore,
526
+ aztecNode: this.stateMachine.node,
527
+ recipientTaggingStore: this.recipientTaggingStore,
528
+ senderAddressBookStore: this.senderAddressBookStore,
529
+ capsuleStore: this.capsuleStore,
530
+ privateEventStore: this.privateEventStore,
531
+ jobId: this.currentJobId,
532
+ scopes,
533
+ });
506
534
  await new WASMSimulator()
507
535
  .executeUserCircuit(toACVMWitness(0, call.args), entryPointArtifact, new Oracle(oracle).toACIRCallback())
508
536
  .catch((err: Error) => {
@@ -1,19 +1,11 @@
1
1
  import { BlockNumber } from '@aztec/foundation/branded-types';
2
2
  import { Fr } from '@aztec/foundation/curves/bn254';
3
3
  import type { ContractStore } from '@aztec/pxe/server';
4
- import { type ContractArtifact, FunctionSelector, FunctionType } from '@aztec/stdlib/abi';
4
+ import { type ContractArtifact, FunctionSelector } from '@aztec/stdlib/abi';
5
5
  import type { AztecAddress } from '@aztec/stdlib/aztec-address';
6
- import {
7
- type ContractClassPublic,
8
- type ContractDataSource,
9
- type ContractInstanceWithAddress,
10
- computePrivateFunctionsRoot,
11
- computePublicBytecodeCommitment,
12
- getContractClassPrivateFunctionFromArtifact,
13
- } from '@aztec/stdlib/contract';
6
+ import type { ContractClassPublic, ContractDataSource, ContractInstanceWithAddress } from '@aztec/stdlib/contract';
14
7
 
15
8
  export class TXEPublicContractDataSource implements ContractDataSource {
16
- #privateFunctionsRoot: Map<string, Buffer> = new Map();
17
9
  constructor(
18
10
  private blockNumber: BlockNumber,
19
11
  private contractStore: ContractStore,
@@ -24,42 +16,24 @@ export class TXEPublicContractDataSource implements ContractDataSource {
24
16
  }
25
17
 
26
18
  async getContractClass(id: Fr): Promise<ContractClassPublic | undefined> {
27
- const contractClass = await this.contractStore.getContractClass(id);
19
+ const contractClass = await this.contractStore.getContractClassWithPreimage(id);
28
20
  if (!contractClass) {
29
21
  return;
30
22
  }
31
- const artifact = await this.contractStore.getContractArtifact(id);
32
- if (!artifact) {
33
- return;
34
- }
35
-
36
- let privateFunctionsRoot;
37
- if (!this.#privateFunctionsRoot.has(id.toString())) {
38
- const privateFunctions = await Promise.all(
39
- artifact.functions
40
- .filter(fn => fn.functionType === FunctionType.PRIVATE)
41
- .map(fn => getContractClassPrivateFunctionFromArtifact(fn)),
42
- );
43
- privateFunctionsRoot = await computePrivateFunctionsRoot(privateFunctions);
44
- this.#privateFunctionsRoot.set(id.toString(), privateFunctionsRoot.toBuffer());
45
- } else {
46
- privateFunctionsRoot = Fr.fromBuffer(this.#privateFunctionsRoot.get(id.toString())!);
47
- }
48
-
49
23
  return {
50
- id,
51
- artifactHash: contractClass!.artifactHash,
52
- packedBytecode: contractClass!.packedBytecode,
53
- privateFunctionsRoot,
54
- version: contractClass!.version,
24
+ id: contractClass.id,
25
+ artifactHash: contractClass.artifactHash,
26
+ packedBytecode: contractClass.packedBytecode,
27
+ privateFunctionsRoot: contractClass.privateFunctionsRoot,
28
+ version: contractClass.version,
55
29
  privateFunctions: [],
56
30
  utilityFunctions: [],
57
31
  };
58
32
  }
59
33
 
60
34
  async getBytecodeCommitment(id: Fr): Promise<Fr | undefined> {
61
- const contractClass = await this.contractStore.getContractClass(id);
62
- return contractClass && computePublicBytecodeCommitment(contractClass.packedBytecode);
35
+ const contractClass = await this.contractStore.getContractClassWithPreimage(id);
36
+ return contractClass?.publicBytecodeCommitment;
63
37
  }
64
38
 
65
39
  async getContract(address: AztecAddress): Promise<ContractInstanceWithAddress | undefined> {
@@ -4,12 +4,12 @@ import {
4
4
  NULLIFIER_SUBTREE_HEIGHT,
5
5
  NUMBER_OF_L1_L2_MESSAGES_PER_ROLLUP,
6
6
  } from '@aztec/constants';
7
- import { BlockNumber } from '@aztec/foundation/branded-types';
7
+ import { BlockNumber, CheckpointNumber, IndexWithinCheckpoint } from '@aztec/foundation/branded-types';
8
8
  import { padArrayEnd } from '@aztec/foundation/collection';
9
9
  import { Fr } from '@aztec/foundation/curves/bn254';
10
- import { Body, L2Block, L2BlockHeader } from '@aztec/stdlib/block';
10
+ import { Body, L2Block } from '@aztec/stdlib/block';
11
11
  import { AppendOnlyTreeSnapshot, MerkleTreeId, type MerkleTreeWriteOperations } from '@aztec/stdlib/trees';
12
- import { GlobalVariables, TxEffect } from '@aztec/stdlib/tx';
12
+ import { BlockHeader, GlobalVariables, TxEffect } from '@aztec/stdlib/tx';
13
13
 
14
14
  /**
15
15
  * Returns a transaction request hash that is valid for transactions that are the only ones in a block.
@@ -46,21 +46,18 @@ export async function insertTxEffectIntoWorldTrees(
46
46
  export async function makeTXEBlockHeader(
47
47
  worldTrees: MerkleTreeWriteOperations,
48
48
  globalVariables: GlobalVariables,
49
- ): Promise<L2BlockHeader> {
49
+ ): Promise<BlockHeader> {
50
50
  const stateReference = await worldTrees.getStateReference();
51
51
  const archiveInfo = await worldTrees.getTreeInfo(MerkleTreeId.ARCHIVE);
52
52
 
53
- return new L2BlockHeader(
54
- new AppendOnlyTreeSnapshot(new Fr(archiveInfo.root), Number(archiveInfo.size)),
55
- Fr.ZERO,
56
- Fr.ZERO,
57
- stateReference,
53
+ return BlockHeader.from({
54
+ lastArchive: new AppendOnlyTreeSnapshot(new Fr(archiveInfo.root), Number(archiveInfo.size)),
55
+ spongeBlobHash: Fr.ZERO,
56
+ state: stateReference,
58
57
  globalVariables,
59
- Fr.ZERO,
60
- Fr.ZERO,
61
- Fr.ZERO,
62
- Fr.ZERO,
63
- );
58
+ totalFees: Fr.ZERO,
59
+ totalManaUsed: Fr.ZERO,
60
+ });
64
61
  }
65
62
 
66
63
  /**
@@ -84,11 +81,17 @@ export async function makeTXEBlock(
84
81
  const header = await makeTXEBlockHeader(worldTrees, globalVariables);
85
82
 
86
83
  // Update the archive tree with this block's header hash
87
- await worldTrees.updateArchive(header.toBlockHeader());
84
+ await worldTrees.updateArchive(header);
88
85
 
89
86
  // Get the new archive state after updating
90
87
  const newArchiveInfo = await worldTrees.getTreeInfo(MerkleTreeId.ARCHIVE);
91
88
  const newArchive = new AppendOnlyTreeSnapshot(new Fr(newArchiveInfo.root), Number(newArchiveInfo.size));
92
89
 
93
- return new L2Block(newArchive, header, new Body(txEffects));
90
+ // L2Block requires checkpointNumber and indexWithinCheckpoint.
91
+ // TXE uses 1-block-per-checkpoint for testing simplicity, so we can use block number as checkpoint number.
92
+ // This uses the deprecated fromBlockNumber method intentionally for the TXE testing environment.
93
+ const checkpointNumber = CheckpointNumber.fromBlockNumber(globalVariables.blockNumber);
94
+ const indexWithinCheckpoint = IndexWithinCheckpoint(0);
95
+
96
+ return new L2Block(newArchive, header, new Body(txEffects), checkpointNumber, indexWithinCheckpoint);
94
97
  }
@@ -4,16 +4,12 @@ import type { ExecutionNoteCache } from '@aztec/pxe/simulator';
4
4
  import { computeNoteHashNonce, computeUniqueNoteHash, siloNoteHash } from '@aztec/stdlib/hash';
5
5
  import { TxEffect, TxHash } from '@aztec/stdlib/tx';
6
6
 
7
- export async function makeTxEffect(
8
- noteCache: ExecutionNoteCache,
9
- protocolNullifier: Fr,
10
- txBlockNumber: BlockNumber,
11
- ): Promise<TxEffect> {
7
+ export async function makeTxEffect(noteCache: ExecutionNoteCache, txBlockNumber: BlockNumber): Promise<TxEffect> {
12
8
  const txEffect = TxEffect.empty();
13
9
 
14
- const { usedProtocolNullifierForNonces } = noteCache.finish();
15
- const nonceGenerator = usedProtocolNullifierForNonces ? protocolNullifier : noteCache.getAllNullifiers()[0];
10
+ noteCache.finish();
16
11
 
12
+ const nonceGenerator = noteCache.getNonceGenerator();
17
13
  txEffect.noteHashes = await Promise.all(
18
14
  noteCache
19
15
  .getAllNotes()
@@ -28,10 +24,6 @@ export async function makeTxEffect(
28
24
  // Nullifiers are already siloed
29
25
  txEffect.nullifiers = noteCache.getAllNullifiers();
30
26
 
31
- if (usedProtocolNullifierForNonces) {
32
- txEffect.nullifiers.unshift(protocolNullifier);
33
- }
34
-
35
27
  txEffect.txHash = new TxHash(new Fr(txBlockNumber));
36
28
 
37
29
  return txEffect;
@@ -1,12 +0,0 @@
1
- import type { ContractArtifact } from '@aztec/aztec.js/abi';
2
- import { Fr } from '@aztec/aztec.js/fields';
3
- import { ContractStore } from '@aztec/pxe/server';
4
- export type ContractArtifactWithHash = ContractArtifact & {
5
- artifactHash: Fr;
6
- };
7
- export declare class TXEContractStore extends ContractStore {
8
- #private;
9
- addContractArtifact(id: Fr, artifact: ContractArtifact | ContractArtifactWithHash): Promise<void>;
10
- getContractArtifact(contractClassId: Fr): Promise<ContractArtifact | ContractArtifactWithHash | undefined>;
11
- }
12
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHhlX2NvbnRyYWN0X3N0b3JlLmQudHMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvdXRpbC90eGVfY29udHJhY3Rfc3RvcmUudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUEsT0FBTyxLQUFLLEVBQUUsZ0JBQWdCLEVBQUUsTUFBTSxxQkFBcUIsQ0FBQztBQUM1RCxPQUFPLEVBQUUsRUFBRSxFQUFFLE1BQU0sd0JBQXdCLENBQUM7QUFDNUMsT0FBTyxFQUFFLGFBQWEsRUFBRSxNQUFNLG1CQUFtQixDQUFDO0FBRWxELE1BQU0sTUFBTSx3QkFBd0IsR0FBRyxnQkFBZ0IsR0FBRztJQUFFLFlBQVksRUFBRSxFQUFFLENBQUE7Q0FBRSxDQUFDO0FBTy9FLHFCQUFhLGdCQUFpQixTQUFRLGFBQWE7O0lBRzNCLG1CQUFtQixDQUN2QyxFQUFFLEVBQUUsRUFBRSxFQUNOLFFBQVEsRUFBRSxnQkFBZ0IsR0FBRyx3QkFBd0IsR0FDcEQsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUtmO0lBRXFCLG1CQUFtQixDQUN2QyxlQUFlLEVBQUUsRUFBRSxHQUNsQixPQUFPLENBQUMsZ0JBQWdCLEdBQUcsd0JBQXdCLEdBQUcsU0FBUyxDQUFDLENBUWxFO0NBQ0YifQ==
@@ -1 +0,0 @@
1
- {"version":3,"file":"txe_contract_store.d.ts","sourceRoot":"","sources":["../../src/util/txe_contract_store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,EAAE,EAAE,MAAM,wBAAwB,CAAC;AAC5C,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,MAAM,MAAM,wBAAwB,GAAG,gBAAgB,GAAG;IAAE,YAAY,EAAE,EAAE,CAAA;CAAE,CAAC;AAO/E,qBAAa,gBAAiB,SAAQ,aAAa;;IAG3B,mBAAmB,CACvC,EAAE,EAAE,EAAE,EACN,QAAQ,EAAE,gBAAgB,GAAG,wBAAwB,GACpD,OAAO,CAAC,IAAI,CAAC,CAKf;IAEqB,mBAAmB,CACvC,eAAe,EAAE,EAAE,GAClB,OAAO,CAAC,gBAAgB,GAAG,wBAAwB,GAAG,SAAS,CAAC,CAQlE;CACF"}
@@ -1,22 +0,0 @@
1
- import { Fr } from '@aztec/aztec.js/fields';
2
- import { ContractStore } from '@aztec/pxe/server';
3
- /*
4
- * A contract store that stores contract artifacts with their hashes. Since
5
- * TXE typically deploys the same contract again and again for multiple tests, caching
6
- * the *very* expensive artifact hash computation improves testing speed significantly.
7
- */ export class TXEContractStore extends ContractStore {
8
- #artifactHashes = new Map();
9
- async addContractArtifact(id, artifact) {
10
- if ('artifactHash' in artifact) {
11
- this.#artifactHashes.set(id.toString(), artifact.artifactHash.toBuffer());
12
- }
13
- await super.addContractArtifact(id, artifact);
14
- }
15
- async getContractArtifact(contractClassId) {
16
- const artifact = await super.getContractArtifact(contractClassId);
17
- if (artifact && this.#artifactHashes.has(contractClassId.toString())) {
18
- artifact.artifactHash = Fr.fromBuffer(this.#artifactHashes.get(contractClassId.toString()));
19
- }
20
- return artifact;
21
- }
22
- }
@@ -1,36 +0,0 @@
1
- import type { ContractArtifact } from '@aztec/aztec.js/abi';
2
- import { Fr } from '@aztec/aztec.js/fields';
3
- import { ContractStore } from '@aztec/pxe/server';
4
-
5
- export type ContractArtifactWithHash = ContractArtifact & { artifactHash: Fr };
6
-
7
- /*
8
- * A contract store that stores contract artifacts with their hashes. Since
9
- * TXE typically deploys the same contract again and again for multiple tests, caching
10
- * the *very* expensive artifact hash computation improves testing speed significantly.
11
- */
12
- export class TXEContractStore extends ContractStore {
13
- #artifactHashes: Map<string, Buffer> = new Map();
14
-
15
- public override async addContractArtifact(
16
- id: Fr,
17
- artifact: ContractArtifact | ContractArtifactWithHash,
18
- ): Promise<void> {
19
- if ('artifactHash' in artifact) {
20
- this.#artifactHashes.set(id.toString(), artifact.artifactHash.toBuffer());
21
- }
22
- await super.addContractArtifact(id, artifact);
23
- }
24
-
25
- public override async getContractArtifact(
26
- contractClassId: Fr,
27
- ): Promise<ContractArtifact | ContractArtifactWithHash | undefined> {
28
- const artifact = await super.getContractArtifact(contractClassId);
29
- if (artifact && this.#artifactHashes.has(contractClassId.toString())) {
30
- (artifact as ContractArtifactWithHash).artifactHash = Fr.fromBuffer(
31
- this.#artifactHashes.get(contractClassId.toString())!,
32
- );
33
- }
34
- return artifact;
35
- }
36
- }