Require sender not in blacklist when withdraw

buckyos · Dec 17, 2024 · 046d840 · 046d840
1 parent 2208853
commit 046d840
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/contracts/dividend.sol b/contracts/dividend.sol
@@ -625,6 +625,10 @@ contract DividendContract is Initializable, UUPSUpgradeable, ReentrancyGuardUpgr
     function withdrawDividends(uint256[] calldata cycleIndexs, address[] calldata tokens) external nonReentrant {
         require(cycleIndexs.length > 0, "No cycle index");
         require(tokens.length > 0, "No token");
+
+        // blacklisted address cannot estimate
+        require(!isBlackListed(msg.sender), "Blacklisted address");
+
         // require(UserStakeRecords[msg.sender].length > 0, "No stake record");
 
         // Display the params