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 }8384 fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {85 Ok(<AllowlistEnabled<T>>::get(contract_address))86 }8788 fn toggle_allowlist(89 &mut self,90 caller: caller,91 contract_address: address,92 enabled: bool,93 ) -> Result<void> {94 <Pallet<T>>::ensure_owner(contract_address, caller)?;95 <Pallet<T>>::toggle_allowlist(contract_address, enabled);96 Ok(())97 }9899 fn toggle_allowed(100 &mut self,101 caller: caller,102 contract_address: address,103 user: address,104 allowed: bool,105 ) -> Result<void> {106 <Pallet<T>>::ensure_owner(contract_address, caller)?;107 <Pallet<T>>::toggle_allowed(contract_address, user, allowed);108 Ok(())109 }110}111112pub struct HelpersOnMethodCall<T: Config>(PhantomData<*const T>);113impl<T: Config> OnMethodCall<T> for HelpersOnMethodCall<T> {114 fn is_reserved(contract: &sp_core::H160) -> bool {115 contract == &T::ContractAddress::get()116 }117118 fn is_used(contract: &sp_core::H160) -> bool {119 contract == &T::ContractAddress::get()120 }121122 fn call(123 source: &sp_core::H160,124 target: &sp_core::H160,125 gas_left: u64,126 input: &[u8],127 value: sp_core::U256,128 ) -> Option<PrecompileResult> {129 130 if <AllowlistEnabled<T>>::get(target) && !<Pallet<T>>::allowed(*target, *source) {131 return Some(Err(PrecompileFailure::Revert {132 exit_status: ExitRevert::Reverted,133 cost: 0,134 output: {135 let mut writer = AbiWriter::new_call(evm_coder::fn_selector!(Error(string)));136 writer.string("Target contract is allowlisted");137 writer.finish()138 },139 }));140 }141142 if target != &T::ContractAddress::get() {143 return None;144 }145146 let helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));147 pallet_evm_coder_substrate::call(*source, helpers, value, input)148 }149150 fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {151 (contract == &T::ContractAddress::get())152 .then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())153 }154}155156pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);157impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {158 fn on_create(owner: H160, contract: H160) {159 <Owner<T>>::insert(contract, owner);160 }161}162163pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);164impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {165 fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {166 let mode = <Pallet<T>>::sponsoring_mode(call.0);167 if mode == SponsoringModeT::Disabled {168 return None;169 }170 if mode == SponsoringModeT::Allowlisted && !<Pallet<T>>::allowed(call.0, *who) {171 return None;172 }173 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;174175 if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {176 let limit = <SponsoringRateLimit<T>>::get(&call.0);177178 let timeout = last_tx_block + limit;179 if block_number < timeout {180 return None;181 }182 }183184 <SponsorBasket<T>>::insert(&call.0, who, block_number);185186 Some(call.0)187 }188}189190generate_stubgen!(contract_helpers_impl, ContractHelpersCall<()>, true);191generate_stubgen!(contract_helpers_iface, ContractHelpersCall<()>, false);