use U256 for data fee

bluealloy · Sep 28, 2023 · ce99cb0 · ce99cb0
1 parent 15adbc4
commit ce99cb0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/crates/primitives/src/env.rs b/crates/primitives/src/env.rs
@@ -424,10 +424,10 @@ impl Env {
     ///
     /// [EIP-4844]: https://eips.ethereum.org/EIPS/eip-4844
     #[inline]
-    pub fn calc_data_fee(&self) -> Option<u128> {
-        self.block
-            .get_blob_gasprice()
-            .map(|blob_gas_price| blob_gas_price as u128 * self.tx.get_total_blob_gas() as u128)
+    pub fn calc_data_fee(&self) -> Option<U256> {
+        self.block.get_blob_gasprice().map(|blob_gas_price| {
+            U256::from(blob_gas_price).saturating_mul(U256::from(self.tx.get_total_blob_gas()))
+        })
     }
 
     /// Validate the block environment.

diff --git a/crates/revm/src/evm_impl.rs b/crates/revm/src/evm_impl.rs
@@ -162,7 +162,7 @@ impl<'a, GSPEC: Spec, DB: Database, const INSPECT: bool> Transact<DB::Error>
         // EIP-4844
         if GSPEC::enabled(CANCUN) {
             let data_fee = env.calc_data_fee().expect("already checked");
-            gas_cost = gas_cost.saturating_add(U256::from(data_fee));
+            gas_cost = gas_cost.saturating_add(data_fee);
         }
 
         caller_account.info.balance = caller_account.info.balance.saturating_sub(gas_cost);