@aztec/archiver 0.0.0-test.1 → 0.0.1-commit.1142ef1

package/dest/archiver/kv_archiver_store/block_store.js

@@ -1,98 +1,393 @@
-import { INITIAL_L2_BLOCK_NUM } from '@aztec/constants';
+import { INITIAL_CHECKPOINT_NUMBER, INITIAL_L2_BLOCK_NUM } from '@aztec/constants';
+import { BlockNumber, CheckpointNumber } from '@aztec/foundation/branded-types';
 import { toArray } from '@aztec/foundation/iterable';
 import { createLogger } from '@aztec/foundation/log';
-import { Body, L2Block, L2BlockHash } from '@aztec/stdlib/block';
+import { BufferReader } from '@aztec/foundation/serialize';
+import { bufferToHex } from '@aztec/foundation/string';
+import { isDefined } from '@aztec/foundation/types';
+import { Body, CheckpointedL2Block, CommitteeAttestation, L2BlockHash, L2BlockNew, deserializeValidateCheckpointResult, serializeValidateCheckpointResult } from '@aztec/stdlib/block';
+import { L1PublishedData } from '@aztec/stdlib/checkpoint';
+import { CheckpointHeader } from '@aztec/stdlib/rollup';
 import { AppendOnlyTreeSnapshot } from '@aztec/stdlib/trees';
-import { BlockHeader, TxReceipt } from '@aztec/stdlib/tx';
+import { BlockHeader, TxHash, TxReceipt, deserializeIndexedTxEffect, serializeIndexedTxEffect } from '@aztec/stdlib/tx';
+import { BlockArchiveNotConsistentError, BlockIndexNotSequentialError, BlockNotFoundError, BlockNumberNotSequentialError, CheckpointNotFoundError, CheckpointNumberNotConsistentError, CheckpointNumberNotSequentialError, InitialBlockNumberNotSequentialError, InitialCheckpointNumberNotSequentialError } from '../errors.js';
 export { TxReceipt } from '@aztec/stdlib/tx';
 /**
  * LMDB implementation of the ArchiverDataStore interface.
  */ export class BlockStore {
     db;
     /** Map block number to block data */ #blocks;
-    /** Map block hash to block body */ #blockBodies;
+    /** Map checkpoint number to checkpoint data */ #checkpoints;
+    /** Map block hash to list of tx hashes */ #blockTxs;
+    /** Tx hash to serialized IndexedTxEffect */ #txEffects;
     /** Stores L1 block number in which the last processed L2 block was included */ #lastSynchedL1Block;
-    /** Stores l2 block number of the last proven block */ #lastProvenL2Block;
-    /** Stores l2 epoch number of the last proven epoch */ #lastProvenL2Epoch;
-    /** Index mapping transaction hash (as a string) to its location in a block */ #txIndex;
+    /** Stores last proven checkpoint */ #lastProvenCheckpoint;
+    /** Stores the pending chain validation status */ #pendingChainValidationStatus;
     /** Index mapping a contract's address (as a string) to its location in a block */ #contractIndex;
+    /** Index mapping block hash to block number */ #blockHashIndex;
+    /** Index mapping block archive to block number */ #blockArchiveIndex;
     #log;
     constructor(db){
         this.db = db;
         this.#log = createLogger('archiver:block_store');
         this.#blocks = db.openMap('archiver_blocks');
-        this.#blockBodies = db.openMap('archiver_block_bodies');
-        this.#txIndex = db.openMap('archiver_tx_index');
+        this.#blockTxs = db.openMap('archiver_block_txs');
+        this.#txEffects = db.openMap('archiver_tx_effects');
         this.#contractIndex = db.openMap('archiver_contract_index');
+        this.#blockHashIndex = db.openMap('archiver_block_hash_index');
+        this.#blockArchiveIndex = db.openMap('archiver_block_archive_index');
         this.#lastSynchedL1Block = db.openSingleton('archiver_last_synched_l1_block');
-        this.#lastProvenL2Block = db.openSingleton('archiver_last_proven_l2_block');
-        this.#lastProvenL2Epoch = db.openSingleton('archiver_last_proven_l2_epoch');
+        this.#lastProvenCheckpoint = db.openSingleton('archiver_last_proven_l2_checkpoint');
+        this.#pendingChainValidationStatus = db.openSingleton('archiver_pending_chain_validation_status');
+        this.#checkpoints = db.openMap('archiver_checkpoints');
     }
     /**
-   * Append new blocks to the store's list.
+   * Append new blocks to the store's list. All blocks must be for the 'current' checkpoint
    * @param blocks - The L2 blocks to be added to the store.
    * @returns True if the operation is successful.
-   */ async addBlocks(blocks) {
+   */ async addBlocks(blocks, opts = {}) {
         if (blocks.length === 0) {
             return true;
         }
         return await this.db.transactionAsync(async ()=>{
+            // Check that the block immediately before the first block to be added is present in the store.
+            const firstBlockNumber = blocks[0].number;
+            const firstBlockCheckpointNumber = blocks[0].checkpointNumber;
+            const firstBlockIndex = blocks[0].indexWithinCheckpoint;
+            const firstBlockLastArchive = blocks[0].header.lastArchive.root;
+            // Extract the latest block and checkpoint numbers
+            const previousBlockNumber = await this.getLatestBlockNumber();
+            const previousCheckpointNumber = await this.getLatestCheckpointNumber();
+            // Check that the first block number is the expected one
+            if (!opts.force && previousBlockNumber !== firstBlockNumber - 1) {
+                throw new InitialBlockNumberNotSequentialError(firstBlockNumber, previousBlockNumber);
+            }
+            // The same check as above but for checkpoints
+            if (!opts.force && previousCheckpointNumber !== firstBlockCheckpointNumber - 1) {
+                throw new InitialCheckpointNumberNotSequentialError(firstBlockCheckpointNumber, previousCheckpointNumber);
+            }
+            // Extract the previous block if there is one and see if it is for the same checkpoint or not
+            const previousBlockResult = await this.getBlock(previousBlockNumber);
+            let expectedFirstblockIndex = 0;
+            let previousBlockIndex = undefined;
+            if (previousBlockResult !== undefined) {
+                if (previousBlockResult.checkpointNumber === firstBlockCheckpointNumber) {
+                    // The previous block is for the same checkpoint, therefore our index should follow it
+                    previousBlockIndex = previousBlockResult.indexWithinCheckpoint;
+                    expectedFirstblockIndex = previousBlockIndex + 1;
+                }
+                if (!previousBlockResult.archive.root.equals(firstBlockLastArchive)) {
+                    throw new BlockArchiveNotConsistentError(firstBlockNumber, previousBlockResult.number, firstBlockLastArchive, previousBlockResult.archive.root);
+                }
+            }
+            // Now check that the first block has the expected index value
+            if (!opts.force && expectedFirstblockIndex !== firstBlockIndex) {
+                throw new BlockIndexNotSequentialError(firstBlockIndex, previousBlockIndex);
+            }
+            // Iterate over blocks array and insert them, checking that the block numbers and indexes are sequential. Also check they are for the correct checkpoint.
+            let previousBlock = undefined;
             for (const block of blocks){
-                await this.#blocks.set(block.data.number, {
-                    header: block.data.header.toBuffer(),
-                    archive: block.data.archive.toBuffer(),
-                    l1: block.l1
-                });
-                for(let i = 0; i < block.data.body.txEffects.length; i++){
-                    const txEffect = block.data.body.txEffects[i];
-                    await this.#txIndex.set(txEffect.txHash.toString(), [
-                        block.data.number,
-                        i
-                    ]);
+                if (!opts.force && previousBlock) {
+                    if (previousBlock.number + 1 !== block.number) {
+                        throw new BlockNumberNotSequentialError(block.number, previousBlock.number);
+                    }
+                    if (previousBlock.indexWithinCheckpoint + 1 !== block.indexWithinCheckpoint) {
+                        throw new BlockIndexNotSequentialError(block.indexWithinCheckpoint, previousBlock.indexWithinCheckpoint);
+                    }
+                    if (!previousBlock.archive.root.equals(block.header.lastArchive.root)) {
+                        throw new BlockArchiveNotConsistentError(block.number, previousBlock.number, block.header.lastArchive.root, previousBlock.archive.root);
+                    }
                 }
-                await this.#blockBodies.set((await block.data.hash()).toString(), block.data.body.toBuffer());
+                if (!opts.force && firstBlockCheckpointNumber !== block.checkpointNumber) {
+                    throw new CheckpointNumberNotConsistentError(block.checkpointNumber, firstBlockCheckpointNumber);
+                }
+                previousBlock = block;
+                await this.addBlockToDatabase(block, block.checkpointNumber, block.indexWithinCheckpoint);
             }
-            await this.#lastSynchedL1Block.set(blocks[blocks.length - 1].l1.blockNumber);
             return true;
         });
     }
     /**
-   * Unwinds blocks from the database
+   * Append new cheskpoints to the store's list.
+   * @param checkpoints - The L2 checkpoints to be added to the store.
+   * @returns True if the operation is successful.
+   */ async addCheckpoints(checkpoints, opts = {}) {
+        if (checkpoints.length === 0) {
+            return true;
+        }
+        return await this.db.transactionAsync(async ()=>{
+            // Check that the checkpoint immediately before the first block to be added is present in the store.
+            const firstCheckpointNumber = checkpoints[0].checkpoint.number;
+            const previousCheckpointNumber = await this.getLatestCheckpointNumber();
+            if (previousCheckpointNumber !== firstCheckpointNumber - 1 && !opts.force) {
+                throw new InitialCheckpointNumberNotSequentialError(firstCheckpointNumber, previousCheckpointNumber);
+            }
+            // Extract the previous checkpoint if there is one
+            let previousCheckpointData = undefined;
+            if (previousCheckpointNumber !== INITIAL_CHECKPOINT_NUMBER - 1) {
+                // There should be a previous checkpoint
+                previousCheckpointData = await this.getCheckpointData(previousCheckpointNumber);
+                if (previousCheckpointData === undefined) {
+                    throw new CheckpointNotFoundError(previousCheckpointNumber);
+                }
+            }
+            let previousBlockNumber = undefined;
+            let previousBlock = undefined;
+            // If we have a previous checkpoint then we need to get the previous block number
+            if (previousCheckpointData !== undefined) {
+                previousBlockNumber = BlockNumber(previousCheckpointData.startBlock + previousCheckpointData.numBlocks - 1);
+                previousBlock = await this.getBlock(previousBlockNumber);
+                if (previousBlock === undefined) {
+                    // We should be able to get the required previous block
+                    throw new BlockNotFoundError(previousBlockNumber);
+                }
+            }
+            // Iterate over checkpoints array and insert them, checking that the block numbers are sequential.
+            let previousCheckpoint = undefined;
+            for (const checkpoint of checkpoints){
+                if (!opts.force && previousCheckpoint && previousCheckpoint.checkpoint.number + 1 !== checkpoint.checkpoint.number) {
+                    throw new CheckpointNumberNotSequentialError(checkpoint.checkpoint.number, previousCheckpoint.checkpoint.number);
+                }
+                previousCheckpoint = checkpoint;
+                // Store every block in the database. the block may already exist, but this has come from chain and is assumed to be correct.
+                for(let i = 0; i < checkpoint.checkpoint.blocks.length; i++){
+                    const block = checkpoint.checkpoint.blocks[i];
+                    if (previousBlock) {
+                        // The blocks should have a sequential block number
+                        if (previousBlock.number !== block.number - 1) {
+                            throw new BlockNumberNotSequentialError(block.number, previousBlock.number);
+                        }
+                        // If the blocks are for the same checkpoint then they should have sequential indexes
+                        if (previousBlock.checkpointNumber === block.checkpointNumber && previousBlock.indexWithinCheckpoint !== block.indexWithinCheckpoint - 1) {
+                            throw new BlockIndexNotSequentialError(block.indexWithinCheckpoint, previousBlock.indexWithinCheckpoint);
+                        }
+                        if (!previousBlock.archive.root.equals(block.header.lastArchive.root)) {
+                            throw new BlockArchiveNotConsistentError(block.number, previousBlock.number, block.header.lastArchive.root, previousBlock.archive.root);
+                        }
+                    } else {
+                        // No previous block, must be block 1 at checkpoint index 0
+                        if (block.indexWithinCheckpoint !== 0) {
+                            throw new BlockIndexNotSequentialError(block.indexWithinCheckpoint, undefined);
+                        }
+                        if (block.number !== INITIAL_L2_BLOCK_NUM) {
+                            throw new BlockNumberNotSequentialError(block.number, undefined);
+                        }
+                    }
+                    previousBlock = block;
+                    await this.addBlockToDatabase(block, checkpoint.checkpoint.number, i);
+                }
+                // Store the checkpoint in the database
+                await this.#checkpoints.set(checkpoint.checkpoint.number, {
+                    header: checkpoint.checkpoint.header.toBuffer(),
+                    archive: checkpoint.checkpoint.archive.toBuffer(),
+                    l1: checkpoint.l1.toBuffer(),
+                    attestations: checkpoint.attestations.map((attestation)=>attestation.toBuffer()),
+                    checkpointNumber: checkpoint.checkpoint.number,
+                    startBlock: checkpoint.checkpoint.blocks[0].number,
+                    numBlocks: checkpoint.checkpoint.blocks.length
+                });
+            }
+            await this.#lastSynchedL1Block.set(checkpoints[checkpoints.length - 1].l1.blockNumber);
+            return true;
+        });
+    }
+    async addBlockToDatabase(block, checkpointNumber, indexWithinCheckpoint) {
+        const blockHash = L2BlockHash.fromField(await block.hash());
+        await this.#blocks.set(block.number, {
+            header: block.header.toBuffer(),
+            blockHash: blockHash.toBuffer(),
+            archive: block.archive.toBuffer(),
+            checkpointNumber,
+            indexWithinCheckpoint
+        });
+        for(let i = 0; i < block.body.txEffects.length; i++){
+            const txEffect = {
+                data: block.body.txEffects[i],
+                l2BlockNumber: block.number,
+                l2BlockHash: blockHash,
+                txIndexInBlock: i
+            };
+            await this.#txEffects.set(txEffect.data.txHash.toString(), serializeIndexedTxEffect(txEffect));
+        }
+        await this.#blockTxs.set(blockHash.toString(), Buffer.concat(block.body.txEffects.map((tx)=>tx.txHash.toBuffer())));
+        // Update indices for block hash and archive
+        await this.#blockHashIndex.set(blockHash.toString(), block.number);
+        await this.#blockArchiveIndex.set(block.archive.root.toString(), block.number);
+    }
+    /**
+   * Unwinds checkpoints from the database
    * @param from -  The tip of the chain, passed for verification purposes,
    *                ensuring that we don't end up deleting something we did not intend
-   * @param blocksToUnwind - The number of blocks we are to unwind
+   * @param checkpointsToUnwind - The number of checkpoints we are to unwind
    * @returns True if the operation is successful
-   */ async unwindBlocks(from, blocksToUnwind) {
+   */ async unwindCheckpoints(from, checkpointsToUnwind) {
         return await this.db.transactionAsync(async ()=>{
-            const last = await this.getSynchedL2BlockNumber();
+            const last = await this.getLatestCheckpointNumber();
             if (from !== last) {
-                throw new Error(`Can only unwind blocks from the tip (requested ${from} but current tip is ${last})`);
+                throw new Error(`Can only unwind checkpoints from the tip (requested ${from} but current tip is ${last})`);
+            }
+            const proven = await this.getProvenCheckpointNumber();
+            if (from - checkpointsToUnwind < proven) {
+                await this.setProvenCheckpointNumber(CheckpointNumber(from - checkpointsToUnwind));
             }
-            for(let i = 0; i < blocksToUnwind; i++){
-                const blockNumber = from - i;
-                const block = await this.getBlock(blockNumber);
-                if (block === undefined) {
-                    this.#log.warn(`Cannot remove block ${blockNumber} from the store since we don't have it`);
+            for(let i = 0; i < checkpointsToUnwind; i++){
+                const checkpointNumber = from - i;
+                const checkpoint = await this.#checkpoints.getAsync(checkpointNumber);
+                if (checkpoint === undefined) {
+                    this.#log.warn(`Cannot remove checkpoint ${checkpointNumber} from the store since we don't have it`);
                     continue;
                 }
-                await this.#blocks.delete(block.data.number);
-                await Promise.all(block.data.body.txEffects.map((tx)=>this.#txIndex.delete(tx.txHash.toString())));
-                const blockHash = (await block.data.hash()).toString();
-                await this.#blockBodies.delete(blockHash);
-                this.#log.debug(`Unwound block ${blockNumber} ${blockHash}`);
+                await this.#checkpoints.delete(checkpointNumber);
+                const maxBlock = checkpoint.startBlock + checkpoint.numBlocks - 1;
+                for(let blockNumber = checkpoint.startBlock; blockNumber <= maxBlock; blockNumber++){
+                    const block = await this.getBlock(BlockNumber(blockNumber));
+                    if (block === undefined) {
+                        this.#log.warn(`Cannot remove block ${blockNumber} from the store since we don't have it`);
+                        continue;
+                    }
+                    await this.#blocks.delete(block.number);
+                    await Promise.all(block.body.txEffects.map((tx)=>this.#txEffects.delete(tx.txHash.toString())));
+                    const blockHash = (await block.hash()).toString();
+                    await this.#blockTxs.delete(blockHash);
+                    // Clean up indices
+                    await this.#blockHashIndex.delete(blockHash);
+                    await this.#blockArchiveIndex.delete(block.archive.root.toString());
+                    this.#log.debug(`Unwound block ${blockNumber} ${blockHash} for checkpoint ${checkpointNumber}`);
+                }
             }
             return true;
         });
     }
+    async getCheckpointData(checkpointNumber) {
+        const checkpointStorage = await this.#checkpoints.getAsync(checkpointNumber);
+        if (!checkpointStorage) {
+            return undefined;
+        }
+        return this.checkpointDataFromCheckpointStorage(checkpointStorage);
+    }
+    async getRangeOfCheckpoints(from, limit) {
+        const checkpoints = [];
+        for(let checkpointNumber = from; checkpointNumber < from + limit; checkpointNumber++){
+            const checkpoint = await this.#checkpoints.getAsync(checkpointNumber);
+            if (!checkpoint) {
+                break;
+            }
+            checkpoints.push(this.checkpointDataFromCheckpointStorage(checkpoint));
+        }
+        return checkpoints;
+    }
+    checkpointDataFromCheckpointStorage(checkpointStorage) {
+        const data = {
+            header: CheckpointHeader.fromBuffer(checkpointStorage.header),
+            archive: AppendOnlyTreeSnapshot.fromBuffer(checkpointStorage.archive),
+            checkpointNumber: CheckpointNumber(checkpointStorage.checkpointNumber),
+            startBlock: checkpointStorage.startBlock,
+            numBlocks: checkpointStorage.numBlocks,
+            l1: L1PublishedData.fromBuffer(checkpointStorage.l1),
+            attestations: checkpointStorage.attestations
+        };
+        return data;
+    }
+    async getBlocksForCheckpoint(checkpointNumber) {
+        const checkpoint = await this.#checkpoints.getAsync(checkpointNumber);
+        if (!checkpoint) {
+            return undefined;
+        }
+        const blocksForCheckpoint = await toArray(this.#blocks.entriesAsync({
+            start: checkpoint.startBlock,
+            end: checkpoint.startBlock + checkpoint.numBlocks
+        }));
+        const converted = await Promise.all(blocksForCheckpoint.map((x)=>this.getBlockFromBlockStorage(x[0], x[1])));
+        return converted.filter(isDefined);
+    }
+    async getProvenBlockNumber() {
+        const provenCheckpointNumber = await this.getProvenCheckpointNumber();
+        if (provenCheckpointNumber === INITIAL_CHECKPOINT_NUMBER - 1) {
+            return BlockNumber(INITIAL_L2_BLOCK_NUM - 1);
+        }
+        const checkpointStorage = await this.#checkpoints.getAsync(provenCheckpointNumber);
+        if (!checkpointStorage) {
+            throw new CheckpointNotFoundError(provenCheckpointNumber);
+        } else {
+            return BlockNumber(checkpointStorage.startBlock + checkpointStorage.numBlocks - 1);
+        }
+    }
+    async getLatestBlockNumber() {
+        const [latestBlocknumber] = await toArray(this.#blocks.keysAsync({
+            reverse: true,
+            limit: 1
+        }));
+        return typeof latestBlocknumber === 'number' ? BlockNumber(latestBlocknumber) : BlockNumber(INITIAL_L2_BLOCK_NUM - 1);
+    }
+    async getLatestCheckpointNumber() {
+        const [latestCheckpointNumber] = await toArray(this.#checkpoints.keysAsync({
+            reverse: true,
+            limit: 1
+        }));
+        if (latestCheckpointNumber === undefined) {
+            return CheckpointNumber(INITIAL_CHECKPOINT_NUMBER - 1);
+        }
+        return CheckpointNumber(latestCheckpointNumber);
+    }
+    async getCheckpointedBlock(number) {
+        const blockStorage = await this.#blocks.getAsync(number);
+        if (!blockStorage) {
+            return undefined;
+        }
+        const checkpoint = await this.#checkpoints.getAsync(blockStorage.checkpointNumber);
+        if (!checkpoint) {
+            return undefined;
+        }
+        const block = await this.getBlockFromBlockStorage(number, blockStorage);
+        if (!block) {
+            return undefined;
+        }
+        return new CheckpointedL2Block(CheckpointNumber(checkpoint.checkpointNumber), block, L1PublishedData.fromBuffer(checkpoint.l1), checkpoint.attestations.map((buf)=>CommitteeAttestation.fromBuffer(buf)));
+    }
+    /**
+   * Gets up to `limit` amount of Checkpointed L2 blocks starting from `from`.
+   * @param start - Number of the first block to return (inclusive).
+   * @param limit - The number of blocks to return.
+   * @returns The requested L2 blocks
+   */ async *getCheckpointedBlocks(start, limit) {
+        const checkpointCache = new Map();
+        for await (const [blockNumber, blockStorage] of this.getBlockStorages(start, limit)){
+            const block = await this.getBlockFromBlockStorage(blockNumber, blockStorage);
+            if (block) {
+                const checkpoint = checkpointCache.get(CheckpointNumber(blockStorage.checkpointNumber)) ?? await this.#checkpoints.getAsync(blockStorage.checkpointNumber);
+                if (checkpoint) {
+                    checkpointCache.set(CheckpointNumber(blockStorage.checkpointNumber), checkpoint);
+                    const checkpointedBlock = new CheckpointedL2Block(CheckpointNumber(checkpoint.checkpointNumber), block, L1PublishedData.fromBuffer(checkpoint.l1), checkpoint.attestations.map((buf)=>CommitteeAttestation.fromBuffer(buf)));
+                    yield checkpointedBlock;
+                }
+            }
+        }
+    }
+    async getCheckpointedBlockByHash(blockHash) {
+        const blockNumber = await this.#blockHashIndex.getAsync(blockHash.toString());
+        if (blockNumber === undefined) {
+            return undefined;
+        }
+        return this.getCheckpointedBlock(BlockNumber(blockNumber));
+    }
+    async getCheckpointedBlockByArchive(archive) {
+        const blockNumber = await this.#blockArchiveIndex.getAsync(archive.toString());
+        if (blockNumber === undefined) {
+            return undefined;
+        }
+        return this.getCheckpointedBlock(BlockNumber(blockNumber));
+    }
     /**
    * Gets up to `limit` amount of L2 blocks starting from `from`.
    * @param start - Number of the first block to return (inclusive).
    * @param limit - The number of blocks to return.
    * @returns The requested L2 blocks
    */ async *getBlocks(start, limit) {
-        for await (const blockStorage of this.#blocks.valuesAsync(this.#computeBlockRange(start, limit))){
-            const block = await this.getBlockFromBlockStorage(blockStorage);
-            yield block;
+        for await (const [blockNumber, blockStorage] of this.getBlockStorages(start, limit)){
+            const block = await this.getBlockFromBlockStorage(blockNumber, blockStorage);
+            if (block) {
+                yield block;
+            }
         }
     }
     /**
@@ -104,7 +399,59 @@ export { TxReceipt } from '@aztec/stdlib/tx';
         if (!blockStorage || !blockStorage.header) {
             return Promise.resolve(undefined);
         }
-        return this.getBlockFromBlockStorage(blockStorage);
+        return this.getBlockFromBlockStorage(blockNumber, blockStorage);
+    }
+    /**
+   * Gets an L2 block by its hash.
+   * @param blockHash - The hash of the block to return.
+   * @returns The requested L2 block.
+   */ async getBlockByHash(blockHash) {
+        const blockNumber = await this.#blockHashIndex.getAsync(blockHash.toString());
+        if (blockNumber === undefined) {
+            return undefined;
+        }
+        return this.getBlock(BlockNumber(blockNumber));
+    }
+    /**
+   * Gets an L2 block by its archive root.
+   * @param archive - The archive root of the block to return.
+   * @returns The requested L2 block.
+   */ async getBlockByArchive(archive) {
+        const blockNumber = await this.#blockArchiveIndex.getAsync(archive.toString());
+        if (blockNumber === undefined) {
+            return undefined;
+        }
+        return this.getBlock(BlockNumber(blockNumber));
+    }
+    /**
+   * Gets a block header by its hash.
+   * @param blockHash - The hash of the block to return.
+   * @returns The requested block header.
+   */ async getBlockHeaderByHash(blockHash) {
+        const blockNumber = await this.#blockHashIndex.getAsync(blockHash.toString());
+        if (blockNumber === undefined) {
+            return undefined;
+        }
+        const blockStorage = await this.#blocks.getAsync(blockNumber);
+        if (!blockStorage || !blockStorage.header) {
+            return undefined;
+        }
+        return BlockHeader.fromBuffer(blockStorage.header);
+    }
+    /**
+   * Gets a block header by its archive root.
+   * @param archive - The archive root of the block to return.
+   * @returns The requested block header.
+   */ async getBlockHeaderByArchive(archive) {
+        const blockNumber = await this.#blockArchiveIndex.getAsync(archive.toString());
+        if (blockNumber === undefined) {
+            return undefined;
+        }
+        const blockStorage = await this.#blocks.getAsync(blockNumber);
+        if (!blockStorage || !blockStorage.header) {
+            return undefined;
+        }
+        return BlockHeader.fromBuffer(blockStorage.header);
     }
     /**
    * Gets the headers for a sequence of L2 blocks.
@@ -112,63 +459,91 @@ export { TxReceipt } from '@aztec/stdlib/tx';
    * @param limit - The number of blocks to return.
    * @returns The requested L2 block headers
    */ async *getBlockHeaders(start, limit) {
-        for await (const blockStorage of this.#blocks.valuesAsync(this.#computeBlockRange(start, limit))){
-            yield BlockHeader.fromBuffer(blockStorage.header);
+        for await (const [blockNumber, blockStorage] of this.getBlockStorages(start, limit)){
+            const header = BlockHeader.fromBuffer(blockStorage.header);
+            if (header.getBlockNumber() !== blockNumber) {
+                throw new Error(`Block number mismatch when retrieving block header from archive (expected ${blockNumber} but got ${header.getBlockNumber()})`);
+            }
+            yield header;
+        }
+    }
+    async *getBlockStorages(start, limit) {
+        let expectedBlockNumber = start;
+        for await (const [blockNumber, blockStorage] of this.#blocks.entriesAsync(this.#computeBlockRange(start, limit))){
+            if (blockNumber !== expectedBlockNumber) {
+                throw new Error(`Block number mismatch when iterating blocks from archive (expected ${expectedBlockNumber} but got ${blockNumber})`);
+            }
+            expectedBlockNumber++;
+            yield [
+                blockNumber,
+                blockStorage
+            ];
         }
     }
-    async getBlockFromBlockStorage(blockStorage) {
+    async getBlockFromBlockStorage(blockNumber, blockStorage) {
         const header = BlockHeader.fromBuffer(blockStorage.header);
         const archive = AppendOnlyTreeSnapshot.fromBuffer(blockStorage.archive);
-        const blockHash = (await header.hash()).toString();
-        const blockBodyBuffer = await this.#blockBodies.getAsync(blockHash);
-        if (blockBodyBuffer === undefined) {
-            throw new Error(`Could not retrieve body for block ${header.globalVariables.blockNumber.toNumber()} ${blockHash}`);
+        const blockHash = blockStorage.blockHash;
+        const blockHashString = bufferToHex(blockHash);
+        const blockTxsBuffer = await this.#blockTxs.getAsync(blockHashString);
+        if (blockTxsBuffer === undefined) {
+            this.#log.warn(`Could not find body for block ${header.globalVariables.blockNumber} ${blockHash}`);
+            return undefined;
         }
-        const body = Body.fromBuffer(blockBodyBuffer);
-        const l2Block = new L2Block(archive, header, body);
-        return {
-            data: l2Block,
-            l1: blockStorage.l1
-        };
+        const txEffects = [];
+        const reader = BufferReader.asReader(blockTxsBuffer);
+        while(!reader.isEmpty()){
+            const txHash = reader.readObject(TxHash);
+            const txEffect = await this.#txEffects.getAsync(txHash.toString());
+            if (txEffect === undefined) {
+                this.#log.warn(`Could not find tx effect for tx ${txHash} in block ${blockNumber}`);
+                return undefined;
+            }
+            txEffects.push(deserializeIndexedTxEffect(txEffect).data);
+        }
+        const body = new Body(txEffects);
+        const block = new L2BlockNew(archive, header, body, CheckpointNumber(blockStorage.checkpointNumber), blockStorage.indexWithinCheckpoint);
+        if (block.number !== blockNumber) {
+            throw new Error(`Block number mismatch when retrieving block from archive (expected ${blockNumber} but got ${block.number} with hash ${blockHashString})`);
+        }
+        return block;
     }
     /**
    * Gets a tx effect.
-   * @param txHash - The txHash of the tx corresponding to the tx effect.
-   * @returns The requested tx effect (or undefined if not found).
+   * @param txHash - The hash of the tx corresponding to the tx effect.
+   * @returns The requested tx effect with block info (or undefined if not found).
    */ async getTxEffect(txHash) {
-        const [blockNumber, txIndex] = await this.getTxLocation(txHash) ?? [];
-        if (typeof blockNumber !== 'number' || typeof txIndex !== 'number') {
+        const buffer = await this.#txEffects.getAsync(txHash.toString());
+        if (!buffer) {
             return undefined;
         }
-        const block = await this.getBlock(blockNumber);
-        if (!block) {
-            return undefined;
-        }
-        return {
-            data: block.data.body.txEffects[txIndex],
-            l2BlockNumber: block.data.number,
-            l2BlockHash: (await block.data.hash()).toString()
-        };
+        return deserializeIndexedTxEffect(buffer);
     }
     /**
    * Gets a receipt of a settled tx.
    * @param txHash - The hash of a tx we try to get the receipt for.
    * @returns The requested tx receipt (or undefined if not found).
    */ async getSettledTxReceipt(txHash) {
-        const [blockNumber, txIndex] = await this.getTxLocation(txHash) ?? [];
-        if (typeof blockNumber !== 'number' || typeof txIndex !== 'number') {
+        const txEffect = await this.getTxEffect(txHash);
+        if (!txEffect) {
             return undefined;
         }
-        const block = await this.getBlock(blockNumber);
-        const tx = block.data.body.txEffects[txIndex];
-        return new TxReceipt(txHash, TxReceipt.statusFromRevertCode(tx.revertCode), '', tx.transactionFee.toBigInt(), L2BlockHash.fromField(await block.data.hash()), block.data.number);
+        return new TxReceipt(txHash, TxReceipt.statusFromRevertCode(txEffect.data.revertCode), '', txEffect.data.transactionFee.toBigInt(), txEffect.l2BlockHash, BlockNumber(txEffect.l2BlockNumber));
     }
     /**
    * Looks up which block included the requested tx effect.
    * @param txHash - The txHash of the tx.
    * @returns The block number and index of the tx.
-   */ getTxLocation(txHash) {
-        return this.#txIndex.getAsync(txHash.toString());
+   */ async getTxLocation(txHash) {
+        const txEffect = await this.#txEffects.getAsync(txHash.toString());
+        if (!txEffect) {
+            return undefined;
+        }
+        const { l2BlockNumber, txIndexInBlock } = deserializeIndexedTxEffect(txEffect);
+        return [
+            l2BlockNumber,
+            txIndexInBlock
+        ];
     }
     /**
    * Looks up which block deployed a particular contract.
@@ -178,14 +553,22 @@ export { TxReceipt } from '@aztec/stdlib/tx';
         return this.#contractIndex.getAsync(contractAddress.toString());
     }
     /**
-   * Gets the number of the latest L2 block processed.
-   * @returns The number of the latest L2 block processed.
-   */ async getSynchedL2BlockNumber() {
+   * Gets the number of the latest L2 block checkpointed.
+   * @returns The number of the latest L2 block checkpointed.
+   */ async getCheckpointedL2BlockNumber() {
+        const latestCheckpointNumber = await this.getLatestCheckpointNumber();
+        const checkpoint = await this.getCheckpointData(latestCheckpointNumber);
+        if (!checkpoint) {
+            return BlockNumber(INITIAL_L2_BLOCK_NUM - 1);
+        }
+        return BlockNumber(checkpoint.startBlock + checkpoint.numBlocks - 1);
+    }
+    async getLatestL2BlockNumber() {
         const [lastBlockNumber] = await toArray(this.#blocks.keysAsync({
             reverse: true,
             limit: 1
         }));
-        return typeof lastBlockNumber === 'number' ? lastBlockNumber : INITIAL_L2_BLOCK_NUM - 1;
+        return typeof lastBlockNumber === 'number' ? BlockNumber(lastBlockNumber) : BlockNumber(INITIAL_L2_BLOCK_NUM - 1);
     }
     /**
    * Gets the most recent L1 block processed.
@@ -196,11 +579,16 @@ export { TxReceipt } from '@aztec/stdlib/tx';
     setSynchedL1BlockNumber(l1BlockNumber) {
         return this.#lastSynchedL1Block.set(l1BlockNumber);
     }
-    async getProvenL2BlockNumber() {
-        return await this.#lastProvenL2Block.getAsync() ?? 0;
+    async getProvenCheckpointNumber() {
+        const [latestCheckpointNumber, provenCheckpointNumber] = await Promise.all([
+            this.getLatestCheckpointNumber(),
+            this.#lastProvenCheckpoint.getAsync()
+        ]);
+        return (provenCheckpointNumber ?? 0) > latestCheckpointNumber ? latestCheckpointNumber : CheckpointNumber(provenCheckpointNumber ?? 0);
     }
-    setProvenL2BlockNumber(blockNumber) {
-        return this.#lastProvenL2Block.set(blockNumber);
+    async setProvenCheckpointNumber(checkpointNumber) {
+        const result = await this.#lastProvenCheckpoint.set(checkpointNumber);
+        return result;
     }
     #computeBlockRange(start, limit) {
         if (limit < 1) {
@@ -214,4 +602,25 @@ export { TxReceipt } from '@aztec/stdlib/tx';
             limit
         };
     }
+    /**
+   * Gets the pending chain validation status.
+   * @returns The validation status or undefined if not set.
+   */ async getPendingChainValidationStatus() {
+        const buffer = await this.#pendingChainValidationStatus.getAsync();
+        if (!buffer) {
+            return undefined;
+        }
+        return deserializeValidateCheckpointResult(buffer);
+    }
+    /**
+   * Sets the pending chain validation status.
+   * @param status - The validation status to store.
+   */ async setPendingChainValidationStatus(status) {
+        if (status) {
+            const buffer = serializeValidateCheckpointResult(status);
+            await this.#pendingChainValidationStatus.set(buffer);
+        } else {
+            await this.#pendingChainValidationStatus.delete();
+        }
+    }
 }