Transfer asset via bridge

Cargo.toml

@@ -30,6 +30,7 @@ members = [
 	"primitives/utility",
 	"polkadot-parachain",
 	"parachains/common",
+	"parachains/pallets/bridge-transfer",
 	"parachains/pallets/parachain-info",
 	"parachains/pallets/ping",
 	"parachains/runtimes/testing/rococo-parachain",

pallets/parachain-system/src/lib.rs

@@ -1042,6 +1042,30 @@ impl<T: Config> Pallet<T> {
 	pub fn set_custom_validation_head_data(head_data: Vec<u8>) {
 		CustomValidationHeadData::<T>::put(head_data);
 	}
+
+	/// Open HRMP channel for using it in benchmarks.
+	///
+	/// The caller assumes that the pallet will accept regular outbound message to the sibling
+	/// `target_parachain` after this call. No other assumptions are made.
+	#[cfg(feature = "runtime-benchmarks")]
+	pub fn open_outbound_hrmp_channel_for_benchmarks(target_parachain: ParaId) {
+		RelevantMessagingState::<T>::put(MessagingStateSnapshot {
+			dmq_mqc_head: Default::default(),
+			relay_dispatch_queue_size: Default::default(),
+			ingress_channels: Default::default(),
+			egress_channels: vec![(
+				target_parachain,
+				cumulus_primitives_core::AbridgedHrmpChannel {
+					max_capacity: 10,
+					max_total_size: 10_000_000_u32,
+					max_message_size: 10_000_000_u32,
+					msg_count: 10,
+					total_size: 10_000_000_u32,
+					mqc_head: None,
+				},
+			)],
+		})
+	}
 }
 
 pub struct ParachainSetCode<T>(sp_std::marker::PhantomData<T>);

parachains/pallets/bridge-transfer/Cargo.toml

@@ -0,0 +1,58 @@
+[package]
+name = "pallet-bridge-transfer"
+version = "0.1.0"
+authors = ["Parity Technologies <admin@parity.io>"]
+edition = "2021"
+license = "Apache-2.0"
+homepage = "https://docs.substrate.io/"
+repository = "https://github.com/paritytech/cumulus/"
+description = "Pallet for message transfer through bridges"
+readme = "README.md"
+
+[dependencies]
+codec = { package = "parity-scale-codec", version = "3.0.0", default-features = false, features = ["derive"] }
+scale-info = { version = "2.3.0", default-features = false, features = ["derive"] }
+log = { version = "0.4.14", default-features = false }
+
+# Substrate
+sp-std = { git = "https://github.com/paritytech/substrate", default-features = false, branch = "master" }
+sp-runtime = { git = "https://github.com/paritytech/substrate", default-features = false, branch = "master" }
+frame-benchmarking = { git = "https://github.com/paritytech/substrate", optional = true, default-features = false, branch = "master" }
+frame-system-benchmarking = { git = "https://github.com/paritytech/substrate", optional = true, default-features = false, branch = "master" }
+frame-support = { git = "https://github.com/paritytech/substrate", default-features = false, branch = "master" }
+frame-system = { git = "https://github.com/paritytech/substrate", default-features = false, branch = "master" }
+
+# Polkadot
+xcm = { git = "https://github.com/paritytech/polkadot", default-features = false, branch = "master" }
+xcm-builder = { git = "https://github.com/paritytech/polkadot", default-features = false, branch = "master" }
+xcm-executor = { git = "https://github.com/paritytech/polkadot", default-features = false, branch = "master" }
+
+[dev-dependencies]
+sp-version = { git = "https://github.com/paritytech/substrate", branch = "master" }
+pallet-balances = { git = "https://github.com/paritytech/substrate", branch = "master" }
+polkadot-parachain = { git = "https://github.com/paritytech/polkadot", branch = "master" }
+cumulus-pallet-xcmp-queue = { path = "../../../pallets/xcmp-queue" }
+
+[features]
+default = ["std"]
+std = [
+	"codec/std",
+	"log/std",
+	"scale-info/std",
+	"sp-std/std",
+	"sp-runtime/std",
+	"frame-support/std",
+	"frame-system/std",
+	"xcm/std",
+	"xcm-builder/std",
+	"xcm-executor/std",
+]
+runtime-benchmarks = [
+	"frame-benchmarking/runtime-benchmarks",
+	"frame-support/runtime-benchmarks",
+	"frame-system-benchmarking/runtime-benchmarks",
+	"frame-system/runtime-benchmarks",
+	"sp-runtime/runtime-benchmarks",
+	"xcm-builder/runtime-benchmarks",
+]
+try-runtime = ["frame-support/try-runtime"]

parachains/pallets/bridge-transfer/src/benchmarking.rs

@@ -0,0 +1,199 @@
+// This file is part of Substrate.
+
+// Copyright (C) 2022 Parity Technologies (UK) Ltd.
+// SPDX-License-Identifier: Apache-2.0
+
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//  http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+//! `BridgeTransfer` pallet benchmarks.
+
+use crate::{
+	AllowedExporters, AllowedReserveLocations, AllowedUniversalAliases, BenchmarkHelper, Call,
+	Config, Event, Pallet, PingMessageBuilder,
+};
+
+use frame_benchmarking::{benchmarks, BenchmarkError};
+use frame_support::{
+	ensure,
+	traits::{EnsureOrigin, Get},
+};
+use sp_std::prelude::*;
+use xcm::prelude::*;
+
+#[cfg(feature = "runtime-benchmarks")]
+impl<T: Config> Pallet<T> {
+	#[cfg(feature = "runtime-benchmarks")]
+	pub fn insert_universal_alias_for_benchmarks((location, junction): (MultiLocation, Junction)) {
+		assert!(matches!(
+			AllowedUniversalAliases::<T>::try_mutate(location, |junctions| junctions
+				.try_insert(junction)),
+			Ok(true)
+		));
+	}
+}
+
+benchmarks! {
+	transfer_asset_via_bridge {
+		let _ = T::TransferAssetOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+		// every asset has its own configuration and ledger, so there's a performance dependency
+		// (be sure to use "worst" of assets)
+		let max_assets_limit = T::MaxAssetsLimit::get();
+		ensure!(max_assets_limit > 0, "MaxAssetsLimit not set up correctly.");
+		let (bridged_network, bridge_config) = T::BenchmarkHelper::bridge_config()
+			.ok_or(BenchmarkError::Stop("missing `bridge_config` data"))?;
+		let (origin, assets, destination) = T::BenchmarkHelper::prepare_asset_transfer()
+			.ok_or(BenchmarkError::Stop("missing `prepare_asset_transfer` data"))?;
+		let assets_count = match &assets {
+			VersionedMultiAssets::V2(assets) => assets.len(),
+			VersionedMultiAssets::V3(assets) => assets.len(),
+		};
+		ensure!(assets_count == max_assets_limit as usize, "`assets` not set up correctly for worst case.");
+		AllowedExporters::<T>::insert(bridged_network, bridge_config);
+	}: _<T::RuntimeOrigin>(origin, Box::new(assets), Box::new(destination))
+	verify {
+		// we don't care about message hash or sender cost here, just check that the transfer has been initiated
+		let actual_event = frame_system::Pallet::<T>::events().pop().map(|r| r.event);
+		let expected_event: <T as Config>::RuntimeEvent = Event::TransferInitiated {
+			message_hash: Default::default(),
+			sender_cost: Default::default(),
+		}.into();
+		assert!(matches!(actual_event, Some(expected_event)));
+	}
+
+	ping_via_bridge {
+		let _ = T::TransferPingOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+
+		let (bridged_network, bridge_config) = T::BenchmarkHelper::bridge_config()
+			.ok_or(BenchmarkError::Stop("missing `bridge_config` data"))?;
+		AllowedExporters::<T>::insert(bridged_network, bridge_config);
+
+		let (origin, destination) = T::BenchmarkHelper::prepare_ping_transfer()
+			.ok_or(BenchmarkError::Stop("missing `prepare_ping_transfer` data"))?;
+
+		let origin_location = T::TransferPingOrigin::ensure_origin(origin.clone()).map_err(|_| BenchmarkError::Stop("invalid `origin`"),
+		)?;
+		let (_, _, destination_location) = Pallet::<T>::ensure_remote_destination(destination.clone()).map_err(|_|
+			BenchmarkError::Stop("invalid `destination_location`"),
+		)?;
+		let _ = T::PingMessageBuilder::try_build(&origin_location, &bridged_network, &destination_location).ok_or(
+			BenchmarkError::Stop("invalid `PingMessageBuilder`"),
+		)?;
+	}: _<T::RuntimeOrigin>(origin, Box::new(destination))
+	verify {
+		// we don't care about message hash or sender cost here, just check that the transfer has been initiated
+		let actual_event = frame_system::Pallet::<T>::events().pop().map(|r| r.event);
+		let expected_event: <T as Config>::RuntimeEvent = Event::TransferInitiated {
+			message_hash: Default::default(),
+			sender_cost: Default::default(),
+		}.into();
+		assert!(matches!(actual_event, Some(expected_event)));
+	}
+
+	add_exporter_config {
+		let origin = T::AdminOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+		let (bridged_network, bridge_config) = T::BenchmarkHelper::bridge_config()
+			.ok_or(BenchmarkError::Stop("missing `bridge_config` data"))?;
+	}: _<T::RuntimeOrigin>(origin, bridged_network, Box::new(bridge_config.clone()))
+	verify {
+		assert_eq!(AllowedExporters::<T>::get(bridged_network), Some(bridge_config));
+	}
+
+	remove_exporter_config {
+		let origin = T::AdminOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+		let (bridged_network, bridge_config) = T::BenchmarkHelper::bridge_config()
+			.ok_or(BenchmarkError::Stop("missing `bridge_config` data"))?;
+		AllowedExporters::<T>::insert(bridged_network, bridge_config);
+	}: _<T::RuntimeOrigin>(origin, bridged_network)
+	verify {
+		assert_eq!(AllowedExporters::<T>::get(bridged_network), None);
+	}
+
+	update_exporter_config {
+		let origin = T::AdminOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+		let (bridged_network, bridge_config) = T::BenchmarkHelper::bridge_config()
+			.ok_or(BenchmarkError::Stop("missing `bridge_config` data"))?;
+		AllowedExporters::<T>::insert(bridged_network, bridge_config);
+
+		let bridge_location_fee = None;
+		let target_location_fee = Some(xcm::VersionedMultiAsset::V3(MultiAsset {
+			id: Concrete(MultiLocation::parent()),
+			fun: Fungible(1_000_0000),
+		}));
+	}: _<T::RuntimeOrigin>(origin, bridged_network, bridge_location_fee.clone().map(Box::new), target_location_fee.clone().map(Box::new))
+	verify {
+		let exporter = AllowedExporters::<T>::get(bridged_network).unwrap();
+		assert_eq!(exporter.bridge_location_fee, bridge_location_fee.map(|fee| MultiAsset::try_from(fee).unwrap()));
+		assert_eq!(exporter.target_location_fee, target_location_fee.map(|fee| MultiAsset::try_from(fee).unwrap()));
+	}
+
+	add_universal_alias {
+		let origin = T::AdminOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+		let (location, junction) = match T::BenchmarkHelper::universal_alias() {
+			Some(alias) => alias,
+			None => match T::UniversalAliasesLimit::get() > 0_u32 {
+				true => return Err(BenchmarkError::Stop("missing `universal_alias` data")),
+				false => return Err(BenchmarkError::Weightless),
+			}
+		};
+	}: _<T::RuntimeOrigin>(origin, Box::new(location.clone()), junction)
+	verify {
+		assert!(AllowedUniversalAliases::<T>::get(&location.try_as().unwrap()).contains(&junction));
+	}
+
+	remove_universal_alias {
+		let origin = T::AdminOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+		let (location, junction) = match T::BenchmarkHelper::universal_alias() {
+			Some(alias) => alias,
+			None => match T::UniversalAliasesLimit::get() > 0_u32 {
+				true => return Err(BenchmarkError::Stop("missing `universal_alias` data")),
+				false => return Err(BenchmarkError::Weightless),
+			}
+		};
+		Pallet::<T>::insert_universal_alias_for_benchmarks((location.clone().try_into().unwrap(), junction));
+	}: _<T::RuntimeOrigin>(origin, Box::new(location.clone()), vec![junction.clone()])
+	verify {
+		assert!(!AllowedUniversalAliases::<T>::get(&location.try_as().unwrap()).contains(&junction));
+	}
+
+	add_reserve_location {
+		let origin = T::AdminOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+		let location = match T::BenchmarkHelper::reserve_location() {
+			Some(location) => location,
+			None => match T::ReserveLocationsLimit::get() > 0_u32 {
+				true => return Err(BenchmarkError::Stop("missing `reserve_location` data")),
+				false => return Err(BenchmarkError::Weightless),
+			}
+		};
+	}: _<T::RuntimeOrigin>(origin, Box::new(location.clone()))
+	verify {
+		assert!(AllowedReserveLocations::<T>::get().contains(&location.try_as().unwrap()));
+	}
+
+	remove_reserve_location {
+		let origin = T::AdminOrigin::try_successful_origin().map_err(|_| BenchmarkError::Weightless)?;
+		let location = match T::BenchmarkHelper::reserve_location() {
+			Some(location) => location,
+			None => match T::ReserveLocationsLimit::get() > 0_u32 {
+				true => return Err(BenchmarkError::Stop("missing `reserve_location` data")),
+				false => return Err(BenchmarkError::Weightless),
+			}
+		};
+		let multilocation: MultiLocation = location.clone().try_into().unwrap();
+		assert!(AllowedReserveLocations::<T>::try_mutate(|locations| locations.try_insert(multilocation)).unwrap());
+	}: _<T::RuntimeOrigin>(origin, vec![location.clone()])
+	verify {
+		assert!(!AllowedReserveLocations::<T>::get().contains(&location.try_as().unwrap()));
+	}
+
+	impl_benchmark_test_suite!(Pallet, crate::tests::new_test_ext(), crate::tests::TestRuntime);
+}