@aztec/p2p 0.0.0-test.0

package/dest/services/service.d.ts

@@ -0,0 +1,87 @@
+/// <reference types="node" resolution-mode="require"/>
+import type { PeerInfo } from '@aztec/stdlib/interfaces/server';
+import type { BlockAttestation, BlockProposal, Gossipable } from '@aztec/stdlib/p2p';
+import type { ENR } from '@chainsafe/enr';
+import type { PeerId } from '@libp2p/interface';
+import type EventEmitter from 'events';
+import type { ReqRespSubProtocol, SubProtocolMap } from './reqresp/interface.js';
+export declare enum PeerDiscoveryState {
+    RUNNING = "running",
+    STOPPED = "stopped"
+}
+/**
+ * The interface for a P2P service implementation.
+ */
+export interface P2PService {
+    /**
+     * Starts the service.
+     * @returns An empty promise.
+     */
+    start(): Promise<void>;
+    /**
+     * Stops the service.
+     * @returns An empty promise.
+     */
+    stop(): Promise<void>;
+    /**
+     * Called to have the given transaction propagated through the P2P network.
+     * @param message - The message to be propagated.
+     */
+    propagate<T extends Gossipable>(message: T): void;
+    /**
+     * Request information from peers via the request response protocol.
+     *
+     * @param protocol - The request response protocol to use
+     * @param request - The request type, corresponding to the protocol
+     * @returns The response type, corresponding to the protocol
+     */
+    sendRequest<Protocol extends ReqRespSubProtocol>(protocol: Protocol, request: InstanceType<SubProtocolMap[Protocol]['request']>): Promise<InstanceType<SubProtocolMap[Protocol]['response']> | undefined>;
+    /**
+     * Send a batch of requests to peers, and return the responses
+     *
+     * @param protocol - The request response protocol to use
+     * @param requests - The requests to send to the peers
+     * @returns The responses to the requests
+     */
+    sendBatchRequest<Protocol extends ReqRespSubProtocol>(protocol: Protocol, requests: InstanceType<SubProtocolMap[Protocol]['request']>[]): Promise<InstanceType<SubProtocolMap[Protocol]['response']>[] | undefined>;
+    registerBlockReceivedCallback(callback: (block: BlockProposal) => Promise<BlockAttestation | undefined>): void;
+    getEnr(): ENR | undefined;
+    getPeers(includePending?: boolean): PeerInfo[];
+}
+/**
+ * The interface for a peer discovery service implementation.
+ */
+export interface PeerDiscoveryService extends EventEmitter {
+    /**
+     * Starts the service.
+     * */
+    start(): Promise<void>;
+    /**
+     * Stops the service.
+     * */
+    stop(): Promise<void>;
+    /**
+     * Gets all peers.
+     * @returns An array of peer ENRs.
+     */
+    getAllPeers(): ENR[];
+    /**
+     * Runs findRandomNode query.
+     */
+    runRandomNodesQuery(): Promise<void>;
+    /**
+     * Checks if the given peer is a bootstrap peer.
+     * @param peerId - The peer ID to check.
+     * @returns True if the peer is a bootstrap peer.
+     */
+    isBootstrapPeer(peerId: PeerId): boolean;
+    /**
+     * Event emitted when a new peer is discovered.
+     */
+    on(event: 'peer:discovered', listener: (enr: ENR) => void): this;
+    emit(event: 'peer:discovered', enr: ENR): boolean;
+    getStatus(): PeerDiscoveryState;
+    getEnr(): ENR | undefined;
+    bootstrapNodes: string[];
+}
+//# sourceMappingURL=service.d.ts.map

package/dest/services/service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.d.ts","sourceRoot":"","sources":["../../src/services/service.ts"],"names":[],"mappings":";AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAChE,OAAO,KAAK,EAAE,gBAAgB,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAErF,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,YAAY,MAAM,QAAQ,CAAC;AAEvC,OAAO,KAAK,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAEjF,oBAAY,kBAAkB;IAC5B,OAAO,YAAY;IACnB,OAAO,YAAY;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB;;;OAGG;IACH,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAEvB;;;OAGG;IACH,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAEtB;;;OAGG;IACH,SAAS,CAAC,CAAC,SAAS,UAAU,EAAE,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC;IAElD;;;;;;OAMG;IACH,WAAW,CAAC,QAAQ,SAAS,kBAAkB,EAC7C,QAAQ,EAAE,QAAQ,EAClB,OAAO,EAAE,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,CAAC,GACzD,OAAO,CAAC,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC;IAE3E;;;;;;OAMG;IACH,gBAAgB,CAAC,QAAQ,SAAS,kBAAkB,EAClD,QAAQ,EAAE,QAAQ,EAClB,QAAQ,EAAE,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,GAC5D,OAAO,CAAC,YAAY,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,GAAG,SAAS,CAAC,CAAC;IAG7E,6BAA6B,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,aAAa,KAAK,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;IAE/G,MAAM,IAAI,GAAG,GAAG,SAAS,CAAC;IAE1B,QAAQ,CAAC,cAAc,CAAC,EAAE,OAAO,GAAG,QAAQ,EAAE,CAAC;CAChD;AAED;;GAEG;AACH,MAAM,WAAW,oBAAqB,SAAQ,YAAY;IACxD;;SAEK;IACL,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAEvB;;SAEK;IACL,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAEtB;;;OAGG;IACH,WAAW,IAAI,GAAG,EAAE,CAAC;IAErB;;OAEG;IACH,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAErC;;;;OAIG;IACH,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;IAEzC;;OAEG;IACH,EAAE,CAAC,KAAK,EAAE,iBAAiB,EAAE,QAAQ,EAAE,CAAC,GAAG,EAAE,GAAG,KAAK,IAAI,GAAG,IAAI,CAAC;IACjE,IAAI,CAAC,KAAK,EAAE,iBAAiB,EAAE,GAAG,EAAE,GAAG,GAAG,OAAO,CAAC;IAElD,SAAS,IAAI,kBAAkB,CAAC;IAEhC,MAAM,IAAI,GAAG,GAAG,SAAS,CAAC;IAE1B,cAAc,EAAE,MAAM,EAAE,CAAC;CAC1B"}

package/dest/services/service.js

@@ -0,0 +1,5 @@
+export var PeerDiscoveryState = /*#__PURE__*/ function(PeerDiscoveryState) {
+    PeerDiscoveryState["RUNNING"] = "running";
+    PeerDiscoveryState["STOPPED"] = "stopped";
+    return PeerDiscoveryState;
+}({});

package/dest/test-helpers/generate-peer-id-private-keys.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Generate a list of peer id private keys
+ * @param numberOfPeers - The number of peer id private keys to generate
+ * @returns A list of peer id private keys
+ */
+export declare function generatePeerIdPrivateKeys(numberOfPeers: number): string[];
+//# sourceMappingURL=generate-peer-id-private-keys.d.ts.map

package/dest/test-helpers/generate-peer-id-private-keys.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-peer-id-private-keys.d.ts","sourceRoot":"","sources":["../../src/test-helpers/generate-peer-id-private-keys.ts"],"names":[],"mappings":"AAEA;;;;GAIG;AACH,wBAAgB,yBAAyB,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM,EAAE,CAOzE"}

package/dest/test-helpers/generate-peer-id-private-keys.js

@@ -0,0 +1,13 @@
+import { generatePrivateKey } from 'viem/accounts';
+/**
+ * Generate a list of peer id private keys
+ * @param numberOfPeers - The number of peer id private keys to generate
+ * @returns A list of peer id private keys
+ */ export function generatePeerIdPrivateKeys(numberOfPeers) {
+    const peerIdPrivateKeys = [];
+    for(let i = 0; i < numberOfPeers; i++){
+        // magic number is multiaddr prefix: https://multiformats.io/multiaddr/
+        peerIdPrivateKeys.push('08021220' + generatePrivateKey().slice(2, 68));
+    }
+    return peerIdPrivateKeys;
+}

package/dest/test-helpers/get-ports.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Get a list of ports for a given number of peers
+ * @param numberOfPeers - The number of peers to get ports for
+ * @returns A list of ports
+ */
+export declare const getPorts: (numberOfPeers: number) => Promise<number[]>;
+//# sourceMappingURL=get-ports.d.ts.map

package/dest/test-helpers/get-ports.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-ports.d.ts","sourceRoot":"","sources":["../../src/test-helpers/get-ports.ts"],"names":[],"mappings":"AAEA;;;;GAIG;AACH,eAAO,MAAM,QAAQ,kBAAmB,MAAM,sBAAwE,CAAC"}

package/dest/test-helpers/get-ports.js

@@ -0,0 +1,8 @@
+import getPort from 'get-port';
+/**
+ * Get a list of ports for a given number of peers
+ * @param numberOfPeers - The number of peers to get ports for
+ * @returns A list of ports
+ */ export const getPorts = (numberOfPeers)=>Promise.all(Array.from({
+        length: numberOfPeers
+    }, ()=>getPort()));

package/dest/test-helpers/index.d.ts

@@ -0,0 +1,6 @@
+export * from './generate-peer-id-private-keys.js';
+export * from './get-ports.js';
+export * from './make-enrs.js';
+export * from './make-test-p2p-clients.js';
+export * from './reqresp-nodes.js';
+//# sourceMappingURL=index.d.ts.map

package/dest/test-helpers/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/test-helpers/index.ts"],"names":[],"mappings":"AAAA,cAAc,oCAAoC,CAAC;AACnD,cAAc,gBAAgB,CAAC;AAC/B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,4BAA4B,CAAC;AAC3C,cAAc,oBAAoB,CAAC"}

package/dest/test-helpers/index.js

@@ -0,0 +1,5 @@
+export * from './generate-peer-id-private-keys.js';
+export * from './get-ports.js';
+export * from './make-enrs.js';
+export * from './make-test-p2p-clients.js';
+export * from './reqresp-nodes.js';

package/dest/test-helpers/make-enrs.d.ts

@@ -0,0 +1,16 @@
+import type { ChainConfig } from '@aztec/stdlib/config';
+/**
+ * Make a list of ENRs for a given list of p2p private keys and ports
+ * @param p2pPrivateKeys - The private keys of the p2p nodes
+ * @param ports - The ports of the p2p nodes
+ * @returns A list of ENRs
+ */
+export declare function makeEnrs(p2pPrivateKeys: string[], ports: number[], config: ChainConfig): Promise<string[]>;
+/**
+ * Make an ENR for a given p2p private key and port
+ * @param p2pPrivateKey - The private key of the p2p node
+ * @param port - The port of the p2p node
+ * @returns The ENR of the p2p node
+ */
+export declare function makeEnr(p2pPrivateKey: string, port: number, config: ChainConfig): Promise<string>;
+//# sourceMappingURL=make-enrs.d.ts.map

package/dest/test-helpers/make-enrs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"make-enrs.d.ts","sourceRoot":"","sources":["../../src/test-helpers/make-enrs.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAQxD;;;;;GAKG;AACH,wBAAsB,QAAQ,CAAC,cAAc,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,WAAW,qBAM5F;AAED;;;;;GAKG;AACH,wBAAsB,OAAO,CAAC,aAAa,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,mBAerF"}

package/dest/test-helpers/make-enrs.js

@@ -0,0 +1,32 @@
+import { SignableENR } from '@chainsafe/enr';
+import { multiaddr } from '@multiformats/multiaddr';
+import { convertToMultiaddr, createLibP2PPeerIdFromPrivateKey } from '../util.js';
+import { setAztecEnrKey } from '../versioning.js';
+/**
+ * Make a list of ENRs for a given list of p2p private keys and ports
+ * @param p2pPrivateKeys - The private keys of the p2p nodes
+ * @param ports - The ports of the p2p nodes
+ * @returns A list of ENRs
+ */ export async function makeEnrs(p2pPrivateKeys, ports, config) {
+    return await Promise.all(p2pPrivateKeys.map((pk, i)=>{
+        return makeEnr(pk, ports[i], config);
+    }));
+}
+/**
+ * Make an ENR for a given p2p private key and port
+ * @param p2pPrivateKey - The private key of the p2p node
+ * @param port - The port of the p2p node
+ * @returns The ENR of the p2p node
+ */ export async function makeEnr(p2pPrivateKey, port, config) {
+    const peerId = await createLibP2PPeerIdFromPrivateKey(p2pPrivateKey);
+    const enr = SignableENR.createFromPeerId(peerId);
+    const udpAnnounceAddress = `127.0.0.1:${port}`;
+    const tcpAnnounceAddress = `127.0.0.1:${port}`;
+    const udpPublicAddr = multiaddr(convertToMultiaddr(udpAnnounceAddress, 'udp'));
+    const tcpPublicAddr = multiaddr(convertToMultiaddr(tcpAnnounceAddress, 'tcp'));
+    // ENRS must include the network and a discoverable address (udp for discv5)
+    setAztecEnrKey(enr, config);
+    enr.setLocationMultiaddr(udpPublicAddr);
+    enr.setLocationMultiaddr(tcpPublicAddr);
+    return enr.encodeTxt();
+}

package/dest/test-helpers/make-test-p2p-clients.d.ts

@@ -0,0 +1,36 @@
+import type { EpochCache } from '@aztec/epoch-cache';
+import { type Logger } from '@aztec/foundation/log';
+import type { WorldStateSynchronizer } from '@aztec/stdlib/interfaces/server';
+import { P2PClientType } from '@aztec/stdlib/p2p';
+import type { P2PClient } from '../client/p2p_client.js';
+import type { P2PConfig } from '../config.js';
+import type { AttestationPool } from '../mem_pools/attestation_pool/attestation_pool.js';
+import type { TxPool } from '../mem_pools/tx_pool/index.js';
+interface MakeTestP2PClientOptions {
+    mockAttestationPool: AttestationPool;
+    mockTxPool: TxPool;
+    mockEpochCache: EpochCache;
+    mockWorldState: WorldStateSynchronizer;
+    alwaysTrueVerifier?: boolean;
+    p2pBaseConfig: P2PConfig;
+    p2pConfigOverrides?: Partial<P2PConfig>;
+    logger?: Logger;
+}
+/**
+ * Creates a single P2P client for testing purposes.
+ * @param peerIdPrivateKey - The private key of the peer.
+ * @param port - The port to run the client on.
+ * @param peers - The peers to connect to.
+ * @param options - The options for the client.
+ * @returns The created client.
+ */
+export declare function makeTestP2PClient(peerIdPrivateKey: string, port: number, peers: string[], { alwaysTrueVerifier, p2pBaseConfig, p2pConfigOverrides, mockAttestationPool, mockTxPool, mockEpochCache, mockWorldState, logger, }: MakeTestP2PClientOptions): Promise<P2PClient<P2PClientType.Full>>;
+/**
+ * Creates a number of P2P clients for testing purposes.
+ * @param numberOfPeers - The number of clients to create.
+ * @param options - The options for the clients.
+ * @returns The created clients.
+ */
+export declare function makeTestP2PClients(numberOfPeers: number, testConfig: MakeTestP2PClientOptions): Promise<P2PClient<P2PClientType.Full>[]>;
+export {};
+//# sourceMappingURL=make-test-p2p-clients.d.ts.map

package/dest/test-helpers/make-test-p2p-clients.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"make-test-p2p-clients.d.ts","sourceRoot":"","sources":["../../src/test-helpers/make-test-p2p-clients.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,KAAK,MAAM,EAAgB,MAAM,uBAAuB,CAAC;AAGlE,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AAC9E,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGlD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mDAAmD,CAAC;AACzF,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAC;AAM5D,UAAU,wBAAwB;IAChC,mBAAmB,EAAE,eAAe,CAAC;IACrC,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,UAAU,CAAC;IAC3B,cAAc,EAAE,sBAAsB,CAAC;IACvC,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,aAAa,EAAE,SAAS,CAAC;IACzB,kBAAkB,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;IACxC,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;;;;;GAOG;AACH,wBAAsB,iBAAiB,CACrC,gBAAgB,EAAE,MAAM,EACxB,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EAAE,EACf,EACE,kBAAyB,EACzB,aAAa,EACb,kBAAuB,EACvB,mBAAmB,EACnB,UAAU,EACV,cAAc,EACd,cAAc,EACd,MAAwC,GACzC,EAAE,wBAAwB,0CA8C5B;AAED;;;;;GAKG;AACH,wBAAsB,kBAAkB,CAAC,aAAa,EAAE,MAAM,EAAE,UAAU,EAAE,wBAAwB,4CAiBnG"}

package/dest/test-helpers/make-test-p2p-clients.js

@@ -0,0 +1,68 @@
+import { MockL2BlockSource } from '@aztec/archiver/test';
+import { createLogger } from '@aztec/foundation/log';
+import { openTmpStore } from '@aztec/kv-store/lmdb-v2';
+import { P2PClientType } from '@aztec/stdlib/p2p';
+import { createP2PClient } from '../client/index.js';
+import { generatePeerIdPrivateKeys } from '../test-helpers/generate-peer-id-private-keys.js';
+import { getPorts } from './get-ports.js';
+import { makeEnrs } from './make-enrs.js';
+import { AlwaysFalseCircuitVerifier, AlwaysTrueCircuitVerifier } from './reqresp-nodes.js';
+/**
+ * Creates a single P2P client for testing purposes.
+ * @param peerIdPrivateKey - The private key of the peer.
+ * @param port - The port to run the client on.
+ * @param peers - The peers to connect to.
+ * @param options - The options for the client.
+ * @returns The created client.
+ */ export async function makeTestP2PClient(peerIdPrivateKey, port, peers, { alwaysTrueVerifier = true, p2pBaseConfig, p2pConfigOverrides = {}, mockAttestationPool, mockTxPool, mockEpochCache, mockWorldState, logger = createLogger('p2p-test-client') }) {
+    const addr = `127.0.0.1:${port}`;
+    const listenAddr = `0.0.0.0:${port}`;
+    // Filter nodes so that we only dial active peers
+    const config = {
+        ...p2pBaseConfig,
+        p2pEnabled: true,
+        peerIdPrivateKey,
+        tcpListenAddress: listenAddr,
+        udpListenAddress: listenAddr,
+        tcpAnnounceAddress: addr,
+        udpAnnounceAddress: addr,
+        bootstrapNodes: peers,
+        peerCheckIntervalMS: 1000,
+        maxPeerCount: 10,
+        bootstrapNodesAsFullPeers: true,
+        ...p2pConfigOverrides
+    };
+    const l2BlockSource = new MockL2BlockSource();
+    await l2BlockSource.createBlocks(100);
+    const proofVerifier = alwaysTrueVerifier ? new AlwaysTrueCircuitVerifier() : new AlwaysFalseCircuitVerifier();
+    const kvStore = await openTmpStore('test');
+    const deps = {
+        txPool: mockTxPool,
+        attestationPool: mockAttestationPool,
+        store: kvStore,
+        logger
+    };
+    const client = await createP2PClient(P2PClientType.Full, config, l2BlockSource, proofVerifier, mockWorldState, mockEpochCache, undefined, deps);
+    await client.start();
+    return client;
+}
+/**
+ * Creates a number of P2P clients for testing purposes.
+ * @param numberOfPeers - The number of clients to create.
+ * @param options - The options for the clients.
+ * @returns The created clients.
+ */ export async function makeTestP2PClients(numberOfPeers, testConfig) {
+    const clients = [];
+    const peerIdPrivateKeys = generatePeerIdPrivateKeys(numberOfPeers);
+    const ports = await getPorts(numberOfPeers);
+    const peerEnrs = await makeEnrs(peerIdPrivateKeys, ports, testConfig.p2pBaseConfig);
+    for(let i = 0; i < numberOfPeers; i++){
+        const client = await makeTestP2PClient(peerIdPrivateKeys[i], ports[i], peerEnrs, {
+            ...testConfig,
+            logger: createLogger(`p2p:${i}`)
+        });
+        clients.push(client);
+    }
+    await Promise.all(clients.map((client)=>client.isReady()));
+    return clients;
+}

package/dest/test-helpers/reqresp-nodes.d.ts

@@ -0,0 +1,66 @@
+import type { EpochCache } from '@aztec/epoch-cache';
+import type { L2BlockSource } from '@aztec/stdlib/block';
+import { type ChainConfig } from '@aztec/stdlib/config';
+import type { ClientProtocolCircuitVerifier, WorldStateSynchronizer } from '@aztec/stdlib/interfaces/server';
+import type { P2PClientType } from '@aztec/stdlib/p2p';
+import type { Tx } from '@aztec/stdlib/tx';
+import { type TelemetryClient } from '@aztec/telemetry-client';
+import { SignableENR } from '@chainsafe/enr';
+import type { PeerId } from '@libp2p/interface';
+import { type Libp2p } from 'libp2p';
+import { BootstrapNode } from '../bootstrap/bootstrap.js';
+import type { BootnodeConfig } from '../config.js';
+import type { MemPools } from '../mem_pools/interface.js';
+import { LibP2PService } from '../services/libp2p/libp2p_service.js';
+import type { PeerScoring } from '../services/peer-manager/peer_scoring.js';
+import { type ReqRespSubProtocolHandlers, type ReqRespSubProtocolValidators } from '../services/reqresp/interface.js';
+import { ReqResp } from '../services/reqresp/reqresp.js';
+/**
+ * Creates a libp2p node, pre configured.
+ * @param boostrapAddrs - an optional list of bootstrap addresses
+ * @returns Lip2p node
+ */
+export declare function createLibp2pNode(boostrapAddrs?: string[], peerId?: PeerId, port?: number, enableGossipSub?: boolean, start?: boolean): Promise<Libp2p>;
+/**
+ * Test Libp2p service
+ * P2P functionality is operational, however everything else is default
+ *
+ *
+ */
+export declare function createTestLibP2PService<T extends P2PClientType>(clientType: T, boostrapAddrs: string[] | undefined, l2BlockSource: L2BlockSource, worldStateSynchronizer: WorldStateSynchronizer, epochCache: EpochCache, mempools: MemPools<T>, telemetry: TelemetryClient, port?: number, peerId?: PeerId, chainConfig?: ChainConfig): Promise<LibP2PService<T>>;
+/**
+ * A p2p / req resp node pairing the req node will always contain the p2p node.
+ * they are provided as a pair to allow access the p2p node directly
+ */
+export type ReqRespNode = {
+    p2p: Libp2p;
+    req: ReqResp;
+};
+export declare const MOCK_SUB_PROTOCOL_HANDLERS: ReqRespSubProtocolHandlers;
+export declare const MOCK_SUB_PROTOCOL_VALIDATORS: ReqRespSubProtocolValidators;
+/**
+ * @param numberOfNodes - the number of nodes to create
+ * @returns An array of the created nodes
+ */
+export declare const createNodes: (peerScoring: PeerScoring, numberOfNodes: number) => Promise<ReqRespNode[]>;
+export declare const startNodes: (nodes: ReqRespNode[], subProtocolHandlers?: ReqRespSubProtocolHandlers, subProtocolValidators?: ReqRespSubProtocolValidators) => Promise<void>;
+export declare const stopNodes: (nodes: ReqRespNode[]) => Promise<void>;
+export declare const createReqResp: (peerScoring: PeerScoring) => Promise<ReqRespNode>;
+export declare const connectToPeers: (nodes: ReqRespNode[]) => Promise<void>;
+export declare class AlwaysTrueCircuitVerifier implements ClientProtocolCircuitVerifier {
+    verifyProof(_tx: Tx): Promise<boolean>;
+}
+export declare class AlwaysFalseCircuitVerifier implements ClientProtocolCircuitVerifier {
+    verifyProof(_tx: Tx): Promise<boolean>;
+}
+export declare function createBootstrapNodeConfig(privateKey: string, port: number, chainConfig: ChainConfig): BootnodeConfig;
+export declare function createBootstrapNodeFromPrivateKey(privateKey: string, port: number, telemetry?: TelemetryClient, chainConfig?: ChainConfig): Promise<BootstrapNode>;
+/**
+ * Create a bootstrap node ENR
+ * @param privateKey - the private key of the bootstrap node
+ * @param port - the port of the bootstrap node
+ * @returns the bootstrap node ENR
+ */
+export declare function getBootstrapNodeEnr(privateKey: string, port: number): Promise<SignableENR>;
+export declare function createBootstrapNode(port: number, telemetry?: TelemetryClient, chainConfig?: ChainConfig): Promise<BootstrapNode>;
+//# sourceMappingURL=reqresp-nodes.d.ts.map

package/dest/test-helpers/reqresp-nodes.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"reqresp-nodes.d.ts","sourceRoot":"","sources":["../../src/test-helpers/reqresp-nodes.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAIrD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,EAAE,KAAK,WAAW,EAAoB,MAAM,sBAAsB,CAAC;AAC1E,OAAO,KAAK,EAAE,6BAA6B,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AAC7G,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,KAAK,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AAC3C,OAAO,EAAE,KAAK,eAAe,EAAsB,MAAM,yBAAyB,CAAC;AAEnF,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAM7C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAKhD,OAAO,EAAE,KAAK,MAAM,EAAoC,MAAM,QAAQ,CAAC;AAEvE,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,KAAK,EAAE,cAAc,EAAa,MAAM,cAAc,CAAC;AAC9D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,sCAAsC,CAAC;AACrE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0CAA0C,CAAC;AAE5E,OAAO,EAEL,KAAK,0BAA0B,EAC/B,KAAK,4BAA4B,EAElC,MAAM,kCAAkC,CAAC;AAE1C,OAAO,EAAE,OAAO,EAAE,MAAM,gCAAgC,CAAC;AAGzD;;;;GAIG;AACH,wBAAsB,gBAAgB,CACpC,aAAa,GAAE,MAAM,EAAO,EAC5B,MAAM,CAAC,EAAE,MAAM,EACf,IAAI,CAAC,EAAE,MAAM,EACb,eAAe,GAAE,OAAe,EAChC,KAAK,GAAE,OAAc,GACpB,OAAO,CAAC,MAAM,CAAC,CAqCjB;AAED;;;;;GAKG;AACH,wBAAsB,uBAAuB,CAAC,CAAC,SAAS,aAAa,EACnE,UAAU,EAAE,CAAC,EACb,aAAa,sBAAe,EAC5B,aAAa,EAAE,aAAa,EAC5B,sBAAsB,EAAE,sBAAsB,EAC9C,UAAU,EAAE,UAAU,EACtB,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,EACrB,SAAS,EAAE,eAAe,EAC1B,IAAI,GAAE,MAAU,EAChB,MAAM,CAAC,EAAE,MAAM,EACf,WAAW,GAAE,WAA8B,6BAkC5C;AAED;;;GAGG;AACH,MAAM,MAAM,WAAW,GAAG;IACxB,GAAG,EAAE,MAAM,CAAC;IACZ,GAAG,EAAE,OAAO,CAAC;CACd,CAAC;AAGF,eAAO,MAAM,0BAA0B,EAAE,0BAMxC,CAAC;AAIF,eAAO,MAAM,4BAA4B,EAAE,4BAM1C,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,WAAW,gBAAiB,WAAW,iBAAiB,MAAM,KAAG,QAAQ,WAAW,EAAE,CAElG,CAAC;AAEF,eAAO,MAAM,UAAU,UACd,WAAW,EAAE,0HAOrB,CAAC;AAEF,eAAO,MAAM,SAAS,UAAiB,WAAW,EAAE,KAAG,QAAQ,IAAI,CAGlE,CAAC;AAGF,eAAO,MAAM,aAAa,gBAAuB,WAAW,KAAG,QAAQ,WAAW,CAWjF,CAAC;AAGF,eAAO,MAAM,cAAc,UAAiB,WAAW,EAAE,KAAG,QAAQ,IAAI,CAUvE,CAAC;AAGF,qBAAa,yBAA0B,YAAW,6BAA6B;IAC7E,WAAW,CAAC,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;CAGvC;AACD,qBAAa,0BAA2B,YAAW,6BAA6B;IAC9E,WAAW,CAAC,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,OAAO,CAAC;CAGvC;AAGD,wBAAgB,yBAAyB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,WAAW,GAAG,cAAc,CAUpH;AAED,wBAAgB,iCAAiC,CAC/C,UAAU,EAAE,MAAM,EAClB,IAAI,EAAE,MAAM,EACZ,SAAS,GAAE,eAAsC,EACjD,WAAW,GAAE,WAA8B,GAC1C,OAAO,CAAC,aAAa,CAAC,CAGxB;AAED;;;;;GAKG;AACH,wBAAsB,mBAAmB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,wBAOzE;AAED,wBAAsB,mBAAmB,CACvC,IAAI,EAAE,MAAM,EACZ,SAAS,GAAE,eAAsC,EACjD,WAAW,GAAE,WAA8B,GAC1C,OAAO,CAAC,aAAa,CAAC,CAKxB"}

package/dest/test-helpers/reqresp-nodes.js

@@ -0,0 +1,207 @@
+import { timesParallel } from '@aztec/foundation/collection';
+import { openTmpStore } from '@aztec/kv-store/lmdb-v2';
+import { emptyChainConfig } from '@aztec/stdlib/config';
+import { getTelemetryClient } from '@aztec/telemetry-client';
+import { SignableENR } from '@chainsafe/enr';
+import { gossipsub } from '@chainsafe/libp2p-gossipsub';
+import { noise } from '@chainsafe/libp2p-noise';
+import { yamux } from '@chainsafe/libp2p-yamux';
+import { bootstrap } from '@libp2p/bootstrap';
+import { identify } from '@libp2p/identify';
+import { createSecp256k1PeerId } from '@libp2p/peer-id-factory';
+import { tcp } from '@libp2p/tcp';
+import { multiaddr } from '@multiformats/multiaddr';
+import getPort from 'get-port';
+import { createLibp2p } from 'libp2p';
+import { BootstrapNode } from '../bootstrap/bootstrap.js';
+import { DiscV5Service } from '../services/discv5/discV5_service.js';
+import { LibP2PService } from '../services/libp2p/libp2p_service.js';
+import { ReqRespSubProtocol, noopValidator } from '../services/reqresp/interface.js';
+import { pingHandler, statusHandler } from '../services/reqresp/protocols/index.js';
+import { ReqResp } from '../services/reqresp/reqresp.js';
+import { convertToMultiaddr, createLibP2PPeerIdFromPrivateKey } from '../util.js';
+/**
+ * Creates a libp2p node, pre configured.
+ * @param boostrapAddrs - an optional list of bootstrap addresses
+ * @returns Lip2p node
+ */ export async function createLibp2pNode(boostrapAddrs = [], peerId, port, enableGossipSub = false, start = true) {
+    port = port ?? await getPort();
+    const options = {
+        start,
+        addresses: {
+            listen: [
+                `/ip4/0.0.0.0/tcp/${port}`
+            ],
+            announce: [
+                `/ip4/0.0.0.0/tcp/${port}`
+            ]
+        },
+        connectionEncryption: [
+            noise()
+        ],
+        streamMuxers: [
+            yamux()
+        ],
+        transports: [
+            tcp()
+        ],
+        services: {
+            identify: identify({
+                protocolPrefix: 'aztec'
+            })
+        }
+    };
+    if (boostrapAddrs.length > 0) {
+        options.peerDiscovery = [
+            bootstrap({
+                list: boostrapAddrs
+            })
+        ];
+    }
+    if (peerId) {
+        options.peerId = peerId;
+    }
+    if (enableGossipSub) {
+        options.services.pubsub = gossipsub({
+            allowPublishToZeroTopicPeers: true
+        });
+    }
+    return await createLibp2p(options);
+}
+/**
+ * Test Libp2p service
+ * P2P functionality is operational, however everything else is default
+ *
+ *
+ */ export async function createTestLibP2PService(clientType, boostrapAddrs = [], l2BlockSource, worldStateSynchronizer, epochCache, mempools, telemetry, port = 0, peerId, chainConfig = emptyChainConfig) {
+    peerId = peerId ?? await createSecp256k1PeerId();
+    const config = {
+        tcpAnnounceAddress: `127.0.0.1:${port}`,
+        udpAnnounceAddress: `127.0.0.1:${port}`,
+        tcpListenAddress: `0.0.0.0:${port}`,
+        udpListenAddress: `0.0.0.0:${port}`,
+        bootstrapNodes: boostrapAddrs,
+        peerCheckIntervalMS: 1000,
+        maxPeerCount: 5,
+        p2pEnabled: true,
+        peerIdPrivateKey: Buffer.from(peerId.privateKey).toString('hex'),
+        bootstrapNodeEnrVersionCheck: false,
+        ...chainConfig
+    };
+    const discoveryService = new DiscV5Service(peerId, config, telemetry);
+    const proofVerifier = new AlwaysTrueCircuitVerifier();
+    // No bootstrap nodes provided as the libp2p service will register them in the constructor
+    const p2pNode = await createLibp2pNode([], peerId, port, /*enable gossip */ true, /**start */ false);
+    return new LibP2PService(clientType, config, p2pNode, discoveryService, mempools, l2BlockSource, epochCache, proofVerifier, worldStateSynchronizer, telemetry);
+}
+// Mock sub protocol handlers
+export const MOCK_SUB_PROTOCOL_HANDLERS = {
+    [ReqRespSubProtocol.PING]: pingHandler,
+    [ReqRespSubProtocol.STATUS]: statusHandler,
+    [ReqRespSubProtocol.TX]: (_msg)=>Promise.resolve(Buffer.from('tx')),
+    [ReqRespSubProtocol.GOODBYE]: (_msg)=>Promise.resolve(Buffer.from('goodbye')),
+    [ReqRespSubProtocol.BLOCK]: (_msg)=>Promise.resolve(Buffer.from('block'))
+};
+// By default, all requests are valid
+// If you want to test an invalid response, you can override the validator
+export const MOCK_SUB_PROTOCOL_VALIDATORS = {
+    [ReqRespSubProtocol.PING]: noopValidator,
+    [ReqRespSubProtocol.STATUS]: noopValidator,
+    [ReqRespSubProtocol.TX]: noopValidator,
+    [ReqRespSubProtocol.GOODBYE]: noopValidator,
+    [ReqRespSubProtocol.BLOCK]: noopValidator
+};
+/**
+ * @param numberOfNodes - the number of nodes to create
+ * @returns An array of the created nodes
+ */ export const createNodes = (peerScoring, numberOfNodes)=>{
+    return timesParallel(numberOfNodes, ()=>createReqResp(peerScoring));
+};
+export const startNodes = async (nodes, subProtocolHandlers = MOCK_SUB_PROTOCOL_HANDLERS, subProtocolValidators = MOCK_SUB_PROTOCOL_VALIDATORS)=>{
+    for (const node of nodes){
+        await node.req.start(subProtocolHandlers, subProtocolValidators);
+    }
+};
+export const stopNodes = async (nodes)=>{
+    const stopPromises = nodes.flatMap((node)=>[
+            node.req.stop(),
+            node.p2p.stop()
+        ]);
+    await Promise.all(stopPromises);
+};
+// Create a req resp node, exposing the underlying p2p node
+export const createReqResp = async (peerScoring)=>{
+    const p2p = await createLibp2pNode();
+    const config = {
+        overallRequestTimeoutMs: 4000,
+        individualRequestTimeoutMs: 2000
+    };
+    const req = new ReqResp(config, p2p, peerScoring);
+    return {
+        p2p,
+        req
+    };
+};
+// Given a node list; hand shake all of the nodes with each other
+export const connectToPeers = async (nodes)=>{
+    for (const node of nodes){
+        for (const otherNode of nodes){
+            if (node === otherNode) {
+                continue;
+            }
+            const addr = otherNode.p2p.getMultiaddrs()[0];
+            await node.p2p.dial(addr);
+        }
+    }
+};
+// Mock circuit verifier for testing - reimplementation from bb to avoid dependency
+export class AlwaysTrueCircuitVerifier {
+    verifyProof(_tx) {
+        return Promise.resolve(true);
+    }
+}
+export class AlwaysFalseCircuitVerifier {
+    verifyProof(_tx) {
+        return Promise.resolve(false);
+    }
+}
+// Bootnodes
+export function createBootstrapNodeConfig(privateKey, port, chainConfig) {
+    return {
+        l1ChainId: chainConfig.l1ChainId,
+        udpListenAddress: `0.0.0.0:${port}`,
+        udpAnnounceAddress: `127.0.0.1:${port}`,
+        peerIdPrivateKey: privateKey,
+        dataDirectory: undefined,
+        dataStoreMapSizeKB: 0,
+        bootstrapNodes: []
+    };
+}
+export function createBootstrapNodeFromPrivateKey(privateKey, port, telemetry = getTelemetryClient(), chainConfig = emptyChainConfig) {
+    const config = createBootstrapNodeConfig(privateKey, port, chainConfig);
+    return startBootstrapNode(config, telemetry);
+}
+/**
+ * Create a bootstrap node ENR
+ * @param privateKey - the private key of the bootstrap node
+ * @param port - the port of the bootstrap node
+ * @returns the bootstrap node ENR
+ */ export async function getBootstrapNodeEnr(privateKey, port) {
+    const peerId = await createLibP2PPeerIdFromPrivateKey(privateKey);
+    const enr = SignableENR.createFromPeerId(peerId);
+    const listenAddrUdp = multiaddr(convertToMultiaddr(`127.0.0.1:${port}`, 'udp'));
+    enr.setLocationMultiaddr(listenAddrUdp);
+    return enr;
+}
+export async function createBootstrapNode(port, telemetry = getTelemetryClient(), chainConfig = emptyChainConfig) {
+    const peerId = await createSecp256k1PeerId();
+    const config = createBootstrapNodeConfig(Buffer.from(peerId.privateKey).toString('hex'), port, chainConfig);
+    return startBootstrapNode(config, telemetry);
+}
+async function startBootstrapNode(config, telemetry) {
+    // Open an ephemeral store that will only exist in memory
+    const store = await openTmpStore('bootstrap-node', true);
+    const bootstrapNode = new BootstrapNode(store, telemetry);
+    await bootstrapNode.start(config);
+    return bootstrapNode;
+}

package/dest/testbench/p2p_client_testbench_worker.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=p2p_client_testbench_worker.d.ts.map

package/dest/testbench/p2p_client_testbench_worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"p2p_client_testbench_worker.d.ts","sourceRoot":"","sources":["../../src/testbench/p2p_client_testbench_worker.ts"],"names":[],"mappings":""}