波场TRON-Rich团队的UsdtBank合约简介

随着波场 DApp 生态的不断发展, DApp开发者和用户的数量急速增长，经济利益的迅速累积，提高智能合约的防攻击能力，越来越成DApp 开发的一个重要考量。因此，波场面向社区，征集DApp 的开源代码，结合其合约源码，以实战的方式，讲解波场智能合约开发时，需要注意的一些安全细节。更多源码征集方式请参见附录。

本期小课堂征集到的是 TRON-Rich 团队的 UsdtBank合约。在分析合约之前的首要事情，就是通过合约验证平台，验证其为真的开源合约。接下来先用小段篇幅对社区的https://troneye.com(以下简称 TRON-Eye)进行解析，以选定合约验证平台。

合约验证的原理在于，Solidity 合约编译后的 bytecode 由可执行bytecode 以及meta-hash两部分组成，同一份合约源码在相同编译环境下多次编译，产生的 bytecode 相同，正确的合约验证方法，应该比对bytecode，从而验证源码是否和链上合约完全一致。

(TRON-Eye 的源码提交页)

TRON-Eye详细阐述了其验证思路，同时还在合约源码展示页支持用户自行编译bytecode并比对，提高了公信力。因此，我们选定 TRON-Eye 作为小课堂的验证平台，校验合约是否真正开源。

( TRON-Eye 的合约源码展示页)

图2所示的，即为本次待考察合约，TRON-Rich 团队的UsdtBank合约代码。接下来就对其源码，进行安全角度的详细解读。

如非必要, 应该禁止被其他合约调用

允许被其他合约调用, 容易被发起回退攻击，尤其是即时返回结果的下注类游戏。攻击合约可以在其合约函数中调用目标合约，如果目标合约立即返回结果，当攻击合约发现返回的结果对自己不利时，主动 revert，回退交易。从而实现“只赢不输”。

/*
* only human is allowed to call this contract
*/
modifier isHuman() {
require((bytes32(msg.sender)) == (bytes32(tx.origin)));
_;
}

UsdtBank 采用了上述代码，判断是否是合约，其原理就是，如果是合约调用的话，msg.sender 是外层合约地址，但是 tx.origin 是合约调用者。当然这段代码先将 address强转 bytes32，浪费了能量，建议直接采用 msg.sender == tx.origin 即可。

function invest(uint256 _referrerCode, uint256 _planId, uint256 _value) 
public whenNotPaused isHuman {
if (_invest(msg.sender, _planId, _referrerCode, _value)) {
emit onInvest(msg.sender, _value);
}}

判断一个地址是否是合约地址

下面是 UsdtBank 使用这个modifier 的方式，可以发现，这个 modifier 仅适合用来限制被调用方是普通用户。那么如果需要判断某个传入的address 参数是人，而不是合约，则需要使用另外一种方式。

function isContract(address account) internal view returns (bool) {
uint256 size;
assembly { size := extcodesize(account) }
return size > 0;
}

Q: 那么为什么 isHuman() 这个 modifier 不使用这种方式呢?

A: 这是因为，通过 extcodesize 方式判断一个地址是否是合约地址，并不准确。当在其他合约的构造函数中读取extcodesize时，这个值总是0.

More: 波场已经提交了一个关于增加 address.type 的 TIP，可以直观准确的判断一个地址类型，欢迎参与该 TIP 的讨论。

小结论：要想完整限制合约中的调用者，以及合约中的地址参数为 Human，目前最好的方式，是结合前述的 isHuman() modifier 以及 isContract().

怎么通过合约，处理TRC20的转账

本期选择 UsdtBank 讲解的一个重要原因是，UsdtBank 是一个支持 USDT参与投注的合约，有助于推广使用TRC20投注游戏合约的正确姿势。

UsdtBank 和 USDT 投注相关的有如下一些代码:

ITRC20publicusdtAddr_;
function setUsdtAddr(address _usdtAddr) public onlyOwner {
require(address(usdtAddr_) == address(0x00));
require(address(_usdtAddr) != address(0x00));
usdtAddr_ = ITRC20(_usdtAddr);
}

上述代码表示，usdtAddress 仅允许初始化的时候，设置一次(谢绝跑路 ^_^)。

function _invest(address _addr, uint256 _planId, uint256 _referrerCode, 
uint256 _amount)
private
notContract(_addr)
returns (bool)
{
usdtAddr_.transferFrom(_addr, address(this), _amount);
….
}

由于 TRC20 token 相对 TRX以及 TRC10 token 最大的区别在于，TRX 和 TRC10的balance存储于address 的 account 中，而 TRC20 token 的 balance存储在 TRC20合约里。直接调用 TRC20合约的 transfer 函数，虽然能够将自己的余额转到另外一个地址名下，但事实上只是在 TRC20合约里发生了两者balance 字段值的修改。所以采用标准TRC20下注，必须使用 Approve 和 TransferFrom 两步分开的方式。虽然这会导致用户签名两次。