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, SponsoringMode, SponsorBasket,8 SponsoringRateLimit, SponsoringModeT,9};10use frame_support::traits::Get;11use up_sponsorship::SponsorshipHandler;12use sp_std::{convert::TryInto, vec::Vec};1314struct ContractHelpers<T: Config>(SubstrateRecorder<T>);15impl<T: Config> WithRecorder<T> for ContractHelpers<T> {16 fn recorder(&self) -> &SubstrateRecorder<T> {17 &self.018 }1920 fn into_recorder(self) -> SubstrateRecorder<T> {21 self.022 }23}2425#[solidity_interface(name = "ContractHelpers")]26impl<T: Config> ContractHelpers<T> {27 fn contract_owner(&self, contract_address: address) -> Result<address> {28 Ok(<Owner<T>>::get(contract_address))29 }3031 fn sponsoring_enabled(&self, contract_address: address) -> Result<bool> {32 Ok(<SelfSponsoring<T>>::get(contract_address))33 }3435 36 fn toggle_sponsoring(37 &mut self,38 caller: caller,39 contract_address: address,40 enabled: bool,41 ) -> Result<void> {42 <Pallet<T>>::ensure_owner(contract_address, caller)?;43 <Pallet<T>>::toggle_sponsoring(contract_address, enabled);44 Ok(())45 }4647 fn set_sponsoring_mode(48 &mut self,49 caller: caller,50 contract_address: address,51 mode: uint8,52 ) -> Result<void> {53 <Pallet<T>>::ensure_owner(contract_address, caller)?;54 let mode = SponsoringModeT::from_eth(mode).ok_or("unknown mode")?;55 <Pallet<T>>::set_sponsoring_mode(contract_address, mode);56 Ok(())57 }5859 fn sponsoring_mode(&self, contract_address: address) -> Result<uint8> {60 Ok(<Pallet<T>>::sponsoring_mode(contract_address).to_eth())61 }6263 fn set_sponsoring_rate_limit(64 &mut self,65 caller: caller,66 contract_address: address,67 rate_limit: uint32,68 ) -> Result<void> {69 <Pallet<T>>::ensure_owner(contract_address, caller)?;70 <Pallet<T>>::set_sponsoring_rate_limit(contract_address, rate_limit.into());71 Ok(())72 }7374 fn get_sponsoring_rate_limit(&self, contract_address: address) -> Result<uint32> {75 Ok(<SponsoringRateLimit<T>>::get(contract_address)76 .try_into()77 .map_err(|_| "rate limit > u32::MAX")?)78 }7980 fn allowed(&self, contract_address: address, user: address) -> Result<bool> {81 self.0.consume_sload()?;82 Ok(<Pallet<T>>::allowed(contract_address, user)83 || !<AllowlistEnabled<T>>::get(contract_address))84 }8586 fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {87 Ok(<AllowlistEnabled<T>>::get(contract_address))88 }8990 fn toggle_allowlist(91 &mut self,92 caller: caller,93 contract_address: address,94 enabled: bool,95 ) -> Result<void> {96 <Pallet<T>>::ensure_owner(contract_address, caller)?;97 <Pallet<T>>::toggle_allowlist(contract_address, enabled);98 Ok(())99 }100101 fn toggle_allowed(102 &mut self,103 caller: caller,104 contract_address: address,105 user: address,106 allowed: bool,107 ) -> Result<void> {108 <Pallet<T>>::ensure_owner(contract_address, caller)?;109 <Pallet<T>>::toggle_allowed(contract_address, user, allowed);110 Ok(())111 }112}113114pub struct HelpersOnMethodCall<T: Config>(PhantomData<*const T>);115impl<T: Config> OnMethodCall<T> for HelpersOnMethodCall<T> {116 fn is_reserved(contract: &sp_core::H160) -> bool {117 contract == &T::ContractAddress::get()118 }119120 fn is_used(contract: &sp_core::H160) -> bool {121 contract == &T::ContractAddress::get()122 }123124 fn call(125 source: &sp_core::H160,126 target: &sp_core::H160,127 gas_left: u64,128 input: &[u8],129 value: sp_core::U256,130 ) -> Option<PrecompileResult> {131 132 if <AllowlistEnabled<T>>::get(target) && !<Pallet<T>>::allowed(*target, *source) {133 return Some(Err(PrecompileFailure::Revert {134 exit_status: ExitRevert::Reverted,135 cost: 0,136 output: {137 let mut writer = AbiWriter::new_call(evm_coder::fn_selector!(Error(string)));138 writer.string("Target contract is allowlisted");139 writer.finish()140 },141 }));142 }143144 if target != &T::ContractAddress::get() {145 return None;146 }147148 let helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));149 pallet_evm_coder_substrate::call(*source, helpers, value, input)150 }151152 fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {153 (contract == &T::ContractAddress::get())154 .then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())155 }156}157158pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);159impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {160 fn on_create(owner: H160, contract: H160) {161 <Owner<T>>::insert(contract, owner);162 }163}164165pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);166impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {167 fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {168 let mode = <Pallet<T>>::sponsoring_mode(call.0);169 if mode == SponsoringModeT::Disabled {170 return None;171 }172 if mode == SponsoringModeT::Allowlisted && !<Pallet<T>>::allowed(call.0, *who) {173 return None;174 }175 let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;176177 if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {178 let limit = <SponsoringRateLimit<T>>::get(&call.0);179180 let timeout = last_tx_block + limit;181 if block_number < timeout {182 return None;183 }184 }185186 <SponsorBasket<T>>::insert(&call.0, who, block_number);187188 Some(call.0)189 }190}191192generate_stubgen!(contract_helpers_impl, ContractHelpersCall<()>, true);193generate_stubgen!(contract_helpers_iface, ContractHelpersCall<()>, false);