-
Notifications
You must be signed in to change notification settings - Fork 670
/
chain.go
117 lines (95 loc) · 2.58 KB
/
chain.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
// Copyright (C) 2019-2024, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package chain
import (
"github.com/ava-labs/avalanchego/database"
"github.com/ava-labs/avalanchego/ids"
"github.com/ava-labs/avalanchego/snow"
"github.com/ava-labs/avalanchego/snow/choices"
"github.com/ava-labs/avalanchego/vms/example/xsvm/state"
xsblock "github.com/ava-labs/avalanchego/vms/example/xsvm/block"
)
var _ Chain = (*chain)(nil)
type Chain interface {
LastAccepted() ids.ID
SetChainState(state snow.State)
GetBlock(blkID ids.ID) (Block, error)
// Creates a fully verifiable and executable block, which can be processed
// by the consensus engine, from a stateless block.
NewBlock(blk *xsblock.Stateless) (Block, error)
}
type chain struct {
chainContext *snow.Context
acceptedState database.Database
// chain state as driven by the consensus engine
chainState snow.State
lastAccepted ids.ID
verifiedBlocks map[ids.ID]*block
}
func New(ctx *snow.Context, db database.Database) (Chain, error) {
// Load the last accepted block data. For a newly created VM, this will be
// the genesis. It is assumed the genesis was processed and stored
// previously during VM initialization.
lastAcceptedID, err := state.GetLastAccepted(db)
if err != nil {
return nil, err
}
c := &chain{
chainContext: ctx,
acceptedState: db,
lastAccepted: lastAcceptedID,
}
lastAccepted, err := c.getBlock(lastAcceptedID)
c.verifiedBlocks = map[ids.ID]*block{
lastAcceptedID: lastAccepted,
}
return c, err
}
func (c *chain) LastAccepted() ids.ID {
return c.lastAccepted
}
func (c *chain) SetChainState(state snow.State) {
c.chainState = state
}
func (c *chain) GetBlock(blkID ids.ID) (Block, error) {
return c.getBlock(blkID)
}
func (c *chain) NewBlock(blk *xsblock.Stateless) (Block, error) {
blkID, err := blk.ID()
if err != nil {
return nil, err
}
if blk, exists := c.verifiedBlocks[blkID]; exists {
return blk, nil
}
blkBytes, err := xsblock.Codec.Marshal(xsblock.CodecVersion, blk)
if err != nil {
return nil, err
}
return &block{
Stateless: blk,
chain: c,
id: blkID,
bytes: blkBytes,
}, nil
}
func (c *chain) getBlock(blkID ids.ID) (*block, error) {
if blk, exists := c.verifiedBlocks[blkID]; exists {
return blk, nil
}
blkBytes, err := state.GetBlock(c.acceptedState, blkID)
if err != nil {
return nil, err
}
stateless, err := xsblock.Parse(blkBytes)
if err != nil {
return nil, err
}
return &block{
Stateless: stateless,
chain: c,
id: blkID,
status: choices.Accepted,
bytes: blkBytes,
}, nil
}