1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

// SPDX-License-Identifier: Apache-2.0
// This file is part of Frontier.
//
// Copyright (c) 2021-2022 Parity Technologies (UK) Ltd.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// 	http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#![cfg_attr(not(feature = "std"), no_std)]
#![warn(unused_crate_dependencies)]

#[cfg(feature = "runtime-benchmarks")]
pub mod benchmarking;
#[cfg(test)]
mod mock;
#[cfg(test)]
mod tests;
pub mod weights;

// Substrate
use frame_support::dispatch::PostDispatchInfo;
use sp_core::H160;
use sp_runtime::traits::Zero;
use sp_std::vec::Vec;
// Frontier
pub use pallet_evm::AddressMapping;

pub use self::{pallet::*, weights::WeightInfo};

#[frame_support::pallet]
pub mod pallet {
	use super::*;
	use frame_support::pallet_prelude::*;
	use frame_system::pallet_prelude::*;

	#[pallet::pallet]
	pub struct Pallet<T>(PhantomData<T>);

	#[pallet::config]
	pub trait Config: frame_system::Config {
		/// Mapping from address to account id.
		type AddressMapping: AddressMapping<Self::AccountId>;
		/// Weight information for extrinsics in this pallet.
		type WeightInfo: WeightInfo;
	}

	#[pallet::error]
	pub enum Error<T> {
		/// Maximum address count exceeded
		MaxAddressCountExceeded,
	}

	#[pallet::call]
	impl<T: Config> Pallet<T> {
		/// Increment `sufficients` for existing accounts having a nonzero `nonce` but zero `sufficients`, `consumers` and `providers` value.
		/// This state was caused by a previous bug in EVM create account dispatchable.
		///
		/// Any accounts in the input list not satisfying the above condition will remain unaffected.
		#[pallet::call_index(0)]
		#[pallet::weight(
			<T as pallet::Config>::WeightInfo::hotfix_inc_account_sufficients(addresses.len().try_into().unwrap_or(u32::MAX))
		)]
		pub fn hotfix_inc_account_sufficients(
			origin: OriginFor<T>,
			addresses: Vec<H160>,
		) -> DispatchResultWithPostInfo {
			const MAX_ADDRESS_COUNT: usize = 1000;

			frame_system::ensure_signed(origin)?;
			ensure!(
				addresses.len() <= MAX_ADDRESS_COUNT,
				Error::<T>::MaxAddressCountExceeded
			);

			for address in addresses {
				let account_id = T::AddressMapping::into_account_id(address);
				let nonce = frame_system::Pallet::<T>::account_nonce(&account_id);
				let refs = frame_system::Pallet::<T>::consumers(&account_id)
					.saturating_add(frame_system::Pallet::<T>::providers(&account_id))
					.saturating_add(frame_system::Pallet::<T>::sufficients(&account_id));

				if !nonce.is_zero() && refs.is_zero() {
					frame_system::Pallet::<T>::inc_sufficients(&account_id);
				}
			}

			Ok(PostDispatchInfo {
				actual_weight: None,
				pays_fee: Pays::Yes,
			})
		}
	}
}