feat: replace DIFFICULTY with PREVRANDAO since merge/EIP-4399

Author: vimpunk

core/src/error.rs

@@ -158,6 +158,10 @@ pub enum ExitError {
 	/// Init code exceeds limit (runtime).
 	#[cfg_attr(feature = "with-codec", codec(index = 7))]
 	InitCodeLimit,
+
+	/// Randomness was not configured for the PREVRANDAO opcode
+	#[cfg_attr(feature = "with-codec", codec(index = 7))]
+	MissingRandomness,
 }
 
 impl From<ExitError> for ExitReason {

core/src/opcode.rs

@@ -217,6 +217,8 @@ impl Opcode {
 	pub const NUMBER: Opcode = Opcode(0x43);
 	/// `DIFFICULTY`
 	pub const DIFFICULTY: Opcode = Opcode(0x44);
+	/// `PREVRANDAO` replaces `DIFFICULTY` since the Merge
+	pub const PREVRANDAO: Opcode = Self::DIFFICULTY;
 	/// `GASLIMIT`
 	pub const GASLIMIT: Opcode = Opcode(0x45);
 	/// `SLOAD`

runtime/src/eval/mod.rs

@@ -40,7 +40,13 @@ pub fn eval<H: Handler>(state: &mut Runtime, opcode: Opcode, handler: &mut H) ->
 		Opcode::COINBASE => system::coinbase(state, handler),
 		Opcode::TIMESTAMP => system::timestamp(state, handler),
 		Opcode::NUMBER => system::number(state, handler),
-		Opcode::DIFFICULTY => system::difficulty(state, handler),
+		Opcode::DIFFICULTY => {
+			if state.config().has_prevrandao {
+				system::prevrandao(state, handler)
+			} else {
+				system::difficulty(state, handler)
+			}
+		}
 		Opcode::GASLIMIT => system::gaslimit(state, handler),
 		Opcode::SLOAD => system::sload(state, handler),
 		Opcode::SSTORE => system::sstore(state, handler),

runtime/src/eval/system.rs

@@ -185,6 +185,22 @@ pub fn difficulty<H: Handler>(runtime: &mut Runtime, handler: &H) -> Control<H>
 	Control::Continue
 }
 
+pub fn prevrandao<H: Handler>(runtime: &mut Runtime, handler: &H) -> Control<H> {
+	match handler.block_randomness() {
+		Some(rand) => {
+			// TODO: clean this up
+			// convert between u256 and h256
+			let mut buf = [0u8; 32];
+			for (i, word) in rand.0.iter().rev().enumerate() {
+				buf[i * 8..(i + 1) * 8].copy_from_slice(&word.to_be_bytes());
+			}
+			push!(runtime, H256(buf));
+			Control::Continue
+		}
+		None => Control::Exit(ExitError::MissingRandomness.into()),
+	}
+}
+
 pub fn gaslimit<H: Handler>(runtime: &mut Runtime, handler: &H) -> Control<H> {
 	push_u256!(runtime, handler.block_gas_limit());
 	Control::Continue

runtime/src/handler.rs

@@ -54,6 +54,8 @@ pub trait Handler {
 	fn block_timestamp(&self) -> U256;
 	/// Get environmental block difficulty.
 	fn block_difficulty(&self) -> U256;
+	/// Get environmental block randomness.
+	fn block_randomness(&self) -> Option<U256>;
 	/// Get environmental gas limit.
 	fn block_gas_limit(&self) -> U256;
 	/// Environmental block base fee.

runtime/src/lib.rs

@@ -114,7 +114,7 @@ pub struct Runtime<'config> {
 	return_data_len: U256,
 	return_data_offset: U256,
 	context: Context,
-	_config: &'config Config,
+	config: &'config Config,
 }
 
 impl<'config> Runtime<'config> {
@@ -132,7 +132,7 @@ impl<'config> Runtime<'config> {
 			return_data_len: U256::zero(),
 			return_data_offset: U256::zero(),
 			context,
-			_config: config,
+			config,
 		}
 	}
 
@@ -146,6 +146,11 @@ impl<'config> Runtime<'config> {
 		&self.context
 	}
 
+	/// Get a reference to the runtime config.
+	pub fn config(&self) -> &'config Config {
+		self.config
+	}
+
 	/// Step the runtime.
 	pub fn step<'a, H: Handler>(
 		&'a mut self,
@@ -287,6 +292,9 @@ pub struct Config {
 	pub has_base_fee: bool,
 	/// Has PUSH0 opcode. See [EIP-3855](https://github.com/ethereum/EIPs/blob/master/EIPS/eip-3855.md)
 	pub has_push0: bool,
+	/// Has PREVRANDAO opcode, replacing the DIFFICULTY opcode.
+	/// See [EIP-4399](https://github.com/ethereum/EIPs/blob/master/EIPS/eip-4399.md)
+	pub has_prevrandao: bool,
 	/// Whether the gasometer is running in estimate mode.
 	pub estimate: bool,
 }
@@ -342,6 +350,7 @@ impl Config {
 			has_ext_code_hash: false,
 			has_base_fee: false,
 			has_push0: false,
+			has_prevrandao: false,
 			estimate: false,
 		}
 	}
@@ -396,6 +405,7 @@ impl Config {
 			has_ext_code_hash: true,
 			has_base_fee: false,
 			has_push0: false,
+			has_prevrandao: false,
 			estimate: false,
 		}
 	}
@@ -410,6 +420,11 @@ impl Config {
 		Self::config_with_derived_values(DerivedConfigInputs::london())
 	}
 
+	/// The Merge (Paris) hard fork configuration.
+	pub const fn merge() -> Config {
+		Self::config_with_derived_values(DerivedConfigInputs::merge())
+	}
+
 	/// Shanghai hard fork configuration.
 	pub const fn shanghai() -> Config {
 		Self::config_with_derived_values(DerivedConfigInputs::shanghai())
@@ -423,6 +438,7 @@ impl Config {
 			decrease_clears_refund,
 			has_base_fee,
 			has_push0,
+			has_prevrandao,
 			disallow_executable_format,
 			warm_coinbase_address,
 			max_initcode_size,
@@ -488,6 +504,7 @@ impl Config {
 			has_ext_code_hash: true,
 			has_base_fee,
 			has_push0,
+			has_prevrandao,
 			estimate: false,
 		}
 	}
@@ -502,6 +519,7 @@ struct DerivedConfigInputs {
 	decrease_clears_refund: bool,
 	has_base_fee: bool,
 	has_push0: bool,
+	has_prevrandao: bool,
 	disallow_executable_format: bool,
 	warm_coinbase_address: bool,
 	max_initcode_size: Option<usize>,
@@ -516,6 +534,7 @@ impl DerivedConfigInputs {
 			decrease_clears_refund: false,
 			has_base_fee: false,
 			has_push0: false,
+			has_prevrandao: false,
 			disallow_executable_format: false,
 			warm_coinbase_address: false,
 			max_initcode_size: None,
@@ -530,6 +549,22 @@ impl DerivedConfigInputs {
 			decrease_clears_refund: true,
 			has_base_fee: true,
 			has_push0: false,
+			has_prevrandao: false,
+			disallow_executable_format: true,
+			warm_coinbase_address: false,
+			max_initcode_size: None,
+		}
+	}
+
+	const fn merge() -> Self {
+		Self {
+			gas_storage_read_warm: 100,
+			gas_sload_cold: 2100,
+			gas_access_list_storage_key: 1900,
+			decrease_clears_refund: true,
+			has_base_fee: true,
+			has_push0: false,
+			has_prevrandao: true,
 			disallow_executable_format: true,
 			warm_coinbase_address: false,
 			max_initcode_size: None,
@@ -544,6 +579,7 @@ impl DerivedConfigInputs {
 			decrease_clears_refund: true,
 			has_base_fee: true,
 			has_push0: true,
+			has_prevrandao: true,
 			disallow_executable_format: true,
 			warm_coinbase_address: true,
 			max_initcode_size: Some(0xC000),

src/backend/memory.rs

@@ -31,6 +31,11 @@ pub struct MemoryVicinity {
 	pub block_gas_limit: U256,
 	/// Environmental base fee per gas.
 	pub block_base_fee_per_gas: U256,
+	/// Environmental randomness.
+	///
+	/// In Ethereum, this is the randomness beacon provided by the beacon
+	/// chain and is only enabled post Merge.
+	pub randomness: Option<U256>,
 }
 
 /// Account information of a memory backend.
@@ -110,6 +115,9 @@ impl<'vicinity> Backend for MemoryBackend<'vicinity> {
 	fn block_difficulty(&self) -> U256 {
 		self.vicinity.block_difficulty
 	}
+	fn block_randomness(&self) -> Option<U256> {
+		self.vicinity.randomness
+	}
 	fn block_gas_limit(&self) -> U256 {
 		self.vicinity.block_gas_limit
 	}

src/backend/mod.rs

@@ -66,6 +66,8 @@ pub trait Backend {
 	fn block_timestamp(&self) -> U256;
 	/// Environmental block difficulty.
 	fn block_difficulty(&self) -> U256;
+	/// Get environmental block randomness.
+	fn block_randomness(&self) -> Option<U256>;
 	/// Environmental block gas limit.
 	fn block_gas_limit(&self) -> U256;
 	/// Environmental block base fee.

src/executor/stack/executor.rs

@@ -1102,6 +1102,9 @@ impl<'config, 'precompiles, S: StackState<'config>, P: PrecompileSet> Handler
 	fn block_difficulty(&self) -> U256 {
 		self.state.block_difficulty()
 	}
+	fn block_randomness(&self) -> Option<U256> {
+		self.state.block_randomness()
+	}
 	fn block_gas_limit(&self) -> U256 {
 		self.state.block_gas_limit()
 	}

src/executor/stack/memory.rs

@@ -421,6 +421,9 @@ impl<'backend, 'config, B: Backend> Backend for MemoryStackState<'backend, 'conf
 	fn block_difficulty(&self) -> U256 {
 		self.backend.block_difficulty()
 	}
+	fn block_randomness(&self) -> Option<U256> {
+		self.backend.block_randomness()
+	}
 	fn block_gas_limit(&self) -> U256 {
 		self.backend.block_gas_limit()
 	}