fix:

add check maintenance mode for claiming the reward in case of any compromise case
icon-community · Jan 23, 2024 · 2f7e6fb · 2f7e6fb
1 parent f507b6d
commit 2f7e6fb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/CPSCore/src/main/java/community/icon/cps/score/cpscore/CPSCore.java b/CPSCore/src/main/java/community/icon/cps/score/cpscore/CPSCore.java
@@ -2397,6 +2397,7 @@ public void removeDenylistPreps() {
     @Override
     @External
     public void claimSponsorBond() {
+        checkMaintenance();
         Address caller = Context.getCaller();
         Context.require(!ArrayDBUtils.containsInArrayDb(caller, blockAddresses), TAG + ": Address is blocked");
         DictDB<String, BigInteger> userAmounts = sponsorBondReturn.at(caller.toString());

diff --git a/CPSTreasury/src/main/java/community/icon/cps/score/cpstreasury/CPSTreasury.java b/CPSTreasury/src/main/java/community/icon/cps/score/cpstreasury/CPSTreasury.java
@@ -339,11 +339,11 @@ public void sendInstallmentToContributor(String ipfsKey, BigInteger milestoneBud
         Address contributorAddress = (Address) proposalData.get(consts.CONTRIBUTOR_ADDRESS);
         String flag = (String) proposalData.get(consts.TOKEN);
 
-        Context.require(milestoneBudget.compareTo(remainingAmount)<= 0,TAG+"Requested budget is greater than remaining amount.");
+        Context.require(milestoneBudget.compareTo(remainingAmount) <= 0, TAG + "Requested budget is greater than remaining amount.");
 //        installmentAmount = remainingAmount.subtract(milestoneBudget);
 
         installmentAmount = milestoneBudget;
-        Context.println("yhe installment is "+ installmentAmount);
+        Context.println("yhe installment is " + installmentAmount);
 
 //        if (_installmentCount == 1) {
 //            installmentAmount = remainingAmount;
@@ -481,10 +481,11 @@ public void disqualifyProject(String ipfsKey) {
     }
 
 
-
     @Override
     @External
     public void claimReward() {
+        boolean checkMaintainance = callScore(Boolean.class, getCpsScore(), "getMaintenanceMode");
+        Context.require(!checkMaintainance, TAG + ": CPS is in maintenance mode");
         Address caller = Context.getCaller();
         List<Address> blockAddresses = callScore(List.class, getCpsScore(), "getBlockedAddresses");
         Context.require(!blockAddresses.contains(caller), TAG + ": Address is blocked");