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

…lance
paritytech · Aug 29, 2023 · 7c83f4a · 7c83f4a
1 parent 783aadc
commit 7c83f4a
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))
 		}
 	}