Add reqresp v2 definition

ChainSafe · Dec 17, 2024 · a8d9f73 · a8d9f73
1 parent 13d6448
commit a8d9f73
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/packages/beacon-node/src/network/reqresp/ReqRespBeaconNode.ts b/packages/beacon-node/src/network/reqresp/ReqRespBeaconNode.ts
@@ -257,6 +257,13 @@ export class ReqRespBeaconNode extends ReqResp {
       );
     }
 
+    if (ForkSeq[fork] >= ForkSeq.electra) {
+      protocolsAtFork.push(
+        [protocols.BlobSidecarsByRootV2(this.config), this.getHandler(ReqRespMethod.BlobSidecarsByRoot)],
+        [protocols.BlobSidecarsByRangeV2(this.config), this.getHandler(ReqRespMethod.BlobSidecarsByRange)]
+      );
+    }
+
     return protocolsAtFork;
   }
 

diff --git a/packages/beacon-node/src/network/reqresp/protocols.ts b/packages/beacon-node/src/network/reqresp/protocols.ts
@@ -63,12 +63,24 @@ export const BlobSidecarsByRange = toProtocol({
   contextBytesType: ContextBytesType.ForkDigest,
 });
 
+export const BlobSidecarsByRangeV2 = toProtocol({
+  method: ReqRespMethod.BlobSidecarsByRange,
+  version: Version.V2,
+  contextBytesType: ContextBytesType.ForkDigest,
+});
+
 export const BlobSidecarsByRoot = toProtocol({
   method: ReqRespMethod.BlobSidecarsByRoot,
   version: Version.V1,
   contextBytesType: ContextBytesType.ForkDigest,
 });
 
+export const BlobSidecarsByRootV2 = toProtocol({
+  method: ReqRespMethod.BlobSidecarsByRoot,
+  version: Version.V2,
+  contextBytesType: ContextBytesType.ForkDigest,
+});
+
 export const LightClientBootstrap = toProtocol({
   method: ReqRespMethod.LightClientBootstrap,
   version: Version.V1,