git.delta.rocks / unique-network / refs/commits / ffd579c349f8

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, SponsorBasket, SponsoringRateLimit, SponsoringModeT,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(<Pallet<T>>::sponsoring_mode(contract_address) != SponsoringModeT::Disabled)32	}3334	/// Deprecated35	fn toggle_sponsoring(36		&mut self,37		caller: caller,38		contract_address: address,39		enabled: bool,40	) -> Result<void> {41		<Pallet<T>>::ensure_owner(contract_address, caller)?;42		<Pallet<T>>::toggle_sponsoring(contract_address, enabled);43		Ok(())44	}4546	fn set_sponsoring_mode(47		&mut self,48		caller: caller,49		contract_address: address,50		mode: uint8,51	) -> Result<void> {52		<Pallet<T>>::ensure_owner(contract_address, caller)?;53		let mode = SponsoringModeT::from_eth(mode).ok_or("unknown mode")?;54		<Pallet<T>>::set_sponsoring_mode(contract_address, mode);55		Ok(())56	}5758	fn sponsoring_mode(&self, contract_address: address) -> Result<uint8> {59		Ok(<Pallet<T>>::sponsoring_mode(contract_address).to_eth())60	}6162	fn set_sponsoring_rate_limit(63		&mut self,64		caller: caller,65		contract_address: address,66		rate_limit: uint32,67	) -> Result<void> {68		<Pallet<T>>::ensure_owner(contract_address, caller)?;69		<Pallet<T>>::set_sponsoring_rate_limit(contract_address, rate_limit.into());70		Ok(())71	}7273	fn get_sponsoring_rate_limit(&self, contract_address: address) -> Result<uint32> {74		Ok(<SponsoringRateLimit<T>>::get(contract_address)75			.try_into()76			.map_err(|_| "rate limit > u32::MAX")?)77	}7879	fn allowed(&self, contract_address: address, user: address) -> Result<bool> {80		self.0.consume_sload()?;81		Ok(<Pallet<T>>::allowed(contract_address, user)82			|| !<AllowlistEnabled<T>>::get(contract_address))83	}8485	fn allowlist_enabled(&self, contract_address: address) -> Result<bool> {86		Ok(<AllowlistEnabled<T>>::get(contract_address))87	}8889	fn toggle_allowlist(90		&mut self,91		caller: caller,92		contract_address: address,93		enabled: bool,94	) -> Result<void> {95		<Pallet<T>>::ensure_owner(contract_address, caller)?;96		<Pallet<T>>::toggle_allowlist(contract_address, enabled);97		Ok(())98	}99100	fn toggle_allowed(101		&mut self,102		caller: caller,103		contract_address: address,104		user: address,105		allowed: bool,106	) -> Result<void> {107		<Pallet<T>>::ensure_owner(contract_address, caller)?;108		<Pallet<T>>::toggle_allowed(contract_address, user, allowed);109		Ok(())110	}111}112113pub struct HelpersOnMethodCall<T: Config>(PhantomData<*const T>);114impl<T: Config> OnMethodCall<T> for HelpersOnMethodCall<T> {115	fn is_reserved(contract: &sp_core::H160) -> bool {116		contract == &T::ContractAddress::get()117	}118119	fn is_used(contract: &sp_core::H160) -> bool {120		contract == &T::ContractAddress::get()121	}122123	fn call(124		source: &sp_core::H160,125		target: &sp_core::H160,126		gas_left: u64,127		input: &[u8],128		value: sp_core::U256,129	) -> Option<PrecompileResult> {130		// TODO: Extract to another OnMethodCall handler131		if <AllowlistEnabled<T>>::get(target) && !<Pallet<T>>::allowed(*target, *source) {132			return Some(Err(PrecompileFailure::Revert {133				exit_status: ExitRevert::Reverted,134				cost: 0,135				output: {136					let mut writer = AbiWriter::new_call(evm_coder::fn_selector!(Error(string)));137					writer.string("Target contract is allowlisted");138					writer.finish()139				},140			}));141		}142143		if target != &T::ContractAddress::get() {144			return None;145		}146147		let helpers = ContractHelpers::<T>(SubstrateRecorder::<T>::new(*target, gas_left));148		pallet_evm_coder_substrate::call(*source, helpers, value, input)149	}150151	fn get_code(contract: &sp_core::H160) -> Option<Vec<u8>> {152		(contract == &T::ContractAddress::get())153			.then(|| include_bytes!("./stubs/ContractHelpers.raw").to_vec())154	}155}156157pub struct HelpersOnCreate<T: Config>(PhantomData<*const T>);158impl<T: Config> OnCreate<T> for HelpersOnCreate<T> {159	fn on_create(owner: H160, contract: H160) {160		<Owner<T>>::insert(contract, owner);161	}162}163164pub struct HelpersContractSponsoring<T: Config>(PhantomData<*const T>);165impl<T: Config> SponsorshipHandler<H160, (H160, Vec<u8>)> for HelpersContractSponsoring<T> {166	fn get_sponsor(who: &H160, call: &(H160, Vec<u8>)) -> Option<H160> {167		let mode = <Pallet<T>>::sponsoring_mode(call.0);168		if mode == SponsoringModeT::Disabled {169			return None;170		}171		if mode == SponsoringModeT::Allowlisted && !<Pallet<T>>::allowed(call.0, *who) {172			return None;173		}174		let block_number = <frame_system::Pallet<T>>::block_number() as T::BlockNumber;175176		if let Some(last_tx_block) = <SponsorBasket<T>>::get(&call.0, who) {177			let limit = <SponsoringRateLimit<T>>::get(&call.0);178179			let timeout = last_tx_block + limit;180			if block_number < timeout {181				return None;182			}183		}184185		<SponsorBasket<T>>::insert(&call.0, who, block_number);186187		Some(call.0)188	}189}190191generate_stubgen!(contract_helpers_impl, ContractHelpersCall<()>, true);192generate_stubgen!(contract_helpers_iface, ContractHelpersCall<()>, false);