fix

Author: pgherveou

substrate/frame/revive/src/evm/block_hash/block_builder.rs

@@ -31,7 +31,7 @@ use crate::{
 use alloc::{vec, vec::Vec};
 
 use codec::{Decode, Encode};
-use frame_support::{weights::Weight, DefaultNoBound};
+use frame_support::DefaultNoBound;
 use scale_info::TypeInfo;
 use sp_core::{keccak_256, H160, H256, U256};
 
@@ -97,7 +97,7 @@ impl<T: crate::Config> EthereumBlockBuilder<T> {
 		&mut self,
 		transaction_encoded: Vec<u8>,
 		success: bool,
-		gas_used: Weight,
+		gas_used: U256,
 		encoded_logs: Vec<u8>,
 		receipt_bloom: LogsBloom,
 	) {
@@ -108,7 +108,7 @@ impl<T: crate::Config> EthereumBlockBuilder<T> {
 		let transaction_type = Self::extract_transaction_type(transaction_encoded.as_slice());
 
 		// Update gas and logs bloom.
-		self.gas_used = self.gas_used.saturating_add(gas_used.ref_time().into());
+		self.gas_used = self.gas_used.saturating_add(gas_used);
 		self.logs_bloom.accrue_bloom(&receipt_bloom);
 
 		// Update the receipt trie.
@@ -120,7 +120,7 @@ impl<T: crate::Config> EthereumBlockBuilder<T> {
 			transaction_type,
 		);
 
-		self.gas_info.push(ReceiptGasInfo { gas_used: gas_used.ref_time().into() });
+		self.gas_info.push(ReceiptGasInfo { gas_used });
 
 		// The first transaction and receipt are returned to be stored in the pallet storage.
 		// The index of the incremental hash builders already expects the next items.

substrate/frame/revive/src/evm/block_storage.rs

@@ -18,7 +18,6 @@ use crate::{
 	evm::block_hash::{AccumulateReceipt, EthereumBlockBuilder, LogsBloom},
 	limits,
 	sp_runtime::traits::One,
-	weights::WeightInfo,
 	BlockHash, Config, EthBlockBuilderIR, EthereumBlock, Event, Pallet, ReceiptInfoData,
 	UniqueSaturatedInto, H160, H256,
 };
@@ -27,7 +26,6 @@ use environmental::environmental;
 use frame_support::{
 	pallet_prelude::{DispatchError, DispatchResultWithPostInfo},
 	storage::with_transaction,
-	weights::Weight,
 };
 use sp_core::U256;
 use sp_runtime::TransactionOutcome;
@@ -41,6 +39,27 @@ pub const BLOCK_HASH_COUNT: u32 = 256;
 // that are needed to construct the final transaction receipt.
 environmental!(receipt: AccumulateReceipt);
 
+/// Result of an Ethereum context call execution.
+pub(crate) struct EthereumCallResult {
+	/// The amount of gas used by the call.
+	pub gas_used: U256,
+	/// The dispatch result with post-dispatch information.
+	pub result: DispatchResultWithPostInfo,
+}
+
+impl EthereumCallResult {
+	/// Create a new `EthereumCallResult` from a native fee and effective gas price.
+	pub(crate) fn new<T: Config>(
+		native_fee: crate::BalanceOf<T>,
+		effective_gas_price: U256,
+		result: DispatchResultWithPostInfo,
+	) -> Self {
+		let eth_fee = Pallet::<T>::convert_native_to_evm(native_fee);
+		let gas_used = eth_fee / effective_gas_price;
+		Self { gas_used, result }
+	}
+}
+
 /// Capture the Ethereum log for the current transaction.
 ///
 /// This method does nothing if called from outside of the ethereum context.
@@ -72,23 +91,19 @@ pub fn bench_with_ethereum_context<R>(f: impl FnOnce() -> R) -> R {
 ///
 /// # Parameters
 /// - transaction_encoded: The RLP encoded transaction bytes.
-/// - call: A closure that executes the transaction logic and returns the gas consumed and result.
+/// - call: A closure that executes the transaction logic and returns an `EthereumCallResult`.
 pub fn with_ethereum_context<T: Config>(
 	transaction_encoded: Vec<u8>,
-	call: impl FnOnce() -> (Weight, DispatchResultWithPostInfo),
+	call: impl FnOnce() -> EthereumCallResult,
 ) -> DispatchResultWithPostInfo {
 	receipt::using(&mut AccumulateReceipt::new(), || {
-		let (err, gas_consumed, mut post_info) =
+		let (err, gas_consumed, post_info) =
 			with_transaction(|| -> TransactionOutcome<Result<_, DispatchError>> {
-				let (gas_consumed, result) = call();
+				let EthereumCallResult { gas_used, result } = call();
 				match result {
-					Ok(post_info) =>
-						TransactionOutcome::Commit(Ok((None, gas_consumed, post_info))),
-					Err(err) => TransactionOutcome::Rollback(Ok((
-						Some(err.error),
-						gas_consumed,
-						err.post_info,
-					))),
+					Ok(post_info) => TransactionOutcome::Commit(Ok((None, gas_used, post_info))),
+					Err(err) =>
+						TransactionOutcome::Rollback(Ok((Some(err.error), gas_used, err.post_info))),
 				}
 			})?;
 
@@ -106,10 +121,6 @@ pub fn with_ethereum_context<T: Config>(
 			deposit_eth_extrinsic_revert_event::<T>(crate::Error::<T>::BenchmarkingError.into());
 
 			crate::block_storage::process_transaction::<T>(transaction_encoded, true, gas_consumed);
-			post_info
-				.actual_weight
-				.as_mut()
-				.map(|w| w.saturating_reduce(T::WeightInfo::deposit_eth_extrinsic_revert_event()));
 			Ok(post_info)
 		}
 	})
@@ -174,11 +185,7 @@ pub fn on_finalize_build_eth_block<T: Config>(
 /// This stores the RLP encoded transaction and receipt details into storage.
 ///
 /// The data is used during the `on_finalize` hook to reconstruct the ETH block.
-pub fn process_transaction<T: Config>(
-	transaction_encoded: Vec<u8>,
-	success: bool,
-	gas_used: Weight,
-) {
+pub fn process_transaction<T: Config>(transaction_encoded: Vec<u8>, success: bool, gas_used: U256) {
 	// Method returns `None` only when called from outside of the ethereum context.
 	// This is not the case here, since this is called from within the
 	// ethereum context.

substrate/frame/revive/src/evm/fees.rs

@@ -105,10 +105,18 @@ pub trait InfoT<T: Config>: seal::Sealed {
 		Zero::zero()
 	}
 
-	/// Makes sure that not too much storage deposit was withdrawn.
-	fn ensure_not_overdrawn(
+	/// Compute the actual post_dispatch fee
+	fn compute_actual_fee(
 		_encoded_len: u32,
 		_info: &DispatchInfo,
+		_result: &DispatchResultWithPostInfo,
+	) -> BalanceOf<T> {
+		Default::default()
+	}
+
+	/// Makes sure that not too much storage deposit was withdrawn.
+	fn ensure_not_overdrawn(
+		_fee: BalanceOf<T>,
 		result: DispatchResultWithPostInfo,
 	) -> DispatchResultWithPostInfo {
 		result
@@ -259,9 +267,29 @@ where
 		.saturating_add(Self::length_to_fee(encoded_len))
 	}
 
-	fn ensure_not_overdrawn(
+	fn compute_actual_fee(
 		encoded_len: u32,
 		info: &DispatchInfo,
+		result: &DispatchResultWithPostInfo,
+	) -> BalanceOf<E::Config> {
+		let mut post_info = match result {
+			Ok(post_info) => post_info,
+			Err(err) => &err.post_info,
+		}
+		.clone();
+
+		post_info
+			.actual_weight
+			.as_mut()
+			.map(|w| w.saturating_accrue(info.extension_weight));
+
+		log::debug!(target: LOG_TARGET, "revive: post_info:{post_info:?}");
+		<TxPallet<E::Config>>::compute_actual_fee(encoded_len, info, &post_info, Zero::zero())
+			.into()
+	}
+
+	fn ensure_not_overdrawn(
+		fee: BalanceOf<E::Config>,
 		result: DispatchResultWithPostInfo,
 	) -> DispatchResultWithPostInfo {
 		// if tx is already failing we can ignore
@@ -270,9 +298,6 @@ where
 			return result;
 		};
 
-		let fee: BalanceOf<E::Config> =
-			<TxPallet<E::Config>>::compute_actual_fee(encoded_len, info, &post_info, Zero::zero())
-				.into();
 		let available = Self::remaining_txfee();
 		if fee > available {
 			log::debug!(target: LOG_TARGET, "Drew too much from the txhold. \

substrate/frame/revive/src/lib.rs

@@ -1267,8 +1267,10 @@ pub mod pallet {
 					output.gas_consumed,
 					base_info.call_weight,
 				);
-				let result = T::FeeInfo::ensure_not_overdrawn(encoded_len, &info, result);
-				(output.gas_consumed, result)
+
+				let native_fee = T::FeeInfo::compute_actual_fee(encoded_len, &info, &result);
+				let result = T::FeeInfo::ensure_not_overdrawn(native_fee, result);
+				block_storage::EthereumCallResult::new::<T>(native_fee, effective_gas_price, result)
 			})
 		}
 
@@ -1329,8 +1331,11 @@ pub mod pallet {
 
 				let result =
 					dispatch_result(output.result, output.gas_consumed, base_info.call_weight);
-				let result = T::FeeInfo::ensure_not_overdrawn(encoded_len, &info, result);
-				(output.gas_consumed, result)
+
+				let native_fee: BalanceOf<T> =
+					T::FeeInfo::compute_actual_fee(encoded_len, &info, &result);
+				let result = T::FeeInfo::ensure_not_overdrawn(native_fee, result);
+				block_storage::EthereumCallResult::new::<T>(native_fee, effective_gas_price, result)
 			})
 		}