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::{ExitRevert, OnCreate, OnMethodCall, PrecompileResult, PrecompileFailure};5use sp_core::H160;6use crate::{7 AllowlistEnabled, Config, Owner, Pallet, SponsorBasket, SponsoringRateLimit, SponsoringModeT,8};9use frame_support::traits::Get;10use up_sponsorship::SponsorshipHandler;11use sp_std::{convert::TryInto, vec::Vec};1213struct ContractHelpers<T: Config>(SubstrateRecorder<T>);14impl<T: Config> WithRecorder<T> for ContractHelpers<T> {15 fn recorder(&self) -> &SubstrateRecorder<T> {16 &self.017 }1819 fn into_recorder(self) -> SubstrateRecorder<T> {20 self.021 }22}2324#[solidity_interface(name = "ContractHelpers")]25impl<T: Config> ContractHelpers<T> {26 fn contract_owner(&self, contract_address: address) -> Result<address> {27 Ok(<Owner<T>>::get(contract_address))28 }2930 fn sponsoring_enabled(&self, contract_address: address) -> Result<bool> {31 Ok(<Pallet<T>>::sponsoring_mode(contract_address) != SponsoringModeT::Disabled)32 }3334 35 fn toggle_sponsoring(36 &mut self,37 caller: caller,38 contract_address: address,39 enabled: bool,40 ) -> Result<void> {41 <Pallet<T>>::ensure_owner(contract_address, caller)?;42 <Pallet<T>>::toggle_sponsoring(contract_address, enabled);43 Ok(())44 }4546 fn set_sponsoring_mode(47 &mut self,48 caller: caller,49 contract_address: address,50 mode: uint8,51 ) -> Result<void> {52 <Pallet<T>>::ensure_owner(contract_address, caller)?;53 let mode = SponsoringModeT::from_eth(mode).ok_or("unknown mode")?;54 <Pallet<T>>::set_sponsoring_mode(contract_address, mode);55 Ok(())56 }5758 fn sponsoring_mode(&self, contract_address: address) -> Result<uint8> {59 Ok(<Pallet<T>>::sponsoring_mode(contract_address).to_eth())60 }6162 fn set_sponsoring_rate_limit(63 &mut self,64 caller: caller,65 contract_address: address,66 rate_limit: uint32,67 ) -> Result<void> {68 <Pallet<T>>::ensure_owner(contract_address, caller)?;69 <Pallet<T>>::set_sponsoring_rate_limit(contract_address, rate_limit.into());70 Ok(())71 }7273 fn get_sponsoring_rate_limit(&self, contract_address: address) -> Result<uint32> {74 Ok(<SponsoringRateLimit<T>>::get(contract_address)75 .try_into()76 .map_err(|_| "rate limit > u32::MAX")?)77 }7879 fn allowed(&self, contract_address: address, user: address) -> Result<bool> {80 self.0.consume_sload()?;81 Ok(<Pallet<T>>::allowed(contract_address, user)82 || !<AllowlistEnabled<T>>::get(contract_address))83 }8485 fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {86 Ok(<AllowlistEnabled<T>>::get(contract_address))87 }8889 fn toggle_allowlist(90 &mut self,91 caller: caller,92 contract_address: address,93 enabled: bool,94 ) -> Result<void> {95 <Pallet<T>>::ensure_owner(contract_address, caller)?;96 <Pallet<T>>::toggle_allowlist(contract_address, enabled);97 Ok(())98 }99100 fn toggle_allowed(101 &mut self,102 caller: caller,103 contract_address: address,104 user: address,105 allowed: bool,106 ) -> Result<void> {107 <Pallet<T>>::ensure_owner(contract_address, caller)?;108 <Pallet<T>>::toggle_allowed(contract_address, user, allowed);109 Ok(())110 }111}112113pub struct HelpersOnMethodCall<T: Config>(PhantomData<*const T>);114impl<T: Config> OnMethodCall<T> for HelpersOnMethodCall<T> {115 fn is_reserved(contract: &sp_core::H160) -> bool {116 contract == &T::ContractAddress::get()117 }118119 fn is_used(contract: &sp_core::H160) -> bool {120 contract == &T::ContractAddress::get()121 }122123 fn call(124 source: &sp_core::H160,125 target: &sp_core::H160,126 gas_left: u64,127 input: &[u8],128 value: sp_core::U256,129 ) -> Option<PrecompileResult> {130 131 if <AllowlistEnabled<T>>::get(target) && !<Pallet<T>>::allowed(*target, *source) {132 return Some(Err(PrecompileFailure::Revert {133 exit_status: ExitRevert::Reverted,134 cost: 0,135 output: {136 let mut writer = AbiWriter::new_call(evm_coder::fn_selector!(Error(string)));137 writer.string("Target contract is allowlisted");138 writer.finish()139 },140 }));141 }142143 if target != &T::ContractAddress::get() {144 return None;145 }146147 let helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));148 pallet_evm_coder_substrate::call(*source, helpers, value, input)149 }150151 fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {152 (contract == &T::ContractAddress::get())153 .then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())154 }155}156157pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);158impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {159 fn on_create(owner: H160, contract: H160) {160 <Owner<T>>::insert(contract, owner);161 }162}163164pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);165impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {166 fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {167 let mode = <Pallet<T>>::sponsoring_mode(call.0);168 if mode == SponsoringModeT::Disabled {169 return None;170 }171 if mode == SponsoringModeT::Allowlisted && !<Pallet<T>>::allowed(call.0, *who) {172 return None;173 }174 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;175176 if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {177 let limit = <SponsoringRateLimit<T>>::get(&call.0);178179 let timeout = last_tx_block + limit;180 if block_number < timeout {181 return None;182 }183 }184185 <SponsorBasket<T>>::insert(&call.0, who, block_number);186187 Some(call.0)188 }189}190191generate_stubgen!(contract_helpers_impl, ContractHelpersCall<()>, true);192generate_stubgen!(contract_helpers_iface, ContractHelpersCall<()>, false);