diff --git a/substrate/core/sr-primitives/src/generic/checked_extrinsic.rs b/substrate/core/sr-primitives/src/generic/checked_extrinsic.rs index 44a0758267468f86fc7c4d60344c998017acecec..08d7b10386721764ca2bad5cf627961b723a6b6a 100644 --- a/substrate/core/sr-primitives/src/generic/checked_extrinsic.rs +++ b/substrate/core/sr-primitives/src/generic/checked_extrinsic.rs @@ -77,14 +77,16 @@ where info: DispatchInfo, len: usize, ) -> Result<DispatchResult, DispatchError> { - let maybe_who = if let Some((id, extra)) = self.signed { - Extra::pre_dispatch(extra, &id, &self.function, info, len)?; - Some(id) + let (maybe_who, pre) = if let Some((id, extra)) = self.signed { + let pre = Extra::pre_dispatch(extra, &id, &self.function, info, len)?; + (Some(id), pre) } else { - Extra::pre_dispatch_unsigned(&self.function, info, len)?; - None + let pre = Extra::pre_dispatch_unsigned(&self.function, info, len)?; + (None, pre) }; - Ok(self.function.dispatch(Origin::from(maybe_who))) + let res = self.function.dispatch(Origin::from(maybe_who)); + Extra::post_dispatch(pre, info, len); + Ok(res) } } diff --git a/substrate/core/sr-primitives/src/generic/unchecked_extrinsic.rs b/substrate/core/sr-primitives/src/generic/unchecked_extrinsic.rs index 17157c4a19b0e43c053d07b277b6a109e439f750..cb9330cfaaff0dad15939ebd9cc42575f7cb70e6 100644 --- a/substrate/core/sr-primitives/src/generic/unchecked_extrinsic.rs +++ b/substrate/core/sr-primitives/src/generic/unchecked_extrinsic.rs @@ -236,6 +236,8 @@ mod tests { type AccountId = u64; type Call = (); type AdditionalSigned = (); + type Pre = (); + fn additional_signed(&self) -> rstd::result::Result<(), &'static str> { Ok(()) } } diff --git a/substrate/core/sr-primitives/src/traits.rs b/substrate/core/sr-primitives/src/traits.rs index 442aa9b0ac89ae45cc50085a53df02f39068c6af..7272bd3488bc7d5859e283e5f2daeb7c1284db28 100644 --- a/substrate/core/sr-primitives/src/traits.rs +++ b/substrate/core/sr-primitives/src/traits.rs @@ -808,6 +808,9 @@ pub trait SignedExtension: /// from the transaction using the `additional_signed` function. type AdditionalSigned: Encode; + /// The type that encodes information that can be passed from pre_dispatch to post-dispatch. + type Pre: Default; + /// Construct any additional data that should be in the signed payload of the transaction. Can /// also perform any pre-signature-verification checks and return an error if needed. fn additional_signed(&self) -> Result<Self::AdditionalSigned, &'static str>; @@ -830,8 +833,8 @@ pub trait SignedExtension: call: &Self::Call, info: DispatchInfo, len: usize, - ) -> Result<(), DispatchError> { - self.validate(who, call, info, len).map(|_| ()) + ) -> Result<Self::Pre, DispatchError> { + self.validate(who, call, info, len).map(|_| Self::Pre::default()) } /// Validate an unsigned transaction for the transaction queue. Normally the default @@ -848,9 +851,16 @@ pub trait SignedExtension: call: &Self::Call, info: DispatchInfo, len: usize, - ) -> Result<(), DispatchError> { - Self::validate_unsigned(call, info, len).map(|_| ()) + ) -> Result<Self::Pre, DispatchError> { + Self::validate_unsigned(call, info, len).map(|_| Self::Pre::default()) } + + /// Do any post-flight stuff for a transaction. + fn post_dispatch( + _pre: Self::Pre, + _info: DispatchInfo, + _len: usize, + ) { } } macro_rules! tuple_impl_indexed { @@ -866,6 +876,7 @@ macro_rules! tuple_impl_indexed { type AccountId = AccountId; type Call = Call; type AdditionalSigned = ($($direct::AdditionalSigned,)+); + type Pre = ($($direct::Pre,)+); fn additional_signed(&self) -> Result<Self::AdditionalSigned, &'static str> { Ok(( $(self.$index.additional_signed()?,)+ )) } @@ -885,9 +896,8 @@ macro_rules! tuple_impl_indexed { call: &Self::Call, info: DispatchInfo, len: usize, - ) -> Result<(), DispatchError> { - $(self.$index.pre_dispatch(who, call, info, len)?;)+ - Ok(()) + ) -> Result<Self::Pre, DispatchError> { + Ok(($(self.$index.pre_dispatch(who, call, info, len)?,)+)) } fn validate_unsigned( call: &Self::Call, @@ -901,9 +911,15 @@ macro_rules! tuple_impl_indexed { call: &Self::Call, info: DispatchInfo, len: usize, - ) -> Result<(), DispatchError> { - $($direct::pre_dispatch_unsigned(call, info, len)?;)+ - Ok(()) + ) -> Result<Self::Pre, DispatchError> { + Ok(($($direct::pre_dispatch_unsigned(call, info, len)?,)+)) + } + fn post_dispatch( + pre: Self::Pre, + info: DispatchInfo, + len: usize, + ) { + $($direct::post_dispatch(pre.$index, info, len);)+ } } @@ -931,6 +947,7 @@ impl SignedExtension for () { type AccountId = u64; type AdditionalSigned = (); type Call = (); + type Pre = (); fn additional_signed(&self) -> rstd::result::Result<(), &'static str> { Ok(()) } } diff --git a/substrate/node/runtime/src/lib.rs b/substrate/node/runtime/src/lib.rs index 9f247c63203acee0c9692acfee26a8185346f28d..05d6845a5703be41a2073e18894a435ce9b346c2 100644 --- a/substrate/node/runtime/src/lib.rs +++ b/substrate/node/runtime/src/lib.rs @@ -80,8 +80,8 @@ pub const VERSION: RuntimeVersion = RuntimeVersion { // and set impl_version to equal spec_version. If only runtime // implementation changes and behavior does not, then leave spec_version as // is and increment impl_version. - spec_version: 134, - impl_version: 134, + spec_version: 135, + impl_version: 135, apis: RUNTIME_API_VERSIONS, }; diff --git a/substrate/srml/balances/src/lib.rs b/substrate/srml/balances/src/lib.rs index 4d0306af28e3116295db7aa7465384f7a88ba0eb..66d0ee028495f411c7dee374012ec0b3455a0489 100644 --- a/substrate/srml/balances/src/lib.rs +++ b/substrate/srml/balances/src/lib.rs @@ -1216,6 +1216,7 @@ impl<T: Trait<I>, I: Instance + Clone + Eq> SignedExtension for TakeFees<T, I> { type AccountId = T::AccountId; type Call = T::Call; type AdditionalSigned = (); + type Pre = (); fn additional_signed(&self) -> rstd::result::Result<(), &'static str> { Ok(()) } fn validate( diff --git a/substrate/srml/system/src/lib.rs b/substrate/srml/system/src/lib.rs index f41a566724cfc9ec7b635e0a9d1ec620a09ed3b2..571ec6ad6ceb010ed5915e3ed490fdf1f03f975b 100644 --- a/substrate/srml/system/src/lib.rs +++ b/substrate/srml/system/src/lib.rs @@ -881,6 +881,7 @@ impl<T: Trait + Send + Sync> SignedExtension for CheckWeight<T> { type AccountId = T::AccountId; type Call = T::Call; type AdditionalSigned = (); + type Pre = (); fn additional_signed(&self) -> rstd::result::Result<(), &'static str> { Ok(()) } @@ -944,6 +945,7 @@ impl<T: Trait> SignedExtension for CheckNonce<T> { type AccountId = T::AccountId; type Call = T::Call; type AdditionalSigned = (); + type Pre = (); fn additional_signed(&self) -> rstd::result::Result<(), &'static str> { Ok(()) } @@ -1017,6 +1019,8 @@ impl<T: Trait + Send + Sync> SignedExtension for CheckEra<T> { type AccountId = T::AccountId; type Call = T::Call; type AdditionalSigned = T::Hash; + type Pre = (); + fn additional_signed(&self) -> Result<Self::AdditionalSigned, &'static str> { let current_u64 = <Module<T>>::block_number().saturated_into::<u64>(); let n = (self.0).0.birth(current_u64).saturated_into::<T::BlockNumber>(); @@ -1047,6 +1051,8 @@ impl<T: Trait + Send + Sync> SignedExtension for CheckGenesis<T> { type AccountId = T::AccountId; type Call = <T as Trait>::Call; type AdditionalSigned = T::Hash; + type Pre = (); + fn additional_signed(&self) -> Result<Self::AdditionalSigned, &'static str> { Ok(<Module<T>>::block_hash(T::BlockNumber::zero())) }