Rust 智能合约养成日记（9）合约升级
- BlockSec
- 更新于 2022-03-29 10:33
- 阅读 2663
往期回顾： Rust智能合约养成日记（1） Rust智能合约养成日记（2）Rust智能合约养成日记（3）Rust智能合约养成日记（4）Rust智能合约养成日记 （5） Rust 智能合约养成日记 （6）Rust 智能合约养成日记 （7）Rust 智能合约养成日记 （8）
往期回顾：
- Rust智能合约养成日记（1）合约状态数据定义与方法实现
- Rust智能合约养成日记（2）编写Rust智能合约单元测试
- Rust智能合约养成日记（3）Rust智能合约部署，函数调用及Explorer的使用
- Rust智能合约养成日记（4）Rust 智能合约整数溢出
- Rust智能合约养成日记 （5）拒绝服务攻击
- Rust智能合约养成日记 （6）拒绝服务攻击
- Rust 智能合约养成日记（7）合约安全之计算精度
- Rust智能合约养成日记（8）合约安全之权限控制
程序维护中的一个基本问题是——缺陷修复总会以（20－50）%的机率引入新的 bug。所以整个过程是前进两步，后退一步。——布鲁克斯《人月神话》
1. 合约升级的必要性
智能合约本质上是程序，既然是程序，缺陷就不可避免。即便是经过大量测试和审计的智能合约，仍然可能会有漏洞。合约的漏洞如果被攻击者利用，可能造成用户资产的损失，导致严重的后果。漏洞的修复往往需要通过合约升级来实现。除了漏洞修复，新特性的加入也需要合约升级。因此合约的可升级性非常有必要。本期Rust智能合约养成日记，将为大家介绍Rust合约的升级方式。
2. Solidity合约常见升级方式
以太坊中，智能合约具有不可变性，一旦被部署到链上，没有人可以改变它。
那么如果合约存在漏洞或合约需要添加新功能，该如何修改合约的代码？解决方案是将新的合约部署到区块链上。
该方法面临的挑战是，solidity每次部署合约后，合约都会被分配一个唯一的地址。因此所有用到了该合约的DApps都需要修改合约地址来适配新的合约。此外，旧版本合约中的状态需要迁移到新版本合约中，状态较为复杂的合约迁移的工作量很大，容易出错，而且复制数据的Gas费用高。
因此，我们通常采用数据和逻辑分离的架构，将数据保存在一个不处理任何逻辑的合约（状态合约）中，所有的逻辑在另一个合约（逻辑合约）中实现。通常合约升级修改的是逻辑，使用该架构只需要升级逻辑合约，不需要担心状态迁移。
为了解决这个问题，可以使用代理合约（Proxy Contract），具体架构如下图所示。
代理合约用来来存储数据，并且使用delegatecall调用逻辑合约A，这样合约A读写的数据都存储在代理合约中。如果需要升级逻辑合约，部署新的合约B，然后发一条交易给代理合约，让代理合约指向新的逻辑合约B即可。
3. NEAR 合约升级常用方法
接下来我们将以StatusMessage项目为例，给大家介绍NEAR 合约的常用升级方法，如下是StatusMessage的合约代码
我们先将编译好的合约部署在测试网上。
transaction 如下：
接着我们调用set_status 方法，向合约中存储数据
$ near call --accountId blocksec_upgrade.testnet statusmessage.blocksec_upgrade.testnet set_status '{"message":"Hi!BlockSec"}'transaction 如下
https://explorer.testnet.near.org/transactions/8QAH66QczyMSEizbfvygj8FVhxThQZUvJ6DY9hsZVyh9接下来我们详细讨论两种不同的合约升级情况
3.1 合约数据结构未被修改
例如我们增加一个函数：
编译后使用deploy重新部署：
会提示该账户已经部署了一个合约，是否继续。输入y确认即可。
接着我们调用get_status方法读取之前写入的数据
near call --accountId blocksec_upgrade.testnet statusmessage.blocksec_upgrade.testnet get_status '{"account_id":"blocksec_upgrade.testnet"}'原来合约中的数据能成功读取：
这是因为NEAR合约可以重复部署，如果一个账户（地址）已经部署过合约，再次调用near deploy命令可以将新的合约代码部署到该账户上。如果我们只修改合约逻辑，不涉及数据结构的修改 ，可以直接使用near deploy部署新的代码。
3.2 合约数据结构被修改
我们将该合约升级，修改了原来的数据结构，去除了records, 新增了taglines和bios
我们尝试再次重新部署：
$ near deploy --accountId statusmessage.blocksec_upgrade.testnet --wasmFile target/wasm32-unknown-unknown/release/status_message.wasm合约还是成功部署了
会发现出错了，错误提示如下
Cannot deserialize the contract state.具体的transaction见
https://explorer.testnet.near.org/transactions/4hQQ1zAwU5bsbfb6tA6DQDqjmFcHsBwaBctdHaPiCKHu具体的transaction见
https://explorer.testnet.near.org/transactions/4hQQ1zAwU5bsbfb6tA6DQDqjmFcHsBwaBctdHaPiCKHu
代码中的#[init(ignore_state)]表示在migrate函数执行前不要加载状态。接着，我们重新部署合约，但是在部署的同时调用migrate方法
我们尝试调用合约新增的方法get_tagline去获取新增的数据taglines
$ near call --accountId blocksec_upgrade.testnet statusmessage.blocksec_upgrade.testnet get_tagline '{"account_id":"blocksec_upgrade.testnet"}'可以看到方法被成功调用，旧的合约数据也被迁移到新的合约
4. 合约升级的安全考量
合约安全升级首先要考虑权限控制，一般合约只能由开发者或DAO升级。上一期Rust 智能合约养成日记（8）合约安全之权限控制介绍了特权函数的访问控制，一般合约的升级函数为only owner函数，确保只能由owner调用。
我们推荐尽可能将合约的owner设置为DAO，通过提案和投票来共同管理合约。因为owner设置为个人账户，合约高度中心化，owner可以随意修改合约数据，还存在owner私钥丢失的风险。
除此之外，开发者在做合约迁移时，还可以考虑以下几点建议
- 在迁移函数前加入
#[init(ignore_state)]，确保执行迁移函数前不加载状态。 - 迁移完成后尽量删除迁移函数，确保迁移函数只被调用一次。
- 新增的数据结构在迁移时完成初始化。
