Skip to content

Commit

Permalink
Box messages proof argument in receive_messages_proof call (#2222)
Browse files Browse the repository at this point in the history
* Box messages proof argument in receive_messages_proof call

* fix benchmark tests
  • Loading branch information
svyatonik authored and bkontur committed Jun 4, 2024
1 parent cb52db0 commit 15a50af
Show file tree
Hide file tree
Showing 7 changed files with 21 additions and 21 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1142,7 +1142,7 @@ pub(crate) mod tests {
fn message_delivery_call(best_message: MessageNonce) -> RuntimeCall {
RuntimeCall::BridgeMessages(MessagesCall::receive_messages_proof {
relayer_id_at_bridged_chain: relayer_account_at_bridged_chain(),
proof: FromBridgedChainMessagesProof {
proof: Box::new(FromBridgedChainMessagesProof {
bridged_header_hash: Default::default(),
storage: Default::default(),
lane: TestLaneId::get(),
Expand All @@ -1153,7 +1153,7 @@ pub(crate) mod tests {
.last_delivered_nonce() +
1,
nonces_end: best_message,
},
}),
messages_count: 1,
dispatch_weight: Weight::zero(),
})
Expand Down
4 changes: 2 additions & 2 deletions bridges/bin/runtime-common/src/messages_call_ext.rs
Original file line number Diff line number Diff line change
Expand Up @@ -415,13 +415,13 @@ mod tests {
messages_count: nonces_end.checked_sub(nonces_start).map(|x| x + 1).unwrap_or(0)
as u32,
dispatch_weight: frame_support::weights::Weight::zero(),
proof: FromBridgedChainMessagesProof {
proof: Box::new(FromBridgedChainMessagesProof {
bridged_header_hash: Default::default(),
storage: Default::default(),
lane: TEST_LANE_ID,
nonces_start,
nonces_end,
},
}),
},
)
.check_obsolete_call()
Expand Down
10 changes: 5 additions & 5 deletions bridges/modules/messages/src/benchmarking.rs
Original file line number Diff line number Diff line change
Expand Up @@ -224,7 +224,7 @@ mod benchmarks {
receive_messages_proof(
RawOrigin::Signed(setup.relayer_id_on_tgt()),
setup.relayer_id_on_src(),
proof,
Box::new(proof),
setup.msgs_count,
dispatch_weight,
);
Expand Down Expand Up @@ -255,7 +255,7 @@ mod benchmarks {
receive_messages_proof(
RawOrigin::Signed(setup.relayer_id_on_tgt()),
setup.relayer_id_on_src(),
proof,
Box::new(proof),
setup.msgs_count,
dispatch_weight,
);
Expand Down Expand Up @@ -296,7 +296,7 @@ mod benchmarks {
receive_messages_proof(
RawOrigin::Signed(setup.relayer_id_on_tgt()),
setup.relayer_id_on_src(),
proof,
Box::new(proof),
setup.msgs_count,
dispatch_weight,
);
Expand Down Expand Up @@ -331,7 +331,7 @@ mod benchmarks {
receive_messages_proof(
RawOrigin::Signed(setup.relayer_id_on_tgt()),
setup.relayer_id_on_src(),
proof,
Box::new(proof),
setup.msgs_count,
dispatch_weight,
);
Expand Down Expand Up @@ -531,7 +531,7 @@ mod benchmarks {
receive_messages_proof(
RawOrigin::Signed(setup.relayer_id_on_tgt()),
setup.relayer_id_on_src(),
proof,
Box::new(proof),
setup.msgs_count,
dispatch_weight,
);
Expand Down
8 changes: 4 additions & 4 deletions bridges/modules/messages/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -194,11 +194,11 @@ pub mod pallet {
/// The call may succeed, but some messages may not be delivered e.g. if they are not fit
/// into the unrewarded relayers vector.
#[pallet::call_index(2)]
#[pallet::weight(T::WeightInfo::receive_messages_proof_weight(proof, *messages_count, *dispatch_weight))]
#[pallet::weight(T::WeightInfo::receive_messages_proof_weight(&**proof, *messages_count, *dispatch_weight))]
pub fn receive_messages_proof(
origin: OriginFor<T>,
relayer_id_at_bridged_chain: AccountIdOf<BridgedChainOf<T, I>>,
proof: FromBridgedChainMessagesProof<HashOf<BridgedChainOf<T, I>>>,
proof: Box<FromBridgedChainMessagesProof<HashOf<BridgedChainOf<T, I>>>>,
messages_count: u32,
dispatch_weight: Weight,
) -> DispatchResultWithPostInfo {
Expand Down Expand Up @@ -226,14 +226,14 @@ pub mod pallet {
// The DeclaredWeight is exactly what's computed here. Unfortunately it is impossible
// to get pre-computed value (and it has been already computed by the executive).
let declared_weight = T::WeightInfo::receive_messages_proof_weight(
&proof,
&*proof,
messages_count,
dispatch_weight,
);
let mut actual_weight = declared_weight;

// verify messages proof && convert proof into messages
let messages = verify_and_decode_messages_proof::<T, I>(proof, messages_count)
let messages = verify_and_decode_messages_proof::<T, I>(*proof, messages_count)
.map_err(|err| {
log::trace!(target: LOG_TARGET, "Rejecting invalid messages proof: {:?}", err,);

Expand Down
8 changes: 4 additions & 4 deletions bridges/modules/messages/src/tests/mock.rs
Original file line number Diff line number Diff line change
Expand Up @@ -223,7 +223,7 @@ impl crate::benchmarking::Config<()> for TestRuntime {
let dispatch_weight =
REGULAR_PAYLOAD.declared_weight * params.message_nonces.checked_len().unwrap_or(0);
(
prepare_messages_proof(
*prepare_messages_proof(
params.message_nonces.into_iter().map(|n| message(n, REGULAR_PAYLOAD)).collect(),
params.outbound_lane_data,
),
Expand Down Expand Up @@ -473,7 +473,7 @@ pub fn run_test<T>(test: impl FnOnce() -> T) -> T {
pub fn prepare_messages_proof(
messages: Vec<Message>,
outbound_lane_data: Option<OutboundLaneData>,
) -> FromBridgedChainMessagesProof<BridgedHeaderHash> {
) -> Box<FromBridgedChainMessagesProof<BridgedHeaderHash>> {
// first - let's generate storage proof
let lane = messages.first().unwrap().key.lane_id;
let nonces_start = messages.first().unwrap().key.nonce;
Expand All @@ -497,13 +497,13 @@ pub fn prepare_messages_proof(
StoredHeaderData { number: 0, state_root: storage_root },
);

FromBridgedChainMessagesProof::<BridgedHeaderHash> {
Box::new(FromBridgedChainMessagesProof::<BridgedHeaderHash> {
bridged_header_hash,
storage,
lane,
nonces_start,
nonces_end,
}
})
}

/// Prepare valid storage proof for given messages and insert appropriate header to the
Expand Down
6 changes: 3 additions & 3 deletions bridges/modules/messages/src/tests/pallet_tests.rs
Original file line number Diff line number Diff line change
Expand Up @@ -666,7 +666,7 @@ fn ref_time_refund_from_receive_messages_proof_works() {
let messages_count = 1;
let pre_dispatch_weight =
<TestRuntime as Config>::WeightInfo::receive_messages_proof_weight(
&proof,
&*proof,
messages_count,
REGULAR_PAYLOAD.declared_weight,
);
Expand Down Expand Up @@ -721,7 +721,7 @@ fn proof_size_refund_from_receive_messages_proof_works() {
let messages_count = 1;
let pre_dispatch_weight =
<TestRuntime as Config>::WeightInfo::receive_messages_proof_weight(
&proof,
&*proof,
messages_count,
REGULAR_PAYLOAD.declared_weight,
);
Expand Down Expand Up @@ -898,7 +898,7 @@ fn test_bridge_messages_call_is_correctly_defined() {
FromBridgedChainMessagesDeliveryProof<BridgedHeaderHash>,
>::receive_messages_proof {
relayer_id_at_bridged_chain: account_id,
proof: message_proof,
proof: *message_proof,
messages_count: 1,
dispatch_weight: REGULAR_PAYLOAD.declared_weight,
};
Expand Down
2 changes: 1 addition & 1 deletion bridges/relays/lib-substrate-relay/src/messages/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -403,7 +403,7 @@ where
) -> CallOf<P::TargetChain> {
let call: CallOf<P::TargetChain> = BridgeMessagesCall::<R, I>::receive_messages_proof {
relayer_id_at_bridged_chain: relayer_id_at_source,
proof: proof.1,
proof: Box::new(proof.1),
messages_count,
dispatch_weight,
}
Expand Down

0 comments on commit 15a50af

Please sign in to comment.