Implement EIP4895: Beacon Chain withdrawals (#2353)

* common: add eip 4895 * block: implement EIP4895 * vm: add EIP4895 * block: eip4895 tests * vm: add eip4895 tests * block: fix trest * vm: fix tests * change withdrawal type to object format and add validator index * fix vm withdrawal spec Co-authored-by: harkamal <[email protected]>
ethereumjs · Nov 4, 2022 · 3f34eff · 3f34eff
1 parent 9b1ab3e
commit 3f34eff
Show file tree

Hide file tree

Showing 12 changed files with 481 additions and 15 deletions.
diff --git a/packages/block/src/block.ts b/packages/block/src/block.ts
@@ -10,21 +10,30 @@ import {
   bufferToHex,
   intToHex,
   isHexPrefixed,
+  toBuffer,
 } from '@ethereumjs/util'
 import { keccak256 } from 'ethereum-cryptography/keccak'
 import { ethers } from 'ethers'
 
 import { blockFromRpc } from './from-rpc'
 import { BlockHeader } from './header'
 
-import type { BlockBuffer, BlockData, BlockOptions, JsonBlock, JsonRpcBlock } from './types'
+import type {
+  BlockBuffer,
+  BlockData,
+  BlockOptions,
+  JsonBlock,
+  JsonRpcBlock,
+  Withdrawal,
+} from './types'
 import type { Common } from '@ethereumjs/common'
 import type {
   FeeMarketEIP1559Transaction,
   Transaction,
   TxOptions,
   TypedTransaction,
 } from '@ethereumjs/tx'
+import type { Address } from '@ethereumjs/util'
 
 /**
  * An object that represents the block.
@@ -33,6 +42,7 @@ export class Block {
   public readonly header: BlockHeader
   public readonly transactions: TypedTransaction[] = []
   public readonly uncleHeaders: BlockHeader[] = []
+  public readonly withdrawals?: Withdrawal[]
   public readonly txTrie = new Trie()
   public readonly _common: Common
 
@@ -43,7 +53,12 @@ export class Block {
    * @param opts
    */
   public static fromBlockData(blockData: BlockData = {}, opts?: BlockOptions) {
-    const { header: headerData, transactions: txsData, uncleHeaders: uhsData } = blockData
+    const {
+      header: headerData,
+      transactions: txsData,
+      uncleHeaders: uhsData,
+      withdrawals,
+    } = blockData
     const header = BlockHeader.fromHeaderData(headerData, opts)
 
     // parse transactions
@@ -78,7 +93,7 @@ export class Block {
       uncleHeaders.push(uh)
     }
 
-    return new Block(header, transactions, uncleHeaders, opts)
+    return new Block(header, transactions, uncleHeaders, opts, withdrawals)
   }
 
   /**
@@ -104,11 +119,11 @@ export class Block {
    * @param opts
    */
   public static fromValuesArray(values: BlockBuffer, opts?: BlockOptions) {
-    if (values.length > 3) {
+    if (values.length > 4) {
       throw new Error('invalid block. More values than expected were received')
     }
 
-    const [headerData, txsData, uhsData] = values
+    const [headerData, txsData, uhsData, withdrawalsData] = values
 
     const header = BlockHeader.fromValuesArray(headerData, opts)
 
@@ -144,7 +159,16 @@ export class Block {
       uncleHeaders.push(BlockHeader.fromValuesArray(uncleHeaderData, uncleOpts))
     }
 
-    return new Block(header, transactions, uncleHeaders, opts)
+    let withdrawals
+    if (withdrawalsData) {
+      withdrawals = <Withdrawal[]>[]
+      for (const withdrawal of withdrawalsData) {
+        const [index, validatorIndex, address, amount] = withdrawal
+        withdrawals.push({ index, validatorIndex, address, amount })
+      }
+    }
+
+    return new Block(header, transactions, uncleHeaders, opts, withdrawals)
   }
 
   /**
@@ -212,7 +236,8 @@ export class Block {
     header?: BlockHeader,
     transactions: TypedTransaction[] = [],
     uncleHeaders: BlockHeader[] = [],
-    opts: BlockOptions = {}
+    opts: BlockOptions = {},
+    withdrawals?: Withdrawal[]
   ) {
     this.header = header ?? BlockHeader.fromHeaderData({}, opts)
     this.transactions = transactions
@@ -234,23 +259,55 @@ export class Block {
       }
     }
 
+    if (this._common.isActivatedEIP(4895) && withdrawals === undefined) {
+      throw new Error('Need a withdrawals field if EIP 4895 is active')
+    } else if (!this._common.isActivatedEIP(4895) && withdrawals !== undefined) {
+      throw new Error('Cannot have a withdrawals field if EIP 4895 is not active')
+    }
+
+    this.withdrawals = withdrawals
+
     const freeze = opts?.freeze ?? true
     if (freeze) {
       Object.freeze(this)
     }
   }
 
+  /**
+   * Convert a withdrawal to a buffer array
+   * @param withdrawal the withdrawal to convert
+   * @returns buffer array of the withdrawal
+   */
+  private withdrawalToBufferArray(withdrawal: Withdrawal): [Buffer, Buffer, Buffer, Buffer] {
+    const { index, validatorIndex, address, amount } = withdrawal
+    let addressBuffer: Buffer
+    if (typeof address === 'string') {
+      addressBuffer = Buffer.from(address.slice(2))
+    } else if (Buffer.isBuffer(address)) {
+      addressBuffer = address
+    } else {
+      addressBuffer = (<Address>address).buf
+    }
+    return [toBuffer(index), toBuffer(validatorIndex), addressBuffer, toBuffer(amount)]
+  }
+
   /**
    * Returns a Buffer Array of the raw Buffers of this block, in order.
    */
   raw(): BlockBuffer {
-    return [
+    const bufferArray = <BlockBuffer>[
       this.header.raw(),
       this.transactions.map((tx) =>
         tx.supports(Capability.EIP2718TypedTransaction) ? tx.serialize() : tx.raw()
       ) as Buffer[],
       this.uncleHeaders.map((uh) => uh.raw()),
     ]
+    if (this.withdrawals) {
+      for (const withdrawal of this.withdrawals) {
+        bufferArray.push(this.withdrawalToBufferArray(withdrawal))
+      }
+    }
+    return bufferArray
   }
 
   /**
@@ -369,6 +426,11 @@ export class Block {
       const msg = this._errorMsg('invalid uncle hash')
       throw new Error(msg)
     }
+
+    if (this._common.isActivatedEIP(4895) && !(await this.validateWithdrawalsTrie())) {
+      const msg = this._errorMsg('invalid withdrawals trie')
+      throw new Error(msg)
+    }
   }
 
   /**
@@ -380,6 +442,23 @@ export class Block {
     return Buffer.from(keccak256(raw)).equals(this.header.uncleHash)
   }
 
+  /**
+   * Validates the withdrawal root
+   */
+  async validateWithdrawalsTrie(): Promise<boolean> {
+    if (!this._common.isActivatedEIP(4895)) {
+      throw new Error('EIP 4895 is not activated')
+    }
+    const trie = new Trie()
+    let index = 0
+    for (const withdrawal of this.withdrawals!) {
+      const withdrawalRLP = RLP.encode(this.withdrawalToBufferArray(withdrawal))
+      await trie.put(Buffer.from('0x' + index.toString(16)), arrToBufArr(withdrawalRLP))
+      index++
+    }
+    return trie.root().equals(this.header.withdrawalsRoot!)
+  }
+
   /**
    * Consistency checks for uncles included in the block, if any.
    *

diff --git a/packages/block/src/header.ts b/packages/block/src/header.ts
@@ -51,6 +51,7 @@ export class BlockHeader {
   public readonly mixHash: Buffer
   public readonly nonce: Buffer
   public readonly baseFeePerGas?: bigint
+  public readonly withdrawalsRoot?: Buffer
 
   public readonly _common: Common
 
@@ -154,6 +155,7 @@ export class BlockHeader {
       mixHash: zeros(32),
       nonce: zeros(8),
       baseFeePerGas: undefined,
+      withdrawalsRoot: undefined,
     }
 
     const parentHash = toType(headerData.parentHash, TypeOutput.Buffer) ?? defaults.parentHash
@@ -176,6 +178,8 @@ export class BlockHeader {
     const nonce = toType(headerData.nonce, TypeOutput.Buffer) ?? defaults.nonce
     let baseFeePerGas =
       toType(headerData.baseFeePerGas, TypeOutput.BigInt) ?? defaults.baseFeePerGas
+    const withdrawalsRoot =
+      toType(headerData.withdrawalsRoot, TypeOutput.Buffer) ?? defaults.withdrawalsRoot
 
     const hardforkByBlockNumber = options.hardforkByBlockNumber ?? false
     if (hardforkByBlockNumber || options.hardforkByTTD !== undefined) {
@@ -198,6 +202,18 @@ export class BlockHeader {
       }
     }
 
+    if (this._common.isActivatedEIP(4895)) {
+      if (withdrawalsRoot === defaults.withdrawalsRoot) {
+        throw new Error('invalid header. withdrawalsRoot should be provided')
+      }
+    } else {
+      if (withdrawalsRoot !== undefined) {
+        throw new Error(
+          'A withdrawalsRoot for a header can only be provied with EIP4895 being activated'
+        )
+      }
+    }
+
     this.parentHash = parentHash
     this.uncleHash = uncleHash
     this.coinbase = coinbase
@@ -214,6 +230,7 @@ export class BlockHeader {
     this.mixHash = mixHash
     this.nonce = nonce
     this.baseFeePerGas = baseFeePerGas
+    this.withdrawalsRoot = withdrawalsRoot
 
     this._genericFormatValidation()
     this._validateDAOExtraData()
@@ -310,6 +327,19 @@ export class BlockHeader {
         }
       }
     }
+
+    if (this._common.isActivatedEIP(4895) === true) {
+      if (this.withdrawalsRoot === undefined) {
+        const msg = this._errorMsg('EIP4895 block has no withdrawalsRoot field')
+        throw new Error(msg)
+      }
+      if (this.withdrawalsRoot?.length !== 32) {
+        const msg = this._errorMsg(
+          `withdrawalsRoot must be 32 bytes, received ${this.withdrawalsRoot!.length} bytes`
+        )
+        throw new Error(msg)
+      }
+    }
   }
 
   /**

diff --git a/packages/block/src/helpers.ts b/packages/block/src/helpers.ts
@@ -37,9 +37,10 @@ export function valuesArrayToHeaderData(values: BlockHeaderBuffer): HeaderData {
     mixHash,
     nonce,
     baseFeePerGas,
+    withdrawalsRoot,
   ] = values
 
-  if (values.length > 16) {
+  if (values.length > 17) {
     throw new Error('invalid header. More values than expected were received')
   }
   if (values.length < 15) {
@@ -63,6 +64,7 @@ export function valuesArrayToHeaderData(values: BlockHeaderBuffer): HeaderData {
     mixHash,
     nonce,
     baseFeePerGas,
+    withdrawalsRoot,
   }
 }
 

diff --git a/packages/block/src/types.ts b/packages/block/src/types.ts
@@ -95,6 +95,14 @@ export interface HeaderData {
   mixHash?: BufferLike
   nonce?: BufferLike
   baseFeePerGas?: BigIntLike
+  withdrawalsRoot?: BufferLike
+}
+
+export type Withdrawal = {
+  index: BigIntLike
+  validatorIndex: BigIntLike
+  address: AddressLike
+  amount: BigIntLike
 }
 
 /**
@@ -107,16 +115,20 @@ export interface BlockData {
   header?: HeaderData
   transactions?: Array<TxData | AccessListEIP2930TxData | FeeMarketEIP1559TxData>
   uncleHeaders?: Array<HeaderData>
+  withdrawals?: Array<Withdrawal>
 }
 
-export type BlockBuffer = [BlockHeaderBuffer, TransactionsBuffer, UncleHeadersBuffer]
+export type BlockBuffer =
+  | [BlockHeaderBuffer, TransactionsBuffer, UncleHeadersBuffer]
+  | [BlockHeaderBuffer, TransactionsBuffer, UncleHeadersBuffer, WithdrawalBuffer]
 export type BlockHeaderBuffer = Buffer[]
-export type BlockBodyBuffer = [TransactionsBuffer, UncleHeadersBuffer]
+export type BlockBodyBuffer = [TransactionsBuffer, UncleHeadersBuffer, WithdrawalBuffer?]
 /**
  * TransactionsBuffer can be an array of serialized txs for Typed Transactions or an array of Buffer Arrays for legacy transactions.
  */
 export type TransactionsBuffer = Buffer[][] | Buffer[]
 export type UncleHeadersBuffer = Buffer[][]
+export type WithdrawalBuffer = Buffer[][]
 
 /**
  * An object with the block's data represented as strings.
@@ -128,6 +140,7 @@ export interface JsonBlock {
   header?: JsonHeader
   transactions?: JsonTx[]
   uncleHeaders?: JsonHeader[]
+  withdrawals?: JsonRpcWithdrawal[]
 }
 
 /**
@@ -150,6 +163,14 @@ export interface JsonHeader {
   mixHash?: string
   nonce?: string
   baseFeePerGas?: string
+  withdrawalsRoot?: string
+}
+
+export interface JsonRpcWithdrawal {
+  index: string // QUANTITY - bigint 8 bytes
+  validatorIndex: string // QUANTITY - bigint 8 bytes
+  address: string // DATA, 20 Bytes  address to withdraw to
+  amount: string // QUANTITY - bigint amount in wei 32 bytes
 }
 
 /*
@@ -177,4 +198,5 @@ export interface JsonRpcBlock {
   transactions: Array<JsonRpcTx | string> // Array of transaction objects, or 32 Bytes transaction hashes depending on the last given parameter.
   uncles: string[] // Array of uncle hashes
   baseFeePerGas?: string // If EIP-1559 is enabled for this block, returns the base fee per gas
+  withdrawals?: Array<JsonRpcWithdrawal> // If EIP-4895 is enabled for this block, array of withdrawals
 }