difftreelog
fix remove unsigned scheduled origins
in: master
1 file changed
pallets/scheduler/src/lib.rsdiffbeforeafterboth1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617// Original license:18// This file is part of Substrate.1920// Copyright (C) 2017-2022 Parity Technologies (UK) Ltd.21// SPDX-License-Identifier: Apache-2.02223// Licensed under the Apache License, Version 2.0 (the "License");24// you may not use this file except in compliance with the License.25// You may obtain a copy of the License at26//27// <http://www.apache.org/licenses/LICENSE-2.0>28//29// Unless required by applicable law or agreed to in writing, software30// distributed under the License is distributed on an "AS IS" BASIS,31// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.32// See the License for the specific language governing permissions and33// limitations under the License.3435//! # Unique scheduler36//! A Pallet for scheduling dispatches.37//!38//! - [`Config`]39//! - [`Call`]40//! - [`Pallet`]41//!42//! ## Overview43//!44//! This Pallet exposes capabilities for scheduling dispatches to occur at a45//! specified block number or at a specified period. These scheduled dispatches46//! should be named and may be canceled.47//!48//! **NOTE:** The unique scheduler is designed for deferred transaction calls by block number.49//! Any user can book a call of a certain transaction to a specific block number.50//! Also possible to book a call with a certain frequency.51//!52//! Key differences from the original pallet:53//! <https://crates.io/crates/pallet-scheduler>54//! Schedule Id restricted by 16 bytes. Identificator for booked call.55//! Priority limited by HARD DEADLINE (<= 63). Calls over maximum weight don't include to block.56//! The maximum weight that may be scheduled per block for any dispatchables of less priority than `schedule::HARD_DEADLINE`.57//! Maybe_periodic limit is 100 calls. Reserved for future sponsored transaction support.58//! At 100 calls reserved amount is not so much and this is avoid potential problems with balance locks.59//! Any account allowed to schedule any calls. Account withdraw implemented through default transaction logic.60//!61//! ## Interface62//!63//! ### Dispatchable Functions64//!65//! * `schedule_named` - augments the `schedule` interface with an additional `Vec<u8>` parameter66//! that can be used for identification.67//! * `cancel_named` - the named complement to the cancel function.6869// Ensure we're `no_std` when compiling for Wasm.70#![cfg_attr(not(feature = "std"), no_std)]7172#[cfg(feature = "runtime-benchmarks")]73mod benchmarking;7475pub mod weights;7677use sp_core::H160;78use codec::{Codec, Decode, Encode};79use frame_system::{self as system, ensure_signed};80pub use pallet::*;81use scale_info::TypeInfo;82use sp_runtime::{83 traits::{BadOrigin, One, Saturating, Zero},84 RuntimeDebug, DispatchErrorWithPostInfo,85};86use sp_std::{borrow::Borrow, cmp::Ordering, marker::PhantomData, prelude::*};8788use frame_support::{89 dispatch::{90 DispatchError, DispatchResult, Dispatchable, UnfilteredDispatchable, Parameter,91 GetDispatchInfo,92 },93 traits::{94 schedule::{self, DispatchTime, MaybeHashed},95 NamedReservableCurrency, EnsureOrigin, Get, IsType, OriginTrait, PrivilegeCmp,96 StorageVersion,97 },98 weights::{Weight},99};100101pub use weights::WeightInfo;102103/// The location of a scheduled task that can be used to remove it.104pub type TaskAddress<BlockNumber> = (BlockNumber, u32);105pub const MAX_TASK_ID_LENGTH_IN_BYTES: u8 = 16;106107pub type ScheduledId = [u8; MAX_TASK_ID_LENGTH_IN_BYTES as usize];108pub type CallOrHashOf<T> =109 MaybeHashed<<T as Config>::RuntimeCall, <T as frame_system::Config>::Hash>;110111/// Information regarding an item to be executed in the future.112#[cfg_attr(any(feature = "std", test), derive(PartialEq, Eq))]113#[derive(Clone, RuntimeDebug, Encode, Decode, TypeInfo)]114pub struct ScheduledV3<Call, BlockNumber, PalletsOrigin, AccountId> {115 /// The unique identity for this task, if there is one.116 maybe_id: Option<ScheduledId>,117 /// This task's priority.118 priority: schedule::Priority,119 /// The call to be dispatched.120 call: Call,121 /// If the call is periodic, then this points to the information concerning that.122 maybe_periodic: Option<schedule::Period<BlockNumber>>,123 /// The origin to dispatch the call.124 origin: PalletsOrigin,125 _phantom: PhantomData<AccountId>,126}127128pub type ScheduledV3Of<T> = ScheduledV3<129 CallOrHashOf<T>,130 <T as frame_system::Config>::BlockNumber,131 <T as Config>::PalletsOrigin,132 <T as frame_system::Config>::AccountId,133>;134135pub type ScheduledOf<T> = ScheduledV3Of<T>;136137/// The current version of Scheduled struct.138pub type Scheduled<Call, BlockNumber, PalletsOrigin, AccountId> =139 ScheduledV3<Call, BlockNumber, PalletsOrigin, AccountId>;140141pub enum ScheduledEnsureOriginSuccess<AccountId> {142 Root,143 Signed(AccountId),144 Unsigned,145}146147#[cfg(feature = "runtime-benchmarks")]148mod preimage_provider {149 use frame_support::traits::PreimageRecipient;150 pub trait PreimageProviderAndMaybeRecipient<H>: PreimageRecipient<H> {}151 impl<H, T: PreimageRecipient<H>> PreimageProviderAndMaybeRecipient<H> for T {}152}153154#[cfg(not(feature = "runtime-benchmarks"))]155mod preimage_provider {156 use frame_support::traits::PreimageProvider;157 pub trait PreimageProviderAndMaybeRecipient<H>: PreimageProvider<H> {}158 impl<H, T: PreimageProvider<H>> PreimageProviderAndMaybeRecipient<H> for T {}159}160161pub use preimage_provider::PreimageProviderAndMaybeRecipient;162163pub(crate) trait MarginalWeightInfo: WeightInfo {164 fn item(periodic: bool, named: bool, resolved: Option<bool>) -> Weight {165 match (periodic, named, resolved) {166 (_, false, None) => Self::on_initialize_aborted(2) - Self::on_initialize_aborted(1),167 (_, true, None) => {168 Self::on_initialize_named_aborted(2) - Self::on_initialize_named_aborted(1)169 }170 (false, false, Some(false)) => Self::on_initialize(2) - Self::on_initialize(1),171 (false, true, Some(false)) => {172 Self::on_initialize_named(2) - Self::on_initialize_named(1)173 }174 (true, false, Some(false)) => {175 Self::on_initialize_periodic(2) - Self::on_initialize_periodic(1)176 }177 (true, true, Some(false)) => {178 Self::on_initialize_periodic_named_resolved(2)179 - Self::on_initialize_periodic_named_resolved(1)180 }181 (false, false, Some(true)) => Self::on_initialize(2) - Self::on_initialize(1),182 (false, true, Some(true)) => {183 Self::on_initialize_named_resolved(2) - Self::on_initialize_named_resolved(1)184 }185 (true, false, Some(true)) => {186 Self::on_initialize_periodic_resolved(2) - Self::on_initialize_periodic_resolved(1)187 }188 (true, true, Some(true)) => {189 Self::on_initialize_periodic_named_resolved(2)190 - Self::on_initialize_periodic_named_resolved(1)191 }192 }193 }194}195impl<T: WeightInfo> MarginalWeightInfo for T {}196197#[frame_support::pallet]198pub mod pallet {199 use super::*;200 use frame_support::{201 dispatch::PostDispatchInfo,202 pallet_prelude::*,203 traits::{204 schedule::{LookupError, LOWEST_PRIORITY},205 PreimageProvider,206 },207 };208 use frame_system::pallet_prelude::*;209210 /// The current storage version.211 const STORAGE_VERSION: StorageVersion = StorageVersion::new(3);212213 #[pallet::pallet]214 #[pallet::generate_store(pub(super) trait Store)]215 #[pallet::storage_version(STORAGE_VERSION)]216 #[pallet::without_storage_info]217 pub struct Pallet<T>(_);218219 /// `system::Config` should always be included in our implied traits.220 #[pallet::config]221 pub trait Config: frame_system::Config {222 /// The overarching event type.223 type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;224225 /// The aggregated origin which the dispatch will take.226 type RuntimeOrigin: OriginTrait<PalletsOrigin = Self::PalletsOrigin>227 + From<Self::PalletsOrigin>228 + IsType<<Self as system::Config>::RuntimeOrigin>;229230 /// The caller origin, overarching type of all pallets origins.231 type PalletsOrigin: From<system::RawOrigin<Self::AccountId>> + Codec + Clone + Eq + TypeInfo;232233 type Currency: NamedReservableCurrency<Self::AccountId, ReserveIdentifier = ScheduledId>;234235 /// The aggregated call type.236 type RuntimeCall: Parameter237 + Dispatchable<238 RuntimeOrigin = <Self as Config>::RuntimeOrigin,239 PostInfo = PostDispatchInfo,240 > + UnfilteredDispatchable<RuntimeOrigin = <Self as system::Config>::RuntimeOrigin>241 + GetDispatchInfo242 + From<system::Call<Self>>;243244 /// The maximum weight that may be scheduled per block for any dispatchables of less245 /// priority than `schedule::HARD_DEADLINE`.246 #[pallet::constant]247 type MaximumWeight: Get<Weight>;248249 /// Required origin to schedule or cancel calls.250 type ScheduleOrigin: EnsureOrigin<251 <Self as system::Config>::RuntimeOrigin,252 Success = ScheduledEnsureOriginSuccess<Self::AccountId>,253 >;254255 /// Required origin to set/change calls' priority.256 type PrioritySetOrigin: EnsureOrigin<<Self as system::Config>::RuntimeOrigin>;257258 /// Compare the privileges of origins.259 ///260 /// This will be used when canceling a task, to ensure that the origin that tries261 /// to cancel has greater or equal privileges as the origin that created the scheduled task.262 ///263 /// For simplicity the [`EqualPrivilegeOnly`](frame_support::traits::EqualPrivilegeOnly) can264 /// be used. This will only check if two given origins are equal.265 type OriginPrivilegeCmp: PrivilegeCmp<Self::PalletsOrigin>;266267 /// The maximum number of scheduled calls in the queue for a single block.268 /// Not strictly enforced, but used for weight estimation.269 #[pallet::constant]270 type MaxScheduledPerBlock: Get<u32>;271272 /// Weight information for extrinsics in this pallet.273 type WeightInfo: WeightInfo;274275 /// The preimage provider with which we look up call hashes to get the call.276 type PreimageProvider: PreimageProviderAndMaybeRecipient<Self::Hash>;277278 /// If `Some` then the number of blocks to postpone execution for when the item is delayed.279 type NoPreimagePostponement: Get<Option<Self::BlockNumber>>;280281 /// Sponsoring function.282 // type SponsorshipHandler: SponsorshipHandler<Self::AccountId, <Self as Config>::Call>;283284 /// The helper type used for custom transaction fee logic.285 type CallExecutor: DispatchCall<Self, H160>;286 }287288 /// A Scheduler-Runtime interface for finer payment handling.289 pub trait DispatchCall<T: frame_system::Config + Config, SelfContainedSignedInfo> {290 /// Reserve (lock) the maximum spendings on a call, calculated from its weight and the repetition count.291 fn reserve_balance(292 id: ScheduledId,293 sponsor: <T as frame_system::Config>::AccountId,294 call: <T as Config>::RuntimeCall,295 count: u32,296 ) -> Result<(), DispatchError>;297298 /// Unreserve (unlock) a certain amount from the payer's reserved funds, returning the change.299 fn pay_for_call(300 id: ScheduledId,301 sponsor: <T as frame_system::Config>::AccountId,302 call: <T as Config>::RuntimeCall,303 ) -> Result<u128, DispatchError>;304305 /// Resolve the call dispatch, including any post-dispatch operations.306 fn dispatch_call(307 signer: Option<T::AccountId>,308 function: <T as Config>::RuntimeCall,309 ) -> Result<310 Result<PostDispatchInfo, DispatchErrorWithPostInfo<PostDispatchInfo>>,311 TransactionValidityError,312 >;313314 /// Release unspent reserved funds in case of a schedule cancel.315 fn cancel_reserve(316 id: ScheduledId,317 sponsor: <T as frame_system::Config>::AccountId,318 ) -> Result<u128, DispatchError>;319 }320321 /// Items to be executed, indexed by the block number that they should be executed on.322 #[pallet::storage]323 pub type Agenda<T: Config> =324 StorageMap<_, Twox64Concat, T::BlockNumber, Vec<Option<ScheduledV3Of<T>>>, ValueQuery>;325326 /// Lookup from identity to the block number and index of the task.327 #[pallet::storage]328 pub(crate) type Lookup<T: Config> =329 StorageMap<_, Twox64Concat, ScheduledId, TaskAddress<T::BlockNumber>>;330331 /// Events type.332 #[pallet::event]333 #[pallet::generate_deposit(pub(super) fn deposit_event)]334 pub enum Event<T: Config> {335 /// Scheduled some task.336 Scheduled { when: T::BlockNumber, index: u32 },337 /// Canceled some task.338 Canceled { when: T::BlockNumber, index: u32 },339 /// Scheduled task's priority has changed340 PriorityChanged {341 when: T::BlockNumber,342 index: u32,343 priority: schedule::Priority,344 },345 /// Dispatched some task.346 Dispatched {347 task: TaskAddress<T::BlockNumber>,348 id: Option<ScheduledId>,349 result: DispatchResult,350 },351 /// The call for the provided hash was not found so the task has been aborted.352 CallLookupFailed {353 task: TaskAddress<T::BlockNumber>,354 id: Option<ScheduledId>,355 error: LookupError,356 },357 }358359 #[pallet::error]360 pub enum Error<T> {361 /// Failed to schedule a call362 FailedToSchedule,363 /// Cannot find the scheduled call.364 NotFound,365 /// Given target block number is in the past.366 TargetBlockNumberInPast,367 /// Reschedule failed because it does not change scheduled time.368 RescheduleNoChange,369 }370371 #[pallet::hooks]372 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {373 /// Execute the scheduled calls374 fn on_initialize(now: T::BlockNumber) -> Weight {375 let limit = T::MaximumWeight::get();376377 let mut queued = Agenda::<T>::take(now)378 .into_iter()379 .enumerate()380 .filter_map(|(index, s)| Some((index as u32, s?)))381 .collect::<Vec<_>>();382383 if queued.len() as u32 > T::MaxScheduledPerBlock::get() {384 log::warn!(385 target: "runtime::scheduler",386 "Warning: This block has more items queued in Scheduler than \387 expected from the runtime configuration. An update might be needed."388 );389 }390391 queued.sort_by_key(|(_, s)| s.priority);392393 let next = now + One::one();394395 let mut total_weight: Weight = T::WeightInfo::on_initialize(0);396 for (order, (index, mut s)) in queued.into_iter().enumerate() {397 let named = s.maybe_id.is_some();398399 let (call, maybe_completed) = s.call.resolved::<T::PreimageProvider>();400 s.call = call;401402 let resolved = if let Some(completed) = maybe_completed {403 T::PreimageProvider::unrequest_preimage(&completed);404 true405 } else {406 false407 };408 let call = match s.call.as_value().cloned() {409 Some(c) => c,410 None => {411 // Preimage not available - postpone until some block.412 total_weight.saturating_accrue(T::WeightInfo::item(false, named, None));413 if let Some(delay) = T::NoPreimagePostponement::get() {414 let until = now.saturating_add(delay);415 if let Some(ref id) = s.maybe_id {416 let index = Agenda::<T>::decode_len(until).unwrap_or(0);417 Lookup::<T>::insert(id, (until, index as u32));418 }419 Agenda::<T>::append(until, Some(s));420 } else if let Some(ref id) = s.maybe_id {421 Lookup::<T>::remove(id);422 }423 continue;424 }425 };426427 let periodic = s.maybe_periodic.is_some();428 let call_weight = call.get_dispatch_info().weight;429 let mut item_weight = T::WeightInfo::item(periodic, named, Some(resolved));430 let origin = <<T as Config>::RuntimeOrigin as From<T::PalletsOrigin>>::from(431 s.origin.clone(),432 )433 .into();434 if ensure_signed(origin).is_ok() {435 // Weights of Signed dispatches expect their signing account to be whitelisted.436 item_weight.saturating_accrue(T::DbWeight::get().reads_writes(1, 1));437 }438439 // We allow a scheduled call if any is true:440 // - It's priority is `HARD_DEADLINE`441 // - It does not push the weight past the limit.442 // - It is the first item in the schedule443 let hard_deadline = s.priority <= schedule::HARD_DEADLINE;444 let test_weight = total_weight445 .saturating_add(call_weight)446 .saturating_add(item_weight);447 if !hard_deadline && order > 0 && test_weight.all_gt(limit) {448 // Cannot be scheduled this block - postpone until next.449 total_weight.saturating_accrue(T::WeightInfo::item(false, named, None));450 if let Some(ref id) = s.maybe_id {451 // NOTE: We could reasonably not do this (in which case there would be one452 // block where the named and delayed item could not be referenced by name),453 // but we will do it anyway since it should be mostly free in terms of454 // weight and it is slightly cleaner.455 let index = Agenda::<T>::decode_len(next).unwrap_or(0);456 Lookup::<T>::insert(id, (next, index as u32));457 }458 Agenda::<T>::append(next, Some(s));459 continue;460 }461462 let scheduled_origin =463 <<T as Config>::RuntimeOrigin as From<T::PalletsOrigin>>::from(464 s.origin.clone(),465 );466 let ensured_origin = T::ScheduleOrigin::ensure_origin(scheduled_origin.into());467468 let r = match ensured_origin {469 Ok(ScheduledEnsureOriginSuccess::Root) => {470 Ok(call.dispatch_bypass_filter(frame_system::RawOrigin::Root.into()))471 }472 Ok(ScheduledEnsureOriginSuccess::Signed(sender)) => {473 // Execute transaction via chain default pipeline474 // That means dispatch will be processed like any user's extrinsic e.g. transaction fees will be taken475 T::CallExecutor::dispatch_call(Some(sender), call.clone())476 }477 Ok(ScheduledEnsureOriginSuccess::Unsigned) => {478 // Unsigned version of the above479 T::CallExecutor::dispatch_call(None, call.clone())480 }481 Err(e) => Ok(Err(e.into())),482 };483484 let mut actual_call_weight: Weight = item_weight;485 let result: Result<_, DispatchError> = match r {486 Ok(o) => match o {487 Ok(di) => {488 actual_call_weight = di.actual_weight.unwrap_or(item_weight);489 Ok(())490 }491 Err(err) => Err(err.error),492 },493 Err(_) => {494 log::error!(495 target: "runtime::scheduler",496 "Warning: Scheduler has failed to execute a post-dispatch transaction. \497 This block might have become invalid.");498 Err(DispatchError::CannotLookup)499 } // todo possibly force a skip/return here, do something with the error500 };501502 total_weight.saturating_accrue(item_weight);503 total_weight.saturating_accrue(actual_call_weight);504505 Self::deposit_event(Event::Dispatched {506 task: (now, index),507 id: s.maybe_id.clone(),508 result,509 });510511 if let &Some((period, count)) = &s.maybe_periodic {512 if count > 1 {513 s.maybe_periodic = Some((period, count - 1));514 } else {515 s.maybe_periodic = None;516 }517 let wake = now + period;518 let is_canceled;519520 // If scheduled is named, place its information in `Lookup`521 if let Some(ref id) = s.maybe_id {522 is_canceled = Lookup::<T>::get(id).is_none();523 let wake_index = Agenda::<T>::decode_len(wake).unwrap_or(0);524525 if !is_canceled {526 Lookup::<T>::insert(id, (wake, wake_index as u32));527 }528 } else {529 is_canceled = false;530 }531532 if !is_canceled {533 Agenda::<T>::append(wake, Some(s));534 }535 } else if let Some(ref id) = s.maybe_id {536 Lookup::<T>::remove(id);537 }538 }539 total_weight540 }541 }542543 #[pallet::call]544 impl<T: Config> Pallet<T> {545 /// Schedule a named task.546 #[pallet::weight(<T as Config>::WeightInfo::schedule_named(T::MaxScheduledPerBlock::get()))]547 pub fn schedule_named(548 origin: OriginFor<T>,549 id: ScheduledId,550 when: T::BlockNumber,551 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,552 priority: Option<schedule::Priority>,553 call: Box<CallOrHashOf<T>>,554 ) -> DispatchResult {555 T::ScheduleOrigin::ensure_origin(origin.clone())?;556557 if priority.is_some() {558 T::PrioritySetOrigin::ensure_origin(origin.clone())?;559 }560561 let origin = <T as Config>::RuntimeOrigin::from(origin);562 Self::do_schedule_named(563 id,564 DispatchTime::At(when),565 maybe_periodic,566 priority.unwrap_or(LOWEST_PRIORITY),567 origin.caller().clone(),568 *call,569 )?;570 Ok(())571 }572573 /// Cancel a named scheduled task.574 #[pallet::weight(<T as Config>::WeightInfo::cancel_named(T::MaxScheduledPerBlock::get()))]575 pub fn cancel_named(origin: OriginFor<T>, id: ScheduledId) -> DispatchResult {576 T::ScheduleOrigin::ensure_origin(origin.clone())?;577 let origin = <T as Config>::RuntimeOrigin::from(origin);578 Self::do_cancel_named(Some(origin.caller().clone()), id)?;579 Ok(())580 }581582 /// Schedule a named task after a delay.583 ///584 /// # <weight>585 /// Same as [`schedule_named`](Self::schedule_named).586 /// # </weight>587 #[pallet::weight(<T as Config>::WeightInfo::schedule_named(T::MaxScheduledPerBlock::get()))]588 pub fn schedule_named_after(589 origin: OriginFor<T>,590 id: ScheduledId,591 after: T::BlockNumber,592 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,593 priority: Option<schedule::Priority>,594 call: Box<CallOrHashOf<T>>,595 ) -> DispatchResult {596 T::ScheduleOrigin::ensure_origin(origin.clone())?;597598 if priority.is_some() {599 T::PrioritySetOrigin::ensure_origin(origin.clone())?;600 }601602 let origin = <T as Config>::RuntimeOrigin::from(origin);603 Self::do_schedule_named(604 id,605 DispatchTime::After(after),606 maybe_periodic,607 priority.unwrap_or(LOWEST_PRIORITY),608 origin.caller().clone(),609 *call,610 )?;611 Ok(())612 }613614 #[pallet::weight(<T as Config>::WeightInfo::change_named_priority(T::MaxScheduledPerBlock::get()))]615 pub fn change_named_priority(616 origin: OriginFor<T>,617 id: ScheduledId,618 priority: schedule::Priority,619 ) -> DispatchResult {620 T::PrioritySetOrigin::ensure_origin(origin.clone())?;621 let origin = <T as Config>::RuntimeOrigin::from(origin);622 Self::do_change_named_priority(origin.caller().clone(), id, priority)623 }624 }625}626627impl<T: Config> Pallet<T> {628 #[cfg(feature = "try-runtime")]629 pub fn pre_migrate_to_v3() -> Result<(), &'static str> {630 Ok(())631 }632633 #[cfg(feature = "try-runtime")]634 pub fn post_migrate_to_v3() -> Result<(), &'static str> {635 use frame_support::dispatch::GetStorageVersion;636637 assert!(Self::current_storage_version() == 3);638 for k in Agenda::<T>::iter_keys() {639 let _ = Agenda::<T>::try_get(k).map_err(|()| "Invalid item in Agenda")?;640 }641 Ok(())642 }643644 /// Helper to migrate scheduler when the pallet origin type has changed.645 pub fn migrate_origin<OldOrigin: Into<T::PalletsOrigin> + codec::Decode>() {646 Agenda::<T>::translate::<647 Vec<Option<Scheduled<CallOrHashOf<T>, T::BlockNumber, OldOrigin, T::AccountId>>>,648 _,649 >(|_, agenda| {650 Some(651 agenda652 .into_iter()653 .map(|schedule| {654 schedule.map(|schedule| Scheduled {655 maybe_id: schedule.maybe_id,656 priority: schedule.priority,657 call: schedule.call,658 maybe_periodic: schedule.maybe_periodic,659 origin: schedule.origin.into(),660 _phantom: Default::default(),661 })662 })663 .collect::<Vec<_>>(),664 )665 });666 }667668 fn resolve_time(when: DispatchTime<T::BlockNumber>) -> Result<T::BlockNumber, DispatchError> {669 let now = frame_system::Pallet::<T>::block_number();670671 let when = match when {672 DispatchTime::At(x) => x,673 // The current block has already completed it's scheduled tasks, so674 // Schedule the task at lest one block after this current block.675 DispatchTime::After(x) => now.saturating_add(x).saturating_add(One::one()),676 };677678 if when <= now {679 return Err(Error::<T>::TargetBlockNumberInPast.into());680 }681682 Ok(when)683 }684685 fn do_schedule_named(686 id: ScheduledId,687 when: DispatchTime<T::BlockNumber>,688 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,689 priority: schedule::Priority,690 origin: T::PalletsOrigin,691 call: CallOrHashOf<T>,692 ) -> Result<TaskAddress<T::BlockNumber>, DispatchError> {693 // ensure id it is unique694 if Lookup::<T>::contains_key(&id) {695 return Err(Error::<T>::FailedToSchedule)?;696 }697698 let when = Self::resolve_time(when)?;699700 call.ensure_requested::<T::PreimageProvider>();701702 // sanitize maybe_periodic703 let maybe_periodic = maybe_periodic704 .filter(|p| p.1 > 1 && !p.0.is_zero())705 // Remove one from the number of repetitions since we will schedule one now.706 .map(|(p, c)| (p, c - 1));707708 let s = Scheduled {709 maybe_id: Some(id.clone()),710 priority,711 call: call.clone(),712 maybe_periodic,713 origin: origin.clone(),714 _phantom: Default::default(),715 };716717 // reserve balance for periodic execution718 // let sender =719 // ensure_signed(<<T as Config>::Origin as From<T::PalletsOrigin>>::from(origin).into())?;720 // let repeats = match maybe_periodic {721 // Some(p) => p.1,722 // None => 1,723 // };724 // let _ = T::CallExecutor::reserve_balance(725 // id.clone(),726 // sender,727 // call.as_value().unwrap().clone(),728 // repeats,729 // );730731 Agenda::<T>::append(when, Some(s));732 let index = Agenda::<T>::decode_len(when).unwrap_or(1) as u32 - 1;733 let address = (when, index);734 Lookup::<T>::insert(&id, &address);735 Self::deposit_event(Event::Scheduled { when, index });736737 Ok(address)738 }739740 fn do_cancel_named(origin: Option<T::PalletsOrigin>, id: ScheduledId) -> DispatchResult {741 Lookup::<T>::try_mutate_exists(id, |lookup| -> DispatchResult {742 if let Some((when, index)) = lookup.take() {743 let i = index as usize;744 Agenda::<T>::try_mutate(when, |agenda| -> DispatchResult {745 if let Some(s) = agenda.get_mut(i) {746 if let (Some(ref o), Some(ref s)) = (origin.clone(), s.borrow()) {747 if matches!(748 T::OriginPrivilegeCmp::cmp_privilege(o, &s.origin),749 Some(Ordering::Less) | None750 ) {751 return Err(BadOrigin.into());752 }753 // release balance reserve754 // let sender = ensure_signed(755 // <<T as Config>::Origin as From<T::PalletsOrigin>>::from(756 // origin.unwrap(),757 // )758 // .into(),759 // )?;760 // let _ = T::CallExecutor::cancel_reserve(id, sender);761762 s.call.ensure_unrequested::<T::PreimageProvider>();763 }764 *s = None;765 }766 Ok(())767 })?;768769 Self::deposit_event(Event::Canceled { when, index });770 Ok(())771 } else {772 Err(Error::<T>::NotFound)?773 }774 })775 }776777 fn do_change_named_priority(778 origin: T::PalletsOrigin,779 id: ScheduledId,780 priority: schedule::Priority,781 ) -> DispatchResult {782 match Lookup::<T>::get(id) {783 Some((when, index)) => {784 let i = index as usize;785 Agenda::<T>::try_mutate(when, |agenda| {786 if let Some(Some(s)) = agenda.get_mut(i) {787 if matches!(788 T::OriginPrivilegeCmp::cmp_privilege(&origin, &s.origin),789 Some(Ordering::Less) | None790 ) {791 return Err(BadOrigin.into());792 }793794 s.priority = priority;795 Self::deposit_event(Event::PriorityChanged {796 when,797 index,798 priority,799 });800 }801 Ok(())802 })803 }804 None => Err(Error::<T>::NotFound.into()),805 }806 }807}1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617// Original license:18// This file is part of Substrate.1920// Copyright (C) 2017-2022 Parity Technologies (UK) Ltd.21// SPDX-License-Identifier: Apache-2.02223// Licensed under the Apache License, Version 2.0 (the "License");24// you may not use this file except in compliance with the License.25// You may obtain a copy of the License at26//27// <http://www.apache.org/licenses/LICENSE-2.0>28//29// Unless required by applicable law or agreed to in writing, software30// distributed under the License is distributed on an "AS IS" BASIS,31// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.32// See the License for the specific language governing permissions and33// limitations under the License.3435//! # Unique scheduler36//! A Pallet for scheduling dispatches.37//!38//! - [`Config`]39//! - [`Call`]40//! - [`Pallet`]41//!42//! ## Overview43//!44//! This Pallet exposes capabilities for scheduling dispatches to occur at a45//! specified block number or at a specified period. These scheduled dispatches46//! should be named and may be canceled.47//!48//! **NOTE:** The unique scheduler is designed for deferred transaction calls by block number.49//! Any user can book a call of a certain transaction to a specific block number.50//! Also possible to book a call with a certain frequency.51//!52//! Key differences from the original pallet:53//! <https://crates.io/crates/pallet-scheduler>54//! Schedule Id restricted by 16 bytes. Identificator for booked call.55//! Priority limited by HARD DEADLINE (<= 63). Calls over maximum weight don't include to block.56//! The maximum weight that may be scheduled per block for any dispatchables of less priority than `schedule::HARD_DEADLINE`.57//! Maybe_periodic limit is 100 calls. Reserved for future sponsored transaction support.58//! At 100 calls reserved amount is not so much and this is avoid potential problems with balance locks.59//! Any account allowed to schedule any calls. Account withdraw implemented through default transaction logic.60//!61//! ## Interface62//!63//! ### Dispatchable Functions64//!65//! * `schedule_named` - augments the `schedule` interface with an additional `Vec<u8>` parameter66//! that can be used for identification.67//! * `cancel_named` - the named complement to the cancel function.6869// Ensure we're `no_std` when compiling for Wasm.70#![cfg_attr(not(feature = "std"), no_std)]7172#[cfg(feature = "runtime-benchmarks")]73mod benchmarking;7475pub mod weights;7677use sp_core::H160;78use codec::{Codec, Decode, Encode};79use frame_system::{self as system, ensure_signed};80pub use pallet::*;81use scale_info::TypeInfo;82use sp_runtime::{83 traits::{BadOrigin, One, Saturating, Zero},84 RuntimeDebug, DispatchErrorWithPostInfo,85};86use sp_std::{borrow::Borrow, cmp::Ordering, marker::PhantomData, prelude::*};8788use frame_support::{89 dispatch::{90 DispatchError, DispatchResult, Dispatchable, UnfilteredDispatchable, Parameter,91 GetDispatchInfo,92 },93 traits::{94 schedule::{self, DispatchTime, MaybeHashed},95 NamedReservableCurrency, EnsureOrigin, Get, IsType, OriginTrait, PrivilegeCmp,96 StorageVersion,97 },98 weights::{Weight},99};100101pub use weights::WeightInfo;102103/// The location of a scheduled task that can be used to remove it.104pub type TaskAddress<BlockNumber> = (BlockNumber, u32);105pub const MAX_TASK_ID_LENGTH_IN_BYTES: u8 = 16;106107pub type ScheduledId = [u8; MAX_TASK_ID_LENGTH_IN_BYTES as usize];108pub type CallOrHashOf<T> =109 MaybeHashed<<T as Config>::RuntimeCall, <T as frame_system::Config>::Hash>;110111/// Information regarding an item to be executed in the future.112#[cfg_attr(any(feature = "std", test), derive(PartialEq, Eq))]113#[derive(Clone, RuntimeDebug, Encode, Decode, TypeInfo)]114pub struct ScheduledV3<Call, BlockNumber, PalletsOrigin, AccountId> {115 /// The unique identity for this task, if there is one.116 maybe_id: Option<ScheduledId>,117 /// This task's priority.118 priority: schedule::Priority,119 /// The call to be dispatched.120 call: Call,121 /// If the call is periodic, then this points to the information concerning that.122 maybe_periodic: Option<schedule::Period<BlockNumber>>,123 /// The origin to dispatch the call.124 origin: PalletsOrigin,125 _phantom: PhantomData<AccountId>,126}127128pub type ScheduledV3Of<T> = ScheduledV3<129 CallOrHashOf<T>,130 <T as frame_system::Config>::BlockNumber,131 <T as Config>::PalletsOrigin,132 <T as frame_system::Config>::AccountId,133>;134135pub type ScheduledOf<T> = ScheduledV3Of<T>;136137/// The current version of Scheduled struct.138pub type Scheduled<Call, BlockNumber, PalletsOrigin, AccountId> =139 ScheduledV3<Call, BlockNumber, PalletsOrigin, AccountId>;140141pub enum ScheduledEnsureOriginSuccess<AccountId> {142 Root,143 Signed(AccountId),144}145146#[cfg(feature = "runtime-benchmarks")]147mod preimage_provider {148 use frame_support::traits::PreimageRecipient;149 pub trait PreimageProviderAndMaybeRecipient<H>: PreimageRecipient<H> {}150 impl<H, T: PreimageRecipient<H>> PreimageProviderAndMaybeRecipient<H> for T {}151}152153#[cfg(not(feature = "runtime-benchmarks"))]154mod preimage_provider {155 use frame_support::traits::PreimageProvider;156 pub trait PreimageProviderAndMaybeRecipient<H>: PreimageProvider<H> {}157 impl<H, T: PreimageProvider<H>> PreimageProviderAndMaybeRecipient<H> for T {}158}159160pub use preimage_provider::PreimageProviderAndMaybeRecipient;161162pub(crate) trait MarginalWeightInfo: WeightInfo {163 fn item(periodic: bool, named: bool, resolved: Option<bool>) -> Weight {164 match (periodic, named, resolved) {165 (_, false, None) => Self::on_initialize_aborted(2) - Self::on_initialize_aborted(1),166 (_, true, None) => {167 Self::on_initialize_named_aborted(2) - Self::on_initialize_named_aborted(1)168 }169 (false, false, Some(false)) => Self::on_initialize(2) - Self::on_initialize(1),170 (false, true, Some(false)) => {171 Self::on_initialize_named(2) - Self::on_initialize_named(1)172 }173 (true, false, Some(false)) => {174 Self::on_initialize_periodic(2) - Self::on_initialize_periodic(1)175 }176 (true, true, Some(false)) => {177 Self::on_initialize_periodic_named_resolved(2)178 - Self::on_initialize_periodic_named_resolved(1)179 }180 (false, false, Some(true)) => Self::on_initialize(2) - Self::on_initialize(1),181 (false, true, Some(true)) => {182 Self::on_initialize_named_resolved(2) - Self::on_initialize_named_resolved(1)183 }184 (true, false, Some(true)) => {185 Self::on_initialize_periodic_resolved(2) - Self::on_initialize_periodic_resolved(1)186 }187 (true, true, Some(true)) => {188 Self::on_initialize_periodic_named_resolved(2)189 - Self::on_initialize_periodic_named_resolved(1)190 }191 }192 }193}194impl<T: WeightInfo> MarginalWeightInfo for T {}195196#[frame_support::pallet]197pub mod pallet {198 use super::*;199 use frame_support::{200 dispatch::PostDispatchInfo,201 pallet_prelude::*,202 traits::{203 schedule::{LookupError, LOWEST_PRIORITY},204 PreimageProvider,205 },206 };207 use frame_system::pallet_prelude::*;208209 /// The current storage version.210 const STORAGE_VERSION: StorageVersion = StorageVersion::new(3);211212 #[pallet::pallet]213 #[pallet::generate_store(pub(super) trait Store)]214 #[pallet::storage_version(STORAGE_VERSION)]215 #[pallet::without_storage_info]216 pub struct Pallet<T>(_);217218 /// `system::Config` should always be included in our implied traits.219 #[pallet::config]220 pub trait Config: frame_system::Config {221 /// The overarching event type.222 type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;223224 /// The aggregated origin which the dispatch will take.225 type RuntimeOrigin: OriginTrait<PalletsOrigin = Self::PalletsOrigin>226 + From<Self::PalletsOrigin>227 + IsType<<Self as system::Config>::RuntimeOrigin>;228229 /// The caller origin, overarching type of all pallets origins.230 type PalletsOrigin: From<system::RawOrigin<Self::AccountId>> + Codec + Clone + Eq + TypeInfo;231232 type Currency: NamedReservableCurrency<Self::AccountId, ReserveIdentifier = ScheduledId>;233234 /// The aggregated call type.235 type RuntimeCall: Parameter236 + Dispatchable<237 RuntimeOrigin = <Self as Config>::RuntimeOrigin,238 PostInfo = PostDispatchInfo,239 > + UnfilteredDispatchable<RuntimeOrigin = <Self as system::Config>::RuntimeOrigin>240 + GetDispatchInfo241 + From<system::Call<Self>>;242243 /// The maximum weight that may be scheduled per block for any dispatchables of less244 /// priority than `schedule::HARD_DEADLINE`.245 #[pallet::constant]246 type MaximumWeight: Get<Weight>;247248 /// Required origin to schedule or cancel calls.249 type ScheduleOrigin: EnsureOrigin<250 <Self as system::Config>::RuntimeOrigin,251 Success = ScheduledEnsureOriginSuccess<Self::AccountId>,252 >;253254 /// Required origin to set/change calls' priority.255 type PrioritySetOrigin: EnsureOrigin<<Self as system::Config>::RuntimeOrigin>;256257 /// Compare the privileges of origins.258 ///259 /// This will be used when canceling a task, to ensure that the origin that tries260 /// to cancel has greater or equal privileges as the origin that created the scheduled task.261 ///262 /// For simplicity the [`EqualPrivilegeOnly`](frame_support::traits::EqualPrivilegeOnly) can263 /// be used. This will only check if two given origins are equal.264 type OriginPrivilegeCmp: PrivilegeCmp<Self::PalletsOrigin>;265266 /// The maximum number of scheduled calls in the queue for a single block.267 /// Not strictly enforced, but used for weight estimation.268 #[pallet::constant]269 type MaxScheduledPerBlock: Get<u32>;270271 /// Weight information for extrinsics in this pallet.272 type WeightInfo: WeightInfo;273274 /// The preimage provider with which we look up call hashes to get the call.275 type PreimageProvider: PreimageProviderAndMaybeRecipient<Self::Hash>;276277 /// If `Some` then the number of blocks to postpone execution for when the item is delayed.278 type NoPreimagePostponement: Get<Option<Self::BlockNumber>>;279280 /// Sponsoring function.281 // type SponsorshipHandler: SponsorshipHandler<Self::AccountId, <Self as Config>::Call>;282283 /// The helper type used for custom transaction fee logic.284 type CallExecutor: DispatchCall<Self, H160>;285 }286287 /// A Scheduler-Runtime interface for finer payment handling.288 pub trait DispatchCall<T: frame_system::Config + Config, SelfContainedSignedInfo> {289 /// Reserve (lock) the maximum spendings on a call, calculated from its weight and the repetition count.290 fn reserve_balance(291 id: ScheduledId,292 sponsor: <T as frame_system::Config>::AccountId,293 call: <T as Config>::RuntimeCall,294 count: u32,295 ) -> Result<(), DispatchError>;296297 /// Unreserve (unlock) a certain amount from the payer's reserved funds, returning the change.298 fn pay_for_call(299 id: ScheduledId,300 sponsor: <T as frame_system::Config>::AccountId,301 call: <T as Config>::RuntimeCall,302 ) -> Result<u128, DispatchError>;303304 /// Resolve the call dispatch, including any post-dispatch operations.305 fn dispatch_call(306 signer: Option<T::AccountId>,307 function: <T as Config>::RuntimeCall,308 ) -> Result<309 Result<PostDispatchInfo, DispatchErrorWithPostInfo<PostDispatchInfo>>,310 TransactionValidityError,311 >;312313 /// Release unspent reserved funds in case of a schedule cancel.314 fn cancel_reserve(315 id: ScheduledId,316 sponsor: <T as frame_system::Config>::AccountId,317 ) -> Result<u128, DispatchError>;318 }319320 /// Items to be executed, indexed by the block number that they should be executed on.321 #[pallet::storage]322 pub type Agenda<T: Config> =323 StorageMap<_, Twox64Concat, T::BlockNumber, Vec<Option<ScheduledV3Of<T>>>, ValueQuery>;324325 /// Lookup from identity to the block number and index of the task.326 #[pallet::storage]327 pub(crate) type Lookup<T: Config> =328 StorageMap<_, Twox64Concat, ScheduledId, TaskAddress<T::BlockNumber>>;329330 /// Events type.331 #[pallet::event]332 #[pallet::generate_deposit(pub(super) fn deposit_event)]333 pub enum Event<T: Config> {334 /// Scheduled some task.335 Scheduled { when: T::BlockNumber, index: u32 },336 /// Canceled some task.337 Canceled { when: T::BlockNumber, index: u32 },338 /// Scheduled task's priority has changed339 PriorityChanged {340 when: T::BlockNumber,341 index: u32,342 priority: schedule::Priority,343 },344 /// Dispatched some task.345 Dispatched {346 task: TaskAddress<T::BlockNumber>,347 id: Option<ScheduledId>,348 result: DispatchResult,349 },350 /// The call for the provided hash was not found so the task has been aborted.351 CallLookupFailed {352 task: TaskAddress<T::BlockNumber>,353 id: Option<ScheduledId>,354 error: LookupError,355 },356 }357358 #[pallet::error]359 pub enum Error<T> {360 /// Failed to schedule a call361 FailedToSchedule,362 /// Cannot find the scheduled call.363 NotFound,364 /// Given target block number is in the past.365 TargetBlockNumberInPast,366 /// Reschedule failed because it does not change scheduled time.367 RescheduleNoChange,368 }369370 #[pallet::hooks]371 impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {372 /// Execute the scheduled calls373 fn on_initialize(now: T::BlockNumber) -> Weight {374 let limit = T::MaximumWeight::get();375376 let mut queued = Agenda::<T>::take(now)377 .into_iter()378 .enumerate()379 .filter_map(|(index, s)| Some((index as u32, s?)))380 .collect::<Vec<_>>();381382 if queued.len() as u32 > T::MaxScheduledPerBlock::get() {383 log::warn!(384 target: "runtime::scheduler",385 "Warning: This block has more items queued in Scheduler than \386 expected from the runtime configuration. An update might be needed."387 );388 }389390 queued.sort_by_key(|(_, s)| s.priority);391392 let next = now + One::one();393394 let mut total_weight: Weight = T::WeightInfo::on_initialize(0);395 for (order, (index, mut s)) in queued.into_iter().enumerate() {396 let named = s.maybe_id.is_some();397398 let (call, maybe_completed) = s.call.resolved::<T::PreimageProvider>();399 s.call = call;400401 let resolved = if let Some(completed) = maybe_completed {402 T::PreimageProvider::unrequest_preimage(&completed);403 true404 } else {405 false406 };407 let call = match s.call.as_value().cloned() {408 Some(c) => c,409 None => {410 // Preimage not available - postpone until some block.411 total_weight.saturating_accrue(T::WeightInfo::item(false, named, None));412 if let Some(delay) = T::NoPreimagePostponement::get() {413 let until = now.saturating_add(delay);414 if let Some(ref id) = s.maybe_id {415 let index = Agenda::<T>::decode_len(until).unwrap_or(0);416 Lookup::<T>::insert(id, (until, index as u32));417 }418 Agenda::<T>::append(until, Some(s));419 } else if let Some(ref id) = s.maybe_id {420 Lookup::<T>::remove(id);421 }422 continue;423 }424 };425426 let periodic = s.maybe_periodic.is_some();427 let call_weight = call.get_dispatch_info().weight;428 let mut item_weight = T::WeightInfo::item(periodic, named, Some(resolved));429 let origin = <<T as Config>::RuntimeOrigin as From<T::PalletsOrigin>>::from(430 s.origin.clone(),431 )432 .into();433 if ensure_signed(origin).is_ok() {434 // Weights of Signed dispatches expect their signing account to be whitelisted.435 item_weight.saturating_accrue(T::DbWeight::get().reads_writes(1, 1));436 }437438 // We allow a scheduled call if any is true:439 // - It's priority is `HARD_DEADLINE`440 // - It does not push the weight past the limit.441 // - It is the first item in the schedule442 let hard_deadline = s.priority <= schedule::HARD_DEADLINE;443 let test_weight = total_weight444 .saturating_add(call_weight)445 .saturating_add(item_weight);446 if !hard_deadline && order > 0 && test_weight.all_gt(limit) {447 // Cannot be scheduled this block - postpone until next.448 total_weight.saturating_accrue(T::WeightInfo::item(false, named, None));449 if let Some(ref id) = s.maybe_id {450 // NOTE: We could reasonably not do this (in which case there would be one451 // block where the named and delayed item could not be referenced by name),452 // but we will do it anyway since it should be mostly free in terms of453 // weight and it is slightly cleaner.454 let index = Agenda::<T>::decode_len(next).unwrap_or(0);455 Lookup::<T>::insert(id, (next, index as u32));456 }457 Agenda::<T>::append(next, Some(s));458 continue;459 }460461 let scheduled_origin =462 <<T as Config>::RuntimeOrigin as From<T::PalletsOrigin>>::from(463 s.origin.clone(),464 );465 let ensured_origin = T::ScheduleOrigin::ensure_origin(scheduled_origin.into());466467 let r = match ensured_origin {468 Ok(ScheduledEnsureOriginSuccess::Root) => {469 Ok(call.dispatch_bypass_filter(frame_system::RawOrigin::Root.into()))470 }471 Ok(ScheduledEnsureOriginSuccess::Signed(sender)) => {472 // Execute transaction via chain default pipeline473 // That means dispatch will be processed like any user's extrinsic e.g. transaction fees will be taken474 T::CallExecutor::dispatch_call(Some(sender), call.clone())475 }476 Err(e) => Ok(Err(e.into())),477 };478479 let mut actual_call_weight: Weight = item_weight;480 let result: Result<_, DispatchError> = match r {481 Ok(o) => match o {482 Ok(di) => {483 actual_call_weight = di.actual_weight.unwrap_or(item_weight);484 Ok(())485 }486 Err(err) => Err(err.error),487 },488 Err(_) => {489 log::error!(490 target: "runtime::scheduler",491 "Warning: Scheduler has failed to execute a post-dispatch transaction. \492 This block might have become invalid.");493 Err(DispatchError::CannotLookup)494 } // todo possibly force a skip/return here, do something with the error495 };496497 total_weight.saturating_accrue(item_weight);498 total_weight.saturating_accrue(actual_call_weight);499500 Self::deposit_event(Event::Dispatched {501 task: (now, index),502 id: s.maybe_id.clone(),503 result,504 });505506 if let &Some((period, count)) = &s.maybe_periodic {507 if count > 1 {508 s.maybe_periodic = Some((period, count - 1));509 } else {510 s.maybe_periodic = None;511 }512 let wake = now + period;513 let is_canceled;514515 // If scheduled is named, place its information in `Lookup`516 if let Some(ref id) = s.maybe_id {517 is_canceled = Lookup::<T>::get(id).is_none();518 let wake_index = Agenda::<T>::decode_len(wake).unwrap_or(0);519520 if !is_canceled {521 Lookup::<T>::insert(id, (wake, wake_index as u32));522 }523 } else {524 is_canceled = false;525 }526527 if !is_canceled {528 Agenda::<T>::append(wake, Some(s));529 }530 } else if let Some(ref id) = s.maybe_id {531 Lookup::<T>::remove(id);532 }533 }534 total_weight535 }536 }537538 #[pallet::call]539 impl<T: Config> Pallet<T> {540 /// Schedule a named task.541 #[pallet::weight(<T as Config>::WeightInfo::schedule_named(T::MaxScheduledPerBlock::get()))]542 pub fn schedule_named(543 origin: OriginFor<T>,544 id: ScheduledId,545 when: T::BlockNumber,546 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,547 priority: Option<schedule::Priority>,548 call: Box<CallOrHashOf<T>>,549 ) -> DispatchResult {550 T::ScheduleOrigin::ensure_origin(origin.clone())?;551552 if priority.is_some() {553 T::PrioritySetOrigin::ensure_origin(origin.clone())?;554 }555556 let origin = <T as Config>::RuntimeOrigin::from(origin);557 Self::do_schedule_named(558 id,559 DispatchTime::At(when),560 maybe_periodic,561 priority.unwrap_or(LOWEST_PRIORITY),562 origin.caller().clone(),563 *call,564 )?;565 Ok(())566 }567568 /// Cancel a named scheduled task.569 #[pallet::weight(<T as Config>::WeightInfo::cancel_named(T::MaxScheduledPerBlock::get()))]570 pub fn cancel_named(origin: OriginFor<T>, id: ScheduledId) -> DispatchResult {571 T::ScheduleOrigin::ensure_origin(origin.clone())?;572 let origin = <T as Config>::RuntimeOrigin::from(origin);573 Self::do_cancel_named(Some(origin.caller().clone()), id)?;574 Ok(())575 }576577 /// Schedule a named task after a delay.578 ///579 /// # <weight>580 /// Same as [`schedule_named`](Self::schedule_named).581 /// # </weight>582 #[pallet::weight(<T as Config>::WeightInfo::schedule_named(T::MaxScheduledPerBlock::get()))]583 pub fn schedule_named_after(584 origin: OriginFor<T>,585 id: ScheduledId,586 after: T::BlockNumber,587 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,588 priority: Option<schedule::Priority>,589 call: Box<CallOrHashOf<T>>,590 ) -> DispatchResult {591 T::ScheduleOrigin::ensure_origin(origin.clone())?;592593 if priority.is_some() {594 T::PrioritySetOrigin::ensure_origin(origin.clone())?;595 }596597 let origin = <T as Config>::RuntimeOrigin::from(origin);598 Self::do_schedule_named(599 id,600 DispatchTime::After(after),601 maybe_periodic,602 priority.unwrap_or(LOWEST_PRIORITY),603 origin.caller().clone(),604 *call,605 )?;606 Ok(())607 }608609 #[pallet::weight(<T as Config>::WeightInfo::change_named_priority(T::MaxScheduledPerBlock::get()))]610 pub fn change_named_priority(611 origin: OriginFor<T>,612 id: ScheduledId,613 priority: schedule::Priority,614 ) -> DispatchResult {615 T::PrioritySetOrigin::ensure_origin(origin.clone())?;616 let origin = <T as Config>::RuntimeOrigin::from(origin);617 Self::do_change_named_priority(origin.caller().clone(), id, priority)618 }619 }620}621622impl<T: Config> Pallet<T> {623 #[cfg(feature = "try-runtime")]624 pub fn pre_migrate_to_v3() -> Result<(), &'static str> {625 Ok(())626 }627628 #[cfg(feature = "try-runtime")]629 pub fn post_migrate_to_v3() -> Result<(), &'static str> {630 use frame_support::dispatch::GetStorageVersion;631632 assert!(Self::current_storage_version() == 3);633 for k in Agenda::<T>::iter_keys() {634 let _ = Agenda::<T>::try_get(k).map_err(|()| "Invalid item in Agenda")?;635 }636 Ok(())637 }638639 /// Helper to migrate scheduler when the pallet origin type has changed.640 pub fn migrate_origin<OldOrigin: Into<T::PalletsOrigin> + codec::Decode>() {641 Agenda::<T>::translate::<642 Vec<Option<Scheduled<CallOrHashOf<T>, T::BlockNumber, OldOrigin, T::AccountId>>>,643 _,644 >(|_, agenda| {645 Some(646 agenda647 .into_iter()648 .map(|schedule| {649 schedule.map(|schedule| Scheduled {650 maybe_id: schedule.maybe_id,651 priority: schedule.priority,652 call: schedule.call,653 maybe_periodic: schedule.maybe_periodic,654 origin: schedule.origin.into(),655 _phantom: Default::default(),656 })657 })658 .collect::<Vec<_>>(),659 )660 });661 }662663 fn resolve_time(when: DispatchTime<T::BlockNumber>) -> Result<T::BlockNumber, DispatchError> {664 let now = frame_system::Pallet::<T>::block_number();665666 let when = match when {667 DispatchTime::At(x) => x,668 // The current block has already completed it's scheduled tasks, so669 // Schedule the task at lest one block after this current block.670 DispatchTime::After(x) => now.saturating_add(x).saturating_add(One::one()),671 };672673 if when <= now {674 return Err(Error::<T>::TargetBlockNumberInPast.into());675 }676677 Ok(when)678 }679680 fn do_schedule_named(681 id: ScheduledId,682 when: DispatchTime<T::BlockNumber>,683 maybe_periodic: Option<schedule::Period<T::BlockNumber>>,684 priority: schedule::Priority,685 origin: T::PalletsOrigin,686 call: CallOrHashOf<T>,687 ) -> Result<TaskAddress<T::BlockNumber>, DispatchError> {688 // ensure id it is unique689 if Lookup::<T>::contains_key(&id) {690 return Err(Error::<T>::FailedToSchedule)?;691 }692693 let when = Self::resolve_time(when)?;694695 call.ensure_requested::<T::PreimageProvider>();696697 // sanitize maybe_periodic698 let maybe_periodic = maybe_periodic699 .filter(|p| p.1 > 1 && !p.0.is_zero())700 // Remove one from the number of repetitions since we will schedule one now.701 .map(|(p, c)| (p, c - 1));702703 let s = Scheduled {704 maybe_id: Some(id.clone()),705 priority,706 call: call.clone(),707 maybe_periodic,708 origin: origin.clone(),709 _phantom: Default::default(),710 };711712 // reserve balance for periodic execution713 // let sender =714 // ensure_signed(<<T as Config>::Origin as From<T::PalletsOrigin>>::from(origin).into())?;715 // let repeats = match maybe_periodic {716 // Some(p) => p.1,717 // None => 1,718 // };719 // let _ = T::CallExecutor::reserve_balance(720 // id.clone(),721 // sender,722 // call.as_value().unwrap().clone(),723 // repeats,724 // );725726 Agenda::<T>::append(when, Some(s));727 let index = Agenda::<T>::decode_len(when).unwrap_or(1) as u32 - 1;728 let address = (when, index);729 Lookup::<T>::insert(&id, &address);730 Self::deposit_event(Event::Scheduled { when, index });731732 Ok(address)733 }734735 fn do_cancel_named(origin: Option<T::PalletsOrigin>, id: ScheduledId) -> DispatchResult {736 Lookup::<T>::try_mutate_exists(id, |lookup| -> DispatchResult {737 if let Some((when, index)) = lookup.take() {738 let i = index as usize;739 Agenda::<T>::try_mutate(when, |agenda| -> DispatchResult {740 if let Some(s) = agenda.get_mut(i) {741 if let (Some(ref o), Some(ref s)) = (origin.clone(), s.borrow()) {742 if matches!(743 T::OriginPrivilegeCmp::cmp_privilege(o, &s.origin),744 Some(Ordering::Less) | None745 ) {746 return Err(BadOrigin.into());747 }748 // release balance reserve749 // let sender = ensure_signed(750 // <<T as Config>::Origin as From<T::PalletsOrigin>>::from(751 // origin.unwrap(),752 // )753 // .into(),754 // )?;755 // let _ = T::CallExecutor::cancel_reserve(id, sender);756757 s.call.ensure_unrequested::<T::PreimageProvider>();758 }759 *s = None;760 }761 Ok(())762 })?;763764 Self::deposit_event(Event::Canceled { when, index });765 Ok(())766 } else {767 Err(Error::<T>::NotFound)?768 }769 })770 }771772 fn do_change_named_priority(773 origin: T::PalletsOrigin,774 id: ScheduledId,775 priority: schedule::Priority,776 ) -> DispatchResult {777 match Lookup::<T>::get(id) {778 Some((when, index)) => {779 let i = index as usize;780 Agenda::<T>::try_mutate(when, |agenda| {781 if let Some(Some(s)) = agenda.get_mut(i) {782 if matches!(783 T::OriginPrivilegeCmp::cmp_privilege(&origin, &s.origin),784 Some(Ordering::Less) | None785 ) {786 return Err(BadOrigin.into());787 }788789 s.priority = priority;790 Self::deposit_event(Event::PriorityChanged {791 when,792 index,793 priority,794 });795 }796 Ok(())797 })798 }799 None => Err(Error::<T>::NotFound.into()),800 }801 }802}