|
1 | 1 | package app |
2 | 2 |
|
3 | 3 | import ( |
4 | | - "bytes" |
5 | 4 | "context" |
6 | | - "errors" |
7 | | - "fmt" |
8 | 5 |
|
9 | | - "cosmossdk.io/collections" |
10 | | - sdkerrors "cosmossdk.io/errors" |
11 | 6 | upgradetypes "cosmossdk.io/x/upgrade/types" |
12 | 7 | "github.com/cosmos/cosmos-sdk/types/module" |
13 | | - |
14 | | - movetypes "github.com/initia-labs/initia/x/move/types" |
15 | | - vmprecompile "github.com/initia-labs/movevm/precompile" |
16 | | - vmtypes "github.com/initia-labs/movevm/types" |
17 | 8 | ) |
18 | 9 |
|
19 | | -const upgradeName = "0.6.0" |
| 10 | +const upgradeName = "0.6.2" |
20 | 11 |
|
21 | 12 | // RegisterUpgradeHandlers returns upgrade handlers |
22 | 13 | func (app *InitiaApp) RegisterUpgradeHandlers(cfg module.Configurator) { |
23 | 14 | app.UpgradeKeeper.SetUpgradeHandler( |
24 | 15 | upgradeName, |
25 | 16 | func(ctx context.Context, _ upgradetypes.Plan, vm module.VersionMap) (module.VersionMap, error) { |
26 | | - |
27 | | - // 1. publish new code module first |
28 | | - codeModuleBz, err := vmprecompile.ReadStdlib("code.mv") |
29 | | - if err != nil { |
30 | | - return nil, err |
31 | | - } |
32 | | - err = app.MoveKeeper.SetModule(ctx, vmtypes.StdAddress, movetypes.MoveModuleNameCode, codeModuleBz[0]) |
33 | | - if err != nil { |
34 | | - return nil, err |
35 | | - } |
36 | | - |
37 | | - // 2. update vm data with new separator and add checksums of each module |
38 | | - |
39 | | - // Previous: |
40 | | - // ModuleSeparator = byte(0) |
41 | | - // ResourceSeparator = byte(1) |
42 | | - // TableEntrySeparator = byte(2) |
43 | | - // TableInfoSeparator = byte(3) |
44 | | - |
45 | | - // Current: |
46 | | - // ModuleSeparator = byte(0) |
47 | | - // ChecksumSeparator = byte(1) |
48 | | - // ResourceSeparator = byte(2) |
49 | | - // TableEntrySeparator = byte(3) |
50 | | - // TableInfoSeparator = byte(4) |
51 | | - |
52 | | - err = app.MoveKeeper.VMStore.Walk(ctx, new(collections.Range[[]byte]).Descending(), func(key, value []byte) (stop bool, err error) { |
53 | | - key = bytes.Clone(key) |
54 | | - |
55 | | - cursor := movetypes.AddressBytesLength |
56 | | - if len(key) <= cursor { |
57 | | - return true, fmt.Errorf("invalid key length: %d", len(key)) |
58 | | - } |
59 | | - |
60 | | - separator := key[cursor] |
61 | | - if separator == movetypes.ModuleSeparator { |
62 | | - checksum := movetypes.ModuleBzToChecksum(value) |
63 | | - value = checksum[:] |
64 | | - } else if separator >= movetypes.TableInfoSeparator { |
65 | | - return true, errors.New("unknown prefix") |
66 | | - } else if err = app.MoveKeeper.VMStore.Remove(ctx, key); err != nil { |
67 | | - return true, err |
68 | | - } |
69 | | - |
70 | | - key[cursor] = key[cursor] + 1 |
71 | | - err = app.MoveKeeper.VMStore.Set(ctx, key, value) |
72 | | - if err != nil { |
73 | | - return true, err |
74 | | - } |
75 | | - |
76 | | - return false, nil |
77 | | - }) |
78 | | - if err != nil { |
79 | | - return nil, err |
80 | | - } |
81 | | - |
82 | | - // 3. update new modules |
83 | | - |
84 | | - codesBz, err := vmprecompile.ReadStdlib("object_code_deployment.mv", "coin.mv", "cosmos.mv", "dex.mv", "json.mv", "bech32.mv", "hash.mv", "collection.mv") |
85 | | - if err != nil { |
86 | | - return nil, err |
87 | | - } |
88 | | - modules := make([]vmtypes.Module, len(codesBz)) |
89 | | - for i, codeBz := range codesBz { |
90 | | - modules[i] = vmtypes.NewModule(codeBz) |
91 | | - } |
92 | | - |
93 | | - err = app.MoveKeeper.PublishModuleBundle(ctx, vmtypes.StdAddress, vmtypes.NewModuleBundle(modules...), movetypes.UpgradePolicy_COMPATIBLE) |
94 | | - if err != nil { |
95 | | - return nil, sdkerrors.Wrap(err, "failed to publish module bundle") |
96 | | - } |
97 | | - |
98 | 17 | return vm, nil |
99 | 18 | }, |
100 | 19 | ) |
|
0 commit comments