@aztec/txe 0.0.1-commit.d3ec352c → 0.0.1-commit.e310a4c8
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.
- package/dest/constants.d.ts +3 -0
- package/dest/constants.d.ts.map +1 -0
- package/dest/constants.js +2 -0
- package/dest/index.d.ts +1 -1
- package/dest/index.d.ts.map +1 -1
- package/dest/index.js +3 -2
- package/dest/oracle/interfaces.d.ts +7 -5
- package/dest/oracle/interfaces.d.ts.map +1 -1
- package/dest/oracle/txe_oracle_public_context.d.ts +5 -5
- package/dest/oracle/txe_oracle_public_context.d.ts.map +1 -1
- package/dest/oracle/txe_oracle_public_context.js +10 -12
- package/dest/oracle/txe_oracle_top_level_context.d.ts +20 -11
- package/dest/oracle/txe_oracle_top_level_context.d.ts.map +1 -1
- package/dest/oracle/txe_oracle_top_level_context.js +105 -66
- package/dest/rpc_translator.d.ts +24 -12
- package/dest/rpc_translator.d.ts.map +1 -1
- package/dest/rpc_translator.js +95 -43
- package/dest/state_machine/archiver.d.ts +21 -57
- package/dest/state_machine/archiver.d.ts.map +1 -1
- package/dest/state_machine/archiver.js +61 -107
- package/dest/state_machine/dummy_p2p_client.d.ts +8 -7
- package/dest/state_machine/dummy_p2p_client.d.ts.map +1 -1
- package/dest/state_machine/dummy_p2p_client.js +16 -11
- package/dest/state_machine/global_variable_builder.d.ts +4 -3
- package/dest/state_machine/global_variable_builder.d.ts.map +1 -1
- package/dest/state_machine/global_variable_builder.js +13 -1
- package/dest/state_machine/index.d.ts +5 -5
- package/dest/state_machine/index.d.ts.map +1 -1
- package/dest/state_machine/index.js +32 -20
- package/dest/state_machine/mock_epoch_cache.d.ts +7 -6
- package/dest/state_machine/mock_epoch_cache.d.ts.map +1 -1
- package/dest/state_machine/mock_epoch_cache.js +10 -7
- package/dest/state_machine/synchronizer.d.ts +3 -3
- package/dest/state_machine/synchronizer.d.ts.map +1 -1
- package/dest/state_machine/synchronizer.js +2 -2
- package/dest/txe_session.d.ts +18 -10
- package/dest/txe_session.d.ts.map +1 -1
- package/dest/txe_session.js +91 -45
- package/dest/util/encoding.d.ts +618 -19
- package/dest/util/encoding.d.ts.map +1 -1
- package/dest/util/encoding.js +1 -1
- package/dest/util/txe_account_store.d.ts +10 -0
- package/dest/util/txe_account_store.d.ts.map +1 -0
- package/dest/util/{txe_account_data_provider.js → txe_account_store.js} +1 -1
- package/dest/util/txe_contract_store.d.ts +12 -0
- package/dest/util/txe_contract_store.d.ts.map +1 -0
- package/dest/util/{txe_contract_data_provider.js → txe_contract_store.js} +3 -3
- package/dest/util/txe_public_contract_data_source.d.ts +5 -5
- package/dest/util/txe_public_contract_data_source.d.ts.map +1 -1
- package/dest/util/txe_public_contract_data_source.js +11 -11
- package/dest/utils/block_creation.d.ts +19 -5
- package/dest/utils/block_creation.d.ts.map +1 -1
- package/dest/utils/block_creation.js +36 -4
- package/dest/utils/tx_effect_creation.d.ts +2 -3
- package/dest/utils/tx_effect_creation.d.ts.map +1 -1
- package/dest/utils/tx_effect_creation.js +4 -7
- package/package.json +16 -16
- package/src/constants.ts +3 -0
- package/src/index.ts +15 -12
- package/src/oracle/interfaces.ts +6 -4
- package/src/oracle/txe_oracle_public_context.ts +10 -17
- package/src/oracle/txe_oracle_top_level_context.ts +168 -91
- package/src/rpc_translator.ts +122 -51
- package/src/state_machine/archiver.ts +57 -131
- package/src/state_machine/dummy_p2p_client.ts +21 -14
- package/src/state_machine/global_variable_builder.ts +19 -2
- package/src/state_machine/index.ts +41 -18
- package/src/state_machine/mock_epoch_cache.ts +10 -11
- package/src/state_machine/synchronizer.ts +3 -4
- package/src/txe_session.ts +183 -78
- package/src/util/encoding.ts +1 -1
- package/src/util/{txe_account_data_provider.ts → txe_account_store.ts} +1 -1
- package/src/util/{txe_contract_data_provider.ts → txe_contract_store.ts} +3 -3
- package/src/util/txe_public_contract_data_source.ts +10 -10
- package/src/utils/block_creation.ts +46 -15
- package/src/utils/tx_effect_creation.ts +4 -12
- package/dest/util/txe_account_data_provider.d.ts +0 -10
- package/dest/util/txe_account_data_provider.d.ts.map +0 -1
- package/dest/util/txe_contract_data_provider.d.ts +0 -12
- package/dest/util/txe_contract_data_provider.d.ts.map +0 -1
|
@@ -3,10 +3,10 @@ import type { EthAddress } from '@aztec/foundation/eth-address';
|
|
|
3
3
|
import type { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
4
4
|
import { GasFees } from '@aztec/stdlib/gas';
|
|
5
5
|
import { makeGlobalVariables } from '@aztec/stdlib/testing';
|
|
6
|
-
import { type GlobalVariableBuilder, GlobalVariables } from '@aztec/stdlib/tx';
|
|
6
|
+
import { type CheckpointGlobalVariables, type GlobalVariableBuilder, GlobalVariables } from '@aztec/stdlib/tx';
|
|
7
7
|
|
|
8
8
|
export class TXEGlobalVariablesBuilder implements GlobalVariableBuilder {
|
|
9
|
-
public
|
|
9
|
+
public getCurrentMinFees(): Promise<GasFees> {
|
|
10
10
|
return Promise.resolve(new GasFees(0, 0));
|
|
11
11
|
}
|
|
12
12
|
|
|
@@ -18,4 +18,21 @@ export class TXEGlobalVariablesBuilder implements GlobalVariableBuilder {
|
|
|
18
18
|
): Promise<GlobalVariables> {
|
|
19
19
|
return Promise.resolve(makeGlobalVariables());
|
|
20
20
|
}
|
|
21
|
+
|
|
22
|
+
public buildCheckpointGlobalVariables(
|
|
23
|
+
_coinbase: EthAddress,
|
|
24
|
+
_feeRecipient: AztecAddress,
|
|
25
|
+
_slotNumber: SlotNumber,
|
|
26
|
+
): Promise<CheckpointGlobalVariables> {
|
|
27
|
+
const vars = makeGlobalVariables();
|
|
28
|
+
return Promise.resolve({
|
|
29
|
+
chainId: vars.chainId,
|
|
30
|
+
version: vars.version,
|
|
31
|
+
slotNumber: vars.slotNumber,
|
|
32
|
+
timestamp: vars.timestamp,
|
|
33
|
+
coinbase: vars.coinbase,
|
|
34
|
+
feeRecipient: vars.feeRecipient,
|
|
35
|
+
gasFees: vars.gasFees,
|
|
36
|
+
});
|
|
37
|
+
}
|
|
21
38
|
}
|
|
@@ -1,10 +1,14 @@
|
|
|
1
1
|
import { type AztecNodeConfig, AztecNodeService } from '@aztec/aztec-node';
|
|
2
2
|
import { TestCircuitVerifier } from '@aztec/bb-prover/test';
|
|
3
|
+
import { CheckpointNumber } from '@aztec/foundation/branded-types';
|
|
4
|
+
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
3
5
|
import { createLogger } from '@aztec/foundation/log';
|
|
4
6
|
import type { AztecAsyncKVStore } from '@aztec/kv-store';
|
|
5
|
-
import {
|
|
6
|
-
import {
|
|
7
|
+
import { AnchorBlockStore } from '@aztec/pxe/server';
|
|
8
|
+
import { L2Block } from '@aztec/stdlib/block';
|
|
9
|
+
import { Checkpoint, L1PublishedData, PublishedCheckpoint } from '@aztec/stdlib/checkpoint';
|
|
7
10
|
import type { AztecNode } from '@aztec/stdlib/interfaces/client';
|
|
11
|
+
import { CheckpointHeader } from '@aztec/stdlib/rollup';
|
|
8
12
|
import { getPackageVersion } from '@aztec/stdlib/update-checker';
|
|
9
13
|
|
|
10
14
|
import { TXEArchiver } from './archiver.js';
|
|
@@ -21,13 +25,13 @@ export class TXEStateMachine {
|
|
|
21
25
|
public node: AztecNode,
|
|
22
26
|
public synchronizer: TXESynchronizer,
|
|
23
27
|
public archiver: TXEArchiver,
|
|
24
|
-
public
|
|
28
|
+
public anchorBlockStore: AnchorBlockStore,
|
|
25
29
|
) {}
|
|
26
30
|
|
|
27
31
|
public static async create(db: AztecAsyncKVStore) {
|
|
28
32
|
const archiver = new TXEArchiver(db);
|
|
29
33
|
const synchronizer = await TXESynchronizer.create();
|
|
30
|
-
const
|
|
34
|
+
const anchorBlockStore = new AnchorBlockStore(db);
|
|
31
35
|
|
|
32
36
|
const aztecNodeConfig = {} as AztecNodeConfig;
|
|
33
37
|
|
|
@@ -54,24 +58,43 @@ export class TXEStateMachine {
|
|
|
54
58
|
log,
|
|
55
59
|
);
|
|
56
60
|
|
|
57
|
-
return new this(node, synchronizer, archiver,
|
|
61
|
+
return new this(node, synchronizer, archiver, anchorBlockStore);
|
|
58
62
|
}
|
|
59
63
|
|
|
60
64
|
public async handleL2Block(block: L2Block) {
|
|
65
|
+
// Create a checkpoint from the block manually
|
|
66
|
+
const checkpoint = new Checkpoint(
|
|
67
|
+
block.archive,
|
|
68
|
+
CheckpointHeader.from({
|
|
69
|
+
lastArchiveRoot: block.header.lastArchive.root,
|
|
70
|
+
inHash: Fr.ZERO,
|
|
71
|
+
blobsHash: Fr.ZERO,
|
|
72
|
+
blockHeadersHash: Fr.ZERO,
|
|
73
|
+
epochOutHash: Fr.ZERO,
|
|
74
|
+
slotNumber: block.header.globalVariables.slotNumber,
|
|
75
|
+
timestamp: block.header.globalVariables.timestamp,
|
|
76
|
+
coinbase: block.header.globalVariables.coinbase,
|
|
77
|
+
feeRecipient: block.header.globalVariables.feeRecipient,
|
|
78
|
+
gasFees: block.header.globalVariables.gasFees,
|
|
79
|
+
totalManaUsed: block.header.totalManaUsed,
|
|
80
|
+
}),
|
|
81
|
+
[block],
|
|
82
|
+
CheckpointNumber.fromBlockNumber(block.number),
|
|
83
|
+
);
|
|
84
|
+
|
|
85
|
+
const publishedCheckpoint = new PublishedCheckpoint(
|
|
86
|
+
checkpoint,
|
|
87
|
+
new L1PublishedData(
|
|
88
|
+
BigInt(block.header.globalVariables.blockNumber),
|
|
89
|
+
block.header.globalVariables.timestamp,
|
|
90
|
+
block.header.globalVariables.blockNumber.toString(),
|
|
91
|
+
),
|
|
92
|
+
[],
|
|
93
|
+
);
|
|
61
94
|
await Promise.all([
|
|
62
|
-
this.synchronizer.handleL2Block(block
|
|
63
|
-
this.archiver.
|
|
64
|
-
|
|
65
|
-
block,
|
|
66
|
-
l1: {
|
|
67
|
-
blockHash: block.header.globalVariables.blockNumber.toString(),
|
|
68
|
-
blockNumber: BigInt(block.header.globalVariables.blockNumber),
|
|
69
|
-
timestamp: block.header.globalVariables.timestamp,
|
|
70
|
-
},
|
|
71
|
-
attestations: [],
|
|
72
|
-
}),
|
|
73
|
-
]),
|
|
74
|
-
this.syncDataProvider.setHeader(block.getBlockHeader()),
|
|
95
|
+
this.synchronizer.handleL2Block(block),
|
|
96
|
+
this.archiver.addCheckpoints([publishedCheckpoint], undefined),
|
|
97
|
+
this.anchorBlockStore.setHeader(block.header),
|
|
75
98
|
]);
|
|
76
99
|
}
|
|
77
100
|
}
|
|
@@ -12,14 +12,16 @@ export class MockEpochCache implements EpochCacheInterface {
|
|
|
12
12
|
committee: undefined,
|
|
13
13
|
seed: 0n,
|
|
14
14
|
epoch: EpochNumber.ZERO,
|
|
15
|
+
isEscapeHatchOpen: false,
|
|
15
16
|
});
|
|
16
17
|
}
|
|
17
18
|
|
|
18
|
-
getEpochAndSlotNow(): EpochAndSlot {
|
|
19
|
+
getEpochAndSlotNow(): EpochAndSlot & { nowMs: bigint } {
|
|
19
20
|
return {
|
|
20
21
|
epoch: EpochNumber.ZERO,
|
|
21
22
|
slot: SlotNumber(0),
|
|
22
23
|
ts: 0n,
|
|
24
|
+
nowMs: 0n,
|
|
23
25
|
};
|
|
24
26
|
}
|
|
25
27
|
|
|
@@ -40,18 +42,15 @@ export class MockEpochCache implements EpochCacheInterface {
|
|
|
40
42
|
return 0n;
|
|
41
43
|
}
|
|
42
44
|
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
nextProposer: EthAddress | undefined;
|
|
46
|
-
currentSlot: SlotNumber;
|
|
47
|
-
nextSlot: SlotNumber;
|
|
48
|
-
}> {
|
|
49
|
-
return Promise.resolve({
|
|
50
|
-
currentProposer: undefined,
|
|
51
|
-
nextProposer: undefined,
|
|
45
|
+
getCurrentAndNextSlot(): { currentSlot: SlotNumber; nextSlot: SlotNumber } {
|
|
46
|
+
return {
|
|
52
47
|
currentSlot: SlotNumber(0),
|
|
53
48
|
nextSlot: SlotNumber(0),
|
|
54
|
-
}
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
|
|
52
|
+
getProposerAttesterAddressInSlot(_slot: SlotNumber): Promise<EthAddress | undefined> {
|
|
53
|
+
return Promise.resolve(undefined);
|
|
55
54
|
}
|
|
56
55
|
|
|
57
56
|
isInCommittee(_slot: SlotTag, _validator: EthAddress): Promise<boolean> {
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { NUMBER_OF_L1_L2_MESSAGES_PER_ROLLUP } from '@aztec/constants';
|
|
2
2
|
import { BlockNumber } from '@aztec/foundation/branded-types';
|
|
3
|
-
import { Fr } from '@aztec/foundation/
|
|
4
|
-
import type {
|
|
3
|
+
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
4
|
+
import type { L2Block } from '@aztec/stdlib/block';
|
|
5
5
|
import type {
|
|
6
6
|
MerkleTreeReadOperations,
|
|
7
7
|
MerkleTreeWriteOperations,
|
|
@@ -23,11 +23,10 @@ export class TXESynchronizer implements WorldStateSynchronizer {
|
|
|
23
23
|
return new this(nativeWorldStateService);
|
|
24
24
|
}
|
|
25
25
|
|
|
26
|
-
public async handleL2Block(block:
|
|
26
|
+
public async handleL2Block(block: L2Block) {
|
|
27
27
|
await this.nativeWorldStateService.handleL2BlockAndMessages(
|
|
28
28
|
block,
|
|
29
29
|
Array(NUMBER_OF_L1_L2_MESSAGES_PER_ROLLUP).fill(0).map(Fr.zero),
|
|
30
|
-
true,
|
|
31
30
|
);
|
|
32
31
|
|
|
33
32
|
this.blockNumber = block.header.globalVariables.blockNumber;
|
package/src/txe_session.ts
CHANGED
|
@@ -1,16 +1,19 @@
|
|
|
1
1
|
import { BlockNumber } from '@aztec/foundation/branded-types';
|
|
2
|
-
import { Fr } from '@aztec/foundation/
|
|
2
|
+
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
6
|
import type { ProtocolContract } from '@aztec/protocol-contracts';
|
|
7
7
|
import {
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
8
|
+
AddressStore,
|
|
9
|
+
CapsuleStore,
|
|
10
|
+
JobCoordinator,
|
|
11
|
+
NoteService,
|
|
12
|
+
NoteStore,
|
|
13
|
+
PrivateEventStore,
|
|
14
|
+
RecipientTaggingStore,
|
|
15
|
+
SenderAddressBookStore,
|
|
16
|
+
SenderTaggingStore,
|
|
14
17
|
} from '@aztec/pxe/server';
|
|
15
18
|
import {
|
|
16
19
|
ExecutionNoteCache,
|
|
@@ -18,34 +21,39 @@ import {
|
|
|
18
21
|
HashedValuesCache,
|
|
19
22
|
type IPrivateExecutionOracle,
|
|
20
23
|
type IUtilityExecutionOracle,
|
|
24
|
+
Oracle,
|
|
21
25
|
PrivateExecutionOracle,
|
|
22
26
|
UtilityExecutionOracle,
|
|
23
27
|
} from '@aztec/pxe/simulator';
|
|
24
|
-
import {
|
|
28
|
+
import {
|
|
29
|
+
ExecutionError,
|
|
30
|
+
WASMSimulator,
|
|
31
|
+
createSimulationError,
|
|
32
|
+
extractCallStack,
|
|
33
|
+
resolveAssertionMessageFromError,
|
|
34
|
+
toACVMWitness,
|
|
35
|
+
} from '@aztec/simulator/client';
|
|
36
|
+
import { FunctionCall, FunctionSelector, FunctionType } from '@aztec/stdlib/abi';
|
|
25
37
|
import type { AuthWitness } from '@aztec/stdlib/auth-witness';
|
|
26
38
|
import { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
27
|
-
import { Body, L2Block } from '@aztec/stdlib/block';
|
|
28
39
|
import { GasSettings } from '@aztec/stdlib/gas';
|
|
29
40
|
import { computeProtocolNullifier } from '@aztec/stdlib/hash';
|
|
30
41
|
import { PrivateContextInputs } from '@aztec/stdlib/kernel';
|
|
31
|
-
import {
|
|
42
|
+
import { makeGlobalVariables } from '@aztec/stdlib/testing';
|
|
32
43
|
import { CallContext, GlobalVariables, TxContext } from '@aztec/stdlib/tx';
|
|
33
44
|
|
|
34
45
|
import { z } from 'zod';
|
|
35
46
|
|
|
47
|
+
import { DEFAULT_ADDRESS } from './constants.js';
|
|
36
48
|
import type { IAvmExecutionOracle, ITxeExecutionOracle } from './oracle/interfaces.js';
|
|
37
49
|
import { TXEOraclePublicContext } from './oracle/txe_oracle_public_context.js';
|
|
38
50
|
import { TXEOracleTopLevelContext } from './oracle/txe_oracle_top_level_context.js';
|
|
39
51
|
import { RPCTranslator } from './rpc_translator.js';
|
|
40
52
|
import { TXEStateMachine } from './state_machine/index.js';
|
|
41
53
|
import type { ForeignCallArgs, ForeignCallResult } from './util/encoding.js';
|
|
42
|
-
import {
|
|
43
|
-
import {
|
|
44
|
-
import {
|
|
45
|
-
getSingleTxBlockRequestHash,
|
|
46
|
-
insertTxEffectIntoWorldTrees,
|
|
47
|
-
makeTXEBlockHeader,
|
|
48
|
-
} from './utils/block_creation.js';
|
|
54
|
+
import { TXEAccountStore } from './util/txe_account_store.js';
|
|
55
|
+
import { TXEContractStore } from './util/txe_contract_store.js';
|
|
56
|
+
import { getSingleTxBlockRequestHash, insertTxEffectIntoWorldTrees, makeTXEBlock } from './utils/block_creation.js';
|
|
49
57
|
import { makeTxEffect } from './utils/tx_effect_creation.js';
|
|
50
58
|
|
|
51
59
|
/**
|
|
@@ -67,7 +75,6 @@ type SessionState =
|
|
|
67
75
|
| {
|
|
68
76
|
name: 'PRIVATE';
|
|
69
77
|
nextBlockGlobalVariables: GlobalVariables;
|
|
70
|
-
protocolNullifier: Fr;
|
|
71
78
|
noteCache: ExecutionNoteCache;
|
|
72
79
|
taggingIndexCache: ExecutionTaggingIndexCache;
|
|
73
80
|
}
|
|
@@ -106,8 +113,6 @@ export interface TXESessionStateHandler {
|
|
|
106
113
|
enterUtilityState(contractAddress?: AztecAddress): Promise<void>;
|
|
107
114
|
}
|
|
108
115
|
|
|
109
|
-
const DEFAULT_ADDRESS = AztecAddress.fromNumber(42);
|
|
110
|
-
|
|
111
116
|
/**
|
|
112
117
|
* A `TXESession` corresponds to a Noir `#[test]` function, and handles all of its oracle calls, stores test-specific
|
|
113
118
|
* state, etc., independent of all other tests running in parallel.
|
|
@@ -124,32 +129,51 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
124
129
|
| IPrivateExecutionOracle
|
|
125
130
|
| IAvmExecutionOracle
|
|
126
131
|
| ITxeExecutionOracle,
|
|
127
|
-
private
|
|
132
|
+
private contractStore: TXEContractStore,
|
|
133
|
+
private noteStore: NoteStore,
|
|
128
134
|
private keyStore: KeyStore,
|
|
129
|
-
private
|
|
130
|
-
private
|
|
135
|
+
private addressStore: AddressStore,
|
|
136
|
+
private accountStore: TXEAccountStore,
|
|
137
|
+
private senderTaggingStore: SenderTaggingStore,
|
|
138
|
+
private recipientTaggingStore: RecipientTaggingStore,
|
|
139
|
+
private senderAddressBookStore: SenderAddressBookStore,
|
|
140
|
+
private capsuleStore: CapsuleStore,
|
|
141
|
+
private privateEventStore: PrivateEventStore,
|
|
142
|
+
private jobCoordinator: JobCoordinator,
|
|
143
|
+
private currentJobId: string,
|
|
131
144
|
private chainId: Fr,
|
|
132
145
|
private version: Fr,
|
|
133
146
|
private nextBlockTimestamp: bigint,
|
|
134
|
-
private pxeOracleInterface: PXEOracleInterface,
|
|
135
147
|
) {}
|
|
136
148
|
|
|
137
149
|
static async init(protocolContracts: ProtocolContract[]) {
|
|
138
150
|
const store = await openTmpStore('txe-session');
|
|
139
151
|
|
|
140
|
-
const
|
|
141
|
-
const
|
|
142
|
-
const
|
|
143
|
-
const
|
|
144
|
-
const
|
|
145
|
-
const
|
|
152
|
+
const addressStore = new AddressStore(store);
|
|
153
|
+
const privateEventStore = new PrivateEventStore(store);
|
|
154
|
+
const contractStore = new TXEContractStore(store);
|
|
155
|
+
const noteStore = new NoteStore(store);
|
|
156
|
+
const senderTaggingStore = new SenderTaggingStore(store);
|
|
157
|
+
const recipientTaggingStore = new RecipientTaggingStore(store);
|
|
158
|
+
const senderAddressBookStore = new SenderAddressBookStore(store);
|
|
159
|
+
const capsuleStore = new CapsuleStore(store);
|
|
146
160
|
const keyStore = new KeyStore(store);
|
|
147
|
-
const
|
|
161
|
+
const accountStore = new TXEAccountStore(store);
|
|
162
|
+
|
|
163
|
+
// Create job coordinator and register staged stores
|
|
164
|
+
const jobCoordinator = new JobCoordinator(store);
|
|
165
|
+
jobCoordinator.registerStores([
|
|
166
|
+
capsuleStore,
|
|
167
|
+
senderTaggingStore,
|
|
168
|
+
recipientTaggingStore,
|
|
169
|
+
privateEventStore,
|
|
170
|
+
noteStore,
|
|
171
|
+
]);
|
|
148
172
|
|
|
149
173
|
// Register protocol contracts.
|
|
150
174
|
for (const { contractClass, instance, artifact } of protocolContracts) {
|
|
151
|
-
await
|
|
152
|
-
await
|
|
175
|
+
await contractStore.addContractArtifact(contractClass.id, artifact);
|
|
176
|
+
await contractStore.addContractInstance(instance);
|
|
153
177
|
}
|
|
154
178
|
|
|
155
179
|
const stateMachine = await TXEStateMachine.create(store);
|
|
@@ -158,25 +182,21 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
158
182
|
const version = new Fr(await stateMachine.node.getVersion());
|
|
159
183
|
const chainId = new Fr(await stateMachine.node.getChainId());
|
|
160
184
|
|
|
161
|
-
const
|
|
162
|
-
stateMachine.node,
|
|
163
|
-
keyStore,
|
|
164
|
-
contractDataProvider,
|
|
165
|
-
noteDataProvider,
|
|
166
|
-
capsuleDataProvider,
|
|
167
|
-
stateMachine.syncDataProvider,
|
|
168
|
-
taggingDataProvider,
|
|
169
|
-
addressDataProvider,
|
|
170
|
-
privateEventDataProvider,
|
|
171
|
-
);
|
|
185
|
+
const initialJobId = jobCoordinator.beginJob();
|
|
172
186
|
|
|
173
187
|
const topLevelOracleHandler = new TXEOracleTopLevelContext(
|
|
174
188
|
stateMachine,
|
|
175
|
-
|
|
189
|
+
contractStore,
|
|
190
|
+
noteStore,
|
|
176
191
|
keyStore,
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
192
|
+
addressStore,
|
|
193
|
+
accountStore,
|
|
194
|
+
senderTaggingStore,
|
|
195
|
+
recipientTaggingStore,
|
|
196
|
+
senderAddressBookStore,
|
|
197
|
+
capsuleStore,
|
|
198
|
+
privateEventStore,
|
|
199
|
+
initialJobId,
|
|
180
200
|
nextBlockTimestamp,
|
|
181
201
|
version,
|
|
182
202
|
chainId,
|
|
@@ -188,14 +208,21 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
188
208
|
createLogger('txe:session'),
|
|
189
209
|
stateMachine,
|
|
190
210
|
topLevelOracleHandler,
|
|
191
|
-
|
|
211
|
+
contractStore,
|
|
212
|
+
noteStore,
|
|
192
213
|
keyStore,
|
|
193
|
-
|
|
194
|
-
|
|
214
|
+
addressStore,
|
|
215
|
+
accountStore,
|
|
216
|
+
senderTaggingStore,
|
|
217
|
+
recipientTaggingStore,
|
|
218
|
+
senderAddressBookStore,
|
|
219
|
+
capsuleStore,
|
|
220
|
+
privateEventStore,
|
|
221
|
+
jobCoordinator,
|
|
222
|
+
initialJobId,
|
|
195
223
|
version,
|
|
196
224
|
chainId,
|
|
197
225
|
nextBlockTimestamp,
|
|
198
|
-
pxeOracleInterface,
|
|
199
226
|
);
|
|
200
227
|
}
|
|
201
228
|
|
|
@@ -252,13 +279,23 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
252
279
|
}
|
|
253
280
|
}
|
|
254
281
|
|
|
282
|
+
// Commit all staged stores from the job that was just completed, then begin a new job
|
|
283
|
+
await this.jobCoordinator.commitJob(this.currentJobId);
|
|
284
|
+
this.currentJobId = this.jobCoordinator.beginJob();
|
|
285
|
+
|
|
255
286
|
this.oracleHandler = new TXEOracleTopLevelContext(
|
|
256
287
|
this.stateMachine,
|
|
257
|
-
this.
|
|
288
|
+
this.contractStore,
|
|
289
|
+
this.noteStore,
|
|
258
290
|
this.keyStore,
|
|
259
|
-
this.
|
|
260
|
-
this.
|
|
261
|
-
this.
|
|
291
|
+
this.addressStore,
|
|
292
|
+
this.accountStore,
|
|
293
|
+
this.senderTaggingStore,
|
|
294
|
+
this.recipientTaggingStore,
|
|
295
|
+
this.senderAddressBookStore,
|
|
296
|
+
this.capsuleStore,
|
|
297
|
+
this.privateEventStore,
|
|
298
|
+
this.currentJobId,
|
|
262
299
|
this.nextBlockTimestamp,
|
|
263
300
|
this.version,
|
|
264
301
|
this.chainId,
|
|
@@ -275,17 +312,14 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
275
312
|
): Promise<PrivateContextInputs> {
|
|
276
313
|
this.exitTopLevelState();
|
|
277
314
|
|
|
278
|
-
// There is no automatic message discovery and contract-driven syncing process in inlined private or utility
|
|
279
|
-
// contexts, which means that known nullifiers are also not searched for, since it is during the tagging sync that
|
|
280
|
-
// we perform this. We therefore search for known nullifiers now, as otherwise notes that were nullified would not
|
|
281
|
-
// be removed from the database.
|
|
282
|
-
// TODO(#12553): make the synchronizer sync here instead and remove this
|
|
283
|
-
await this.pxeOracleInterface.syncNoteNullifiers(contractAddress);
|
|
284
|
-
|
|
285
315
|
// Private execution has two associated block numbers: the anchor block (i.e. the historical block that is used to
|
|
286
316
|
// build the proof), and the *next* block, i.e. the one we'll create once the execution ends, and which will contain
|
|
287
317
|
// a single transaction with the effects of what was done in the test.
|
|
288
318
|
const anchorBlock = await this.stateMachine.node.getBlockHeader(anchorBlockNumber ?? 'latest');
|
|
319
|
+
|
|
320
|
+
await new NoteService(this.noteStore, this.stateMachine.node, anchorBlock!, this.currentJobId).syncNoteNullifiers(
|
|
321
|
+
contractAddress,
|
|
322
|
+
);
|
|
289
323
|
const latestBlock = await this.stateMachine.node.getBlockHeader('latest');
|
|
290
324
|
|
|
291
325
|
const nextBlockGlobalVariables = makeGlobalVariables(undefined, {
|
|
@@ -300,17 +334,29 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
300
334
|
const noteCache = new ExecutionNoteCache(protocolNullifier);
|
|
301
335
|
const taggingIndexCache = new ExecutionTaggingIndexCache();
|
|
302
336
|
|
|
337
|
+
const utilityExecutor = this.utilityExecutorForContractSync(anchorBlock);
|
|
303
338
|
this.oracleHandler = new PrivateExecutionOracle(
|
|
304
339
|
Fr.ZERO,
|
|
305
340
|
new TxContext(this.chainId, this.version, GasSettings.empty()),
|
|
306
341
|
new CallContext(AztecAddress.ZERO, contractAddress, FunctionSelector.empty(), false),
|
|
307
342
|
anchorBlock!,
|
|
343
|
+
utilityExecutor,
|
|
308
344
|
[],
|
|
309
345
|
[],
|
|
310
346
|
new HashedValuesCache(),
|
|
311
347
|
noteCache,
|
|
312
348
|
taggingIndexCache,
|
|
313
|
-
this.
|
|
349
|
+
this.contractStore,
|
|
350
|
+
this.noteStore,
|
|
351
|
+
this.keyStore,
|
|
352
|
+
this.addressStore,
|
|
353
|
+
this.stateMachine.node,
|
|
354
|
+
this.senderTaggingStore,
|
|
355
|
+
this.recipientTaggingStore,
|
|
356
|
+
this.senderAddressBookStore,
|
|
357
|
+
this.capsuleStore,
|
|
358
|
+
this.privateEventStore,
|
|
359
|
+
this.currentJobId,
|
|
314
360
|
);
|
|
315
361
|
|
|
316
362
|
// We store the note and tagging index caches fed into the PrivateExecutionOracle (along with some other auxiliary
|
|
@@ -318,7 +364,7 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
318
364
|
// difference resides in that the simulator has all information needed in order to run the simulation, while ours
|
|
319
365
|
// will be ongoing as the different oracles will be invoked from the Noir test, until eventually the private
|
|
320
366
|
// execution finishes.
|
|
321
|
-
this.state = { name: 'PRIVATE', nextBlockGlobalVariables,
|
|
367
|
+
this.state = { name: 'PRIVATE', nextBlockGlobalVariables, noteCache, taggingIndexCache };
|
|
322
368
|
this.logger.debug(`Entered state ${this.state.name}`);
|
|
323
369
|
|
|
324
370
|
return (this.oracleHandler as PrivateExecutionOracle).getPrivateContextInputs();
|
|
@@ -351,14 +397,36 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
351
397
|
async enterUtilityState(contractAddress: AztecAddress = DEFAULT_ADDRESS) {
|
|
352
398
|
this.exitTopLevelState();
|
|
353
399
|
|
|
400
|
+
const anchorBlockHeader = await this.stateMachine.anchorBlockStore.getBlockHeader();
|
|
401
|
+
|
|
354
402
|
// There is no automatic message discovery and contract-driven syncing process in inlined private or utility
|
|
355
403
|
// contexts, which means that known nullifiers are also not searched for, since it is during the tagging sync that
|
|
356
404
|
// we perform this. We therefore search for known nullifiers now, as otherwise notes that were nullified would not
|
|
357
405
|
// be removed from the database.
|
|
358
406
|
// TODO(#12553): make the synchronizer sync here instead and remove this
|
|
359
|
-
await
|
|
360
|
-
|
|
361
|
-
|
|
407
|
+
await new NoteService(
|
|
408
|
+
this.noteStore,
|
|
409
|
+
this.stateMachine.node,
|
|
410
|
+
anchorBlockHeader,
|
|
411
|
+
this.currentJobId,
|
|
412
|
+
).syncNoteNullifiers(contractAddress);
|
|
413
|
+
|
|
414
|
+
this.oracleHandler = new UtilityExecutionOracle(
|
|
415
|
+
contractAddress,
|
|
416
|
+
[],
|
|
417
|
+
[],
|
|
418
|
+
anchorBlockHeader,
|
|
419
|
+
this.contractStore,
|
|
420
|
+
this.noteStore,
|
|
421
|
+
this.keyStore,
|
|
422
|
+
this.addressStore,
|
|
423
|
+
this.stateMachine.node,
|
|
424
|
+
this.recipientTaggingStore,
|
|
425
|
+
this.senderAddressBookStore,
|
|
426
|
+
this.capsuleStore,
|
|
427
|
+
this.privateEventStore,
|
|
428
|
+
this.currentJobId,
|
|
429
|
+
);
|
|
362
430
|
|
|
363
431
|
this.state = { name: 'UTILITY' };
|
|
364
432
|
this.logger.debug(`Entered state ${this.state.name}`);
|
|
@@ -392,21 +460,14 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
392
460
|
|
|
393
461
|
// We rely on the note cache to determine the effects of the transaction. This is incomplete as it doesn't private
|
|
394
462
|
// logs (other effects like enqueued public calls don't need to be considered since those are not allowed).
|
|
395
|
-
|
|
396
|
-
|
|
397
|
-
this.state.protocolNullifier,
|
|
398
|
-
this.state.nextBlockGlobalVariables.blockNumber,
|
|
399
|
-
);
|
|
463
|
+
|
|
464
|
+
const txEffect = await makeTxEffect(this.state.noteCache, this.state.nextBlockGlobalVariables.blockNumber);
|
|
400
465
|
|
|
401
466
|
// We build a block holding just this transaction
|
|
402
467
|
const forkedWorldTrees = await this.stateMachine.synchronizer.nativeWorldStateService.fork();
|
|
403
468
|
await insertTxEffectIntoWorldTrees(txEffect, forkedWorldTrees);
|
|
404
469
|
|
|
405
|
-
const block =
|
|
406
|
-
makeAppendOnlyTreeSnapshot(),
|
|
407
|
-
await makeTXEBlockHeader(forkedWorldTrees, this.state.nextBlockGlobalVariables),
|
|
408
|
-
new Body([txEffect]),
|
|
409
|
-
);
|
|
470
|
+
const block = await makeTXEBlock(forkedWorldTrees, this.state.nextBlockGlobalVariables, [txEffect]);
|
|
410
471
|
await this.stateMachine.handleL2Block(block);
|
|
411
472
|
|
|
412
473
|
await forkedWorldTrees.close();
|
|
@@ -431,4 +492,48 @@ export class TXESession implements TXESessionStateHandler {
|
|
|
431
492
|
throw new Error(`Expected to be in state 'UTILITY', but got '${this.state.name}' instead`);
|
|
432
493
|
}
|
|
433
494
|
}
|
|
495
|
+
|
|
496
|
+
private utilityExecutorForContractSync(anchorBlock: any) {
|
|
497
|
+
return async (call: FunctionCall) => {
|
|
498
|
+
const entryPointArtifact = await this.contractStore.getFunctionArtifactWithDebugMetadata(call.to, call.selector);
|
|
499
|
+
if (entryPointArtifact.functionType !== FunctionType.UTILITY) {
|
|
500
|
+
throw new Error(`Cannot run ${entryPointArtifact.functionType} function as utility`);
|
|
501
|
+
}
|
|
502
|
+
|
|
503
|
+
try {
|
|
504
|
+
const oracle = new UtilityExecutionOracle(
|
|
505
|
+
call.to,
|
|
506
|
+
[],
|
|
507
|
+
[],
|
|
508
|
+
anchorBlock!,
|
|
509
|
+
this.contractStore,
|
|
510
|
+
this.noteStore,
|
|
511
|
+
this.keyStore,
|
|
512
|
+
this.addressStore,
|
|
513
|
+
this.stateMachine.node,
|
|
514
|
+
this.recipientTaggingStore,
|
|
515
|
+
this.senderAddressBookStore,
|
|
516
|
+
this.capsuleStore,
|
|
517
|
+
this.privateEventStore,
|
|
518
|
+
this.currentJobId,
|
|
519
|
+
);
|
|
520
|
+
await new WASMSimulator()
|
|
521
|
+
.executeUserCircuit(toACVMWitness(0, call.args), entryPointArtifact, new Oracle(oracle).toACIRCallback())
|
|
522
|
+
.catch((err: Error) => {
|
|
523
|
+
err.message = resolveAssertionMessageFromError(err, entryPointArtifact);
|
|
524
|
+
throw new ExecutionError(
|
|
525
|
+
err.message,
|
|
526
|
+
{
|
|
527
|
+
contractAddress: call.to,
|
|
528
|
+
functionSelector: call.selector,
|
|
529
|
+
},
|
|
530
|
+
extractCallStack(err, entryPointArtifact.debug),
|
|
531
|
+
{ cause: err },
|
|
532
|
+
);
|
|
533
|
+
});
|
|
534
|
+
} catch (err) {
|
|
535
|
+
throw createSimulationError(err instanceof Error ? err : new Error('Unknown error contract data sync'));
|
|
536
|
+
}
|
|
537
|
+
};
|
|
538
|
+
}
|
|
434
539
|
}
|
package/src/util/encoding.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
+
import { Fr } from '@aztec/foundation/curves/bn254';
|
|
1
2
|
import type { EthAddress } from '@aztec/foundation/eth-address';
|
|
2
|
-
import { Fr } from '@aztec/foundation/fields';
|
|
3
3
|
import { hexToBuffer } from '@aztec/foundation/string';
|
|
4
4
|
import { type ContractArtifact, ContractArtifactSchema } from '@aztec/stdlib/abi';
|
|
5
5
|
import { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
@@ -2,7 +2,7 @@ import type { AztecAsyncKVStore, AztecAsyncMap } from '@aztec/kv-store';
|
|
|
2
2
|
import type { AztecAddress } from '@aztec/stdlib/aztec-address';
|
|
3
3
|
import { CompleteAddress } from '@aztec/stdlib/contract';
|
|
4
4
|
|
|
5
|
-
export class
|
|
5
|
+
export class TXEAccountStore {
|
|
6
6
|
#accounts: AztecAsyncMap<string, Buffer>;
|
|
7
7
|
|
|
8
8
|
constructor(store: AztecAsyncKVStore) {
|
|
@@ -1,15 +1,15 @@
|
|
|
1
1
|
import type { ContractArtifact } from '@aztec/aztec.js/abi';
|
|
2
2
|
import { Fr } from '@aztec/aztec.js/fields';
|
|
3
|
-
import {
|
|
3
|
+
import { ContractStore } from '@aztec/pxe/server';
|
|
4
4
|
|
|
5
5
|
export type ContractArtifactWithHash = ContractArtifact & { artifactHash: Fr };
|
|
6
6
|
|
|
7
7
|
/*
|
|
8
|
-
* A contract
|
|
8
|
+
* A contract store that stores contract artifacts with their hashes. Since
|
|
9
9
|
* TXE typically deploys the same contract again and again for multiple tests, caching
|
|
10
10
|
* the *very* expensive artifact hash computation improves testing speed significantly.
|
|
11
11
|
*/
|
|
12
|
-
export class
|
|
12
|
+
export class TXEContractStore extends ContractStore {
|
|
13
13
|
#artifactHashes: Map<string, Buffer> = new Map();
|
|
14
14
|
|
|
15
15
|
public override async addContractArtifact(
|