@aztec/p2p 0.0.0-test.0

package/dest/config.js

@@ -0,0 +1,193 @@
+import { booleanConfigHelper, getConfigFromMappings, getDefaultConfig, numberConfigHelper, pickConfigMappings } from '@aztec/foundation/config';
+import { dataConfigMappings } from '@aztec/kv-store/config';
+import { chainConfigMappings } from '@aztec/stdlib/config';
+import { p2pReqRespConfigMappings } from './services/reqresp/config.js';
+export const p2pConfigMappings = {
+    p2pEnabled: {
+        env: 'P2P_ENABLED',
+        description: 'A flag dictating whether the P2P subsystem should be enabled.',
+        ...booleanConfigHelper()
+    },
+    blockCheckIntervalMS: {
+        env: 'P2P_BLOCK_CHECK_INTERVAL_MS',
+        description: 'The frequency in which to check for new L2 blocks.',
+        ...numberConfigHelper(100)
+    },
+    debugDisableColocationPenalty: {
+        env: 'DEBUG_P2P_DISABLE_COLOCATION_PENALTY',
+        description: 'DEBUG: Disable colocation penalty - NEVER set to true in production',
+        ...booleanConfigHelper(false)
+    },
+    peerCheckIntervalMS: {
+        env: 'P2P_PEER_CHECK_INTERVAL_MS',
+        description: 'The frequency in which to check for new peers.',
+        ...numberConfigHelper(30_000)
+    },
+    l2QueueSize: {
+        env: 'P2P_L2_QUEUE_SIZE',
+        description: 'Size of queue of L2 blocks to store.',
+        ...numberConfigHelper(1_000)
+    },
+    tcpListenAddress: {
+        env: 'P2P_TCP_LISTEN_ADDR',
+        defaultValue: '0.0.0.0:40400',
+        description: 'The listen address for TCP. Format: <IP_ADDRESS>:<PORT>.'
+    },
+    udpListenAddress: {
+        env: 'P2P_UDP_LISTEN_ADDR',
+        defaultValue: '0.0.0.0:40400',
+        description: 'The listen address for UDP. Format: <IP_ADDRESS>:<PORT>.'
+    },
+    tcpAnnounceAddress: {
+        env: 'P2P_TCP_ANNOUNCE_ADDR',
+        description: 'The announce address for TCP. Format: <IP_ADDRESS>:<PORT>. Leave IP_ADDRESS blank to query for public IP.'
+    },
+    udpAnnounceAddress: {
+        env: 'P2P_UDP_ANNOUNCE_ADDR',
+        description: 'The announce address for UDP. Format: <IP_ADDRESS>:<PORT>. Leave IP_ADDRESS blank to query for public IP.'
+    },
+    peerIdPrivateKey: {
+        env: 'PEER_ID_PRIVATE_KEY',
+        description: 'An optional peer id private key. If blank, will generate a random key.'
+    },
+    bootstrapNodes: {
+        env: 'BOOTSTRAP_NODES',
+        parseEnv: (val)=>val.split(','),
+        description: 'A list of bootstrap peer ENRs to connect to. Separated by commas.'
+    },
+    bootstrapNodeEnrVersionCheck: {
+        env: 'P2P_BOOTSTRAP_NODE_ENR_VERSION_CHECK',
+        description: 'Whether to check the version of the bootstrap node ENR.',
+        ...booleanConfigHelper()
+    },
+    bootstrapNodesAsFullPeers: {
+        env: 'P2P_BOOTSTRAP_NODES_AS_FULL_PEERS',
+        description: 'Whether to consider our configured bootnodes as full peers',
+        ...booleanConfigHelper(false)
+    },
+    maxPeerCount: {
+        env: 'P2P_MAX_PEERS',
+        description: 'The maximum number of peers to connect to.',
+        ...numberConfigHelper(100)
+    },
+    queryForIp: {
+        env: 'P2P_QUERY_FOR_IP',
+        description: 'If announceUdpAddress or announceTcpAddress are not provided, query for the IP address of the machine. Default is false.',
+        ...booleanConfigHelper()
+    },
+    keepProvenTxsInPoolFor: {
+        env: 'P2P_TX_POOL_KEEP_PROVEN_FOR',
+        description: 'How many blocks have to pass after a block is proven before its txs are deleted (zero to delete immediately once proven)',
+        ...numberConfigHelper(0)
+    },
+    keepAttestationsInPoolFor: {
+        env: 'P2P_ATTESTATION_POOL_KEEP_FOR',
+        description: 'How many slots to keep attestations for.',
+        ...numberConfigHelper(96)
+    },
+    gossipsubInterval: {
+        env: 'P2P_GOSSIPSUB_INTERVAL_MS',
+        description: 'The interval of the gossipsub heartbeat to perform maintenance tasks.',
+        ...numberConfigHelper(700)
+    },
+    gossipsubD: {
+        env: 'P2P_GOSSIPSUB_D',
+        description: 'The D parameter for the gossipsub protocol.',
+        ...numberConfigHelper(8)
+    },
+    gossipsubDlo: {
+        env: 'P2P_GOSSIPSUB_DLO',
+        description: 'The Dlo parameter for the gossipsub protocol.',
+        ...numberConfigHelper(4)
+    },
+    gossipsubDhi: {
+        env: 'P2P_GOSSIPSUB_DHI',
+        description: 'The Dhi parameter for the gossipsub protocol.',
+        ...numberConfigHelper(12)
+    },
+    gossipsubDLazy: {
+        env: 'P2P_GOSSIPSUB_DLAZY',
+        description: 'The Dlazy parameter for the gossipsub protocol.',
+        ...numberConfigHelper(8)
+    },
+    gossipsubFloodPublish: {
+        env: 'P2P_GOSSIPSUB_FLOOD_PUBLISH',
+        description: 'Whether to flood publish messages. - For testing purposes only',
+        ...booleanConfigHelper(true)
+    },
+    gossipsubMcacheLength: {
+        env: 'P2P_GOSSIPSUB_MCACHE_LENGTH',
+        description: 'The number of gossipsub interval message cache windows to keep.',
+        ...numberConfigHelper(6)
+    },
+    gossipsubMcacheGossip: {
+        env: 'P2P_GOSSIPSUB_MCACHE_GOSSIP',
+        description: 'How many message cache windows to include when gossiping with other pears.',
+        ...numberConfigHelper(3)
+    },
+    gossipsubTxTopicWeight: {
+        env: 'P2P_GOSSIPSUB_TX_TOPIC_WEIGHT',
+        description: 'The weight of the tx topic for the gossipsub protocol.',
+        ...numberConfigHelper(1)
+    },
+    gossipsubTxInvalidMessageDeliveriesWeight: {
+        env: 'P2P_GOSSIPSUB_TX_INVALID_MESSAGE_DELIVERIES_WEIGHT',
+        description: 'The weight of the tx invalid message deliveries for the gossipsub protocol.',
+        ...numberConfigHelper(-20)
+    },
+    gossipsubTxInvalidMessageDeliveriesDecay: {
+        env: 'P2P_GOSSIPSUB_TX_INVALID_MESSAGE_DELIVERIES_DECAY',
+        description: 'Determines how quickly the penalty for invalid message deliveries decays over time. Between 0 and 1.',
+        ...numberConfigHelper(0.5)
+    },
+    peerPenaltyValues: {
+        env: 'P2P_PEER_PENALTY_VALUES',
+        parseEnv: (val)=>val.split(',').map(Number),
+        description: 'The values for the peer scoring system. Passed as a comma separated list of values in order: low, mid, high tolerance errors.',
+        defaultValue: [
+            2,
+            10,
+            50
+        ]
+    },
+    doubleSpendSeverePeerPenaltyWindow: {
+        env: 'P2P_DOUBLE_SPEND_SEVERE_PEER_PENALTY_WINDOW',
+        description: 'The "age" (in L2 blocks) of a tx after which we heavily penalize a peer for sending it.',
+        ...numberConfigHelper(30)
+    },
+    blockRequestBatchSize: {
+        env: 'P2P_BLOCK_REQUEST_BATCH_SIZE',
+        description: 'The number of blocks to fetch in a single batch.',
+        ...numberConfigHelper(20)
+    },
+    archivedTxLimit: {
+        env: 'P2P_ARCHIVED_TX_LIMIT',
+        description: 'The number of transactions that will be archived. If the limit is set to 0 then archiving will be disabled.',
+        ...numberConfigHelper(0)
+    },
+    ...p2pReqRespConfigMappings,
+    ...chainConfigMappings
+};
+/**
+ * Gets the config values for p2p client from environment variables.
+ * @returns The config values for p2p client.
+ */ export function getP2PConfigFromEnv() {
+    return getConfigFromMappings(p2pConfigMappings);
+}
+export function getP2PDefaultConfig() {
+    return getDefaultConfig(p2pConfigMappings);
+}
+const bootnodeConfigKeys = [
+    'udpAnnounceAddress',
+    'peerIdPrivateKey',
+    'udpListenAddress',
+    'dataDirectory',
+    'dataStoreMapSizeKB',
+    'bootstrapNodes',
+    'l1ChainId'
+];
+export const bootnodeConfigMappings = pickConfigMappings({
+    ...p2pConfigMappings,
+    ...dataConfigMappings,
+    ...chainConfigMappings
+}, bootnodeConfigKeys);

package/dest/enr/generate-enr.d.ts

@@ -0,0 +1,9 @@
+import type { LogFn } from '@aztec/foundation/log';
+import { SignableENR } from '@chainsafe/enr';
+import type { PeerId } from '@libp2p/interface';
+export declare function createBootnodeENRandPeerId(privateKey: string, udpAnnounceAddress: string, l1ChainId: number): Promise<{
+    enr: SignableENR;
+    peerId: PeerId;
+}>;
+export declare function printENR(enr: string, log: LogFn): Promise<void>;
+//# sourceMappingURL=generate-enr.d.ts.map

package/dest/enr/generate-enr.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-enr.d.ts","sourceRoot":"","sources":["../../src/enr/generate-enr.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAGnD,OAAO,EAAO,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAOhD,wBAAsB,0BAA0B,CAC9C,UAAU,EAAE,MAAM,EAClB,kBAAkB,EAAE,MAAM,EAC1B,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC;IAAE,GAAG,EAAE,WAAW,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAAC,CAa/C;AAED,wBAAsB,QAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,iBAQrD"}

package/dest/enr/generate-enr.js

@@ -0,0 +1,30 @@
+import { emptyChainConfig } from '@aztec/stdlib/config';
+import { ENR, SignableENR } from '@chainsafe/enr';
+import { multiaddr } from '@multiformats/multiaddr';
+import { AZTEC_ENR_KEY } from '../types/index.js';
+import { convertToMultiaddr, createLibP2PPeerIdFromPrivateKey } from '../util.js';
+import { setAztecEnrKey } from '../versioning.js';
+export async function createBootnodeENRandPeerId(privateKey, udpAnnounceAddress, l1ChainId) {
+    const peerId = await createLibP2PPeerIdFromPrivateKey(privateKey);
+    const enr = SignableENR.createFromPeerId(peerId);
+    const publicAddr = multiaddr(convertToMultiaddr(udpAnnounceAddress, 'udp'));
+    enr.setLocationMultiaddr(publicAddr);
+    const config = {
+        ...emptyChainConfig,
+        l1ChainId
+    };
+    setAztecEnrKey(enr, config);
+    return {
+        enr,
+        peerId
+    };
+}
+export async function printENR(enr, log) {
+    const decoded = ENR.decodeTxt(enr);
+    log(`PeerID: ${await decoded.peerId()}`);
+    log(`IP: ${decoded.ip}`);
+    log(`UDP: ${decoded.udp}`);
+    log(`TCP: ${decoded.tcp}`);
+    const aztec = decoded.kvs.get(AZTEC_ENR_KEY);
+    log(`Aztec version: ${aztec?.toString()}`);
+}

package/dest/enr/index.d.ts

@@ -0,0 +1,2 @@
+export * from './generate-enr.js';
+//# sourceMappingURL=index.d.ts.map

package/dest/enr/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/enr/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC"}

package/dest/enr/index.js

@@ -0,0 +1 @@
+export * from './generate-enr.js';

package/dest/errors/reqresp.error.d.ts

@@ -0,0 +1,28 @@
+/** Individual request timeout error
+ *
+ * This error will be thrown when a request to a specific peer times out.
+ * @category Errors
+ */
+export declare class IndividualReqRespTimeoutError extends Error {
+    constructor();
+}
+/** Collective request timeout error
+ *
+ * This error will be thrown when a req resp request times out regardless of the peer.
+ * @category Errors
+ */
+export declare class CollectiveReqRespTimeoutError extends Error {
+    constructor();
+}
+/** Invalid response error
+ *
+ * This error will be thrown when a response is received that is not valid.
+ *
+ * This error does not need to be punished as message validators will handle punishing invalid
+ * requests
+ * @category Errors
+ */
+export declare class InvalidResponseError extends Error {
+    constructor();
+}
+//# sourceMappingURL=reqresp.error.d.ts.map

package/dest/errors/reqresp.error.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reqresp.error.d.ts","sourceRoot":"","sources":["../../src/errors/reqresp.error.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,qBAAa,6BAA8B,SAAQ,KAAK;;CAIvD;AAED;;;;GAIG;AACH,qBAAa,6BAA8B,SAAQ,KAAK;;CAIvD;AAED;;;;;;;GAOG;AACH,qBAAa,oBAAqB,SAAQ,KAAK;;CAI9C"}

package/dest/errors/reqresp.error.js

@@ -0,0 +1,30 @@
+/** Individual request timeout error
+ *
+ * This error will be thrown when a request to a specific peer times out.
+ * @category Errors
+ */ export class IndividualReqRespTimeoutError extends Error {
+    constructor(){
+        super(`Request to peer timed out`);
+    }
+}
+/** Collective request timeout error
+ *
+ * This error will be thrown when a req resp request times out regardless of the peer.
+ * @category Errors
+ */ export class CollectiveReqRespTimeoutError extends Error {
+    constructor(){
+        super(`Request to all peers timed out`);
+    }
+}
+/** Invalid response error
+ *
+ * This error will be thrown when a response is received that is not valid.
+ *
+ * This error does not need to be punished as message validators will handle punishing invalid
+ * requests
+ * @category Errors
+ */ export class InvalidResponseError extends Error {
+    constructor(){
+        super(`Invalid response received`);
+    }
+}

package/dest/index.d.ts

@@ -0,0 +1,8 @@
+export * from './bootstrap/bootstrap.js';
+export * from './client/index.js';
+export * from './config.js';
+export * from './mem_pools/attestation_pool/index.js';
+export * from './mem_pools/tx_pool/index.js';
+export * from './msg_validators/index.js';
+export * from './services/index.js';
+//# sourceMappingURL=index.d.ts.map

package/dest/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,0BAA0B,CAAC;AACzC,cAAc,mBAAmB,CAAC;AAClC,cAAc,aAAa,CAAC;AAC5B,cAAc,uCAAuC,CAAC;AACtD,cAAc,8BAA8B,CAAC;AAC7C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,qBAAqB,CAAC"}

package/dest/index.js

@@ -0,0 +1,7 @@
+export * from './bootstrap/bootstrap.js';
+export * from './client/index.js';
+export * from './config.js';
+export * from './mem_pools/attestation_pool/index.js';
+export * from './mem_pools/tx_pool/index.js';
+export * from './msg_validators/index.js';
+export * from './services/index.js';

package/dest/mem_pools/attestation_pool/attestation_pool.d.ts

@@ -0,0 +1,57 @@
+import type { BlockAttestation } from '@aztec/stdlib/p2p';
+/**
+ * An Attestation Pool contains attestations collected by a validator
+ *
+ * Attestations that are observed via the p2p network are stored for requests
+ * from the validator to produce a block, or to serve to other peers.
+ */
+export interface AttestationPool {
+    /**
+     * AddAttestation
+     *
+     * @param attestations - Attestations to add into the pool
+     */
+    addAttestations(attestations: BlockAttestation[]): Promise<void>;
+    /**
+     * DeleteAttestation
+     *
+     * @param attestations - Attestations to remove from the pool
+     */
+    deleteAttestations(attestations: BlockAttestation[]): Promise<void>;
+    /**
+     * Delete Attestations with a slot number smaller than the given slot
+     *
+     * Removes all attestations associated with a slot
+     *
+     * @param slot - The oldest slot to keep.
+     */
+    deleteAttestationsOlderThan(slot: bigint): Promise<void>;
+    /**
+     * Delete Attestations for slot
+     *
+     * Removes all attestations associated with a slot
+     *
+     * @param slot - The slot to delete.
+     */
+    deleteAttestationsForSlot(slot: bigint): Promise<void>;
+    /**
+     * Delete Attestations for slot and proposal
+     *
+     * Removes all attestations associated with a slot and proposal
+     *
+     * @param slot - The slot to delete.
+     * @param proposalId - The proposal to delete.
+     */
+    deleteAttestationsForSlotAndProposal(slot: bigint, proposalId: string): Promise<void>;
+    /**
+     * Get Attestations for slot
+     *
+     * Retrieve all of the attestations observed pertaining to a given slot
+     *
+     * @param slot - The slot to query
+     * @param proposalId - The proposal to query
+     * @return BlockAttestations
+     */
+    getAttestationsForSlot(slot: bigint, proposalId: string): Promise<BlockAttestation[]>;
+}
+//# sourceMappingURL=attestation_pool.d.ts.map

package/dest/mem_pools/attestation_pool/attestation_pool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"attestation_pool.d.ts","sourceRoot":"","sources":["../../../src/mem_pools/attestation_pool/attestation_pool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAE1D;;;;;GAKG;AACH,MAAM,WAAW,eAAe;IAC9B;;;;OAIG;IACH,eAAe,CAAC,YAAY,EAAE,gBAAgB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjE;;;;OAIG;IACH,kBAAkB,CAAC,YAAY,EAAE,gBAAgB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEpE;;;;;;OAMG;IACH,2BAA2B,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEzD;;;;;;OAMG;IACH,yBAAyB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEvD;;;;;;;OAOG;IACH,oCAAoC,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEtF;;;;;;;;OAQG;IACH,sBAAsB,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC;CACvF"}

package/dest/mem_pools/attestation_pool/attestation_pool.js

@@ -0,0 +1,6 @@
+/**
+ * An Attestation Pool contains attestations collected by a validator
+ *
+ * Attestations that are observed via the p2p network are stored for requests
+ * from the validator to produce a block, or to serve to other peers.
+ */ export { };

package/dest/mem_pools/attestation_pool/attestation_pool_test_suite.d.ts

@@ -0,0 +1,3 @@
+import type { AttestationPool } from './attestation_pool.js';
+export declare function describeAttestationPool(getAttestationPool: () => AttestationPool): void;
+//# sourceMappingURL=attestation_pool_test_suite.d.ts.map

package/dest/mem_pools/attestation_pool/attestation_pool_test_suite.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"attestation_pool_test_suite.d.ts","sourceRoot":"","sources":["../../../src/mem_pools/attestation_pool/attestation_pool_test_suite.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAK7D,wBAAgB,uBAAuB,CAAC,kBAAkB,EAAE,MAAM,eAAe,QAuNhF"}

package/dest/mem_pools/attestation_pool/attestation_pool_test_suite.js

@@ -0,0 +1,195 @@
+import { Secp256k1Signer } from '@aztec/foundation/crypto';
+import { Fr } from '@aztec/foundation/fields';
+import { TxHash } from '@aztec/stdlib/tx';
+import { jest } from '@jest/globals';
+import { mock } from 'jest-mock-extended';
+import { mockAttestation } from './mocks.js';
+const NUMBER_OF_SIGNERS_PER_TEST = 4;
+export function describeAttestationPool(getAttestationPool) {
+    let ap;
+    let signers;
+    // Check that metrics are recorded correctly
+    let metricsMock;
+    beforeEach(()=>{
+        ap = getAttestationPool();
+        signers = Array.from({
+            length: NUMBER_OF_SIGNERS_PER_TEST
+        }, ()=>Secp256k1Signer.random());
+        metricsMock = mock();
+        // Can i overwrite this like this??
+        ap.metrics = metricsMock;
+    });
+    const createAttestationsForSlot = (slotNumber)=>{
+        const archive = Fr.random();
+        return Promise.all(signers.map((signer)=>mockAttestation(signer, slotNumber, archive)));
+    };
+    // We compare buffers as the objects can have cached values attached to them which are not serialised
+    // using array containing as the kv store does not respect insertion order
+    const compareAttestations = (a1, a2)=>{
+        const a1Buffer = a1.map((attestation)=>attestation.toBuffer());
+        const a2Buffer = a2.map((attestation)=>attestation.toBuffer());
+        expect(a1Buffer.length).toBe(a2Buffer.length);
+        expect(a1Buffer).toEqual(expect.arrayContaining(a2Buffer));
+    };
+    it('should add attestations to pool', async ()=>{
+        const slotNumber = 420;
+        const archive = Fr.random();
+        const attestations = await Promise.all(signers.map((signer)=>mockAttestation(signer, slotNumber, archive)));
+        await ap.addAttestations(attestations);
+        // Check metrics have been updated.
+        expect(metricsMock.recordAddedObjects).toHaveBeenCalledWith(attestations.length);
+        const retreivedAttestations = await ap.getAttestationsForSlot(BigInt(slotNumber), archive.toString());
+        expect(retreivedAttestations.length).toBe(NUMBER_OF_SIGNERS_PER_TEST);
+        compareAttestations(retreivedAttestations, attestations);
+        // Delete by slot
+        await ap.deleteAttestationsForSlot(BigInt(slotNumber));
+        expect(metricsMock.recordRemovedObjects).toHaveBeenCalledWith(attestations.length);
+        const retreivedAttestationsAfterDelete = await ap.getAttestationsForSlot(BigInt(slotNumber), archive.toString());
+        expect(retreivedAttestationsAfterDelete.length).toBe(0);
+    });
+    it('Should handle duplicate proposals in a slot', async ()=>{
+        const slotNumber = 420;
+        const archive = Fr.random();
+        const txs = [
+            0,
+            1,
+            2,
+            3,
+            4,
+            5
+        ].map(()=>TxHash.random());
+        // Use the same signer for all attestations
+        const attestations = [];
+        const signer = signers[0];
+        for(let i = 0; i < NUMBER_OF_SIGNERS_PER_TEST; i++){
+            attestations.push(await mockAttestation(signer, slotNumber, archive, txs));
+        }
+        await ap.addAttestations(attestations);
+        const retreivedAttestations = await ap.getAttestationsForSlot(BigInt(slotNumber), archive.toString());
+        expect(retreivedAttestations.length).toBe(1);
+        expect(retreivedAttestations[0].toBuffer()).toEqual(attestations[0].toBuffer());
+        expect(retreivedAttestations[0].payload.txHashes).toEqual(txs);
+        expect((await retreivedAttestations[0].getSender()).toString()).toEqual(signer.address.toString());
+    });
+    it('Should store attestations by differing slot', async ()=>{
+        const slotNumbers = [
+            1,
+            2,
+            3,
+            4
+        ];
+        const attestations = await Promise.all(signers.map((signer, i)=>mockAttestation(signer, slotNumbers[i])));
+        await ap.addAttestations(attestations);
+        for (const attestation of attestations){
+            const slot = attestation.payload.header.globalVariables.slotNumber;
+            const archive = attestation.archive.toString();
+            const retreivedAttestations = await ap.getAttestationsForSlot(slot.toBigInt(), archive);
+            expect(retreivedAttestations.length).toBe(1);
+            expect(retreivedAttestations[0].toBuffer()).toEqual(attestation.toBuffer());
+            expect(retreivedAttestations[0].payload.header.globalVariables.slotNumber).toEqual(slot);
+        }
+    });
+    it('Should store attestations by differing slot and archive', async ()=>{
+        const slotNumbers = [
+            1,
+            1,
+            2,
+            3
+        ];
+        const archives = [
+            Fr.random(),
+            Fr.random(),
+            Fr.random(),
+            Fr.random()
+        ];
+        const attestations = await Promise.all(signers.map((signer, i)=>mockAttestation(signer, slotNumbers[i], archives[i])));
+        await ap.addAttestations(attestations);
+        for (const attestation of attestations){
+            const slot = attestation.payload.header.globalVariables.slotNumber;
+            const proposalId = attestation.archive.toString();
+            const retreivedAttestations = await ap.getAttestationsForSlot(slot.toBigInt(), proposalId);
+            expect(retreivedAttestations.length).toBe(1);
+            expect(retreivedAttestations[0].toBuffer()).toEqual(attestation.toBuffer());
+            expect(retreivedAttestations[0].payload.header.globalVariables.slotNumber).toEqual(slot);
+        }
+    });
+    it('Should delete attestations', async ()=>{
+        const slotNumber = 420;
+        const archive = Fr.random();
+        const attestations = await Promise.all(signers.map((signer)=>mockAttestation(signer, slotNumber, archive)));
+        const proposalId = attestations[0].archive.toString();
+        await ap.addAttestations(attestations);
+        expect(metricsMock.recordAddedObjects).toHaveBeenCalledWith(attestations.length);
+        const retreivedAttestations = await ap.getAttestationsForSlot(BigInt(slotNumber), proposalId);
+        expect(retreivedAttestations.length).toBe(NUMBER_OF_SIGNERS_PER_TEST);
+        compareAttestations(retreivedAttestations, attestations);
+        await ap.deleteAttestations(attestations);
+        expect(metricsMock.recordRemovedObjects).toHaveBeenCalledWith(attestations.length);
+        const gottenAfterDelete = await ap.getAttestationsForSlot(BigInt(slotNumber), proposalId);
+        expect(gottenAfterDelete.length).toBe(0);
+    });
+    it('Should blanket delete attestations per slot', async ()=>{
+        const slotNumber = 420;
+        const archive = Fr.random();
+        const attestations = await Promise.all(signers.map((signer)=>mockAttestation(signer, slotNumber, archive)));
+        const proposalId = attestations[0].archive.toString();
+        await ap.addAttestations(attestations);
+        const retreivedAttestations = await ap.getAttestationsForSlot(BigInt(slotNumber), proposalId);
+        expect(retreivedAttestations.length).toBe(NUMBER_OF_SIGNERS_PER_TEST);
+        compareAttestations(retreivedAttestations, attestations);
+        await ap.deleteAttestationsForSlot(BigInt(slotNumber));
+        const retreivedAttestationsAfterDelete = await ap.getAttestationsForSlot(BigInt(slotNumber), proposalId);
+        expect(retreivedAttestationsAfterDelete.length).toBe(0);
+    });
+    it('Should blanket delete attestations per slot and proposal', async ()=>{
+        const slotNumber = 420;
+        const archive = Fr.random();
+        const attestations = await Promise.all(signers.map((signer)=>mockAttestation(signer, slotNumber, archive)));
+        const proposalId = attestations[0].archive.toString();
+        // Add another set of attestations with a different proposalId, yet the same slot
+        const archive2 = Fr.random();
+        const attestations2 = await Promise.all(signers.map((signer)=>mockAttestation(signer, slotNumber, archive2)));
+        const proposalId2 = attestations2[0].archive.toString();
+        await ap.addAttestations(attestations);
+        await ap.addAttestations(attestations2);
+        expect(metricsMock.recordAddedObjects).toHaveBeenCalledWith(attestations.length);
+        expect(metricsMock.recordAddedObjects).toHaveBeenCalledWith(attestations2.length);
+        const retreivedAttestations = await ap.getAttestationsForSlot(BigInt(slotNumber), proposalId);
+        expect(retreivedAttestations.length).toBe(NUMBER_OF_SIGNERS_PER_TEST);
+        compareAttestations(retreivedAttestations, attestations);
+        await ap.deleteAttestationsForSlotAndProposal(BigInt(slotNumber), proposalId);
+        expect(metricsMock.recordRemovedObjects).toHaveBeenCalledWith(attestations.length);
+        const retreivedAttestationsAfterDelete = await ap.getAttestationsForSlot(BigInt(slotNumber), proposalId);
+        expect(retreivedAttestationsAfterDelete.length).toBe(0);
+        const retreivedAttestationsAfterDeleteForOtherProposal = await ap.getAttestationsForSlot(BigInt(slotNumber), proposalId2);
+        expect(retreivedAttestationsAfterDeleteForOtherProposal.length).toBe(NUMBER_OF_SIGNERS_PER_TEST);
+        compareAttestations(retreivedAttestationsAfterDeleteForOtherProposal, attestations2);
+    });
+    it('Should delete attestations older than a given slot', async ()=>{
+        const slotNumbers = [
+            1,
+            2,
+            3,
+            69,
+            72,
+            74,
+            88,
+            420
+        ];
+        const attestations = (await Promise.all(slotNumbers.map((slotNumber)=>createAttestationsForSlot(slotNumber)))).flat();
+        const proposalId = attestations[0].archive.toString();
+        await ap.addAttestations(attestations);
+        const attestationsForSlot1 = await ap.getAttestationsForSlot(BigInt(1), proposalId);
+        expect(attestationsForSlot1.length).toBe(signers.length);
+        const deleteAttestationsSpy = jest.spyOn(ap, 'deleteAttestationsForSlot');
+        await ap.deleteAttestationsOlderThan(BigInt(73));
+        const attestationsForSlot1AfterDelete = await ap.getAttestationsForSlot(BigInt(1), proposalId);
+        expect(attestationsForSlot1AfterDelete.length).toBe(0);
+        expect(deleteAttestationsSpy).toHaveBeenCalledTimes(5);
+        expect(deleteAttestationsSpy).toHaveBeenCalledWith(BigInt(1));
+        expect(deleteAttestationsSpy).toHaveBeenCalledWith(BigInt(2));
+        expect(deleteAttestationsSpy).toHaveBeenCalledWith(BigInt(3));
+        expect(deleteAttestationsSpy).toHaveBeenCalledWith(BigInt(69));
+        expect(deleteAttestationsSpy).toHaveBeenCalledWith(BigInt(72));
+    });
+}

package/dest/mem_pools/attestation_pool/index.d.ts

@@ -0,0 +1,3 @@
+export * from './attestation_pool.js';
+export * from './memory_attestation_pool.js';
+//# sourceMappingURL=index.d.ts.map

package/dest/mem_pools/attestation_pool/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/mem_pools/attestation_pool/index.ts"],"names":[],"mappings":"AAAA,cAAc,uBAAuB,CAAC;AACtC,cAAc,8BAA8B,CAAC"}

package/dest/mem_pools/attestation_pool/index.js

@@ -0,0 +1,2 @@
+export * from './attestation_pool.js';
+export * from './memory_attestation_pool.js';

package/dest/mem_pools/attestation_pool/kv_attestation_pool.d.ts

@@ -0,0 +1,22 @@
+import type { AztecAsyncKVStore } from '@aztec/kv-store';
+import { BlockAttestation } from '@aztec/stdlib/p2p';
+import { type TelemetryClient } from '@aztec/telemetry-client';
+import type { AttestationPool } from './attestation_pool.js';
+export declare class KvAttestationPool implements AttestationPool {
+    private store;
+    private log;
+    private metrics;
+    private attestations;
+    private proposalsForSlot;
+    private attestationsForProposal;
+    constructor(store: AztecAsyncKVStore, telemetry?: TelemetryClient, log?: import("@aztec/foundation/log").Logger);
+    private getProposalKey;
+    private getAttestationKey;
+    addAttestations(attestations: BlockAttestation[]): Promise<void>;
+    getAttestationsForSlot(slot: bigint, proposalId: string): Promise<BlockAttestation[]>;
+    deleteAttestationsOlderThan(oldestSlot: bigint): Promise<void>;
+    deleteAttestationsForSlot(slot: bigint): Promise<void>;
+    deleteAttestationsForSlotAndProposal(slot: bigint, proposalId: string): Promise<void>;
+    deleteAttestations(attestations: BlockAttestation[]): Promise<void>;
+}
+//# sourceMappingURL=kv_attestation_pool.d.ts.map

package/dest/mem_pools/attestation_pool/kv_attestation_pool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"kv_attestation_pool.d.ts","sourceRoot":"","sources":["../../../src/mem_pools/attestation_pool/kv_attestation_pool.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,iBAAiB,EAAqC,MAAM,iBAAiB,CAAC;AAC5F,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,KAAK,eAAe,EAAsB,MAAM,yBAAyB,CAAC;AAGnF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAE7D,qBAAa,iBAAkB,YAAW,eAAe;IAQrD,OAAO,CAAC,KAAK;IAEb,OAAO,CAAC,GAAG;IATb,OAAO,CAAC,OAAO,CAAwC;IAEvD,OAAO,CAAC,YAAY,CAAgC;IACpD,OAAO,CAAC,gBAAgB,CAAqC;IAC7D,OAAO,CAAC,uBAAuB,CAAqC;gBAG1D,KAAK,EAAE,iBAAiB,EAChC,SAAS,GAAE,eAAsC,EACzC,GAAG,yCAAyC;IAStD,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,iBAAiB;IAIZ,eAAe,CAAC,YAAY,EAAE,gBAAgB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAsBhE,sBAAsB,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAsBrF,2BAA2B,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAO9D,yBAAyB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAuBtD,oCAAoC,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBrF,kBAAkB,CAAC,YAAY,EAAE,gBAAgB,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAkBjF"}