1use core::marker::PhantomData;2use evm_coder::{abi::AbiWriter, execution::Result, generate_stubgen, solidity_interface, types::*};3use pallet_evm_coder_substrate::{SubstrateRecorder, WithRecorder};4use pallet_evm::{5 ExitReason, ExitRevert, OnCreate, OnMethodCall, PrecompileOutput, PrecompileResult,6 PrecompileFailure,7};8use sp_core::H160;9use crate::{10 AllowlistEnabled, Config, Owner, Pallet, SelfSponsoring, SponsorBasket, SponsoringRateLimit,11};12use frame_support::traits::Get;13use up_sponsorship::SponsorshipHandler;14use sp_std::{convert::TryInto, vec::Vec};1516struct ContractHelpers<T: Config>(SubstrateRecorder<T>);17impl<T: Config> WithRecorder<T> for ContractHelpers<T> {18 fn recorder(&self) -> &SubstrateRecorder<T> {19 &self.020 }2122 fn into_recorder(self) -> SubstrateRecorder<T> {23 self.024 }25}2627#[solidity_interface(name = "ContractHelpers")]28impl<T: Config> ContractHelpers<T> {29 fn contract_owner(&self, contract_address: address) -> Result<address> {30 Ok(<Owner<T>>::get(contract_address))31 }3233 fn sponsoring_enabled(&self, contract_address: address) -> Result<bool> {34 Ok(<SelfSponsoring<T>>::get(contract_address))35 }3637 fn toggle_sponsoring(38 &mut self,39 caller: caller,40 contract_address: address,41 enabled: bool,42 ) -> Result<void> {43 <Pallet<T>>::ensure_owner(contract_address, caller)?;44 <Pallet<T>>::toggle_sponsoring(contract_address, enabled);45 Ok(())46 }4748 fn set_sponsoring_rate_limit(49 &mut self,50 caller: caller,51 contract_address: address,52 rate_limit: uint32,53 ) -> Result<void> {54 <Pallet<T>>::ensure_owner(contract_address, caller)?;55 <Pallet<T>>::set_sponsoring_rate_limit(contract_address, rate_limit.into());56 Ok(())57 }5859 fn get_sponsoring_rate_limit(&self, contract_address: address) -> Result<uint32> {60 Ok(<SponsoringRateLimit<T>>::get(contract_address)61 .try_into()62 .map_err(|_| "rate limit > u32::MAX")?)63 }6465 fn allowed(&self, contract_address: address, user: address) -> Result<bool> {66 self.0.consume_sload()?;67 Ok(<Pallet<T>>::allowed(contract_address, user)68 || !<AllowlistEnabled<T>>::get(contract_address))69 }7071 fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {72 Ok(<AllowlistEnabled<T>>::get(contract_address))73 }7475 fn toggle_allowlist(76 &mut self,77 caller: caller,78 contract_address: address,79 enabled: bool,80 ) -> Result<void> {81 <Pallet<T>>::ensure_owner(contract_address, caller)?;82 <Pallet<T>>::toggle_allowlist(contract_address, enabled);83 Ok(())84 }8586 fn toggle_allowed(87 &mut self,88 caller: caller,89 contract_address: address,90 user: address,91 allowed: bool,92 ) -> Result<void> {93 <Pallet<T>>::ensure_owner(contract_address, caller)?;94 <Pallet<T>>::toggle_allowed(contract_address, user, allowed);95 Ok(())96 }97}9899pub struct HelpersOnMethodCall<T: Config>(PhantomData<*const T>);100impl<T: Config> OnMethodCall<T> for HelpersOnMethodCall<T> {101 fn is_reserved(contract: &sp_core::H160) -> bool {102 contract == &T::ContractAddress::get()103 }104105 fn is_used(contract: &sp_core::H160) -> bool {106 contract == &T::ContractAddress::get()107 }108109 fn call(110 source: &sp_core::H160,111 target: &sp_core::H160,112 gas_left: u64,113 input: &[u8],114 value: sp_core::U256,115 ) -> Option<PrecompileResult> {116 117 if <AllowlistEnabled<T>>::get(target) && !<Pallet<T>>::allowed(*target, *source) {118 return Some(Err(PrecompileFailure::Revert {119 exit_status: ExitRevert::Reverted,120 cost: 0,121 output: {122 let mut writer = AbiWriter::new_call(evm_coder::fn_selector!(Error(string)));123 writer.string("Target contract is allowlisted");124 writer.finish()125 },126 }));127 }128129 if target != &T::ContractAddress::get() {130 return None;131 }132133 let helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));134 pallet_evm_coder_substrate::call(*source, helpers, value, input)135 }136137 fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {138 (contract == &T::ContractAddress::get())139 .then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())140 }141}142143pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);144impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {145 fn on_create(owner: H160, contract: H160) {146 <Owner<T>>::insert(contract, owner);147 }148}149150pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);151impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {152 fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {153 if !<SelfSponsoring<T>>::get(&call.0) {154 return None;155 }156 if !<Pallet<T>>::allowed(call.0, *who) {157 return None;158 }159 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;160161 if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {162 let limit = <SponsoringRateLimit<T>>::get(&call.0);163164 let timeout = last_tx_block + limit.into();165 if block_number < timeout {166 return None;167 }168 }169170 <SponsorBasket<T>>::insert(&call.0, who, block_number);171172 Some(call.0)173 }174}175176generate_stubgen!(contract_helpers_impl, ContractHelpersCall<()>, true);177generate_stubgen!(contract_helpers_iface, ContractHelpersCall<()>, false);