fix: sherlock-audit/2024-06-leveraged-vaults-judging#73

notional-finance · Aug 9, 2024 · a78f2cc · a78f2cc
1 parent 121fe6a
commit a78f2cc
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/contracts/vaults/common/VaultRewarderLib.sol b/contracts/vaults/common/VaultRewarderLib.sol
@@ -313,16 +313,18 @@ contract VaultRewarderLib is IVaultRewarder {
             // prevent normal vault operations from working. Failures may include a
             // lack of balances or some sort of blacklist that prevents an account
             // from receiving tokens.
-            try IEIP20NonStandard(rewardToken).transfer(account, rewardToClaim) {
-                bool success = TokenUtils.checkReturnCode();
-                if (success) {
-                    emit VaultRewardTransfer(rewardToken, account, rewardToClaim);
-                } else {
+            if (rewardToken.code.length > 0) {
+                try IEIP20NonStandard(rewardToken).transfer(account, rewardToClaim) {
+                    bool success = TokenUtils.checkReturnCode();
+                    if (success) {
+                        emit VaultRewardTransfer(rewardToken, account, rewardToClaim);
+                    } else {
+                        emit VaultRewardTransfer(rewardToken, account, 0);
+                    }
+                // Emits zero tokens transferred if the transfer fails.
+                } catch {
                     emit VaultRewardTransfer(rewardToken, account, 0);
                 }
-            // Emits zero tokens transferred if the transfer fails.
-            } catch {
-                emit VaultRewardTransfer(rewardToken, account, 0);
             }
         }
     }