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::{DispatchError, DispatchResult, Dispatchable, GetDispatchInfo, Parameter},81 traits::{82 schedule::{self, DispatchTime, LOWEST_PRIORITY},83 EnsureOrigin, Get, IsType, OriginTrait, PrivilegeCmp, StorageVersion, PreimageRecipient,84 ConstU32, UnfilteredDispatchable,85 },86 weights::{Weight, PostDispatchInfo}, unsigned::TransactionValidityError,87};8889use frame_system::{self as system};90use scale_info::TypeInfo;91use sp_runtime::{92 traits::{BadOrigin, One, Saturating, Zero, Hash},93 BoundedVec, RuntimeDebug, DispatchErrorWithPostInfo,94};95use sp_core::H160;96use sp_std::{borrow::Borrow, cmp::Ordering, marker::PhantomData, prelude::*};97pub use weights::WeightInfo;9899pub use pallet::*;100101102pub type PeriodicIndex = u32;103104pub type TaskAddress<BlockNumber> = (BlockNumber, u32);105106pub type EncodedCall = BoundedVec<u8, ConstU32<128>>;107108#[derive(Clone, Eq, PartialEq, Encode, Decode, RuntimeDebug, TypeInfo, MaxEncodedLen)]109#[scale_info(skip_type_params(T))]110pub enum ScheduledCall<T: Config> {111 Inline(EncodedCall),112 PreimageLookup { hash: T::Hash, unbounded_len: u32 },113}114115impl<T: Config> ScheduledCall<T> {116 pub fn new(call: <T as Config>::Call) -> Result<Self, DispatchError> {117 let encoded = call.encode();118 let len = encoded.len();119120 match EncodedCall::try_from(encoded.clone()) {121 Ok(bounded) => Ok(Self::Inline(bounded)),122 Err(_) => {123 let hash = <T as system::Config>::Hashing::hash_of(&encoded);124 <T as Config>::Preimages::note_preimage(125 encoded126 .try_into()127 .map_err(|_| <Error<T>>::TooBigScheduledCall)?,128 );129130 Ok(Self::PreimageLookup {131 hash,132 unbounded_len: len as u32,133 })134 }135 }136 }137138 139 pub fn lookup_len(&self) -> Option<u32> {140 match self {141 Self::Inline(..) => None,142 Self::PreimageLookup { unbounded_len, .. } => Some(*unbounded_len),143 }144 }145146 147 pub fn lookup_needed(&self) -> bool {148 match self {149 Self::Inline(_) => false,150 Self::PreimageLookup { .. } => true,151 }152 }153154 fn decode(mut data: &[u8]) -> Result<<T as Config>::Call, DispatchError> {155 <T as Config>::Call::decode(&mut data)156 .map_err(|_| <Error<T>>::ScheduledCallCorrupted.into())157 }158}159160pub trait SchedulerPreimages<T: Config>: PreimageRecipient<T::Hash> {161 fn drop(call: &ScheduledCall<T>);162163 fn peek(164 call: &ScheduledCall<T>,165 ) -> Result<(<T as pallet::Config>::Call, Option<u32>), DispatchError>;166167 168 169 170 fn realize(171 call: &ScheduledCall<T>,172 ) -> Result<(<T as pallet::Config>::Call, Option<u32>), DispatchError>;173}174175impl<T: Config, PP: PreimageRecipient<T::Hash>> SchedulerPreimages<T> for PP {176 fn drop(call: &ScheduledCall<T>) {177 match call {178 ScheduledCall::Inline(_) => {}179 ScheduledCall::PreimageLookup { hash, .. } => Self::unrequest_preimage(hash),180 }181 }182183 fn peek(184 call: &ScheduledCall<T>,185 ) -> Result<(<T as pallet::Config>::Call, Option<u32>), DispatchError> {186 match call {187 ScheduledCall::Inline(data) => Ok((ScheduledCall::<T>::decode(data)?, None)),188 ScheduledCall::PreimageLookup {189 hash,190 unbounded_len,191 } => {192 let (preimage, len) = Self::get_preimage(hash)193 .ok_or(<Error<T>>::PreimageNotFound)194 .map(|preimage| (preimage, *unbounded_len))?;195196 Ok((ScheduledCall::<T>::decode(preimage.as_slice())?, Some(len)))197 }198 }199 }200201 fn realize(202 call: &ScheduledCall<T>,203 ) -> Result<(<T as pallet::Config>::Call, Option<u32>), DispatchError> {204 let r = Self::peek(call)?;205 Self::drop(call);206 Ok(r)207 }208}209210pub enum ScheduledEnsureOriginSuccess<AccountId> {211 Root,212 Signed(AccountId),213 Unsigned,214}215216pub type TaskName = [u8; 32];217218219#[cfg_attr(any(feature = "std", test), derive(PartialEq, Eq))]220#[derive(Clone, RuntimeDebug, Encode, Decode, MaxEncodedLen, TypeInfo)]221pub struct Scheduled<Name, Call, BlockNumber, PalletsOrigin, AccountId> {222 223 maybe_id: Option<Name>,224225 226 priority: schedule::Priority,227228 229 call: Call,230231 232 maybe_periodic: Option<schedule::Period<BlockNumber>>,233234 235 origin: PalletsOrigin,236 _phantom: PhantomData<AccountId>,237}238239pub type ScheduledOf<T> = Scheduled<240 TaskName,241 ScheduledCall<T>,242 <T as frame_system::Config>::BlockNumber,243 <T as Config>::PalletsOrigin,244 <T as frame_system::Config>::AccountId,245>;246247struct WeightCounter {248 used: Weight,249 limit: Weight,250}251252impl WeightCounter {253 fn check_accrue(&mut self, w: Weight) -> bool {254 let test = self.used.saturating_add(w);255 if test > self.limit {256 false257 } else {258 self.used = test;259 true260 }261 }262263 fn can_accrue(&mut self, w: Weight) -> bool {264 self.used.saturating_add(w) <= self.limit265 }266}267268pub(crate) trait MarginalWeightInfo: WeightInfo {269 fn service_task(maybe_lookup_len: Option<usize>, named: bool, periodic: bool) -> Weight {270 let base = Self::service_task_base();271 let mut total = match maybe_lookup_len {272 None => base,273 Some(l) => Self::service_task_fetched(l as u32),274 };275 if named {276 total.saturating_accrue(Self::service_task_named().saturating_sub(base));277 }278 if periodic {279 total.saturating_accrue(Self::service_task_periodic().saturating_sub(base));280 }281 total282 }283}284285impl<T: WeightInfo> MarginalWeightInfo for T {}286287#[frame_support::pallet]288pub mod pallet {289 use super::*;290 use frame_support::{dispatch::PostDispatchInfo, pallet_prelude::*};291 use system::pallet_prelude::*;292293 294 const STORAGE_VERSION: StorageVersion = StorageVersion::new(0);295296 #[pallet::pallet]297 #[pallet::generate_store(pub(super) trait Store)]298 #[pallet::storage_version(STORAGE_VERSION)]299 pub struct Pallet<T>(_);300301 #[pallet::config]302 pub trait Config: frame_system::Config {303 type Event: From<Event<Self>> + IsType<<Self as frame_system::Config>::Event>;304305 306 type Origin: OriginTrait<PalletsOrigin = Self::PalletsOrigin>307 + From<Self::PalletsOrigin>308 + IsType<<Self as system::Config>::Origin>309 + Clone;310311 312 type PalletsOrigin: From<system::RawOrigin<Self::AccountId>>313 + Codec314 + Clone315 + Eq316 + TypeInfo317 + MaxEncodedLen;318319 320 type Call: Parameter321 + Dispatchable<Origin = <Self as Config>::Origin, PostInfo = PostDispatchInfo>322 + UnfilteredDispatchable<Origin = <Self as system::Config>::Origin>323 + GetDispatchInfo324 + From<system::Call<Self>>;325326 327 #[pallet::constant]328 type MaximumWeight: Get<Weight>;329330 331 type ScheduleOrigin: EnsureOrigin<332 <Self as system::Config>::Origin,333 Success = ScheduledEnsureOriginSuccess<Self::AccountId>,334 >;335336 337 338 339 340 341 342 343 type OriginPrivilegeCmp: PrivilegeCmp<Self::PalletsOrigin>;344345 346 #[pallet::constant]347 type MaxScheduledPerBlock: Get<u32>;348349 350 type WeightInfo: WeightInfo;351352 353 type Preimages: SchedulerPreimages<Self>;354355 356 type CallExecutor: DispatchCall<Self, H160>;357358 359 type PrioritySetOrigin: EnsureOrigin<<Self as system::Config>::Origin>;360 }361362 #[pallet::storage]363 pub type IncompleteSince<T: Config> = StorageValue<_, T::BlockNumber>;364365 366 #[pallet::storage]367 pub type Agenda<T: Config> = StorageMap<368 _,369 Twox64Concat,370 T::BlockNumber,371 BoundedVec<Option<ScheduledOf<T>>, T::MaxScheduledPerBlock>,372 ValueQuery,373 >;374375 376 #[pallet::storage]377 pub(crate) type Lookup<T: Config> =378 StorageMap<_, Twox64Concat, TaskName, TaskAddress<T::BlockNumber>>;379380 381 #[pallet::event]382 #[pallet::generate_deposit(pub(super) fn deposit_event)]383 pub enum Event<T: Config> {384 385 Scheduled { when: T::BlockNumber, index: u32 },386 387 Canceled { when: T::BlockNumber, index: u32 },388 389 Dispatched {390 task: TaskAddress<T::BlockNumber>,391 id: Option<[u8; 32]>,392 result: DispatchResult,393 },394 395 PriorityChanged {396 when: T::BlockNumber,397 index: u32,398 priority: schedule::Priority,399 },400 401 CallUnavailable {402 task: TaskAddress<T::BlockNumber>,403 id: Option<[u8; 32]>,404 },405 406 PeriodicFailed {407 task: TaskAddress<T::BlockNumber>,408 id: Option<[u8; 32]>,409 },410 411 PermanentlyOverweight {412 task: TaskAddress<T::BlockNumber>,413 id: Option<[u8; 32]>,414 },415 }416417 #[pallet::error]418 pub enum Error<T> {419 420 FailedToSchedule,421 422 AgendaIsExhausted,423 424 ScheduledCallCorrupted,425 426 PreimageNotFound,427 428 TooBigScheduledCall,429 430 NotFound,431 432 TargetBlockNumberInPast,433 434 RescheduleNoChange,435 436 Named,437 }438439 #[pallet::hooks]440 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {441 442 fn on_initialize(now: T::BlockNumber) -> Weight {443 let mut weight_counter = WeightCounter {444 used: Weight::zero(),445 limit: T::MaximumWeight::get(),446 };447 Self::service_agendas(&mut weight_counter, now, u32::max_value());448 weight_counter.used449 }450 }451452 #[pallet::call]453 impl<T: Config> Pallet<T> {454 455 #[pallet::weight(<T as Config>::WeightInfo::schedule(T::MaxScheduledPerBlock::get()))]456 pub fn schedule(457 origin: OriginFor<T>,458 when: T::BlockNumber,459 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,460 priority: Option<schedule::Priority>,461 call: Box<<T as Config>::Call>,462 ) -> DispatchResult {463 T::ScheduleOrigin::ensure_origin(origin.clone())?;464465 if priority.is_some() {466 T::PrioritySetOrigin::ensure_origin(origin.clone())?;467 }468469 let origin = <T as Config>::Origin::from(origin);470 Self::do_schedule(471 DispatchTime::At(when),472 maybe_periodic,473 priority.unwrap_or(LOWEST_PRIORITY),474 origin.caller().clone(),475 <ScheduledCall<T>>::new(*call)?,476 )?;477 Ok(())478 }479480 481 #[pallet::weight(<T as Config>::WeightInfo::cancel(T::MaxScheduledPerBlock::get()))]482 pub fn cancel(origin: OriginFor<T>, when: T::BlockNumber, index: u32) -> DispatchResult {483 T::ScheduleOrigin::ensure_origin(origin.clone())?;484 let origin = <T as Config>::Origin::from(origin);485 Self::do_cancel(Some(origin.caller().clone()), (when, index))?;486 Ok(())487 }488489 490 #[pallet::weight(<T as Config>::WeightInfo::schedule_named(T::MaxScheduledPerBlock::get()))]491 pub fn schedule_named(492 origin: OriginFor<T>,493 id: TaskName,494 when: T::BlockNumber,495 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,496 priority: Option<schedule::Priority>,497 call: Box<<T as Config>::Call>,498 ) -> DispatchResult {499 T::ScheduleOrigin::ensure_origin(origin.clone())?;500501 if priority.is_some() {502 T::PrioritySetOrigin::ensure_origin(origin.clone())?;503 }504505 let origin = <T as Config>::Origin::from(origin);506 Self::do_schedule_named(507 id,508 DispatchTime::At(when),509 maybe_periodic,510 priority.unwrap_or(LOWEST_PRIORITY),511 origin.caller().clone(),512 <ScheduledCall<T>>::new(*call)?,513 )?;514 Ok(())515 }516517 518 #[pallet::weight(<T as Config>::WeightInfo::cancel_named(T::MaxScheduledPerBlock::get()))]519 pub fn cancel_named(origin: OriginFor<T>, id: TaskName) -> DispatchResult {520 T::ScheduleOrigin::ensure_origin(origin.clone())?;521 let origin = <T as Config>::Origin::from(origin);522 Self::do_cancel_named(Some(origin.caller().clone()), id)?;523 Ok(())524 }525526 527 528 529 530 531 #[pallet::weight(<T as Config>::WeightInfo::schedule(T::MaxScheduledPerBlock::get()))]532 pub fn schedule_after(533 origin: OriginFor<T>,534 after: T::BlockNumber,535 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,536 priority: Option<schedule::Priority>,537 call: Box<<T as Config>::Call>,538 ) -> DispatchResult {539 T::ScheduleOrigin::ensure_origin(origin.clone())?;540541 if priority.is_some() {542 T::PrioritySetOrigin::ensure_origin(origin.clone())?;543 }544545 let origin = <T as Config>::Origin::from(origin);546 Self::do_schedule(547 DispatchTime::After(after),548 maybe_periodic,549 priority.unwrap_or(LOWEST_PRIORITY),550 origin.caller().clone(),551 <ScheduledCall<T>>::new(*call)?,552 )?;553 Ok(())554 }555556 557 558 559 560 561 #[pallet::weight(<T as Config>::WeightInfo::schedule_named(T::MaxScheduledPerBlock::get()))]562 pub fn schedule_named_after(563 origin: OriginFor<T>,564 id: TaskName,565 after: T::BlockNumber,566 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,567 priority: Option<schedule::Priority>,568 call: Box<<T as Config>::Call>,569 ) -> DispatchResult {570 T::ScheduleOrigin::ensure_origin(origin.clone())?;571572 if priority.is_some() {573 T::PrioritySetOrigin::ensure_origin(origin.clone())?;574 }575576 let origin = <T as Config>::Origin::from(origin);577 Self::do_schedule_named(578 id,579 DispatchTime::After(after),580 maybe_periodic,581 priority.unwrap_or(LOWEST_PRIORITY),582 origin.caller().clone(),583 <ScheduledCall<T>>::new(*call)?,584 )?;585 Ok(())586 }587588 #[pallet::weight(<T as Config>::WeightInfo::change_named_priority(T::MaxScheduledPerBlock::get()))]589 pub fn change_named_priority(590 origin: OriginFor<T>,591 id: TaskName,592 priority: schedule::Priority,593 ) -> DispatchResult {594 T::PrioritySetOrigin::ensure_origin(origin.clone())?;595 let origin = <T as Config>::Origin::from(origin);596 Self::do_change_named_priority(origin.caller().clone(), id, priority)597 }598 }599}600601impl<T: Config> Pallet<T> {602 fn resolve_time(when: DispatchTime<T::BlockNumber>) -> Result<T::BlockNumber, DispatchError> {603 let now = frame_system::Pallet::<T>::block_number();604605 let when = match when {606 DispatchTime::At(x) => x,607 608 609 DispatchTime::After(x) => now.saturating_add(x).saturating_add(One::one()),610 };611612 if when <= now {613 return Err(Error::<T>::TargetBlockNumberInPast.into());614 }615616 Ok(when)617 }618619 fn place_task(620 when: T::BlockNumber,621 what: ScheduledOf<T>,622 ) -> Result<TaskAddress<T::BlockNumber>, (DispatchError, ScheduledOf<T>)> {623 let maybe_name = what.maybe_id;624 let index = Self::push_to_agenda(when, what)?;625 let address = (when, index);626 if let Some(name) = maybe_name {627 Lookup::<T>::insert(name, address)628 }629 Self::deposit_event(Event::Scheduled {630 when: address.0,631 index: address.1,632 });633 Ok(address)634 }635636 fn push_to_agenda(637 when: T::BlockNumber,638 what: ScheduledOf<T>,639 ) -> Result<u32, (DispatchError, ScheduledOf<T>)> {640 let mut agenda = Agenda::<T>::get(when);641 let index = if (agenda.len() as u32) < T::MaxScheduledPerBlock::get() {642 643 let _ = agenda.try_push(Some(what));644 agenda.len() as u32 - 1645 } else {646 if let Some(hole_index) = agenda.iter().position(|i| i.is_none()) {647 agenda[hole_index] = Some(what);648 hole_index as u32649 } else {650 return Err((<Error<T>>::AgendaIsExhausted.into(), what));651 }652 };653 Agenda::<T>::insert(when, agenda);654 Ok(index)655 }656657 fn do_schedule(658 when: DispatchTime<T::BlockNumber>,659 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,660 priority: schedule::Priority,661 origin: T::PalletsOrigin,662 call: ScheduledCall<T>,663 ) -> Result<TaskAddress<T::BlockNumber>, DispatchError> {664 let when = Self::resolve_time(when)?;665666 667 let maybe_periodic = maybe_periodic668 .filter(|p| p.1 > 1 && !p.0.is_zero())669 670 .map(|(p, c)| (p, c - 1));671 let task = Scheduled {672 maybe_id: None,673 priority,674 call,675 maybe_periodic,676 origin,677 _phantom: PhantomData,678 };679 Self::place_task(when, task).map_err(|x| x.0)680 }681682 fn do_cancel(683 origin: Option<T::PalletsOrigin>,684 (when, index): TaskAddress<T::BlockNumber>,685 ) -> Result<(), DispatchError> {686 let scheduled = Agenda::<T>::try_mutate(when, |agenda| {687 agenda.get_mut(index as usize).map_or(688 Ok(None),689 |s| -> Result<Option<Scheduled<_, _, _, _, _>>, DispatchError> {690 if let (Some(ref o), Some(ref s)) = (origin, s.borrow()) {691 if matches!(692 T::OriginPrivilegeCmp::cmp_privilege(o, &s.origin),693 Some(Ordering::Less) | None694 ) {695 return Err(BadOrigin.into());696 }697 };698 Ok(s.take())699 },700 )701 })?;702 if let Some(s) = scheduled {703 T::Preimages::drop(&s.call);704705 if let Some(id) = s.maybe_id {706 Lookup::<T>::remove(id);707 }708 Self::deposit_event(Event::Canceled { when, index });709 Ok(())710 } else {711 return Err(Error::<T>::NotFound.into());712 }713 }714715 fn do_schedule_named(716 id: TaskName,717 when: DispatchTime<T::BlockNumber>,718 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,719 priority: schedule::Priority,720 origin: T::PalletsOrigin,721 call: ScheduledCall<T>,722 ) -> Result<TaskAddress<T::BlockNumber>, DispatchError> {723 724 if Lookup::<T>::contains_key(&id) {725 return Err(Error::<T>::FailedToSchedule.into());726 }727728 let when = Self::resolve_time(when)?;729730 731 let maybe_periodic = maybe_periodic732 .filter(|p| p.1 > 1 && !p.0.is_zero())733 734 .map(|(p, c)| (p, c - 1));735736 let task = Scheduled {737 maybe_id: Some(id),738 priority,739 call,740 maybe_periodic,741 origin,742 _phantom: Default::default(),743 };744 Self::place_task(when, task).map_err(|x| x.0)745 }746747 fn do_cancel_named(origin: Option<T::PalletsOrigin>, id: TaskName) -> DispatchResult {748 Lookup::<T>::try_mutate_exists(id, |lookup| -> DispatchResult {749 if let Some((when, index)) = lookup.take() {750 let i = index as usize;751 Agenda::<T>::try_mutate(when, |agenda| -> DispatchResult {752 if let Some(s) = agenda.get_mut(i) {753 if let (Some(ref o), Some(ref s)) = (origin, s.borrow()) {754 if matches!(755 T::OriginPrivilegeCmp::cmp_privilege(o, &s.origin),756 Some(Ordering::Less) | None757 ) {758 return Err(BadOrigin.into());759 }760 T::Preimages::drop(&s.call);761 }762 *s = None;763 }764 Ok(())765 })?;766 Self::deposit_event(Event::Canceled { when, index });767 Ok(())768 } else {769 return Err(Error::<T>::NotFound.into());770 }771 })772 }773774 fn do_change_named_priority(775 origin: T::PalletsOrigin,776 id: TaskName,777 priority: schedule::Priority,778 ) -> DispatchResult {779 match Lookup::<T>::get(id) {780 Some((when, index)) => {781 let i = index as usize;782 Agenda::<T>::try_mutate(when, |agenda| {783 if let Some(Some(s)) = agenda.get_mut(i) {784 if matches!(785 T::OriginPrivilegeCmp::cmp_privilege(&origin, &s.origin),786 Some(Ordering::Less) | None787 ) {788 return Err(BadOrigin.into());789 }790791 s.priority = priority;792 Self::deposit_event(Event::PriorityChanged {793 when,794 index,795 priority,796 });797 }798 Ok(())799 })800 }801 None => Err(Error::<T>::NotFound.into()),802 }803 }804}805806enum ServiceTaskError {807 808 Unavailable,809 810 Overweight,811}812use ServiceTaskError::*;813814815pub trait DispatchCall<T: frame_system::Config + Config, SelfContainedSignedInfo> {816 817 fn dispatch_call(818 signer: Option<T::AccountId>,819 function: <T as Config>::Call,820 ) -> Result<821 Result<PostDispatchInfo, DispatchErrorWithPostInfo<PostDispatchInfo>>,822 TransactionValidityError,823 >;824}825826impl<T: Config> Pallet<T> {827 828 fn service_agendas(weight: &mut WeightCounter, now: T::BlockNumber, max: u32) {829 if !weight.check_accrue(T::WeightInfo::service_agendas_base()) {830 return;831 }832833 let mut incomplete_since = now + One::one();834 let mut when = IncompleteSince::<T>::take().unwrap_or(now);835 let mut executed = 0;836837 let max_items = T::MaxScheduledPerBlock::get();838 let mut count_down = max;839 let service_agenda_base_weight = T::WeightInfo::service_agenda_base(max_items);840 while count_down > 0 && when <= now && weight.can_accrue(service_agenda_base_weight) {841 if !Self::service_agenda(weight, &mut executed, now, when, u32::max_value()) {842 incomplete_since = incomplete_since.min(when);843 }844 when.saturating_inc();845 count_down.saturating_dec();846 }847 incomplete_since = incomplete_since.min(when);848 if incomplete_since <= now {849 IncompleteSince::<T>::put(incomplete_since);850 }851 }852853 854 855 fn service_agenda(856 weight: &mut WeightCounter,857 executed: &mut u32,858 now: T::BlockNumber,859 when: T::BlockNumber,860 max: u32,861 ) -> bool {862 let mut agenda = Agenda::<T>::get(when);863 let mut ordered = agenda864 .iter()865 .enumerate()866 .filter_map(|(index, maybe_item)| {867 maybe_item868 .as_ref()869 .map(|item| (index as u32, item.priority))870 })871 .collect::<Vec<_>>();872 ordered.sort_by_key(|k| k.1);873 let within_limit =874 weight.check_accrue(T::WeightInfo::service_agenda_base(ordered.len() as u32));875 debug_assert!(876 within_limit,877 "weight limit should have been checked in advance"878 );879880 881 let mut postponed = (ordered.len() as u32).saturating_sub(max);882 883 let mut dropped = 0;884885 for (agenda_index, _) in ordered.into_iter().take(max as usize) {886 let task = match agenda[agenda_index as usize].take() {887 None => continue,888 Some(t) => t,889 };890 let base_weight = T::WeightInfo::service_task(891 task.call.lookup_len().map(|x| x as usize),892 task.maybe_id.is_some(),893 task.maybe_periodic.is_some(),894 );895 if !weight.can_accrue(base_weight) {896 postponed += 1;897 break;898 }899 let result = Self::service_task(weight, now, when, agenda_index, *executed == 0, task);900 agenda[agenda_index as usize] = match result {901 Err((Unavailable, slot)) => {902 dropped += 1;903 slot904 }905 Err((Overweight, slot)) => {906 postponed += 1;907 slot908 }909 Ok(()) => {910 *executed += 1;911 None912 }913 };914 }915 if postponed > 0 || dropped > 0 {916 Agenda::<T>::insert(when, agenda);917 } else {918 Agenda::<T>::remove(when);919 }920 postponed == 0921 }922923 924 925 926 927 928 929 fn service_task(930 weight: &mut WeightCounter,931 now: T::BlockNumber,932 when: T::BlockNumber,933 agenda_index: u32,934 is_first: bool,935 mut task: ScheduledOf<T>,936 ) -> Result<(), (ServiceTaskError, Option<ScheduledOf<T>>)> {937 let (call, lookup_len) = match T::Preimages::peek(&task.call) {938 Ok(c) => c,939 Err(_) => {940 if let Some(ref id) = task.maybe_id {941 Lookup::<T>::remove(id);942 }943944 return Err((Unavailable, Some(task)));945 },946 };947948 weight.check_accrue(T::WeightInfo::service_task(949 lookup_len.map(|x| x as usize),950 task.maybe_id.is_some(),951 task.maybe_periodic.is_some(),952 ));953954 match Self::execute_dispatch(weight, task.origin.clone(), call) {955 Err(Unavailable) => {956 debug_assert!(false, "Checked to exist with `peek`");957958 if let Some(ref id) = task.maybe_id {959 Lookup::<T>::remove(id);960 }961962 Self::deposit_event(Event::CallUnavailable {963 task: (when, agenda_index),964 id: task.maybe_id,965 });966 Err((Unavailable, Some(task)))967 }968 Err(Overweight) if is_first => {969 T::Preimages::drop(&task.call);970971 if let Some(ref id) = task.maybe_id {972 Lookup::<T>::remove(id);973 }974975 Self::deposit_event(Event::PermanentlyOverweight {976 task: (when, agenda_index),977 id: task.maybe_id,978 });979 Err((Unavailable, Some(task)))980 }981 Err(Overweight) => {982 983 Err((Overweight, Some(task)))984 },985 Ok(result) => {986 Self::deposit_event(Event::Dispatched {987 task: (when, agenda_index),988 id: task.maybe_id,989 result,990 });991992 let is_canceled = task.maybe_id.as_ref()993 .map(|id| !Lookup::<T>::contains_key(id))994 .unwrap_or(false);995996 match &task.maybe_periodic {997 &Some((period, count)) if !is_canceled => {998 if count > 1 {999 task.maybe_periodic = Some((period, count - 1));1000 } else {1001 task.maybe_periodic = None;1002 }1003 let wake = now.saturating_add(period);1004 match Self::place_task(wake, task) {1005 Ok(_) => {}1006 Err((_, task)) => {1007 1008 1009 T::Preimages::drop(&task.call);1010 Self::deposit_event(Event::PeriodicFailed {1011 task: (when, agenda_index),1012 id: task.maybe_id,1013 });1014 }1015 }1016 },1017 _ => {1018 if let Some(ref id) = task.maybe_id {1019 Lookup::<T>::remove(id);1020 }10211022 T::Preimages::drop(&task.call)1023 },1024 }1025 Ok(())1026 }1027 }1028 }10291030 1031 1032 1033 1034 1035 1036 fn execute_dispatch(1037 weight: &mut WeightCounter,1038 origin: T::PalletsOrigin,1039 call: <T as Config>::Call,1040 ) -> Result<DispatchResult, ServiceTaskError> {1041 let dispatch_origin: <T as Config>::Origin = origin.into();1042 let base_weight = match dispatch_origin.clone().as_signed() {1043 Some(_) => T::WeightInfo::execute_dispatch_signed(),1044 _ => T::WeightInfo::execute_dispatch_unsigned(),1045 };1046 let call_weight = call.get_dispatch_info().weight;1047 1048 let max_weight = base_weight.saturating_add(call_weight);10491050 if !weight.can_accrue(max_weight) {1051 return Err(Overweight);1052 }10531054 1055 1056 let ensured_origin = T::ScheduleOrigin::ensure_origin(dispatch_origin.into());10571058 let r = match ensured_origin {1059 Ok(ScheduledEnsureOriginSuccess::Root) => {1060 Ok(call.dispatch_bypass_filter(frame_system::RawOrigin::Root.into()))1061 },1062 Ok(ScheduledEnsureOriginSuccess::Signed(sender)) => {1063 1064 1065 T::CallExecutor::dispatch_call(Some(sender), call.clone())1066 },1067 Ok(ScheduledEnsureOriginSuccess::Unsigned) => {1068 1069 T::CallExecutor::dispatch_call(None, call.clone())1070 }1071 Err(e) => Ok(Err(e.into())),1072 };10731074 let (maybe_actual_call_weight, result) = match r {1075 Ok(result) => match result {1076 Ok(post_info) => (post_info.actual_weight, Ok(())),1077 Err(error_and_info) => (1078 error_and_info.post_info.actual_weight,1079 Err(error_and_info.error),1080 ),1081 },1082 Err(_) => {1083 log::error!(1084 target: "runtime::scheduler",1085 "Warning: Scheduler has failed to execute a post-dispatch transaction. \1086 This block might have become invalid.");1087 (None, Err(DispatchError::CannotLookup))1088 }1089 };1090 let call_weight = maybe_actual_call_weight.unwrap_or(call_weight);1091 weight.check_accrue(base_weight);1092 weight.check_accrue(call_weight);1093 Ok(result)1094 }1095}