@aztec/p2p 0.0.0-test.1 → 0.0.1-commit.24de95ac

package/dest/mem_pools/tx_pool/tx_pool_test_suite.js

@@ -1,38 +1,89 @@
 import { unfreeze } from '@aztec/foundation/types';
 import { GasFees } from '@aztec/stdlib/gas';
 import { mockTx } from '@aztec/stdlib/testing';
+import { BlockHeader, GlobalVariables } from '@aztec/stdlib/tx';
 /**
  * Tests a TxPool implementation.
  * @param getTxPool - Gets a fresh TxPool
  */ export function describeTxPool(getTxPool) {
     let pool;
+    const minedBlockHeader = BlockHeader.empty({
+        globalVariables: GlobalVariables.empty({
+            blockNumber: 1,
+            timestamp: 0n
+        })
+    });
     beforeEach(()=>{
         pool = getTxPool();
     });
-    it('Adds txs to the pool as pending', async ()=>{
+    afterEach(()=>{
+        pool.removeAllListeners('txs-added');
+    });
+    it('adds txs to the pool as pending', async ()=>{
         const tx1 = await mockTx();
         await pool.addTxs([
             tx1
         ]);
-        const poolTx = await pool.getTxByHash(await tx1.getTxHash());
-        expect(await poolTx.getTxHash()).toEqual(await tx1.getTxHash());
-        await expect(pool.getTxStatus(await tx1.getTxHash())).resolves.toEqual('pending');
+        const poolTx = await pool.getTxByHash(tx1.getTxHash());
+        expect(poolTx.getTxHash()).toEqual(tx1.getTxHash());
+        await expect(pool.getTxStatus(tx1.getTxHash())).resolves.toEqual('pending');
         await expect(pool.getPendingTxHashes()).resolves.toEqual([
-            await tx1.getTxHash()
+            tx1.getTxHash()
         ]);
+        await expect(pool.getPendingTxCount()).resolves.toEqual(1);
     });
-    it('Removes txs from the pool', async ()=>{
-        const tx1 = await mockTx();
+    it('emits txs-added event with new txs', async ()=>{
+        const tx1 = await mockTx(); // existing and pending
+        const tx2 = await mockTx(); // mined but not known
+        const tx3 = await mockTx(); // brand new
         await pool.addTxs([
             tx1
         ]);
+        await pool.markAsMined([
+            tx2.getTxHash()
+        ], minedBlockHeader);
+        let txsFromEvent = undefined;
+        pool.once('txs-added', ({ txs })=>{
+            txsFromEvent = txs;
+        });
+        await pool.addTxs([
+            tx1,
+            tx2,
+            tx3
+        ]);
+        expect(txsFromEvent).toBeDefined();
+        expect(txsFromEvent).toHaveLength(2);
+        expect(txsFromEvent).toEqual(expect.arrayContaining([
+            tx2,
+            tx3
+        ]));
+    });
+    it('permanently deletes pending txs and soft-deletes mined txs', async ()=>{
+        const pendingTx = await mockTx(1);
+        const minedTx = await mockTx(2);
+        await pool.addTxs([
+            pendingTx,
+            minedTx
+        ]);
+        await pool.markAsMined([
+            minedTx.getTxHash()
+        ], minedBlockHeader);
+        // Delete a pending tx - should be permanently deleted
+        await pool.deleteTxs([
+            pendingTx.getTxHash()
+        ]);
+        await expect(pool.getTxByHash(pendingTx.getTxHash())).resolves.toBeUndefined();
+        await expect(pool.getTxStatus(pendingTx.getTxHash())).resolves.toBeUndefined();
+        // Delete a mined tx - should be soft-deleted (still in storage)
         await pool.deleteTxs([
-            await tx1.getTxHash()
+            minedTx.getTxHash()
         ]);
-        await expect(pool.getTxByHash(await tx1.getTxHash())).resolves.toBeFalsy();
-        await expect(pool.getTxStatus(await tx1.getTxHash())).resolves.toBeUndefined();
+        await expect(pool.getTxByHash(minedTx.getTxHash())).resolves.toBeDefined();
+        await expect(pool.getTxStatus(minedTx.getTxHash())).resolves.toEqual('deleted');
+        await expect(pool.getMinedTxHashes()).resolves.toEqual([]);
+        await expect(pool.getPendingTxCount()).resolves.toEqual(0);
     });
-    it('Marks txs as mined', async ()=>{
+    it('marks txs as mined', async ()=>{
         const tx1 = await mockTx(1);
         const tx2 = await mockTx(2);
         await pool.addTxs([
@@ -40,21 +91,22 @@ import { mockTx } from '@aztec/stdlib/testing';
             tx2
         ]);
         await pool.markAsMined([
-            await tx1.getTxHash()
-        ], 1);
-        await expect(pool.getTxByHash(await tx1.getTxHash())).resolves.toEqual(tx1);
-        await expect(pool.getTxStatus(await tx1.getTxHash())).resolves.toEqual('mined');
+            tx1.getTxHash()
+        ], minedBlockHeader);
+        await expect(pool.getTxByHash(tx1.getTxHash())).resolves.toEqual(tx1);
+        await expect(pool.getTxStatus(tx1.getTxHash())).resolves.toEqual('mined');
         await expect(pool.getMinedTxHashes()).resolves.toEqual([
             [
-                await tx1.getTxHash(),
+                tx1.getTxHash(),
                 1
             ]
         ]);
         await expect(pool.getPendingTxHashes()).resolves.toEqual([
-            await tx2.getTxHash()
+            tx2.getTxHash()
         ]);
+        await expect(pool.getPendingTxCount()).resolves.toEqual(1);
     });
-    it('Marks txs as pending after being mined', async ()=>{
+    it('marks txs as pending after being mined', async ()=>{
         const tx1 = await mockTx(1);
         const tx2 = await mockTx(2);
         await pool.addTxs([
@@ -62,35 +114,36 @@ import { mockTx } from '@aztec/stdlib/testing';
             tx2
         ]);
         await pool.markAsMined([
-            await tx1.getTxHash()
-        ], 1);
+            tx1.getTxHash()
+        ], minedBlockHeader);
         await pool.markMinedAsPending([
-            await tx1.getTxHash()
+            tx1.getTxHash()
         ]);
         await expect(pool.getMinedTxHashes()).resolves.toEqual([]);
         const pending = await pool.getPendingTxHashes();
         expect(pending).toHaveLength(2);
         expect(pending).toEqual(expect.arrayContaining([
-            await tx1.getTxHash(),
-            await tx2.getTxHash()
+            tx1.getTxHash(),
+            tx2.getTxHash()
         ]));
+        await expect(pool.getPendingTxCount()).resolves.toEqual(2);
     });
-    it('Only marks txs as pending if they are known', async ()=>{
+    it('only marks txs as pending if they are known', async ()=>{
         const tx1 = await mockTx(1);
         // simulate a situation where not all peers have all the txs
         const tx2 = await mockTx(2);
-        const someTxHashThatThisPeerDidNotSee = await tx2.getTxHash();
+        const someTxHashThatThisPeerDidNotSee = tx2.getTxHash();
         await pool.addTxs([
             tx1
         ]);
         // this peer knows that tx2 was mined, but it does not have the tx object
         await pool.markAsMined([
-            await tx1.getTxHash(),
+            tx1.getTxHash(),
             someTxHashThatThisPeerDidNotSee
-        ], 1);
+        ], minedBlockHeader);
         expect(await pool.getMinedTxHashes()).toEqual(expect.arrayContaining([
             [
-                await tx1.getTxHash(),
+                tx1.getTxHash(),
                 1
             ],
             [
@@ -100,15 +153,16 @@ import { mockTx } from '@aztec/stdlib/testing';
         ]));
         // reorg: both txs should now become available again
         await pool.markMinedAsPending([
-            await tx1.getTxHash(),
+            tx1.getTxHash(),
             someTxHashThatThisPeerDidNotSee
         ]);
         await expect(pool.getMinedTxHashes()).resolves.toEqual([]);
         await expect(pool.getPendingTxHashes()).resolves.toEqual([
-            await tx1.getTxHash()
+            tx1.getTxHash()
         ]); // tx2 is not in the pool
+        await expect(pool.getPendingTxCount()).resolves.toEqual(1);
     });
-    it('Returns all transactions in the pool', async ()=>{
+    it('returns all transactions in the pool', async ()=>{
         const tx1 = await mockTx(1);
         const tx2 = await mockTx(2);
         const tx3 = await mockTx(3);
@@ -124,8 +178,9 @@ import { mockTx } from '@aztec/stdlib/testing';
             tx2,
             tx3
         ]));
+        await expect(pool.getPendingTxCount()).resolves.toEqual(3);
     });
-    it('Returns all txHashes in the pool', async ()=>{
+    it('returns all txHashes in the pool', async ()=>{
         const tx1 = await mockTx(1);
         const tx2 = await mockTx(2);
         const tx3 = await mockTx(3);
@@ -135,14 +190,73 @@ import { mockTx } from '@aztec/stdlib/testing';
             tx3
         ]);
         const poolTxHashes = await pool.getAllTxHashes();
+        const expectedHashes = [
+            tx1,
+            tx2,
+            tx3
+        ].map((tx)=>tx.getTxHash());
         expect(poolTxHashes).toHaveLength(3);
-        expect(poolTxHashes).toEqual(expect.arrayContaining([
-            await tx1.getTxHash(),
-            await tx2.getTxHash(),
-            await tx3.getTxHash()
+        expect(poolTxHashes).toEqual(expect.arrayContaining(expectedHashes));
+        await expect(pool.getPendingTxCount()).resolves.toEqual(3);
+    });
+    it('returns txs by their hash', async ()=>{
+        const tx1 = await mockTx(1);
+        const tx2 = await mockTx(2);
+        const tx3 = await mockTx(3);
+        await pool.addTxs([
+            tx1,
+            tx2,
+            tx3
+        ]);
+        const requestedTxs = await pool.getTxsByHash([
+            tx1.getTxHash(),
+            tx3.getTxHash()
+        ]);
+        expect(requestedTxs).toHaveLength(2);
+        expect(requestedTxs).toEqual(expect.arrayContaining([
+            tx1,
+            tx3
         ]));
     });
-    it('Returns pending tx hashes sorted by priority', async ()=>{
+    it('returns a large number of transactions by their hash', async ()=>{
+        const numTxs = 1000;
+        const txs = await Promise.all(Array.from({
+            length: numTxs
+        }, (_, i)=>mockTx(i)));
+        const hashes = txs.map((tx)=>tx.getTxHash());
+        await pool.addTxs(txs);
+        const requestedTxs = await pool.getTxsByHash(hashes);
+        expect(requestedTxs).toHaveLength(numTxs);
+        expect(requestedTxs).toEqual(expect.arrayContaining(txs));
+    });
+    it('returns whether or not txs exist', async ()=>{
+        const tx1 = await mockTx(1);
+        const tx2 = await mockTx(2);
+        const tx3 = await mockTx(3);
+        await pool.addTxs([
+            tx1,
+            tx2,
+            tx3
+        ]);
+        const tx4 = await mockTx(4);
+        const tx5 = await mockTx(5);
+        const availability = await pool.hasTxs([
+            tx1.getTxHash(),
+            tx2.getTxHash(),
+            tx3.getTxHash(),
+            tx4.getTxHash(),
+            tx5.getTxHash()
+        ]);
+        expect(availability).toHaveLength(5);
+        expect(availability).toEqual(expect.arrayContaining([
+            true,
+            true,
+            true,
+            false,
+            false
+        ]));
+    });
+    it('returns pending tx hashes sorted by priority', async ()=>{
         const withPriorityFee = (tx, fee)=>{
             unfreeze(tx.data.constants.txContext.gasSettings).maxPriorityFeesPerGas = new GasFees(fee, fee);
             return tx;
@@ -159,11 +273,122 @@ import { mockTx } from '@aztec/stdlib/testing';
         ]);
         const poolTxHashes = await pool.getPendingTxHashes();
         expect(poolTxHashes).toHaveLength(4);
-        expect(poolTxHashes).toEqual(await Promise.all([
+        expect(poolTxHashes).toEqual([
             tx4,
             tx1,
             tx3,
             tx2
-        ].map((tx)=>tx.getTxHash())));
+        ].map((tx)=>tx.getTxHash()));
+    });
+    describe('soft-delete', ()=>{
+        it('soft-deletes mined txs and keeps them in storage', async ()=>{
+            const txs = await Promise.all([
+                mockTx(1),
+                mockTx(2),
+                mockTx(3)
+            ]);
+            await pool.addTxs(txs);
+            // Mark first tx as mined
+            await pool.markAsMined([
+                txs[0].getTxHash()
+            ], minedBlockHeader);
+            // Verify initial state
+            await expect(pool.getPendingTxCount()).resolves.toBe(2);
+            await expect(pool.getTxByHash(txs[0].getTxHash())).resolves.toBeDefined();
+            await expect(pool.getTxByHash(txs[1].getTxHash())).resolves.toBeDefined();
+            // Delete mined tx - should be soft-deleted
+            await pool.deleteTxs([
+                txs[0].getTxHash()
+            ]);
+            // Delete pending tx - should be permanently deleted
+            await pool.deleteTxs([
+                txs[1].getTxHash()
+            ]);
+            // Verify mined tx still exists in storage but has 'deleted' status
+            await expect(pool.getTxByHash(txs[0].getTxHash())).resolves.toBeDefined();
+            await expect(pool.getTxStatus(txs[0].getTxHash())).resolves.toEqual('deleted');
+            // Verify pending tx is permanently deleted
+            await expect(pool.getTxByHash(txs[1].getTxHash())).resolves.toBeUndefined();
+            await expect(pool.getTxStatus(txs[1].getTxHash())).resolves.toBeUndefined();
+            // Verify remaining pending count
+            await expect(pool.getPendingTxCount()).resolves.toBe(1);
+            // Verify pending hashes don't include deleted txs
+            const pendingHashes = await pool.getPendingTxHashes();
+            expect(pendingHashes).toHaveLength(1);
+            expect(pendingHashes.map((h)=>h.toString())).toContain(txs[2].getTxHash().toString());
+        });
+        it('cleans up old deleted mined transactions', async ()=>{
+            const txs = await Promise.all([
+                mockTx(1),
+                mockTx(2),
+                mockTx(3)
+            ]);
+            await pool.addTxs(txs);
+            // Mark first two as mined in block 1
+            await pool.markAsMined([
+                txs[0].getTxHash(),
+                txs[1].getTxHash()
+            ], minedBlockHeader);
+            // Soft-delete mined transactions
+            await pool.deleteTxs([
+                txs[0].getTxHash(),
+                txs[1].getTxHash()
+            ]);
+            // Clean up deleted mined txs from block 1 and earlier
+            const deletedCount = await pool.cleanupDeletedMinedTxs(1);
+            // Verify old transactions are permanently deleted
+            expect(deletedCount).toBe(2);
+            await expect(pool.getTxByHash(txs[0].getTxHash())).resolves.toBeUndefined();
+            await expect(pool.getTxByHash(txs[1].getTxHash())).resolves.toBeUndefined();
+            await expect(pool.getTxByHash(txs[2].getTxHash())).resolves.toBeDefined();
+        });
+        it('does not clean up recent deleted mined transactions', async ()=>{
+            const txs = await Promise.all([
+                mockTx(1),
+                mockTx(2)
+            ]);
+            await pool.addTxs(txs);
+            // Mark as mined in block 2
+            const laterBlockHeader = BlockHeader.empty({
+                globalVariables: GlobalVariables.empty({
+                    blockNumber: 2,
+                    timestamp: 0n
+                })
+            });
+            await pool.markAsMined([
+                txs[0].getTxHash()
+            ], laterBlockHeader);
+            // Soft-delete a mined transaction
+            await pool.deleteTxs([
+                txs[0].getTxHash()
+            ]);
+            // Try to clean up with block 1 (before the mined block)
+            const deletedCount = await pool.cleanupDeletedMinedTxs(1);
+            // Verify no transactions were cleaned up
+            expect(deletedCount).toBe(0);
+            await expect(pool.getTxByHash(txs[0].getTxHash())).resolves.toBeDefined();
+        });
+        it('restores deleted mined tx when it is mined again', async ()=>{
+            const tx = await mockTx(1);
+            await pool.addTxs([
+                tx
+            ]);
+            // Mark as mined
+            await pool.markAsMined([
+                tx.getTxHash()
+            ], minedBlockHeader);
+            // Soft-delete it
+            await pool.deleteTxs([
+                tx.getTxHash()
+            ]);
+            await expect(pool.getTxStatus(tx.getTxHash())).resolves.toEqual('deleted');
+            // Mark as mined again (e.g., after a reorg)
+            await pool.markAsMined([
+                tx.getTxHash()
+            ], minedBlockHeader);
+            // Should be back to mined status
+            await expect(pool.getTxStatus(tx.getTxHash())).resolves.toEqual('mined');
+            await expect(pool.getTxByHash(tx.getTxHash())).resolves.toBeDefined();
+        });
     });
 }

package/dest/msg_validators/attestation_validator/attestation_validator.d.ts

@@ -2,6 +2,7 @@ import type { EpochCacheInterface } from '@aztec/epoch-cache';
 import { type BlockAttestation, type P2PValidator, PeerErrorSeverity } from '@aztec/stdlib/p2p';
 export declare class AttestationValidator implements P2PValidator<BlockAttestation> {
     private epochCache;
+    private logger;
     constructor(epochCache: EpochCacheInterface);
     validate(message: BlockAttestation): Promise<PeerErrorSeverity | undefined>;
 }

package/dest/msg_validators/attestation_validator/attestation_validator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"attestation_validator.d.ts","sourceRoot":"","sources":["../../../src/msg_validators/attestation_validator/attestation_validator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,KAAK,gBAAgB,EAAE,KAAK,YAAY,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAEhG,qBAAa,oBAAqB,YAAW,YAAY,CAAC,gBAAgB,CAAC;IACzE,OAAO,CAAC,UAAU,CAAsB;gBAE5B,UAAU,EAAE,mBAAmB;IAIrC,QAAQ,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC;CAelF"}
+{"version":3,"file":"attestation_validator.d.ts","sourceRoot":"","sources":["../../../src/msg_validators/attestation_validator/attestation_validator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAG9D,OAAO,EAAE,KAAK,gBAAgB,EAAE,KAAK,YAAY,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAEhG,qBAAa,oBAAqB,YAAW,YAAY,CAAC,gBAAgB,CAAC;IACzE,OAAO,CAAC,UAAU,CAAsB;IACxC,OAAO,CAAC,MAAM,CAAS;gBAEX,UAAU,EAAE,mBAAmB;IAKrC,QAAQ,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC;CAwDlF"}

package/dest/msg_validators/attestation_validator/attestation_validator.js

@@ -1,19 +1,55 @@
+import { NoCommitteeError } from '@aztec/ethereum';
+import { createLogger } from '@aztec/foundation/log';
 import { PeerErrorSeverity } from '@aztec/stdlib/p2p';
 export class AttestationValidator {
     epochCache;
+    logger;
     constructor(epochCache){
         this.epochCache = epochCache;
+        this.logger = createLogger('p2p:attestation-validator');
     }
     async validate(message) {
-        const { currentSlot, nextSlot } = await this.epochCache.getProposerInCurrentOrNextSlot();
-        const slotNumberBigInt = message.payload.header.globalVariables.slotNumber.toBigInt();
-        if (slotNumberBigInt !== currentSlot && slotNumberBigInt !== nextSlot) {
-            return PeerErrorSeverity.HighToleranceError;
+        const slotNumberBigInt = message.payload.header.slotNumber.toBigInt();
+        try {
+            const { currentProposer, nextProposer, currentSlot, nextSlot } = await this.epochCache.getProposerAttesterAddressInCurrentOrNextSlot();
+            if (slotNumberBigInt !== currentSlot && slotNumberBigInt !== nextSlot) {
+                this.logger.warn(`Attestation slot ${slotNumberBigInt} is not current (${currentSlot}) or next (${nextSlot}) slot`);
+                return PeerErrorSeverity.HighToleranceError;
+            }
+            // Verify the signature is valid
+            const attester = message.getSender();
+            if (attester === undefined) {
+                this.logger.warn(`Invalid signature in attestation for slot ${slotNumberBigInt}`);
+                return PeerErrorSeverity.LowToleranceError;
+            }
+            // Verify the attester is in the committee for this slot
+            if (!await this.epochCache.isInCommittee(slotNumberBigInt, attester)) {
+                this.logger.warn(`Attester ${attester.toString()} is not in committee for slot ${slotNumberBigInt}`);
+                return PeerErrorSeverity.HighToleranceError;
+            }
+            // Verify the proposer signature matches the expected proposer for this slot
+            const proposer = message.getProposer();
+            const expectedProposer = slotNumberBigInt === currentSlot ? currentProposer : nextProposer;
+            if (!expectedProposer) {
+                this.logger.warn(`No proposer defined for slot ${slotNumberBigInt}`);
+                return PeerErrorSeverity.HighToleranceError;
+            }
+            if (!proposer) {
+                this.logger.warn(`Invalid proposer signature in attestation for slot ${slotNumberBigInt}`);
+                return PeerErrorSeverity.LowToleranceError;
+            }
+            if (!proposer.equals(expectedProposer)) {
+                this.logger.warn(`Proposer signature mismatch in attestation. ` + `Expected ${expectedProposer?.toString() ?? 'none'} but got ${proposer.toString()} for slot ${slotNumberBigInt}`);
+                return PeerErrorSeverity.HighToleranceError;
+            }
+            return undefined;
+        } catch (e) {
+            // People shouldn't be sending us attestations if the committee doesn't exist
+            if (e instanceof NoCommitteeError) {
+                this.logger.warn(`No committee exists for attestation for slot ${slotNumberBigInt}`);
+                return PeerErrorSeverity.LowToleranceError;
+            }
+            throw e;
         }
-        const attester = await message.getSender();
-        if (!await this.epochCache.isInCommittee(attester)) {
-            return PeerErrorSeverity.HighToleranceError;
-        }
-        return undefined;
     }
 }

package/dest/msg_validators/block_proposal_validator/block_proposal_validator.d.ts

@@ -2,7 +2,11 @@ import type { EpochCacheInterface } from '@aztec/epoch-cache';
 import { type BlockProposal, type P2PValidator, PeerErrorSeverity } from '@aztec/stdlib/p2p';
 export declare class BlockProposalValidator implements P2PValidator<BlockProposal> {
     private epochCache;
-    constructor(epochCache: EpochCacheInterface);
+    private logger;
+    private txsPermitted;
+    constructor(epochCache: EpochCacheInterface, opts: {
+        txsPermitted: boolean;
+    });
     validate(block: BlockProposal): Promise<PeerErrorSeverity | undefined>;
 }
 //# sourceMappingURL=block_proposal_validator.d.ts.map

package/dest/msg_validators/block_proposal_validator/block_proposal_validator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"block_proposal_validator.d.ts","sourceRoot":"","sources":["../../../src/msg_validators/block_proposal_validator/block_proposal_validator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,KAAK,aAAa,EAAE,KAAK,YAAY,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAE7F,qBAAa,sBAAuB,YAAW,YAAY,CAAC,aAAa,CAAC;IACxE,OAAO,CAAC,UAAU,CAAsB;gBAE5B,UAAU,EAAE,mBAAmB;IAIrC,QAAQ,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC;CAkB7E"}
+{"version":3,"file":"block_proposal_validator.d.ts","sourceRoot":"","sources":["../../../src/msg_validators/block_proposal_validator/block_proposal_validator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAG9D,OAAO,EAAE,KAAK,aAAa,EAAE,KAAK,YAAY,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAE7F,qBAAa,sBAAuB,YAAW,YAAY,CAAC,aAAa,CAAC;IACxE,OAAO,CAAC,UAAU,CAAsB;IACxC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,YAAY,CAAU;gBAElB,UAAU,EAAE,mBAAmB,EAAE,IAAI,EAAE;QAAE,YAAY,EAAE,OAAO,CAAA;KAAE;IAMtE,QAAQ,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC;CAgE7E"}

package/dest/msg_validators/block_proposal_validator/block_proposal_validator.js

@@ -1,21 +1,70 @@
+import { NoCommitteeError } from '@aztec/ethereum';
+import { createLogger } from '@aztec/foundation/log';
 import { PeerErrorSeverity } from '@aztec/stdlib/p2p';
 export class BlockProposalValidator {
     epochCache;
-    constructor(epochCache){
+    logger;
+    txsPermitted;
+    constructor(epochCache, opts){
         this.epochCache = epochCache;
+        this.txsPermitted = opts.txsPermitted;
+        this.logger = createLogger('p2p:block_proposal_validator');
     }
     async validate(block) {
-        const { currentProposer, nextProposer, currentSlot, nextSlot } = await this.epochCache.getProposerInCurrentOrNextSlot();
-        // Check that the attestation is for the current or next slot
-        const slotNumberBigInt = block.payload.header.globalVariables.slotNumber.toBigInt();
-        if (slotNumberBigInt !== currentSlot && slotNumberBigInt !== nextSlot) {
-            return PeerErrorSeverity.HighToleranceError;
+        try {
+            // Check signature validity first - invalid signatures are a high-severity issue
+            const proposer = block.getSender();
+            if (!proposer) {
+                this.logger.debug(`Penalizing peer for block proposal with invalid signature`);
+                return PeerErrorSeverity.MidToleranceError;
+            }
+            // Check if transactions are permitted when the proposal contains transaction hashes
+            if (!this.txsPermitted && block.txHashes.length > 0) {
+                this.logger.debug(`Penalizing peer for block proposal with ${block.txHashes.length} transaction(s) when transactions are not permitted`);
+                return PeerErrorSeverity.MidToleranceError;
+            }
+            const { currentProposer, nextProposer, currentSlot, nextSlot } = await this.epochCache.getProposerAttesterAddressInCurrentOrNextSlot();
+            // Check that the attestation is for the current or next slot
+            const slotNumberBigInt = block.payload.header.slotNumber.toBigInt();
+            if (slotNumberBigInt !== currentSlot && slotNumberBigInt !== nextSlot) {
+                this.logger.debug(`Penalizing peer for invalid slot number ${slotNumberBigInt}`, {
+                    currentSlot,
+                    nextSlot
+                });
+                return PeerErrorSeverity.HighToleranceError;
+            }
+            // Check that the block proposal is from the current or next proposer
+            if (slotNumberBigInt === currentSlot && currentProposer !== undefined && !proposer.equals(currentProposer)) {
+                this.logger.debug(`Penalizing peer for invalid proposer for current slot ${slotNumberBigInt}`, {
+                    currentProposer,
+                    nextProposer,
+                    proposer: proposer.toString()
+                });
+                return PeerErrorSeverity.MidToleranceError;
+            }
+            if (slotNumberBigInt === nextSlot && nextProposer !== undefined && !proposer.equals(nextProposer)) {
+                this.logger.debug(`Penalizing peer for invalid proposer for next slot ${slotNumberBigInt}`, {
+                    currentProposer,
+                    nextProposer,
+                    proposer: proposer.toString()
+                });
+                return PeerErrorSeverity.MidToleranceError;
+            }
+            // Validate tx hashes for all txs embedded in the proposal
+            if (!(await Promise.all(block.txs?.map((tx)=>tx.validateTxHash()) ?? [])).every((v)=>v)) {
+                this.logger.warn(`Penalizing peer for invalid tx hashes in block proposal`, {
+                    proposer,
+                    slotNumber: slotNumberBigInt
+                });
+                return PeerErrorSeverity.LowToleranceError;
+            }
+            return undefined;
+        } catch (e) {
+            // People shouldn't be sending us block proposals if the committee doesn't exist
+            if (e instanceof NoCommitteeError) {
+                return PeerErrorSeverity.LowToleranceError;
+            }
+            throw e;
         }
-        // Check that the block proposal is from the current or next proposer
-        const proposer = await block.getSender();
-        if (!proposer.equals(currentProposer) && !proposer.equals(nextProposer)) {
-            return PeerErrorSeverity.HighToleranceError;
-        }
-        return undefined;
     }
 }

package/dest/msg_validators/msg_seen_validator/msg_seen_validator.d.ts

@@ -0,0 +1,10 @@
+export declare class MessageSeenValidator {
+    private queueLength;
+    private queue;
+    private writePointer;
+    private seenMessages;
+    constructor(queueLength: number);
+    addMessage(msgId: string): boolean;
+    size(): number;
+}
+//# sourceMappingURL=msg_seen_validator.d.ts.map

package/dest/msg_validators/msg_seen_validator/msg_seen_validator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"msg_seen_validator.d.ts","sourceRoot":"","sources":["../../../src/msg_validators/msg_seen_validator/msg_seen_validator.ts"],"names":[],"mappings":"AACA,qBAAa,oBAAoB;IAKnB,OAAO,CAAC,WAAW;IAJ/B,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,YAAY,CAAK;IACzB,OAAO,CAAC,YAAY,CAA0B;gBAE1B,WAAW,EAAE,MAAM;IAQhC,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAkBlC,IAAI;CAGZ"}

package/dest/msg_validators/msg_seen_validator/msg_seen_validator.js

@@ -0,0 +1,36 @@
+// Implements a queue of message IDs
+export class MessageSeenValidator {
+    queueLength;
+    queue;
+    writePointer;
+    seenMessages;
+    constructor(queueLength){
+        this.queueLength = queueLength;
+        this.writePointer = 0;
+        this.seenMessages = new Set();
+        if (this.queueLength <= 0) {
+            throw new Error('Queue length must be greater than 0');
+        }
+        this.queue = new Array(this.queueLength);
+    }
+    // Adds a message if not seen before. Returns true if added, false if already seen.
+    addMessage(msgId) {
+        // Check if the message is already in the cache
+        if (this.seenMessages.has(msgId)) {
+            return false;
+        }
+        // If we are at the cache limit, remove the oldest msg ID
+        if (this.seenMessages.size >= this.queueLength) {
+            const msgToRemove = this.queue[this.writePointer];
+            this.seenMessages.delete(msgToRemove);
+        }
+        // Insert the message into the cache and the queue
+        this.seenMessages.add(msgId);
+        this.queue[this.writePointer] = msgId;
+        this.writePointer = this.writePointer === this.queueLength - 1 ? 0 : this.writePointer + 1;
+        return true;
+    }
+    size() {
+        return this.seenMessages.size;
+    }
+}

package/dest/msg_validators/tx_validator/allowed_public_setup.d.ts

@@ -0,0 +1,3 @@
+import type { AllowedElement } from '@aztec/stdlib/interfaces/server';
+export declare function getDefaultAllowedSetupFunctions(): Promise<AllowedElement[]>;
+//# sourceMappingURL=allowed_public_setup.d.ts.map

package/dest/msg_validators/tx_validator/allowed_public_setup.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"allowed_public_setup.d.ts","sourceRoot":"","sources":["../../../src/msg_validators/tx_validator/allowed_public_setup.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AAGtE,wBAAsB,+BAA+B,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CA2BjF"}