audit fix02

2020-12-09 11:20:27 +08:00
parent ca692ee548
commit fc39f709f2
6 changed files with 41 additions and 5 deletions
--- a/contracts/DODOVendingMachine/impl/DVMTrader.sol
+++ b/contracts/DODOVendingMachine/impl/DVMTrader.sol
@@ -103,7 +103,12 @@ contract DVMTrader is DVMVault {
        uint256 quoteAmount,
        address assetTo,
        bytes calldata data
-    ) external preventReentrant {
+    ) 
+        external 
+        preventReentrant 
+        isSellAllow(assetTo)
+        isBuyAllow(assetTo)  
+    {
        _transferBaseOut(assetTo, baseAmount);
        _transferQuoteOut(assetTo, quoteAmount);