@aztec-labs/prover-node 6.0.0-nightly.20260829

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 (90) hide show
  1. package/README.md +573 -0
  2. package/dest/actions/download-epoch-proving-job.d.ts +18 -0
  3. package/dest/actions/download-epoch-proving-job.d.ts.map +1 -0
  4. package/dest/actions/download-epoch-proving-job.js +37 -0
  5. package/dest/actions/index.d.ts +3 -0
  6. package/dest/actions/index.d.ts.map +1 -0
  7. package/dest/actions/index.js +2 -0
  8. package/dest/actions/rerun-epoch-proving-job.d.ts +23 -0
  9. package/dest/actions/rerun-epoch-proving-job.d.ts.map +1 -0
  10. package/dest/actions/rerun-epoch-proving-job.js +264 -0
  11. package/dest/actions/upload-epoch-proof-failure.d.ts +15 -0
  12. package/dest/actions/upload-epoch-proof-failure.d.ts.map +1 -0
  13. package/dest/actions/upload-epoch-proof-failure.js +78 -0
  14. package/dest/bin/run-failed-epoch.d.ts +2 -0
  15. package/dest/bin/run-failed-epoch.d.ts.map +1 -0
  16. package/dest/bin/run-failed-epoch.js +68 -0
  17. package/dest/checkpoint-store.d.ts +95 -0
  18. package/dest/checkpoint-store.d.ts.map +1 -0
  19. package/dest/checkpoint-store.js +178 -0
  20. package/dest/config.d.ts +31 -0
  21. package/dest/config.d.ts.map +1 -0
  22. package/dest/config.js +143 -0
  23. package/dest/factory.d.ts +36 -0
  24. package/dest/factory.d.ts.map +1 -0
  25. package/dest/factory.js +105 -0
  26. package/dest/index.d.ts +8 -0
  27. package/dest/index.d.ts.map +1 -0
  28. package/dest/index.js +7 -0
  29. package/dest/job/checkpoint-prover.d.ts +169 -0
  30. package/dest/job/checkpoint-prover.d.ts.map +1 -0
  31. package/dest/job/checkpoint-prover.js +413 -0
  32. package/dest/job/epoch-proving-job-data.d.ts +20 -0
  33. package/dest/job/epoch-proving-job-data.d.ts.map +1 -0
  34. package/dest/job/epoch-proving-job-data.js +61 -0
  35. package/dest/job/epoch-session.d.ts +160 -0
  36. package/dest/job/epoch-session.d.ts.map +1 -0
  37. package/dest/job/epoch-session.js +744 -0
  38. package/dest/job/top-tree-job.d.ts +82 -0
  39. package/dest/job/top-tree-job.d.ts.map +1 -0
  40. package/dest/job/top-tree-job.js +152 -0
  41. package/dest/metrics.d.ts +88 -0
  42. package/dest/metrics.d.ts.map +1 -0
  43. package/dest/metrics.js +264 -0
  44. package/dest/monitors/epoch-monitor.d.ts +43 -0
  45. package/dest/monitors/epoch-monitor.d.ts.map +1 -0
  46. package/dest/monitors/epoch-monitor.js +111 -0
  47. package/dest/monitors/index.d.ts +2 -0
  48. package/dest/monitors/index.d.ts.map +1 -0
  49. package/dest/monitors/index.js +1 -0
  50. package/dest/proof-publishing-service.d.ts +161 -0
  51. package/dest/proof-publishing-service.d.ts.map +1 -0
  52. package/dest/proof-publishing-service.js +335 -0
  53. package/dest/prover-node-publisher.d.ts +71 -0
  54. package/dest/prover-node-publisher.d.ts.map +1 -0
  55. package/dest/prover-node-publisher.js +346 -0
  56. package/dest/prover-node.d.ts +185 -0
  57. package/dest/prover-node.d.ts.map +1 -0
  58. package/dest/prover-node.js +1056 -0
  59. package/dest/prover-publisher-factory.d.ts +27 -0
  60. package/dest/prover-publisher-factory.d.ts.map +1 -0
  61. package/dest/prover-publisher-factory.js +29 -0
  62. package/dest/session-manager.d.ts +158 -0
  63. package/dest/session-manager.d.ts.map +1 -0
  64. package/dest/session-manager.js +493 -0
  65. package/dest/test/index.d.ts +13 -0
  66. package/dest/test/index.d.ts.map +1 -0
  67. package/dest/test/index.js +3 -0
  68. package/package.json +102 -0
  69. package/src/actions/download-epoch-proving-job.ts +43 -0
  70. package/src/actions/index.ts +2 -0
  71. package/src/actions/rerun-epoch-proving-job.ts +224 -0
  72. package/src/actions/upload-epoch-proof-failure.ts +87 -0
  73. package/src/bin/run-failed-epoch.ts +78 -0
  74. package/src/checkpoint-store.ts +212 -0
  75. package/src/config.ts +208 -0
  76. package/src/factory.ts +191 -0
  77. package/src/index.ts +7 -0
  78. package/src/job/checkpoint-prover.ts +551 -0
  79. package/src/job/epoch-proving-job-data.ts +94 -0
  80. package/src/job/epoch-session.ts +466 -0
  81. package/src/job/top-tree-job.ts +227 -0
  82. package/src/metrics.ts +323 -0
  83. package/src/monitors/epoch-monitor.ts +110 -0
  84. package/src/monitors/index.ts +1 -0
  85. package/src/proof-publishing-service.ts +427 -0
  86. package/src/prover-node-publisher.ts +473 -0
  87. package/src/prover-node.ts +830 -0
  88. package/src/prover-publisher-factory.ts +48 -0
  89. package/src/session-manager.ts +593 -0
  90. package/src/test/index.ts +14 -0
@@ -0,0 +1,224 @@
1
+ import { createArchiverStore, createContractDataSource } from '@aztec-labs/archiver';
2
+ import type { L1ContractsConfig } from '@aztec-labs/ethereum/config';
3
+ import { BlockNumber } from '@aztec-labs/foundation/branded-types';
4
+ import type { Logger } from '@aztec-labs/foundation/log';
5
+ import { DateProvider } from '@aztec-labs/foundation/timer';
6
+ import { type ProverClientConfig, createProverClient } from '@aztec-labs/prover-client';
7
+ import { ProverBrokerConfig, createAndStartProvingBroker } from '@aztec-labs/prover-client/broker';
8
+ import { getLastSiblingPath } from '@aztec-labs/prover-client/helpers';
9
+ import { type CheckpointSubTreeProofs, ChonkCache } from '@aztec-labs/prover-client/orchestrator';
10
+ import { AvmSimulatorPool, PublicProcessorFactory } from '@aztec-labs/simulator/server';
11
+ import type { L2Block } from '@aztec-labs/stdlib/block';
12
+ import { getEpochAtSlot, getSlotRangeForEpoch } from '@aztec-labs/stdlib/epoch-helpers';
13
+ import type { ITxProvider } from '@aztec-labs/stdlib/interfaces/server';
14
+ import type { DataStoreConfig } from '@aztec-labs/stdlib/kv-store';
15
+ import { MerkleTreeId } from '@aztec-labs/stdlib/trees';
16
+ import type { Tx, TxHash } from '@aztec-labs/stdlib/tx';
17
+ import type { GenesisData } from '@aztec-labs/stdlib/world-state';
18
+ import { getTelemetryClient } from '@aztec-labs/telemetry-client';
19
+ import { createWorldState } from '@aztec-labs/world-state';
20
+ import { readFileSync } from 'fs';
21
+
22
+ import { CheckpointProver } from '../job/checkpoint-prover.js';
23
+ import { type EpochProvingJobData, deserializeEpochProvingJobData } from '../job/epoch-proving-job-data.js';
24
+ import { EpochSession, type SessionSpec } from '../job/epoch-session.js';
25
+ import { ProverNodeJobMetrics } from '../metrics.js';
26
+
27
+ type RerunConfig = DataStoreConfig &
28
+ ProverBrokerConfig &
29
+ ProverClientConfig &
30
+ Pick<L1ContractsConfig, 'aztecEpochDuration'>;
31
+
32
+ /**
33
+ * Given a local folder where `downloadEpochProvingJob` was called, creates a new archiver and world state
34
+ * using the state snapshots, and creates a new epoch proving session to prove the downloaded proving job.
35
+ * Proving is done with a local proving broker and agents as specified by the config.
36
+ */
37
+ export async function rerunEpochProvingJob(localPath: string, log: Logger, config: RerunConfig, genesis?: GenesisData) {
38
+ await using ctx = await createRerunContext(localPath, log, config, genesis);
39
+ const { jobData, prover, metrics } = ctx;
40
+
41
+ log.info(`Rerunning epoch proving for epoch ${jobData.epochNumber}`);
42
+
43
+ const provers: CheckpointProver[] = [];
44
+ for (let i = 0; i < jobData.checkpoints.length; i++) {
45
+ provers.push(await buildCheckpointProver(ctx, i, log));
46
+ }
47
+
48
+ // Local rerun never publishes — stub the service so submit() always resolves 'published'
49
+ // and withdraw is a no-op.
50
+ const publishingService = {
51
+ submit: () => Promise.resolve('published' as const),
52
+ withdraw: () => {},
53
+ };
54
+
55
+ const l1Constants = { epochDuration: config.aztecEpochDuration };
56
+ const [fromSlot, toSlot] = getSlotRangeForEpoch(jobData.epochNumber, l1Constants);
57
+ const spec: SessionSpec = { kind: 'full', epochNumber: jobData.epochNumber, fromSlot, toSlot };
58
+
59
+ const session = new EpochSession(spec, provers, {
60
+ proverFactory: prover,
61
+ proverId: prover.getProverId(),
62
+ publishingService,
63
+ metrics,
64
+ dateProvider: new DateProvider(),
65
+ deadline: undefined,
66
+ config: {},
67
+ bindings: log.getBindings(),
68
+ });
69
+
70
+ const finalState = await session.start();
71
+ log.info(`Completed proving for epoch ${jobData.epochNumber} with status ${finalState}`, {
72
+ derivedEpoch: getEpochAtSlot(provers[0].slotNumber, l1Constants),
73
+ });
74
+ return finalState;
75
+ }
76
+
77
+ /**
78
+ * Re-proves a single downloaded checkpoint proving job (as uploaded by a `CheckpointProver` failure).
79
+ * Reconstructs just that checkpoint's sub-tree prover from the snapshot and awaits its block proofs — no
80
+ * epoch top-tree, no L1 submission — so a checkpoint-level failure can be reproduced offline in isolation.
81
+ * Returns the block-rollup proof outputs on success; throws if the checkpoint fails to prove again.
82
+ */
83
+ export async function rerunCheckpointProvingJob(
84
+ localPath: string,
85
+ log: Logger,
86
+ config: RerunConfig,
87
+ genesis?: GenesisData,
88
+ ): Promise<CheckpointSubTreeProofs['blockProofOutputs']> {
89
+ await using ctx = await createRerunContext(localPath, log, config, genesis);
90
+ const { jobData } = ctx;
91
+ const checkpointNumber = jobData.checkpoints[0].number;
92
+
93
+ log.info(`Rerunning checkpoint proving for checkpoint ${checkpointNumber} (epoch ${jobData.epochNumber})`);
94
+
95
+ const prover = await buildCheckpointProver(ctx, 0, log);
96
+ try {
97
+ const { blockProofOutputs } = await prover.whenSubTreeProofsReady();
98
+ log.info(`Completed proving for checkpoint ${checkpointNumber} with ${blockProofOutputs.length} block proof(s)`);
99
+ return blockProofOutputs;
100
+ } finally {
101
+ prover.cancel({ routine: true });
102
+ await prover.whenDone();
103
+ }
104
+ }
105
+
106
+ /** Everything a rerun needs, reconstructed from the downloaded snapshot + job data. */
107
+ type RerunContext = {
108
+ jobData: EpochProvingJobData;
109
+ metrics: ProverNodeJobMetrics;
110
+ worldState: Awaited<ReturnType<typeof createWorldState>>;
111
+ publicProcessorFactory: PublicProcessorFactory;
112
+ prover: Awaited<ReturnType<typeof createProverClient>>;
113
+ chonkCache: ChonkCache;
114
+ txProvider: ITxProvider;
115
+ } & AsyncDisposable;
116
+
117
+ /**
118
+ * Rebuilds the offline proving environment from a downloaded job: world state + archiver from the
119
+ * snapshots, a local proving broker + client, a chonk cache, and a tx provider that replays the job's txs.
120
+ */
121
+ async function createRerunContext(
122
+ localPath: string,
123
+ log: Logger,
124
+ config: RerunConfig,
125
+ genesis?: GenesisData,
126
+ ): Promise<RerunContext> {
127
+ const jobData = deserializeEpochProvingJobData(readFileSync(localPath));
128
+ log.info(`Loaded proving job data for epoch ${jobData.epochNumber}`);
129
+
130
+ const telemetry = getTelemetryClient();
131
+ const metrics = new ProverNodeJobMetrics(telemetry.getMeter('prover-job'), telemetry.getTracer('prover-job'));
132
+ const worldState = await createWorldState(config, genesis);
133
+ const initialBlockHash = await worldState.getInitialHeader().hash();
134
+ const archiver = await createArchiverStore(config, initialBlockHash);
135
+ const avmSimulator = await AvmSimulatorPool.spawn({ wsdbIpcPath: worldState.getIpcPath() });
136
+ const publicProcessorFactory = new PublicProcessorFactory(
137
+ createContractDataSource(archiver),
138
+ avmSimulator,
139
+ undefined,
140
+ undefined,
141
+ log.getBindings(),
142
+ );
143
+ const broker = await createAndStartProvingBroker(config, telemetry);
144
+ const prover = await createProverClient(config, worldState, broker, telemetry);
145
+ const chonkCache = new ChonkCache(log.getBindings());
146
+ const txProvider = makeReplayingTxProvider(jobData.txs);
147
+
148
+ return {
149
+ jobData,
150
+ metrics,
151
+ worldState,
152
+ publicProcessorFactory,
153
+ prover,
154
+ chonkCache,
155
+ txProvider,
156
+ async [Symbol.asyncDispose]() {
157
+ await avmSimulator[Symbol.asyncDispose]();
158
+ await worldState[Symbol.asyncDispose]();
159
+ },
160
+ };
161
+ }
162
+
163
+ /** Reconstructs the `CheckpointProver` for the checkpoint at `index` in the job, ready to prove. */
164
+ async function buildCheckpointProver(ctx: RerunContext, index: number, log: Logger): Promise<CheckpointProver> {
165
+ const { jobData, worldState, prover, chonkCache, publicProcessorFactory, txProvider, metrics } = ctx;
166
+ const checkpoint = jobData.checkpoints[index];
167
+ const previousBlockHeader =
168
+ index === 0 ? jobData.previousBlockHeader : jobData.checkpoints[index - 1].blocks.at(-1)!.header;
169
+ const previousInboxRollingHash =
170
+ index === 0 ? jobData.previousInboxRollingHash : jobData.checkpoints[index - 1].header.inboxRollingHash;
171
+ const l1ToL2Messages = jobData.l1ToL2Messages[checkpoint.number] ?? [];
172
+ const previousArchiveSiblingPath = await getLastSiblingPath(
173
+ MerkleTreeId.ARCHIVE,
174
+ worldState.getSnapshot(BlockNumber(checkpoint.blocks[0].number - 1)),
175
+ );
176
+ const attestations = checkpoint.number === jobData.checkpoints.at(-1)!.number ? jobData.attestations : [];
177
+ return new CheckpointProver(
178
+ {
179
+ checkpoint,
180
+ epochNumber: jobData.epochNumber,
181
+ attestations,
182
+ previousBlockHeader,
183
+ l1ToL2Messages,
184
+ previousInboxRollingHash,
185
+ previousArchiveSiblingPath,
186
+ },
187
+ {
188
+ proverFactory: prover,
189
+ chonkCache,
190
+ publicProcessorFactory,
191
+ dbProvider: worldState,
192
+ txProvider,
193
+ dateProvider: new DateProvider(),
194
+ proverId: prover.getProverId(),
195
+ metrics,
196
+ txGatheringTimeoutMs: 120_000,
197
+ deadline: undefined,
198
+ log,
199
+ },
200
+ );
201
+ }
202
+
203
+ /** Build a synthetic ITxProvider that returns the supplied txs map by lookup. */
204
+ function makeReplayingTxProvider(txs: Map<string, Tx>): ITxProvider {
205
+ const lookup = (hashes: TxHash[]) => {
206
+ const found: Tx[] = [];
207
+ const missing: TxHash[] = [];
208
+ for (const hash of hashes) {
209
+ const tx = txs.get(hash.toString());
210
+ if (tx) {
211
+ found.push(tx);
212
+ } else {
213
+ missing.push(hash);
214
+ }
215
+ }
216
+ return { txs: found, missingTxs: missing };
217
+ };
218
+ return {
219
+ getAvailableTxs: hashes => Promise.resolve(lookup(hashes)),
220
+ hasTxs: hashes => Promise.resolve(hashes.map(h => txs.has(h.toString()))),
221
+ getTxsForBlockProposal: () => Promise.resolve({ txs: [], missingTxs: [] }),
222
+ getTxsForBlock: (block: L2Block) => Promise.resolve(lookup(block.body.txEffects.map(e => e.txHash))),
223
+ };
224
+ }
@@ -0,0 +1,87 @@
1
+ import { ARCHIVER_DB_VERSION, type Archiver } from '@aztec-labs/archiver';
2
+ import { tryRmDir } from '@aztec-labs/foundation/fs';
3
+ import { jsonStringify } from '@aztec-labs/foundation/json-rpc';
4
+ import type { Logger } from '@aztec-labs/foundation/log';
5
+ import { isoDate } from '@aztec-labs/foundation/string';
6
+ import { buildSnapshotMetadata, createBackups } from '@aztec-labs/node-lib/actions';
7
+ import type { ChainConfig } from '@aztec-labs/stdlib/config';
8
+ import { type FileStore, createFileStore } from '@aztec-labs/stdlib/file-store';
9
+ import type { WorldStateSynchronizer } from '@aztec-labs/stdlib/interfaces/server';
10
+ import type { DataStoreConfig } from '@aztec-labs/stdlib/kv-store';
11
+ import { type UploadSnapshotMetadata, getBasePath, uploadSnapshotData } from '@aztec-labs/stdlib/snapshots';
12
+ import { WORLD_STATE_DB_VERSION } from '@aztec-labs/world-state';
13
+ import { mkdtemp } from 'fs/promises';
14
+ import { tmpdir } from 'os';
15
+ import { dirname, join } from 'path';
16
+
17
+ import { type EpochProvingJobData, serializeEpochProvingJobData } from '../job/epoch-proving-job-data.js';
18
+
19
+ type UploadEpochProofConfig = Pick<ChainConfig, 'l1ChainId' | 'rollupVersion'> & Pick<DataStoreConfig, 'dataDirectory'>;
20
+
21
+ /** Whether uploaded data to the file store should be of public access. */
22
+ const PUBLIC_UPLOADS = true;
23
+
24
+ /**
25
+ * Uploads a snapshot of world state and archiver (requires pausing them) along with the proving job data,
26
+ * so we can download and re-run the job later under the same conditions.
27
+ * @param location The location to upload the data to (used to create the `FileStore`).
28
+ */
29
+ export async function uploadEpochProofFailure(
30
+ location: string,
31
+ jobId: string,
32
+ jobData: EpochProvingJobData,
33
+ archiver: Archiver,
34
+ worldState: WorldStateSynchronizer,
35
+ config: UploadEpochProofConfig,
36
+ log: Logger,
37
+ ) {
38
+ const epochNumber = jobData.epochNumber;
39
+ log.warn(`Uploading epoch proof failure for ${epochNumber} to ${location}`, { epochNumber, jobId, location });
40
+
41
+ const backupDir = await mkdtemp(join(config.dataDirectory ?? tmpdir(), 'epoch-proof-data-'));
42
+ const store = await createFileStore(location);
43
+ if (!store) {
44
+ throw new Error(`Failed to create file store for epoch proof failure upload for location ${location}.`);
45
+ }
46
+
47
+ try {
48
+ const versions = { archiver: ARCHIVER_DB_VERSION, worldState: WORLD_STATE_DB_VERSION };
49
+ const uploadMetadata = await buildSnapshotMetadata(archiver, config);
50
+ const paths = await createBackups(backupDir, archiver, worldState, log);
51
+
52
+ const basePath = `${getBasePath(uploadMetadata)}/${epochNumber}-${isoDate()}-${jobId}`;
53
+ const pathFor = (key: string) => `${basePath}/${key}.db`;
54
+ const [metadata, dataUrl, metadataUrl] = await Promise.all([
55
+ uploadSnapshotData(paths, versions, uploadMetadata, store, { pathFor, private: !PUBLIC_UPLOADS }),
56
+ uploadJobData(jobData, store, basePath),
57
+ uploadSnapshotMetadata(uploadMetadata, store, basePath),
58
+ ] as const);
59
+
60
+ const baseUrl = dirname(metadataUrl);
61
+ log.warn(`Uploaded epoch ${epochNumber} proof failure data to ${baseUrl}`, {
62
+ epochNumber,
63
+ location,
64
+ basePath,
65
+ metadataUrl,
66
+ dataUrl,
67
+ metadata,
68
+ jobId,
69
+ });
70
+ return baseUrl;
71
+ } finally {
72
+ log.info(`Cleaning up backup dir ${backupDir}`);
73
+ await tryRmDir(backupDir, log);
74
+ }
75
+ }
76
+
77
+ async function uploadJobData(jobData: EpochProvingJobData, store: FileStore, basePath: string) {
78
+ const data = serializeEpochProvingJobData(jobData);
79
+ const path = `${basePath}/data.bin`;
80
+ return await store.save(path, data, { compress: true, public: PUBLIC_UPLOADS });
81
+ }
82
+
83
+ async function uploadSnapshotMetadata(metadata: UploadSnapshotMetadata, store: FileStore, basePath: string) {
84
+ const data = Buffer.from(jsonStringify(metadata), 'utf-8');
85
+ const path = `${basePath}/metadata.json`;
86
+ return await store.save(path, data, { compress: false, public: PUBLIC_UPLOADS });
87
+ }
@@ -0,0 +1,78 @@
1
+ /* eslint-disable no-console */
2
+ import { getL1ContractsConfigEnvVars } from '@aztec-labs/ethereum/config';
3
+ import { EthAddress } from '@aztec-labs/foundation/eth-address';
4
+ import { jsonParseWithSchema, jsonStringify } from '@aztec-labs/foundation/json-rpc';
5
+ import { createLogger } from '@aztec-labs/foundation/log';
6
+ import { downloadEpochProvingJob, getProverNodeConfigFromEnv, rerunEpochProvingJob } from '@aztec-labs/prover-node';
7
+ import { type UploadSnapshotMetadata, UploadSnapshotMetadataSchema } from '@aztec-labs/stdlib/snapshots';
8
+ import { existsSync, mkdirSync } from 'fs';
9
+ import { readFile, writeFile } from 'fs/promises';
10
+ import { basename, join } from 'path';
11
+
12
+ const logger = createLogger('prover-node:run-failed-epoch');
13
+
14
+ function printUsage() {
15
+ console.error('Usage: run-failed-epoch <proof-uri> [out-dir=./data]');
16
+ }
17
+
18
+ async function rerunFailedEpoch(provingJobUrl: string, baseLocalDir: string) {
19
+ const localDir = join(baseLocalDir, basename(provingJobUrl));
20
+ const jobPath = join(localDir, 'data.bin');
21
+ const dataDir = join(localDir, 'state');
22
+
23
+ const env = getProverNodeConfigFromEnv();
24
+ const l1Config = getL1ContractsConfigEnvVars();
25
+ const config = {
26
+ ...env,
27
+ ...l1Config,
28
+ dataDirectory: dataDir,
29
+ dataStoreMapSizeKb: env.dataStoreMapSizeKb ?? 1024 * 1024,
30
+ proverId: env.proverId ?? EthAddress.random(),
31
+ };
32
+
33
+ let metadata: UploadSnapshotMetadata;
34
+ const metadataPath = join(localDir, 'metadata.json');
35
+ if (existsSync(metadataPath)) {
36
+ logger.info(`Using downloaded data`);
37
+ metadata = jsonParseWithSchema(await readFile(metadataPath, 'utf-8'), UploadSnapshotMetadataSchema);
38
+ } else {
39
+ logger.info(`Downloading epoch proving job data and state from ${provingJobUrl} to ${localDir}`);
40
+ metadata = await downloadEpochProvingJob(provingJobUrl!, logger, {
41
+ jobDataDownloadPath: jobPath,
42
+ dataDirectory: dataDir,
43
+ });
44
+ await writeFile(metadataPath, jsonStringify(metadata, true));
45
+ logger.info(`Download to ${localDir} complete`);
46
+ }
47
+
48
+ logger.info(`Rerunning proving job from ${jobPath} with state from ${dataDir}`, metadata);
49
+ const result = await rerunEpochProvingJob(jobPath, logger, {
50
+ ...config,
51
+ rollupAddress: metadata.rollupAddress,
52
+ rollupVersion: metadata.rollupVersion,
53
+ });
54
+
55
+ console.error(`Epoch proving job complete with result ${result}`);
56
+ }
57
+
58
+ async function main() {
59
+ if (process.argv[2] === '--help') {
60
+ printUsage();
61
+ return;
62
+ }
63
+
64
+ const uri = process.argv[2];
65
+ const outDir = process.argv[3] || './data';
66
+ if (!uri) {
67
+ printUsage();
68
+ throw new Error('Missing URL to epoch proving job');
69
+ }
70
+
71
+ mkdirSync(outDir, { recursive: true });
72
+ await rerunFailedEpoch(uri, outDir);
73
+ }
74
+
75
+ main().catch(err => {
76
+ console.error(err);
77
+ process.exit(1);
78
+ });
@@ -0,0 +1,212 @@
1
+ import type { BlockNumber, CheckpointNumber, EpochNumber, SlotNumber } from '@aztec-labs/foundation/branded-types';
2
+ import { type Logger, type LoggerBindings, createLogger } from '@aztec-labs/foundation/log';
3
+ import type { L2BlockSource } from '@aztec-labs/stdlib/block';
4
+ import type { Checkpoint } from '@aztec-labs/stdlib/checkpoint';
5
+ import { type L1RollupConstants, getEpochAtSlot, getSlotRangeForEpoch } from '@aztec-labs/stdlib/epoch-helpers';
6
+
7
+ import {
8
+ CheckpointProver,
9
+ type CheckpointProverArgs,
10
+ type CheckpointProverDeps,
11
+ type CheckpointProverTestHooks,
12
+ } from './job/checkpoint-prover.js';
13
+
14
+ /** Register-time data needed to construct a `CheckpointProver` (everything except the checkpoint + epoch). */
15
+ export type RegisterCheckpointData = Omit<CheckpointProverArgs, 'checkpoint' | 'epochNumber'>;
16
+
17
+ /** Factory used by the store to construct new provers. Tests can inject a stub. */
18
+ export type CheckpointProverFactory = (args: CheckpointProverArgs, deps: CheckpointProverDeps) => CheckpointProver;
19
+
20
+ /**
21
+ * Prover-node-wide registry of `CheckpointProver` instances, content-addressed by
22
+ * `(checkpoint number, slot, checkpoint archive root)`.
23
+ *
24
+ * The store survives every epoch / session boundary. A prover lives from its first
25
+ * `addOrUpdate` call until either:
26
+ * - its checkpoint is pruned by an L1 reorg (`cancelAndRemoveAboveBlock`), or
27
+ * - its epoch's proof-submission window has closed (`reapExpired`), so the proof could no
28
+ * longer be accepted on L1 even if produced.
29
+ *
30
+ * A prover's sub-tree work forks world-state per block and does not survive a prune of a base
31
+ * block, so there is nothing to preserve across a reorg: a pruned prover is cancelled and
32
+ * dropped, and a re-add (even of identical content) constructs a fresh prover.
33
+ */
34
+ export class CheckpointStore {
35
+ private readonly provers = new Map<string, CheckpointProver>();
36
+ /**
37
+ * Teardowns of provers already removed from `provers` (by prune or reap), awaited on `stop()`.
38
+ * Keyed by a monotonic id rather than the prover's content id: a prune-then-re-add can leave two
39
+ * teardowns for the same content id in flight at once, which a content-id key would clobber.
40
+ */
41
+ private readonly pendingTeardowns = new Map<number, Promise<void>>();
42
+ private nextTeardownId = 0;
43
+ /** Test-only hooks injected into every prover this store constructs. */
44
+ private testHooks: CheckpointProverTestHooks = {};
45
+ private readonly log: Logger;
46
+
47
+ constructor(
48
+ private readonly l2BlockSource: Pick<L2BlockSource, 'getL1Constants'>,
49
+ private readonly proverDeps: Omit<CheckpointProverDeps, 'log'>,
50
+ bindings?: LoggerBindings,
51
+ private readonly proverFactoryFn: CheckpointProverFactory = (args, deps) => new CheckpointProver(args, deps),
52
+ ) {
53
+ this.log = createLogger('prover-node:checkpoint-store', bindings);
54
+ }
55
+
56
+ public start(): Promise<void> {
57
+ return Promise.resolve();
58
+ }
59
+
60
+ public async stop(): Promise<void> {
61
+ // Cancel every live prover, then await both their teardown and any still in flight for provers
62
+ // already removed by a prune or reap.
63
+ const provers = Array.from(this.provers.values());
64
+ this.provers.clear();
65
+ for (const prover of provers) {
66
+ prover.cancel();
67
+ }
68
+ await Promise.allSettled([...provers.map(p => p.whenDone()), ...this.pendingTeardowns.values()]);
69
+ }
70
+
71
+ /**
72
+ * Tracks the teardown of a prover just removed from the store so `stop()` can await it. The entry
73
+ * removes itself once teardown settles, so the map stays bounded by the number in flight.
74
+ */
75
+ private trackTeardown(prover: CheckpointProver): void {
76
+ const id = this.nextTeardownId++;
77
+ const done = prover.whenDone();
78
+ this.pendingTeardowns.set(id, done);
79
+ void done.finally(() => this.pendingTeardowns.delete(id));
80
+ }
81
+
82
+ /**
83
+ * Registers a checkpoint with the store. If a prover already exists for the
84
+ * `(number, slot, archive root)` content key it is reused (an at-least-once re-registration of
85
+ * still-canonical content); otherwise a new prover is constructed.
86
+ */
87
+ public async addOrUpdate(checkpoint: Checkpoint, data: RegisterCheckpointData): Promise<CheckpointProver> {
88
+ const l1Constants = await this.l2BlockSource.getL1Constants();
89
+ const epochNumber = getEpochAtSlot(checkpoint.header.slotNumber, l1Constants);
90
+ const id = CheckpointProver.idFor(checkpoint);
91
+
92
+ const existing = this.provers.get(id);
93
+ if (existing) {
94
+ return existing;
95
+ }
96
+
97
+ // At most one canonical checkpoint per slot. A different checkpoint at the same slot means the
98
+ // caller forgot to prune the old chain before adding the replacement — surface it rather than
99
+ // silently creating a parallel canonical chain. A pruned checkpoint has already been removed,
100
+ // so every prover still in the store is canonical.
101
+ for (const prover of this.provers.values()) {
102
+ if (prover.slotNumber === checkpoint.header.slotNumber) {
103
+ throw new Error(
104
+ `Cannot add checkpoint ${checkpoint.number} (archive ${checkpoint.archive.root}) at slot ${checkpoint.header.slotNumber}: ` +
105
+ `a different checkpoint already occupies this slot. Prune it first.`,
106
+ );
107
+ }
108
+ }
109
+
110
+ const prover = this.proverFactoryFn(
111
+ { ...data, checkpoint, epochNumber },
112
+ { ...this.proverDeps, checkpointProveOverride: this.testHooks.checkpointProveOverride, log: this.log },
113
+ );
114
+ this.provers.set(id, prover);
115
+ return prover;
116
+ }
117
+
118
+ /**
119
+ * Installs test-only hooks applied to every prover constructed from now on. Used by the e2e harness to
120
+ * force a checkpoint sub-tree failure without monkey-patching the prover factory.
121
+ */
122
+ public setTestHooks(hooks: CheckpointProverTestHooks): void {
123
+ this.testHooks = hooks;
124
+ }
125
+
126
+ /**
127
+ * Cancels and removes every prover that holds a block above the prune target. A checkpoint is orphaned by a prune to
128
+ * block `targetBlockNumber` iff its last block sits above the target — including a checkpoint whose range straddles
129
+ * the target (partially orphaned), which block-range marking catches without boundary ambiguity. Keying off the
130
+ * surviving block number (rather than a checkpoint number) is correct even when the source has already
131
+ * re-checkpointed past the divergence: the prune event reports the highest surviving block, which by construction
132
+ * survives on the source, whereas the source's current checkpointed tip can sit above the prune target.
133
+ *
134
+ * The prover's in-flight sub-tree work forks world-state per block and faults once its base block is pruned, so it
135
+ * cannot be reused; it is cancelled (aborting the fork reads) and dropped. A subsequent re-add constructs a fresh
136
+ * prover. Returns the removed provers.
137
+ */
138
+ public cancelAndRemoveAboveBlock(targetBlockNumber: BlockNumber): CheckpointProver[] {
139
+ const affected: CheckpointProver[] = [];
140
+ for (const [id, prover] of Array.from(this.provers.entries())) {
141
+ const lastBlockNumber = prover.checkpoint.blocks.at(-1)!.number;
142
+ if (lastBlockNumber > targetBlockNumber) {
143
+ prover.cancel();
144
+ this.trackTeardown(prover);
145
+ this.provers.delete(id);
146
+ affected.push(prover);
147
+ }
148
+ }
149
+ return affected;
150
+ }
151
+
152
+ /**
153
+ * Drops provers whose epoch is at or below the supplied expired epoch. Once an epoch's
154
+ * proof-submission window has closed, its proof can no longer be accepted on L1, so the
155
+ * prover is no longer needed.
156
+ */
157
+ public reapExpired(expiredEpoch: EpochNumber): void {
158
+ const reaped: { id: string; checkpointNumber: CheckpointNumber; epochNumber: EpochNumber }[] = [];
159
+ for (const [id, prover] of Array.from(this.provers.entries())) {
160
+ if (prover.epochNumber <= expiredEpoch) {
161
+ reaped.push({ id, checkpointNumber: prover.checkpoint.number, epochNumber: prover.epochNumber });
162
+ prover.cancel({ routine: true });
163
+ this.trackTeardown(prover);
164
+ this.provers.delete(id);
165
+ }
166
+ }
167
+ if (reaped.length > 0) {
168
+ this.log.info(`Reaped ${reaped.length} expired CheckpointProver(s) for expiredEpoch ${expiredEpoch}`, {
169
+ expiredEpoch,
170
+ reapedCount: reaped.length,
171
+ reaped,
172
+ });
173
+ }
174
+ }
175
+
176
+ /** Returns the prover with the supplied id, or undefined. */
177
+ public get(id: string): CheckpointProver | undefined {
178
+ return this.provers.get(id);
179
+ }
180
+
181
+ /** Returns the prover for the supplied checkpoint (by its content-addressed id), or undefined. */
182
+ public getByCheckpoint(checkpoint: Checkpoint): CheckpointProver | undefined {
183
+ return this.provers.get(CheckpointProver.idFor(checkpoint));
184
+ }
185
+
186
+ /** Every prover currently in the store, in insertion order. */
187
+ public listAll(): CheckpointProver[] {
188
+ return Array.from(this.provers.values());
189
+ }
190
+
191
+ /** Provers in the store, sorted by checkpoint number. */
192
+ public list(): CheckpointProver[] {
193
+ return Array.from(this.provers.values()).sort((a, b) => a.checkpoint.number - b.checkpoint.number);
194
+ }
195
+
196
+ /**
197
+ * Provers whose slot is in the supplied epoch's slot range, sorted by checkpoint number.
198
+ */
199
+ public async listForEpoch(epoch: EpochNumber): Promise<CheckpointProver[]> {
200
+ const l1Constants = await this.l2BlockSource.getL1Constants();
201
+ const [fromSlot, toSlot] = getSlotRangeForEpoch(epoch, l1Constants);
202
+ return this.listInSlotRange(fromSlot, toSlot);
203
+ }
204
+
205
+ /** Provers whose slot falls within `[fromSlot, toSlot]`, sorted by checkpoint number. */
206
+ public listInSlotRange(fromSlot: SlotNumber, toSlot: SlotNumber): CheckpointProver[] {
207
+ return this.list().filter(p => p.slotNumber >= fromSlot && p.slotNumber <= toSlot);
208
+ }
209
+ }
210
+
211
+ /** Sub-set of `L1RollupConstants` actually consumed by the store's slot helpers. */
212
+ export type CheckpointStoreL1Constants = Pick<L1RollupConstants, 'epochDuration'>;