git.delta.rocks / unique-network / refs/commits / 44bd99aa7cbe

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, true))63	}6465	fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {66		self.0.consume_sload()?;67		Ok(<AllowlistEnabled<T>>::get(contract_address))68	}6970	fn toggle_allowlist(71		&mut self,72		caller: caller,73		contract_address: address,74		enabled: bool,75	) -> Result<void> {76		self.0.consume_sload()?;77		<Pallet<T>>::ensure_owner(contract_address, caller)?;78		self.0.consume_sstore()?;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		self.0.consume_sload()?;91		<Pallet<T>>::ensure_owner(contract_address, caller)?;92		self.0.consume_sstore()?;93		<Pallet<T>>::toggle_allowed(contract_address, user, allowed);94		Ok(())95	}96}9798pub struct HelpersOnMethodCall<T: Config>(PhantomData<*const T>);99impl<T: Config> OnMethodCall<T> for HelpersOnMethodCall<T> {100	fn is_reserved(contract: &sp_core::H160) -> bool {101		contract == &T::ContractAddress::get()102	}103104	fn is_used(contract: &sp_core::H160) -> bool {105		contract == &T::ContractAddress::get()106	}107108	fn call(109		source: &sp_core::H160,110		target: &sp_core::H160,111		gas_left: u64,112		input: &[u8],113		value: sp_core::U256,114	) -> Option<PrecompileOutput> {115		// TODO: Extract to another OnMethodCall handler116		if !<Pallet<T>>::allowed(*target, *source, true) {117			return Some(PrecompileOutput {118				exit_status: ExitReason::Revert(ExitRevert::Reverted),119				cost: 0,120				output: {121					let mut writer = AbiWriter::new_call(evm_coder::fn_selector!(Error(string)));122					writer.string("Target contract is allowlisted");123					writer.finish()124				},125				logs: sp_std::vec![],126			});127		}128129		if target != &T::ContractAddress::get() {130			return None;131		}132133		let mut helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));134		let result = pallet_evm_coder_substrate::call_internal(*source, &mut helpers, value, input);135		helpers.0.evm_to_precompile_output(result)136	}137138	fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {139		(contract == &T::ContractAddress::get())140			.then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())141	}142}143144pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);145impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {146	fn on_create(owner: H160, contract: H160) {147		<Owner<T>>::insert(contract, owner);148	}149}150151pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);152impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {153	fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {154		if <SelfSponsoring<T>>::get(&call.0) && <Pallet<T>>::allowed(call.0, *who, false) {155			let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;156			if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {157				let rate_limit = <SponsoringRateLimit<T>>::get(&call.0);158				let limit_time = last_tx_block + rate_limit;159160				if block_number > limit_time {161					<SponsorBasket<T>>::insert(&call.0, who, block_number);162					return Some(call.0);163				}164			} else {165				<SponsorBasket<T>>::insert(&call.0, who, block_number);166				return Some(call.0);167			}168		}169		None170	}171}172173generate_stubgen!(contract_helpers_impl, ContractHelpersCall, true);174generate_stubgen!(contract_helpers_iface, ContractHelpersCall, false);