38 lines
1.5 KiB
Nim
38 lines
1.5 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2018 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or http://opensource.org/licenses/MIT)
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
import
|
|
chronicles, strformat, eth/common, # GasInt
|
|
../../errors, ../../vm_types
|
|
|
|
logScope:
|
|
topics = "vm gas"
|
|
|
|
proc init*(m: var GasMeter, startGas: GasInt) =
|
|
m.startGas = startGas
|
|
m.gasRemaining = m.startGas
|
|
m.gasRefunded = 0
|
|
|
|
proc consumeGas*(gasMeter: var GasMeter; amount: GasInt; reason: string) =
|
|
if amount > gasMeter.gasRemaining:
|
|
raise newException(OutOfGas,
|
|
&"Out of gas: Needed {amount} - Remaining {gasMeter.gasRemaining} - Reason: {reason}")
|
|
gasMeter.gasRemaining -= amount
|
|
trace "GAS CONSUMPTION", total = gasMeter.gasRemaining + amount, amount, remaining = gasMeter.gasRemaining, reason
|
|
|
|
proc returnGas*(gasMeter: var GasMeter; amount: GasInt) =
|
|
gasMeter.gasRemaining += amount
|
|
trace "GAS RETURNED", consumed = gasMeter.gasRemaining - amount, amount, remaining = gasMeter.gasRemaining
|
|
|
|
proc refundGas*(gasMeter: var GasMeter; amount: GasInt) =
|
|
gasMeter.gasRefunded += amount
|
|
trace "GAS REFUND", consumed = gasMeter.gasRemaining - amount, amount, refunded = gasMeter.gasRefunded
|
|
|
|
proc resetGas*(gasMeter: var GasMeter) =
|
|
gasMeter.gasRemaining = gasMeter.startGas
|
|
gasMeter.gasRefunded = 0
|