NoBey/uniswap-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

move from invariant -> k (compute sqrt lazily)

Browse Source 
use getReserves more places

improve comment quality
This commit is contained in:
Noah Zinsmeister
2020-01-29 13:15:07 -05:00
parent 452b531205
commit 80ce1e48c0
3 changed files with 45 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
contracts/UniswapV2Exchange.sol
View File

@ -17,18 +17,13 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
address public token0; address public token0;
address public token1; address public token1;
uint112 private reserve0; // single storage slot, (jointly) access via getReserves uint112 private reserve0; // uses single storage slot, accessible via getReserves
uint112 private reserve1; // single storage slot, (jointly) access via getReserves uint112 private reserve1; // uses single storage slot, accessible via getReserves
uint32 private blockTimestampLast; // single storage slot, (jointly) access via getReserves uint32 private blockTimestampLast; // uses single storage slot, accessible via getReserves
uint public price0CumulativeLast; uint public price0CumulativeLast;
uint public price1CumulativeLast; uint public price1CumulativeLast;
uint public invariantLast; uint public kLast; // reserve0 * reserve1, as of immediately after the most recent liquidity event
event Mint(address indexed sender, uint amount0, uint amount1);
event Burn(address indexed sender, uint amount0, uint amount1, address indexed to);
event Swap(address indexed sender, address indexed tokenIn, uint amountIn, uint amountOut, address indexed to);
event Sync(uint112 reserve0, uint112 reserve1);
bool private unlocked = true; bool private unlocked = true;
modifier lock() { modifier lock() {
@ -38,30 +33,35 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
unlocked = true; unlocked = true;
} }
function getReserves() public view returns (uint112 _reserve0, uint112 _reserve1, uint32 _blockTimestampLast) {
_reserve0 = reserve0;
_reserve1 = reserve1;
_blockTimestampLast = blockTimestampLast;
}
function _safeTransfer(address token, address to, uint value) private { function _safeTransfer(address token, address to, uint value) private {
(bool success, bytes memory data) = token.call(abi.encodeWithSelector(selector, to, value)); (bool success, bytes memory data) = token.call(abi.encodeWithSelector(selector, to, value));
require(success && (data.length == 0 || abi.decode(data, (bool))), 'UniswapV2: TRANSFER_FAILED'); require(success && (data.length == 0 || abi.decode(data, (bool))), 'UniswapV2: TRANSFER_FAILED');
} }
event Mint(address indexed sender, uint amount0, uint amount1);
event Burn(address indexed sender, uint amount0, uint amount1, address indexed to);
event Swap(address indexed sender, address indexed tokenIn, uint amountIn, uint amountOut, address indexed to);
event Sync(uint112 reserve0, uint112 reserve1);
constructor() public { constructor() public {
factory = msg.sender; factory = msg.sender;
} }
function initialize(address _token0, address _token1) external { function initialize(address _token0, address _token1) external {
require(msg.sender == factory, 'UniswapV2: FORBIDDEN'); require(msg.sender == factory, 'UniswapV2: FORBIDDEN'); // sufficient check
token0 = _token0; token0 = _token0;
token1 = _token1; token1 = _token1;
} }
function getReserves() external view returns (uint112 _reserve0, uint112 _reserve1, uint32 _blockTimestampLast) { // update reserves and, on the first call per block, price accumulators
_reserve0 = reserve0;
_reserve1 = reserve1;
_blockTimestampLast = blockTimestampLast;
}
// update reserves and, on the first time this function is called per block, price accumulators
function _update(uint balance0, uint balance1, uint112 _reserve0, uint112 _reserve1) private { function _update(uint balance0, uint balance1, uint112 _reserve0, uint112 _reserve1) private {
require(balance0 <= uint112(-1) && balance1 <= uint112(-1), 'UniswapV2: BALANCE_OVERFLOW'); require(balance0 <= uint112(-1) && balance1 <= uint112(-1), 'UniswapV2: OVERFLOW');
uint32 blockTimestamp = uint32(block.timestamp % 2**32); uint32 blockTimestamp = uint32(block.timestamp % 2**32);
uint32 timeElapsed = blockTimestamp - blockTimestampLast; // overflow is desired uint32 timeElapsed = blockTimestamp - blockTimestampLast; // overflow is desired
if (timeElapsed > 0 && _reserve0 != 0 && _reserve1 != 0) { if (timeElapsed > 0 && _reserve0 != 0 && _reserve1 != 0) {
@ -75,17 +75,18 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
emit Sync(reserve0, reserve1); emit Sync(reserve0, reserve1);
} }
// mint liquidity equivalent to 20% of newly accumulated fees // if fee is on, mint liquidity equivalent to 20% of growth in sqrt(k)
function _mintFee(uint112 _reserve0, uint112 _reserve1) private returns (bool feeOn) { function _mintFee(uint112 _reserve0, uint112 _reserve1) private returns (bool feeOn) {
address feeTo = IUniswapV2Factory(factory).feeTo(); address feeTo = IUniswapV2Factory(factory).feeTo();
feeOn = feeTo != address(0); feeOn = feeTo != address(0);
if (feeOn) { if (feeOn) {
uint _invariantLast = invariantLast; // gas savings uint _kLast = kLast; // gas savings
if (_invariantLast != 0) { if (_kLast > 0) {
uint invariant = Math.sqrt(uint(_reserve0).mul(_reserve1)); uint rootK = Math.sqrt(uint(_reserve0).mul(_reserve1));
if (invariant > _invariantLast) { uint rootKLast = Math.sqrt(_kLast);
uint numerator = totalSupply.mul(invariant.sub(_invariantLast)); if (rootK > rootKLast) {
uint denominator = invariant.mul(4).add(_invariantLast); uint numerator = totalSupply.mul(rootK.sub(rootKLast));
uint denominator = rootK.mul(4).add(rootKLast);
uint liquidity = numerator / denominator; uint liquidity = numerator / denominator;
if (liquidity > 0) _mint(feeTo, liquidity); if (liquidity > 0) _mint(feeTo, liquidity);
} }
@ -93,17 +94,15 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
} }
} }
// mint liquidity
function mint(address to) external lock returns (uint liquidity) { function mint(address to) external lock returns (uint liquidity) {
uint112 _reserve0 = reserve0; // gas savings (uint112 _reserve0, uint112 _reserve1,) = getReserves(); // gas savings
uint112 _reserve1 = reserve1; // gas savings
uint balance0 = IERC20(token0).balanceOf(address(this)); uint balance0 = IERC20(token0).balanceOf(address(this));
uint balance1 = IERC20(token1).balanceOf(address(this)); uint balance1 = IERC20(token1).balanceOf(address(this));
uint amount0 = balance0.sub(_reserve0); uint amount0 = balance0.sub(_reserve0);
uint amount1 = balance1.sub(_reserve1); uint amount1 = balance1.sub(_reserve1);
bool feeOn = _mintFee(_reserve0, _reserve1); bool feeOn = _mintFee(_reserve0, _reserve1);
uint _totalSupply = totalSupply; // gas savings uint _totalSupply = totalSupply; // gas savings, must be defined here since totalSupply can update in _mintFee
liquidity = _totalSupply == 0 liquidity = _totalSupply == 0
? Math.sqrt(amount0.mul(amount1)) ? Math.sqrt(amount0.mul(amount1))
: Math.min(amount0.mul(_totalSupply) / _reserve0, amount1.mul(_totalSupply) / _reserve1); : Math.min(amount0.mul(_totalSupply) / _reserve0, amount1.mul(_totalSupply) / _reserve1);
@ -111,14 +110,12 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
_mint(to, liquidity); _mint(to, liquidity);
_update(balance0, balance1, _reserve0, _reserve1); _update(balance0, balance1, _reserve0, _reserve1);
if (feeOn) invariantLast = Math.sqrt(uint(reserve0).mul(reserve1)); if (feeOn) kLast = uint(reserve0).mul(reserve1);
emit Mint(msg.sender, amount0, amount1); emit Mint(msg.sender, amount0, amount1);
} }
// burn liquidity
function burn(address to) external lock returns (uint amount0, uint amount1) { function burn(address to) external lock returns (uint amount0, uint amount1) {
uint112 _reserve0 = reserve0; // gas savings (uint112 _reserve0, uint112 _reserve1,) = getReserves(); // gas savings
uint112 _reserve1 = reserve1; // gas savings
address _token0 = token0; // gas savings address _token0 = token0; // gas savings
address _token1 = token1; // gas savings address _token1 = token1; // gas savings
uint balance0 = IERC20(_token0).balanceOf(address(this)); uint balance0 = IERC20(_token0).balanceOf(address(this));
@ -126,9 +123,9 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
uint liquidity = balanceOf[address(this)]; uint liquidity = balanceOf[address(this)];
bool feeOn = _mintFee(_reserve0, _reserve1); bool feeOn = _mintFee(_reserve0, _reserve1);
uint _totalSupply = totalSupply; // gas savings uint _totalSupply = totalSupply; // gas savings, must be defined here since totalSupply can update in _mintFee
amount0 = liquidity.mul(balance0) / _totalSupply; // use balances instead of reserves to address force sends amount0 = liquidity.mul(balance0) / _totalSupply; // using balances ensures pro-rata distribution
amount1 = liquidity.mul(balance1) / _totalSupply; // use balances instead of reserves to address force sends amount1 = liquidity.mul(balance1) / _totalSupply; // using balances ensures pro-rata distribution
require(amount0 > 0 && amount1 > 0, 'UniswapV2: INSUFFICIENT_LIQUIDITY_BURNED'); require(amount0 > 0 && amount1 > 0, 'UniswapV2: INSUFFICIENT_LIQUIDITY_BURNED');
_burn(address(this), liquidity); _burn(address(this), liquidity);
_safeTransfer(_token0, to, amount0); _safeTransfer(_token0, to, amount0);
@ -137,14 +134,13 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
balance1 = IERC20(_token1).balanceOf(address(this)); balance1 = IERC20(_token1).balanceOf(address(this));
_update(balance0, balance1, _reserve0, _reserve1); _update(balance0, balance1, _reserve0, _reserve1);
if (feeOn) invariantLast = Math.sqrt(uint(reserve0).mul(reserve1)); if (feeOn) kLast = uint(reserve0).mul(reserve1);
emit Burn(msg.sender, amount0, amount1, to); emit Burn(msg.sender, amount0, amount1, to);
} }
// swap tokens
function swap(address tokenIn, uint amountOut, address to) external lock { function swap(address tokenIn, uint amountOut, address to) external lock {
uint112 _reserve0 = reserve0; // gas savings require(amountOut > 0, 'UniswapV2: INSUFFICIENT_OUTPUT_AMOUNT');
uint112 _reserve1 = reserve1; // gas savings (uint112 _reserve0, uint112 _reserve1,) = getReserves(); // gas savings
address _token0 = token0; // gas savings address _token0 = token0; // gas savings
address _token1 = token1; // gas savings address _token1 = token1; // gas savings
uint balance0; uint balance0;
@ -152,7 +148,7 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
uint amountIn; uint amountIn;
if (tokenIn == _token0) { if (tokenIn == _token0) {
require(0 < amountOut && amountOut < _reserve1, 'UniswapV2: INVALID_OUTPUT_AMOUNT'); require(amountOut < _reserve1, 'UniswapV2: INSUFFICIENT_LIQUIDITY');
balance0 = IERC20(_token0).balanceOf(address(this)); balance0 = IERC20(_token0).balanceOf(address(this));
amountIn = balance0.sub(_reserve0); amountIn = balance0.sub(_reserve0);
require(amountIn > 0, 'UniswapV2: INSUFFICIENT_INPUT_AMOUNT'); require(amountIn > 0, 'UniswapV2: INSUFFICIENT_INPUT_AMOUNT');
@ -161,7 +157,7 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
balance1 = IERC20(_token1).balanceOf(address(this)); balance1 = IERC20(_token1).balanceOf(address(this));
} else { } else {
require(tokenIn == _token1, 'UniswapV2: INVALID_INPUT_TOKEN'); require(tokenIn == _token1, 'UniswapV2: INVALID_INPUT_TOKEN');
require(0 < amountOut && amountOut < _reserve0, 'UniswapV2: INVALID_OUTPUT_AMOUNT'); require(amountOut < _reserve0, 'UniswapV2: INSUFFICIENT_LIQUIDITY');
balance1 = IERC20(_token1).balanceOf(address(this)); balance1 = IERC20(_token1).balanceOf(address(this));
amountIn = balance1.sub(_reserve1); amountIn = balance1.sub(_reserve1);
require(amountIn > 0, 'UniswapV2: INSUFFICIENT_INPUT_AMOUNT'); require(amountIn > 0, 'UniswapV2: INSUFFICIENT_INPUT_AMOUNT');

4
contracts/UniswapV2Factory.sol
View File

@ -25,7 +25,7 @@ contract UniswapV2Factory is IUniswapV2Factory {
require(tokenA != tokenB, 'UniswapV2: IDENTICAL_ADDRESSES'); require(tokenA != tokenB, 'UniswapV2: IDENTICAL_ADDRESSES');
(address token0, address token1) = tokenA < tokenB ? (tokenA, tokenB) : (tokenB, tokenA); (address token0, address token1) = tokenA < tokenB ? (tokenA, tokenB) : (tokenB, tokenA);
require(token0 != address(0), 'UniswapV2: ZERO_ADDRESS'); require(token0 != address(0), 'UniswapV2: ZERO_ADDRESS');
require(getExchange[token0][token1] == address(0), 'UniswapV2: EXCHANGE_EXISTS'); require(getExchange[token0][token1] == address(0), 'UniswapV2: EXCHANGE_EXISTS'); // single check is sufficient
bytes memory exchangeBytecode = type(UniswapV2Exchange).creationCode; bytes memory exchangeBytecode = type(UniswapV2Exchange).creationCode;
bytes32 salt = keccak256(abi.encodePacked(token0, token1)); bytes32 salt = keccak256(abi.encodePacked(token0, token1));
assembly { assembly {
@ -33,7 +33,7 @@ contract UniswapV2Factory is IUniswapV2Factory {
} }
IUniswapV2Exchange(exchange).initialize(token0, token1); IUniswapV2Exchange(exchange).initialize(token0, token1);
getExchange[token0][token1] = exchange; getExchange[token0][token1] = exchange;
getExchange[token1][token0] = exchange; getExchange[token1][token0] = exchange; // populate mapping in the reverse direction
allExchanges.push(exchange); allExchanges.push(exchange);
emit ExchangeCreated(token0, token1, exchange, allExchanges.length); emit ExchangeCreated(token0, token1, exchange, allExchanges.length);
} }

1
contracts/interfaces/IUniswapV2Exchange.sol
View File

@ -13,6 +13,7 @@ interface IUniswapV2Exchange {
function getReserves() external view returns (uint112 reserve0, uint112 reserve1, uint32 blockTimestampLast); function getReserves() external view returns (uint112 reserve0, uint112 reserve1, uint32 blockTimestampLast);
function price0CumulativeLast() external view returns (uint); function price0CumulativeLast() external view returns (uint);
function price1CumulativeLast() external view returns (uint); function price1CumulativeLast() external view returns (uint);
function kLast() external view returns (uint);
function mint(address to) external returns (uint liquidity); function mint(address to) external returns (uint liquidity);
function burn(address to) external returns (uint amount0, uint amount1); function burn(address to) external returns (uint amount0, uint amount1);