@aztec/prover-client 3.0.0-nightly.20250917 → 3.0.0-nightly.20250919

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 (65) hide show
  1. package/dest/block-factory/light.d.ts +5 -3
  2. package/dest/block-factory/light.d.ts.map +1 -1
  3. package/dest/block-factory/light.js +16 -9
  4. package/dest/mocks/fixtures.d.ts +3 -1
  5. package/dest/mocks/fixtures.d.ts.map +1 -1
  6. package/dest/mocks/fixtures.js +19 -2
  7. package/dest/mocks/test_context.d.ts +30 -9
  8. package/dest/mocks/test_context.d.ts.map +1 -1
  9. package/dest/mocks/test_context.js +68 -15
  10. package/dest/orchestrator/block-building-helpers.d.ts +16 -14
  11. package/dest/orchestrator/block-building-helpers.d.ts.map +1 -1
  12. package/dest/orchestrator/block-building-helpers.js +69 -66
  13. package/dest/orchestrator/block-proving-state.d.ts +59 -52
  14. package/dest/orchestrator/block-proving-state.d.ts.map +1 -1
  15. package/dest/orchestrator/block-proving-state.js +215 -187
  16. package/dest/orchestrator/checkpoint-proving-state.d.ts +62 -0
  17. package/dest/orchestrator/checkpoint-proving-state.d.ts.map +1 -0
  18. package/dest/orchestrator/checkpoint-proving-state.js +208 -0
  19. package/dest/orchestrator/epoch-proving-state.d.ts +32 -25
  20. package/dest/orchestrator/epoch-proving-state.d.ts.map +1 -1
  21. package/dest/orchestrator/epoch-proving-state.js +132 -81
  22. package/dest/orchestrator/orchestrator.d.ts +25 -24
  23. package/dest/orchestrator/orchestrator.d.ts.map +1 -1
  24. package/dest/orchestrator/orchestrator.js +331 -203
  25. package/dest/orchestrator/tx-proving-state.d.ts +3 -3
  26. package/dest/orchestrator/tx-proving-state.d.ts.map +1 -1
  27. package/dest/orchestrator/tx-proving-state.js +5 -5
  28. package/dest/prover-client/server-epoch-prover.d.ts +8 -7
  29. package/dest/prover-client/server-epoch-prover.d.ts.map +1 -1
  30. package/dest/prover-client/server-epoch-prover.js +7 -7
  31. package/dest/proving_broker/broker_prover_facade.d.ts +18 -13
  32. package/dest/proving_broker/broker_prover_facade.d.ts.map +1 -1
  33. package/dest/proving_broker/broker_prover_facade.js +36 -21
  34. package/dest/proving_broker/fixtures.js +1 -1
  35. package/dest/proving_broker/proof_store/index.d.ts +1 -0
  36. package/dest/proving_broker/proof_store/index.d.ts.map +1 -1
  37. package/dest/proving_broker/proof_store/index.js +1 -0
  38. package/dest/proving_broker/proving_broker.d.ts.map +1 -1
  39. package/dest/proving_broker/proving_broker.js +27 -16
  40. package/dest/proving_broker/proving_job_controller.d.ts.map +1 -1
  41. package/dest/proving_broker/proving_job_controller.js +34 -14
  42. package/dest/test/mock_prover.d.ts +18 -13
  43. package/dest/test/mock_prover.d.ts.map +1 -1
  44. package/dest/test/mock_prover.js +31 -16
  45. package/package.json +16 -17
  46. package/src/block-factory/light.ts +33 -9
  47. package/src/mocks/fixtures.ts +25 -7
  48. package/src/mocks/test_context.ts +113 -21
  49. package/src/orchestrator/block-building-helpers.ts +107 -93
  50. package/src/orchestrator/block-proving-state.ts +232 -244
  51. package/src/orchestrator/checkpoint-proving-state.ts +294 -0
  52. package/src/orchestrator/epoch-proving-state.ts +169 -121
  53. package/src/orchestrator/orchestrator.ts +483 -262
  54. package/src/orchestrator/tx-proving-state.ts +6 -6
  55. package/src/prover-client/server-epoch-prover.ts +30 -16
  56. package/src/proving_broker/broker_prover_facade.ts +151 -83
  57. package/src/proving_broker/fixtures.ts +1 -1
  58. package/src/proving_broker/proof_store/index.ts +1 -0
  59. package/src/proving_broker/proving_broker.ts +34 -16
  60. package/src/proving_broker/proving_job_controller.ts +34 -14
  61. package/src/test/mock_prover.ts +125 -54
  62. package/dest/bin/get-proof-inputs.d.ts +0 -2
  63. package/dest/bin/get-proof-inputs.d.ts.map +0 -1
  64. package/dest/bin/get-proof-inputs.js +0 -51
  65. package/src/bin/get-proof-inputs.ts +0 -59
@@ -1,268 +1,304 @@
1
- import { BatchedBlobAccumulator, BlobAccumulatorPublicInputs, SpongeBlob } from '@aztec/blob-lib';
2
- import { BLOBS_PER_BLOCK, FIELDS_PER_BLOB, NUM_BASE_PARITY_PER_ROOT_PARITY } from '@aztec/constants';
3
- import { padArrayEnd } from '@aztec/foundation/collection';
4
- import { BLS12Point, Fr } from '@aztec/foundation/fields';
1
+ import { NUM_BASE_PARITY_PER_ROOT_PARITY } from '@aztec/constants';
2
+ import { assertLength } from '@aztec/foundation/serialize';
5
3
  import { UnbalancedTreeStore } from '@aztec/foundation/trees';
6
- import { getVKIndex, getVKSiblingPath, getVKTreeRoot } from '@aztec/noir-protocol-circuits-types/vk-tree';
7
- import { protocolContractTreeRoot } from '@aztec/protocol-contracts';
8
- import { RootParityInput, RootParityInputs } from '@aztec/stdlib/parity';
9
- import { BlockConstantData, BlockRootRollupBlobData, BlockRootRollupData, BlockRootRollupInputs, EmptyBlockRootRollupInputs, EpochConstantData, MergeRollupInputs, PaddingBlockRootRollupInputs, PreviousRollupData, SingleTxBlockRootRollupInputs } from '@aztec/stdlib/rollup';
10
- import { StateReference } from '@aztec/stdlib/tx';
11
- import { VkData } from '@aztec/stdlib/vks';
12
- import { accumulateBlobs, buildBlobHints, buildHeaderFromCircuitOutputs, getEmptyBlockBlobsHash } from './block-building-helpers.js';
4
+ import { getBlockBlobFields } from '@aztec/stdlib/block';
5
+ import { ParityRootPrivateInputs } from '@aztec/stdlib/parity';
6
+ import { BlockRootEmptyTxFirstRollupPrivateInputs, BlockRootFirstRollupPrivateInputs, BlockRootRollupPrivateInputs, BlockRootSingleTxFirstRollupPrivateInputs, BlockRootSingleTxRollupPrivateInputs, TxMergeRollupPrivateInputs } from '@aztec/stdlib/rollup';
7
+ import { GlobalVariables } from '@aztec/stdlib/tx';
8
+ import { buildHeaderFromCircuitOutputs, toProofData } from './block-building-helpers.js';
13
9
  /**
14
10
  * The current state of the proving schedule for a given block. Managed by ProvingState.
15
11
  * Contains the raw inputs and intermediate state to generate every constituent proof in the tree.
16
12
  */ export class BlockProvingState {
17
13
  index;
18
- globalVariables;
19
- newL1ToL2Messages;
20
- l1ToL2MessageTreeSnapshot;
21
- l1ToL2MessageSubtreeSiblingPath;
22
- l1ToL2MessageTreeSnapshotAfterInsertion;
23
- lastArchiveSnapshot;
24
- lastArchiveSiblingPath;
25
- newArchiveSiblingPath;
26
- previousBlockHeader;
27
- proverId;
28
- parentEpoch;
29
- baseOrMergeProvingOutputs;
30
- baseParityProvingOutputs;
31
- rootParityProvingOutput;
32
- blockRootProvingOutput;
33
- blockRootRollupStarted;
34
- block;
35
- spongeBlobState;
36
- startBlobAccumulator;
37
- endBlobAccumulator;
38
- blobsHash;
14
+ blockNumber;
39
15
  totalNumTxs;
16
+ constants;
17
+ timestamp;
18
+ lastArchiveTreeSnapshot;
19
+ lastArchiveSiblingPath;
20
+ lastL1ToL2MessageTreeSnapshot;
21
+ lastL1ToL2MessageSubtreeSiblingPath;
22
+ newL1ToL2MessageTreeSnapshot;
23
+ headerOfLastBlockInPreviousCheckpoint;
24
+ startSpongeBlob;
25
+ parentCheckpoint;
26
+ baseOrMergeProofs;
27
+ baseParityProofs;
28
+ rootParityProof;
29
+ blockRootProof;
30
+ builtBlockHeader;
31
+ endSpongeBlob;
40
32
  txs;
33
+ isFirstBlock;
41
34
  error;
42
- constructor(index, globalVariables, newL1ToL2Messages, l1ToL2MessageTreeSnapshot, l1ToL2MessageSubtreeSiblingPath, l1ToL2MessageTreeSnapshotAfterInsertion, lastArchiveSnapshot, lastArchiveSiblingPath, newArchiveSiblingPath, previousBlockHeader, proverId, parentEpoch){
35
+ constructor(index, blockNumber, totalNumTxs, constants, timestamp, lastArchiveTreeSnapshot, lastArchiveSiblingPath, lastL1ToL2MessageTreeSnapshot, lastL1ToL2MessageSubtreeSiblingPath, newL1ToL2MessageTreeSnapshot, headerOfLastBlockInPreviousCheckpoint, startSpongeBlob, parentCheckpoint){
43
36
  this.index = index;
44
- this.globalVariables = globalVariables;
45
- this.newL1ToL2Messages = newL1ToL2Messages;
46
- this.l1ToL2MessageTreeSnapshot = l1ToL2MessageTreeSnapshot;
47
- this.l1ToL2MessageSubtreeSiblingPath = l1ToL2MessageSubtreeSiblingPath;
48
- this.l1ToL2MessageTreeSnapshotAfterInsertion = l1ToL2MessageTreeSnapshotAfterInsertion;
49
- this.lastArchiveSnapshot = lastArchiveSnapshot;
37
+ this.blockNumber = blockNumber;
38
+ this.totalNumTxs = totalNumTxs;
39
+ this.constants = constants;
40
+ this.timestamp = timestamp;
41
+ this.lastArchiveTreeSnapshot = lastArchiveTreeSnapshot;
50
42
  this.lastArchiveSiblingPath = lastArchiveSiblingPath;
51
- this.newArchiveSiblingPath = newArchiveSiblingPath;
52
- this.previousBlockHeader = previousBlockHeader;
53
- this.proverId = proverId;
54
- this.parentEpoch = parentEpoch;
55
- this.baseOrMergeProvingOutputs = new UnbalancedTreeStore(0);
56
- this.blockRootRollupStarted = false;
57
- this.txs = [];
58
- this.baseParityProvingOutputs = Array.from({
43
+ this.lastL1ToL2MessageTreeSnapshot = lastL1ToL2MessageTreeSnapshot;
44
+ this.lastL1ToL2MessageSubtreeSiblingPath = lastL1ToL2MessageSubtreeSiblingPath;
45
+ this.newL1ToL2MessageTreeSnapshot = newL1ToL2MessageTreeSnapshot;
46
+ this.headerOfLastBlockInPreviousCheckpoint = headerOfLastBlockInPreviousCheckpoint;
47
+ this.startSpongeBlob = startSpongeBlob;
48
+ this.parentCheckpoint = parentCheckpoint;
49
+ this.baseOrMergeProofs = new UnbalancedTreeStore(0);
50
+ this.baseParityProofs = Array.from({
59
51
  length: NUM_BASE_PARITY_PER_ROOT_PARITY
60
52
  }).map((_)=>undefined);
61
- this.totalNumTxs = 0;
62
- if (this.blockNumber == parentEpoch.firstBlockNumber) {
63
- this.startBlobAccumulator = BatchedBlobAccumulator.newWithChallenges(parentEpoch.finalBlobBatchingChallenges);
53
+ this.txs = [];
54
+ this.isFirstBlock = index === 0;
55
+ if (!totalNumTxs && !this.isFirstBlock) {
56
+ throw new Error(`Cannot create a block with 0 txs, unless it's the first block.`);
64
57
  }
58
+ this.baseOrMergeProofs = new UnbalancedTreeStore(totalNumTxs);
65
59
  }
66
- get blockNumber() {
67
- return this.globalVariables.blockNumber;
68
- }
69
- startNewBlock(numTxs, numBlobFields) {
70
- if (this.spongeBlobState) {
71
- throw new Error(`Block ${this.blockNumber} already initalised.`);
72
- }
73
- this.baseOrMergeProvingOutputs = new UnbalancedTreeStore(numTxs);
74
- // Initialize the sponge which will eventually absorb all tx effects to be added to the blob.
75
- // Like l1 to l2 messages, we need to know beforehand how many effects will be absorbed.
76
- this.spongeBlobState = SpongeBlob.init(numBlobFields);
77
- this.totalNumTxs = numTxs;
60
+ get epochNumber() {
61
+ return this.parentCheckpoint.epochNumber;
78
62
  }
79
63
  // Adds a transaction to the proving state, returns it's index
80
64
  addNewTx(tx) {
81
- if (!this.spongeBlobState) {
82
- throw new Error(`Invalid block proving state, call startNewBlock before adding transactions.`);
65
+ if (!this.isAcceptingTxs()) {
66
+ throw new Error(`Cannot add more txs to block ${this.blockNumber}.`);
83
67
  }
84
68
  const txIndex = this.txs.length;
85
69
  this.txs[txIndex] = tx;
86
70
  return txIndex;
87
71
  }
72
+ isAcceptingTxs() {
73
+ return this.txs.length < this.totalNumTxs;
74
+ }
75
+ getProcessedTxs() {
76
+ return this.txs.map((t)=>t.processedTx);
77
+ }
78
+ tryStartProvingBase(txIndex) {
79
+ if (this.baseOrMergeProofs.getLeaf(txIndex)?.isProving) {
80
+ return false;
81
+ } else {
82
+ this.baseOrMergeProofs.setLeaf(txIndex, {
83
+ isProving: true
84
+ });
85
+ return true;
86
+ }
87
+ }
88
88
  setBaseRollupProof(txIndex, provingOutput) {
89
- return this.baseOrMergeProvingOutputs.setLeaf(txIndex, provingOutput);
89
+ return this.baseOrMergeProofs.setLeaf(txIndex, {
90
+ provingOutput
91
+ });
92
+ }
93
+ tryStartProvingMerge(location) {
94
+ if (this.baseOrMergeProofs.getNode(location)?.isProving) {
95
+ return false;
96
+ } else {
97
+ this.baseOrMergeProofs.setNode(location, {
98
+ isProving: true
99
+ });
100
+ return true;
101
+ }
90
102
  }
91
103
  setMergeRollupProof(location, provingOutput) {
92
- this.baseOrMergeProvingOutputs.setNode(location, provingOutput);
104
+ this.baseOrMergeProofs.setNode(location, {
105
+ provingOutput
106
+ });
107
+ }
108
+ tryStartProvingBaseParity(index) {
109
+ if (this.baseParityProofs[index]?.isProving) {
110
+ return false;
111
+ } else {
112
+ this.baseParityProofs[index] = {
113
+ isProving: true
114
+ };
115
+ return true;
116
+ }
93
117
  }
94
118
  // Stores a set of root parity inputs at the given index
95
119
  setBaseParityProof(index, provingOutput) {
96
120
  if (index >= NUM_BASE_PARITY_PER_ROOT_PARITY) {
97
121
  throw new Error(`Unable to set a base parity proofs at index ${index}. Expected at most ${NUM_BASE_PARITY_PER_ROOT_PARITY} proofs.`);
98
122
  }
99
- this.baseParityProvingOutputs[index] = provingOutput;
123
+ this.baseParityProofs[index] = {
124
+ provingOutput
125
+ };
126
+ }
127
+ tryStartProvingRootParity() {
128
+ if (this.rootParityProof?.isProving) {
129
+ return false;
130
+ } else {
131
+ this.rootParityProof = {
132
+ isProving: true
133
+ };
134
+ return true;
135
+ }
100
136
  }
101
137
  setRootParityProof(provingOutput) {
102
- this.rootParityProvingOutput = provingOutput;
138
+ this.rootParityProof = {
139
+ provingOutput
140
+ };
141
+ }
142
+ tryStartProvingBlockRoot() {
143
+ if (this.blockRootProof?.isProving) {
144
+ return false;
145
+ } else {
146
+ this.blockRootProof = {
147
+ isProving: true
148
+ };
149
+ return true;
150
+ }
103
151
  }
104
152
  setBlockRootRollupProof(provingOutput) {
105
- this.blockRootProvingOutput = provingOutput;
153
+ this.blockRootProof = {
154
+ provingOutput
155
+ };
156
+ return this.parentCheckpoint.setBlockRootRollupProof(this.index, provingOutput);
106
157
  }
107
- setBlock(block) {
108
- this.block = block;
158
+ getBlockRootRollupOutput() {
159
+ return this.blockRootProof?.provingOutput?.inputs;
109
160
  }
110
- setStartBlobAccumulator(accumulator) {
111
- this.startBlobAccumulator = accumulator;
161
+ setBuiltBlockHeader(blockHeader) {
162
+ this.builtBlockHeader = blockHeader;
112
163
  }
113
- setEndBlobAccumulator(accumulator) {
114
- this.endBlobAccumulator = accumulator;
164
+ getBuiltBlockHeader() {
165
+ return this.builtBlockHeader;
115
166
  }
116
- async accumulateBlobs() {
117
- if (!this.block || !this.startBlobAccumulator) {
118
- // We only want to accumulate once we have all txs, so we wait until the block is set.
119
- return;
167
+ getGlobalVariables() {
168
+ if (this.txs.length) {
169
+ return this.txs[0].processedTx.globalVariables;
120
170
  }
121
- const endBlobAccumulator = await accumulateBlobs(this.allTxs.map((t)=>t.processedTx), this.startBlobAccumulator);
122
- this.setEndBlobAccumulator(endBlobAccumulator);
171
+ const constants = this.constants;
172
+ return GlobalVariables.from({
173
+ chainId: constants.chainId,
174
+ version: constants.version,
175
+ blockNumber: this.blockNumber,
176
+ slotNumber: constants.slotNumber,
177
+ timestamp: this.timestamp,
178
+ coinbase: constants.coinbase,
179
+ feeRecipient: constants.feeRecipient,
180
+ gasFees: constants.gasFees
181
+ });
182
+ }
183
+ getStartSpongeBlob() {
184
+ return this.startSpongeBlob;
123
185
  }
124
- // Returns the complete set of transaction proving state objects
125
- get allTxs() {
126
- return this.txs;
186
+ setEndSpongeBlob(endSpongeBlob) {
187
+ this.endSpongeBlob = endSpongeBlob;
127
188
  }
128
- /** Returns the block number as an epoch number. Used for prioritizing proof requests. */ get epochNumber() {
129
- return this.parentEpoch.epochNumber;
189
+ getEndSpongeBlob() {
190
+ return this.endSpongeBlob;
191
+ }
192
+ getBlockBlobFields() {
193
+ return getBlockBlobFields(this.txs.map((t)=>t.processedTx.txEffect));
130
194
  }
131
195
  getParentLocation(location) {
132
- return this.baseOrMergeProvingOutputs.getParentLocation(location);
196
+ return this.baseOrMergeProofs.getParentLocation(location);
133
197
  }
134
198
  getMergeRollupInputs(mergeLocation) {
135
- const [left, right] = this.baseOrMergeProvingOutputs.getChildren(mergeLocation);
199
+ const [left, right] = this.baseOrMergeProofs.getChildren(mergeLocation).map((c)=>c?.provingOutput);
136
200
  if (!left || !right) {
137
- throw new Error('At lease one child is not ready.');
201
+ throw new Error('At least one child is not ready for the merge rollup.');
138
202
  }
139
- return new MergeRollupInputs([
140
- this.#getPreviousRollupData(left),
141
- this.#getPreviousRollupData(right)
203
+ return new TxMergeRollupPrivateInputs([
204
+ toProofData(left),
205
+ toProofData(right)
142
206
  ]);
143
207
  }
144
- async getBlockRootRollupTypeAndInputs() {
145
- if (!this.rootParityProvingOutput) {
146
- throw new Error('Root parity is not ready.');
208
+ getBlockRootRollupTypeAndInputs() {
209
+ const provingOutputs = this.#getChildProvingOutputsForBlockRoot();
210
+ if (!provingOutputs.every((p)=>!!p)) {
211
+ throw new Error('At least one child is not ready for the block root rollup.');
147
212
  }
148
- const proofs = this.#getChildProofsForBlockRoot();
149
- const nonEmptyProofs = proofs.filter((p)=>!!p);
150
- if (proofs.length !== nonEmptyProofs.length) {
151
- throw new Error('At lease one child is not ready for the block root.');
213
+ const previousRollups = provingOutputs.map((p)=>toProofData(p));
214
+ if (this.isFirstBlock) {
215
+ return this.#getFirstBlockRootRollupTypeAndInputs(previousRollups);
152
216
  }
153
- const data = this.#getBlockRootRollupData();
154
- if (this.totalNumTxs === 0) {
155
- const constants = BlockConstantData.from({
156
- lastArchive: this.lastArchiveSnapshot,
157
- newL1ToL2: this.l1ToL2MessageTreeSnapshotAfterInsertion,
158
- globalVariables: this.globalVariables,
159
- vkTreeRoot: getVKTreeRoot(),
160
- protocolContractTreeRoot
161
- });
162
- this.blobsHash = await getEmptyBlockBlobsHash();
217
+ const [leftRollup, rightRollup] = previousRollups;
218
+ if (!rightRollup) {
219
+ return {
220
+ rollupType: 'rollup-block-root-single-tx',
221
+ inputs: new BlockRootSingleTxRollupPrivateInputs(leftRollup, this.lastArchiveSiblingPath)
222
+ };
223
+ } else {
163
224
  return {
164
- rollupType: 'empty-block-root-rollup',
165
- inputs: EmptyBlockRootRollupInputs.from({
166
- data,
167
- constants
168
- })
225
+ rollupType: 'rollup-block-root',
226
+ inputs: new BlockRootRollupPrivateInputs([
227
+ leftRollup,
228
+ rightRollup
229
+ ], this.lastArchiveSiblingPath)
169
230
  };
170
231
  }
171
- const previousRollupData = await Promise.all(nonEmptyProofs.map((p)=>this.#getPreviousRollupData(p)));
172
- const blobData = await this.#getBlockRootRollupBlobData();
173
- this.blobsHash = blobData.blobsHash;
174
- if (previousRollupData.length === 1) {
232
+ }
233
+ #getFirstBlockRootRollupTypeAndInputs([leftRollup, rightRollup]) {
234
+ if (!this.rootParityProof?.provingOutput) {
235
+ throw new Error('Root parity is not ready.');
236
+ }
237
+ const l1ToL2Roots = toProofData(this.rootParityProof.provingOutput);
238
+ if (!leftRollup) {
175
239
  return {
176
- rollupType: 'single-tx-block-root-rollup',
177
- inputs: new SingleTxBlockRootRollupInputs(previousRollupData, data, blobData)
240
+ rollupType: 'rollup-block-root-first-empty-tx',
241
+ inputs: new BlockRootEmptyTxFirstRollupPrivateInputs(l1ToL2Roots, this.lastArchiveTreeSnapshot, this.headerOfLastBlockInPreviousCheckpoint.state, this.constants, this.startSpongeBlob, this.timestamp, this.lastL1ToL2MessageSubtreeSiblingPath, this.lastArchiveSiblingPath)
242
+ };
243
+ } else if (!rightRollup) {
244
+ return {
245
+ rollupType: 'rollup-block-root-first-single-tx',
246
+ inputs: new BlockRootSingleTxFirstRollupPrivateInputs(l1ToL2Roots, leftRollup, this.lastL1ToL2MessageTreeSnapshot, this.lastL1ToL2MessageSubtreeSiblingPath, this.lastArchiveSiblingPath)
178
247
  };
179
248
  } else {
180
249
  return {
181
- rollupType: 'block-root-rollup',
182
- inputs: new BlockRootRollupInputs(previousRollupData, data, blobData)
250
+ rollupType: 'rollup-block-root-first',
251
+ inputs: new BlockRootFirstRollupPrivateInputs(l1ToL2Roots, [
252
+ leftRollup,
253
+ rightRollup
254
+ ], this.lastL1ToL2MessageTreeSnapshot, this.lastL1ToL2MessageSubtreeSiblingPath, this.lastArchiveSiblingPath)
183
255
  };
184
256
  }
185
257
  }
186
- getPaddingBlockRootInputs() {
187
- const constants = EpochConstantData.from({
188
- vkTreeRoot: getVKTreeRoot(),
189
- protocolContractTreeRoot,
190
- proverId: this.proverId.toField()
191
- });
192
- return PaddingBlockRootRollupInputs.from({
193
- constants
194
- });
195
- }
196
- getRootParityInputs() {
197
- if (!this.baseParityProvingOutputs.every((p)=>!!p)) {
258
+ getParityRootInputs() {
259
+ const baseParityProvingOutputs = this.baseParityProofs.filter((p)=>!!p?.provingOutput).map((p)=>p.provingOutput);
260
+ if (baseParityProvingOutputs.length !== this.baseParityProofs.length) {
198
261
  throw new Error('At lease one base parity is not ready.');
199
262
  }
200
- const children = this.baseParityProvingOutputs.map((p)=>this.#getRootParityData(p));
201
- return new RootParityInputs(children);
263
+ const children = baseParityProvingOutputs.map((p)=>toProofData(p));
264
+ return new ParityRootPrivateInputs(assertLength(children, NUM_BASE_PARITY_PER_ROOT_PARITY));
202
265
  }
203
266
  // Returns a specific transaction proving state
204
267
  getTxProvingState(txIndex) {
205
268
  return this.txs[txIndex];
206
269
  }
207
270
  async buildHeaderFromProvingOutputs() {
208
- const previousRollupData = this.totalNumTxs === 0 ? [] : await Promise.all(this.#getChildProofsForBlockRoot().map((p)=>this.#getPreviousRollupData(p)));
209
- let endPartialState = this.previousBlockHeader.state.partial;
210
- if (this.totalNumTxs !== 0) {
211
- const previousRollupData = this.#getChildProofsForBlockRoot();
212
- const lastRollup = previousRollupData[previousRollupData.length - 1];
213
- if (!lastRollup) {
214
- throw new Error('End state of the block is not available. Last rollup is not ready yet.');
215
- }
216
- endPartialState = lastRollup.inputs.end;
271
+ if (!this.blockRootProof?.provingOutput) {
272
+ throw new Error('Block root rollup is not ready.');
217
273
  }
218
- const endState = new StateReference(this.l1ToL2MessageTreeSnapshotAfterInsertion, endPartialState);
219
- return buildHeaderFromCircuitOutputs(previousRollupData.map((d)=>d.baseOrMergeRollupPublicInputs), this.rootParityProvingOutput.inputs, this.blockRootProvingOutput.inputs, this.blobsHash, endState);
274
+ return await buildHeaderFromCircuitOutputs(this.blockRootProof.provingOutput.inputs);
220
275
  }
221
276
  isReadyForMergeRollup(location) {
222
- return this.baseOrMergeProvingOutputs.getSibling(location) !== undefined;
277
+ return !!this.baseOrMergeProofs.getSibling(location)?.provingOutput;
223
278
  }
224
279
  // Returns true if we have sufficient inputs to execute the block root rollup
225
280
  isReadyForBlockRootRollup() {
226
- const childProofs = this.#getChildProofsForBlockRoot();
227
- return this.block !== undefined && this.rootParityProvingOutput !== undefined && this.endBlobAccumulator !== undefined && childProofs.every((p)=>!!p);
281
+ const childProofs = this.#getChildProvingOutputsForBlockRoot();
282
+ return (!this.isFirstBlock || !!this.rootParityProof?.provingOutput) && childProofs.every((p)=>!!p);
228
283
  }
229
284
  // Returns true if we have sufficient root parity inputs to execute the root parity circuit
230
285
  isReadyForRootParity() {
231
- return this.baseParityProvingOutputs.every((p)=>!!p);
286
+ return this.baseParityProofs.every((p)=>!!p?.provingOutput);
232
287
  }
233
288
  isComplete() {
234
- return !!this.blockRootProvingOutput;
289
+ return !!this.blockRootProof;
235
290
  }
236
- // Returns whether the proving state is still valid
237
291
  verifyState() {
238
- return this.parentEpoch.verifyState();
292
+ return this.parentCheckpoint.verifyState();
293
+ }
294
+ getError() {
295
+ return this.error;
239
296
  }
240
297
  reject(reason) {
241
298
  this.error = reason;
242
- this.parentEpoch.reject(reason);
243
- }
244
- #getBlockRootRollupData() {
245
- return BlockRootRollupData.from({
246
- l1ToL2Roots: this.#getRootParityData(this.rootParityProvingOutput),
247
- l1ToL2MessageSubtreeSiblingPath: this.l1ToL2MessageSubtreeSiblingPath,
248
- previousArchiveSiblingPath: this.lastArchiveSiblingPath,
249
- newArchiveSiblingPath: this.newArchiveSiblingPath,
250
- previousBlockHeader: this.previousBlockHeader,
251
- startBlobAccumulator: BlobAccumulatorPublicInputs.fromBatchedBlobAccumulator(this.startBlobAccumulator),
252
- finalBlobChallenges: this.startBlobAccumulator.finalBlobChallenges,
253
- proverId: this.proverId.toField()
254
- });
299
+ this.parentCheckpoint.reject(reason);
255
300
  }
256
- async #getBlockRootRollupBlobData() {
257
- const txEffects = this.txs.map((txProvingState)=>txProvingState.processedTx.txEffect);
258
- const { blobFields, blobCommitments, blobsHash } = await buildBlobHints(txEffects);
259
- return BlockRootRollupBlobData.from({
260
- blobFields: padArrayEnd(blobFields, Fr.ZERO, FIELDS_PER_BLOB * BLOBS_PER_BLOCK),
261
- blobCommitments: padArrayEnd(blobCommitments, BLS12Point.ZERO, BLOBS_PER_BLOCK),
262
- blobsHash
263
- });
264
- }
265
- #getChildProofsForBlockRoot() {
301
+ #getChildProvingOutputsForBlockRoot() {
266
302
  if (this.totalNumTxs === 0) {
267
303
  return [];
268
304
  }
@@ -272,15 +308,7 @@ import { accumulateBlobs, buildBlobHints, buildHeaderFromCircuitOutputs, getEmpt
272
308
  };
273
309
  // If there's only 1 tx, its base rollup proof will be stored at the root.
274
310
  return this.totalNumTxs === 1 ? [
275
- this.baseOrMergeProvingOutputs.getNode(rootLocation)
276
- ] : this.baseOrMergeProvingOutputs.getChildren(rootLocation);
277
- }
278
- #getPreviousRollupData({ inputs, proof, verificationKey }) {
279
- const leafIndex = getVKIndex(verificationKey.keyAsFields);
280
- const vkData = new VkData(verificationKey, leafIndex, getVKSiblingPath(leafIndex));
281
- return new PreviousRollupData(inputs, proof, vkData);
282
- }
283
- #getRootParityData({ inputs, proof, verificationKey }) {
284
- return new RootParityInput(proof, verificationKey.keyAsFields, getVKSiblingPath(getVKIndex(verificationKey)), inputs);
311
+ this.baseOrMergeProofs.getNode(rootLocation)?.provingOutput
312
+ ] : this.baseOrMergeProofs.getChildren(rootLocation).map((c)=>c?.provingOutput);
285
313
  }
286
314
  }
@@ -0,0 +1,62 @@
1
+ import { BatchedBlobAccumulator, type FinalBlobBatchingChallenges } from '@aztec/blob-lib';
2
+ import { type ARCHIVE_HEIGHT, type L1_TO_L2_MSG_SUBTREE_SIBLING_PATH_LENGTH, type NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH } from '@aztec/constants';
3
+ import { Fr } from '@aztec/foundation/fields';
4
+ import type { Tuple } from '@aztec/foundation/serialize';
5
+ import { type TreeNodeLocation } from '@aztec/foundation/trees';
6
+ import type { PublicInputsAndRecursiveProof } from '@aztec/stdlib/interfaces/server';
7
+ import { ParityBasePrivateInputs } from '@aztec/stdlib/parity';
8
+ import { BlockMergeRollupPrivateInputs, BlockRollupPublicInputs, CheckpointConstantData, CheckpointRollupPublicInputs, CheckpointRootRollupPrivateInputs, CheckpointRootSingleBlockRollupPrivateInputs } from '@aztec/stdlib/rollup';
9
+ import type { CircuitName } from '@aztec/stdlib/stats';
10
+ import type { AppendOnlyTreeSnapshot } from '@aztec/stdlib/trees';
11
+ import type { BlockHeader } from '@aztec/stdlib/tx';
12
+ import type { UInt64 } from '@aztec/stdlib/types';
13
+ import { BlockProvingState } from './block-proving-state.js';
14
+ import type { EpochProvingState } from './epoch-proving-state.js';
15
+ export declare class CheckpointProvingState {
16
+ #private;
17
+ readonly index: number;
18
+ readonly constants: CheckpointConstantData;
19
+ readonly totalNumBlocks: number;
20
+ private readonly totalNumBlobFields;
21
+ private readonly finalBlobBatchingChallenges;
22
+ private readonly headerOfLastBlockInPreviousCheckpoint;
23
+ private readonly lastArchiveSiblingPath;
24
+ private readonly l1ToL2Messages;
25
+ private readonly lastL1ToL2MessageTreeSnapshot;
26
+ private readonly lastL1ToL2MessageSubtreeSiblingPath;
27
+ private readonly newL1ToL2MessageTreeSnapshot;
28
+ private readonly newL1ToL2MessageSubtreeSiblingPath;
29
+ parentEpoch: EpochProvingState;
30
+ private onBlobAccumulatorSet;
31
+ private blockProofs;
32
+ private checkpointRootProof;
33
+ private blocks;
34
+ private startBlobAccumulator;
35
+ private endBlobAccumulator;
36
+ private error;
37
+ readonly firstBlockNumber: number;
38
+ constructor(index: number, constants: CheckpointConstantData, totalNumBlocks: number, totalNumBlobFields: number, finalBlobBatchingChallenges: FinalBlobBatchingChallenges, headerOfLastBlockInPreviousCheckpoint: BlockHeader, lastArchiveSiblingPath: Tuple<Fr, typeof ARCHIVE_HEIGHT>, l1ToL2Messages: Fr[], lastL1ToL2MessageTreeSnapshot: AppendOnlyTreeSnapshot, lastL1ToL2MessageSubtreeSiblingPath: Tuple<Fr, typeof L1_TO_L2_MSG_SUBTREE_SIBLING_PATH_LENGTH>, newL1ToL2MessageTreeSnapshot: AppendOnlyTreeSnapshot, newL1ToL2MessageSubtreeSiblingPath: Tuple<Fr, typeof L1_TO_L2_MSG_SUBTREE_SIBLING_PATH_LENGTH>, parentEpoch: EpochProvingState, onBlobAccumulatorSet: (checkpoint: CheckpointProvingState) => void);
39
+ get epochNumber(): number;
40
+ startNewBlock(blockNumber: number, timestamp: UInt64, totalNumTxs: number, lastArchiveTreeSnapshot: AppendOnlyTreeSnapshot, lastArchiveSiblingPath: Tuple<Fr, typeof ARCHIVE_HEIGHT>): BlockProvingState;
41
+ isAcceptingBlocks(): boolean;
42
+ setBlockRootRollupProof(blockIndex: number, provingOutput: PublicInputsAndRecursiveProof<BlockRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>): TreeNodeLocation;
43
+ tryStartProvingBlockMerge(location: TreeNodeLocation): boolean;
44
+ setBlockMergeRollupProof(location: TreeNodeLocation, provingOutput: PublicInputsAndRecursiveProof<BlockRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>): void;
45
+ tryStartProvingCheckpointRoot(): boolean;
46
+ setCheckpointRootRollupProof(provingOutput: PublicInputsAndRecursiveProof<CheckpointRollupPublicInputs, typeof NESTED_RECURSIVE_ROLLUP_HONK_PROOF_LENGTH>): TreeNodeLocation;
47
+ getBaseParityInputs(baseParityIndex: number): ParityBasePrivateInputs;
48
+ accumulateBlobs(startBlobAccumulator: BatchedBlobAccumulator): Promise<BatchedBlobAccumulator | undefined>;
49
+ getEndBlobAccumulator(): BatchedBlobAccumulator | undefined;
50
+ getParentLocation(location: TreeNodeLocation): TreeNodeLocation;
51
+ getBlockMergeRollupInputs(mergeLocation: TreeNodeLocation): BlockMergeRollupPrivateInputs;
52
+ getCheckpointRootRollupType(): CircuitName;
53
+ getCheckpointRootRollupInputs(): Promise<CheckpointRootSingleBlockRollupPrivateInputs | CheckpointRootRollupPrivateInputs>;
54
+ getBlockProvingStateByBlockNumber(blockNumber: number): BlockProvingState | undefined;
55
+ isReadyForBlockMerge(location: TreeNodeLocation): boolean;
56
+ isReadyForCheckpointRoot(): boolean;
57
+ verifyState(): boolean;
58
+ getError(): string | undefined;
59
+ cancel(): void;
60
+ reject(reason: string): void;
61
+ }
62
+ //# sourceMappingURL=checkpoint-proving-state.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"checkpoint-proving-state.d.ts","sourceRoot":"","sources":["../../src/orchestrator/checkpoint-proving-state.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,EAEtB,KAAK,2BAA2B,EAEjC,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,KAAK,cAAc,EAGnB,KAAK,wCAAwC,EAC7C,KAAK,yCAAyC,EAE/C,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAc,EAAE,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,6BAA6B,CAAC;AACzD,OAAO,EAAE,KAAK,gBAAgB,EAAuB,MAAM,yBAAyB,CAAC;AACrF,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,iCAAiC,CAAC;AACrF,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EACL,6BAA6B,EAC7B,uBAAuB,EACvB,sBAAsB,EACtB,4BAA4B,EAE5B,iCAAiC,EACjC,4CAA4C,EAC7C,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAClE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAGlD,OAAO,EAAE,iBAAiB,EAAmB,MAAM,0BAA0B,CAAC;AAC9E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAElE,qBAAa,sBAAsB;;aAcf,KAAK,EAAE,MAAM;aACb,SAAS,EAAE,sBAAsB;aACjC,cAAc,EAAE,MAAM;IACtC,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IACnC,OAAO,CAAC,QAAQ,CAAC,2BAA2B;IAC5C,OAAO,CAAC,QAAQ,CAAC,qCAAqC;IACtD,OAAO,CAAC,QAAQ,CAAC,sBAAsB;IACvC,OAAO,CAAC,QAAQ,CAAC,cAAc;IAE/B,OAAO,CAAC,QAAQ,CAAC,6BAA6B;IAC9C,OAAO,CAAC,QAAQ,CAAC,mCAAmC;IAEpD,OAAO,CAAC,QAAQ,CAAC,4BAA4B;IAC7C,OAAO,CAAC,QAAQ,CAAC,kCAAkC;IAC5C,WAAW,EAAE,iBAAiB;IACrC,OAAO,CAAC,oBAAoB;IA5B9B,OAAO,CAAC,WAAW,CAEjB;IACF,OAAO,CAAC,mBAAmB,CAEb;IACd,OAAO,CAAC,MAAM,CAAyC;IACvD,OAAO,CAAC,oBAAoB,CAAqC;IACjE,OAAO,CAAC,kBAAkB,CAAqC;IAC/D,OAAO,CAAC,KAAK,CAAqB;IAClC,SAAgB,gBAAgB,EAAE,MAAM,CAAC;gBAGvB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,sBAAsB,EACjC,cAAc,EAAE,MAAM,EACrB,kBAAkB,EAAE,MAAM,EAC1B,2BAA2B,EAAE,2BAA2B,EACxD,qCAAqC,EAAE,WAAW,EAClD,sBAAsB,EAAE,KAAK,CAAC,EAAE,EAAE,OAAO,cAAc,CAAC,EACxD,cAAc,EAAE,EAAE,EAAE,EAEpB,6BAA6B,EAAE,sBAAsB,EACrD,mCAAmC,EAAE,KAAK,CAAC,EAAE,EAAE,OAAO,wCAAwC,CAAC,EAE/F,4BAA4B,EAAE,sBAAsB,EACpD,kCAAkC,EAAE,KAAK,CAAC,EAAE,EAAE,OAAO,wCAAwC,CAAC,EACxG,WAAW,EAAE,iBAAiB,EAC7B,oBAAoB,EAAE,CAAC,UAAU,EAAE,sBAAsB,KAAK,IAAI;IAM5E,IAAW,WAAW,IAAI,MAAM,CAE/B;IAEM,aAAa,CAClB,WAAW,EAAE,MAAM,EACnB,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,EACnB,uBAAuB,EAAE,sBAAsB,EAC/C,sBAAsB,EAAE,KAAK,CAAC,EAAE,EAAE,OAAO,cAAc,CAAC,GACvD,iBAAiB;IA2Cb,iBAAiB;IAIjB,uBAAuB,CAC5B,UAAU,EAAE,MAAM,EAClB,aAAa,EAAE,6BAA6B,CAC1C,uBAAuB,EACvB,OAAO,yCAAyC,CACjD,GACA,gBAAgB;IAIZ,yBAAyB,CAAC,QAAQ,EAAE,gBAAgB;IASpD,wBAAwB,CAC7B,QAAQ,EAAE,gBAAgB,EAC1B,aAAa,EAAE,6BAA6B,CAC1C,uBAAuB,EACvB,OAAO,yCAAyC,CACjD;IAKI,6BAA6B;IAS7B,4BAA4B,CACjC,aAAa,EAAE,6BAA6B,CAC1C,4BAA4B,EAC5B,OAAO,yCAAyC,CACjD,GACA,gBAAgB;IAKZ,mBAAmB,CAAC,eAAe,EAAE,MAAM;IAYrC,eAAe,CAAC,oBAAoB,EAAE,sBAAsB;IAclE,qBAAqB;IAIrB,iBAAiB,CAAC,QAAQ,EAAE,gBAAgB;IAI5C,yBAAyB,CAAC,aAAa,EAAE,gBAAgB;IASzD,2BAA2B,IAAI,WAAW;IAIpC,6BAA6B;IA8BnC,iCAAiC,CAAC,WAAW,EAAE,MAAM;IAKrD,oBAAoB,CAAC,QAAQ,EAAE,gBAAgB;IAI/C,wBAAwB;IAKxB,WAAW;IAIX,QAAQ;IAKR,MAAM;IAIN,MAAM,CAAC,MAAM,EAAE,MAAM;CAW7B"}