Extend error handling on MaticX contract

stader-labs · Sep 25, 2024 · 8ca1048 · 8ca1048
1 parent 790940d
commit 8ca1048
Show file tree

Hide file tree

Showing 2 changed files with 328 additions and 6 deletions.
diff --git a/contracts/MaticX.sol b/contracts/MaticX.sol
@@ -412,6 +412,7 @@ contract MaticX is
 		uint256 _toValidatorId,
 		uint256 _amount
 	) external override whenNotPaused onlyRole(DEFAULT_ADMIN_ROLE) {
+		require(_amount > 0, "Amount is zero");
 		require(
 			IValidatorRegistry(validatorRegistry).validatorIdExists(
 				_fromValidatorId
@@ -590,9 +591,11 @@ contract MaticX is
 		override
 		returns (uint256)
 	{
-		uint256 totalStake;
 		uint256[] memory validators = IValidatorRegistry(validatorRegistry)
 			.getValidators();
+
+		uint256 totalStake;
+
 		for (uint256 i = 0; i < validators.length; ++i) {
 			address validatorShare = IStakeManager(stakeManager)
 				.getValidatorContract(validators[i]);