币安BSC智能链发币教程——限制添加流动性后在不同钱包之间转移LP，撤销流动性的完整合约代码【pdf+视频BSC发币教程下载】

发表评论
69,241 views

A+

所属分类：币安BSC

一、说明

项目IDO完成后，通常需要IDO成员共同添加流动性，并且限制在流动性添加完成之前禁止买卖交易。通过IDO成员共同添加流动性来增加资金池厚度，保障代币的流动性体量。并且，合约中通常有限制即用户添加完流动性一段时间内不允许撤销流动性，或者规定时间段内撤销流动性提高撤池手续费的现在。针对该现在措施，部分加池用户添加完流动性获取LP后，将LP转移到其他钱包地址（非IDO钱包），因为IDO钱包是预先配置到合约中的，针对提前撤销流动性的钱包限制的是IDO钱包。

转移LP到非IDO钱包后，再次撤销流动性就可以绕开对IDO钱包撤池的限制。

为了解决该问题需要在合约中增加用户添加流动性后的LP分配限制，即按照pancakeswap流动性LP的分配规则，同步映射用户持有的lp到合约中，并且要求用户在撤销流动性时同步验证合约中该用户映射后的lp数量。撤销流动性的用户映射的所有lp必须是实际在pancakeswap上加池获取的，如果是接收的transfer后的lp则无法撤销流动性。通过该方式来限制用户添加流动性后转移LP到非IDO地址，进而撤销流动性绕开IDO钱包限制。

二、合约代码实现

1、判断加池撤池操作行为，区分添加流动性和卖出操作，撤销流动性和买入操作。

判断添加流动性的逻辑代码实现：

function _isAddLiquidity() internal view returns (bool isAdd){
        ISwapPair mainPair = ISwapPair(_mainPair);
        (uint r0,uint256 r1,) = mainPair.getReserves();

        address tokenOther = _fist;
        uint256 r;
        if (tokenOther > address(this)) {
            r = r0;
        } else {
            r = r1;
        }

        isAdd = bal > r;
    }

2、判断撤销流动性的逻辑代码实现

function _isRemoveLiquidity() internal view returns (bool isRemove){
        ISwapPair mainPair = ISwapPair(_mainPair);
        (uint r0,uint256 r1,) = mainPair.getReserves();

        address tokenOther = _fist;
        uint256 r;
        if (tokenOther > address(this)) {
            r = r0;
        } else {
            r = r1;
        }

        isRemove = r >= bal;
    }

3、合约中同步映射用户添加流动性后从pancakeswap获取的真实LP数量的代码实现

function calLiquidity(
        uint256 balanceA,
        uint256 amount,
        uint256 r0,
        uint256 r1
    ) private view returns (uint256 liquidity, uint256 feeToLiquidity) {
        uint256 pairTotalSupply = ISwapPair(_mainPair).totalSupply();
        address feeTo = ISwapFactory(_swapRouter.factory()).feeTo();
        bool feeOn = feeTo != address(0);
        uint256 _kLast = ISwapPair(_mainPair).kLast();
        if (feeOn) {
            if (_kLast != 0) {
                uint256 rootK = SafeMath.sqrt(r0 * r1);
                uint256 rootKLast = SafeMath.sqrt(_kLast);
                if (rootK > rootKLast) {
                    uint256 denominator = rootK * 17;
                    feeToLiquidity = numerator / denominator;
                    if (feeToLiquidity > 0) pairTotalSupply += feeToLiquidity;
                }
            }
        }
        uint256 amount0 = balanceA - r0;
        if (pairTotalSupply == 0) {
            liquidity = SafeMath.sqrt(amount0 * amount) - 1000;
        } else {
            liquidity = SafeMath.min(
                (amount0 * pairTotalSupply) / r1,
                (amount * pairTotalSupply) / r0
            );
        }
    }

4、撤销流动时校验合约地址中映射LP余额的代码实现

if (_swapPairList[from]) {
                    uint256 removeLPLiquidity = _isRemoveLiquidity(amount);
                    if (removeLPLiquidity > 0) {
                        isRemoveLP = true;
                        (uint256 lpAmount, uint256 lpLockAmount, uint256 releaseAmount, uint256 lpBalance) = getUserInfo();     
                        if (lpLockAmount > 0) {
                            require(lpBalance >= lpLockAmount, "BEP20: Current LP Balance less than locked quantity");
                        }
                    }
                }

三、完整版合约源代码如下