Skip to content

Commit

Permalink
feat: add ApplyAuthorizedUpgradeOrigin
Browse files Browse the repository at this point in the history
  • Loading branch information
mrshiposha committed Jan 9, 2024
1 parent 0ff3f4d commit fa2a50c
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 3 deletions.
3 changes: 2 additions & 1 deletion cumulus/pallets/parachain-system/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -693,9 +693,10 @@ pub mod pallet {
note = "To be removed after June 2024. Migrate to `frame_system::apply_authorized_upgrade`."
)]
pub fn enact_authorized_upgrade(
_: OriginFor<T>,
origin: OriginFor<T>,
code: Vec<u8>,
) -> DispatchResultWithPostInfo {
<T as frame_system::Config>::ApplyAuthorizedUpgradeOrigin::ensure_origin(origin)?;
let post = frame_system::Pallet::<T>::do_apply_authorize_upgrade(code)?;
Ok(post)
}
Expand Down
23 changes: 21 additions & 2 deletions substrate/frame/system/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -257,7 +257,7 @@ pub mod pallet {

/// Default implementations of [`DefaultConfig`], which can be used to implement [`Config`].
pub mod config_preludes {
use super::{inject_runtime_type, DefaultConfig};
use super::{inject_runtime_type, DefaultConfig, EnsureAlways};
use frame_support::derive_impl;

/// Provides a viable default config that can be used with
Expand Down Expand Up @@ -298,6 +298,7 @@ pub mod pallet {
type BaseCallFilter = frame_support::traits::Everything;
type BlockHashCount = frame_support::traits::ConstU64<10>;
type OnSetCode = ();
type ApplyAuthorizedUpgradeOrigin = EnsureAlways;
}

/// Default configurations of this pallet in a solo-chain environment.
Expand Down Expand Up @@ -392,6 +393,8 @@ pub mod pallet {

/// The set code logic, just the default since we're not a parachain.
type OnSetCode = ();

type ApplyAuthorizedUpgradeOrigin = EnsureAlways;
}

/// Default configurations of this pallet in a relay-chain environment.
Expand Down Expand Up @@ -568,6 +571,8 @@ pub mod pallet {
#[pallet::no_default_bounds]
type OnSetCode: SetCode<Self>;

type ApplyAuthorizedUpgradeOrigin: EnsureOrigin<Self::RuntimeOrigin>;

/// The maximum number of consumers allowed on a single account.
type MaxConsumers: ConsumerLimits;
}
Expand Down Expand Up @@ -757,9 +762,10 @@ pub mod pallet {
#[pallet::call_index(11)]
#[pallet::weight((T::SystemWeightInfo::apply_authorized_upgrade(), DispatchClass::Operational))]
pub fn apply_authorized_upgrade(
_: OriginFor<T>,
origin: OriginFor<T>,
code: Vec<u8>,
) -> DispatchResultWithPostInfo {
T::ApplyAuthorizedUpgradeOrigin::ensure_origin(origin)?;
let post = Self::do_apply_authorize_upgrade(code)?;
Ok(post)
}
Expand Down Expand Up @@ -1259,6 +1265,19 @@ impl_ensure_origin_with_arg_ignoring_arg! {
{}
}

pub struct EnsureAlways;
impl<O> EnsureOrigin<O> for EnsureAlways {
type Success = ();
fn try_origin(_: O) -> Result<(), O> {
Ok(())
}

#[cfg(feature = "runtime-benchmarks")]
fn try_successful_origin() -> Result<O, ()> {
Ok(O::from(RawOrigin::None))
}
}

#[docify::export]
/// Ensure that the origin `o` represents a signed extrinsic (i.e. transaction).
/// Returns `Ok` with the account that signed the extrinsic or an `Err` otherwise.
Expand Down

0 comments on commit fa2a50c

Please sign in to comment.