change transfer to call

iotexproject · Mar 25, 2024 · c807ea2 · c807ea2
1 parent 6033864
commit c807ea2
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 6 deletions.
diff --git a/contracts/RewardsDistributor.sol b/contracts/RewardsDistributor.sol
@@ -145,7 +145,8 @@ contract RewardsDistributor is IRewardsDistributor, ReentrancyGuard {
         (_timestamp >= _locked.end && !_locked.isPermanent)
       ) {
         address _owner = ve.ownerOf(_tokenId);
-        payable(_owner).transfer(amount);
+        (bool success, ) = payable(_owner).call{value: amount}("");
+        require(success, "transfer rewards failed.");
       } else {
         ve.depositFor{value: amount}(_tokenId, amount);
       }
@@ -175,7 +176,8 @@ contract RewardsDistributor is IRewardsDistributor, ReentrancyGuard {
           (_timestamp >= _locked.end && !_locked.isPermanent)
         ) {
           address _owner = ve.ownerOf(_tokenId);
-          payable(_owner).transfer(amount);
+          (bool success, ) = payable(_owner).call{value: amount}("");
+          require(success, "transfer rewards failed.");
         } else {
           ve.depositFor{value: amount}(_tokenId, amount);
         }

diff --git a/contracts/gauges/Gauge.sol b/contracts/gauges/Gauge.sol
@@ -76,7 +76,8 @@ contract Gauge is IGauge, ERC2771Context, ReentrancyGuard {
     uint256 reward = rewards[_account];
     if (reward > 0) {
       rewards[_account] = 0;
-      payable(_account).transfer(reward);
+      (bool success, ) = payable(_account).call{value: reward}("");
+      require(success, "transfer rewards failed.");
       emit ClaimRewards(_account, reward);
     }
   }