@aztec/txe 5.0.0-private.20260318 → 5.0.0-rc.1

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 (154) hide show
  1. package/dest/AuthRegistry-CPGFQR26.js +3 -0
  2. package/dest/AuthRegistry-CPGFQR26.js.map +7 -0
  3. package/dest/ContractClassRegistry-EHVIHGEK.js +3 -0
  4. package/dest/ContractClassRegistry-EHVIHGEK.js.map +7 -0
  5. package/dest/ContractInstanceRegistry-DWZDXHRG.js +3 -0
  6. package/dest/ContractInstanceRegistry-DWZDXHRG.js.map +7 -0
  7. package/dest/FeeJuice-MI32ZO7B.js +3 -0
  8. package/dest/FeeJuice-MI32ZO7B.js.map +7 -0
  9. package/dest/HandshakeRegistry-3KSP3ITH.js +3 -0
  10. package/dest/HandshakeRegistry-3KSP3ITH.js.map +7 -0
  11. package/dest/MultiCallEntrypoint-IU7HYFYE.js +3 -0
  12. package/dest/MultiCallEntrypoint-IU7HYFYE.js.map +7 -0
  13. package/dest/SchnorrAccount-6TUE7JX4.js +3 -0
  14. package/dest/SchnorrAccount-6TUE7JX4.js.map +7 -0
  15. package/dest/SchnorrInitializerlessAccount-S3DU2DJK.js +3 -0
  16. package/dest/SchnorrInitializerlessAccount-S3DU2DJK.js.map +7 -0
  17. package/dest/bin/check_txe_oracle_version.d.ts +2 -0
  18. package/dest/bin/check_txe_oracle_version.d.ts.map +1 -0
  19. package/dest/bin/check_txe_oracle_version.js +61 -0
  20. package/dest/bin/index.js +3 -30
  21. package/dest/bin/index.js.map +7 -0
  22. package/dest/bin/oracle_test_server.d.ts +3 -0
  23. package/dest/bin/oracle_test_server.d.ts.map +1 -0
  24. package/dest/bin/oracle_test_server.js +41 -0
  25. package/dest/chunk-5U25VAFR.js +265 -0
  26. package/dest/chunk-5U25VAFR.js.map +7 -0
  27. package/dest/chunk-BJVAAXNA.js +3 -0
  28. package/dest/chunk-BJVAAXNA.js.map +7 -0
  29. package/dest/chunk-UPW55EJX.js +304 -0
  30. package/dest/chunk-UPW55EJX.js.map +7 -0
  31. package/dest/constants.d.ts +5 -1
  32. package/dest/constants.d.ts.map +1 -1
  33. package/dest/constants.js +8 -0
  34. package/dest/dispatcher_pool.d.ts +67 -0
  35. package/dest/dispatcher_pool.d.ts.map +1 -0
  36. package/dest/dispatcher_pool.js +286 -0
  37. package/dest/index.d.ts +51 -7
  38. package/dest/index.d.ts.map +1 -1
  39. package/dest/index.js +70 -190
  40. package/dest/metafile.json +38829 -0
  41. package/dest/msgpackr_fr_extension.d.ts +2 -0
  42. package/dest/msgpackr_fr_extension.d.ts.map +1 -0
  43. package/dest/msgpackr_fr_extension.js +21 -0
  44. package/dest/oracle/interfaces.d.ts +33 -8
  45. package/dest/oracle/interfaces.d.ts.map +1 -1
  46. package/dest/oracle/test-resolver/fixtures.d.ts +43 -0
  47. package/dest/oracle/test-resolver/fixtures.d.ts.map +1 -0
  48. package/dest/oracle/test-resolver/fixtures.js +39 -0
  49. package/dest/oracle/test-resolver/index.d.ts +9 -0
  50. package/dest/oracle/test-resolver/index.d.ts.map +1 -0
  51. package/dest/oracle/test-resolver/index.js +33 -0
  52. package/dest/oracle/test-resolver/resolver.d.ts +34 -0
  53. package/dest/oracle/test-resolver/resolver.d.ts.map +1 -0
  54. package/dest/oracle/test-resolver/resolver.js +114 -0
  55. package/dest/oracle/txe_oracle_public_context.d.ts +26 -2
  56. package/dest/oracle/txe_oracle_public_context.d.ts.map +1 -1
  57. package/dest/oracle/txe_oracle_public_context.js +43 -1
  58. package/dest/oracle/txe_oracle_registry.d.ts +14 -0
  59. package/dest/oracle/txe_oracle_registry.d.ts.map +1 -0
  60. package/dest/oracle/txe_oracle_registry.js +562 -0
  61. package/dest/oracle/txe_oracle_top_level_context.d.ts +32 -18
  62. package/dest/oracle/txe_oracle_top_level_context.d.ts.map +1 -1
  63. package/dest/oracle/txe_oracle_top_level_context.js +151 -55
  64. package/dest/oracle/txe_oracle_version.d.ts +17 -0
  65. package/dest/oracle/txe_oracle_version.d.ts.map +1 -0
  66. package/dest/oracle/txe_oracle_version.js +14 -0
  67. package/dest/oracle/txe_private_execution_oracle.d.ts +17 -0
  68. package/dest/oracle/txe_private_execution_oracle.d.ts.map +1 -0
  69. package/dest/oracle/txe_private_execution_oracle.js +15 -0
  70. package/dest/rpc_server.d.ts +14 -0
  71. package/dest/rpc_server.d.ts.map +1 -0
  72. package/dest/rpc_server.js +78 -0
  73. package/dest/rpc_translator.d.ts +103 -230
  74. package/dest/rpc_translator.d.ts.map +1 -1
  75. package/dest/rpc_translator.js +697 -616
  76. package/dest/server.bundle.js +3 -0
  77. package/dest/server.bundle.js.map +7 -0
  78. package/dest/state_machine/archiver.d.ts +4 -3
  79. package/dest/state_machine/archiver.d.ts.map +1 -1
  80. package/dest/state_machine/archiver.js +26 -15
  81. package/dest/state_machine/dummy_p2p_client.d.ts +14 -7
  82. package/dest/state_machine/dummy_p2p_client.d.ts.map +1 -1
  83. package/dest/state_machine/dummy_p2p_client.js +19 -4
  84. package/dest/state_machine/global_variable_builder.d.ts +9 -4
  85. package/dest/state_machine/global_variable_builder.d.ts.map +1 -1
  86. package/dest/state_machine/global_variable_builder.js +9 -3
  87. package/dest/state_machine/index.d.ts +4 -2
  88. package/dest/state_machine/index.d.ts.map +1 -1
  89. package/dest/state_machine/index.js +11 -3
  90. package/dest/state_machine/mock_epoch_cache.d.ts +16 -3
  91. package/dest/state_machine/mock_epoch_cache.d.ts.map +1 -1
  92. package/dest/state_machine/mock_epoch_cache.js +29 -2
  93. package/dest/state_machine/synchronizer.js +1 -1
  94. package/dest/txe_session.d.ts +85 -17
  95. package/dest/txe_session.d.ts.map +1 -1
  96. package/dest/txe_session.js +245 -40
  97. package/dest/utils/encoding.d.ts +191 -0
  98. package/dest/utils/encoding.d.ts.map +1 -0
  99. package/dest/{util → utils}/encoding.js +7 -2
  100. package/dest/{util → utils}/expected_failure_error.d.ts +1 -1
  101. package/dest/utils/expected_failure_error.d.ts.map +1 -0
  102. package/dest/{util → utils}/txe_account_store.d.ts +1 -1
  103. package/dest/utils/txe_account_store.d.ts.map +1 -0
  104. package/dest/utils/txe_artifact_resolver.d.ts +37 -0
  105. package/dest/utils/txe_artifact_resolver.d.ts.map +1 -0
  106. package/dest/utils/txe_artifact_resolver.js +161 -0
  107. package/dest/utils/txe_public_contract_data_source.d.ts +20 -0
  108. package/dest/utils/txe_public_contract_data_source.d.ts.map +1 -0
  109. package/dest/{util → utils}/txe_public_contract_data_source.js +1 -3
  110. package/dest/worker.bundle.js +3 -0
  111. package/dest/worker.bundle.js.map +7 -0
  112. package/dest/worker.d.ts +2 -0
  113. package/dest/worker.d.ts.map +1 -0
  114. package/dest/worker.js +92 -0
  115. package/package.json +38 -21
  116. package/src/bin/check_txe_oracle_version.ts +70 -0
  117. package/src/bin/index.ts +11 -2
  118. package/src/bin/oracle_test_server.ts +51 -0
  119. package/src/constants.ts +10 -0
  120. package/src/dispatcher_pool.ts +317 -0
  121. package/src/index.ts +97 -227
  122. package/src/msgpackr_fr_extension.ts +23 -0
  123. package/src/oracle/interfaces.ts +29 -7
  124. package/src/oracle/test-resolver/fixtures.ts +84 -0
  125. package/src/oracle/test-resolver/index.ts +45 -0
  126. package/src/oracle/test-resolver/resolver.ts +165 -0
  127. package/src/oracle/txe_oracle_public_context.ts +60 -0
  128. package/src/oracle/txe_oracle_registry.ts +401 -0
  129. package/src/oracle/txe_oracle_top_level_context.ts +185 -64
  130. package/src/oracle/txe_oracle_version.ts +17 -0
  131. package/src/oracle/txe_private_execution_oracle.ts +30 -0
  132. package/src/rpc_server.ts +87 -0
  133. package/src/rpc_translator.ts +767 -892
  134. package/src/state_machine/archiver.ts +38 -16
  135. package/src/state_machine/dummy_p2p_client.ts +35 -11
  136. package/src/state_machine/global_variable_builder.ts +18 -3
  137. package/src/state_machine/index.ts +17 -5
  138. package/src/state_machine/mock_epoch_cache.ts +38 -3
  139. package/src/state_machine/synchronizer.ts +1 -1
  140. package/src/txe_session.ts +437 -50
  141. package/src/{util → utils}/encoding.ts +8 -2
  142. package/src/utils/txe_artifact_resolver.ts +217 -0
  143. package/src/{util → utils}/txe_public_contract_data_source.ts +0 -2
  144. package/src/worker.ts +98 -0
  145. package/dest/util/encoding.d.ts +0 -720
  146. package/dest/util/encoding.d.ts.map +0 -1
  147. package/dest/util/expected_failure_error.d.ts.map +0 -1
  148. package/dest/util/txe_account_store.d.ts.map +0 -1
  149. package/dest/util/txe_public_contract_data_source.d.ts +0 -20
  150. package/dest/util/txe_public_contract_data_source.d.ts.map +0 -1
  151. /package/dest/{util → utils}/expected_failure_error.js +0 -0
  152. /package/dest/{util → utils}/txe_account_store.js +0 -0
  153. /package/src/{util → utils}/expected_failure_error.ts +0 -0
  154. /package/src/{util → utils}/txe_account_store.ts +0 -0
package/src/index.ts CHANGED
@@ -1,62 +1,33 @@
1
- import { SchnorrAccountContractArtifact } from '@aztec/accounts/schnorr';
2
- import { type NoirCompiledContract, loadContractArtifact } from '@aztec/aztec.js/abi';
3
- import { AztecAddress } from '@aztec/aztec.js/addresses';
4
- import {
5
- type ContractInstanceWithAddress,
6
- getContractInstanceFromInstantiationParams,
7
- } from '@aztec/aztec.js/contracts';
8
1
  import { Fr } from '@aztec/aztec.js/fields';
9
- import { PublicKeys, deriveKeys } from '@aztec/aztec.js/keys';
10
- import { createSafeJsonRpcServer } from '@aztec/foundation/json-rpc/server';
11
2
  import type { Logger } from '@aztec/foundation/log';
12
- import { openTmpStore } from '@aztec/kv-store/lmdb-v2';
13
- import { protocolContractNames } from '@aztec/protocol-contracts';
14
- import { BundledProtocolContractsProvider } from '@aztec/protocol-contracts/providers/bundle';
15
- import { ContractStore } from '@aztec/pxe/server';
16
- import { computeArtifactHash } from '@aztec/stdlib/contract';
17
- import type { ContractArtifactWithHash } from '@aztec/stdlib/contract';
3
+ import { cloneEphemeralStoreFrom } from '@aztec/kv-store/lmdb-v2';
4
+ import type { ProtocolContractName } from '@aztec/protocol-contracts';
5
+ import { ContractStore } from '@aztec/pxe/client/lazy';
18
6
  import type { ApiSchemaFor } from '@aztec/stdlib/schemas';
19
7
  import { zodFor } from '@aztec/stdlib/schemas';
20
8
 
21
- import { createHash } from 'crypto';
22
- import { createReadStream } from 'fs';
23
- import { readFile, readdir } from 'fs/promises';
24
- import { join, parse } from 'path';
9
+ import { join } from 'path';
25
10
  import { z } from 'zod';
26
11
 
12
+ // Side-effect import: registers the msgpackr Fr extension for the bundled `Fr` class. Must
13
+ // be loaded before any `sendMessage` call. See msgpackr_fr_extension.ts for the why.
14
+ import './msgpackr_fr_extension.js';
27
15
  import { type TXEOracleFunctionName, TXESession } from './txe_session.js';
28
16
  import {
29
17
  type ForeignCallArgs,
30
18
  ForeignCallArgsSchema,
31
- type ForeignCallArray,
32
19
  type ForeignCallResult,
33
20
  ForeignCallResultSchema,
34
- type ForeignCallSingle,
35
- addressFromSingle,
36
- fromArray,
37
- fromSingle,
38
- toSingle,
39
- } from './util/encoding.js';
21
+ } from './utils/encoding.js';
22
+ import { TXEArtifactResolver } from './utils/txe_artifact_resolver.js';
40
23
 
41
- const sessions = new Map<number, TXESession>();
24
+ // Protocol contracts TXE registers in its contract store. Only AuthRegistry is needed for the
25
+ // current test suites; add a contract here if a lookup against a `0x000…00X` address fails.
26
+ export const TXE_REQUIRED_PROTOCOL_CONTRACTS: ProtocolContractName[] = [];
42
27
 
43
- /*
44
- * TXE typically has to load the same contract artifacts over and over again for multiple tests,
45
- * so we cache them here to avoid loading from disk repeatedly.
46
- *
47
- * The in-flight map coalesces concurrent requests for the same cache key so that
48
- * computeArtifactHash (very expensive) is only run once even under parallelism.
49
- */
50
- const TXEArtifactsCache = new Map<
51
- string,
52
- { artifact: ContractArtifactWithHash; instance: ContractInstanceWithAddress }
53
- >();
54
- const TXEArtifactsCacheInFlight = new Map<
55
- string,
56
- Promise<{ artifact: ContractArtifactWithHash; instance: ContractInstanceWithAddress }>
57
- >();
28
+ const sessions = new Map<number, TXESession>();
58
29
 
59
- type TXEForeignCallInput = {
30
+ export type TXEForeignCallInput = {
60
31
  session_id: number;
61
32
  function: TXEOracleFunctionName;
62
33
  root_path: string;
@@ -64,10 +35,13 @@ type TXEForeignCallInput = {
64
35
  inputs: ForeignCallArgs;
65
36
  };
66
37
 
67
- const TXEForeignCallInputSchema = zodFor<TXEForeignCallInput>()(
38
+ export const TXEForeignCallInputSchema: z.ZodType<TXEForeignCallInput> = zodFor<TXEForeignCallInput>()(
68
39
  z.object({
40
+ // Nargo generates session_id as a u64, which may exceed Number.MAX_SAFE_INTEGER.
41
+ // Zod 4's `.int()` enforces the safe-integer bound, so we drop it here and only require
42
+ // the value to be a non-negative number (it is used solely as a Map key).
69
43
  // eslint-disable-next-line camelcase
70
- session_id: z.number().int().nonnegative(),
44
+ session_id: z.number().nonnegative(),
71
45
  function: z.string() as z.ZodType<TXEOracleFunctionName>,
72
46
  // eslint-disable-next-line camelcase
73
47
  root_path: z.string(),
@@ -77,205 +51,101 @@ const TXEForeignCallInputSchema = zodFor<TXEForeignCallInput>()(
77
51
  }),
78
52
  );
79
53
 
80
- class TXEDispatcher {
81
- private contractStore!: ContractStore;
82
-
83
- constructor(private logger: Logger) {}
84
-
85
- private fastHashFile(path: string) {
86
- return new Promise(resolve => {
87
- const fd = createReadStream(path);
88
- const hash = createHash('sha1');
89
- hash.setEncoding('hex');
90
-
91
- fd.on('end', function () {
92
- hash.end();
93
- resolve(hash.read());
94
- });
95
-
96
- fd.pipe(hash);
97
- });
98
- }
99
-
100
- async #processDeployInputs({ inputs, root_path: rootPath, package_name: packageName }: TXEForeignCallInput) {
101
- const [contractPath, initializer] = inputs.slice(0, 2).map(input =>
102
- fromArray(input as ForeignCallArray)
103
- .map(char => String.fromCharCode(char.toNumber()))
104
- .join(''),
105
- );
106
-
107
- const decodedArgs = fromArray(inputs[3] as ForeignCallArray);
108
- const secret = fromSingle(inputs[4] as ForeignCallSingle);
109
- const salt = fromSingle(inputs[5] as ForeignCallSingle);
110
- const deployer = addressFromSingle(inputs[6] as ForeignCallSingle);
111
- const publicKeys = secret.equals(Fr.ZERO) ? PublicKeys.default() : (await deriveKeys(secret)).publicKeys;
112
- const publicKeysHash = await publicKeys.hash();
113
-
114
- let artifactPath = '';
115
- const { dir: contractDirectory, base: contractFilename } = parse(contractPath);
116
- if (contractDirectory) {
117
- if (contractDirectory.includes('@')) {
118
- // We're deploying a contract that belongs in a workspace
119
- // env.deploy("../path/to/workspace/root@packageName/contractName")
120
- const [workspace, pkg] = contractDirectory.split('@');
121
- const targetPath = join(rootPath, workspace, '/target');
122
- this.logger.debug(`Looking for compiled artifact in workspace ${targetPath}`);
123
- artifactPath = join(targetPath, `${pkg}-${contractFilename}.json`);
124
- } else {
125
- // We're deploying a standalone external contract
126
- // env.deploy("../path/to/contract/root/contractName")
127
- const targetPath = join(rootPath, contractDirectory, '/target');
128
- this.logger.debug(`Looking for compiled artifact in ${targetPath}`);
129
- [artifactPath] = (await readdir(targetPath)).filter(file => file.endsWith(`-${contractFilename}.json`));
130
- }
131
- } else {
132
- // We're deploying a local contract
133
- // env.deploy("contractName")
134
- artifactPath = join(rootPath, './target', `${packageName}-${contractFilename}.json`);
135
- }
136
-
137
- const fileHash = await this.fastHashFile(artifactPath);
138
-
139
- const cacheKey = `${contractDirectory ?? ''}-${contractFilename}-${initializer}-${decodedArgs
140
- .map(arg => arg.toString())
141
- .join('-')}-${publicKeysHash}-${salt}-${deployer}-${fileHash}`;
142
-
143
- let instance;
144
- let artifact: ContractArtifactWithHash;
145
-
146
- if (TXEArtifactsCache.has(cacheKey)) {
147
- this.logger.debug(`Using cached artifact for ${cacheKey}`);
148
- ({ artifact, instance } = TXEArtifactsCache.get(cacheKey)!);
149
- } else {
150
- if (!TXEArtifactsCacheInFlight.has(cacheKey)) {
151
- this.logger.debug(`Loading compiled artifact ${artifactPath}`);
152
- const compute = async () => {
153
- const artifactJSON = JSON.parse(await readFile(artifactPath, 'utf-8')) as NoirCompiledContract;
154
- const artifactWithoutHash = loadContractArtifact(artifactJSON);
155
- const computedArtifact: ContractArtifactWithHash = {
156
- ...artifactWithoutHash,
157
- // Artifact hash is *very* expensive to compute, so we do it here once
158
- // and the TXE contract data provider can cache it
159
- artifactHash: await computeArtifactHash(artifactWithoutHash),
160
- };
161
- this.logger.debug(
162
- `Deploy ${computedArtifact.name} with initializer ${initializer}(${decodedArgs}) and public keys hash ${publicKeysHash.toString()}`,
163
- );
164
- const computedInstance = await getContractInstanceFromInstantiationParams(computedArtifact, {
165
- constructorArgs: decodedArgs,
166
- skipArgsDecoding: true,
167
- salt,
168
- publicKeys,
169
- constructorArtifact: initializer ? initializer : undefined,
170
- deployer,
171
- });
172
- const result = { artifact: computedArtifact, instance: computedInstance };
173
- TXEArtifactsCache.set(cacheKey, result);
174
- TXEArtifactsCacheInFlight.delete(cacheKey);
175
- return result;
176
- };
177
- TXEArtifactsCacheInFlight.set(cacheKey, compute());
178
- }
179
- ({ artifact, instance } = await TXEArtifactsCacheInFlight.get(cacheKey)!);
180
- }
54
+ export interface TXEDispatcherOptions {
55
+ /**
56
+ * Path to an LMDB directory holding the required protocol contracts (see
57
+ * {@link TXE_REQUIRED_PROTOCOL_CONTRACTS}) and the SchnorrAccount artifact. When set, the
58
+ * dispatcher clones this directory into a fresh tmpdir on first use instead of registering
59
+ * the contracts itself.
60
+ */
61
+ contractStoreSourceDir: string;
62
+ /**
63
+ * Class id (hex) of the SchnorrAccount artifact pre-registered in the shared LMDB. The
64
+ * {@link TXEArtifactResolver} looks the artifact up from the cloned store via this class id
65
+ * instead of recomputing it via `getSchnorrAccountContractArtifact()` + `computeArtifactHash()`.
66
+ */
67
+ schnorrClassId: string;
68
+ }
181
69
 
182
- inputs.splice(0, 1, artifact, instance, toSingle(secret));
70
+ export class TXEDispatcher {
71
+ private contractStore!: ContractStore;
72
+ private artifactResolver!: TXEArtifactResolver;
73
+ private readonly contractStoreSourceDir: string;
74
+ private readonly schnorrClassId: Fr;
75
+
76
+ constructor(
77
+ private logger: Logger,
78
+ opts: TXEDispatcherOptions,
79
+ ) {
80
+ this.contractStoreSourceDir = opts.contractStoreSourceDir;
81
+ this.schnorrClassId = Fr.fromString(opts.schnorrClassId);
183
82
  }
184
83
 
185
- async #processAddAccountInputs({ inputs }: TXEForeignCallInput) {
186
- const secret = fromSingle(inputs[0] as ForeignCallSingle);
187
-
188
- const cacheKey = `SchnorrAccountContract-${secret}`;
189
-
190
- let artifact: ContractArtifactWithHash;
191
- let instance;
192
-
193
- if (TXEArtifactsCache.has(cacheKey)) {
194
- this.logger.debug(`Using cached artifact for ${cacheKey}`);
195
- ({ artifact, instance } = TXEArtifactsCache.get(cacheKey)!);
196
- } else {
197
- if (!TXEArtifactsCacheInFlight.has(cacheKey)) {
198
- const compute = async () => {
199
- const keys = await deriveKeys(secret);
200
- const args = [
201
- keys.publicKeys.masterIncomingViewingPublicKey.x,
202
- keys.publicKeys.masterIncomingViewingPublicKey.y,
203
- ];
204
- const computedArtifact: ContractArtifactWithHash = {
205
- ...SchnorrAccountContractArtifact,
206
- // Artifact hash is *very* expensive to compute, so we do it here once
207
- // and the TXE contract data provider can cache it
208
- artifactHash: await computeArtifactHash(SchnorrAccountContractArtifact),
209
- };
210
- const computedInstance = await getContractInstanceFromInstantiationParams(computedArtifact, {
211
- constructorArgs: args,
212
- skipArgsDecoding: true,
213
- salt: Fr.ONE,
214
- publicKeys: keys.publicKeys,
215
- constructorArtifact: 'constructor',
216
- deployer: AztecAddress.ZERO,
217
- });
218
- const result = { artifact: computedArtifact, instance: computedInstance };
219
- TXEArtifactsCache.set(cacheKey, result);
220
- TXEArtifactsCacheInFlight.delete(cacheKey);
221
- return result;
222
- };
223
- TXEArtifactsCacheInFlight.set(cacheKey, compute());
224
- }
225
- ({ artifact, instance } = await TXEArtifactsCacheInFlight.get(cacheKey)!);
84
+ /**
85
+ * Clones the pre-populated LMDB at `contractStoreSourceDir` into a fresh per-instance tmpdir
86
+ * on first use, so this dispatcher has a writable store already containing the required
87
+ * protocol contracts + SchnorrAccount. Idempotent — subsequent calls are no-ops.
88
+ */
89
+ private async warmUp(): Promise<void> {
90
+ if (this.contractStore) {
91
+ return;
226
92
  }
227
-
228
- inputs.splice(0, 0, artifact, instance);
93
+ const t0 = Date.now();
94
+ const kvStore = await cloneEphemeralStoreFrom(
95
+ join(this.contractStoreSourceDir, 'data.mdb'),
96
+ 'txe-contracts',
97
+ undefined,
98
+ 2,
99
+ );
100
+ this.contractStore = new ContractStore(kvStore);
101
+ this.artifactResolver = new TXEArtifactResolver(this.contractStore, this.schnorrClassId);
102
+ this.logger.debug('Cloned shared protocol-contracts store', { totalMs: Date.now() - t0 });
229
103
  }
230
104
 
231
105
  // eslint-disable-next-line camelcase
232
106
  async resolve_foreign_call(callData: TXEForeignCallInput): Promise<ForeignCallResult> {
233
- const { session_id: sessionId, function: functionName, inputs } = callData;
107
+ const {
108
+ session_id: sessionId,
109
+ function: functionName,
110
+ inputs,
111
+ root_path: rootPath,
112
+ package_name: packageName,
113
+ } = callData;
234
114
  this.logger.debug(`Calling ${functionName} on session ${sessionId}`);
235
115
 
236
116
  if (!sessions.has(sessionId)) {
237
117
  this.logger.debug(`Creating new session ${sessionId}`);
238
- if (!this.contractStore) {
239
- const kvStore = await openTmpStore('txe-contracts');
240
- this.contractStore = new ContractStore(kvStore);
241
- const provider = new BundledProtocolContractsProvider();
242
- for (const name of protocolContractNames) {
243
- const { instance, artifact } = await provider.getProtocolContractArtifact(name);
244
- await this.contractStore.addContractArtifact(artifact);
245
- await this.contractStore.addContractInstance(instance);
246
- }
247
- this.logger.debug('Registered protocol contracts in shared contract store');
248
- }
249
- sessions.set(sessionId, await TXESession.init(this.contractStore));
250
- }
251
-
252
- switch (functionName) {
253
- case 'aztec_txe_deploy': {
254
- await this.#processDeployInputs(callData);
255
- break;
256
- }
257
- case 'aztec_txe_addAccount': {
258
- await this.#processAddAccountInputs(callData);
259
- break;
260
- }
118
+ await this.warmUp();
119
+ sessions.set(sessionId, await TXESession.init(this.contractStore, this.artifactResolver, rootPath, packageName));
261
120
  }
262
121
 
263
122
  return await sessions.get(sessionId)!.processFunction(functionName, inputs);
264
123
  }
124
+
125
+ /**
126
+ * Releases a session and its resources (per-session LMDB + `NativeWorldStateService`).
127
+ * Called by the dispatcher pool when nargo closes its TCP connection for a test (see
128
+ * `rpc_server.ts`'s socket tracker). No-op if the session was never created — that happens
129
+ * when nargo opens a connection but errors before sending a request.
130
+ */
131
+ async disposeSession(sessionId: number): Promise<void> {
132
+ const session = sessions.get(sessionId);
133
+ if (!session) {
134
+ return;
135
+ }
136
+ sessions.delete(sessionId);
137
+ await session.dispose();
138
+ }
265
139
  }
266
140
 
267
- const TXEDispatcherApiSchema: ApiSchemaFor<TXEDispatcher> = {
141
+ /** Diagnostic-only: number of sessions currently held by this worker. */
142
+ export function activeSessionCount(): number {
143
+ return sessions.size;
144
+ }
145
+
146
+ export const TXEDispatcherApiSchema: ApiSchemaFor<TXEDispatcher> = {
268
147
  // eslint-disable-next-line camelcase
269
- resolve_foreign_call: z.function().args(TXEForeignCallInputSchema).returns(ForeignCallResultSchema),
148
+ resolve_foreign_call: z.function({ input: z.tuple([TXEForeignCallInputSchema]), output: ForeignCallResultSchema }),
149
+ // disposeSession is invoked over IPC from the worker, not via RPC; required by ApiSchemaFor.
150
+ disposeSession: z.function({ input: z.tuple([z.number().nonnegative()]), output: z.void() }),
270
151
  };
271
-
272
- /**
273
- * Creates an RPC server that forwards calls to the TXE.
274
- * @param logger - Logger to output to
275
- * @returns A TXE RPC server.
276
- */
277
- export function createTXERpcServer(logger: Logger) {
278
- return createSafeJsonRpcServer(new TXEDispatcher(logger), TXEDispatcherApiSchema, {
279
- http200OnError: true,
280
- });
281
- }
@@ -0,0 +1,23 @@
1
+ import { Fr } from '@aztec/foundation/curves/bn254';
2
+
3
+ import { addExtension } from 'msgpackr';
4
+
5
+ // `@aztec/native/msgpack_channel` registers a msgpackr extension that packs `Fr` instances to
6
+ // their raw 32-byte buffer for the C++ world-state side to deserialize. Because `@aztec/native`
7
+ // is externalized in TXE's esbuild config (it loads a `.node` binary that can't be bundled),
8
+ // the `Fr` class the extension is keyed on belongs to the *external* `@aztec/foundation`
9
+ // module instance — a different class identity than the `Fr` bundled into the TXE worker /
10
+ // bin. msgpackr's extension match uses `instanceof`, so bundled `Fr` instances slip through
11
+ // and msgpackr falls back to `Fr.toJSON()` (returns `Fr.toString()` — a `0x...` hex string).
12
+ // The C++ side then sees a string where a 32-byte binary was expected and throws
13
+ // `msgpack::type_error` whose `what()` is literally `"std::bad_cast"`.
14
+ //
15
+ // Fix: register the same extension under the BUNDLED `Fr` class identity. msgpackr's extension
16
+ // table holds an entry per `Class`, so both Fr classes now resolve to the same `write` callback
17
+ // and produce the same wire format. Imported as a side-effect from `./index.js` so both the
18
+ // worker (via `./worker.ts`) and the main-thread RPC server (via `./rpc_server.ts`) get the
19
+ // registration before any `sendMessage` runs.
20
+ addExtension({
21
+ Class: Fr,
22
+ write: (fr: Fr) => fr.toBuffer(),
23
+ });
@@ -1,11 +1,11 @@
1
- import type { ContractArtifact } from '@aztec/aztec.js/abi';
2
1
  import { CompleteAddress } from '@aztec/aztec.js/addresses';
3
- import type { ContractInstanceWithAddress } from '@aztec/aztec.js/contracts';
4
2
  import { TxHash } from '@aztec/aztec.js/tx';
5
3
  import { BlockNumber } from '@aztec/foundation/branded-types';
6
4
  import type { Fr } from '@aztec/foundation/curves/bn254';
7
5
  import type { EventSelector, FunctionSelector } from '@aztec/stdlib/abi';
8
6
  import type { AztecAddress } from '@aztec/stdlib/aztec-address';
7
+ import type { GasSettings } from '@aztec/stdlib/gas';
8
+ import type { PrivateLog } from '@aztec/stdlib/logs';
9
9
  import type { UInt64 } from '@aztec/stdlib/types';
10
10
 
11
11
  // These interfaces complement the ones defined in PXE, and combined with those contain the full list of oracles used by
@@ -36,6 +36,15 @@ export interface IAvmExecutionOracle {
36
36
  nullifierExists(siloedNullifier: Fr): Promise<boolean>;
37
37
  storageWrite(slot: Fr, value: Fr): Promise<void>;
38
38
  storageRead(slot: Fr, contractAddress: AztecAddress): Promise<Fr>;
39
+ getContractInstanceDeployer(address: AztecAddress): Promise<{ member: Fr; exists: boolean }>;
40
+ getContractInstanceClassId(address: AztecAddress): Promise<{ member: Fr; exists: boolean }>;
41
+ getContractInstanceInitializationHash(address: AztecAddress): Promise<{ member: Fr; exists: boolean }>;
42
+ getContractInstanceImmutablesHash(address: AztecAddress): Promise<{ member: Fr; exists: boolean }>;
43
+ returndataSize(): Promise<Fr>;
44
+ returndataCopy(rdOffset: number, copySize: number): Promise<Fr[]>;
45
+ call(l2Gas: number, daGas: number, address: AztecAddress, argsLength: number, args: Fr[]): Promise<Fr[]>;
46
+ staticCall(l2Gas: number, daGas: number, address: AztecAddress, argsLength: number, args: Fr[]): Promise<Fr[]>;
47
+ successCopy(): Promise<Fr>;
39
48
  }
40
49
 
41
50
  /**
@@ -49,37 +58,50 @@ export interface ITxeExecutionOracle {
49
58
  getNextBlockTimestamp(): Promise<UInt64>;
50
59
  advanceBlocksBy(blocks: number): Promise<void>;
51
60
  advanceTimestampBy(duration: UInt64): void;
52
- deploy(artifact: ContractArtifact, instance: ContractInstanceWithAddress, foreignSecret: Fr): Promise<void>;
61
+ deploy(
62
+ contractPath: string,
63
+ initializer: string,
64
+ args: Fr[],
65
+ secret: Fr,
66
+ salt: Fr,
67
+ deployer: AztecAddress,
68
+ ): Promise<Fr[]>;
53
69
  createAccount(secret: Fr): Promise<CompleteAddress>;
54
- addAccount(artifact: ContractArtifact, instance: ContractInstanceWithAddress, secret: Fr): Promise<CompleteAddress>;
70
+ addAccount(secret: Fr): Promise<CompleteAddress>;
55
71
  addAuthWitness(address: AztecAddress, messageHash: Fr): Promise<void>;
56
72
  getLastBlockTimestamp(): Promise<bigint>;
57
73
  getLastTxEffects(): Promise<{
58
74
  txHash: TxHash;
59
75
  noteHashes: Fr[];
60
76
  nullifiers: Fr[];
77
+ privateLogs: PrivateLog[];
61
78
  }>;
62
79
  getPrivateEvents(selector: EventSelector, contractAddress: AztecAddress, scope: AztecAddress): Promise<Fr[][]>;
63
80
  privateCallNewFlow(
64
- from: AztecAddress,
81
+ from: AztecAddress | undefined,
65
82
  targetContractAddress: AztecAddress,
66
83
  functionSelector: FunctionSelector,
67
84
  args: Fr[],
68
85
  argsHash: Fr,
69
86
  isStaticCall: boolean,
87
+ additionalScopes: AztecAddress[],
70
88
  jobId: string,
71
- ): Promise<Fr[]>;
89
+ authorizedUtilityCallTargets: AztecAddress[],
90
+ gasSettings: GasSettings,
91
+ ): Promise<{ returnValues: Fr[]; offchainEffects: Fr[][] }>;
72
92
  executeUtilityFunction(
73
93
  targetContractAddress: AztecAddress,
74
94
  functionSelector: FunctionSelector,
75
95
  args: Fr[],
76
96
  jobId: string,
97
+ authorizedUtilityCallTargets: AztecAddress[],
77
98
  ): Promise<Fr[]>;
78
99
  publicCallNewFlow(
79
- from: AztecAddress,
100
+ from: AztecAddress | undefined,
80
101
  targetContractAddress: AztecAddress,
81
102
  calldata: Fr[],
82
103
  isStaticCall: boolean,
104
+ gasSettings: GasSettings,
83
105
  ): Promise<Fr[]>;
84
106
  // TODO(F-335): Drop this from here as it's not a real oracle handler - it's only called from
85
107
  // RPCTranslator::txeGetPrivateEvents and never from Noir.
@@ -0,0 +1,84 @@
1
+ /* eslint-disable camelcase */
2
+ import { AztecAddress } from '@aztec/aztec.js/addresses';
3
+ import { Fr } from '@aztec/foundation/curves/bn254';
4
+ import { type NamedValue, Option } from '@aztec/pxe/simulator';
5
+
6
+ import { TXE_ORACLE_REGISTRY } from '../txe_oracle_registry.js';
7
+
8
+ /**
9
+ * Fixture definitions for oracle roundtrip tests.
10
+ *
11
+ * Each key is an oracle name from TXE_ORACLE_REGISTRY, and the value is an array of scenarios. The resolver selects a
12
+ * scenario (by name for multi-scenario oracles, or automatically for single-scenario ones), verifies that the
13
+ * deserialized inputs match the fixture's expected values, and returns the fixture output.
14
+ *
15
+ * To add a new oracle test:
16
+ * 1. Add a fixture entry here with the expected inputs and output.
17
+ * 2. Add a Noir `#[oracle_test]` function that calls the oracle with the same input values and asserts the output.
18
+ */
19
+ export const ORACLE_TEST_FIXTURES: OracleTestFixtures = {
20
+ aztec_avm_storageRead: [
21
+ makeScenario({
22
+ inputs: { slot: new Fr(10), contractAddress: AztecAddress.fromNumber(1) },
23
+ output: new Fr(42),
24
+ }),
25
+ ],
26
+
27
+ aztec_prv_getSenderForTags: [
28
+ makeScenario({
29
+ scenario: 'some',
30
+ inputs: {},
31
+ output: Option.some(AztecAddress.fromNumber(42)),
32
+ }),
33
+ makeScenario({
34
+ scenario: 'none',
35
+ inputs: {},
36
+ output: Option.none(AztecAddress.ZERO),
37
+ }),
38
+ ],
39
+ };
40
+
41
+ /**
42
+ * Converts a `NamedValue` tuple (from `deserializeParams`) to a record.
43
+ * `[NamedValue<'slot', Fr>, NamedValue<'addr', AztecAddress>]` → `{ slot: Fr, addr: AztecAddress }`.
44
+ */
45
+ type NamedValuesToRecord<T extends readonly NamedValue[]> = {
46
+ [K in T[number] as K extends NamedValue<infer N, any> ? N : never]: K extends NamedValue<string, infer V> ? V : never;
47
+ };
48
+
49
+ /** Extracts the deserialized inputs record type from an oracle registry entry. */
50
+ type OracleInputs<K extends keyof typeof TXE_ORACLE_REGISTRY> = (typeof TXE_ORACLE_REGISTRY)[K] extends {
51
+ deserializeParams(...args: any[]): infer P;
52
+ }
53
+ ? P extends readonly NamedValue[]
54
+ ? NamedValuesToRecord<P>
55
+ : Record<string, never>
56
+ : Record<string, never>;
57
+
58
+ /** Extracts the return type from an oracle registry entry's serializeReturn method. */
59
+ type OracleOutput<K extends keyof typeof TXE_ORACLE_REGISTRY> = (typeof TXE_ORACLE_REGISTRY)[K] extends {
60
+ serializeReturn(result: infer R): any;
61
+ }
62
+ ? R
63
+ : void;
64
+
65
+ /** A single test scenario for an oracle, typed against the oracle registry. */
66
+ export interface OracleTestScenario<K extends keyof typeof TXE_ORACLE_REGISTRY = keyof typeof TXE_ORACLE_REGISTRY> {
67
+ /** Scenario name for selection via `#[oracle_test("name")]`. Required when an oracle has multiple scenarios. */
68
+ scenario?: string;
69
+ /** Expected deserialized inputs. */
70
+ inputs: OracleInputs<K>;
71
+ /** Return value to serialize back to Noir. */
72
+ output: OracleOutput<K>;
73
+ }
74
+
75
+ type OracleTestFixtures = {
76
+ [K in keyof typeof TXE_ORACLE_REGISTRY]?: OracleTestScenario<K>[];
77
+ };
78
+
79
+ /** Creates a typed fixture scenario for a given oracle. */
80
+ function makeScenario<K extends keyof typeof TXE_ORACLE_REGISTRY>(
81
+ scenario: OracleTestScenario<K>,
82
+ ): OracleTestScenario<K> {
83
+ return scenario;
84
+ }
@@ -0,0 +1,45 @@
1
+ /* eslint-disable camelcase */
2
+ import { createSafeJsonRpcServer } from '@aztec/foundation/json-rpc/server';
3
+ import type { Logger } from '@aztec/foundation/log';
4
+ import type { ApiSchemaFor } from '@aztec/foundation/schemas';
5
+ import { zodFor } from '@aztec/stdlib/schemas';
6
+
7
+ import { z } from 'zod';
8
+
9
+ import { ForeignCallArgsSchema, type ForeignCallResult, ForeignCallResultSchema } from '../../utils/encoding.js';
10
+ import { TXE_ORACLE_REGISTRY } from '../txe_oracle_registry.js';
11
+ import { ORACLE_TEST_FIXTURES } from './fixtures.js';
12
+ import { type OracleTestCallInput, OracleTestResolver } from './resolver.js';
13
+
14
+ const OracleTestCallInputSchema = zodFor<OracleTestCallInput>()(
15
+ z.object({
16
+ session_id: z.number().nonnegative(),
17
+ function: z.string(),
18
+ root_path: z.string(),
19
+ package_name: z.string(),
20
+ inputs: ForeignCallArgsSchema,
21
+ }),
22
+ );
23
+
24
+ /** Narrowed interface exposing only the RPC method. */
25
+ interface OracleTestRpcHandler {
26
+ resolve_foreign_call(callData: OracleTestCallInput): Promise<ForeignCallResult>;
27
+ }
28
+
29
+ const OracleTestRpcHandlerSchema: ApiSchemaFor<OracleTestRpcHandler> = {
30
+ resolve_foreign_call: z.function({
31
+ input: z.tuple([OracleTestCallInputSchema]),
32
+ output: ForeignCallResultSchema,
33
+ }),
34
+ };
35
+
36
+ /** Creates an RPC server backed by the oracle test resolver. */
37
+ export function createOracleTestRpcServer(logger: Logger) {
38
+ const resolver = new OracleTestResolver(TXE_ORACLE_REGISTRY, ORACLE_TEST_FIXTURES, logger);
39
+ const server = createSafeJsonRpcServer<OracleTestRpcHandler>(resolver, OracleTestRpcHandlerSchema, {
40
+ http200OnError: true,
41
+ });
42
+ return { server, resolver };
43
+ }
44
+
45
+ export { OracleTestResolver } from './resolver.js';