-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
9c5a319
commit 9dafd13
Showing
14 changed files
with
269 additions
and
77 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,7 +1,126 @@ | ||
| use wasmer::wasmparser::Operator; | ||
| use wasmer::{wasmparser::Operator, FunctionEnvMut, WASM_PAGE_SIZE}; | ||
| use wasmer_middlewares::metering::{get_remaining_points, set_remaining_points, MeteringPoints}; | ||
|
|
||
| use crate::{errors::Result, RuntimeError, VmContext}; | ||
|
|
||
| pub fn is_accounting(operator: &Operator) -> bool { | ||
| matches!( | ||
| operator, | ||
| Operator::Loop { .. } | ||
| | Operator::End | ||
| | Operator::If { .. } | ||
| | Operator::Else | ||
| | Operator::Br { .. } | ||
| | Operator::BrTable { .. } | ||
| | Operator::BrIf { .. } | ||
| | Operator::Call { .. } | ||
| | Operator::CallIndirect { .. } | ||
| | Operator::Return | ||
| | Operator::Throw { .. } | ||
| | Operator::ThrowRef | ||
| | Operator::Rethrow { .. } | ||
| | Operator::Delegate { .. } | ||
| | Operator::Catch { .. } | ||
| | Operator::ReturnCall { .. } | ||
| | Operator::ReturnCallIndirect { .. } | ||
| | Operator::BrOnCast { .. } | ||
| | Operator::BrOnCastFail { .. } | ||
| | Operator::CallRef { .. } | ||
| | Operator::ReturnCallRef { .. } | ||
| | Operator::BrOnNull { .. } | ||
| | Operator::BrOnNonNull { .. } | ||
| ) | ||
| } | ||
|
|
||
| const GAS_PER_OPERATION: u64 = 115; | ||
| const GAS_ACCOUNTING_MULTIPLIER: u64 = 14; | ||
| const GAS_MEMORY_GROW_BASE: u64 = 1_000_000; | ||
|
|
||
| // Gas for reading and writing bytes | ||
| pub const GAS_PER_BYTE: u64 = 10_000; | ||
| const GAS_PER_BYTE_EXECUTION_RESULT: u64 = 10_000_000; | ||
|
|
||
| const GAS_HTTP_FETCH_BASE: u64 = 1_000_000_000; | ||
| const GAS_BN254_VERIFY_BASE: u64 = 10_000_000; | ||
| const GAS_PROXY_HTTP_FETCH_BASE: u64 = 5_000_000_000; | ||
| const GAS_SECP256K1_BASE: u64 = 10_000_000; | ||
| const GAS_KECCAK256_BASE: u64 = 10_000_000; | ||
| pub const GAS_STARTUP: u64 = 5_000_000_000_000; | ||
|
|
||
| /// Gas cost for each operator | ||
| /// TODO: For now we give everything an equal gas cost, we should expand this | ||
| pub fn get_wasm_operation_cost(_operator: &Operator) -> u64 { | ||
| 1 | ||
| pub fn get_wasm_operation_gas_cost(operator: &Operator) -> u64 { | ||
| if is_accounting(operator) { | ||
| return GAS_PER_OPERATION * GAS_ACCOUNTING_MULTIPLIER; | ||
| } | ||
|
|
||
| match operator { | ||
| Operator::MemoryGrow { mem, mem_byte: _ } => { | ||
| GAS_MEMORY_GROW_BASE + ((WASM_PAGE_SIZE as u64 * *mem as u64) * GAS_PER_BYTE) | ||
| } | ||
| _ => GAS_PER_OPERATION, | ||
| } | ||
| } | ||
|
|
||
| #[derive(Debug)] | ||
| pub enum ExternalCallType { | ||
| /// Takes as argument the bytes length | ||
| ExecutionResult(u64), | ||
| /// Takes as argument the bytes length | ||
| HttpFetchRequest(u64), | ||
| /// Takes as argument the bytes length | ||
| HttpFetchResponse(u64), | ||
| /// Takes as argument the length of the message | ||
| Bn254Verify(u64), | ||
| /// Takes as argument the bytes length | ||
| ProxyHttpFetchRequest(u64), | ||
| /// Takes as argument the length of the message | ||
| Secp256k1Verify(u64), | ||
| /// Takes as argument the length of the message | ||
| Keccak256(u64), | ||
| } | ||
|
|
||
| pub fn check_enough_gas(gas_cost: u64, remaining_gas: u64, gas_limit: u64) -> Result<u64> { | ||
| let gas_used = gas_limit - remaining_gas; | ||
|
|
||
| if (gas_cost + gas_used) > gas_limit { | ||
| return Err(RuntimeError::OutOfGas); | ||
| } | ||
|
|
||
| Ok(remaining_gas - gas_cost) | ||
| } | ||
|
|
||
| pub fn apply_gas_cost(external_call_type: ExternalCallType, env: &mut FunctionEnvMut<'_, VmContext>) -> Result<()> { | ||
| let context: &VmContext = env.data(); | ||
| let instance = match &context.instance { | ||
| None => Err(RuntimeError::VmHostError( | ||
| "Instance on VmContext was not set".to_string(), | ||
| )), | ||
| Some(v) => Ok(v.clone()), | ||
| }?; | ||
|
|
||
| if let Some(gas_limit) = context.call_data.gas_limit { | ||
| let remaining_gas = match get_remaining_points(env, &instance) { | ||
| MeteringPoints::Exhausted => 0, | ||
| MeteringPoints::Remaining(remaining_gas) => remaining_gas, | ||
| }; | ||
|
|
||
| let gas_cost = match external_call_type { | ||
| ExternalCallType::ExecutionResult(bytes_length) => GAS_PER_BYTE_EXECUTION_RESULT * bytes_length, | ||
| ExternalCallType::HttpFetchRequest(bytes_length) => GAS_HTTP_FETCH_BASE + (GAS_PER_BYTE * bytes_length), | ||
| ExternalCallType::HttpFetchResponse(bytes_length) => GAS_PER_BYTE * bytes_length, | ||
| ExternalCallType::Bn254Verify(bytes_length) => GAS_BN254_VERIFY_BASE + (GAS_PER_BYTE * bytes_length), | ||
| ExternalCallType::ProxyHttpFetchRequest(bytes_length) => { | ||
| GAS_PROXY_HTTP_FETCH_BASE + (GAS_PER_BYTE * bytes_length) | ||
| } | ||
| ExternalCallType::Secp256k1Verify(bytes_length) => { | ||
| GAS_SECP256K1_BASE + GAS_KECCAK256_BASE + (GAS_PER_BYTE * bytes_length) | ||
| } | ||
| ExternalCallType::Keccak256(bytes_length) => GAS_KECCAK256_BASE + (GAS_PER_BYTE * bytes_length), | ||
| }; | ||
|
|
||
| let gas_left = check_enough_gas(gas_cost, remaining_gas, gas_limit)?; | ||
| set_remaining_points(env, &instance, gas_left); | ||
| } | ||
|
|
||
| Ok(()) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.