47 lines
2.2 KiB
Nim
47 lines
2.2 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
|
|
strformat, eth_common, # GasInt
|
|
../../logging, ../../errors, ../../vm_types
|
|
|
|
proc init*(m: var GasMeter, startGas: GasInt) =
|
|
m.startGas = startGas
|
|
m.gasRemaining = m.startGas
|
|
m.gasRefunded = 0
|
|
m.logger = logging.getLogger("gas")
|
|
|
|
proc consumeGas*(gasMeter: var GasMeter; amount: GasInt; reason: string) =
|
|
#if amount < 0.u256:
|
|
# raise newException(ValidationError, "Gas consumption amount must be positive")
|
|
# Alternatively: use a range type `range[0'i64 .. high(int64)]`
|
|
# https://github.com/status-im/nimbus/issues/35#issuecomment-391726518
|
|
if amount > gasMeter.gasRemaining:
|
|
raise newException(OutOfGas,
|
|
&"Out of gas: Needed {amount} - Remaining {gasMeter.gasRemaining} - Reason: {reason}")
|
|
gasMeter.gasRemaining -= amount
|
|
gasMeter.logger.trace(
|
|
&"GAS CONSUMPTION: {gasMeter.gasRemaining + amount} - {amount} -> {gasMeter.gasRemaining} ({reason})")
|
|
|
|
proc returnGas*(gasMeter: var GasMeter; amount: GasInt) =
|
|
#if amount < 0.int256:
|
|
# raise newException(ValidationError, "Gas return amount must be positive")
|
|
# Alternatively: use a range type `range[0'i64 .. high(int64)]`
|
|
# https://github.com/status-im/nimbus/issues/35#issuecomment-391726518
|
|
gasMeter.gasRemaining += amount
|
|
gasMeter.logger.trace(
|
|
&"GAS RETURNED: {gasMeter.gasRemaining - amount} + {amount} -> {gasMeter.gasRemaining}")
|
|
|
|
proc refundGas*(gasMeter: var GasMeter; amount: GasInt) =
|
|
#if amount < 0.int256:
|
|
# raise newException(ValidationError, "Gas refund amount must be positive")
|
|
# Alternatively: use a range type `range[0'i64 .. high(int64)]`
|
|
# https://github.com/status-im/nimbus/issues/35#issuecomment-391726518
|
|
gasMeter.gasRefunded += amount
|
|
gasMeter.logger.trace(
|
|
&"GAS REFUND: {gasMeter.gasRemaining - amount} + {amount} -> {gasMeter.gasRefunded}")
|