git.delta.rocks / unique-network / refs/commits / 22251a0f47f9

1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617#![cfg_attr(not(feature = "std"), no_std)]1819use core::marker::PhantomData;20use fp_evm::WithdrawReason;21use frame_support::traits::{Currency, IsSubType};22pub use pallet::*;23use pallet_evm::{EVMCurrencyAdapter, EnsureAddressOrigin};24use sp_core::{H160, U256};25use sp_runtime::TransactionOutcome;26use up_sponsorship::SponsorshipHandler;27use up_evm_mapping::EvmBackwardsAddressMapping;28use pallet_evm::AddressMapping;2930#[frame_support::pallet]31pub mod pallet {32	use super::*;3334	use frame_support::traits::Currency;35	use sp_std::vec::Vec;3637	#[pallet::config]38	pub trait Config: frame_system::Config {39		type EvmSponsorshipHandler: SponsorshipHandler<Self::AccountId, (H160, Vec<u8>)>;40		type Currency: Currency<Self::AccountId>;41		type EvmBackwardsAddressMapping: EvmBackwardsAddressMapping<Self::AccountId>;42		type EvmAddressMapping: AddressMapping<Self::AccountId>;43	}4445	#[pallet::pallet]46	#[pallet::generate_store(pub(super) trait Store)]47	pub struct Pallet<T>(_);48}4950type NegativeImbalanceOf<C, T> =51	<C as Currency<<T as frame_system::Config>::AccountId>>::NegativeImbalance;5253pub struct ChargeEvmLiquidityInfo<T>54where55	T: Config,56	T: pallet_evm::Config,57{58	who: H160,59	negative_imbalance: NegativeImbalanceOf<<T as Config>::Currency, T>,60}6162pub struct TransactionValidityHack<T: Config>(PhantomData<*const T>);63impl<T: Config> fp_evm::TransactionValidityHack<T::AccountId> for TransactionValidityHack<T> {64	fn who_pays_fee(origin: H160, reason: &WithdrawReason) -> Option<T::AccountId> {65		match reason {66			WithdrawReason::Call { target, input } => {67				// This method is only used for checking, we shouldn't touch storage in it68				frame_support::storage::with_transaction(|| {69					let origin_sub = T::EvmAddressMapping::into_account_id(origin);70					TransactionOutcome::Rollback(T::EvmSponsorshipHandler::get_sponsor(71						&origin_sub,72						&(*target, input.clone()),73					))74				})75			}76			_ => None,77		}78	}79}80pub struct OnChargeTransaction<T: Config>(PhantomData<*const T>);81impl<T> pallet_evm::OnChargeEVMTransaction<T> for OnChargeTransaction<T>82where83	T: Config,84	T: pallet_evm::Config,85{86	type LiquidityInfo = Option<ChargeEvmLiquidityInfo<T>>;8788	fn withdraw_fee(89		who: &T::AccountId,90		reason: WithdrawReason,91		fee: U256,92	) -> core::result::Result<Self::LiquidityInfo, pallet_evm::Error<T>> {93		let mut who_pays_fee = who.clone();94		if let WithdrawReason::Call { target, input } = &reason {95			who_pays_fee = T::EvmSponsorshipHandler::get_sponsor(who, &(*target, input.clone()))96				.unwrap_or(who_pays_fee);97		}9899		let negative_imbalance = EVMCurrencyAdapter::<<T as Config>::Currency, ()>::withdraw_fee(100			&who_pays_fee,101			reason,102			fee,103		)?;104105		let who_pays_fee_eth = T::EvmBackwardsAddressMapping::from_account_id(who_pays_fee);106		Ok(negative_imbalance.map(|i| ChargeEvmLiquidityInfo {107			who: who_pays_fee_eth,108			negative_imbalance: i,109		}))110	}111112	fn correct_and_deposit_fee(113		who: &T::AccountId,114		corrected_fee: U256,115		already_withdrawn: Self::LiquidityInfo,116	) {117		<EVMCurrencyAdapter<<T as Config>::Currency, ()> as pallet_evm::OnChargeEVMTransaction<T>>::correct_and_deposit_fee(118			&already_withdrawn.as_ref().map(|e| T::EvmAddressMapping::into_account_id(e.who)).unwrap_or(who.clone()),119			corrected_fee,120			already_withdrawn.map(|e| e.negative_imbalance),121		)122	}123124	fn pay_priority_fee(tip: U256) {125		<EVMCurrencyAdapter<<T as Config>::Currency, ()> as pallet_evm::OnChargeEVMTransaction<T>>::pay_priority_fee(tip)126	}127}128129/// Implements sponsoring for evm calls performed from pallet-evm (via api.tx.ethereum.transact/api.tx.evm.call)130pub struct BridgeSponsorshipHandler<T>(PhantomData<T>);131impl<T, C> SponsorshipHandler<T::AccountId, C> for BridgeSponsorshipHandler<T>132where133	T: Config + pallet_evm::Config,134	C: IsSubType<pallet_evm::Call<T>>,135{136	fn get_sponsor(who: &T::AccountId, call: &C) -> Option<T::AccountId> {137		match call.is_sub_type()? {138			pallet_evm::Call::call {139				source,140				target,141				input,142				..143			} => {144				let _ = T::CallOrigin::ensure_address_origin(145					source,146					<frame_system::RawOrigin<T::AccountId>>::Signed(who.clone()).into(),147				)148				.ok()?;149				// Effects from EvmSponsorshipHandler are applied in OnChargeEvmTransaction by pallet_evm::runner150				// TODO: Should we implement simulation mode (test, but do not apply effects) in `up-sponsorship`?151				let sponsor = frame_support::storage::with_transaction(|| {152					TransactionOutcome::Rollback(T::EvmSponsorshipHandler::get_sponsor(153						&who,154						&(*target, input.clone()),155					))156				})?;157				Some(sponsor)158			}159			_ => None,160		}161	}162}