1#![cfg_attr(not(feature = "std"), no_std)]
99
100extern crate alloc;
101
102use alloc::{borrow::Cow, boxed::Box, vec, vec::Vec};
103use core::{fmt::Debug, marker::PhantomData};
104use pallet_prelude::{BlockNumberFor, HeaderFor};
105#[cfg(feature = "std")]
106use serde::Serialize;
107use sp_io::hashing::blake2_256;
108#[cfg(feature = "runtime-benchmarks")]
109use sp_runtime::traits::TrailingZeroInput;
110use sp_runtime::{
111 generic,
112 traits::{
113 self, AsTransactionAuthorizedOrigin, AtLeast32Bit, BadOrigin, BlockNumberProvider, Bounded,
114 CheckEqual, Dispatchable, Hash, Header, Lookup, LookupError, MaybeDisplay,
115 MaybeSerializeDeserialize, Member, One, Saturating, SimpleBitOps, StaticLookup, Zero,
116 },
117 transaction_validity::{
118 InvalidTransaction, TransactionLongevity, TransactionSource, TransactionValidity,
119 ValidTransaction,
120 },
121 DispatchError,
122};
123use sp_version::RuntimeVersion;
124
125use codec::{Decode, DecodeWithMemTracking, Encode, EncodeLike, FullCodec, MaxEncodedLen};
126#[cfg(feature = "std")]
127use frame_support::traits::BuildGenesisConfig;
128use frame_support::{
129 defensive,
130 dispatch::{
131 extract_actual_pays_fee, extract_actual_weight, DispatchClass, DispatchInfo,
132 DispatchResult, DispatchResultWithPostInfo, GetDispatchInfo, PerDispatchClass,
133 PostDispatchInfo,
134 },
135 ensure, impl_ensure_origin_with_arg_ignoring_arg,
136 migrations::MultiStepMigrator,
137 pallet_prelude::Pays,
138 storage::{self, StorageStreamIter},
139 traits::{
140 ConstU32, Contains, EnsureOrigin, EnsureOriginWithArg, Get, HandleLifetime,
141 OnKilledAccount, OnNewAccount, OnRuntimeUpgrade, OriginTrait, PalletInfo, SortedMembers,
142 StoredMap, TypedGet,
143 },
144 Parameter,
145};
146use scale_info::TypeInfo;
147use sp_core::storage::well_known_keys;
148use sp_runtime::{
149 traits::{DispatchInfoOf, PostDispatchInfoOf},
150 transaction_validity::TransactionValidityError,
151};
152use sp_weights::{RuntimeDbWeight, Weight, WeightMeter};
153
154#[cfg(any(feature = "std", test))]
155use sp_io::TestExternalities;
156
157pub mod limits;
158#[cfg(test)]
159pub(crate) mod mock;
160
161pub mod offchain;
162
163mod extensions;
164#[cfg(feature = "std")]
165pub mod mocking;
166#[cfg(test)]
167mod tests;
168pub mod weights;
169
170pub mod migrations;
171
172pub use extensions::{
173 authorize_call::AuthorizeCall,
174 check_genesis::CheckGenesis,
175 check_mortality::CheckMortality,
176 check_non_zero_sender::CheckNonZeroSender,
177 check_nonce::{CheckNonce, ValidNonceInfo},
178 check_spec_version::CheckSpecVersion,
179 check_tx_version::CheckTxVersion,
180 check_weight::CheckWeight,
181 weight_reclaim::WeightReclaim,
182 weights::SubstrateWeight as SubstrateExtensionsWeight,
183 WeightInfo as ExtensionsWeightInfo,
184};
185pub use extensions::check_mortality::CheckMortality as CheckEra;
187pub use frame_support::dispatch::RawOrigin;
188use frame_support::traits::{Authorize, PostInherents, PostTransactions, PreInherents};
189use sp_core::storage::StateVersion;
190pub use weights::WeightInfo;
191
192const LOG_TARGET: &str = "runtime::system";
193
194pub fn extrinsics_root<H: Hash, E: codec::Encode>(
199 extrinsics: &[E],
200 state_version: StateVersion,
201) -> H::Output {
202 extrinsics_data_root::<H>(extrinsics.iter().map(codec::Encode::encode).collect(), state_version)
203}
204
205pub fn extrinsics_data_root<H: Hash>(xts: Vec<Vec<u8>>, state_version: StateVersion) -> H::Output {
210 H::ordered_trie_root(xts, state_version)
211}
212
213pub type ConsumedWeight = PerDispatchClass<Weight>;
215
216pub use pallet::*;
217
218pub trait SetCode<T: Config> {
220 fn set_code(code: Vec<u8>) -> DispatchResult;
222}
223
224impl<T: Config> SetCode<T> for () {
225 fn set_code(code: Vec<u8>) -> DispatchResult {
226 <Pallet<T>>::update_code_in_storage(&code);
227 Ok(())
228 }
229}
230
231pub trait ConsumerLimits {
233 fn max_consumers() -> RefCount;
235 fn max_overflow() -> RefCount;
241}
242
243impl<const Z: u32> ConsumerLimits for ConstU32<Z> {
244 fn max_consumers() -> RefCount {
245 Z
246 }
247 fn max_overflow() -> RefCount {
248 Z
249 }
250}
251
252impl<MaxNormal: Get<u32>, MaxOverflow: Get<u32>> ConsumerLimits for (MaxNormal, MaxOverflow) {
253 fn max_consumers() -> RefCount {
254 MaxNormal::get()
255 }
256 fn max_overflow() -> RefCount {
257 MaxOverflow::get()
258 }
259}
260
261#[derive(Decode, Encode, Default, PartialEq, Eq, MaxEncodedLen, TypeInfo)]
264#[scale_info(skip_type_params(T))]
265pub struct CodeUpgradeAuthorization<T>
266where
267 T: Config,
268{
269 code_hash: T::Hash,
271 check_version: bool,
273}
274
275#[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
276impl<T> CodeUpgradeAuthorization<T>
277where
278 T: Config,
279{
280 pub fn code_hash(&self) -> &T::Hash {
281 &self.code_hash
282 }
283}
284
285#[derive(
289 Clone, Copy, Eq, PartialEq, Default, Debug, Encode, Decode, DecodeWithMemTracking, TypeInfo,
290)]
291pub struct DispatchEventInfo {
292 pub weight: Weight,
294 pub class: DispatchClass,
296 pub pays_fee: Pays,
298}
299
300#[frame_support::pallet]
301pub mod pallet {
302 use crate::{self as frame_system, pallet_prelude::*, *};
303 use codec::HasCompact;
304 use frame_support::pallet_prelude::*;
305
306 pub mod config_preludes {
308 use super::{inject_runtime_type, DefaultConfig};
309 use frame_support::{derive_impl, traits::Get};
310
311 pub struct TestBlockHashCount<C: Get<u32>>(core::marker::PhantomData<C>);
317 impl<I: From<u32>, C: Get<u32>> Get<I> for TestBlockHashCount<C> {
318 fn get() -> I {
319 C::get().into()
320 }
321 }
322
323 pub struct TestDefaultConfig;
330
331 #[frame_support::register_default_impl(TestDefaultConfig)]
332 impl DefaultConfig for TestDefaultConfig {
333 type Nonce = u32;
334 type Hash = sp_core::hash::H256;
335 type Hashing = sp_runtime::traits::BlakeTwo256;
336 type AccountId = u64;
337 type Lookup = sp_runtime::traits::IdentityLookup<Self::AccountId>;
338 type MaxConsumers = frame_support::traits::ConstU32<16>;
339 type AccountData = ();
340 type OnNewAccount = ();
341 type OnKilledAccount = ();
342 type SystemWeightInfo = ();
343 type ExtensionsWeightInfo = ();
344 type SS58Prefix = ();
345 type Version = ();
346 type BlockWeights = ();
347 type BlockLength = ();
348 type DbWeight = ();
349 #[inject_runtime_type]
350 type RuntimeEvent = ();
351 #[inject_runtime_type]
352 type RuntimeOrigin = ();
353 #[inject_runtime_type]
354 type RuntimeCall = ();
355 #[inject_runtime_type]
356 type PalletInfo = ();
357 #[inject_runtime_type]
358 type RuntimeTask = ();
359 type BaseCallFilter = frame_support::traits::Everything;
360 type BlockHashCount = TestBlockHashCount<frame_support::traits::ConstU32<10>>;
361 type OnSetCode = ();
362 type SingleBlockMigrations = ();
363 type MultiBlockMigrator = ();
364 type PreInherents = ();
365 type PostInherents = ();
366 type PostTransactions = ();
367 }
368
369 pub struct SolochainDefaultConfig;
383
384 #[frame_support::register_default_impl(SolochainDefaultConfig)]
385 impl DefaultConfig for SolochainDefaultConfig {
386 type Nonce = u32;
388
389 type Hash = sp_core::hash::H256;
391
392 type Hashing = sp_runtime::traits::BlakeTwo256;
394
395 type AccountId = sp_runtime::AccountId32;
397
398 type Lookup = sp_runtime::traits::AccountIdLookup<Self::AccountId, ()>;
400
401 type MaxConsumers = frame_support::traits::ConstU32<128>;
403
404 type AccountData = ();
406
407 type OnNewAccount = ();
409
410 type OnKilledAccount = ();
412
413 type SystemWeightInfo = ();
415
416 type ExtensionsWeightInfo = ();
418
419 type SS58Prefix = ();
421
422 type Version = ();
424
425 type BlockWeights = ();
427
428 type BlockLength = ();
430
431 type DbWeight = ();
433
434 #[inject_runtime_type]
436 type RuntimeEvent = ();
437
438 #[inject_runtime_type]
440 type RuntimeOrigin = ();
441
442 #[inject_runtime_type]
445 type RuntimeCall = ();
446
447 #[inject_runtime_type]
449 type RuntimeTask = ();
450
451 #[inject_runtime_type]
453 type PalletInfo = ();
454
455 type BaseCallFilter = frame_support::traits::Everything;
457
458 type BlockHashCount = TestBlockHashCount<frame_support::traits::ConstU32<256>>;
461
462 type OnSetCode = ();
464 type SingleBlockMigrations = ();
465 type MultiBlockMigrator = ();
466 type PreInherents = ();
467 type PostInherents = ();
468 type PostTransactions = ();
469 }
470
471 pub struct RelayChainDefaultConfig;
473
474 #[derive_impl(SolochainDefaultConfig as DefaultConfig, no_aggregated_types)]
476 #[frame_support::register_default_impl(RelayChainDefaultConfig)]
477 impl DefaultConfig for RelayChainDefaultConfig {}
478
479 pub struct ParaChainDefaultConfig;
481
482 #[derive_impl(SolochainDefaultConfig as DefaultConfig, no_aggregated_types)]
484 #[frame_support::register_default_impl(ParaChainDefaultConfig)]
485 impl DefaultConfig for ParaChainDefaultConfig {}
486 }
487
488 #[pallet::config(with_default, frame_system_config)]
490 #[pallet::disable_frame_system_supertrait_check]
491 pub trait Config: 'static + Eq + Clone {
492 #[pallet::no_default_bounds]
494 type RuntimeEvent: Parameter
495 + Member
496 + From<Event<Self>>
497 + Debug
498 + IsType<<Self as frame_system::Config>::RuntimeEvent>;
499
500 #[pallet::no_default_bounds]
511 type BaseCallFilter: Contains<Self::RuntimeCall>;
512
513 #[pallet::constant]
515 type BlockWeights: Get<limits::BlockWeights>;
516
517 #[pallet::constant]
519 type BlockLength: Get<limits::BlockLength>;
520
521 #[pallet::no_default_bounds]
523 type RuntimeOrigin: Into<Result<RawOrigin<Self::AccountId>, Self::RuntimeOrigin>>
524 + From<RawOrigin<Self::AccountId>>
525 + Clone
526 + OriginTrait<Call = Self::RuntimeCall, AccountId = Self::AccountId>
527 + AsTransactionAuthorizedOrigin;
528
529 #[docify::export(system_runtime_call)]
530 #[pallet::no_default_bounds]
532 type RuntimeCall: Parameter
533 + Dispatchable<RuntimeOrigin = Self::RuntimeOrigin>
534 + Debug
535 + GetDispatchInfo
536 + From<Call<Self>>
537 + Authorize;
538
539 #[pallet::no_default_bounds]
541 type RuntimeTask: Task;
542
543 type Nonce: Parameter
545 + HasCompact<Type: DecodeWithMemTracking>
546 + Member
547 + MaybeSerializeDeserialize
548 + Debug
549 + Default
550 + MaybeDisplay
551 + AtLeast32Bit
552 + Copy
553 + MaxEncodedLen;
554
555 type Hash: Parameter
557 + Member
558 + MaybeSerializeDeserialize
559 + Debug
560 + MaybeDisplay
561 + SimpleBitOps
562 + Ord
563 + Default
564 + Copy
565 + CheckEqual
566 + core::hash::Hash
567 + AsRef<[u8]>
568 + AsMut<[u8]>
569 + MaxEncodedLen;
570
571 type Hashing: Hash<Output = Self::Hash> + TypeInfo;
573
574 type AccountId: Parameter
576 + Member
577 + MaybeSerializeDeserialize
578 + Debug
579 + MaybeDisplay
580 + Ord
581 + MaxEncodedLen;
582
583 type Lookup: StaticLookup<Target = Self::AccountId>;
590
591 #[pallet::no_default]
594 type Block: Parameter + Member + traits::Block<Hash = Self::Hash>;
595
596 #[pallet::constant]
598 #[pallet::no_default_bounds]
599 type BlockHashCount: Get<BlockNumberFor<Self>>;
600
601 #[pallet::constant]
603 type DbWeight: Get<RuntimeDbWeight>;
604
605 #[pallet::constant]
607 type Version: Get<RuntimeVersion>;
608
609 #[pallet::no_default_bounds]
616 type PalletInfo: PalletInfo;
617
618 type AccountData: Member + FullCodec + Clone + Default + TypeInfo + MaxEncodedLen;
621
622 type OnNewAccount: OnNewAccount<Self::AccountId>;
624
625 type OnKilledAccount: OnKilledAccount<Self::AccountId>;
629
630 type SystemWeightInfo: WeightInfo;
632
633 type ExtensionsWeightInfo: extensions::WeightInfo;
635
636 #[pallet::constant]
642 type SS58Prefix: Get<u16>;
643
644 #[pallet::no_default_bounds]
652 type OnSetCode: SetCode<Self>;
653
654 type MaxConsumers: ConsumerLimits;
656
657 type SingleBlockMigrations: OnRuntimeUpgrade;
664
665 type MultiBlockMigrator: MultiStepMigrator;
670
671 type PreInherents: PreInherents;
675
676 type PostInherents: PostInherents;
680
681 type PostTransactions: PostTransactions;
685 }
686
687 #[pallet::pallet]
688 pub struct Pallet<T>(_);
689
690 #[pallet::hooks]
691 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {
692 #[cfg(feature = "std")]
693 fn integrity_test() {
694 T::BlockWeights::get().validate().expect("The weights are invalid.");
695 }
696 }
697
698 #[pallet::call(weight = <T as Config>::SystemWeightInfo)]
699 impl<T: Config> Pallet<T> {
700 #[pallet::call_index(0)]
704 #[pallet::weight(T::SystemWeightInfo::remark(remark.len() as u32))]
705 pub fn remark(_origin: OriginFor<T>, remark: Vec<u8>) -> DispatchResultWithPostInfo {
706 let _ = remark; Ok(().into())
708 }
709
710 #[pallet::call_index(1)]
712 #[pallet::weight((T::SystemWeightInfo::set_heap_pages(), DispatchClass::Operational))]
713 pub fn set_heap_pages(origin: OriginFor<T>, pages: u64) -> DispatchResultWithPostInfo {
714 ensure_root(origin)?;
715 storage::unhashed::put_raw(well_known_keys::HEAP_PAGES, &pages.encode());
716 Self::deposit_log(generic::DigestItem::RuntimeEnvironmentUpdated);
717 Ok(().into())
718 }
719
720 #[pallet::call_index(2)]
722 #[pallet::weight((T::SystemWeightInfo::set_code(), DispatchClass::Operational))]
723 pub fn set_code(origin: OriginFor<T>, code: Vec<u8>) -> DispatchResultWithPostInfo {
724 ensure_root(origin)?;
725 Self::can_set_code(&code, true).into_result()?;
726 T::OnSetCode::set_code(code)?;
727 Ok(Some(T::BlockWeights::get().max_block).into())
729 }
730
731 #[pallet::call_index(3)]
736 #[pallet::weight((T::SystemWeightInfo::set_code(), DispatchClass::Operational))]
737 pub fn set_code_without_checks(
738 origin: OriginFor<T>,
739 code: Vec<u8>,
740 ) -> DispatchResultWithPostInfo {
741 ensure_root(origin)?;
742 Self::can_set_code(&code, false).into_result()?;
743 T::OnSetCode::set_code(code)?;
744 Ok(Some(T::BlockWeights::get().max_block).into())
745 }
746
747 #[pallet::call_index(4)]
749 #[pallet::weight((
750 T::SystemWeightInfo::set_storage(items.len() as u32),
751 DispatchClass::Operational,
752 ))]
753 pub fn set_storage(
754 origin: OriginFor<T>,
755 items: Vec<KeyValue>,
756 ) -> DispatchResultWithPostInfo {
757 ensure_root(origin)?;
758 for i in &items {
759 storage::unhashed::put_raw(&i.0, &i.1);
760 }
761 Ok(().into())
762 }
763
764 #[pallet::call_index(5)]
766 #[pallet::weight((
767 T::SystemWeightInfo::kill_storage(keys.len() as u32),
768 DispatchClass::Operational,
769 ))]
770 pub fn kill_storage(origin: OriginFor<T>, keys: Vec<Key>) -> DispatchResultWithPostInfo {
771 ensure_root(origin)?;
772 for key in &keys {
773 storage::unhashed::kill(key);
774 }
775 Ok(().into())
776 }
777
778 #[pallet::call_index(6)]
783 #[pallet::weight((
784 T::SystemWeightInfo::kill_prefix(subkeys.saturating_add(1)),
785 DispatchClass::Operational,
786 ))]
787 pub fn kill_prefix(
788 origin: OriginFor<T>,
789 prefix: Key,
790 subkeys: u32,
791 ) -> DispatchResultWithPostInfo {
792 ensure_root(origin)?;
793 let _ = storage::unhashed::clear_prefix(&prefix, Some(subkeys), None);
794 Ok(().into())
795 }
796
797 #[pallet::call_index(7)]
799 #[pallet::weight(T::SystemWeightInfo::remark_with_event(remark.len() as u32))]
800 pub fn remark_with_event(
801 origin: OriginFor<T>,
802 remark: Vec<u8>,
803 ) -> DispatchResultWithPostInfo {
804 let who = ensure_signed(origin)?;
805 let hash = T::Hashing::hash(&remark[..]);
806 Self::deposit_event(Event::Remarked { sender: who, hash });
807 Ok(().into())
808 }
809
810 #[cfg(feature = "experimental")]
811 #[pallet::call_index(8)]
812 #[pallet::weight(task.weight())]
813 pub fn do_task(_origin: OriginFor<T>, task: T::RuntimeTask) -> DispatchResultWithPostInfo {
814 if !task.is_valid() {
815 return Err(Error::<T>::InvalidTask.into());
816 }
817
818 Self::deposit_event(Event::TaskStarted { task: task.clone() });
819 if let Err(err) = task.run() {
820 Self::deposit_event(Event::TaskFailed { task, err });
821 return Err(Error::<T>::FailedTask.into());
822 }
823
824 Self::deposit_event(Event::TaskCompleted { task });
826
827 Ok(().into())
829 }
830
831 #[pallet::call_index(9)]
836 #[pallet::weight((T::SystemWeightInfo::authorize_upgrade(), DispatchClass::Operational))]
837 pub fn authorize_upgrade(origin: OriginFor<T>, code_hash: T::Hash) -> DispatchResult {
838 ensure_root(origin)?;
839 Self::do_authorize_upgrade(code_hash, true);
840 Ok(())
841 }
842
843 #[pallet::call_index(10)]
852 #[pallet::weight((T::SystemWeightInfo::authorize_upgrade(), DispatchClass::Operational))]
853 pub fn authorize_upgrade_without_checks(
854 origin: OriginFor<T>,
855 code_hash: T::Hash,
856 ) -> DispatchResult {
857 ensure_root(origin)?;
858 Self::do_authorize_upgrade(code_hash, false);
859 Ok(())
860 }
861
862 #[pallet::call_index(11)]
872 #[pallet::weight((T::SystemWeightInfo::apply_authorized_upgrade(), DispatchClass::Operational))]
873 pub fn apply_authorized_upgrade(
874 _: OriginFor<T>,
875 code: Vec<u8>,
876 ) -> DispatchResultWithPostInfo {
877 let res = Self::validate_code_is_authorized(&code)?;
878 AuthorizedUpgrade::<T>::kill();
879
880 match Self::can_set_code(&code, res.check_version) {
881 CanSetCodeResult::Ok => {},
882 CanSetCodeResult::MultiBlockMigrationsOngoing => {
883 return Err(Error::<T>::MultiBlockMigrationsOngoing.into())
884 },
885 CanSetCodeResult::InvalidVersion(error) => {
886 Self::deposit_event(Event::RejectedInvalidAuthorizedUpgrade {
888 code_hash: res.code_hash,
889 error: error.into(),
890 });
891
892 return Ok(Pays::No.into());
894 },
895 };
896 T::OnSetCode::set_code(code)?;
897
898 Ok(PostDispatchInfo {
899 actual_weight: Some(T::BlockWeights::get().max_block),
901 pays_fee: Pays::No,
903 })
904 }
905 }
906
907 #[pallet::event]
909 pub enum Event<T: Config> {
910 ExtrinsicSuccess { dispatch_info: DispatchEventInfo },
912 ExtrinsicFailed { dispatch_error: DispatchError, dispatch_info: DispatchEventInfo },
914 CodeUpdated,
916 NewAccount { account: T::AccountId },
918 KilledAccount { account: T::AccountId },
920 Remarked { sender: T::AccountId, hash: T::Hash },
922 #[cfg(feature = "experimental")]
923 TaskStarted { task: T::RuntimeTask },
925 #[cfg(feature = "experimental")]
926 TaskCompleted { task: T::RuntimeTask },
928 #[cfg(feature = "experimental")]
929 TaskFailed { task: T::RuntimeTask, err: DispatchError },
931 UpgradeAuthorized { code_hash: T::Hash, check_version: bool },
933 RejectedInvalidAuthorizedUpgrade { code_hash: T::Hash, error: DispatchError },
935 }
936
937 #[pallet::error]
939 pub enum Error<T> {
940 InvalidSpecName,
943 SpecVersionNeedsToIncrease,
946 FailedToExtractRuntimeVersion,
950 NonDefaultComposite,
952 NonZeroRefCount,
954 CallFiltered,
956 MultiBlockMigrationsOngoing,
958 #[cfg(feature = "experimental")]
959 InvalidTask,
961 #[cfg(feature = "experimental")]
962 FailedTask,
964 NothingAuthorized,
966 Unauthorized,
968 }
969
970 #[pallet::origin]
972 pub type Origin<T> = RawOrigin<<T as Config>::AccountId>;
973
974 #[pallet::storage]
976 #[pallet::getter(fn account)]
977 pub type Account<T: Config> = StorageMap<
978 _,
979 Blake2_128Concat,
980 T::AccountId,
981 AccountInfo<T::Nonce, T::AccountData>,
982 ValueQuery,
983 >;
984
985 #[pallet::storage]
987 #[pallet::whitelist_storage]
988 pub(super) type ExtrinsicCount<T: Config> = StorageValue<_, u32>;
989
990 #[pallet::storage]
992 #[pallet::whitelist_storage]
993 pub type InherentsApplied<T: Config> = StorageValue<_, bool, ValueQuery>;
994
995 #[pallet::storage]
997 #[pallet::whitelist_storage]
998 #[pallet::getter(fn block_weight)]
999 pub type BlockWeight<T: Config> = StorageValue<_, ConsumedWeight, ValueQuery>;
1000
1001 #[pallet::storage]
1005 #[pallet::whitelist_storage]
1006 pub type BlockSize<T: Config> = StorageValue<_, u32>;
1007
1008 #[pallet::storage]
1010 #[pallet::getter(fn block_hash)]
1011 pub type BlockHash<T: Config> =
1012 StorageMap<_, Twox64Concat, BlockNumberFor<T>, T::Hash, ValueQuery>;
1013
1014 #[pallet::storage]
1016 #[pallet::getter(fn extrinsic_data)]
1017 #[pallet::unbounded]
1018 pub(super) type ExtrinsicData<T: Config> =
1019 StorageMap<_, Twox64Concat, u32, Vec<u8>, ValueQuery>;
1020
1021 #[pallet::storage]
1023 #[pallet::whitelist_storage]
1024 #[pallet::getter(fn block_number)]
1025 pub(super) type Number<T: Config> = StorageValue<_, BlockNumberFor<T>, ValueQuery>;
1026
1027 #[pallet::storage]
1029 #[pallet::getter(fn parent_hash)]
1030 pub(super) type ParentHash<T: Config> = StorageValue<_, T::Hash, ValueQuery>;
1031
1032 #[pallet::storage]
1034 #[pallet::whitelist_storage]
1035 #[pallet::unbounded]
1036 #[pallet::getter(fn digest)]
1037 pub(super) type Digest<T: Config> = StorageValue<_, generic::Digest, ValueQuery>;
1038
1039 #[pallet::storage]
1047 #[pallet::whitelist_storage]
1048 #[pallet::disable_try_decode_storage]
1049 #[pallet::unbounded]
1050 pub(super) type Events<T: Config> =
1051 StorageValue<_, Vec<Box<EventRecord<T::RuntimeEvent, T::Hash>>>, ValueQuery>;
1052
1053 #[pallet::storage]
1055 #[pallet::whitelist_storage]
1056 #[pallet::getter(fn event_count)]
1057 pub(super) type EventCount<T: Config> = StorageValue<_, EventIndex, ValueQuery>;
1058
1059 #[pallet::storage]
1070 #[pallet::unbounded]
1071 #[pallet::getter(fn event_topics)]
1072 pub(super) type EventTopics<T: Config> =
1073 StorageMap<_, Blake2_128Concat, T::Hash, Vec<(BlockNumberFor<T>, EventIndex)>, ValueQuery>;
1074
1075 #[pallet::storage]
1077 #[pallet::unbounded]
1078 pub type LastRuntimeUpgrade<T: Config> = StorageValue<_, LastRuntimeUpgradeInfo>;
1079
1080 #[pallet::storage]
1082 pub(super) type BlocksTillUpgrade<T: Config> = StorageValue<_, u8>;
1083
1084 #[pallet::storage]
1086 pub(super) type UpgradedToU32RefCount<T: Config> = StorageValue<_, bool, ValueQuery>;
1087
1088 #[pallet::storage]
1091 pub(super) type UpgradedToTripleRefCount<T: Config> = StorageValue<_, bool, ValueQuery>;
1092
1093 #[pallet::storage]
1095 #[pallet::whitelist_storage]
1096 pub(super) type ExecutionPhase<T: Config> = StorageValue<_, Phase>;
1097
1098 #[pallet::storage]
1100 #[pallet::getter(fn authorized_upgrade)]
1101 pub(super) type AuthorizedUpgrade<T: Config> =
1102 StorageValue<_, CodeUpgradeAuthorization<T>, OptionQuery>;
1103
1104 #[pallet::storage]
1112 #[pallet::whitelist_storage]
1113 pub type ExtrinsicWeightReclaimed<T: Config> = StorageValue<_, Weight, ValueQuery>;
1114
1115 #[derive(frame_support::DefaultNoBound)]
1116 #[pallet::genesis_config]
1117 pub struct GenesisConfig<T: Config> {
1118 #[serde(skip)]
1119 pub _config: core::marker::PhantomData<T>,
1120 }
1121
1122 #[pallet::genesis_build]
1123 impl<T: Config> BuildGenesisConfig for GenesisConfig<T> {
1124 fn build(&self) {
1125 <BlockHash<T>>::insert::<_, T::Hash>(BlockNumberFor::<T>::zero(), hash69());
1126 <ParentHash<T>>::put::<T::Hash>(hash69());
1127 <LastRuntimeUpgrade<T>>::put(LastRuntimeUpgradeInfo::from(T::Version::get()));
1128 <UpgradedToU32RefCount<T>>::put(true);
1129 <UpgradedToTripleRefCount<T>>::put(true);
1130
1131 sp_io::storage::set(well_known_keys::EXTRINSIC_INDEX, &0u32.encode());
1132 }
1133 }
1134
1135 #[pallet::validate_unsigned]
1136 impl<T: Config> sp_runtime::traits::ValidateUnsigned for Pallet<T> {
1137 type Call = Call<T>;
1138 fn validate_unsigned(source: TransactionSource, call: &Self::Call) -> TransactionValidity {
1139 if let Call::apply_authorized_upgrade { ref code } = call {
1140 if let Ok(res) = Self::validate_code_is_authorized(&code[..]) {
1141 if Self::can_set_code(&code, false).is_ok() {
1142 return Ok(ValidTransaction {
1143 priority: u64::max_value(),
1144 requires: Vec::new(),
1145 provides: vec![res.code_hash.encode()],
1146 longevity: TransactionLongevity::max_value(),
1147 propagate: true,
1148 });
1149 }
1150 }
1151 }
1152
1153 #[cfg(feature = "experimental")]
1154 if let Call::do_task { ref task } = call {
1155 if source == TransactionSource::InBlock || source == TransactionSource::Local {
1163 if task.is_valid() {
1164 return Ok(ValidTransaction {
1165 priority: u64::max_value(),
1166 requires: Vec::new(),
1167 provides: vec![T::Hashing::hash_of(&task.encode()).as_ref().to_vec()],
1168 longevity: TransactionLongevity::max_value(),
1169 propagate: false,
1170 });
1171 }
1172 }
1173 }
1174
1175 #[cfg(not(feature = "experimental"))]
1176 let _ = source;
1177
1178 Err(InvalidTransaction::Call.into())
1179 }
1180 }
1181}
1182
1183pub type Key = Vec<u8>;
1184pub type KeyValue = (Vec<u8>, Vec<u8>);
1185
1186#[derive(Encode, Decode, Debug, TypeInfo, MaxEncodedLen)]
1188#[cfg_attr(feature = "std", derive(Serialize, PartialEq, Eq, Clone))]
1189pub enum Phase {
1190 ApplyExtrinsic(u32),
1192 Finalization,
1194 Initialization,
1196}
1197
1198impl Default for Phase {
1199 fn default() -> Self {
1200 Self::Initialization
1201 }
1202}
1203
1204#[derive(Encode, Decode, Debug, TypeInfo)]
1206#[cfg_attr(feature = "std", derive(Serialize, PartialEq, Eq, Clone))]
1207pub struct EventRecord<E: Parameter + Member, T> {
1208 pub phase: Phase,
1210 pub event: E,
1212 pub topics: Vec<T>,
1214}
1215
1216fn hash69<T: AsMut<[u8]> + Default>() -> T {
1219 let mut h = T::default();
1220 h.as_mut().iter_mut().for_each(|byte| *byte = 69);
1221 h
1222}
1223
1224type EventIndex = u32;
1229
1230pub type RefCount = u32;
1232
1233#[derive(Clone, Eq, PartialEq, Default, Debug, Encode, Decode, TypeInfo, MaxEncodedLen)]
1235pub struct AccountInfo<Nonce, AccountData> {
1236 pub nonce: Nonce,
1238 pub consumers: RefCount,
1241 pub providers: RefCount,
1244 pub sufficients: RefCount,
1247 pub data: AccountData,
1250}
1251
1252#[derive(Debug, Encode, Decode, TypeInfo)]
1255#[cfg_attr(feature = "std", derive(PartialEq))]
1256pub struct LastRuntimeUpgradeInfo {
1257 pub spec_version: codec::Compact<u32>,
1258 pub spec_name: Cow<'static, str>,
1259}
1260
1261impl LastRuntimeUpgradeInfo {
1262 pub fn was_upgraded(&self, current: &RuntimeVersion) -> bool {
1266 current.spec_version > self.spec_version.0 || current.spec_name != self.spec_name
1267 }
1268}
1269
1270impl From<RuntimeVersion> for LastRuntimeUpgradeInfo {
1271 fn from(version: RuntimeVersion) -> Self {
1272 Self { spec_version: version.spec_version.into(), spec_name: version.spec_name }
1273 }
1274}
1275
1276pub struct EnsureRoot<AccountId>(core::marker::PhantomData<AccountId>);
1278impl<O: OriginTrait, AccountId> EnsureOrigin<O> for EnsureRoot<AccountId> {
1279 type Success = ();
1280 fn try_origin(o: O) -> Result<Self::Success, O> {
1281 match o.as_system_ref() {
1282 Some(RawOrigin::Root) => Ok(()),
1283 _ => Err(o),
1284 }
1285 }
1286
1287 #[cfg(feature = "runtime-benchmarks")]
1288 fn try_successful_origin() -> Result<O, ()> {
1289 Ok(O::root())
1290 }
1291}
1292
1293impl_ensure_origin_with_arg_ignoring_arg! {
1294 impl< { O: .., AccountId: Decode, T } >
1295 EnsureOriginWithArg<O, T> for EnsureRoot<AccountId>
1296 {}
1297}
1298
1299pub struct EnsureRootWithSuccess<AccountId, Success>(
1301 core::marker::PhantomData<(AccountId, Success)>,
1302);
1303impl<O: OriginTrait, AccountId, Success: TypedGet> EnsureOrigin<O>
1304 for EnsureRootWithSuccess<AccountId, Success>
1305{
1306 type Success = Success::Type;
1307 fn try_origin(o: O) -> Result<Self::Success, O> {
1308 match o.as_system_ref() {
1309 Some(RawOrigin::Root) => Ok(Success::get()),
1310 _ => Err(o),
1311 }
1312 }
1313
1314 #[cfg(feature = "runtime-benchmarks")]
1315 fn try_successful_origin() -> Result<O, ()> {
1316 Ok(O::root())
1317 }
1318}
1319
1320impl_ensure_origin_with_arg_ignoring_arg! {
1321 impl< { O: .., AccountId: Decode, Success: TypedGet, T } >
1322 EnsureOriginWithArg<O, T> for EnsureRootWithSuccess<AccountId, Success>
1323 {}
1324}
1325
1326pub struct EnsureWithSuccess<Ensure, AccountId, Success>(
1328 core::marker::PhantomData<(Ensure, AccountId, Success)>,
1329);
1330
1331impl<O: OriginTrait, Ensure: EnsureOrigin<O>, AccountId, Success: TypedGet> EnsureOrigin<O>
1332 for EnsureWithSuccess<Ensure, AccountId, Success>
1333{
1334 type Success = Success::Type;
1335
1336 fn try_origin(o: O) -> Result<Self::Success, O> {
1337 Ensure::try_origin(o).map(|_| Success::get())
1338 }
1339
1340 #[cfg(feature = "runtime-benchmarks")]
1341 fn try_successful_origin() -> Result<O, ()> {
1342 Ensure::try_successful_origin()
1343 }
1344}
1345
1346pub struct EnsureSigned<AccountId>(core::marker::PhantomData<AccountId>);
1348impl<O: OriginTrait<AccountId = AccountId>, AccountId: Decode + Clone> EnsureOrigin<O>
1349 for EnsureSigned<AccountId>
1350{
1351 type Success = AccountId;
1352 fn try_origin(o: O) -> Result<Self::Success, O> {
1353 match o.as_system_ref() {
1354 Some(RawOrigin::Signed(who)) => Ok(who.clone()),
1355 _ => Err(o),
1356 }
1357 }
1358
1359 #[cfg(feature = "runtime-benchmarks")]
1360 fn try_successful_origin() -> Result<O, ()> {
1361 let zero_account_id =
1362 AccountId::decode(&mut TrailingZeroInput::zeroes()).map_err(|_| ())?;
1363 Ok(O::signed(zero_account_id))
1364 }
1365}
1366
1367impl_ensure_origin_with_arg_ignoring_arg! {
1368 impl< { O: OriginTrait<AccountId = AccountId>, AccountId: Decode + Clone, T } >
1369 EnsureOriginWithArg<O, T> for EnsureSigned<AccountId>
1370 {}
1371}
1372
1373pub struct EnsureSignedBy<Who, AccountId>(core::marker::PhantomData<(Who, AccountId)>);
1375impl<
1376 O: OriginTrait<AccountId = AccountId>,
1377 Who: SortedMembers<AccountId>,
1378 AccountId: PartialEq + Clone + Ord + Decode,
1379 > EnsureOrigin<O> for EnsureSignedBy<Who, AccountId>
1380{
1381 type Success = AccountId;
1382 fn try_origin(o: O) -> Result<Self::Success, O> {
1383 match o.as_system_ref() {
1384 Some(RawOrigin::Signed(ref who)) if Who::contains(who) => Ok(who.clone()),
1385 _ => Err(o),
1386 }
1387 }
1388
1389 #[cfg(feature = "runtime-benchmarks")]
1390 fn try_successful_origin() -> Result<O, ()> {
1391 let first_member = match Who::sorted_members().first() {
1392 Some(account) => account.clone(),
1393 None => AccountId::decode(&mut TrailingZeroInput::zeroes()).map_err(|_| ())?,
1394 };
1395 Ok(O::signed(first_member))
1396 }
1397}
1398
1399impl_ensure_origin_with_arg_ignoring_arg! {
1400 impl< { O: OriginTrait<AccountId = AccountId>, Who: SortedMembers<AccountId>, AccountId: PartialEq + Clone + Ord + Decode, T } >
1401 EnsureOriginWithArg<O, T> for EnsureSignedBy<Who, AccountId>
1402 {}
1403}
1404
1405pub struct EnsureNone<AccountId>(core::marker::PhantomData<AccountId>);
1407impl<O: OriginTrait<AccountId = AccountId>, AccountId> EnsureOrigin<O> for EnsureNone<AccountId> {
1408 type Success = ();
1409 fn try_origin(o: O) -> Result<Self::Success, O> {
1410 match o.as_system_ref() {
1411 Some(RawOrigin::None) => Ok(()),
1412 _ => Err(o),
1413 }
1414 }
1415
1416 #[cfg(feature = "runtime-benchmarks")]
1417 fn try_successful_origin() -> Result<O, ()> {
1418 Ok(O::none())
1419 }
1420}
1421
1422impl_ensure_origin_with_arg_ignoring_arg! {
1423 impl< { O: OriginTrait<AccountId = AccountId>, AccountId, T } >
1424 EnsureOriginWithArg<O, T> for EnsureNone<AccountId>
1425 {}
1426}
1427
1428pub struct EnsureNever<Success>(core::marker::PhantomData<Success>);
1430impl<O, Success> EnsureOrigin<O> for EnsureNever<Success> {
1431 type Success = Success;
1432 fn try_origin(o: O) -> Result<Self::Success, O> {
1433 Err(o)
1434 }
1435
1436 #[cfg(feature = "runtime-benchmarks")]
1437 fn try_successful_origin() -> Result<O, ()> {
1438 Err(())
1439 }
1440}
1441
1442impl_ensure_origin_with_arg_ignoring_arg! {
1443 impl< { O, Success, T } >
1444 EnsureOriginWithArg<O, T> for EnsureNever<Success>
1445 {}
1446}
1447
1448#[docify::export]
1449pub fn ensure_signed<OuterOrigin, AccountId>(o: OuterOrigin) -> Result<AccountId, BadOrigin>
1452where
1453 OuterOrigin: Into<Result<RawOrigin<AccountId>, OuterOrigin>>,
1454{
1455 match o.into() {
1456 Ok(RawOrigin::Signed(t)) => Ok(t),
1457 _ => Err(BadOrigin),
1458 }
1459}
1460
1461pub fn ensure_signed_or_root<OuterOrigin, AccountId>(
1465 o: OuterOrigin,
1466) -> Result<Option<AccountId>, BadOrigin>
1467where
1468 OuterOrigin: Into<Result<RawOrigin<AccountId>, OuterOrigin>>,
1469{
1470 match o.into() {
1471 Ok(RawOrigin::Root) => Ok(None),
1472 Ok(RawOrigin::Signed(t)) => Ok(Some(t)),
1473 _ => Err(BadOrigin),
1474 }
1475}
1476
1477pub fn ensure_root<OuterOrigin, AccountId>(o: OuterOrigin) -> Result<(), BadOrigin>
1479where
1480 OuterOrigin: Into<Result<RawOrigin<AccountId>, OuterOrigin>>,
1481{
1482 match o.into() {
1483 Ok(RawOrigin::Root) => Ok(()),
1484 _ => Err(BadOrigin),
1485 }
1486}
1487
1488pub fn ensure_none<OuterOrigin, AccountId>(o: OuterOrigin) -> Result<(), BadOrigin>
1490where
1491 OuterOrigin: Into<Result<RawOrigin<AccountId>, OuterOrigin>>,
1492{
1493 match o.into() {
1494 Ok(RawOrigin::None) => Ok(()),
1495 _ => Err(BadOrigin),
1496 }
1497}
1498
1499pub fn ensure_authorized<OuterOrigin, AccountId>(o: OuterOrigin) -> Result<(), BadOrigin>
1502where
1503 OuterOrigin: Into<Result<RawOrigin<AccountId>, OuterOrigin>>,
1504{
1505 match o.into() {
1506 Ok(RawOrigin::Authorized) => Ok(()),
1507 _ => Err(BadOrigin),
1508 }
1509}
1510
1511#[derive(Debug)]
1513pub enum RefStatus {
1514 Referenced,
1515 Unreferenced,
1516}
1517
1518#[derive(Eq, PartialEq, Debug)]
1520pub enum IncRefStatus {
1521 Created,
1523 Existed,
1525}
1526
1527#[derive(Eq, PartialEq, Debug)]
1529pub enum DecRefStatus {
1530 Reaped,
1532 Exists,
1534}
1535
1536pub enum CanSetCodeResult<T: Config> {
1538 Ok,
1540 MultiBlockMigrationsOngoing,
1542 InvalidVersion(Error<T>),
1544}
1545
1546impl<T: Config> CanSetCodeResult<T> {
1547 pub fn into_result(self) -> Result<(), DispatchError> {
1549 match self {
1550 Self::Ok => Ok(()),
1551 Self::MultiBlockMigrationsOngoing => {
1552 Err(Error::<T>::MultiBlockMigrationsOngoing.into())
1553 },
1554 Self::InvalidVersion(err) => Err(err.into()),
1555 }
1556 }
1557
1558 pub fn is_ok(&self) -> bool {
1560 matches!(self, Self::Ok)
1561 }
1562}
1563
1564impl<T: Config> Pallet<T> {
1565 #[doc = docify::embed!("src/tests.rs", last_runtime_upgrade_spec_version_usage)]
1579 pub fn last_runtime_upgrade_spec_version() -> u32 {
1580 LastRuntimeUpgrade::<T>::get().map_or(0, |l| l.spec_version.0)
1581 }
1582
1583 pub fn account_exists(who: &T::AccountId) -> bool {
1585 Account::<T>::contains_key(who)
1586 }
1587
1588 pub fn update_code_in_storage(code: &[u8]) {
1595 match T::Version::get().system_version {
1596 0..=2 => {
1597 storage::unhashed::put_raw(well_known_keys::CODE, code);
1598 },
1599 _ => {
1600 BlocksTillUpgrade::<T>::put(2u8);
1601 storage::unhashed::put_raw(well_known_keys::PENDING_CODE, code);
1602 },
1603 }
1604 Self::deposit_log(generic::DigestItem::RuntimeEnvironmentUpdated);
1605 Self::deposit_event(Event::CodeUpdated);
1606 }
1607
1608 pub fn maybe_apply_pending_code_upgrade() {
1613 let Some(remaining) = BlocksTillUpgrade::<T>::get() else { return };
1614
1615 let remaining = remaining.saturating_sub(1);
1616
1617 if remaining > 0 {
1618 BlocksTillUpgrade::<T>::put(remaining);
1619 return;
1620 }
1621
1622 BlocksTillUpgrade::<T>::kill();
1623
1624 let Some(new_code) = storage::unhashed::get_raw(well_known_keys::PENDING_CODE) else {
1625 defensive!("BlocksTillUpgrade is set but no pending code found");
1627 return;
1628 };
1629
1630 storage::unhashed::put_raw(well_known_keys::CODE, &new_code);
1631 storage::unhashed::kill(well_known_keys::PENDING_CODE);
1632 }
1633
1634 pub fn inherents_applied() -> bool {
1636 InherentsApplied::<T>::get()
1637 }
1638
1639 pub fn note_inherents_applied() {
1644 InherentsApplied::<T>::put(true);
1645 }
1646
1647 #[deprecated = "Use `inc_consumers` instead"]
1649 pub fn inc_ref(who: &T::AccountId) {
1650 let _ = Self::inc_consumers(who);
1651 }
1652
1653 #[deprecated = "Use `dec_consumers` instead"]
1656 pub fn dec_ref(who: &T::AccountId) {
1657 let _ = Self::dec_consumers(who);
1658 }
1659
1660 #[deprecated = "Use `consumers` instead"]
1662 pub fn refs(who: &T::AccountId) -> RefCount {
1663 Self::consumers(who)
1664 }
1665
1666 #[deprecated = "Use `!is_provider_required` instead"]
1668 pub fn allow_death(who: &T::AccountId) -> bool {
1669 !Self::is_provider_required(who)
1670 }
1671
1672 pub fn inc_providers(who: &T::AccountId) -> IncRefStatus {
1674 Account::<T>::mutate(who, |a| {
1675 if a.providers == 0 && a.sufficients == 0 {
1676 a.providers = 1;
1678 Self::on_created_account(who.clone(), a);
1679 IncRefStatus::Created
1680 } else {
1681 a.providers = a.providers.saturating_add(1);
1682 IncRefStatus::Existed
1683 }
1684 })
1685 }
1686
1687 pub fn dec_providers(who: &T::AccountId) -> Result<DecRefStatus, DispatchError> {
1691 Account::<T>::try_mutate_exists(who, |maybe_account| {
1692 if let Some(mut account) = maybe_account.take() {
1693 if account.providers == 0 {
1694 log::error!(
1696 target: LOG_TARGET,
1697 "Logic error: Unexpected underflow in reducing provider",
1698 );
1699 account.providers = 1;
1700 }
1701 match (account.providers, account.consumers, account.sufficients) {
1702 (1, 0, 0) => {
1703 Pallet::<T>::on_killed_account(who.clone());
1706 Ok(DecRefStatus::Reaped)
1707 },
1708 (1, c, _) if c > 0 => {
1709 Err(DispatchError::ConsumerRemaining)
1711 },
1712 (x, _, _) => {
1713 account.providers = x - 1;
1716 *maybe_account = Some(account);
1717 Ok(DecRefStatus::Exists)
1718 },
1719 }
1720 } else {
1721 log::error!(
1722 target: LOG_TARGET,
1723 "Logic error: Account already dead when reducing provider",
1724 );
1725 Ok(DecRefStatus::Reaped)
1726 }
1727 })
1728 }
1729
1730 pub fn inc_sufficients(who: &T::AccountId) -> IncRefStatus {
1732 Account::<T>::mutate(who, |a| {
1733 if a.providers + a.sufficients == 0 {
1734 a.sufficients = 1;
1736 Self::on_created_account(who.clone(), a);
1737 IncRefStatus::Created
1738 } else {
1739 a.sufficients = a.sufficients.saturating_add(1);
1740 IncRefStatus::Existed
1741 }
1742 })
1743 }
1744
1745 pub fn dec_sufficients(who: &T::AccountId) -> DecRefStatus {
1749 Account::<T>::mutate_exists(who, |maybe_account| {
1750 if let Some(mut account) = maybe_account.take() {
1751 if account.sufficients == 0 {
1752 log::error!(
1754 target: LOG_TARGET,
1755 "Logic error: Unexpected underflow in reducing sufficients",
1756 );
1757 }
1758 match (account.sufficients, account.providers) {
1759 (0, 0) | (1, 0) => {
1760 Pallet::<T>::on_killed_account(who.clone());
1761 DecRefStatus::Reaped
1762 },
1763 (x, _) => {
1764 account.sufficients = x.saturating_sub(1);
1765 *maybe_account = Some(account);
1766 DecRefStatus::Exists
1767 },
1768 }
1769 } else {
1770 log::error!(
1771 target: LOG_TARGET,
1772 "Logic error: Account already dead when reducing provider",
1773 );
1774 DecRefStatus::Reaped
1775 }
1776 })
1777 }
1778
1779 pub fn providers(who: &T::AccountId) -> RefCount {
1781 Account::<T>::get(who).providers
1782 }
1783
1784 pub fn sufficients(who: &T::AccountId) -> RefCount {
1786 Account::<T>::get(who).sufficients
1787 }
1788
1789 pub fn reference_count(who: &T::AccountId) -> RefCount {
1791 let a = Account::<T>::get(who);
1792 a.providers + a.sufficients
1793 }
1794
1795 pub fn inc_consumers(who: &T::AccountId) -> Result<(), DispatchError> {
1800 Account::<T>::try_mutate(who, |a| {
1801 if a.providers > 0 {
1802 if a.consumers < T::MaxConsumers::max_consumers() {
1803 a.consumers = a.consumers.saturating_add(1);
1804 Ok(())
1805 } else {
1806 Err(DispatchError::TooManyConsumers)
1807 }
1808 } else {
1809 Err(DispatchError::NoProviders)
1810 }
1811 })
1812 }
1813
1814 pub fn inc_consumers_without_limit(who: &T::AccountId) -> Result<(), DispatchError> {
1818 Account::<T>::try_mutate(who, |a| {
1819 if a.providers > 0 {
1820 a.consumers = a.consumers.saturating_add(1);
1821 Ok(())
1822 } else {
1823 Err(DispatchError::NoProviders)
1824 }
1825 })
1826 }
1827
1828 pub fn dec_consumers(who: &T::AccountId) {
1831 Account::<T>::mutate(who, |a| {
1832 if a.consumers > 0 {
1833 a.consumers -= 1;
1834 } else {
1835 log::error!(
1836 target: LOG_TARGET,
1837 "Logic error: Unexpected underflow in reducing consumer",
1838 );
1839 }
1840 })
1841 }
1842
1843 pub fn consumers(who: &T::AccountId) -> RefCount {
1845 Account::<T>::get(who).consumers
1846 }
1847
1848 pub fn is_provider_required(who: &T::AccountId) -> bool {
1850 Account::<T>::get(who).consumers != 0
1851 }
1852
1853 pub fn can_dec_provider(who: &T::AccountId) -> bool {
1855 let a = Account::<T>::get(who);
1856 a.consumers == 0 || a.providers > 1
1857 }
1858
1859 pub fn can_accrue_consumers(who: &T::AccountId, amount: u32) -> bool {
1862 let a = Account::<T>::get(who);
1863 match a.consumers.checked_add(amount) {
1864 Some(c) => a.providers > 0 && c <= T::MaxConsumers::max_consumers(),
1865 None => false,
1866 }
1867 }
1868
1869 pub fn can_inc_consumer(who: &T::AccountId) -> bool {
1872 Self::can_accrue_consumers(who, 1)
1873 }
1874
1875 pub fn deposit_event(event: impl Into<T::RuntimeEvent>) {
1879 Self::deposit_event_indexed(&[], event.into());
1880 }
1881
1882 pub fn deposit_event_indexed(topics: &[T::Hash], event: T::RuntimeEvent) {
1890 let block_number = Self::block_number();
1891
1892 if block_number.is_zero() {
1894 return;
1895 }
1896
1897 let phase = ExecutionPhase::<T>::get().unwrap_or_default();
1898 let event = EventRecord { phase, event, topics: topics.to_vec() };
1899
1900 let event_idx = {
1902 let old_event_count = EventCount::<T>::get();
1903 let new_event_count = match old_event_count.checked_add(1) {
1904 None => return,
1907 Some(nc) => nc,
1908 };
1909 EventCount::<T>::put(new_event_count);
1910 old_event_count
1911 };
1912
1913 Events::<T>::append(event);
1914
1915 for topic in topics {
1916 <EventTopics<T>>::append(topic, &(block_number, event_idx));
1917 }
1918 }
1919
1920 pub fn extrinsic_index() -> Option<u32> {
1922 storage::unhashed::get(well_known_keys::EXTRINSIC_INDEX)
1923 }
1924
1925 pub fn extrinsic_count() -> u32 {
1927 ExtrinsicCount::<T>::get().unwrap_or_default()
1928 }
1929
1930 pub fn block_size() -> u32 {
1932 BlockSize::<T>::get().unwrap_or_default()
1933 }
1934
1935 pub fn execution_phase() -> Option<Phase> {
1937 ExecutionPhase::<T>::get()
1938 }
1939
1940 pub fn register_extra_weight_unchecked(weight: Weight, class: DispatchClass) {
1956 BlockWeight::<T>::mutate(|current_weight| {
1957 current_weight.accrue(weight, class);
1958 });
1959 }
1960
1961 pub fn initialize(number: &BlockNumberFor<T>, parent_hash: &T::Hash, digest: &generic::Digest) {
1968 let expected_block_number = Self::block_number() + One::one();
1969 assert_eq!(expected_block_number, *number, "Block number must be strictly increasing.");
1970
1971 ExecutionPhase::<T>::put(Phase::Initialization);
1973 storage::unhashed::put(well_known_keys::EXTRINSIC_INDEX, &0u32);
1974 Self::initialize_intra_block_entropy(parent_hash);
1975 <Number<T>>::put(number);
1976 <Digest<T>>::put(digest);
1977 <ParentHash<T>>::put(parent_hash);
1978 <BlockHash<T>>::insert(*number - One::one(), parent_hash);
1979
1980 BlockWeight::<T>::kill();
1982
1983 let digest_size = digest.encoded_size();
1986 let empty_header = <<T as Config>::Block as traits::Block>::Header::new(
1987 *number,
1988 Default::default(),
1989 Default::default(),
1990 *parent_hash,
1991 Default::default(),
1992 );
1993 let empty_header_size = empty_header.encoded_size();
1994 let overhead = digest_size.saturating_add(empty_header_size) as u32;
1995 BlockSize::<T>::put(overhead);
1996
1997 let max_total_header = T::BlockLength::get().max_header_size();
1999 assert!(
2000 overhead <= max_total_header as u32,
2001 "Header size ({overhead}) exceeds max header size limit ({max_total_header})"
2002 );
2003 }
2004
2005 pub fn initialize_intra_block_entropy(parent_hash: &T::Hash) {
2009 let entropy = (b"frame_system::initialize", parent_hash).using_encoded(blake2_256);
2010 storage::unhashed::put_raw(well_known_keys::INTRABLOCK_ENTROPY, &entropy[..]);
2011 }
2012
2013 pub fn resource_usage_report() {
2017 log::debug!(
2018 target: LOG_TARGET,
2019 "[{:?}] {} extrinsics, block size: {} (normal {}%, op: {}%, mandatory {}%) / normal weight:\
2020 {} (ref_time: {}%, proof_size: {}%) op weight {} (ref_time {}%, proof_size {}%) / \
2021 mandatory weight {} (ref_time: {}%, proof_size: {}%)",
2022 Self::block_number(),
2023 Self::extrinsic_count(),
2024 Self::block_size(),
2025 sp_runtime::Percent::from_rational(
2026 Self::block_size(),
2027 *T::BlockLength::get().max.get(DispatchClass::Normal)
2028 ).deconstruct(),
2029 sp_runtime::Percent::from_rational(
2030 Self::block_size(),
2031 *T::BlockLength::get().max.get(DispatchClass::Operational)
2032 ).deconstruct(),
2033 sp_runtime::Percent::from_rational(
2034 Self::block_size(),
2035 *T::BlockLength::get().max.get(DispatchClass::Mandatory)
2036 ).deconstruct(),
2037 Self::block_weight().get(DispatchClass::Normal),
2038 sp_runtime::Percent::from_rational(
2039 Self::block_weight().get(DispatchClass::Normal).ref_time(),
2040 T::BlockWeights::get().get(DispatchClass::Normal).max_total.unwrap_or(Bounded::max_value()).ref_time()
2041 ).deconstruct(),
2042 sp_runtime::Percent::from_rational(
2043 Self::block_weight().get(DispatchClass::Normal).proof_size(),
2044 T::BlockWeights::get().get(DispatchClass::Normal).max_total.unwrap_or(Bounded::max_value()).proof_size()
2045 ).deconstruct(),
2046 Self::block_weight().get(DispatchClass::Operational),
2047 sp_runtime::Percent::from_rational(
2048 Self::block_weight().get(DispatchClass::Operational).ref_time(),
2049 T::BlockWeights::get().get(DispatchClass::Operational).max_total.unwrap_or(Bounded::max_value()).ref_time()
2050 ).deconstruct(),
2051 sp_runtime::Percent::from_rational(
2052 Self::block_weight().get(DispatchClass::Operational).proof_size(),
2053 T::BlockWeights::get().get(DispatchClass::Operational).max_total.unwrap_or(Bounded::max_value()).proof_size()
2054 ).deconstruct(),
2055 Self::block_weight().get(DispatchClass::Mandatory),
2056 sp_runtime::Percent::from_rational(
2057 Self::block_weight().get(DispatchClass::Mandatory).ref_time(),
2058 T::BlockWeights::get().get(DispatchClass::Mandatory).max_total.unwrap_or(Bounded::max_value()).ref_time()
2059 ).deconstruct(),
2060 sp_runtime::Percent::from_rational(
2061 Self::block_weight().get(DispatchClass::Mandatory).proof_size(),
2062 T::BlockWeights::get().get(DispatchClass::Mandatory).max_total.unwrap_or(Bounded::max_value()).proof_size()
2063 ).deconstruct(),
2064 );
2065 }
2066
2067 pub fn finalize() -> HeaderFor<T> {
2070 Self::resource_usage_report();
2071 ExecutionPhase::<T>::kill();
2072 BlockSize::<T>::kill();
2073 storage::unhashed::kill(well_known_keys::INTRABLOCK_ENTROPY);
2074 InherentsApplied::<T>::kill();
2075
2076 let number = <Number<T>>::get();
2087 let parent_hash = <ParentHash<T>>::get();
2088 let digest = <Digest<T>>::get();
2089
2090 let extrinsics = (0..ExtrinsicCount::<T>::take().unwrap_or_default())
2091 .map(ExtrinsicData::<T>::take)
2092 .collect();
2093 let extrinsics_root_state_version = T::Version::get().extrinsics_root_state_version();
2094 let extrinsics_root =
2095 extrinsics_data_root::<T::Hashing>(extrinsics, extrinsics_root_state_version);
2096
2097 let block_hash_count = T::BlockHashCount::get();
2099 let to_remove = number.saturating_sub(block_hash_count).saturating_sub(One::one());
2100
2101 if !to_remove.is_zero() {
2103 <BlockHash<T>>::remove(to_remove);
2104 }
2105
2106 let version = T::Version::get().state_version();
2107 let storage_root = T::Hash::decode(&mut &sp_io::storage::root(version)[..])
2108 .expect("Node is configured to use the same hash; qed");
2109
2110 HeaderFor::<T>::new(number, extrinsics_root, storage_root, parent_hash, digest)
2111 }
2112
2113 pub fn deposit_log(item: generic::DigestItem) {
2118 BlockSize::<T>::mutate(|len| {
2119 *len = Some(len.unwrap_or(0).saturating_add(item.encoded_size() as u32));
2120 });
2121 <Digest<T>>::append(item);
2122 }
2123
2124 #[cfg(any(feature = "std", test))]
2126 pub fn externalities() -> TestExternalities {
2127 TestExternalities::new(sp_core::storage::Storage {
2128 top: [
2129 (<BlockHash<T>>::hashed_key_for(BlockNumberFor::<T>::zero()), [69u8; 32].encode()),
2130 (<Number<T>>::hashed_key().to_vec(), BlockNumberFor::<T>::one().encode()),
2131 (<ParentHash<T>>::hashed_key().to_vec(), [69u8; 32].encode()),
2132 ]
2133 .into_iter()
2134 .collect(),
2135 children_default: Default::default(),
2136 })
2137 }
2138
2139 #[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2147 pub fn events() -> Vec<EventRecord<T::RuntimeEvent, T::Hash>> {
2148 Self::read_events_no_consensus().map(|e| *e).collect()
2150 }
2151
2152 pub fn event_no_consensus(index: usize) -> Option<T::RuntimeEvent> {
2157 Self::read_events_no_consensus().nth(index).map(|e| e.event.clone())
2158 }
2159
2160 pub fn read_events_no_consensus(
2165 ) -> impl Iterator<Item = Box<EventRecord<T::RuntimeEvent, T::Hash>>> {
2166 Events::<T>::stream_iter()
2167 }
2168
2169 pub fn read_events_for_pallet<E>() -> Vec<E>
2174 where
2175 T::RuntimeEvent: TryInto<E>,
2176 {
2177 Events::<T>::get()
2178 .into_iter()
2179 .map(|er| er.event)
2180 .filter_map(|e| e.try_into().ok())
2181 .collect::<_>()
2182 }
2183
2184 #[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2193 pub fn run_to_block_with<AllPalletsWithSystem>(
2194 n: BlockNumberFor<T>,
2195 mut hooks: RunToBlockHooks<T>,
2196 ) where
2197 AllPalletsWithSystem: frame_support::traits::OnInitialize<BlockNumberFor<T>>
2198 + frame_support::traits::OnFinalize<BlockNumberFor<T>>,
2199 {
2200 let mut bn = Self::block_number();
2201
2202 while bn < n {
2203 if !bn.is_zero() {
2205 (hooks.before_finalize)(bn);
2206 AllPalletsWithSystem::on_finalize(bn);
2207 (hooks.after_finalize)(bn);
2208 }
2209
2210 bn += One::one();
2211
2212 Self::set_block_number(bn);
2213 (hooks.before_initialize)(bn);
2214 AllPalletsWithSystem::on_initialize(bn);
2215 (hooks.after_initialize)(bn);
2216 }
2217 }
2218
2219 #[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2221 pub fn run_to_block<AllPalletsWithSystem>(n: BlockNumberFor<T>)
2222 where
2223 AllPalletsWithSystem: frame_support::traits::OnInitialize<BlockNumberFor<T>>
2224 + frame_support::traits::OnFinalize<BlockNumberFor<T>>,
2225 {
2226 Self::run_to_block_with::<AllPalletsWithSystem>(n, Default::default());
2227 }
2228
2229 #[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2232 pub fn set_block_number(n: BlockNumberFor<T>) {
2233 <Number<T>>::put(n);
2234 }
2235
2236 #[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2238 pub fn set_extrinsic_index(extrinsic_index: u32) {
2239 storage::unhashed::put(well_known_keys::EXTRINSIC_INDEX, &extrinsic_index)
2240 }
2241
2242 #[cfg(any(feature = "std", test))]
2245 pub fn set_parent_hash(n: T::Hash) {
2246 <ParentHash<T>>::put(n);
2247 }
2248
2249 #[cfg(any(feature = "std", test))]
2251 pub fn set_block_consumed_resources(weight: Weight, len: usize) {
2252 BlockWeight::<T>::mutate(|current_weight| {
2253 current_weight.set(weight, DispatchClass::Normal)
2254 });
2255 BlockSize::<T>::put(len as u32);
2256 }
2257
2258 pub fn reset_events() {
2263 <Events<T>>::kill();
2264 EventCount::<T>::kill();
2265 let _ = <EventTopics<T>>::clear(u32::max_value(), None);
2266 }
2267
2268 #[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2272 #[track_caller]
2273 pub fn assert_has_event(event: T::RuntimeEvent) {
2274 let warn = if Self::block_number().is_zero() {
2275 "WARNING: block number is zero, and events are not registered at block number zero.\n"
2276 } else {
2277 ""
2278 };
2279
2280 let events = Self::events();
2281 assert!(
2282 events.iter().any(|record| record.event == event),
2283 "{warn}expected event {event:?} not found in events {events:?}",
2284 );
2285 }
2286
2287 #[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2291 #[track_caller]
2292 pub fn assert_last_event(event: T::RuntimeEvent) {
2293 let warn = if Self::block_number().is_zero() {
2294 "WARNING: block number is zero, and events are not registered at block number zero.\n"
2295 } else {
2296 ""
2297 };
2298
2299 let last_event = Self::events()
2300 .last()
2301 .expect(&alloc::format!("{warn}events expected"))
2302 .event
2303 .clone();
2304 assert_eq!(
2305 last_event, event,
2306 "{warn}expected event {event:?} is not equal to the last event {last_event:?}",
2307 );
2308 }
2309
2310 pub fn runtime_version() -> RuntimeVersion {
2312 T::Version::get()
2313 }
2314
2315 pub fn account_nonce(who: impl EncodeLike<T::AccountId>) -> T::Nonce {
2317 Account::<T>::get(who).nonce
2318 }
2319
2320 pub fn inc_account_nonce(who: impl EncodeLike<T::AccountId>) {
2322 Account::<T>::mutate(who, |a| a.nonce += T::Nonce::one());
2323 }
2324
2325 pub fn note_extrinsic(encoded_xt: Vec<u8>) {
2330 ExtrinsicData::<T>::insert(Self::extrinsic_index().unwrap_or_default(), encoded_xt);
2331 }
2332
2333 pub fn note_applied_extrinsic(r: &DispatchResultWithPostInfo, info: DispatchInfo) {
2339 let weight = extract_actual_weight(r, &info)
2340 .saturating_add(T::BlockWeights::get().get(info.class).base_extrinsic);
2341 let class = info.class;
2342 let pays_fee = extract_actual_pays_fee(r, &info);
2343 let dispatch_event_info = DispatchEventInfo { weight, class, pays_fee };
2344
2345 Self::deposit_event(match r {
2346 Ok(_) => Event::ExtrinsicSuccess { dispatch_info: dispatch_event_info },
2347 Err(err) => {
2348 log::trace!(
2349 target: LOG_TARGET,
2350 "Extrinsic failed at block({:?}): {:?}",
2351 Self::block_number(),
2352 err,
2353 );
2354 Event::ExtrinsicFailed {
2355 dispatch_error: err.error,
2356 dispatch_info: dispatch_event_info,
2357 }
2358 },
2359 });
2360
2361 log::trace!(
2362 target: LOG_TARGET,
2363 "Used block weight: {:?}",
2364 BlockWeight::<T>::get(),
2365 );
2366
2367 log::trace!(
2368 target: LOG_TARGET,
2369 "Used block length: {:?}",
2370 Pallet::<T>::block_size(),
2371 );
2372
2373 let next_extrinsic_index = Self::extrinsic_index().unwrap_or_default() + 1u32;
2374
2375 storage::unhashed::put(well_known_keys::EXTRINSIC_INDEX, &next_extrinsic_index);
2376 ExecutionPhase::<T>::put(Phase::ApplyExtrinsic(next_extrinsic_index));
2377 ExtrinsicWeightReclaimed::<T>::kill();
2378 }
2379
2380 pub fn note_finished_extrinsics() {
2383 let extrinsic_index: u32 =
2384 storage::unhashed::take(well_known_keys::EXTRINSIC_INDEX).unwrap_or_default();
2385 ExtrinsicCount::<T>::put(extrinsic_index);
2386 ExecutionPhase::<T>::put(Phase::Finalization);
2387 }
2388
2389 pub fn note_finished_initialize() {
2392 ExecutionPhase::<T>::put(Phase::ApplyExtrinsic(0))
2393 }
2394
2395 pub fn on_created_account(who: T::AccountId, _a: &mut AccountInfo<T::Nonce, T::AccountData>) {
2397 T::OnNewAccount::on_new_account(&who);
2398 Self::deposit_event(Event::NewAccount { account: who });
2399 }
2400
2401 fn on_killed_account(who: T::AccountId) {
2403 T::OnKilledAccount::on_killed_account(&who);
2404 Self::deposit_event(Event::KilledAccount { account: who });
2405 }
2406
2407 pub fn can_set_code(code: &[u8], check_version: bool) -> CanSetCodeResult<T> {
2411 if T::MultiBlockMigrator::ongoing() {
2412 return CanSetCodeResult::MultiBlockMigrationsOngoing;
2413 }
2414
2415 if check_version {
2416 let current_version = T::Version::get();
2417 let Some(new_version) = sp_io::misc::runtime_version(code)
2418 .and_then(|v| RuntimeVersion::decode(&mut &v[..]).ok())
2419 else {
2420 return CanSetCodeResult::InvalidVersion(Error::<T>::FailedToExtractRuntimeVersion);
2421 };
2422
2423 cfg_if::cfg_if! {
2424 if #[cfg(all(feature = "runtime-benchmarks", not(test)))] {
2425 core::hint::black_box((new_version, current_version));
2427 } else {
2428 if new_version.spec_name != current_version.spec_name {
2429 return CanSetCodeResult::InvalidVersion(Error::<T>::InvalidSpecName)
2430 }
2431
2432 if new_version.spec_version <= current_version.spec_version {
2433 return CanSetCodeResult::InvalidVersion(Error::<T>::SpecVersionNeedsToIncrease)
2434 }
2435 }
2436 }
2437 }
2438
2439 CanSetCodeResult::Ok
2440 }
2441
2442 pub fn do_authorize_upgrade(code_hash: T::Hash, check_version: bool) {
2444 AuthorizedUpgrade::<T>::put(CodeUpgradeAuthorization { code_hash, check_version });
2445 Self::deposit_event(Event::UpgradeAuthorized { code_hash, check_version });
2446 }
2447
2448 fn validate_code_is_authorized(
2452 code: &[u8],
2453 ) -> Result<CodeUpgradeAuthorization<T>, DispatchError> {
2454 let authorization = AuthorizedUpgrade::<T>::get().ok_or(Error::<T>::NothingAuthorized)?;
2455 let actual_hash = T::Hashing::hash(code);
2456 ensure!(actual_hash == authorization.code_hash, Error::<T>::Unauthorized);
2457 Ok(authorization)
2458 }
2459
2460 pub fn reclaim_weight(
2465 info: &DispatchInfoOf<T::RuntimeCall>,
2466 post_info: &PostDispatchInfoOf<T::RuntimeCall>,
2467 ) -> Result<(), TransactionValidityError>
2468 where
2469 T::RuntimeCall: Dispatchable<Info = DispatchInfo, PostInfo = PostDispatchInfo>,
2470 {
2471 let already_reclaimed = crate::ExtrinsicWeightReclaimed::<T>::get();
2472 let unspent = post_info.calc_unspent(info);
2473 let accurate_reclaim = already_reclaimed.max(unspent);
2474 let to_reclaim_more = accurate_reclaim.saturating_sub(already_reclaimed);
2476 if to_reclaim_more != Weight::zero() {
2477 crate::BlockWeight::<T>::mutate(|current_weight| {
2478 current_weight.reduce(to_reclaim_more, info.class);
2479 });
2480 crate::ExtrinsicWeightReclaimed::<T>::put(accurate_reclaim);
2481 }
2482
2483 Ok(())
2484 }
2485
2486 pub fn remaining_block_weight() -> WeightMeter {
2488 let limit = T::BlockWeights::get().max_block;
2489 let consumed = BlockWeight::<T>::get().total();
2490
2491 WeightMeter::with_consumed_and_limit(consumed, limit)
2492 }
2493}
2494
2495pub fn unique(entropy: impl Encode) -> [u8; 32] {
2498 let mut last = [0u8; 32];
2499 sp_io::storage::read(well_known_keys::INTRABLOCK_ENTROPY, &mut last[..], 0);
2500 let next = (b"frame_system::unique", entropy, last).using_encoded(blake2_256);
2501 sp_io::storage::set(well_known_keys::INTRABLOCK_ENTROPY, &next);
2502 next
2503}
2504
2505pub struct Provider<T>(PhantomData<T>);
2507impl<T: Config> HandleLifetime<T::AccountId> for Provider<T> {
2508 fn created(t: &T::AccountId) -> Result<(), DispatchError> {
2509 Pallet::<T>::inc_providers(t);
2510 Ok(())
2511 }
2512 fn killed(t: &T::AccountId) -> Result<(), DispatchError> {
2513 Pallet::<T>::dec_providers(t).map(|_| ())
2514 }
2515}
2516
2517pub struct SelfSufficient<T>(PhantomData<T>);
2519impl<T: Config> HandleLifetime<T::AccountId> for SelfSufficient<T> {
2520 fn created(t: &T::AccountId) -> Result<(), DispatchError> {
2521 Pallet::<T>::inc_sufficients(t);
2522 Ok(())
2523 }
2524 fn killed(t: &T::AccountId) -> Result<(), DispatchError> {
2525 Pallet::<T>::dec_sufficients(t);
2526 Ok(())
2527 }
2528}
2529
2530pub struct Consumer<T>(PhantomData<T>);
2532impl<T: Config> HandleLifetime<T::AccountId> for Consumer<T> {
2533 fn created(t: &T::AccountId) -> Result<(), DispatchError> {
2534 Pallet::<T>::inc_consumers(t)
2535 }
2536 fn killed(t: &T::AccountId) -> Result<(), DispatchError> {
2537 Pallet::<T>::dec_consumers(t);
2538 Ok(())
2539 }
2540}
2541
2542impl<T: Config> BlockNumberProvider for Pallet<T> {
2543 type BlockNumber = BlockNumberFor<T>;
2544
2545 fn current_block_number() -> Self::BlockNumber {
2546 Pallet::<T>::block_number()
2547 }
2548
2549 #[cfg(feature = "runtime-benchmarks")]
2550 fn set_block_number(n: BlockNumberFor<T>) {
2551 Self::set_block_number(n)
2552 }
2553}
2554
2555impl<T: Config> StoredMap<T::AccountId, T::AccountData> for Pallet<T> {
2561 fn get(k: &T::AccountId) -> T::AccountData {
2562 Account::<T>::get(k).data
2563 }
2564
2565 fn try_mutate_exists<R, E: From<DispatchError>>(
2566 k: &T::AccountId,
2567 f: impl FnOnce(&mut Option<T::AccountData>) -> Result<R, E>,
2568 ) -> Result<R, E> {
2569 let account = Account::<T>::get(k);
2570 let is_default = account.data == T::AccountData::default();
2571 let mut some_data = if is_default { None } else { Some(account.data) };
2572 let result = f(&mut some_data)?;
2573 if Self::providers(k) > 0 || Self::sufficients(k) > 0 {
2574 Account::<T>::mutate(k, |a| a.data = some_data.unwrap_or_default());
2575 } else {
2576 Account::<T>::remove(k)
2577 }
2578 Ok(result)
2579 }
2580}
2581
2582pub fn split_inner<T, R, S>(
2584 option: Option<T>,
2585 splitter: impl FnOnce(T) -> (R, S),
2586) -> (Option<R>, Option<S>) {
2587 match option {
2588 Some(inner) => {
2589 let (r, s) = splitter(inner);
2590 (Some(r), Some(s))
2591 },
2592 None => (None, None),
2593 }
2594}
2595
2596pub struct ChainContext<T>(PhantomData<T>);
2597impl<T> Default for ChainContext<T> {
2598 fn default() -> Self {
2599 ChainContext(PhantomData)
2600 }
2601}
2602
2603impl<T: Config> Lookup for ChainContext<T> {
2604 type Source = <T::Lookup as StaticLookup>::Source;
2605 type Target = <T::Lookup as StaticLookup>::Target;
2606
2607 fn lookup(&self, s: Self::Source) -> Result<Self::Target, LookupError> {
2608 <T::Lookup as StaticLookup>::lookup(s)
2609 }
2610}
2611
2612#[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2614pub struct RunToBlockHooks<'a, T>
2615where
2616 T: 'a + Config,
2617{
2618 before_initialize: Box<dyn 'a + FnMut(BlockNumberFor<T>)>,
2619 after_initialize: Box<dyn 'a + FnMut(BlockNumberFor<T>)>,
2620 before_finalize: Box<dyn 'a + FnMut(BlockNumberFor<T>)>,
2621 after_finalize: Box<dyn 'a + FnMut(BlockNumberFor<T>)>,
2622}
2623
2624#[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2625impl<'a, T> RunToBlockHooks<'a, T>
2626where
2627 T: 'a + Config,
2628{
2629 pub fn before_initialize<F>(mut self, f: F) -> Self
2631 where
2632 F: 'a + FnMut(BlockNumberFor<T>),
2633 {
2634 self.before_initialize = Box::new(f);
2635 self
2636 }
2637 pub fn after_initialize<F>(mut self, f: F) -> Self
2639 where
2640 F: 'a + FnMut(BlockNumberFor<T>),
2641 {
2642 self.after_initialize = Box::new(f);
2643 self
2644 }
2645 pub fn before_finalize<F>(mut self, f: F) -> Self
2647 where
2648 F: 'a + FnMut(BlockNumberFor<T>),
2649 {
2650 self.before_finalize = Box::new(f);
2651 self
2652 }
2653 pub fn after_finalize<F>(mut self, f: F) -> Self
2655 where
2656 F: 'a + FnMut(BlockNumberFor<T>),
2657 {
2658 self.after_finalize = Box::new(f);
2659 self
2660 }
2661}
2662
2663#[cfg(any(feature = "std", feature = "runtime-benchmarks", test))]
2664impl<'a, T> Default for RunToBlockHooks<'a, T>
2665where
2666 T: Config,
2667{
2668 fn default() -> Self {
2669 Self {
2670 before_initialize: Box::new(|_| {}),
2671 after_initialize: Box::new(|_| {}),
2672 before_finalize: Box::new(|_| {}),
2673 after_finalize: Box::new(|_| {}),
2674 }
2675 }
2676}
2677
2678pub mod pallet_prelude {
2680 pub use crate::{
2681 ensure_authorized, ensure_none, ensure_root, ensure_signed, ensure_signed_or_root,
2682 };
2683
2684 pub type OriginFor<T> = <T as crate::Config>::RuntimeOrigin;
2686
2687 pub type HeaderFor<T> =
2689 <<T as crate::Config>::Block as sp_runtime::traits::HeaderProvider>::HeaderT;
2690
2691 pub type BlockNumberFor<T> = <HeaderFor<T> as sp_runtime::traits::Header>::Number;
2693
2694 pub type ExtrinsicFor<T> =
2696 <<T as crate::Config>::Block as sp_runtime::traits::Block>::Extrinsic;
2697
2698 pub type RuntimeCallFor<T> = <T as crate::Config>::RuntimeCall;
2700
2701 pub type AccountIdFor<T> = <T as crate::Config>::AccountId;
2703}