Access and set FuelCosts

stellar · Jun 1, 2023 · 1a2bc7f · 1a2bc7f
1 parent 10212cf
commit 1a2bc7f
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/crates/wasmi/src/engine/config.rs b/crates/wasmi/src/engine/config.rs
@@ -390,6 +390,11 @@ impl Config {
         self
     }
 
+    pub fn set_fuel_costs(&mut self, costs: FuelCosts) -> &mut Self {
+        self.fuel_costs = costs;
+        self
+    }
+
     /// Returns the [`FuelConsumptionMode`] for the [`Engine`].
     ///
     /// Returns `None` if fuel metering is disabled for the [`Engine`].

diff --git a/crates/wasmi/src/engine/executor.rs b/crates/wasmi/src/engine/executor.rs
@@ -727,7 +727,8 @@ impl<'ctx, 'engine> Executor<'ctx, 'engine> {
         }
     }
 
-    /// Consume an amount of memory fuel specified by `delta`.
+    /// Consume an amount of memory fuel specified by `bytes`. The memory fuel cost
+    /// is always 1 per byte.
     #[inline(always)]
     fn consume_mem_fuel(&mut self, bytes: u64) -> Result<u64, TrapCode>
     {

diff --git a/crates/wasmi/src/engine/mod.rs b/crates/wasmi/src/engine/mod.rs
@@ -19,7 +19,7 @@ mod tests;
 pub use self::{
     bytecode::DropKeep,
     code_map::CompiledFunc,
-    config::{Config, FuelConsumptionMode},
+    config::{Config, FuelConsumptionMode, FuelCosts},
     func_builder::{
         FuncBuilder,
         FuncTranslatorAllocations,

diff --git a/crates/wasmi/src/lib.rs b/crates/wasmi/src/lib.rs
@@ -123,6 +123,7 @@ pub use self::{
         Config,
         Engine,
         FuelConsumptionMode,
+        FuelCosts,
         ResumableCall,
         ResumableInvocation,
         StackLimits,