From 5dedf7a92bcd216e33705754896c65f394d97999 Mon Sep 17 00:00:00 2001 From: Noah Zinsmeister Date: Thu, 21 Nov 2019 12:10:58 -0500 Subject: [PATCH] add require to rageQuit --- contracts/UniswapV2.sol | 1 + 1 file changed, 1 insertion(+) diff --git a/contracts/UniswapV2.sol b/contracts/UniswapV2.sol index 025d266..eddd138 100644 --- 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); }