feat: tainted transaction reporting

engine/src/witness/btc/deposits.rs

@@ -194,7 +194,9 @@ pub mod tests {
 	fn fake_details(
 		deposit_address: DepositAddress,
 	) -> DepositChannelDetails<state_chain_runtime::Runtime, BitcoinInstance> {
+		use cf_chains::{btc::ScriptPubkey, ForeignChainAddress};
 		DepositChannelDetails::<_, BitcoinInstance> {
+			owner: AccountId32::new([0xab; 32]),
 			opened_at: 1,
 			expires_at: 10,
 			deposit_channel: DepositChannel {
@@ -205,6 +207,7 @@ pub mod tests {
 			},
 			action: ChannelAction::<AccountId32>::LiquidityProvision {
 				lp_account: AccountId32::new([0xab; 32]),
+				refund_address: Some(ForeignChainAddress::Btc(ScriptPubkey::P2PKH([0; 20]))),
 			},
 			boost_fee: 0,
 			boost_status: BoostStatus::NotBoosted,

state-chain/pallets/cf-ingress-egress/src/benchmarking.rs

@@ -7,6 +7,7 @@ use cf_chains::{
 	benchmarking_value::{BenchmarkValue, BenchmarkValueExtended},
 	DepositChannel,
 };
+use cf_primitives::AccountRole;
 use cf_traits::AccountRoleRegistry;
 use frame_benchmarking::v2::*;
 use frame_support::{
@@ -61,6 +62,7 @@ mod benchmarks {
 		DepositChannelLookup::<T, I>::insert(
 			&deposit_address,
 			DepositChannelDetails {
+				owner: account("doogle", 0, 0),
 				opened_at: block_number,
 				expires_at: block_number,
 				deposit_channel:
@@ -71,6 +73,7 @@ mod benchmarks {
 					.unwrap(),
 				action: ChannelAction::<T::AccountId>::LiquidityProvision {
 					lp_account: account("doogle", 0, 0),
+					refund_address: None,
 				},
 				boost_fee: 0,
 				boost_status: BoostStatus::NotBoosted,
@@ -102,6 +105,7 @@ mod benchmarks {
 			let block_number = TargetChainBlockNumber::<T, I>::benchmark_value();
 			let mut channel =
 				DepositChannelDetails::<T, I> {
+					owner: account("doogle", 0, 0),
 					opened_at: block_number,
 					expires_at: block_number,
 					deposit_channel: DepositChannel::generate_new::<
@@ -110,6 +114,7 @@ mod benchmarks {
 					.unwrap(),
 					action: ChannelAction::<T::AccountId>::LiquidityProvision {
 						lp_account: account("doogle", 0, 0),
+						refund_address: None,
 					},
 					boost_fee: 0,
 					boost_status: BoostStatus::NotBoosted,
@@ -227,7 +232,10 @@ mod benchmarks {
 		let (_channel_id, deposit_address, ..) = Pallet::<T, I>::open_channel(
 			lp_account,
 			asset,
-			ChannelAction::LiquidityProvision { lp_account: lp_account.clone() },
+			ChannelAction::LiquidityProvision {
+				lp_account: lp_account.clone(),
+				refund_address: None,
+			},
 			fee_tier,
 		)
 		.unwrap();
@@ -401,7 +409,10 @@ mod benchmarks {
 		let (_channel_id, deposit_address, ..) = Pallet::<T, I>::open_channel(
 			&boosters[0],
 			asset,
-			ChannelAction::LiquidityProvision { lp_account: boosters[0].clone() },
+			ChannelAction::LiquidityProvision {
+				lp_account: boosters[0].clone(),
+				refund_address: None,
+			},
 			TIER_5_BPS,
 		)
 		.unwrap();
@@ -506,6 +517,26 @@ mod benchmarks {
 		assert_eq!(BoostPools::<T, I>::iter().count(), 1);
 	}
 
+	#[benchmark]
+	fn mark_transaction_as_tainted() {
+		let caller =
+			T::AccountRoleRegistry::whitelisted_caller_with_role(AccountRole::Broker).unwrap();
+		let tx_id = <<T as Config<I>>::TargetChain as Chain>::DepositDetails::benchmark_value();
+
+		#[block]
+		{
+			assert_ok!(Pallet::<T, I>::mark_transaction_as_tainted_inner(
+				caller.clone(),
+				tx_id.clone(),
+			));
+		}
+
+		assert!(
+			TaintedTransactions::<T, I>::get(caller, tx_id).is_some(),
+			"No tainted transactions found"
+		);
+	}
+
 	#[cfg(test)]
 	use crate::mock_eth::*;
 
@@ -550,5 +581,8 @@ mod benchmarks {
 		new_test_ext().execute_with(|| {
 			_contract_ccm_swap_request::<Test, ()>(true);
 		});
+		new_test_ext().execute_with(|| {
+			_mark_transaction_as_tainted::<Test, ()>(true);
+		});
 	}
 }