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, 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>);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(<SelfSponsoring<T>>::get(contract_address))32 }3334 fn toggle_sponsoring(35 &mut self,36 caller: caller,37 contract_address: address,38 enabled: bool,39 ) -> Result<void> {40 <Pallet<T>>::ensure_owner(contract_address, caller)?;41 <Pallet<T>>::toggle_sponsoring(contract_address, enabled);42 Ok(())43 }4445 fn set_sponsoring_rate_limit(46 &mut self,47 caller: caller,48 contract_address: address,49 rate_limit: uint32,50 ) -> Result<void> {51 <Pallet<T>>::ensure_owner(contract_address, caller)?;52 <Pallet<T>>::set_sponsoring_rate_limit(contract_address, rate_limit.into());53 Ok(())54 }5556 fn get_sponsoring_rate_limit(&self, contract_address: address) -> Result<uint32> {57 Ok(<SponsoringRateLimit<T>>::get(contract_address)58 .try_into()59 .map_err(|_| "rate limit > u32::MAX")?)60 }6162 fn allowed(&self, contract_address: address, user: address) -> Result<bool> {63 self.0.consume_sload()?;64 Ok(<Pallet<T>>::allowed(contract_address, user)65 || !<AllowlistEnabled<T>>::get(contract_address))66 }6768 fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {69 Ok(<AllowlistEnabled<T>>::get(contract_address))70 }7172 fn toggle_allowlist(73 &mut self,74 caller: caller,75 contract_address: address,76 enabled: bool,77 ) -> Result<void> {78 <Pallet<T>>::ensure_owner(contract_address, caller)?;79 <Pallet<T>>::toggle_allowlist(contract_address, enabled);80 Ok(())81 }8283 fn toggle_allowed(84 &mut self,85 caller: caller,86 contract_address: address,87 user: address,88 allowed: bool,89 ) -> Result<void> {90 <Pallet<T>>::ensure_owner(contract_address, caller)?;91 <Pallet<T>>::toggle_allowed(contract_address, user, allowed);92 Ok(())93 }94}9596pub struct HelpersOnMethodCall<T: Config>(PhantomData<*const T>);97impl<T: Config> OnMethodCall<T> for HelpersOnMethodCall<T> {98 fn is_reserved(contract: &sp_core::H160) -> bool {99 contract == &T::ContractAddress::get()100 }101102 fn is_used(contract: &sp_core::H160) -> bool {103 contract == &T::ContractAddress::get()104 }105106 fn call(107 source: &sp_core::H160,108 target: &sp_core::H160,109 gas_left: u64,110 input: &[u8],111 value: sp_core::U256,112 ) -> Option<PrecompileResult> {113 114 if <AllowlistEnabled<T>>::get(target) && !<Pallet<T>>::allowed(*target, *source) {115 return Some(Err(PrecompileFailure::Revert {116 exit_status: ExitRevert::Reverted,117 cost: 0,118 output: {119 let mut writer = AbiWriter::new_call(evm_coder::fn_selector!(Error(string)));120 writer.string("Target contract is allowlisted");121 writer.finish()122 },123 }));124 }125126 if target != &T::ContractAddress::get() {127 return None;128 }129130 let helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));131 pallet_evm_coder_substrate::call(*source, helpers, value, input)132 }133134 fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {135 (contract == &T::ContractAddress::get())136 .then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())137 }138}139140pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);141impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {142 fn on_create(owner: H160, contract: H160) {143 <Owner<T>>::insert(contract, owner);144 }145}146147pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);148impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {149 fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {150 if !<SelfSponsoring<T>>::get(&call.0) {151 return None;152 }153 if !<Pallet<T>>::allowed(call.0, *who) {154 return None;155 }156 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;157158 if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {159 let limit = <SponsoringRateLimit<T>>::get(&call.0);160161 let timeout = last_tx_block + limit;162 if block_number < timeout {163 return None;164 }165 }166167 <SponsorBasket<T>>::insert(&call.0, who, block_number);168169 Some(call.0)170 }171}172173generate_stubgen!(contract_helpers_impl, ContractHelpersCall<()>, true);174generate_stubgen!(contract_helpers_iface, ContractHelpersCall<()>, false);