Skip to content

Commit

Permalink
remove duplicated arm and fix version index (paritytech#6884)
Browse files Browse the repository at this point in the history
* remove duplicated arm

* annotate the version index

* add tests

* fmt

# Conflicts:
#	xcm/Cargo.toml
  • Loading branch information
xlc authored and tgmichel committed Mar 22, 2023
1 parent 89dcd36 commit 3628ce6
Show file tree
Hide file tree
Showing 4 changed files with 207 additions and 85 deletions.
2 changes: 2 additions & 0 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 2 additions & 0 deletions xcm/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,8 @@ xcm-procedural = { path = "procedural" }

[dev-dependencies]
sp-io = { git = "https://github.com/purestake/substrate", branch = "moonbeam-polkadot-v0.9.38" }
hex = "0.4.3"
hex-literal = "0.3.4"

[features]
default = ["std"]
Expand Down
105 changes: 20 additions & 85 deletions xcm/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,9 @@ pub mod latest {
mod double_encoded;
pub use double_encoded::DoubleEncoded;

#[cfg(test)]
mod tests;

/// Maximum nesting level for XCM decoding.
pub const MAX_XCM_DECODE_DEPTH: u32 = 8;

Expand Down Expand Up @@ -73,6 +76,7 @@ pub trait TryAs<T> {

macro_rules! versioned_type {
($(#[$attr:meta])* pub enum $n:ident {
$(#[$index3:meta])+
V3($v3:ty),
}) => {
#[derive(Derivative, Encode, Decode, TypeInfo)]
Expand All @@ -86,7 +90,7 @@ macro_rules! versioned_type {
#[codec(decode_bound())]
$(#[$attr])*
pub enum $n {
#[codec(index = 0)]
$(#[$index3])*
V3($v3),
}
impl $n {
Expand Down Expand Up @@ -131,7 +135,9 @@ macro_rules! versioned_type {
};

($(#[$attr:meta])* pub enum $n:ident {
$(#[$index2:meta])+
V2($v2:ty),
$(#[$index3:meta])+
V3($v3:ty),
}) => {
#[derive(Derivative, Encode, Decode, TypeInfo)]
Expand All @@ -145,9 +151,9 @@ macro_rules! versioned_type {
#[codec(decode_bound())]
$(#[$attr])*
pub enum $n {
#[codec(index = 0)]
$(#[$index2])*
V2($v2),
#[codec(index = 1)]
$(#[$index3])*
V3($v3),
}
impl $n {
Expand Down Expand Up @@ -216,101 +222,22 @@ macro_rules! versioned_type {
}
}
};

($(#[$attr:meta])* pub enum $n:ident {
V2($v2:ty),
V3($v3:ty),
}) => {
#[derive(Derivative, Encode, Decode, TypeInfo)]
#[derivative(Clone(bound = ""), Eq(bound = ""), PartialEq(bound = ""), Debug(bound = ""))]
#[codec(encode_bound())]
#[codec(decode_bound())]
$(#[$attr])*
pub enum $n {
#[codec(index = 1)]
V2($v2),
#[codec(index = 2)]
V3($v3),
}
impl $n {
pub fn try_as<T>(&self) -> Result<&T, ()> where Self: TryAs<T> {
<Self as TryAs<T>>::try_as(&self)
}
}
impl TryAs<$v2> for $n {
fn try_as(&self) -> Result<&$v2, ()> {
match &self {
Self::V2(ref x) => Ok(x),
_ => Err(()),
}
}
}
impl TryAs<$v3> for $n {
fn try_as(&self) -> Result<&$v3, ()> {
match &self {
Self::V3(ref x) => Ok(x),
_ => Err(()),
}
}
}
impl IntoVersion for $n {
fn into_version(self, n: Version) -> Result<Self, ()> {
Ok(match n {
2 => Self::V2(self.try_into()?),
3 => Self::V3(self.try_into()?),
_ => return Err(()),
})
}
}
impl From<$v2> for $n {
fn from(x: $v2) -> Self {
$n::V2(x)
}
}
impl<T: Into<$v3>> From<T> for $n {
fn from(x: T) -> Self {
$n::V3(x.into())
}
}
impl TryFrom<$n> for $v2 {
type Error = ();
fn try_from(x: $n) -> Result<Self, ()> {
use $n::*;
match x {
V2(x) => Ok(x),
V3(x) => x.try_into(),
}
}
}
impl TryFrom<$n> for $v3 {
type Error = ();
fn try_from(x: $n) -> Result<Self, ()> {
use $n::*;
match x {
V2(x) => x.try_into(),
V3(x) => Ok(x),
}
}
}
impl MaxEncodedLen for $n {
fn max_encoded_len() -> usize {
<$v3>::max_encoded_len()
}
}
}
}

versioned_type! {
/// A single version's `Response` value, together with its version code.
pub enum VersionedAssetId {
#[codec(index = 3)]
V3(v3::AssetId),
}
}

versioned_type! {
/// A single version's `Response` value, together with its version code.
pub enum VersionedResponse {
#[codec(index = 2)]
V2(v2::Response),
#[codec(index = 3)]
V3(v3::Response),
}
}
Expand All @@ -319,31 +246,39 @@ versioned_type! {
/// A single `MultiLocation` value, together with its version code.
#[derive(Ord, PartialOrd)]
pub enum VersionedMultiLocation {
#[codec(index = 1)] // v2 is same as v1 and therefore re-using the v1 index
V2(v2::MultiLocation),
#[codec(index = 3)]
V3(v3::MultiLocation),
}
}

versioned_type! {
/// A single `InteriorMultiLocation` value, together with its version code.
pub enum VersionedInteriorMultiLocation {
#[codec(index = 2)] // while this is same as v1::Junctions, VersionedInteriorMultiLocation is introduced in v3
V2(v2::InteriorMultiLocation),
#[codec(index = 3)]
V3(v3::InteriorMultiLocation),
}
}

versioned_type! {
/// A single `MultiAsset` value, together with its version code.
pub enum VersionedMultiAsset {
#[codec(index = 1)] // v2 is same as v1 and therefore re-using the v1 index
V2(v2::MultiAsset),
#[codec(index = 3)]
V3(v3::MultiAsset),
}
}

versioned_type! {
/// A single `MultiAssets` value, together with its version code.
pub enum VersionedMultiAssets {
#[codec(index = 1)] // v2 is same as v1 and therefore re-using the v1 index
V2(v2::MultiAssets),
#[codec(index = 3)]
V3(v3::MultiAssets),
}
}
Expand Down
183 changes: 183 additions & 0 deletions xcm/src/tests.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,183 @@
// Copyright 2021 Parity Technologies (UK) Ltd.
// This file is part of Polkadot.

// Polkadot is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Polkadot is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Polkadot. If not, see <http://www.gnu.org/licenses/>.

use crate::*;
use alloc::vec;

#[test]
fn encode_decode_versioned_asset_id_v3() {
let asset_id = VersionedAssetId::V3(v3::AssetId::Abstract([1; 32]));
let encoded = asset_id.encode();

assert_eq!(
encoded,
hex_literal::hex!("03010101010101010101010101010101010101010101010101010101010101010101"),
"encode format changed"
);
assert_eq!(encoded[0], 3, "bad version number");

let decoded = VersionedAssetId::decode(&mut &encoded[..]).unwrap();
assert_eq!(asset_id, decoded);
}

#[test]
fn encode_decode_versioned_response_v2() {
let response = VersionedResponse::V2(v2::Response::Null);
let encoded = response.encode();

assert_eq!(encoded, hex_literal::hex!("0200"), "encode format changed");
assert_eq!(encoded[0], 2, "bad version number");

let decoded = VersionedResponse::decode(&mut &encoded[..]).unwrap();
assert_eq!(response, decoded);
}

#[test]
fn encode_decode_versioned_response_v3() {
let response = VersionedResponse::V3(v3::Response::Null);
let encoded = response.encode();

assert_eq!(encoded, hex_literal::hex!("0300"), "encode format changed");
assert_eq!(encoded[0], 3, "bad version number");

let decoded = VersionedResponse::decode(&mut &encoded[..]).unwrap();
assert_eq!(response, decoded);
}

#[test]
fn encode_decode_versioned_multi_location_v2() {
let location = VersionedMultiLocation::V2(v2::MultiLocation::new(0, v2::Junctions::Here));
let encoded = location.encode();

assert_eq!(encoded, hex_literal::hex!("010000"), "encode format changed");
assert_eq!(encoded[0], 1, "bad version number"); // this is introduced in v1

let decoded = VersionedMultiLocation::decode(&mut &encoded[..]).unwrap();
assert_eq!(location, decoded);
}

#[test]
fn encode_decode_versioned_multi_location_v3() {
let location = VersionedMultiLocation::V3(v3::MultiLocation::new(0, v3::Junctions::Here));
let encoded = location.encode();

assert_eq!(encoded, hex_literal::hex!("030000"), "encode format changed");
assert_eq!(encoded[0], 3, "bad version number");

let decoded = VersionedMultiLocation::decode(&mut &encoded[..]).unwrap();
assert_eq!(location, decoded);
}

#[test]
fn encode_decode_versioned_interior_multi_location_v2() {
let location = VersionedInteriorMultiLocation::V2(v2::InteriorMultiLocation::Here);
let encoded = location.encode();

assert_eq!(encoded, hex_literal::hex!("0200"), "encode format changed");
assert_eq!(encoded[0], 2, "bad version number");

let decoded = VersionedInteriorMultiLocation::decode(&mut &encoded[..]).unwrap();
assert_eq!(location, decoded);
}

#[test]
fn encode_decode_versioned_interior_multi_location_v3() {
let location = VersionedInteriorMultiLocation::V3(v3::InteriorMultiLocation::Here);
let encoded = location.encode();

assert_eq!(encoded, hex_literal::hex!("0300"), "encode format changed");
assert_eq!(encoded[0], 3, "bad version number");

let decoded = VersionedInteriorMultiLocation::decode(&mut &encoded[..]).unwrap();
assert_eq!(location, decoded);
}

#[test]
fn encode_decode_versioned_multi_asset_v2() {
let asset = VersionedMultiAsset::V2(v2::MultiAsset::from(((0, v2::Junctions::Here), 1)));
let encoded = asset.encode();

assert_eq!(encoded, hex_literal::hex!("010000000004"), "encode format changed");
assert_eq!(encoded[0], 1, "bad version number");

let decoded = VersionedMultiAsset::decode(&mut &encoded[..]).unwrap();
assert_eq!(asset, decoded);
}

#[test]
fn encode_decode_versioned_multi_asset_v3() {
let asset = VersionedMultiAsset::V3(v3::MultiAsset::from((v3::MultiLocation::default(), 1)));
let encoded = asset.encode();

assert_eq!(encoded, hex_literal::hex!("030000000004"), "encode format changed");
assert_eq!(encoded[0], 3, "bad version number");

let decoded = VersionedMultiAsset::decode(&mut &encoded[..]).unwrap();
assert_eq!(asset, decoded);
}

#[test]
fn encode_decode_versioned_multi_assets_v2() {
let assets = VersionedMultiAssets::V2(v2::MultiAssets::from(vec![v2::MultiAsset::from((
(0, v2::Junctions::Here),
1,
))]));
let encoded = assets.encode();

assert_eq!(encoded, hex_literal::hex!("01040000000004"), "encode format changed");
assert_eq!(encoded[0], 1, "bad version number");

let decoded = VersionedMultiAssets::decode(&mut &encoded[..]).unwrap();
assert_eq!(assets, decoded);
}

#[test]
fn encode_decode_versioned_multi_assets_v3() {
let assets = VersionedMultiAssets::V3(v3::MultiAssets::from(vec![
(v3::MultiAsset::from((v3::MultiLocation::default(), 1))),
]));
let encoded = assets.encode();

assert_eq!(encoded, hex_literal::hex!("03040000000004"), "encode format changed");
assert_eq!(encoded[0], 3, "bad version number");

let decoded = VersionedMultiAssets::decode(&mut &encoded[..]).unwrap();
assert_eq!(assets, decoded);
}

#[test]
fn encode_decode_versioned_xcm_v2() {
let xcm = VersionedXcm::V2(v2::Xcm::<()>::new());
let encoded = xcm.encode();

assert_eq!(encoded, hex_literal::hex!("0200"), "encode format changed");
assert_eq!(encoded[0], 2, "bad version number");

let decoded = VersionedXcm::decode(&mut &encoded[..]).unwrap();
assert_eq!(xcm, decoded);
}

#[test]
fn encode_decode_versioned_xcm_v3() {
let xcm = VersionedXcm::V3(v3::Xcm::<()>::new());
let encoded = xcm.encode();

assert_eq!(encoded, hex_literal::hex!("0300"), "encode format changed");
assert_eq!(encoded[0], 3, "bad version number");

let decoded = VersionedXcm::decode(&mut &encoded[..]).unwrap();
assert_eq!(xcm, decoded);
}

0 comments on commit 3628ce6

Please sign in to comment.