add check when converting to stable coin

masa-finance · Nov 10, 2022 · 446378d · 446378d
1 parent 3a57a42
commit 446378d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/contracts/dex/PayDexAMM.sol b/contracts/dex/PayDexAMM.sol
@@ -126,6 +126,12 @@ abstract contract PayDexAMM is Ownable {
         view
         returns (uint256)
     {
+        require(
+            token == wrappedNativeToken ||
+                token == utilityToken ||
+                erc20token[token],
+            "INVALID_TOKEN"
+        );
         return _estimateSwapAmount(token, stableCoin, amount);
     }