1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768#![cfg_attr(not(feature = "std"), no_std)]6970#[cfg(feature = "runtime-benchmarks")]71mod benchmarking;72#[cfg(test)]73mod mock;74#[cfg(test)]75mod tests;76pub mod weights;7778use codec::{Codec, Decode, Encode, MaxEncodedLen};79use frame_support::{80 dispatch::{81 DispatchError, DispatchResult, Dispatchable, GetDispatchInfo, Parameter, PostDispatchInfo,82 },83 traits::{84 schedule::{self, DispatchTime, LOWEST_PRIORITY},85 EnsureOrigin, Get, IsType, OriginTrait, PrivilegeCmp, StorageVersion, PreimageRecipient,86 ConstU32, UnfilteredDispatchable,87 },88 weights::Weight,89 unsigned::TransactionValidityError,90};9192use frame_system::{self as system};93use scale_info::TypeInfo;94use sp_runtime::{95 traits::{BadOrigin, One, Saturating, Zero, Hash},96 BoundedVec, RuntimeDebug, DispatchErrorWithPostInfo,97};98use sp_core::H160;99use sp_std::{borrow::Borrow, cmp::Ordering, marker::PhantomData, prelude::*};100pub use weights::WeightInfo;101102pub use pallet::*;103104105pub type PeriodicIndex = u32;106107pub type TaskAddress<BlockNumber> = (BlockNumber, u32);108109pub type EncodedCall = BoundedVec<u8, ConstU32<128>>;110111#[derive(Clone, Eq, PartialEq, Encode, Decode, RuntimeDebug, TypeInfo, MaxEncodedLen)]112#[scale_info(skip_type_params(T))]113pub enum ScheduledCall<T: Config> {114 Inline(EncodedCall),115 PreimageLookup { hash: T::Hash, unbounded_len: u32 },116}117118impl<T: Config> ScheduledCall<T> {119 pub fn new(call: <T as Config>::RuntimeCall) -> Result<Self, DispatchError> {120 let encoded = call.encode();121 let len = encoded.len();122123 match EncodedCall::try_from(encoded.clone()) {124 Ok(bounded) => Ok(Self::Inline(bounded)),125 Err(_) => {126 let hash = <T as system::Config>::Hashing::hash_of(&encoded);127 <T as Config>::Preimages::note_preimage(128 encoded129 .try_into()130 .map_err(|_| <Error<T>>::TooBigScheduledCall)?,131 );132133 Ok(Self::PreimageLookup {134 hash,135 unbounded_len: len as u32,136 })137 }138 }139 }140141 142 pub fn lookup_len(&self) -> Option<u32> {143 match self {144 Self::Inline(..) => None,145 Self::PreimageLookup { unbounded_len, .. } => Some(*unbounded_len),146 }147 }148149 150 pub fn lookup_needed(&self) -> bool {151 match self {152 Self::Inline(_) => false,153 Self::PreimageLookup { .. } => true,154 }155 }156157 fn decode(mut data: &[u8]) -> Result<<T as Config>::RuntimeCall, DispatchError> {158 <T as Config>::RuntimeCall::decode(&mut data)159 .map_err(|_| <Error<T>>::ScheduledCallCorrupted.into())160 }161}162163pub trait SchedulerPreimages<T: Config>: PreimageRecipient<T::Hash> {164 fn drop(call: &ScheduledCall<T>);165166 fn peek(167 call: &ScheduledCall<T>,168 ) -> Result<(<T as pallet::Config>::RuntimeCall, Option<u32>), DispatchError>;169170 171 172 173 fn realize(174 call: &ScheduledCall<T>,175 ) -> Result<(<T as pallet::Config>::RuntimeCall, Option<u32>), DispatchError>;176}177178impl<T: Config, PP: PreimageRecipient<T::Hash>> SchedulerPreimages<T> for PP {179 fn drop(call: &ScheduledCall<T>) {180 match call {181 ScheduledCall::Inline(_) => {}182 ScheduledCall::PreimageLookup { hash, .. } => Self::unrequest_preimage(hash),183 }184 }185186 fn peek(187 call: &ScheduledCall<T>,188 ) -> Result<(<T as pallet::Config>::RuntimeCall, Option<u32>), DispatchError> {189 match call {190 ScheduledCall::Inline(data) => Ok((ScheduledCall::<T>::decode(data)?, None)),191 ScheduledCall::PreimageLookup {192 hash,193 unbounded_len,194 } => {195 let (preimage, len) = Self::get_preimage(hash)196 .ok_or(<Error<T>>::PreimageNotFound)197 .map(|preimage| (preimage, *unbounded_len))?;198199 Ok((ScheduledCall::<T>::decode(preimage.as_slice())?, Some(len)))200 }201 }202 }203204 fn realize(205 call: &ScheduledCall<T>,206 ) -> Result<(<T as pallet::Config>::RuntimeCall, Option<u32>), DispatchError> {207 let r = Self::peek(call)?;208 Self::drop(call);209 Ok(r)210 }211}212213pub enum ScheduledEnsureOriginSuccess<AccountId> {214 Root,215 Signed(AccountId),216}217218pub type TaskName = [u8; 32];219220221#[cfg_attr(any(feature = "std", test), derive(PartialEq, Eq))]222#[derive(Clone, RuntimeDebug, Encode, Decode, MaxEncodedLen, TypeInfo)]223pub struct Scheduled<Name, Call, BlockNumber, PalletsOrigin, AccountId> {224 225 maybe_id: Option<Name>,226227 228 priority: schedule::Priority,229230 231 call: Call,232233 234 maybe_periodic: Option<schedule::Period<BlockNumber>>,235236 237 origin: PalletsOrigin,238 _phantom: PhantomData<AccountId>,239}240241pub type ScheduledOf<T> = Scheduled<242 TaskName,243 ScheduledCall<T>,244 <T as frame_system::Config>::BlockNumber,245 <T as Config>::PalletsOrigin,246 <T as frame_system::Config>::AccountId,247>;248249struct WeightCounter {250 used: Weight,251 limit: Weight,252}253254impl WeightCounter {255 fn check_accrue(&mut self, w: Weight) -> bool {256 let test = self.used.saturating_add(w);257 if test.any_gt(self.limit) {258 false259 } else {260 self.used = test;261 true262 }263 }264265 fn can_accrue(&mut self, w: Weight) -> bool {266 self.used.saturating_add(w).all_lte(self.limit)267 }268}269270pub(crate) trait MarginalWeightInfo: WeightInfo {271 fn service_task(maybe_lookup_len: Option<usize>, named: bool, periodic: bool) -> Weight {272 let base = Self::service_task_base();273 let mut total = match maybe_lookup_len {274 None => base,275 Some(l) => Self::service_task_fetched(l as u32),276 };277 if named {278 total.saturating_accrue(Self::service_task_named().saturating_sub(base));279 }280 if periodic {281 total.saturating_accrue(Self::service_task_periodic().saturating_sub(base));282 }283 total284 }285}286287impl<T: WeightInfo> MarginalWeightInfo for T {}288289#[frame_support::pallet]290pub mod pallet {291 use super::*;292 use frame_support::{dispatch::PostDispatchInfo, pallet_prelude::*};293 use system::pallet_prelude::*;294295 296 const STORAGE_VERSION: StorageVersion = StorageVersion::new(0);297298 #[pallet::pallet]299 #[pallet::generate_store(pub(super) trait Store)]300 #[pallet::storage_version(STORAGE_VERSION)]301 pub struct Pallet<T>(_);302303 #[pallet::config]304 pub trait Config: frame_system::Config {305 type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;306307 308 type RuntimeOrigin: OriginTrait<PalletsOrigin = Self::PalletsOrigin>309 + From<Self::PalletsOrigin>310 + IsType<<Self as system::Config>::RuntimeOrigin>311 + Clone;312313 314 type PalletsOrigin: From<system::RawOrigin<Self::AccountId>>315 + Codec316 + Clone317 + Eq318 + TypeInfo319 + MaxEncodedLen;320321 322 type RuntimeCall: Parameter323 + Dispatchable<324 RuntimeOrigin = <Self as Config>::RuntimeOrigin,325 PostInfo = PostDispatchInfo,326 > + UnfilteredDispatchable<RuntimeOrigin = <Self as system::Config>::RuntimeOrigin>327 + GetDispatchInfo328 + From<system::Call<Self>>;329330 331 #[pallet::constant]332 type MaximumWeight: Get<Weight>;333334 335 type ScheduleOrigin: EnsureOrigin<336 <Self as system::Config>::RuntimeOrigin,337 Success = ScheduledEnsureOriginSuccess<Self::AccountId>,338 >;339340 341 342 343 344 345 346 347 type OriginPrivilegeCmp: PrivilegeCmp<Self::PalletsOrigin>;348349 350 #[pallet::constant]351 type MaxScheduledPerBlock: Get<u32>;352353 354 type WeightInfo: WeightInfo;355356 357 type Preimages: SchedulerPreimages<Self>;358359 360 type CallExecutor: DispatchCall<Self, H160>;361362 363 type PrioritySetOrigin: EnsureOrigin<<Self as system::Config>::RuntimeOrigin>;364 }365366 #[pallet::storage]367 pub type IncompleteSince<T: Config> = StorageValue<_, T::BlockNumber>;368369 370 #[pallet::storage]371 pub type Agenda<T: Config> = StorageMap<372 _,373 Twox64Concat,374 T::BlockNumber,375 BoundedVec<Option<ScheduledOf<T>>, T::MaxScheduledPerBlock>,376 ValueQuery,377 >;378379 380 #[pallet::storage]381 pub(crate) type Lookup<T: Config> =382 StorageMap<_, Twox64Concat, TaskName, TaskAddress<T::BlockNumber>>;383384 385 #[pallet::event]386 #[pallet::generate_deposit(pub(super) fn deposit_event)]387 pub enum Event<T: Config> {388 389 Scheduled { when: T::BlockNumber, index: u32 },390 391 Canceled { when: T::BlockNumber, index: u32 },392 393 Dispatched {394 task: TaskAddress<T::BlockNumber>,395 id: Option<[u8; 32]>,396 result: DispatchResult,397 },398 399 PriorityChanged {400 when: T::BlockNumber,401 index: u32,402 priority: schedule::Priority,403 },404 405 CallUnavailable {406 task: TaskAddress<T::BlockNumber>,407 id: Option<[u8; 32]>,408 },409 410 PeriodicFailed {411 task: TaskAddress<T::BlockNumber>,412 id: Option<[u8; 32]>,413 },414 415 PermanentlyOverweight {416 task: TaskAddress<T::BlockNumber>,417 id: Option<[u8; 32]>,418 },419 }420421 #[pallet::error]422 pub enum Error<T> {423 424 FailedToSchedule,425 426 AgendaIsExhausted,427 428 ScheduledCallCorrupted,429 430 PreimageNotFound,431 432 TooBigScheduledCall,433 434 NotFound,435 436 TargetBlockNumberInPast,437 438 Named,439 }440441 #[pallet::hooks]442 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {443 444 fn on_initialize(now: T::BlockNumber) -> Weight {445 let mut weight_counter = WeightCounter {446 used: Weight::zero(),447 limit: T::MaximumWeight::get(),448 };449 Self::service_agendas(&mut weight_counter, now, u32::max_value());450 weight_counter.used451 }452 }453454 #[pallet::call]455 impl<T: Config> Pallet<T> {456 457 #[pallet::weight(<T as Config>::WeightInfo::schedule(T::MaxScheduledPerBlock::get()))]458 pub fn schedule(459 origin: OriginFor<T>,460 when: T::BlockNumber,461 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,462 priority: Option<schedule::Priority>,463 call: Box<<T as Config>::RuntimeCall>,464 ) -> DispatchResult {465 T::ScheduleOrigin::ensure_origin(origin.clone())?;466467 if priority.is_some() {468 T::PrioritySetOrigin::ensure_origin(origin.clone())?;469 }470471 let origin = <T as Config>::RuntimeOrigin::from(origin);472 Self::do_schedule(473 DispatchTime::At(when),474 maybe_periodic,475 priority.unwrap_or(LOWEST_PRIORITY),476 origin.caller().clone(),477 <ScheduledCall<T>>::new(*call)?,478 )?;479 Ok(())480 }481482 483 #[pallet::weight(<T as Config>::WeightInfo::cancel(T::MaxScheduledPerBlock::get()))]484 pub fn cancel(origin: OriginFor<T>, when: T::BlockNumber, index: u32) -> DispatchResult {485 T::ScheduleOrigin::ensure_origin(origin.clone())?;486 let origin = <T as Config>::RuntimeOrigin::from(origin);487 Self::do_cancel(Some(origin.caller().clone()), (when, index))?;488 Ok(())489 }490491 492 #[pallet::weight(<T as Config>::WeightInfo::schedule_named(T::MaxScheduledPerBlock::get()))]493 pub fn schedule_named(494 origin: OriginFor<T>,495 id: TaskName,496 when: T::BlockNumber,497 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,498 priority: Option<schedule::Priority>,499 call: Box<<T as Config>::RuntimeCall>,500 ) -> DispatchResult {501 T::ScheduleOrigin::ensure_origin(origin.clone())?;502503 if priority.is_some() {504 T::PrioritySetOrigin::ensure_origin(origin.clone())?;505 }506507 let origin = <T as Config>::RuntimeOrigin::from(origin);508 Self::do_schedule_named(509 id,510 DispatchTime::At(when),511 maybe_periodic,512 priority.unwrap_or(LOWEST_PRIORITY),513 origin.caller().clone(),514 <ScheduledCall<T>>::new(*call)?,515 )?;516 Ok(())517 }518519 520 #[pallet::weight(<T as Config>::WeightInfo::cancel_named(T::MaxScheduledPerBlock::get()))]521 pub fn cancel_named(origin: OriginFor<T>, id: TaskName) -> DispatchResult {522 T::ScheduleOrigin::ensure_origin(origin.clone())?;523 let origin = <T as Config>::RuntimeOrigin::from(origin);524 Self::do_cancel_named(Some(origin.caller().clone()), id)?;525 Ok(())526 }527528 529 530 531 532 533 #[pallet::weight(<T as Config>::WeightInfo::schedule(T::MaxScheduledPerBlock::get()))]534 pub fn schedule_after(535 origin: OriginFor<T>,536 after: T::BlockNumber,537 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,538 priority: Option<schedule::Priority>,539 call: Box<<T as Config>::RuntimeCall>,540 ) -> DispatchResult {541 T::ScheduleOrigin::ensure_origin(origin.clone())?;542543 if priority.is_some() {544 T::PrioritySetOrigin::ensure_origin(origin.clone())?;545 }546547 let origin = <T as Config>::RuntimeOrigin::from(origin);548 Self::do_schedule(549 DispatchTime::After(after),550 maybe_periodic,551 priority.unwrap_or(LOWEST_PRIORITY),552 origin.caller().clone(),553 <ScheduledCall<T>>::new(*call)?,554 )?;555 Ok(())556 }557558 559 560 561 562 563 #[pallet::weight(<T as Config>::WeightInfo::schedule_named(T::MaxScheduledPerBlock::get()))]564 pub fn schedule_named_after(565 origin: OriginFor<T>,566 id: TaskName,567 after: T::BlockNumber,568 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,569 priority: Option<schedule::Priority>,570 call: Box<<T as Config>::RuntimeCall>,571 ) -> DispatchResult {572 T::ScheduleOrigin::ensure_origin(origin.clone())?;573574 if priority.is_some() {575 T::PrioritySetOrigin::ensure_origin(origin.clone())?;576 }577578 let origin = <T as Config>::RuntimeOrigin::from(origin);579 Self::do_schedule_named(580 id,581 DispatchTime::After(after),582 maybe_periodic,583 priority.unwrap_or(LOWEST_PRIORITY),584 origin.caller().clone(),585 <ScheduledCall<T>>::new(*call)?,586 )?;587 Ok(())588 }589590 #[pallet::weight(<T as Config>::WeightInfo::change_named_priority(T::MaxScheduledPerBlock::get()))]591 pub fn change_named_priority(592 origin: OriginFor<T>,593 id: TaskName,594 priority: schedule::Priority,595 ) -> DispatchResult {596 T::PrioritySetOrigin::ensure_origin(origin.clone())?;597 let origin = <T as Config>::RuntimeOrigin::from(origin);598 Self::do_change_named_priority(origin.caller().clone(), id, priority)599 }600 }601}602603impl<T: Config> Pallet<T> {604 fn resolve_time(when: DispatchTime<T::BlockNumber>) -> Result<T::BlockNumber, DispatchError> {605 let now = frame_system::Pallet::<T>::block_number();606607 let when = match when {608 DispatchTime::At(x) => x,609 610 611 DispatchTime::After(x) => now.saturating_add(x).saturating_add(One::one()),612 };613614 if when <= now {615 return Err(Error::<T>::TargetBlockNumberInPast.into());616 }617618 Ok(when)619 }620621 fn place_task(622 when: T::BlockNumber,623 what: ScheduledOf<T>,624 ) -> Result<TaskAddress<T::BlockNumber>, (DispatchError, ScheduledOf<T>)> {625 let maybe_name = what.maybe_id;626 let index = Self::push_to_agenda(when, what)?;627 let address = (when, index);628 if let Some(name) = maybe_name {629 Lookup::<T>::insert(name, address)630 }631 Self::deposit_event(Event::Scheduled {632 when: address.0,633 index: address.1,634 });635 Ok(address)636 }637638 fn push_to_agenda(639 when: T::BlockNumber,640 what: ScheduledOf<T>,641 ) -> Result<u32, (DispatchError, ScheduledOf<T>)> {642 let mut agenda = Agenda::<T>::get(when);643 let index = if (agenda.len() as u32) < T::MaxScheduledPerBlock::get() {644 645 let _ = agenda.try_push(Some(what));646 agenda.len() as u32 - 1647 } else {648 if let Some(hole_index) = agenda.iter().position(|i| i.is_none()) {649 agenda[hole_index] = Some(what);650 hole_index as u32651 } else {652 return Err((<Error<T>>::AgendaIsExhausted.into(), what));653 }654 };655 Agenda::<T>::insert(when, agenda);656 Ok(index)657 }658659 fn do_schedule(660 when: DispatchTime<T::BlockNumber>,661 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,662 priority: schedule::Priority,663 origin: T::PalletsOrigin,664 call: ScheduledCall<T>,665 ) -> Result<TaskAddress<T::BlockNumber>, DispatchError> {666 let when = Self::resolve_time(when)?;667668 669 let maybe_periodic = maybe_periodic670 .filter(|p| p.1 > 1 && !p.0.is_zero())671 672 .map(|(p, c)| (p, c - 1));673 let task = Scheduled {674 maybe_id: None,675 priority,676 call,677 maybe_periodic,678 origin,679 _phantom: PhantomData,680 };681 Self::place_task(when, task).map_err(|x| x.0)682 }683684 fn do_cancel(685 origin: Option<T::PalletsOrigin>,686 (when, index): TaskAddress<T::BlockNumber>,687 ) -> Result<(), DispatchError> {688 let scheduled = Agenda::<T>::try_mutate(when, |agenda| {689 agenda.get_mut(index as usize).map_or(690 Ok(None),691 |s| -> Result<Option<Scheduled<_, _, _, _, _>>, DispatchError> {692 if let (Some(ref o), Some(ref s)) = (origin, s.borrow()) {693 if matches!(694 T::OriginPrivilegeCmp::cmp_privilege(o, &s.origin),695 Some(Ordering::Less) | None696 ) {697 return Err(BadOrigin.into());698 }699 };700 Ok(s.take())701 },702 )703 })?;704 if let Some(s) = scheduled {705 T::Preimages::drop(&s.call);706707 if let Some(id) = s.maybe_id {708 Lookup::<T>::remove(id);709 }710 Self::deposit_event(Event::Canceled { when, index });711 Ok(())712 } else {713 return Err(Error::<T>::NotFound.into());714 }715 }716717 fn do_schedule_named(718 id: TaskName,719 when: DispatchTime<T::BlockNumber>,720 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,721 priority: schedule::Priority,722 origin: T::PalletsOrigin,723 call: ScheduledCall<T>,724 ) -> Result<TaskAddress<T::BlockNumber>, DispatchError> {725 726 if Lookup::<T>::contains_key(&id) {727 return Err(Error::<T>::FailedToSchedule.into());728 }729730 let when = Self::resolve_time(when)?;731732 733 let maybe_periodic = maybe_periodic734 .filter(|p| p.1 > 1 && !p.0.is_zero())735 736 .map(|(p, c)| (p, c - 1));737738 let task = Scheduled {739 maybe_id: Some(id),740 priority,741 call,742 maybe_periodic,743 origin,744 _phantom: Default::default(),745 };746 Self::place_task(when, task).map_err(|x| x.0)747 }748749 fn do_cancel_named(origin: Option<T::PalletsOrigin>, id: TaskName) -> DispatchResult {750 Lookup::<T>::try_mutate_exists(id, |lookup| -> DispatchResult {751 if let Some((when, index)) = lookup.take() {752 let i = index as usize;753 Agenda::<T>::try_mutate(when, |agenda| -> DispatchResult {754 if let Some(s) = agenda.get_mut(i) {755 if let (Some(ref o), Some(ref s)) = (origin, s.borrow()) {756 if matches!(757 T::OriginPrivilegeCmp::cmp_privilege(o, &s.origin),758 Some(Ordering::Less) | None759 ) {760 return Err(BadOrigin.into());761 }762 T::Preimages::drop(&s.call);763 }764 *s = None;765 }766 Ok(())767 })?;768 Self::deposit_event(Event::Canceled { when, index });769 Ok(())770 } else {771 return Err(Error::<T>::NotFound.into());772 }773 })774 }775776 fn do_change_named_priority(777 origin: T::PalletsOrigin,778 id: TaskName,779 priority: schedule::Priority,780 ) -> DispatchResult {781 match Lookup::<T>::get(id) {782 Some((when, index)) => {783 let i = index as usize;784 Agenda::<T>::try_mutate(when, |agenda| {785 if let Some(Some(s)) = agenda.get_mut(i) {786 if matches!(787 T::OriginPrivilegeCmp::cmp_privilege(&origin, &s.origin),788 Some(Ordering::Less) | None789 ) {790 return Err(BadOrigin.into());791 }792793 s.priority = priority;794 Self::deposit_event(Event::PriorityChanged {795 when,796 index,797 priority,798 });799 }800 Ok(())801 })802 }803 None => Err(Error::<T>::NotFound.into()),804 }805 }806}807808enum ServiceTaskError {809 810 Unavailable,811 812 Overweight,813}814use ServiceTaskError::*;815816817pub trait DispatchCall<T: frame_system::Config + Config, SelfContainedSignedInfo> {818 819 fn dispatch_call(820 signer: Option<T::AccountId>,821 function: <T as Config>::RuntimeCall,822 ) -> Result<823 Result<PostDispatchInfo, DispatchErrorWithPostInfo<PostDispatchInfo>>,824 TransactionValidityError,825 >;826}827828impl<T: Config> Pallet<T> {829 830 fn service_agendas(weight: &mut WeightCounter, now: T::BlockNumber, max: u32) {831 if !weight.check_accrue(T::WeightInfo::service_agendas_base()) {832 return;833 }834835 let mut incomplete_since = now + One::one();836 let mut when = IncompleteSince::<T>::take().unwrap_or(now);837 let mut executed = 0;838839 let max_items = T::MaxScheduledPerBlock::get();840 let mut count_down = max;841 let service_agenda_base_weight = T::WeightInfo::service_agenda_base(max_items);842 while count_down > 0 && when <= now && weight.can_accrue(service_agenda_base_weight) {843 if !Self::service_agenda(weight, &mut executed, now, when, u32::max_value()) {844 incomplete_since = incomplete_since.min(when);845 }846 when.saturating_inc();847 count_down.saturating_dec();848 }849 incomplete_since = incomplete_since.min(when);850 if incomplete_since <= now {851 IncompleteSince::<T>::put(incomplete_since);852 }853 }854855 856 857 fn service_agenda(858 weight: &mut WeightCounter,859 executed: &mut u32,860 now: T::BlockNumber,861 when: T::BlockNumber,862 max: u32,863 ) -> bool {864 let mut agenda = Agenda::<T>::get(when);865 let mut ordered = agenda866 .iter()867 .enumerate()868 .filter_map(|(index, maybe_item)| {869 maybe_item870 .as_ref()871 .map(|item| (index as u32, item.priority))872 })873 .collect::<Vec<_>>();874 ordered.sort_by_key(|k| k.1);875 let within_limit =876 weight.check_accrue(T::WeightInfo::service_agenda_base(ordered.len() as u32));877 debug_assert!(878 within_limit,879 "weight limit should have been checked in advance"880 );881882 883 let mut postponed = (ordered.len() as u32).saturating_sub(max);884 885 let mut dropped = 0;886887 for (agenda_index, _) in ordered.into_iter().take(max as usize) {888 let task = match agenda[agenda_index as usize].take() {889 None => continue,890 Some(t) => t,891 };892 let base_weight = T::WeightInfo::service_task(893 task.call.lookup_len().map(|x| x as usize),894 task.maybe_id.is_some(),895 task.maybe_periodic.is_some(),896 );897 if !weight.can_accrue(base_weight) {898 postponed += 1;899 break;900 }901 let result = Self::service_task(weight, now, when, agenda_index, *executed == 0, task);902 agenda[agenda_index as usize] = match result {903 Err((Unavailable, slot)) => {904 dropped += 1;905 slot906 }907 Err((Overweight, slot)) => {908 postponed += 1;909 slot910 }911 Ok(()) => {912 *executed += 1;913 None914 }915 };916 }917 if postponed > 0 || dropped > 0 {918 Agenda::<T>::insert(when, agenda);919 } else {920 Agenda::<T>::remove(when);921 }922 postponed == 0923 }924925 926 927 928 929 930 931 fn service_task(932 weight: &mut WeightCounter,933 now: T::BlockNumber,934 when: T::BlockNumber,935 agenda_index: u32,936 is_first: bool,937 mut task: ScheduledOf<T>,938 ) -> Result<(), (ServiceTaskError, Option<ScheduledOf<T>>)> {939 let (call, lookup_len) = match T::Preimages::peek(&task.call) {940 Ok(c) => c,941 Err(_) => {942 if let Some(ref id) = task.maybe_id {943 Lookup::<T>::remove(id);944 }945946 return Err((Unavailable, Some(task)));947 }948 };949950 weight.check_accrue(T::WeightInfo::service_task(951 lookup_len.map(|x| x as usize),952 task.maybe_id.is_some(),953 task.maybe_periodic.is_some(),954 ));955956 match Self::execute_dispatch(weight, task.origin.clone(), call) {957 Err(Unavailable) => {958 debug_assert!(false, "Checked to exist with `peek`");959960 if let Some(ref id) = task.maybe_id {961 Lookup::<T>::remove(id);962 }963964 Self::deposit_event(Event::CallUnavailable {965 task: (when, agenda_index),966 id: task.maybe_id,967 });968 Err((Unavailable, Some(task)))969 }970 Err(Overweight) if is_first && !Self::is_runtime_upgraded() => {971 T::Preimages::drop(&task.call);972973 if let Some(ref id) = task.maybe_id {974 Lookup::<T>::remove(id);975 }976977 Self::deposit_event(Event::PermanentlyOverweight {978 task: (when, agenda_index),979 id: task.maybe_id,980 });981 Err((Unavailable, Some(task)))982 }983 Err(Overweight) => {984 985 Err((Overweight, Some(task)))986 }987 Ok(result) => {988 Self::deposit_event(Event::Dispatched {989 task: (when, agenda_index),990 id: task.maybe_id,991 result,992 });993994 let is_canceled = task995 .maybe_id996 .as_ref()997 .map(|id| !Lookup::<T>::contains_key(id))998 .unwrap_or(false);9991000 match &task.maybe_periodic {1001 &Some((period, count)) if !is_canceled => {1002 if count > 1 {1003 task.maybe_periodic = Some((period, count - 1));1004 } else {1005 task.maybe_periodic = None;1006 }1007 let wake = now.saturating_add(period);1008 match Self::place_task(wake, task) {1009 Ok(_) => {}1010 Err((_, task)) => {1011 1012 1013 T::Preimages::drop(&task.call);1014 Self::deposit_event(Event::PeriodicFailed {1015 task: (when, agenda_index),1016 id: task.maybe_id,1017 });1018 }1019 }1020 }1021 _ => {1022 if let Some(ref id) = task.maybe_id {1023 Lookup::<T>::remove(id);1024 }10251026 T::Preimages::drop(&task.call)1027 }1028 }1029 Ok(())1030 }1031 }1032 }10331034 fn is_runtime_upgraded() -> bool {1035 let last = system::LastRuntimeUpgrade::<T>::get();1036 let current = T::Version::get();10371038 last.map(|v| v.was_upgraded(¤t)).unwrap_or(true)1039 }10401041 1042 1043 1044 1045 1046 1047 fn execute_dispatch(1048 weight: &mut WeightCounter,1049 origin: T::PalletsOrigin,1050 call: <T as Config>::RuntimeCall,1051 ) -> Result<DispatchResult, ServiceTaskError> {1052 let dispatch_origin: <T as Config>::RuntimeOrigin = origin.into();1053 let base_weight = match dispatch_origin.clone().as_signed() {1054 Some(_) => T::WeightInfo::execute_dispatch_signed(),1055 _ => T::WeightInfo::execute_dispatch_unsigned(),1056 };1057 let call_weight = call.get_dispatch_info().weight;1058 1059 let max_weight = base_weight.saturating_add(call_weight);10601061 if !weight.can_accrue(max_weight) {1062 return Err(Overweight);1063 }10641065 1066 1067 let ensured_origin = T::ScheduleOrigin::ensure_origin(dispatch_origin.into());10681069 let r = match ensured_origin {1070 Ok(ScheduledEnsureOriginSuccess::Root) => {1071 Ok(call.dispatch_bypass_filter(frame_system::RawOrigin::Root.into()))1072 }1073 Ok(ScheduledEnsureOriginSuccess::Signed(sender)) => {1074 1075 1076 T::CallExecutor::dispatch_call(Some(sender), call.clone())1077 }1078 Err(e) => Ok(Err(e.into())),1079 };10801081 let (maybe_actual_call_weight, result) = match r {1082 Ok(result) => match result {1083 Ok(post_info) => (post_info.actual_weight, Ok(())),1084 Err(error_and_info) => (1085 error_and_info.post_info.actual_weight,1086 Err(error_and_info.error),1087 ),1088 },1089 Err(_) => {1090 log::error!(1091 target: "runtime::scheduler",1092 "Warning: Scheduler has failed to execute a post-dispatch transaction. \1093 This block might have become invalid.");1094 (None, Err(DispatchError::CannotLookup))1095 }1096 };1097 let call_weight = maybe_actual_call_weight.unwrap_or(call_weight);1098 weight.check_accrue(base_weight);1099 weight.check_accrue(call_weight);1100 Ok(result)1101 }1102}