看起来很简单的ERC20转账,处理不好就有可能导致不可估量的损失
看起来很简单的 ERC20 转账,处理不好就有可能导致不可估量的损失 因为每天能用来写东西的时间比较有限,我准备分三个小文去讲,分别讲 转入、转出、记账 今天讲 ERC20 token 的转入处理
看下面的代码,你感觉会不会有问题?
function deposit(address token, uint256 amount) external {
//...
address depositor = msg.sender;
IERC20(token).safeTransferFrom(depositor, address(this), amount);
emit Deposit(token, amount, depositor);
}
这是我从曾经经手过的一个项目中截取的代码,忽略掉了对 token 和 amount 的参数校验。用户调用 deposit 函数把钱存入合约之后,合约触发 Deposit 事件,服务端程序会扫描 Deposit 事件落库,之后用户可以申请提款,提款金额和落库金额有关。
正如你所想的那样,一切都是工作正常的,直到有一天,这个系统想要支持 PAXG,用户似乎可以提取比存入更多的 PAXG Token。问题出在哪里呢? 我们看看 PAXG 的转账代码实现:
function _transfer(address _from, address _to, uint256 _value) internal returns (uint256) {
uint256 _fee = getFeeFor(_value);
uint256 _principle = _value.sub(_fee);
balances[_from] = balances[_from].sub(_value);
balances[_to] = balances[_to].add(_principle); // 重点看看这里的 _principle 和 参数 _value 有什么不同
emit Transfer(_from, _to, _principle);
emit Transfer(_from, feeRecipient, _fee);
if (_fee > 0) {
balances[feeRecipient] = balances[feeRecipient].add(_fee);
emit FeeCollected(_from, feeRecipient, _fee);
}
return _principle;
}
看明白了么?PAXG 每笔转账都会从转账金额中扣除一部分手续费,真正转到目标账户的是扣除手续费之后的金额。
所以啊,永远不要假定调用 transferFrom 实际转入的就是你告诉它要转入的金额。
怎么修复这个问题呢?其实也很简单,直接看代码吧
function deposit(address token, uint256 amount) external {
//...
address depositor = msg.sender;
uint256 balanceBefore = IERC20(token).balanceOf(address(this));
IERC20(token).safeTransferFrom(depositor, address(this), amount);
uint256 balanceAfter = IERC20(token).balanceOf(address(this));
uint256 actualAmount = balanceAfter - balanceBefore;
emit Deposit(token, actualAmount, depositor);
}
如果觉得我的文章对您有用,请随意打赏。你的支持将鼓励我继续创作!