@lodestar/beacon-node 1.35.0-dev.f80d2d52da → 1.35.0-dev.fd1dac853d

package/src/chain/produceBlock/produceBlockBody.ts

@@ -0,0 +1,797 @@
+import {ChainForkConfig} from "@lodestar/config";
+import {
+  ForkName,
+  ForkPostBellatrix,
+  ForkPostDeneb,
+  ForkPostFulu,
+  ForkSeq,
+  isForkPostAltair,
+  isForkPostBellatrix,
+} from "@lodestar/params";
+import {
+  CachedBeaconStateAllForks,
+  CachedBeaconStateBellatrix,
+  CachedBeaconStateCapella,
+  CachedBeaconStateExecutions,
+  computeTimeAtSlot,
+  getCurrentEpoch,
+  getExpectedWithdrawals,
+  getRandaoMix,
+  isMergeTransitionComplete,
+} from "@lodestar/state-transition";
+import {
+  BLSPubkey,
+  BLSSignature,
+  BeaconBlock,
+  BeaconBlockBody,
+  BlindedBeaconBlock,
+  BlindedBeaconBlockBody,
+  BlobsBundle,
+  Bytes32,
+  ExecutionPayload,
+  ExecutionPayloadHeader,
+  Root,
+  RootHex,
+  SSEPayloadAttributes,
+  Slot,
+  ValidatorIndex,
+  Wei,
+  altair,
+  capella,
+  deneb,
+  electra,
+  fulu,
+  ssz,
+  sszTypesFor,
+} from "@lodestar/types";
+import {Logger, sleep, toHex, toPubkeyHex, toRootHex} from "@lodestar/utils";
+import {ZERO_HASH, ZERO_HASH_HEX} from "../../constants/index.js";
+import {IEth1ForBlockProduction} from "../../eth1/index.js";
+import {numToQuantity} from "../../eth1/provider/utils.js";
+import {
+  IExecutionBuilder,
+  IExecutionEngine,
+  PayloadAttributes,
+  PayloadId,
+  getExpectedGasLimit,
+} from "../../execution/index.js";
+import {fromGraffitiBytes} from "../../util/graffiti.js";
+import {kzg} from "../../util/kzg.js";
+import type {BeaconChain} from "../chain.js";
+import {CommonBlockBody} from "../interface.js";
+import {validateBlobsAndKzgCommitments, validateCellsAndKzgCommitments} from "./validateBlobsAndKzgCommitments.js";
+
+// Time to provide the EL to generate a payload from new payload id
+const PAYLOAD_GENERATION_TIME_MS = 500;
+
+export enum PayloadPreparationType {
+  Fresh = "Fresh",
+  Cached = "Cached",
+  Reorged = "Reorged",
+  Blinded = "Blinded",
+}
+
+/**
+ * Block production steps tracked in metrics
+ */
+export enum BlockProductionStep {
+  proposerSlashing = "proposerSlashing",
+  attesterSlashings = "attesterSlashings",
+  voluntaryExits = "voluntaryExits",
+  blsToExecutionChanges = "blsToExecutionChanges",
+  attestations = "attestations",
+  eth1DataAndDeposits = "eth1DataAndDeposits",
+  syncAggregate = "syncAggregate",
+  executionPayload = "executionPayload",
+}
+
+export type BlockAttributes = {
+  randaoReveal: BLSSignature;
+  graffiti: Bytes32;
+  slot: Slot;
+  parentBlockRoot: Root;
+  parentSlot: Slot;
+  feeRecipient?: string;
+};
+
+export enum BlockType {
+  Full = "Full",
+  Blinded = "Blinded",
+}
+export type AssembledBodyType<T extends BlockType> = T extends BlockType.Full
+  ? BeaconBlockBody
+  : BlindedBeaconBlockBody;
+export type AssembledBlockType<T extends BlockType> = T extends BlockType.Full ? BeaconBlock : BlindedBeaconBlock;
+
+export type ProduceFullFulu = {
+  type: BlockType.Full;
+  fork: ForkPostFulu;
+  executionPayload: ExecutionPayload<ForkPostFulu>;
+  blobsBundle: BlobsBundle<ForkPostFulu>;
+  cells: fulu.Cell[][];
+};
+export type ProduceFullDeneb = {
+  type: BlockType.Full;
+  fork: ForkName.deneb | ForkName.electra;
+  executionPayload: ExecutionPayload<ForkPostDeneb>;
+  blobsBundle: BlobsBundle<ForkPostDeneb>;
+};
+export type ProduceFullBellatrix = {
+  type: BlockType.Full;
+  fork: ForkName.bellatrix | ForkName.capella;
+  executionPayload: ExecutionPayload<ForkPostBellatrix>;
+};
+export type ProduceFullPhase0 = {
+  type: BlockType.Full;
+  fork: ForkName.phase0 | ForkName.altair;
+};
+export type ProduceBlinded = {
+  type: BlockType.Blinded;
+  fork: ForkName;
+};
+
+// The results of block production returned by `produceBlockBody`
+// The types are defined separately so typecasting can be used
+
+/** The result of local block production, everything that's not the block itself */
+export type ProduceResult =
+  | ProduceFullFulu
+  | ProduceFullDeneb
+  | ProduceFullBellatrix
+  | ProduceFullPhase0
+  | ProduceBlinded;
+
+export async function produceBlockBody<T extends BlockType>(
+  this: BeaconChain,
+  blockType: T,
+  currentState: CachedBeaconStateAllForks,
+  blockAttr: BlockAttributes & {
+    proposerIndex: ValidatorIndex;
+    proposerPubKey: BLSPubkey;
+    commonBlockBodyPromise: Promise<CommonBlockBody>;
+  }
+): Promise<{
+  body: AssembledBodyType<T>;
+  produceResult: ProduceResult;
+  executionPayloadValue: Wei;
+  shouldOverrideBuilder?: boolean;
+}> {
+  const {
+    slot: blockSlot,
+    feeRecipient: requestedFeeRecipient,
+    parentBlockRoot,
+    proposerIndex,
+    proposerPubKey,
+    commonBlockBodyPromise,
+  } = blockAttr;
+  let executionPayloadValue: Wei;
+  let blockBody: AssembledBodyType<T>;
+  // even though shouldOverrideBuilder is relevant for the engine response, for simplicity of typing
+  // we just return it undefined for the builder which anyway doesn't get consumed downstream
+  let shouldOverrideBuilder: boolean | undefined;
+  const fork = currentState.config.getForkName(blockSlot);
+  const produceResult = {
+    type: blockType,
+    fork,
+  } as ProduceResult;
+
+  const logMeta: Record<string, string | number | bigint> = {
+    fork,
+    blockType,
+    slot: blockSlot,
+  };
+  this.logger.verbose("Producing beacon block body", logMeta);
+
+  if (isForkPostBellatrix(fork)) {
+    const safeBlockHash = this.forkChoice.getJustifiedBlock().executionPayloadBlockHash ?? ZERO_HASH_HEX;
+    const finalizedBlockHash = this.forkChoice.getFinalizedBlock().executionPayloadBlockHash ?? ZERO_HASH_HEX;
+    const feeRecipient = requestedFeeRecipient ?? this.beaconProposerCache.getOrDefault(proposerIndex);
+    const feeRecipientType = requestedFeeRecipient
+      ? "requested"
+      : this.beaconProposerCache.get(proposerIndex)
+        ? "cached"
+        : "default";
+
+    Object.assign(logMeta, {feeRecipientType, feeRecipient});
+
+    if (blockType === BlockType.Blinded) {
+      if (!this.executionBuilder) throw Error("External builder not configured");
+      const executionBuilder = this.executionBuilder;
+
+      const builderPromise = (async () => {
+        const endExecutionPayloadHeader = this.metrics?.builderBlockProductionTimeSteps.startTimer();
+        // This path will not be used in the production, but is here just for merge mock
+        // tests because merge-mock requires an fcU to be issued prior to fetch payload
+        // header.
+        if (executionBuilder.issueLocalFcUWithFeeRecipient !== undefined) {
+          await prepareExecutionPayload(
+            this,
+            this.logger,
+            fork,
+            parentBlockRoot,
+            safeBlockHash,
+            finalizedBlockHash ?? ZERO_HASH_HEX,
+            currentState as CachedBeaconStateBellatrix,
+            executionBuilder.issueLocalFcUWithFeeRecipient
+          );
+        }
+
+        // For MeV boost integration, this is where the execution header will be
+        // fetched from the payload id and a blinded block will be produced instead of
+        // fullblock for the validator to sign
+        this.logger.verbose("Fetching execution payload header from builder", {
+          slot: blockSlot,
+          proposerPubKey: toHex(proposerPubKey),
+        });
+        const headerRes = await prepareExecutionPayloadHeader(
+          this,
+          fork,
+          currentState as CachedBeaconStateBellatrix,
+          proposerPubKey
+        );
+
+        endExecutionPayloadHeader?.({
+          step: BlockProductionStep.executionPayload,
+        });
+
+        return headerRes;
+      })();
+
+      const [builderRes, commonBlockBody] = await Promise.all([builderPromise, commonBlockBodyPromise]);
+      blockBody = Object.assign({}, commonBlockBody) as AssembledBodyType<BlockType.Blinded>;
+
+      (blockBody as BlindedBeaconBlockBody).executionPayloadHeader = builderRes.header;
+      executionPayloadValue = builderRes.executionPayloadValue;
+
+      const fetchedTime = Date.now() / 1000 - computeTimeAtSlot(this.config, blockSlot, this.genesisTime);
+      const prepType = PayloadPreparationType.Blinded;
+      this.metrics?.blockPayload.payloadFetchedTime.observe({prepType}, fetchedTime);
+      this.logger.verbose("Fetched execution payload header from builder", {
+        slot: blockSlot,
+        executionPayloadValue,
+        prepType,
+        fetchedTime,
+      });
+
+      const targetGasLimit = executionBuilder.getValidatorRegistration(proposerPubKey)?.gasLimit;
+      if (!targetGasLimit) {
+        // This should only happen if cache was cleared due to restart of beacon node
+        this.logger.warn("Failed to get validator registration, could not check header gas limit", {
+          slot: blockSlot,
+          proposerIndex,
+          proposerPubKey: toPubkeyHex(proposerPubKey),
+        });
+      } else {
+        const headerGasLimit = builderRes.header.gasLimit;
+        const parentGasLimit = (currentState as CachedBeaconStateBellatrix).latestExecutionPayloadHeader.gasLimit;
+        const expectedGasLimit = getExpectedGasLimit(parentGasLimit, targetGasLimit);
+
+        const lowerBound = Math.min(parentGasLimit, expectedGasLimit);
+        const upperBound = Math.max(parentGasLimit, expectedGasLimit);
+
+        if (headerGasLimit < lowerBound || headerGasLimit > upperBound) {
+          throw Error(
+            `Header gas limit ${headerGasLimit} is outside of acceptable range [${lowerBound}, ${upperBound}]`
+          );
+        }
+
+        if (headerGasLimit !== expectedGasLimit) {
+          this.logger.warn("Header gas limit does not match expected value", {
+            slot: blockSlot,
+            headerGasLimit,
+            expectedGasLimit,
+            parentGasLimit,
+            targetGasLimit,
+          });
+        }
+      }
+
+      if (ForkSeq[fork] >= ForkSeq.deneb) {
+        const {blobKzgCommitments} = builderRes;
+        if (blobKzgCommitments === undefined) {
+          throw Error(`Invalid builder getHeader response for fork=${fork}, missing blobKzgCommitments`);
+        }
+
+        (blockBody as deneb.BlindedBeaconBlockBody).blobKzgCommitments = blobKzgCommitments;
+        Object.assign(logMeta, {blobs: blobKzgCommitments.length});
+      }
+
+      if (ForkSeq[fork] >= ForkSeq.electra) {
+        const {executionRequests} = builderRes;
+        if (executionRequests === undefined) {
+          throw Error(`Invalid builder getHeader response for fork=${fork}, missing executionRequests`);
+        }
+        (blockBody as electra.BlindedBeaconBlockBody).executionRequests = executionRequests;
+      }
+    }
+
+    // blockType === BlockType.Full
+    else {
+      const enginePromise = (async () => {
+        const endExecutionPayload = this.metrics?.executionBlockProductionTimeSteps.startTimer();
+
+        this.logger.verbose("Preparing execution payload from engine", {
+          slot: blockSlot,
+          parentBlockRoot: toRootHex(parentBlockRoot),
+          feeRecipient,
+        });
+        // https://github.com/ethereum/consensus-specs/blob/dev/specs/deneb/validator.md#constructing-the-beaconblockbody
+        const prepareRes = await prepareExecutionPayload(
+          this,
+          this.logger,
+          fork,
+          parentBlockRoot,
+          safeBlockHash,
+          finalizedBlockHash ?? ZERO_HASH_HEX,
+          currentState as CachedBeaconStateExecutions,
+          feeRecipient
+        );
+
+        if (prepareRes.isPremerge) {
+          return {
+            ...prepareRes,
+            executionPayload: sszTypesFor(fork).ExecutionPayload.defaultValue(),
+            executionPayloadValue: BigInt(0),
+          };
+        }
+
+        const {prepType, payloadId} = prepareRes;
+        Object.assign(logMeta, {executionPayloadPrepType: prepType});
+
+        if (prepType !== PayloadPreparationType.Cached) {
+          // Wait for 500ms to allow EL to add some txs to the payload
+          // the pitfalls of this have been put forward here, but 500ms delay for block proposal
+          // seems marginal even with unhealthy network
+          //
+          // See: https://discord.com/channels/595666850260713488/892088344438255616/1009882079632314469
+          await sleep(PAYLOAD_GENERATION_TIME_MS);
+        }
+
+        this.logger.verbose("Fetching execution payload from engine", {slot: blockSlot, payloadId});
+        const payloadRes = await this.executionEngine.getPayload(fork, payloadId);
+
+        endExecutionPayload?.({
+          step: BlockProductionStep.executionPayload,
+        });
+
+        return {...prepareRes, ...payloadRes};
+      })().catch((e) => {
+        // catch payload fetch here, because there is still a recovery path possible if we
+        // are pre-merge. We don't care the same for builder segment as the execution block
+        // will takeover if the builder flow was activated and errors
+        this.metrics?.blockPayload.payloadFetchErrors.inc();
+
+        if (!isMergeTransitionComplete(currentState as CachedBeaconStateBellatrix)) {
+          this.logger?.warn(
+            "Fetch payload from the execution failed, however since we are still pre-merge proceeding with an empty one.",
+            {},
+            e as Error
+          );
+          // ok we don't have an execution payload here, so we can assign an empty one
+          // if pre-merge
+          return {
+            isPremerge: true as const,
+            executionPayload: sszTypesFor(fork).ExecutionPayload.defaultValue(),
+            executionPayloadValue: BigInt(0),
+          };
+        }
+        // since merge transition is complete, we need a valid payload even if with an
+        // empty (transactions) one. defaultValue isn't gonna cut it!
+        throw e;
+      });
+
+      const [engineRes, commonBlockBody] = await Promise.all([enginePromise, commonBlockBodyPromise]);
+      blockBody = Object.assign({}, commonBlockBody) as AssembledBodyType<BlockType.Blinded>;
+
+      if (engineRes.isPremerge) {
+        (blockBody as BeaconBlockBody<ForkPostBellatrix>).executionPayload = engineRes.executionPayload;
+        executionPayloadValue = engineRes.executionPayloadValue;
+      } else {
+        const {prepType, payloadId, executionPayload, blobsBundle, executionRequests} = engineRes;
+        shouldOverrideBuilder = engineRes.shouldOverrideBuilder;
+
+        (blockBody as BeaconBlockBody<ForkPostBellatrix>).executionPayload = executionPayload;
+        (produceResult as ProduceFullBellatrix).executionPayload = executionPayload;
+        executionPayloadValue = engineRes.executionPayloadValue;
+        Object.assign(logMeta, {transactions: executionPayload.transactions.length, shouldOverrideBuilder});
+
+        const fetchedTime = Date.now() / 1000 - computeTimeAtSlot(this.config, blockSlot, this.genesisTime);
+        this.metrics?.blockPayload.payloadFetchedTime.observe({prepType}, fetchedTime);
+        this.logger.verbose("Fetched execution payload from engine", {
+          slot: blockSlot,
+          executionPayloadValue,
+          prepType,
+          payloadId,
+          fetchedTime,
+          executionHeadBlockHash: toRootHex(engineRes.executionPayload.blockHash),
+        });
+        if (executionPayload.transactions.length === 0) {
+          this.metrics?.blockPayload.emptyPayloads.inc({prepType});
+        }
+
+        if (ForkSeq[fork] >= ForkSeq.fulu) {
+          if (blobsBundle === undefined) {
+            throw Error(`Missing blobsBundle response from getPayload at fork=${fork}`);
+          }
+          // NOTE: Even though the fulu.BlobsBundle type is superficially the same as deneb.BlobsBundle, it is NOT.
+          // In fulu, proofs are _cell_ proofs, vs in deneb they are _blob_ proofs.
+
+          const cells = blobsBundle.blobs.map((blob) => kzg.computeCells(blob));
+          if (this.opts.sanityCheckExecutionEngineBlobs) {
+            await validateCellsAndKzgCommitments(blobsBundle.commitments, blobsBundle.proofs, cells);
+          }
+
+          (blockBody as deneb.BeaconBlockBody).blobKzgCommitments = blobsBundle.commitments;
+          (produceResult as ProduceFullFulu).blobsBundle = blobsBundle;
+          (produceResult as ProduceFullFulu).cells = cells;
+
+          Object.assign(logMeta, {blobs: blobsBundle.commitments.length});
+        } else if (ForkSeq[fork] >= ForkSeq.deneb) {
+          if (blobsBundle === undefined) {
+            throw Error(`Missing blobsBundle response from getPayload at fork=${fork}`);
+          }
+
+          if (this.opts.sanityCheckExecutionEngineBlobs) {
+            await validateBlobsAndKzgCommitments(blobsBundle.commitments, blobsBundle.proofs, blobsBundle.blobs);
+          }
+
+          (blockBody as deneb.BeaconBlockBody).blobKzgCommitments = blobsBundle.commitments;
+          (produceResult as ProduceFullDeneb).blobsBundle = blobsBundle;
+
+          Object.assign(logMeta, {blobs: blobsBundle.commitments.length});
+        }
+
+        if (ForkSeq[fork] >= ForkSeq.electra) {
+          if (executionRequests === undefined) {
+            throw Error(`Missing executionRequests response from getPayload at fork=${fork}`);
+          }
+          (blockBody as electra.BeaconBlockBody).executionRequests = executionRequests;
+        }
+      }
+    }
+  } else {
+    const commonBlockBody = await commonBlockBodyPromise;
+    blockBody = Object.assign({}, commonBlockBody) as AssembledBodyType<T>;
+    executionPayloadValue = BigInt(0);
+  }
+
+  const {graffiti, attestations, deposits, voluntaryExits, attesterSlashings, proposerSlashings} = blockBody;
+
+  Object.assign(logMeta, {
+    graffiti: fromGraffitiBytes(graffiti),
+    attestations: attestations.length,
+    deposits: deposits.length,
+    voluntaryExits: voluntaryExits.length,
+    attesterSlashings: attesterSlashings.length,
+    proposerSlashings: proposerSlashings.length,
+  });
+
+  if (isForkPostAltair(fork)) {
+    const {syncAggregate} = blockBody as altair.BeaconBlockBody;
+    Object.assign(logMeta, {
+      syncAggregateParticipants: syncAggregate.syncCommitteeBits.getTrueBitIndexes().length,
+    });
+  }
+
+  if (ForkSeq[fork] >= ForkSeq.capella) {
+    const {blsToExecutionChanges, executionPayload} = blockBody as capella.BeaconBlockBody;
+    Object.assign(logMeta, {
+      blsToExecutionChanges: blsToExecutionChanges.length,
+    });
+
+    // withdrawals are only available in full body
+    if (blockType === BlockType.Full) {
+      Object.assign(logMeta, {
+        withdrawals: executionPayload.withdrawals.length,
+      });
+    }
+  }
+
+  Object.assign(logMeta, {executionPayloadValue});
+  this.logger.verbose("Produced beacon block body", logMeta);
+
+  return {body: blockBody as AssembledBodyType<T>, produceResult, executionPayloadValue, shouldOverrideBuilder};
+}
+
+/**
+ * Produce ExecutionPayload for pre-merge, merge, and post-merge.
+ *
+ * Expects `eth1MergeBlockFinder` to be actively searching for blocks well in advance to being called.
+ *
+ * @returns PayloadId = pow block found, null = pow NOT found
+ */
+export async function prepareExecutionPayload(
+  chain: {
+    eth1: IEth1ForBlockProduction;
+    executionEngine: IExecutionEngine;
+    config: ChainForkConfig;
+  },
+  logger: Logger,
+  fork: ForkPostBellatrix,
+  parentBlockRoot: Root,
+  safeBlockHash: RootHex,
+  finalizedBlockHash: RootHex,
+  state: CachedBeaconStateExecutions,
+  suggestedFeeRecipient: string
+): Promise<{isPremerge: true} | {isPremerge: false; prepType: PayloadPreparationType; payloadId: PayloadId}> {
+  const parentHashRes = await getExecutionPayloadParentHash(chain, state);
+  if (parentHashRes.isPremerge) {
+    // Return null only if the execution is pre-merge
+    return {isPremerge: true};
+  }
+
+  const {parentHash} = parentHashRes;
+  const timestamp = computeTimeAtSlot(chain.config, state.slot, state.genesisTime);
+  const prevRandao = getRandaoMix(state, state.epochCtx.epoch);
+
+  const payloadIdCached = chain.executionEngine.payloadIdCache.get({
+    headBlockHash: toRootHex(parentHash),
+    finalizedBlockHash,
+    timestamp: numToQuantity(timestamp),
+    prevRandao: toHex(prevRandao),
+    suggestedFeeRecipient,
+  });
+
+  // prepareExecutionPayload will throw error via notifyForkchoiceUpdate if
+  // the EL returns Syncing on this request to prepare a payload
+  // TODO: Handle only this case, DO NOT put a generic try / catch that discards all errors
+  let payloadId: PayloadId | null;
+  let prepType: PayloadPreparationType;
+
+  if (payloadIdCached) {
+    payloadId = payloadIdCached;
+    prepType = PayloadPreparationType.Cached;
+  } else {
+    // If there was a payload assigned to this timestamp, it would imply that there some sort
+    // of payload reorg, i.e. head, fee recipient or any other fcu param changed
+    if (chain.executionEngine.payloadIdCache.hasPayload({timestamp: numToQuantity(timestamp)})) {
+      prepType = PayloadPreparationType.Reorged;
+    } else {
+      prepType = PayloadPreparationType.Fresh;
+    }
+
+    const attributes: PayloadAttributes = preparePayloadAttributes(fork, chain, {
+      prepareState: state,
+      prepareSlot: state.slot,
+      parentBlockRoot,
+      feeRecipient: suggestedFeeRecipient,
+    });
+
+    payloadId = await chain.executionEngine.notifyForkchoiceUpdate(
+      fork,
+      toRootHex(parentHash),
+      safeBlockHash,
+      finalizedBlockHash,
+      attributes
+    );
+    logger.verbose("Prepared payload id from execution engine", {payloadId});
+  }
+
+  // Should never happen, notifyForkchoiceUpdate() with payload attributes always
+  // returns payloadId
+  if (payloadId === null) {
+    throw Error("notifyForkchoiceUpdate returned payloadId null");
+  }
+
+  // We are only returning payloadId here because prepareExecutionPayload is also called from
+  // prepareNextSlot, which is an advance call to execution engine to start building payload
+  // Actual payload isn't produced till getPayload is called.
+  return {isPremerge: false, payloadId, prepType};
+}
+
+async function prepareExecutionPayloadHeader(
+  chain: {
+    eth1: IEth1ForBlockProduction;
+    executionBuilder?: IExecutionBuilder;
+    config: ChainForkConfig;
+  },
+  fork: ForkPostBellatrix,
+  state: CachedBeaconStateBellatrix,
+  proposerPubKey: BLSPubkey
+): Promise<{
+  header: ExecutionPayloadHeader;
+  executionPayloadValue: Wei;
+  blobKzgCommitments?: deneb.BlobKzgCommitments;
+  executionRequests?: electra.ExecutionRequests;
+}> {
+  if (!chain.executionBuilder) {
+    throw Error("executionBuilder required");
+  }
+
+  const parentHashRes = await getExecutionPayloadParentHash(chain, state);
+  if (parentHashRes.isPremerge) {
+    throw Error("External builder disabled pre-merge");
+  }
+
+  const {parentHash} = parentHashRes;
+  return chain.executionBuilder.getHeader(fork, state.slot, parentHash, proposerPubKey);
+}
+
+export async function getExecutionPayloadParentHash(
+  chain: {
+    eth1: IEth1ForBlockProduction;
+    config: ChainForkConfig;
+  },
+  state: CachedBeaconStateExecutions
+): Promise<{isPremerge: true} | {isPremerge: false; parentHash: Root}> {
+  // Use different POW block hash parent for block production based on merge status.
+  // Returned value of null == using an empty ExecutionPayload value
+  if (isMergeTransitionComplete(state)) {
+    // Post-merge, normal payload
+    return {isPremerge: false, parentHash: state.latestExecutionPayloadHeader.blockHash};
+  }
+
+  if (
+    !ssz.Root.equals(chain.config.TERMINAL_BLOCK_HASH, ZERO_HASH) &&
+    getCurrentEpoch(state) < chain.config.TERMINAL_BLOCK_HASH_ACTIVATION_EPOCH
+  ) {
+    throw new Error(
+      `InvalidMergeTBH epoch: expected >= ${
+        chain.config.TERMINAL_BLOCK_HASH_ACTIVATION_EPOCH
+      }, actual: ${getCurrentEpoch(state)}`
+    );
+  }
+
+  const terminalPowBlockHash = await chain.eth1.getTerminalPowBlock();
+  if (terminalPowBlockHash === null) {
+    // Pre-merge, no prepare payload call is needed
+    return {isPremerge: true};
+  }
+  // Signify merge via producing on top of the last PoW block
+  return {isPremerge: false, parentHash: terminalPowBlockHash};
+}
+
+export async function getPayloadAttributesForSSE(
+  fork: ForkPostBellatrix,
+  chain: {
+    eth1: IEth1ForBlockProduction;
+    config: ChainForkConfig;
+  },
+  {
+    prepareState,
+    prepareSlot,
+    parentBlockRoot,
+    feeRecipient,
+  }: {prepareState: CachedBeaconStateExecutions; prepareSlot: Slot; parentBlockRoot: Root; feeRecipient: string}
+): Promise<SSEPayloadAttributes> {
+  const parentHashRes = await getExecutionPayloadParentHash(chain, prepareState);
+
+  if (!parentHashRes.isPremerge) {
+    const {parentHash} = parentHashRes;
+    const payloadAttributes = preparePayloadAttributes(fork, chain, {
+      prepareState,
+      prepareSlot,
+      parentBlockRoot,
+      feeRecipient,
+    });
+
+    const ssePayloadAttributes: SSEPayloadAttributes = {
+      proposerIndex: prepareState.epochCtx.getBeaconProposer(prepareSlot),
+      proposalSlot: prepareSlot,
+      parentBlockNumber: prepareState.latestExecutionPayloadHeader.blockNumber,
+      parentBlockRoot,
+      parentBlockHash: parentHash,
+      payloadAttributes,
+    };
+    return ssePayloadAttributes;
+  }
+
+  throw Error("The execution is still pre-merge");
+}
+
+function preparePayloadAttributes(
+  fork: ForkPostBellatrix,
+  chain: {
+    config: ChainForkConfig;
+  },
+  {
+    prepareState,
+    prepareSlot,
+    parentBlockRoot,
+    feeRecipient,
+  }: {
+    prepareState: CachedBeaconStateExecutions;
+    prepareSlot: Slot;
+    parentBlockRoot: Root;
+    feeRecipient: string;
+  }
+): SSEPayloadAttributes["payloadAttributes"] {
+  const timestamp = computeTimeAtSlot(chain.config, prepareSlot, prepareState.genesisTime);
+  const prevRandao = getRandaoMix(prepareState, prepareState.epochCtx.epoch);
+  const payloadAttributes = {
+    timestamp,
+    prevRandao,
+    suggestedFeeRecipient: feeRecipient,
+  };
+
+  if (ForkSeq[fork] >= ForkSeq.capella) {
+    // withdrawals logic is now fork aware as it changes on electra fork post capella
+    (payloadAttributes as capella.SSEPayloadAttributes["payloadAttributes"]).withdrawals = getExpectedWithdrawals(
+      ForkSeq[fork],
+      prepareState as CachedBeaconStateCapella
+    ).withdrawals;
+  }
+
+  if (ForkSeq[fork] >= ForkSeq.deneb) {
+    (payloadAttributes as deneb.SSEPayloadAttributes["payloadAttributes"]).parentBeaconBlockRoot = parentBlockRoot;
+  }
+
+  return payloadAttributes;
+}
+
+export async function produceCommonBlockBody<T extends BlockType>(
+  this: BeaconChain,
+  blockType: T,
+  currentState: CachedBeaconStateAllForks,
+  {
+    randaoReveal,
+    graffiti,
+    slot,
+    parentSlot,
+    parentBlockRoot,
+  }: BlockAttributes & {
+    parentSlot: Slot;
+  }
+): Promise<CommonBlockBody> {
+  const stepsMetrics =
+    blockType === BlockType.Full
+      ? this.metrics?.executionBlockProductionTimeSteps
+      : this.metrics?.builderBlockProductionTimeSteps;
+
+  const fork = currentState.config.getForkName(slot);
+
+  // TODO:
+  // Iterate through the naive aggregation pool and ensure all the attestations from there
+  // are included in the operation pool.
+  // for (const attestation of db.attestationPool.getAll()) {
+  //   try {
+  //     opPool.insertAttestation(attestation);
+  //   } catch (e) {
+  //     // Don't stop block production if there's an error, just create a log.
+  //     logger.error("Attestation did not transfer to op pool", {}, e);
+  //   }
+  // }
+  const [attesterSlashings, proposerSlashings, voluntaryExits, blsToExecutionChanges] =
+    this.opPool.getSlashingsAndExits(currentState, blockType, this.metrics);
+
+  const endAttestations = stepsMetrics?.startTimer();
+  const attestations = this.aggregatedAttestationPool.getAttestationsForBlock(fork, this.forkChoice, currentState);
+  endAttestations?.({
+    step: BlockProductionStep.attestations,
+  });
+
+  const endEth1DataAndDeposits = stepsMetrics?.startTimer();
+  const {eth1Data, deposits} = await this.eth1.getEth1DataAndDeposits(currentState);
+  endEth1DataAndDeposits?.({
+    step: BlockProductionStep.eth1DataAndDeposits,
+  });
+
+  const blockBody: Omit<CommonBlockBody, "blsToExecutionChanges" | "syncAggregate"> = {
+    randaoReveal,
+    graffiti,
+    eth1Data,
+    proposerSlashings,
+    attesterSlashings,
+    attestations,
+    deposits,
+    voluntaryExits,
+  };
+
+  if (ForkSeq[fork] >= ForkSeq.capella) {
+    (blockBody as CommonBlockBody).blsToExecutionChanges = blsToExecutionChanges;
+  }
+
+  const endSyncAggregate = stepsMetrics?.startTimer();
+  if (ForkSeq[fork] >= ForkSeq.altair) {
+    const syncAggregate = this.syncContributionAndProofPool.getAggregate(parentSlot, parentBlockRoot);
+    this.metrics?.production.producedSyncAggregateParticipants.observe(
+      syncAggregate.syncCommitteeBits.getTrueBitIndexes().length
+    );
+    (blockBody as CommonBlockBody).syncAggregate = syncAggregate;
+  }
+  endSyncAggregate?.({
+    step: BlockProductionStep.syncAggregate,
+  });
+
+  return blockBody as CommonBlockBody;
+}