Rust 智能合约养成日记(9)合约升级

  • BlockSec
  • 更新于 2022-03-29 10:33
  • 阅读 1993

往期回顾: Rust智能合约养成日记(1) Rust智能合约养成日记(2)Rust智能合约养成日记(3)Rust智能合约养成日记(4)Rust智能合约养成日记 (5) Rust 智能合约养成日记 (6)Rust 智能合约养成日记 (7)Rust 智能合约养成日记 (8)

往期回顾:

程序维护中的一个基本问题是——缺陷修复总会以(20-50)%的机率引入新的 bug。所以整个过程是前进两步,后退一步。——布鲁克斯《人月神话》


1. 合约升级的必要性

智能合约本质上是程序,既然是程序,缺陷就不可避免。即便是经过大量测试和审计的智能合约,仍然可能会有漏洞。合约的漏洞如果被攻击者利用,可能造成用户资产的损失,导致严重的后果。漏洞的修复往往需要通过合约升级来实现。除了漏洞修复,新特性的加入也需要合约升级。因此合约的可升级性非常有必要。本期Rust智能合约养成日记,将为大家介绍Rust合约的升级方式。


2. Solidity合约常见升级方式

以太坊中,智能合约具有不可变性,一旦被部署到链上,没有人可以改变它。

那么如果合约存在漏洞或合约需要添加新功能,该如何修改合约的代码?解决方案是将新的合约部署到区块链上。

该方法面临的挑战是,solidity每次部署合约后,合约都会被分配一个唯一的地址。因此所有用到了该合约的DApps都需要修改合约地址来适配新的合约。此外,旧版本合约中的状态需要迁移到新版本合约中,状态较为复杂的合约迁移的工作量很大,容易出错,而且复制数据的Gas费用高。

因此,我们通常采用数据和逻辑分离的架构,将数据保存在一个不处理任何逻辑的合约(状态合约)中,所有的逻辑在另一个合约(逻辑合约)中实现。通常合约升级修改的是逻辑,使用该架构只需要升级逻辑合约,不需要担心状态迁移。

为了解决这个问题,可以使用代理合约(Proxy Contract),具体架构如下图所示。

2.png

代理合约用来来存储数据,并且使用delegatecall调用逻辑合约A,这样合约A读写的数据都存储在代理合约中。如果需要升级逻辑合约,部署新的合约B,然后发一条交易给代理合约,让代理合约指向新的逻辑合约B即可。


3. NEAR 合约升级常用方法

接下来我们将以StatusMessage项目为例,给大家介绍NEAR 合约的常用升级方法,如下是StatusMessage的合约代码

3.png

我们先将编译好的合约部署在测试网上。

4.png

5.png

transaction 如下: 6.png

接着我们调用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 合约数据结构未被修改

例如我们增加一个函数:

7.png

编译后使用deploy重新部署:

8.png

会提示该账户已经部署了一个合约,是否继续。输入y确认即可。

9.png

接着我们调用get_status方法读取之前写入的数据

near call --accountId blocksec_upgrade.testnet statusmessage.blocksec_upgrade.testnet get_status '{"account_id":"blocksec_upgrade.testnet"}'

原来合约中的数据能成功读取:

10.png

这是因为NEAR合约可以重复部署,如果一个账户(地址)已经部署过合约,再次调用near deploy命令可以将新的合约代码部署到该账户上。如果我们只修改合约逻辑,不涉及数据结构的修改 ,可以直接使用near deploy部署新的代码。

3.2 合约数据结构被修改

我们将该合约升级,修改了原来的数据结构,去除了records, 新增了taglines和bios

11.png

我们尝试再次重新部署:

$ near deploy --accountId statusmessage.blocksec_upgrade.testnet --wasmFile target/wasm32-unknown-unknown/release/status_message.wasm

合约还是成功部署了

12.png

会发现出错了,错误提示如下

Cannot deserialize the contract state.

具体的transaction见

https://explorer.testnet.near.org/transactions/4hQQ1zAwU5bsbfb6tA6DQDqjmFcHsBwaBctdHaPiCKHu

具体的transaction见

https://explorer.testnet.near.org/transactions/4hQQ1zAwU5bsbfb6tA6DQDqjmFcHsBwaBctdHaPiCKHu

13.png

代码中的#[init(ignore_state)]表示在migrate函数执行前不要加载状态。接着,我们重新部署合约,但是在部署的同时调用migrate方法

14.png

我们尝试调用合约新增的方法get_tagline去获取新增的数据taglines

$ near call --accountId blocksec_upgrade.testnet statusmessage.blocksec_upgrade.testnet get_tagline '{"account_id":"blocksec_upgrade.testnet"}'

可以看到方法被成功调用,旧的合约数据也被迁移到新的合约

15.png

4. 合约升级的安全考量

合约安全升级首先要考虑权限控制,一般合约只能由开发者或DAO升级。上一期Rust 智能合约养成日记(8)合约安全之权限控制介绍了特权函数的访问控制,一般合约的升级函数为only owner函数,确保只能由owner调用。

我们推荐尽可能将合约的owner设置为DAO,通过提案和投票来共同管理合约。因为owner设置为个人账户,合约高度中心化,owner可以随意修改合约数据,还存在owner私钥丢失的风险。

除此之外,开发者在做合约迁移时,还可以考虑以下几点建议

  • 在迁移函数前加入#[init(ignore_state)],确保执行迁移函数前不加载状态。
  • 迁移完成后尽量删除迁移函数,确保迁移函数只被调用一次。
  • 新增的数据结构在迁移时完成初始化。
  • 640.jpg
点赞 0
收藏 0
分享
本文参与登链社区写作激励计划 ,好文好收益,欢迎正在阅读的你也加入。

0 条评论

请先 登录 后评论
BlockSec
BlockSec
江湖只有他的大名,没有他的介绍。