mirror of
https://github.com/pezkuwichain/revive-differential-tests.git
synced 2026-04-24 05:27:56 +00:00
Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 83c20b1be3 | |||
| 075c8235a7 | |||
| 43e0d0e592 | |||
| 2373872230 | |||
| e3723e780a |
Generated
-1
@@ -4012,7 +4012,6 @@ version = "0.1.0"
|
|||||||
dependencies = [
|
dependencies = [
|
||||||
"alloy",
|
"alloy",
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"futures",
|
|
||||||
"once_cell",
|
"once_cell",
|
||||||
"tokio",
|
"tokio",
|
||||||
"tracing",
|
"tracing",
|
||||||
|
|||||||
@@ -25,7 +25,6 @@ alloy-primitives = "1.2.1"
|
|||||||
alloy-sol-types = "1.2.1"
|
alloy-sol-types = "1.2.1"
|
||||||
anyhow = "1.0"
|
anyhow = "1.0"
|
||||||
clap = { version = "4", features = ["derive"] }
|
clap = { version = "4", features = ["derive"] }
|
||||||
futures = { version = "0.3.31" }
|
|
||||||
hex = "0.4.3"
|
hex = "0.4.3"
|
||||||
reqwest = { version = "0.12.15", features = ["blocking", "json"] }
|
reqwest = { version = "0.12.15", features = ["blocking", "json"] }
|
||||||
once_cell = "1.21"
|
once_cell = "1.21"
|
||||||
|
|||||||
+117
-221
@@ -1,16 +1,13 @@
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
use alloy::{
|
use alloy::{
|
||||||
hex,
|
json_abi::JsonAbi,
|
||||||
json_abi::{Function, JsonAbi},
|
|
||||||
network::TransactionBuilder,
|
network::TransactionBuilder,
|
||||||
primitives::{Address, Bytes},
|
primitives::{Address, Bytes, U256},
|
||||||
rpc::types::TransactionRequest,
|
rpc::types::TransactionRequest,
|
||||||
};
|
};
|
||||||
use alloy_primitives::U256;
|
|
||||||
use alloy_sol_types::SolValue;
|
|
||||||
use semver::VersionReq;
|
use semver::VersionReq;
|
||||||
use serde::{Deserialize, de::Deserializer};
|
use serde::Deserialize;
|
||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
|
|
||||||
#[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq)]
|
#[derive(Clone, Debug, Default, Deserialize, Eq, PartialEq)]
|
||||||
@@ -20,7 +17,6 @@ pub struct Input {
|
|||||||
pub comment: Option<String>,
|
pub comment: Option<String>,
|
||||||
#[serde(default = "default_instance")]
|
#[serde(default = "default_instance")]
|
||||||
pub instance: String,
|
pub instance: String,
|
||||||
#[serde(deserialize_with = "deserialize_method")]
|
|
||||||
pub method: Method,
|
pub method: Method,
|
||||||
pub calldata: Option<Calldata>,
|
pub calldata: Option<Calldata>,
|
||||||
pub expected: Option<Expected>,
|
pub expected: Option<Expected>,
|
||||||
@@ -48,58 +44,28 @@ pub struct ExpectedOutput {
|
|||||||
#[serde(untagged)]
|
#[serde(untagged)]
|
||||||
pub enum Calldata {
|
pub enum Calldata {
|
||||||
Single(String),
|
Single(String),
|
||||||
Compound(Vec<CalldataArg>),
|
Compound(Vec<String>),
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone, Debug, Deserialize, Eq, PartialEq)]
|
|
||||||
#[serde(untagged)]
|
|
||||||
pub enum CalldataArg {
|
|
||||||
Literal(String),
|
|
||||||
/// For example: `Contract.address`
|
|
||||||
AddressRef(String),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Specify how the contract is called.
|
/// Specify how the contract is called.
|
||||||
#[derive(Debug, Default, Clone, Eq, PartialEq)]
|
#[derive(Debug, Default, Deserialize, Clone, Eq, PartialEq)]
|
||||||
pub enum Method {
|
pub enum Method {
|
||||||
/// Initiate a deploy transaction, calling contracts constructor.
|
/// Initiate a deploy transaction, calling contracts constructor.
|
||||||
///
|
///
|
||||||
/// Indicated by `#deployer`.
|
/// Indicated by `#deployer`.
|
||||||
|
#[serde(rename = "#deployer")]
|
||||||
Deployer,
|
Deployer,
|
||||||
|
|
||||||
/// Does not calculate and insert a function selector.
|
/// Does not calculate and insert a function selector.
|
||||||
///
|
///
|
||||||
/// Indicated by `#fallback`.
|
/// Indicated by `#fallback`.
|
||||||
#[default]
|
#[default]
|
||||||
|
#[serde(rename = "#fallback")]
|
||||||
Fallback,
|
Fallback,
|
||||||
/// Call the public function with this selector.
|
|
||||||
///
|
|
||||||
/// Calculates the selector if neither deployer or fallback matches.
|
|
||||||
Function([u8; 4]),
|
|
||||||
}
|
|
||||||
|
|
||||||
fn deserialize_method<'de, D>(deserializer: D) -> Result<Method, D::Error>
|
/// Call the public function with the given name.
|
||||||
where
|
#[serde(untagged)]
|
||||||
D: Deserializer<'de>,
|
FunctionName(String),
|
||||||
{
|
|
||||||
Ok(match String::deserialize(deserializer)?.as_str() {
|
|
||||||
"#deployer" => Method::Deployer,
|
|
||||||
"#fallback" => Method::Fallback,
|
|
||||||
signature => {
|
|
||||||
let signature = if signature.ends_with(')') {
|
|
||||||
signature.to_string()
|
|
||||||
} else {
|
|
||||||
format!("{signature}()")
|
|
||||||
};
|
|
||||||
match Function::parse(&signature) {
|
|
||||||
Ok(function) => Method::Function(function.selector().0),
|
|
||||||
Err(error) => {
|
|
||||||
return Err(serde::de::Error::custom(format!(
|
|
||||||
"parsing function signature '{signature}' error: {error}"
|
|
||||||
)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Input {
|
impl Input {
|
||||||
@@ -119,7 +85,7 @@ impl Input {
|
|||||||
deployed_abis: &HashMap<String, JsonAbi>,
|
deployed_abis: &HashMap<String, JsonAbi>,
|
||||||
deployed_contracts: &HashMap<String, Address>,
|
deployed_contracts: &HashMap<String, Address>,
|
||||||
) -> anyhow::Result<Bytes> {
|
) -> anyhow::Result<Bytes> {
|
||||||
let Method::Function(selector) = self.method else {
|
let Method::FunctionName(ref function_name) = self.method else {
|
||||||
return Ok(Bytes::default()); // fallback or deployer — no input
|
return Ok(Bytes::default()); // fallback or deployer — no input
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -129,14 +95,17 @@ impl Input {
|
|||||||
|
|
||||||
tracing::trace!("ABI found for instance: {}", &self.instance);
|
tracing::trace!("ABI found for instance: {}", &self.instance);
|
||||||
|
|
||||||
// Find function by selector
|
// We follow the same logic that's implemented in the matter-labs-tester where they resolve
|
||||||
|
// the function name into a function selector and they assume that he function doesn't have
|
||||||
|
// any existing overloads.
|
||||||
|
// https://github.com/matter-labs/era-compiler-tester/blob/1dfa7d07cba0734ca97e24704f12dd57f6990c2c/compiler_tester/src/test/case/input/mod.rs#L158-L190
|
||||||
let function = abi
|
let function = abi
|
||||||
.functions()
|
.functions()
|
||||||
.find(|f| f.selector().0 == selector)
|
.find(|function| function.name.starts_with(function_name))
|
||||||
.ok_or_else(|| {
|
.ok_or_else(|| {
|
||||||
anyhow::anyhow!(
|
anyhow::anyhow!(
|
||||||
"Function with selector {:?} not found in ABI for the instance {:?}",
|
"Function with name {:?} not found in ABI for the instance {:?}",
|
||||||
selector,
|
function_name,
|
||||||
&self.instance
|
&self.instance
|
||||||
)
|
)
|
||||||
})?;
|
})?;
|
||||||
@@ -161,61 +130,28 @@ impl Input {
|
|||||||
&self.instance
|
&self.instance
|
||||||
);
|
);
|
||||||
|
|
||||||
let mut encoded = selector.to_vec();
|
// Allocating a vector that we will be using for the calldata. The vector size will be:
|
||||||
|
// 4 bytes for the function selector.
|
||||||
|
// function.inputs.len() * 32 bytes for the arguments (each argument is a U256).
|
||||||
|
//
|
||||||
|
// We're using indices in the following code in order to avoid the need for us to allocate
|
||||||
|
// a new buffer for each one of the resolved arguments.
|
||||||
|
let mut calldata = Vec::<u8>::with_capacity(4 + calldata_args.len() * 32);
|
||||||
|
calldata.extend(function.selector().0);
|
||||||
|
|
||||||
for (i, param) in function.inputs.iter().enumerate() {
|
for (arg_idx, arg) in calldata_args.iter().enumerate() {
|
||||||
let arg = calldata_args.get(i).unwrap();
|
match resolve_argument(arg, deployed_contracts) {
|
||||||
let encoded_arg = match arg {
|
Ok(resolved) => {
|
||||||
CalldataArg::Literal(value) => match param.ty.as_str() {
|
calldata.extend(resolved.to_be_bytes::<32>());
|
||||||
"uint256" | "uint" => {
|
}
|
||||||
let val: U256 = value.parse()?;
|
Err(error) => {
|
||||||
val.abi_encode()
|
tracing::error!(arg, arg_idx, ?error, "Failed to resolve argument");
|
||||||
}
|
return Err(error);
|
||||||
"uint24" => {
|
|
||||||
let val: u32 = value.parse()?;
|
|
||||||
(val & 0xFFFFFF).abi_encode()
|
|
||||||
}
|
|
||||||
"bool" => {
|
|
||||||
let val: bool = value.parse()?;
|
|
||||||
val.abi_encode()
|
|
||||||
}
|
|
||||||
"address" => {
|
|
||||||
let addr: Address = value.parse()?;
|
|
||||||
addr.abi_encode()
|
|
||||||
}
|
|
||||||
"string" => value.abi_encode(),
|
|
||||||
"bytes32" => {
|
|
||||||
let val = hex::decode(value.trim_start_matches("0x"))?;
|
|
||||||
let mut fixed = [0u8; 32];
|
|
||||||
fixed[..val.len()].copy_from_slice(&val);
|
|
||||||
fixed.abi_encode()
|
|
||||||
}
|
|
||||||
"uint256[]" | "uint[]" => {
|
|
||||||
let nums: Vec<u64> = serde_json::from_str(value)?;
|
|
||||||
nums.abi_encode()
|
|
||||||
}
|
|
||||||
"bytes" => {
|
|
||||||
let val = hex::decode(value.trim_start_matches("0x"))?;
|
|
||||||
val.abi_encode()
|
|
||||||
}
|
|
||||||
_ => anyhow::bail!("Unsupported type: {}", param.ty),
|
|
||||||
},
|
|
||||||
CalldataArg::AddressRef(name) => {
|
|
||||||
let contract_name = name.trim_end_matches(".address");
|
|
||||||
let addr = deployed_contracts
|
|
||||||
.get(contract_name)
|
|
||||||
.copied()
|
|
||||||
.ok_or_else(|| {
|
|
||||||
anyhow::anyhow!("Address for '{}' not found", contract_name)
|
|
||||||
})?;
|
|
||||||
addr.abi_encode()
|
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
encoded.extend(encoded_arg);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(Bytes::from(encoded))
|
Ok(calldata.into())
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Parse this input into a legacy transaction.
|
/// Parse this input into a legacy transaction.
|
||||||
@@ -244,12 +180,72 @@ fn default_caller() -> Address {
|
|||||||
"90F8bf6A479f320ead074411a4B0e7944Ea8c9C1".parse().unwrap()
|
"90F8bf6A479f320ead074411a4B0e7944Ea8c9C1".parse().unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// This function takes in the string calldata argument provided in the JSON input and resolves it
|
||||||
|
/// into a [`U256`] which is later used to construct the calldata.
|
||||||
|
///
|
||||||
|
/// # Note
|
||||||
|
///
|
||||||
|
/// This piece of code is taken from the matter-labs-tester repository which is licensed under MIT
|
||||||
|
/// or Apache. The original source code can be found here:
|
||||||
|
/// https://github.com/matter-labs/era-compiler-tester/blob/0ed598a27f6eceee7008deab3ff2311075a2ec69/compiler_tester/src/test/case/input/value.rs#L43-L146
|
||||||
|
fn resolve_argument(
|
||||||
|
value: &str,
|
||||||
|
deployed_contracts: &HashMap<String, Address>,
|
||||||
|
) -> anyhow::Result<U256> {
|
||||||
|
if let Some(instance) = value.strip_suffix(".address") {
|
||||||
|
Ok(U256::from_be_slice(
|
||||||
|
deployed_contracts
|
||||||
|
.get(instance)
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("Instance `{}` not found", instance))?
|
||||||
|
.as_ref(),
|
||||||
|
))
|
||||||
|
} else if let Some(value) = value.strip_prefix('-') {
|
||||||
|
let value = U256::from_str_radix(value, 10)
|
||||||
|
.map_err(|error| anyhow::anyhow!("Invalid decimal literal after `-`: {}", error))?;
|
||||||
|
if value > U256::ONE << 255u8 {
|
||||||
|
anyhow::bail!("Decimal literal after `-` is too big");
|
||||||
|
}
|
||||||
|
let value = value
|
||||||
|
.checked_sub(U256::ONE)
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("`-0` is invalid literal"))?;
|
||||||
|
Ok(U256::MAX.checked_sub(value).expect("Always valid"))
|
||||||
|
} else if let Some(value) = value.strip_prefix("0x") {
|
||||||
|
Ok(U256::from_str_radix(value, 16)
|
||||||
|
.map_err(|error| anyhow::anyhow!("Invalid hexadecimal literal: {}", error))?)
|
||||||
|
} else {
|
||||||
|
// TODO: This is a set of "variables" that we need to be able to resolve to be fully in
|
||||||
|
// compliance with the matter labs tester but we currently do not resolve them. We need to
|
||||||
|
// add logic that does their resolution in the future, perhaps through some kind of system
|
||||||
|
// context API that we pass down to the resolution function that allows it to make calls to
|
||||||
|
// the node to perform these resolutions.
|
||||||
|
let is_unsupported = [
|
||||||
|
"$CHAIN_ID",
|
||||||
|
"$GAS_LIMIT",
|
||||||
|
"$COINBASE",
|
||||||
|
"$DIFFICULTY",
|
||||||
|
"$BLOCK_HASH",
|
||||||
|
"$BLOCK_TIMESTAMP",
|
||||||
|
]
|
||||||
|
.iter()
|
||||||
|
.any(|var| value.starts_with(var));
|
||||||
|
|
||||||
|
if is_unsupported {
|
||||||
|
tracing::error!(value, "Unsupported variable used");
|
||||||
|
anyhow::bail!("Encountered {value} which is currently unsupported by the framework");
|
||||||
|
} else {
|
||||||
|
Ok(U256::from_str_radix(value, 10)
|
||||||
|
.map_err(|error| anyhow::anyhow!("Invalid decimal literal: {}", error))?)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
use alloy::json_abi::JsonAbi;
|
use alloy::json_abi::JsonAbi;
|
||||||
use alloy_primitives::{address, keccak256};
|
use alloy_primitives::address;
|
||||||
|
use alloy_sol_types::SolValue;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
@@ -267,16 +263,18 @@ mod tests {
|
|||||||
"#;
|
"#;
|
||||||
|
|
||||||
let parsed_abi: JsonAbi = serde_json::from_str(raw_metadata).unwrap();
|
let parsed_abi: JsonAbi = serde_json::from_str(raw_metadata).unwrap();
|
||||||
let selector = keccak256("store(uint256)".as_bytes())[0..4]
|
let selector = parsed_abi
|
||||||
.try_into()
|
.function("store")
|
||||||
.unwrap();
|
.unwrap()
|
||||||
|
.first()
|
||||||
|
.unwrap()
|
||||||
|
.selector()
|
||||||
|
.0;
|
||||||
|
|
||||||
let input = Input {
|
let input = Input {
|
||||||
instance: "Contract".to_string(),
|
instance: "Contract".to_string(),
|
||||||
method: Method::Function(selector),
|
method: Method::FunctionName("store".to_owned()),
|
||||||
calldata: Some(Calldata::Compound(vec![CalldataArg::Literal(
|
calldata: Some(Calldata::Compound(vec!["42".into()])),
|
||||||
"42".to_string(),
|
|
||||||
)])),
|
|
||||||
..Default::default()
|
..Default::default()
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -294,112 +292,6 @@ mod tests {
|
|||||||
assert_eq!(decoded.0, 42);
|
assert_eq!(decoded.0, 42);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_encoded_input_bool() {
|
|
||||||
let raw_abi = r#"[
|
|
||||||
{
|
|
||||||
"inputs": [{"name": "flag", "type": "bool"}],
|
|
||||||
"name": "toggle",
|
|
||||||
"outputs": [],
|
|
||||||
"stateMutability": "nonpayable",
|
|
||||||
"type": "function"
|
|
||||||
}
|
|
||||||
]"#;
|
|
||||||
|
|
||||||
let parsed_abi: JsonAbi = serde_json::from_str(raw_abi).unwrap();
|
|
||||||
let selector = keccak256("toggle(bool)".as_bytes())[0..4]
|
|
||||||
.try_into()
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let input = Input {
|
|
||||||
instance: "Contract".to_string(),
|
|
||||||
method: Method::Function(selector),
|
|
||||||
calldata: Some(Calldata::Compound(vec![CalldataArg::Literal(
|
|
||||||
"true".to_string(),
|
|
||||||
)])),
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut abis = HashMap::new();
|
|
||||||
abis.insert("Contract".to_string(), parsed_abi);
|
|
||||||
let contracts = HashMap::new();
|
|
||||||
|
|
||||||
let encoded = input.encoded_input(&abis, &contracts).unwrap();
|
|
||||||
assert!(encoded.0.starts_with(&selector));
|
|
||||||
|
|
||||||
type T = (bool,);
|
|
||||||
let decoded: T = T::abi_decode(&encoded.0[4..]).unwrap();
|
|
||||||
assert_eq!(decoded.0, true);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_encoded_input_string() {
|
|
||||||
let raw_abi = r#"[
|
|
||||||
{
|
|
||||||
"inputs": [{"name": "msg", "type": "string"}],
|
|
||||||
"name": "echo",
|
|
||||||
"outputs": [],
|
|
||||||
"stateMutability": "nonpayable",
|
|
||||||
"type": "function"
|
|
||||||
}
|
|
||||||
]"#;
|
|
||||||
|
|
||||||
let parsed_abi: JsonAbi = serde_json::from_str(raw_abi).unwrap();
|
|
||||||
let selector = keccak256("echo(string)".as_bytes())[0..4]
|
|
||||||
.try_into()
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let input = Input {
|
|
||||||
instance: "Contract".to_string(),
|
|
||||||
method: Method::Function(selector),
|
|
||||||
calldata: Some(Calldata::Compound(vec![CalldataArg::Literal(
|
|
||||||
"hello".to_string(),
|
|
||||||
)])),
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut abis = HashMap::new();
|
|
||||||
abis.insert("Contract".to_string(), parsed_abi);
|
|
||||||
let contracts = HashMap::new();
|
|
||||||
|
|
||||||
let encoded = input.encoded_input(&abis, &contracts).unwrap();
|
|
||||||
assert!(encoded.0.starts_with(&selector));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_encoded_input_uint256_array() {
|
|
||||||
let raw_abi = r#"[
|
|
||||||
{
|
|
||||||
"inputs": [{"name": "arr", "type": "uint256[]"}],
|
|
||||||
"name": "sum",
|
|
||||||
"outputs": [],
|
|
||||||
"stateMutability": "nonpayable",
|
|
||||||
"type": "function"
|
|
||||||
}
|
|
||||||
]"#;
|
|
||||||
|
|
||||||
let parsed_abi: JsonAbi = serde_json::from_str(raw_abi).unwrap();
|
|
||||||
let selector = keccak256("sum(uint256[])".as_bytes())[0..4]
|
|
||||||
.try_into()
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
let input = Input {
|
|
||||||
instance: "Contract".to_string(),
|
|
||||||
method: Method::Function(selector),
|
|
||||||
calldata: Some(Calldata::Compound(vec![CalldataArg::Literal(
|
|
||||||
"[1,2,3]".to_string(),
|
|
||||||
)])),
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut abis = HashMap::new();
|
|
||||||
abis.insert("Contract".to_string(), parsed_abi);
|
|
||||||
let contracts = HashMap::new();
|
|
||||||
|
|
||||||
let encoded = input.encoded_input(&abis, &contracts).unwrap();
|
|
||||||
assert!(encoded.0.starts_with(&selector));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_encoded_input_address() {
|
fn test_encoded_input_address() {
|
||||||
let raw_abi = r#"[
|
let raw_abi = r#"[
|
||||||
@@ -413,16 +305,20 @@ mod tests {
|
|||||||
]"#;
|
]"#;
|
||||||
|
|
||||||
let parsed_abi: JsonAbi = serde_json::from_str(raw_abi).unwrap();
|
let parsed_abi: JsonAbi = serde_json::from_str(raw_abi).unwrap();
|
||||||
let selector = keccak256("send(address)".as_bytes())[0..4]
|
let selector = parsed_abi
|
||||||
.try_into()
|
.function("send")
|
||||||
.unwrap();
|
.unwrap()
|
||||||
|
.first()
|
||||||
|
.unwrap()
|
||||||
|
.selector()
|
||||||
|
.0;
|
||||||
|
|
||||||
let input = Input {
|
let input: Input = Input {
|
||||||
instance: "Contract".to_string(),
|
instance: "Contract".to_string(),
|
||||||
method: Method::Function(selector),
|
method: Method::FunctionName("send".to_owned()),
|
||||||
calldata: Some(Calldata::Compound(vec![CalldataArg::Literal(
|
calldata: Some(Calldata::Compound(vec![
|
||||||
"0x1000000000000000000000000000000000000001".to_string(),
|
"0x1000000000000000000000000000000000000001".to_string(),
|
||||||
)])),
|
])),
|
||||||
..Default::default()
|
..Default::default()
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -11,7 +11,6 @@ rust-version.workspace = true
|
|||||||
[dependencies]
|
[dependencies]
|
||||||
alloy = { workspace = true }
|
alloy = { workspace = true }
|
||||||
anyhow = { workspace = true }
|
anyhow = { workspace = true }
|
||||||
futures = { workspace = true }
|
|
||||||
tracing = { workspace = true }
|
tracing = { workspace = true }
|
||||||
once_cell = { workspace = true }
|
once_cell = { workspace = true }
|
||||||
tokio = { workspace = true }
|
tokio = { workspace = true }
|
||||||
|
|||||||
@@ -1,221 +0,0 @@
|
|||||||
//! The alloy crate __requires__ a tokio runtime.
|
|
||||||
//! We contain any async rust right here.
|
|
||||||
|
|
||||||
use std::{any::Any, panic::AssertUnwindSafe, pin::Pin, thread};
|
|
||||||
|
|
||||||
use futures::FutureExt;
|
|
||||||
use once_cell::sync::Lazy;
|
|
||||||
use tokio::{
|
|
||||||
runtime::Builder,
|
|
||||||
sync::{mpsc::UnboundedSender, oneshot},
|
|
||||||
};
|
|
||||||
|
|
||||||
/// A blocking async executor.
|
|
||||||
///
|
|
||||||
/// This struct exposes the abstraction of a blocking async executor. It is a global and static
|
|
||||||
/// executor which means that it doesn't require for new instances of it to be created, it's a
|
|
||||||
/// singleton and can be accessed by any thread that wants to perform some async computation on the
|
|
||||||
/// blocking executor thread.
|
|
||||||
///
|
|
||||||
/// The API of the blocking executor is created in a way so that it's very natural, simple to use,
|
|
||||||
/// and unbounded to specific tasks or return types. The following is an example of using this
|
|
||||||
/// executor to drive an async computation:
|
|
||||||
///
|
|
||||||
/// ```rust
|
|
||||||
/// use revive_dt_node_interaction::*;
|
|
||||||
///
|
|
||||||
/// fn blocking_function() {
|
|
||||||
/// let result = BlockingExecutor::execute(async move {
|
|
||||||
/// tokio::time::sleep(std::time::Duration::from_secs(1)).await;
|
|
||||||
/// 0xFFu8
|
|
||||||
/// })
|
|
||||||
/// .expect("Computation failed");
|
|
||||||
///
|
|
||||||
/// assert_eq!(result, 0xFF);
|
|
||||||
/// }
|
|
||||||
/// ```
|
|
||||||
///
|
|
||||||
/// Users get to pass in their async tasks without needing to worry about putting them in a [`Box`],
|
|
||||||
/// [`Pin`], needing to perform down-casting, or the internal channel mechanism used by the runtime.
|
|
||||||
/// To the user, it just looks like a function that converts some async code into sync code.
|
|
||||||
///
|
|
||||||
/// This struct also handled panics that occur in the passed futures and converts them into errors
|
|
||||||
/// that can be handled by the user. This is done to allow the executor to be robust.
|
|
||||||
///
|
|
||||||
/// Internally, the executor communicates with the tokio runtime thread through channels which carry
|
|
||||||
/// the [`TaskMessage`] and the results of the execution.
|
|
||||||
pub struct BlockingExecutor;
|
|
||||||
|
|
||||||
impl BlockingExecutor {
|
|
||||||
pub fn execute<R>(future: impl Future<Output = R> + Send + 'static) -> Result<R, anyhow::Error>
|
|
||||||
where
|
|
||||||
R: Send + 'static,
|
|
||||||
{
|
|
||||||
// Note: The blocking executor is a singleton and therefore we store its state in a static
|
|
||||||
// so that it's assigned only once. Additionally, when we set the state of the executor we
|
|
||||||
// spawn the thread where the async runtime runs.
|
|
||||||
static STATE: Lazy<ExecutorState> = Lazy::new(|| {
|
|
||||||
tracing::trace!("Initializing the BlockingExecutor state");
|
|
||||||
|
|
||||||
// All communication with the tokio runtime thread happens over mspc channels where the
|
|
||||||
// producers here are the threads that want to run async tasks and the consumer here is
|
|
||||||
// the tokio runtime thread.
|
|
||||||
let (tx, mut rx) = tokio::sync::mpsc::unbounded_channel::<TaskMessage>();
|
|
||||||
|
|
||||||
thread::spawn(move || {
|
|
||||||
let runtime = Builder::new_current_thread()
|
|
||||||
.enable_all()
|
|
||||||
.build()
|
|
||||||
.expect("Failed to create the async runtime");
|
|
||||||
|
|
||||||
runtime.block_on(async move {
|
|
||||||
while let Some(TaskMessage {
|
|
||||||
future: task,
|
|
||||||
response_tx: response_channel,
|
|
||||||
}) = rx.recv().await
|
|
||||||
{
|
|
||||||
tracing::trace!("Received a new future to execute");
|
|
||||||
tokio::spawn(async move {
|
|
||||||
// One of the things that the blocking executor does is that it allows
|
|
||||||
// us to catch panics if they occur. By wrapping the given future in an
|
|
||||||
// AssertUnwindSafe::catch_unwind we are able to catch all panic unwinds
|
|
||||||
// in the given future and convert them into errors.
|
|
||||||
let task = AssertUnwindSafe(task).catch_unwind();
|
|
||||||
|
|
||||||
let result = task.await;
|
|
||||||
let _ = response_channel.send(result);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
})
|
|
||||||
});
|
|
||||||
|
|
||||||
ExecutorState { tx }
|
|
||||||
});
|
|
||||||
|
|
||||||
// We need to perform blocking synchronous communication between the current thread and the
|
|
||||||
// tokio runtime thread with the result of the async computation and the oneshot channels
|
|
||||||
// from tokio allows us to do that. The sender side of the channel will be given to the
|
|
||||||
// tokio runtime thread to send the result when the computation is completed and the receive
|
|
||||||
// side of the channel will be kept with this thread to await for the response of the async
|
|
||||||
// task to come back.
|
|
||||||
let (response_tx, response_rx) =
|
|
||||||
oneshot::channel::<Result<Box<dyn Any + Send>, Box<dyn Any + Send>>>();
|
|
||||||
|
|
||||||
// The tokio runtime thread expects a Future<Output = Box<dyn Any + Send>> + Send to be
|
|
||||||
// sent to it to execute. However, this function has a typed Future<Output = R> + Send and
|
|
||||||
// therefore we need to change the type of the future to fit what the runtime thread expects
|
|
||||||
// in the task message. In doing this conversion, we lose some of the type information since
|
|
||||||
// we're converting R => dyn Any. However, we will perform down-casting on the result to
|
|
||||||
// convert it back into R.
|
|
||||||
let future = Box::pin(async move { Box::new(future.await) as Box<dyn Any + Send> });
|
|
||||||
|
|
||||||
let task = TaskMessage::new(future, response_tx);
|
|
||||||
if let Err(error) = STATE.tx.send(task) {
|
|
||||||
tracing::error!(?error, "Failed to send the task to the blocking executor");
|
|
||||||
anyhow::bail!("Failed to send the task to the blocking executor: {error:?}")
|
|
||||||
}
|
|
||||||
|
|
||||||
let result = match response_rx.blocking_recv() {
|
|
||||||
Ok(result) => result,
|
|
||||||
Err(error) => {
|
|
||||||
tracing::error!(
|
|
||||||
?error,
|
|
||||||
"Failed to get the response from the blocking executor"
|
|
||||||
);
|
|
||||||
anyhow::bail!("Failed to get the response from the blocking executor: {error:?}")
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
match result.map(|result| {
|
|
||||||
*result
|
|
||||||
.downcast::<R>()
|
|
||||||
.expect("Type mismatch in the downcast")
|
|
||||||
}) {
|
|
||||||
Ok(result) => Ok(result),
|
|
||||||
Err(error) => {
|
|
||||||
tracing::error!(
|
|
||||||
?error,
|
|
||||||
"Failed to downcast the returned result into the expected type"
|
|
||||||
);
|
|
||||||
anyhow::bail!(
|
|
||||||
"Failed to downcast the returned result into the expected type: {error:?}"
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Represents the state of the async runtime. This runtime is designed to be a singleton runtime
|
|
||||||
/// which means that in the current running program there's just a single thread that has an async
|
|
||||||
/// runtime.
|
|
||||||
struct ExecutorState {
|
|
||||||
/// The sending side of the task messages channel. This is used by all of the other threads to
|
|
||||||
/// communicate with the async runtime thread.
|
|
||||||
tx: UnboundedSender<TaskMessage>,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Represents a message that contains an asynchronous task that's to be executed by the runtime
|
|
||||||
/// as well as a way for the runtime to report back on the result of the execution.
|
|
||||||
struct TaskMessage {
|
|
||||||
/// The task that's being requested to run. This is a future that returns an object that does
|
|
||||||
/// implement [`Any`] and [`Send`] to allow it to be sent between the requesting thread and the
|
|
||||||
/// async thread.
|
|
||||||
future: Pin<Box<dyn Future<Output = Box<dyn Any + Send>> + Send>>,
|
|
||||||
|
|
||||||
/// A one shot sender channel where the sender of the task is expecting to hear back on the
|
|
||||||
/// result of the task.
|
|
||||||
response_tx: oneshot::Sender<Result<Box<dyn Any + Send>, Box<dyn Any + Send>>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl TaskMessage {
|
|
||||||
pub fn new(
|
|
||||||
future: Pin<Box<dyn Future<Output = Box<dyn Any + Send>> + Send>>,
|
|
||||||
response_tx: oneshot::Sender<Result<Box<dyn Any + Send>, Box<dyn Any + Send>>>,
|
|
||||||
) -> Self {
|
|
||||||
Self {
|
|
||||||
future,
|
|
||||||
response_tx,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod test {
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn simple_future_works() {
|
|
||||||
// Act
|
|
||||||
let result = BlockingExecutor::execute(async move {
|
|
||||||
tokio::time::sleep(std::time::Duration::from_secs(1)).await;
|
|
||||||
0xFFu8
|
|
||||||
})
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
// Assert
|
|
||||||
assert_eq!(result, 0xFFu8);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
#[allow(unreachable_code, clippy::unreachable)]
|
|
||||||
fn panics_in_futures_are_caught() {
|
|
||||||
// Act
|
|
||||||
let result = BlockingExecutor::execute(async move {
|
|
||||||
panic!("This is a panic!");
|
|
||||||
0xFFu8
|
|
||||||
});
|
|
||||||
|
|
||||||
// Assert
|
|
||||||
assert!(result.is_err());
|
|
||||||
|
|
||||||
// Act
|
|
||||||
let result = BlockingExecutor::execute(async move {
|
|
||||||
tokio::time::sleep(std::time::Duration::from_secs(1)).await;
|
|
||||||
0xFFu8
|
|
||||||
})
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
// Assert
|
|
||||||
assert_eq!(result, 0xFFu8)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -3,9 +3,12 @@
|
|||||||
use alloy::primitives::Address;
|
use alloy::primitives::Address;
|
||||||
use alloy::rpc::types::trace::geth::{DiffMode, GethTrace};
|
use alloy::rpc::types::trace::geth::{DiffMode, GethTrace};
|
||||||
use alloy::rpc::types::{TransactionReceipt, TransactionRequest};
|
use alloy::rpc::types::{TransactionReceipt, TransactionRequest};
|
||||||
|
use tokio_runtime::TO_TOKIO;
|
||||||
|
|
||||||
mod blocking_executor;
|
pub mod nonce;
|
||||||
pub use blocking_executor::*;
|
mod tokio_runtime;
|
||||||
|
pub mod trace;
|
||||||
|
pub mod transaction;
|
||||||
|
|
||||||
/// An interface for all interactions with Ethereum compatible nodes.
|
/// An interface for all interactions with Ethereum compatible nodes.
|
||||||
pub trait EthereumNode {
|
pub trait EthereumNode {
|
||||||
|
|||||||
@@ -0,0 +1,55 @@
|
|||||||
|
use std::pin::Pin;
|
||||||
|
|
||||||
|
use alloy::{
|
||||||
|
primitives::Address,
|
||||||
|
providers::{Provider, ProviderBuilder},
|
||||||
|
};
|
||||||
|
use tokio::sync::oneshot;
|
||||||
|
|
||||||
|
use crate::{TO_TOKIO, tokio_runtime::AsyncNodeInteraction};
|
||||||
|
|
||||||
|
pub type Task = Pin<Box<dyn Future<Output = anyhow::Result<u64>> + Send>>;
|
||||||
|
|
||||||
|
pub(crate) struct Nonce {
|
||||||
|
sender: oneshot::Sender<anyhow::Result<u64>>,
|
||||||
|
task: Task,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AsyncNodeInteraction for Nonce {
|
||||||
|
type Output = anyhow::Result<u64>;
|
||||||
|
|
||||||
|
fn split(
|
||||||
|
self,
|
||||||
|
) -> (
|
||||||
|
std::pin::Pin<Box<dyn Future<Output = Self::Output> + Send>>,
|
||||||
|
oneshot::Sender<Self::Output>,
|
||||||
|
) {
|
||||||
|
(self.task, self.sender)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// This is like `trace_transaction`, just for nonces.
|
||||||
|
pub fn fetch_onchain_nonce(
|
||||||
|
connection: String,
|
||||||
|
wallet: alloy::network::EthereumWallet,
|
||||||
|
address: Address,
|
||||||
|
) -> anyhow::Result<u64> {
|
||||||
|
let sender = TO_TOKIO.lock().unwrap().nonce_sender.clone();
|
||||||
|
|
||||||
|
let (tx, rx) = oneshot::channel();
|
||||||
|
let task: Task = Box::pin(async move {
|
||||||
|
let provider = ProviderBuilder::new()
|
||||||
|
.wallet(wallet)
|
||||||
|
.connect(&connection)
|
||||||
|
.await?;
|
||||||
|
let onchain = provider.get_transaction_count(address).await?;
|
||||||
|
Ok(onchain)
|
||||||
|
});
|
||||||
|
|
||||||
|
sender
|
||||||
|
.blocking_send(Nonce { task, sender: tx })
|
||||||
|
.expect("not in async context");
|
||||||
|
|
||||||
|
rx.blocking_recv()
|
||||||
|
.unwrap_or_else(|err| anyhow::bail!("nonce fetch failed: {err}"))
|
||||||
|
}
|
||||||
@@ -0,0 +1,87 @@
|
|||||||
|
//! The alloy crate __requires__ a tokio runtime.
|
||||||
|
//! We contain any async rust right here.
|
||||||
|
|
||||||
|
use once_cell::sync::Lazy;
|
||||||
|
use std::pin::Pin;
|
||||||
|
use std::sync::Mutex;
|
||||||
|
use std::thread;
|
||||||
|
use tokio::runtime::Runtime;
|
||||||
|
use tokio::spawn;
|
||||||
|
use tokio::sync::{mpsc, oneshot};
|
||||||
|
use tokio::task::JoinError;
|
||||||
|
|
||||||
|
use crate::nonce::Nonce;
|
||||||
|
use crate::trace::Trace;
|
||||||
|
use crate::transaction::Transaction;
|
||||||
|
|
||||||
|
pub(crate) static TO_TOKIO: Lazy<Mutex<TokioRuntime>> =
|
||||||
|
Lazy::new(|| Mutex::new(TokioRuntime::spawn()));
|
||||||
|
|
||||||
|
/// Common interface for executing async node interactions from a non-async context.
|
||||||
|
#[allow(clippy::type_complexity)]
|
||||||
|
pub(crate) trait AsyncNodeInteraction: Send + 'static {
|
||||||
|
type Output: Send;
|
||||||
|
|
||||||
|
//// Returns the task and the output sender.
|
||||||
|
fn split(
|
||||||
|
self,
|
||||||
|
) -> (
|
||||||
|
Pin<Box<dyn Future<Output = Self::Output> + Send>>,
|
||||||
|
oneshot::Sender<Self::Output>,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) struct TokioRuntime {
|
||||||
|
pub(crate) transaction_sender: mpsc::Sender<Transaction>,
|
||||||
|
pub(crate) trace_sender: mpsc::Sender<Trace>,
|
||||||
|
pub(crate) nonce_sender: mpsc::Sender<Nonce>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TokioRuntime {
|
||||||
|
fn spawn() -> Self {
|
||||||
|
let rt = Runtime::new().expect("should be able to create the tokio runtime");
|
||||||
|
let (transaction_sender, transaction_receiver) = mpsc::channel::<Transaction>(1024);
|
||||||
|
let (trace_sender, trace_receiver) = mpsc::channel::<Trace>(1024);
|
||||||
|
let (nonce_sender, nonce_receiver) = mpsc::channel::<Nonce>(1024);
|
||||||
|
|
||||||
|
thread::spawn(move || {
|
||||||
|
rt.block_on(async move {
|
||||||
|
let transaction_task = spawn(interaction::<Transaction>(transaction_receiver));
|
||||||
|
let trace_task = spawn(interaction::<Trace>(trace_receiver));
|
||||||
|
let nonce_task = spawn(interaction::<Nonce>(nonce_receiver));
|
||||||
|
|
||||||
|
if let Err(error) = transaction_task.await {
|
||||||
|
tracing::error!("tokio transaction task failed: {error}");
|
||||||
|
}
|
||||||
|
if let Err(error) = trace_task.await {
|
||||||
|
tracing::error!("tokio trace transaction task failed: {error}");
|
||||||
|
}
|
||||||
|
if let Err(error) = nonce_task.await {
|
||||||
|
tracing::error!("tokio nonce task failed: {error}");
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
Self {
|
||||||
|
transaction_sender,
|
||||||
|
trace_sender,
|
||||||
|
nonce_sender,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn interaction<T>(mut receiver: mpsc::Receiver<T>) -> Result<(), JoinError>
|
||||||
|
where
|
||||||
|
T: AsyncNodeInteraction,
|
||||||
|
{
|
||||||
|
while let Some(task) = receiver.recv().await {
|
||||||
|
spawn(async move {
|
||||||
|
let (task, sender) = task.split();
|
||||||
|
sender
|
||||||
|
.send(task.await)
|
||||||
|
.unwrap_or_else(|_| panic!("failed to send task output"));
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
@@ -0,0 +1,43 @@
|
|||||||
|
//! Trace transactions in a sync context.
|
||||||
|
|
||||||
|
use std::pin::Pin;
|
||||||
|
|
||||||
|
use alloy::rpc::types::trace::geth::GethTrace;
|
||||||
|
use tokio::sync::oneshot;
|
||||||
|
|
||||||
|
use crate::TO_TOKIO;
|
||||||
|
use crate::tokio_runtime::AsyncNodeInteraction;
|
||||||
|
|
||||||
|
pub type Task = Pin<Box<dyn Future<Output = anyhow::Result<GethTrace>> + Send>>;
|
||||||
|
|
||||||
|
pub(crate) struct Trace {
|
||||||
|
sender: oneshot::Sender<anyhow::Result<GethTrace>>,
|
||||||
|
task: Task,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AsyncNodeInteraction for Trace {
|
||||||
|
type Output = anyhow::Result<GethTrace>;
|
||||||
|
|
||||||
|
fn split(
|
||||||
|
self,
|
||||||
|
) -> (
|
||||||
|
std::pin::Pin<Box<dyn Future<Output = Self::Output> + Send>>,
|
||||||
|
oneshot::Sender<Self::Output>,
|
||||||
|
) {
|
||||||
|
(self.task, self.sender)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Execute some [Task] that return a [GethTrace] result.
|
||||||
|
pub fn trace_transaction(task: Task) -> anyhow::Result<GethTrace> {
|
||||||
|
let task_sender = TO_TOKIO.lock().unwrap().trace_sender.clone();
|
||||||
|
let (sender, receiver) = oneshot::channel();
|
||||||
|
|
||||||
|
task_sender
|
||||||
|
.blocking_send(Trace { task, sender })
|
||||||
|
.expect("we are not calling this from an async context");
|
||||||
|
|
||||||
|
receiver
|
||||||
|
.blocking_recv()
|
||||||
|
.unwrap_or_else(|error| anyhow::bail!("no trace received: {error}"))
|
||||||
|
}
|
||||||
@@ -0,0 +1,46 @@
|
|||||||
|
//! Execute transactions in a sync context.
|
||||||
|
|
||||||
|
use std::pin::Pin;
|
||||||
|
|
||||||
|
use alloy::rpc::types::TransactionReceipt;
|
||||||
|
use tokio::sync::oneshot;
|
||||||
|
|
||||||
|
use crate::TO_TOKIO;
|
||||||
|
use crate::tokio_runtime::AsyncNodeInteraction;
|
||||||
|
|
||||||
|
pub type Task = Pin<Box<dyn Future<Output = anyhow::Result<TransactionReceipt>> + Send>>;
|
||||||
|
|
||||||
|
pub(crate) struct Transaction {
|
||||||
|
receipt_sender: oneshot::Sender<anyhow::Result<TransactionReceipt>>,
|
||||||
|
task: Task,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AsyncNodeInteraction for Transaction {
|
||||||
|
type Output = anyhow::Result<TransactionReceipt>;
|
||||||
|
|
||||||
|
fn split(
|
||||||
|
self,
|
||||||
|
) -> (
|
||||||
|
Pin<Box<dyn Future<Output = Self::Output> + Send>>,
|
||||||
|
oneshot::Sender<Self::Output>,
|
||||||
|
) {
|
||||||
|
(self.task, self.receipt_sender)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Execute some [Task] that returns a [TransactionReceipt].
|
||||||
|
pub fn execute_transaction(task: Task) -> anyhow::Result<TransactionReceipt> {
|
||||||
|
let request_sender = TO_TOKIO.lock().unwrap().transaction_sender.clone();
|
||||||
|
let (receipt_sender, receipt_receiver) = oneshot::channel();
|
||||||
|
|
||||||
|
request_sender
|
||||||
|
.blocking_send(Transaction {
|
||||||
|
receipt_sender,
|
||||||
|
task,
|
||||||
|
})
|
||||||
|
.expect("we are not calling this from an async context");
|
||||||
|
|
||||||
|
receipt_receiver
|
||||||
|
.blocking_recv()
|
||||||
|
.unwrap_or_else(|error| anyhow::bail!("no receipt received: {error}"))
|
||||||
|
}
|
||||||
+9
-14
@@ -23,7 +23,10 @@ use alloy::{
|
|||||||
},
|
},
|
||||||
};
|
};
|
||||||
use revive_dt_config::Arguments;
|
use revive_dt_config::Arguments;
|
||||||
use revive_dt_node_interaction::{BlockingExecutor, EthereumNode};
|
use revive_dt_node_interaction::{
|
||||||
|
EthereumNode, nonce::fetch_onchain_nonce, trace::trace_transaction,
|
||||||
|
transaction::execute_transaction,
|
||||||
|
};
|
||||||
use tracing::Level;
|
use tracing::Level;
|
||||||
|
|
||||||
use crate::Node;
|
use crate::Node;
|
||||||
@@ -202,7 +205,7 @@ impl EthereumNode for Instance {
|
|||||||
let connection_string = self.connection_string();
|
let connection_string = self.connection_string();
|
||||||
let wallet = self.wallet.clone();
|
let wallet = self.wallet.clone();
|
||||||
|
|
||||||
BlockingExecutor::execute(async move {
|
execute_transaction(Box::pin(async move {
|
||||||
let outer_span = tracing::debug_span!("Submitting transaction", ?transaction,);
|
let outer_span = tracing::debug_span!("Submitting transaction", ?transaction,);
|
||||||
let _outer_guard = outer_span.enter();
|
let _outer_guard = outer_span.enter();
|
||||||
|
|
||||||
@@ -281,7 +284,7 @@ impl EthereumNode for Instance {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})?
|
}))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tracing::instrument(skip_all, fields(geth_node_id = self.id))]
|
#[tracing::instrument(skip_all, fields(geth_node_id = self.id))]
|
||||||
@@ -297,14 +300,14 @@ impl EthereumNode for Instance {
|
|||||||
});
|
});
|
||||||
let wallet = self.wallet.clone();
|
let wallet = self.wallet.clone();
|
||||||
|
|
||||||
BlockingExecutor::execute(async move {
|
trace_transaction(Box::pin(async move {
|
||||||
Ok(ProviderBuilder::new()
|
Ok(ProviderBuilder::new()
|
||||||
.wallet(wallet)
|
.wallet(wallet)
|
||||||
.connect(&connection_string)
|
.connect(&connection_string)
|
||||||
.await?
|
.await?
|
||||||
.debug_trace_transaction(transaction.transaction_hash, trace_options)
|
.debug_trace_transaction(transaction.transaction_hash, trace_options)
|
||||||
.await?)
|
.await?)
|
||||||
})?
|
}))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tracing::instrument(skip_all, fields(geth_node_id = self.id))]
|
#[tracing::instrument(skip_all, fields(geth_node_id = self.id))]
|
||||||
@@ -326,15 +329,7 @@ impl EthereumNode for Instance {
|
|||||||
let connection_string = self.connection_string.clone();
|
let connection_string = self.connection_string.clone();
|
||||||
let wallet = self.wallet.clone();
|
let wallet = self.wallet.clone();
|
||||||
|
|
||||||
let onchain_nonce = BlockingExecutor::execute::<anyhow::Result<_>>(async move {
|
let onchain_nonce = fetch_onchain_nonce(connection_string, wallet, address)?;
|
||||||
ProviderBuilder::new()
|
|
||||||
.wallet(wallet)
|
|
||||||
.connect(&connection_string)
|
|
||||||
.await?
|
|
||||||
.get_transaction_count(address)
|
|
||||||
.await
|
|
||||||
.map_err(Into::into)
|
|
||||||
})??;
|
|
||||||
|
|
||||||
let mut nonces = self.nonces.lock().unwrap();
|
let mut nonces = self.nonces.lock().unwrap();
|
||||||
let current = nonces.entry(address).or_insert(onchain_nonce);
|
let current = nonces.entry(address).or_insert(onchain_nonce);
|
||||||
|
|||||||
@@ -27,7 +27,10 @@ use sp_runtime::AccountId32;
|
|||||||
use tracing::Level;
|
use tracing::Level;
|
||||||
|
|
||||||
use revive_dt_config::Arguments;
|
use revive_dt_config::Arguments;
|
||||||
use revive_dt_node_interaction::{BlockingExecutor, EthereumNode};
|
use revive_dt_node_interaction::{
|
||||||
|
EthereumNode, nonce::fetch_onchain_nonce, trace::trace_transaction,
|
||||||
|
transaction::execute_transaction,
|
||||||
|
};
|
||||||
|
|
||||||
use crate::Node;
|
use crate::Node;
|
||||||
|
|
||||||
@@ -337,7 +340,7 @@ impl EthereumNode for KitchensinkNode {
|
|||||||
tracing::debug!("Submitting transaction: {transaction:#?}");
|
tracing::debug!("Submitting transaction: {transaction:#?}");
|
||||||
|
|
||||||
tracing::info!("Submitting tx to kitchensink");
|
tracing::info!("Submitting tx to kitchensink");
|
||||||
let receipt = BlockingExecutor::execute(async move {
|
let receipt = execute_transaction(Box::pin(async move {
|
||||||
Ok(ProviderBuilder::new()
|
Ok(ProviderBuilder::new()
|
||||||
.wallet(wallet)
|
.wallet(wallet)
|
||||||
.connect(&url)
|
.connect(&url)
|
||||||
@@ -346,7 +349,7 @@ impl EthereumNode for KitchensinkNode {
|
|||||||
.await?
|
.await?
|
||||||
.get_receipt()
|
.get_receipt()
|
||||||
.await?)
|
.await?)
|
||||||
})?;
|
}));
|
||||||
tracing::info!(?receipt, "Submitted tx to kitchensink");
|
tracing::info!(?receipt, "Submitted tx to kitchensink");
|
||||||
receipt
|
receipt
|
||||||
}
|
}
|
||||||
@@ -365,14 +368,14 @@ impl EthereumNode for KitchensinkNode {
|
|||||||
|
|
||||||
let wallet = self.wallet.clone();
|
let wallet = self.wallet.clone();
|
||||||
|
|
||||||
BlockingExecutor::execute(async move {
|
trace_transaction(Box::pin(async move {
|
||||||
Ok(ProviderBuilder::new()
|
Ok(ProviderBuilder::new()
|
||||||
.wallet(wallet)
|
.wallet(wallet)
|
||||||
.connect(&url)
|
.connect(&url)
|
||||||
.await?
|
.await?
|
||||||
.debug_trace_transaction(transaction.transaction_hash, trace_options)
|
.debug_trace_transaction(transaction.transaction_hash, trace_options)
|
||||||
.await?)
|
.await?)
|
||||||
})?
|
}))
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tracing::instrument(skip_all, fields(kitchensink_node_id = self.id))]
|
#[tracing::instrument(skip_all, fields(kitchensink_node_id = self.id))]
|
||||||
@@ -391,15 +394,7 @@ impl EthereumNode for KitchensinkNode {
|
|||||||
let url = self.rpc_url.clone();
|
let url = self.rpc_url.clone();
|
||||||
let wallet = self.wallet.clone();
|
let wallet = self.wallet.clone();
|
||||||
|
|
||||||
let onchain_nonce = BlockingExecutor::execute::<anyhow::Result<_>>(async move {
|
let onchain_nonce = fetch_onchain_nonce(url, wallet, address)?;
|
||||||
ProviderBuilder::new()
|
|
||||||
.wallet(wallet)
|
|
||||||
.connect(&url)
|
|
||||||
.await?
|
|
||||||
.get_transaction_count(address)
|
|
||||||
.await
|
|
||||||
.map_err(Into::into)
|
|
||||||
})??;
|
|
||||||
|
|
||||||
let mut nonces = self.nonces.lock().unwrap();
|
let mut nonces = self.nonces.lock().unwrap();
|
||||||
let current = nonces.entry(address).or_insert(onchain_nonce);
|
let current = nonces.entry(address).or_insert(onchain_nonce);
|
||||||
|
|||||||
Reference in New Issue
Block a user