使用 CREATE2 进行确定性部署

介绍

作为 2019 年 Constantinople fork 的一部分,CREATE2 作为 EIP-1014 的一部分被纳入 EVM。CREATE2 允许您根据部署者控制的参数将智能合约部署到确定性地址。因此,它经常被提及为实现“假设性”部署的工具,您可以与尚未创建的地址进行交互,因为 CREATE2 可以保证已知代码可以放置在该地址。这与 CREATE 操作码形成对比,其中部署的合约地址是部署者 nonce 的函数。使用 CREATE2,即使地址具有不同的 nonce,您也可以使用相同的部署者帐户将合约部署到多个网络上。

在本教程中,我们将:

  1. 查看 CREATE2 工厂实现。
  2. 使用传统的部署方法部署工厂。
  3. 使用已部署的工厂来反过来在确定性地址部署一个简单的计数器合约。
  4. 通过在 Foundry 中编写简单的测试来模拟这一系列事件。

先决条件

  1. 需要对 Solidity 和 Foundry 有一定了解,并建议对内联汇编有一定了解。请参考 官方 Solidity 文档了解内联汇编的基础知识。
  2. 确保您的系统上已经 安装 了 Foundry。
  3. 初始化一个新的 Foundry 项目。

CREATE2 工厂

src 目录下创建一个名为 Create2.sol 的文件。 像这样初始化一个名为 Create2 的合约:

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.20;

contract Create2 {

    error Create2EmptyBytecode();

    error Create2FailedDeployment();
}

这些错误旨在对工厂部署执行一些合理性检查,并在触发时回滚整个交易。如果传递给 deploy 函数的字节码为空,则会触发 Create2EmptyBytecode() 错误,如果部署由于任何原因失败,则会触发 Create2FailedDeployment() 错误。

ℹ️ 注意

请注意,由于多种原因,CREATE2 部署可能会失败。例如,如果字节码无效,或者如果在计算地址时已经部署了合约。如果您的构造函数由于任何原因回滚,您的部署也可能会失败。

接下来,创建一个名为 deploy 的函数:

function deploy(bytes32 salt, bytes memory creationCode) external payable returns (address addr) {
 
 }

此函数接受 2 个输入:

  1. 用于计算最终地址的 salt。基本上,这可以是我们想要的任何随机值。
  2. 我们想要部署的合约的创建代码。

成功部署后,新部署的合约地址将被返回。

ℹ️ 注意

您可以向使用 CREATE2 部署的合约发送 ETH,但前提是它具有可支付的构造函数。如果您尝试在没有可支付构造函数的情况下向其发送 ETH,交易将回滚。

在部署函数的顶部添加此回滚语句。我们希望我们的函数在不满足以下条件时拒绝部署请求:

    if (creationCode.length == 0) {
        revert Create2EmptyBytecode();
    }

接下来,我们将使用内联汇编调用 CREATE2 操作码,可以使用 assembly 关键字来完成:

要从内联汇编中调用 CREATE2 操作码 ,我们需要传入 4 个参数:

    assembly {
        addr := create2(callvalue(), add(creationCode, 0x20), mload(creationCode), salt)
    }
  1. 我们希望作为部署的一部分发送到新地址的 ETH 数量。在这里,我们只传入 callvalue(),这是作为交易的一部分发送到工厂合约的 ETH 数量。可以将其视为 msg.value 的低级版本。
  2. 第二个和第三个参数是我们的字节码所在内存范围。add(bytecode, 0x20) 接受一个对 bytes 变量 bytecode 在内存中的位置的引用,并跳过 32 字节(十六进制中的 0x20)以指向实际的字节码。

ℹ️ 注意

Solidity 中的 bytes 类型是一个动态大小的字节数组,其中内存的前 32 个字节表示数组的长度,其余字节表示实际数据。因此,当我们传入对 bytes 变量的引用时,我们需要跳过前 32 个字节以指向实际数据。 在 evm.codes 上阅读有关 addmload 操作码的更多信息。

最后,如果由于任何原因部署失败,我们将回滚整个交易,在这种情况下,CREATE2 操作码将返回一个 0 地址:

        if (addr == address(0)) {
            revert Create2FailedDeployment();
        }

最后,让我们创建一个名为 computeAddress 的视图函数。此函数应接受 saltcreationCode 作为参数,并返回使用 deploy 函数将部署的合约的地址:

function computeAddress(bytes32 salt, bytes32 creationCodeHash) external view returns (address addr) {

 }

在函数内部,粘贴以下代码,该代码使用内联汇编来通过执行与 CREATE2 操作码相同的计算来计算地址:

    address contractAddress = address(this);
        
    assembly {
        let ptr := mload(0x40)

        mstore(add(ptr, 0x40), creationCodeHash)
        mstore(add(ptr, 0x20), salt)
        mstore(ptr, contractAddress)
        let start := add(ptr, 0x0b)
        mstore8(start, 0xff)
        addr := keccak256(start, 85)
    }

在尝试理解此处的汇编代码之前,让我们看一下 CREATE2 操作码用于计算地址的公式:

keccak256(0xff ++ address ++ salt ++ keccak256(bytecode))[12:]

0xff 是一个硬编码的前缀,用于防止使用 CREATECREATE2 部署的地址之间的哈希冲突。address 参数是调用 CREATE2 操作码的合约的地址,在我们的情况下是工厂合约。这 4 个参数被连接在一起,并使用 keccak256 生成一个 32 字节的哈希。前 12 个字节被截断,剩下的 20 个字节被用作部署合约的地址。

computeAddress 函数中的整个汇编代码尝试重新创建相同的公式,尽管没有调用 CREATE2 操作码:

  1. mload(0x40) 将空闲内存指针加载到内存中。这是指向内存数组中下一个空闲内存槽的指针。在 Solidity 文档中了解更多信息。
  2. mstore(add(ptr, 0x40), bytecodeHash)bytecodeHash 存储在由 ptr + 0x40 指向的内存位置,即 ptr+ 64 字节
  3. mstore(add(ptr, 0x20), salt)salt 存储在由 ptr + 0x20 指向的内存位置。
  4. mstore(ptr, contractAddress)contractAddress 存储在由 ptr 指向的内存位置。

ℹ️ 注意

请记住,传递给 computeAddress 函数的所有参数都是 32 字节长,并且作为 32 字节值存储在内存中。然而,Solidity 中的地址长度为 20 字节,并且作为 32 字节值存储在内存中,其中前 12 字节被替换为 0。因此,当我们需要跳过 12 个字节以指向实际地址时。

  1. let start := add(ptr, 0x0b) 创建一个名为 start 的新变量,指向内存位置 ptr + 0x0b ,即 ptr + 11 字节
  2. 最后,可以使用 mstore8 操作码在内存位置存储单个字节。在这里,我们将值 0xff 存储在由 start 指向的内存位置,它占据内存槽的第 12 个字节。
  3. 所有值都打包到其正确的内存位置后,我们现在可以在从 start 开始的内存槽上调用 keccak256,并将内存槽的长度作为第二个参数传递。这将返回一个 32 字节的哈希,我们可以截断以获得最终地址。 ℹ️ 注意

您可以在此处查看此工厂实现的完整代码 此处。 还可以查看 OpenZeppelin 的 CREATE2 库的实现,该实现已被用作本教程的灵感来源。 最后,Forge 提供了一些 CREATE2 地址计算辅助函数。 查看它们

测试我们的工厂

test 目录中创建一个名为 Create2.t.sol 的文件。 像这样初始化一个名为 Create2Test 的合约:

// SPDX-License-Identifier: UNLICENSED
pragma solidity ^0.8.20;

import "forge-std/Test.sol";
import {Counter} from "../src/Counter.sol";
import {Create2} from "../src/Create2.sol";

contract Create2Test is Test {

}

初始化以下状态变量和 setUp() 函数:

    Create2 internal create2;
    Counter internal counter;

    function setUp() public {
        create2 = new Create2();
        counter = new Counter();
    }

创建一个名为 testDeterministicDeploy() 的新函数,该函数:

  1. 部署 Create2 合约的新实例。
  2. 向一个特定地址分配 100 ETH,我们将使用该地址来冒充所有后续调用的调用者,使用 prank 作弊码。
  3. 设置 saltbytecode 参数
  4. 使用先前部署的 Create2 合约在确定性地址上部署 Counter 合约。
  5. 检查合约是否部署在正确的地址上,通过断言计算的地址是否等于部署的地址。
    function testDeterministicDeploy() public {
        vm.deal(address(0x1), 100 ether);

        vm.startPrank(address(0x1));  
        bytes32 salt = "12345";
        bytes memory creationCode = abi.encodePacked(type(Counter).creationCode);

        address computedAddress = create2.computeAddress(salt, keccak256(creationCode));
        address deployedAddress = create2.deploy(salt , creationCode);
        vm.stopPrank();

        assertEq(computedAddress, deployedAddress);  
    }

保存所有文件,并使用 forge test --match-path test/Create2.t.sol -vvvv 运行测试。 您的测试应该通过,没有任何错误。

ℹ️ 注意

本指南旨在帮助理解 CREATE2。在大多数情况下,您不需要编写和使用自己的部署器,可以使用现有的确定性部署器。在 forge 脚本中,使用 new MyContract{salt: salt}() 将使用地址为 0x4e59b44847b379578588920ca78fbf26c0b4956c 的确定性部署器。