使用Echidna测试智能合约库 - Trail of Bits博客

Alex Groce
2020年8月17日
区块链, 模糊测试

本文将展示如何使用Echidna模糊测试器测试智能合约。具体内容包括：

• 通过差分模糊测试变体发现我们在Set Protocol审计期间发现的漏洞
• 为您自己的智能合约库指定和检查有用属性

我们将演示如何通过crytic.io完成所有这些操作，该平台提供GitHub集成和额外的安全检查。

库可能引入风险

在单个智能合约中发现漏洞至关重要：一个合约可能管理重要的经济资源（无论是代币还是以太币），漏洞造成的损失可能达数百万美元。但可以说，以太坊区块链上存在比任何单个合约更重要的代码：库代码。

库可能被许多高价值合约共享，因此，例如SafeMath中一个微妙的未知漏洞可能允许攻击者利用不止一个而是多个关键合约。这种基础设施代码的关键性在区块链环境之外已得到充分理解——广泛使用的库（如TLS或sqlite）中的漏洞具有传染性，可能感染所有依赖该漏洞库的代码。

库测试通常侧重于检测内存安全漏洞。然而在区块链上，我们不太担心避免栈溢出或从包含私钥的区域执行memcpy；我们最担心的是库代码的语义正确性。智能合约在“代码即法律”的金融世界中运行，如果库在某种情况下计算出错误结果，这种“法律漏洞”可能传播到调用合约，并允许攻击者使合约行为异常。

此类漏洞可能产生除使库产生错误结果之外的其他后果；如果攻击者可以强制库代码意外回退，他们就获得了潜在拒绝服务攻击的关键。如果攻击者可以使库函数进入失控循环，他们可以将拒绝服务与昂贵的gas消耗结合起来。

这就是Trail of Bits在管理地址数组的库旧版本中发现的漏洞本质，如对Set Protocol代码的审计所述。

有问题的代码如下：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15

/**
* 返回是否存在重复项。以O(n^2)运行。
* @param A 要搜索的数组
* @return 如果存在重复项则返回true，否则返回false
*/
function hasDuplicate(address[] memory A) returns (bool) {
   for (uint256 i = 0; i < A.length - 1; i++) {
     for (uint256 j = i + 1; j < A.length; j++) {
       if (A[i] == A[j]) {
          return true;
       }
     }
 }
 return false;
}

问题是如果A.length为0（A为空），则A.length - 1下溢，外部（i）循环遍历整个uint256值集。在这种情况下，内部（j）循环不会执行，因此我们有一个紧密循环（基本上）永远不执行任何操作。当然，这个过程总是会耗尽gas，而进行hasDuplicate调用的交易将失败。如果攻击者可以在正确的位置产生空数组，那么使用hasDuplicate对地址数组执行某些不变量的合约可能会被禁用——可能是永久性的。

库

有关详细信息，请参阅我们的示例代码，并查看有关使用Echidna的教程。

在高层次上，该库提供了用于管理地址数组的便捷函数。典型用例涉及使用地址白名单进行访问控制。AddressArrayUtils.sol有19个要测试的函数：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19

function indexOf(address[] memory A, address a)
function contains(address[] memory A, address a)
function indexOfFromEnd(address[] A, address a)
function extend(address[] memory A, address[] memory B)
function append(address[] memory A, address a)
function sExtend(address[] storage A, address[] storage B)
function intersect(address[] memory A, address[] memory B)
function union(address[] memory A, address[] memory B)
function unionB(address[] memory A, address[] memory B)
function difference(address[] memory A, address[] memory B)
function sReverse(address[] storage A)
function pop(address[] memory A, uint256 index)
function remove(address[] memory A, address a)
function sPop(address[] storage A, uint256 index)
function sPopCheap(address[] storage A, uint256 index)
function sRemoveCheap(address[] storage A, address a)
function hasDuplicate(address[] memory A)
function isEqual(address[] memory A, address[] memory B)
function argGet(address[] memory A, uint256[] memory indexArray)

看起来很多，但许多函数效果相似，因为AddressArrayUtils提供了功能版本（对内存数组参数进行操作）和变异版本（需要存储数组）的extend、reverse、pop和remove。您可以看到，一旦我们为pop编写了测试，为sPop编写测试可能不会太困难。

基于属性的模糊测试101

我们的工作是获取我们感兴趣的函数——这里是所有函数——并：

1. 弄清楚每个函数的作用
2. 编写测试确保函数做到这一点！

当然，一种方法是编写大量单元测试，但这有问题。如果我们想彻底测试库，这将是一项繁重的工作，而且坦白说，我们可能做不好。我们确定能想到每个边缘情况吗？即使我们尝试覆盖所有源代码，像hasDuplicate漏洞这样涉及缺失源代码的漏洞也很容易被遗漏。

我们希望使用基于属性的测试来指定所有可能输入的一般行为，然后生成大量输入。编写行为的一般描述比编写任何具体的“给定输入X，函数应该做/返回Y”测试更困难。但是编写所有所需的具体测试的工作量将过高。最重要的是，即使是非常完善的手动单元测试也无法找到攻击者寻找的那种奇怪边缘情况漏洞。

Echidna测试工具：hasDuplicate

测试库的代码最明显的事情是它比库本身还大！在这种情况下并不罕见。不要被吓倒；与库不同，测试工具作为进行中的工作处理，并慢慢改进和扩展，效果很好。测试开发本质上是增量的，如果您有像Echidna这样的工具来放大您的投入，即使小的努力也能提供相当大的好处。

对于一个具体示例，让我们看看hasDuplicate漏洞。我们想检查：

• 如果存在重复项，hasDuplicate报告它
• 如果不存在重复项，hasDuplicate报告不存在

我们可以重新实现hasDuplicate本身，但这在一般情况下没有太大帮助（在这里，它可能让我们找到漏洞）。如果我们有另一个独立开发的高质量地址数组实用程序库，我们可以比较它，这种方法称为差分测试。不幸的是，我们通常没有这样的参考库。

我们这里的方法是通过在库中寻找另一个可以不调用hasDuplicate检测重复项的函数来应用较弱版本的差分测试。为此，我们将使用indexOf和indexOfFromEnd来检查项目索引（从0开始）是否与从数组末尾执行搜索时的索引相同：

1
2
3
4
5
6
7
8
9

for (uint i = 0; i < addrs1.length; i++) {
  (i1, b) = AddressArrayUtils.indexOf(addrs1, addrs1[i]);
  (i2, b) = AddressArrayUtils.indexOfFromEnd(addrs1, addrs1[i]);
  if (i1 != (i2-1)) { // -1 因为fromEnd返回偏移1
    hasDup = true;
  }
}
return hasDup == AddressArrayUtils.hasDuplicate(addrs1);
}

在我们的addressarrayutils演示中查看完整示例代码

此代码遍历addrs1并找到每个元素首次出现的索引。如果没有重复项，这当然总是i本身。代码然后找到元素最后出现的索引（即从末尾）。如果这两个索引不同，则存在重复项。在Echidna中，属性只是布尔Solidity函数，如果属性满足通常返回true（我们将在下面看到例外），如果它们回退或返回false则失败。现在我们的hasDuplicate测试正在测试hasDuplicate和两个indexOf函数。如果它们不一致，Echidna会告诉我们。

现在我们可以添加几个要模糊化的函数来设置addrs1。

让我们在Crytic上运行此属性：

hasDuplicate的属性测试在Crytic中失败

首先，crytic_hasDuplicate失败：

1
2
3

crytic_hasDuplicate: failed!
  Call sequence:
    set_addr(0x0)

触发交易序列极其简单：不向addrs1添加任何内容，然后对其调用hasDuplicate。就这样——由此产生的失控循环将耗尽您的gas预算，Crytic/Echidna将告诉您属性失败。当Echidna将失败最小化为最简单的可能序列时，会产生0x0地址。

我们的其他属性（crytic_revert_remove和crytic_remove）通过，这很好。如果我们修复hasDuplicate中的漏洞，那么我们的测试都将通过：

所有三个属性测试现在在Crytic中通过

crytic_hasDuplicate: fuzzing (2928/10000)告诉我们，由于昂贵的hasDuplicate属性没有快速失败，在我们达到五分钟超时之前，每个属性最多10,000个测试中只执行了3,000个。

Echidna测试工具：库的其余部分

现在我们已经看到了一个测试示例，以下是构建其余测试的一些基本建议（正如我们在addressarrayutils_demo存储库中所做的）：

• 尝试不同的方式计算相同的东西。您拥有的函数的“差分”版本越多，就越有可能发现其中一个错误。例如，查看我们交叉检查indexOf、contains和indexOfFromEnd的所有方式。
• 测试回退。如果您在属性名称前添加前缀_revert_，如我们这里所做，则属性仅在所有调用都回退时通过。这确保代码在应该失败时失败。
• 不要忘记检查明显的简单不变量，例如，数组与自身的差始终为空（ourEqual(AddressArrayUtils.difference(addrs1, addrs1), empty)）。
• 其他测试中的不变量检查和前提条件也可以作为测试函数的交叉检查。请注意，hasDuplicate在许多并非旨在检查hasDuplicate的测试中被调用；只是知道数组无重复可以建立许多其他行为的额外不变量，例如，在任何位置删除地址X后，数组将不再包含X。

开始使用Crytic

您可以通过下载和安装工具或使用我们的docker构建在自己的设备上运行Echidna测试——但使用Crytic平台将基于属性的Echidna测试、Slither静态分析（包括公共版本Slither中不可用的新分析器）、可升级性检查以及您自己的单元测试集成到与版本控制相连的无缝环境中。此外，addressarrayutils_demo存储库显示了基于属性测试所需的一切：它可以像创建最小的Truffle设置、添加带有Echidna属性的crytic.sol文件以及在Crytic的存储库配置中打开基于属性的测试一样简单。

立即注册Crytic，如果您有问题，请加入我们的Slack频道（#crytic）或在Twitter上关注@CryticCI。