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 }6667 fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {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 <Pallet<T>>::ensure_owner(contract_address, caller)?;78 <Pallet<T>>::toggle_allowlist(contract_address, enabled);79 Ok(())80 }8182 fn toggle_allowed(83 &mut self,84 caller: caller,85 contract_address: address,86 user: address,87 allowed: bool,88 ) -> Result<void> {89 <Pallet<T>>::ensure_owner(contract_address, caller)?;90 <Pallet<T>>::toggle_allowed(contract_address, user, allowed);91 Ok(())92 }93}9495pub struct HelpersOnMethodCall<T: Config>(PhantomData<*const T>);96impl<T: Config> OnMethodCall<T> for HelpersOnMethodCall<T> {97 fn is_reserved(contract: &sp_core::H160) -> bool {98 contract == &T::ContractAddress::get()99 }100101 fn is_used(contract: &sp_core::H160) -> bool {102 contract == &T::ContractAddress::get()103 }104105 fn call(106 source: &sp_core::H160,107 target: &sp_core::H160,108 gas_left: u64,109 input: &[u8],110 value: sp_core::U256,111 ) -> Option<PrecompileResult> {112 113 if <AllowlistEnabled<T>>::get(target) && !<Pallet<T>>::allowed(*target, *source) {114 return Some(Err(PrecompileFailure::Revert {115 exit_status: ExitRevert::Reverted,116 cost: 0,117 output: {118 let mut writer = AbiWriter::new_call(evm_coder::fn_selector!(Error(string)));119 writer.string("Target contract is allowlisted");120 writer.finish()121 },122 }));123 }124125 if target != &T::ContractAddress::get() {126 return None;127 }128129 let helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));130 pallet_evm_coder_substrate::call(*source, helpers, value, input)131 }132133 fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {134 (contract == &T::ContractAddress::get())135 .then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())136 }137}138139pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);140impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {141 fn on_create(owner: H160, contract: H160) {142 <Owner<T>>::insert(contract, owner);143 }144}145146pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);147impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {148 fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {149 if !<SelfSponsoring<T>>::get(&call.0) {150 return None;151 }152 if !<Pallet<T>>::allowed(call.0, *who) {153 return None;154 }155 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;156157 if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {158 let limit = <SponsoringRateLimit<T>>::get(&call.0);159160 let timeout = last_tx_block + limit;161 if block_number < timeout {162 return None;163 }164 }165166 <SponsorBasket<T>>::insert(&call.0, who, block_number);167168 Some(call.0)169 }170}171172generate_stubgen!(contract_helpers_impl, ContractHelpersCall<()>, true);173generate_stubgen!(contract_helpers_iface, ContractHelpersCall<()>, false);