add require to rageQuit

2019-11-21 12:10:58 -05:00
parent e669a81a33
commit 5dedf7a92b
1 changed files with 1 additions and 0 deletions
--- a/contracts/UniswapV2.sol
+++ b/contracts/UniswapV2.sol
@ -221,6 +221,7 @@ contract UniswapV2 is IUniswapV2, ERC20("Uniswap V2", "UNI-V2", 18, 0), SafeTran
            amounts.token0 = amountOutput = (liquidity.mul(reserves.token0) / totalSupply).downcast128();
            safeTransfer(token0, recipient, amounts.token0);
        } else {
+            require(output == token1, "UniswapV2: INVALID_OUTPUT");
            amounts.token1 = amountOutput = (liquidity.mul(reserves.token1) / totalSupply).downcast128();
            safeTransfer(token1, recipient, amounts.token1);
        }