check user has enough balance and limit max transfer to their free ba…

…lance
paritytech · Aug 29, 2023 · 767cc3a · 767cc3a
1 parent 56cafc4
commit 767cc3a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/substrate/frame/nomination-pools/src/lib.rs b/substrate/frame/nomination-pools/src/lib.rs
@@ -2660,7 +2660,8 @@ pub mod pallet {
 			#[pallet::compact] max_transfer: BalanceOf<T>,
 		) -> DispatchResult {
 			let who = ensure_signed(origin)?;
-			Self::do_top_up_reward_deficit(who, pool_id, max_transfer)
+			let caller_balance = T::Currency::free_balance(&who);
+			Self::do_top_up_reward_deficit(who, pool_id, max_transfer.min(caller_balance))
 		}
 	}