corner case around transition protocol

Signed-off-by: Danno Ferrin <[email protected]>
hyperledger · Jan 12, 2023 · a080bff · a080bff
1 parent 18d336d
commit a080bff
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/...rc/main/java/org/hyperledger/besu/ethereum/mainnet/ScheduleBasedBlockHeaderFunctions.java b/...rc/main/java/org/hyperledger/besu/ethereum/mainnet/ScheduleBasedBlockHeaderFunctions.java
@@ -47,6 +47,12 @@ public ParsedExtraData parseExtraData(final BlockHeader header) {
   }
 
   private BlockHeaderFunctions getBlockHeaderFunctions(final SealableBlockHeader header) {
-    return protocolSchedule.getByBlockHeader(header).getBlockHeaderFunctions();
+    if (protocolSchedule instanceof ProtocolSchedule) {
+      return ((ProtocolSchedule) protocolSchedule)
+          .getByBlockNumber(header.getNumber())
+          .getBlockHeaderFunctions();
+    } else {
+      return protocolSchedule.getByBlockHeader(header).getBlockHeaderFunctions();
+    }
   }
 }