1use core::marker::PhantomData;2use evm_coder::{abi::AbiWriter, execution::Result, generate_stubgen, solidity_interface, types::*};3use pallet_evm_coder_substrate::SubstrateRecorder;4use pallet_evm::{ExitReason, ExitRevert, OnCreate, OnMethodCall, PrecompileOutput};5use sp_core::H160;6use crate::{7 AllowlistEnabled, Config, Owner, Pallet, SelfSponsoring, SponsorBasket, SponsoringRateLimit,8};9use frame_support::traits::Get;10use up_sponsorship::SponsorshipHandler;11use sp_std::{convert::TryInto, vec::Vec};1213struct ContractHelpers<T: Config>(SubstrateRecorder<T>);1415#[solidity_interface(name = "ContractHelpers")]16impl<T: Config> ContractHelpers<T> {17 fn contract_owner(&self, contract_address: address) -> Result<address> {18 self.0.consume_sload()?;19 Ok(<Owner<T>>::get(contract_address))20 }2122 fn sponsoring_enabled(&self, contract_address: address) -> Result<bool> {23 self.0.consume_sload()?;24 Ok(<SelfSponsoring<T>>::get(contract_address))25 }2627 fn toggle_sponsoring(28 &mut self,29 caller: caller,30 contract_address: address,31 enabled: bool,32 ) -> Result<void> {33 self.0.consume_sload()?;34 <Pallet<T>>::ensure_owner(contract_address, caller)?;35 self.0.consume_sstore()?;36 <Pallet<T>>::toggle_sponsoring(contract_address, enabled);37 Ok(())38 }3940 fn set_sponsoring_rate_limit(41 &mut self,42 caller: caller,43 contract_address: address,44 rate_limit: uint32,45 ) -> Result<void> {46 self.0.consume_sload()?;47 <Pallet<T>>::ensure_owner(contract_address, caller)?;48 self.0.consume_sstore()?;49 <Pallet<T>>::set_sponsoring_rate_limit(contract_address, rate_limit.into());50 Ok(())51 }5253 fn get_sponsoring_rate_limit(&self, contract_address: address) -> Result<uint32> {54 self.0.consume_sload()?;55 Ok(<SponsoringRateLimit<T>>::get(contract_address)56 .try_into()57 .map_err(|_| "rate limit > u32::MAX")?)58 }5960 fn allowed(&self, contract_address: address, user: address) -> Result<bool> {61 self.0.consume_sload()?;62 Ok(<Pallet<T>>::allowed(contract_address, user)63 || !<AllowlistEnabled<T>>::get(contract_address))64 }6566 fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {67 self.0.consume_sload()?;68 Ok(<AllowlistEnabled<T>>::get(contract_address))69 }7071 fn toggle_allowlist(72 &mut self,73 caller: caller,74 contract_address: address,75 enabled: bool,76 ) -> Result<void> {77 self.0.consume_sload()?;78 <Pallet<T>>::ensure_owner(contract_address, caller)?;79 self.0.consume_sstore()?;80 <Pallet<T>>::toggle_allowlist(contract_address, enabled);81 Ok(())82 }8384 fn toggle_allowed(85 &mut self,86 caller: caller,87 contract_address: address,88 user: address,89 allowed: bool,90 ) -> Result<void> {91 self.0.consume_sload()?;92 <Pallet<T>>::ensure_owner(contract_address, caller)?;93 self.0.consume_sstore()?;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<PrecompileOutput> {116 117 if <AllowlistEnabled<T>>::get(target) && !<Pallet<T>>::allowed(*target, *source) {118 return Some(PrecompileOutput {119 exit_status: ExitReason::Revert(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 logs: sp_std::vec![],127 });128 }129130 if target != &T::ContractAddress::get() {131 return None;132 }133134 let mut helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));135 let result = pallet_evm_coder_substrate::call_internal(*source, &mut helpers, value, input);136 helpers.0.evm_to_precompile_output(result)137 }138139 fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {140 (contract == &T::ContractAddress::get())141 .then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())142 }143}144145pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);146impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {147 fn on_create(owner: H160, contract: H160) {148 <Owner<T>>::insert(contract, owner);149 }150}151152pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);153impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {154 fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {155 if !<SelfSponsoring<T>>::get(&call.0) {156 return None;157 }158 if !<Pallet<T>>::allowed(call.0, *who) {159 return None;160 }161 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;162163 if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {164 let limit = <SponsoringRateLimit<T>>::get(&call.0);165166 let timeout = last_tx_block + limit.into();167 if block_number < timeout {168 return None;169 }170 }171172 <SponsorBasket<T>>::insert(&call.0, who, block_number);173174 Some(call.0)175 }176}177178generate_stubgen!(contract_helpers_impl, ContractHelpersCall, true);179generate_stubgen!(contract_helpers_iface, ContractHelpersCall, false);