mirror of https://github.com/poanetwork/gecko.git
155 lines
3.7 KiB
Go
155 lines
3.7 KiB
Go
// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package platformvm
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/ava-labs/gecko/database"
|
|
"github.com/ava-labs/gecko/database/versiondb"
|
|
"github.com/ava-labs/gecko/ids"
|
|
"github.com/ava-labs/gecko/snow/choices"
|
|
"github.com/ava-labs/gecko/vms/components/core"
|
|
)
|
|
|
|
var (
|
|
errConflictingParentTxs = errors.New("block contains a transaction that conflicts with a transaction in a parent block")
|
|
)
|
|
|
|
// AtomicTx is an operation that can be decided without being proposed, but must have special control over database commitment
|
|
type AtomicTx interface {
|
|
initialize(vm *VM) error
|
|
|
|
ID() ids.ID
|
|
|
|
// UTXOs this tx consumes
|
|
InputUTXOs() ids.Set
|
|
|
|
// Attempt to verify this transaction with the provided state. The provided
|
|
// database can be modified arbitrarily.
|
|
SemanticVerify(database.Database) error
|
|
|
|
Accept(database.Batch) error
|
|
}
|
|
|
|
// AtomicBlock being accepted results in the transaction contained in the
|
|
// block to be accepted and committed to the chain.
|
|
type AtomicBlock struct {
|
|
SingleDecisionBlock `serialize:"true"`
|
|
|
|
Tx AtomicTx `serialize:"true"`
|
|
|
|
inputs ids.Set
|
|
}
|
|
|
|
// initialize this block
|
|
func (ab *AtomicBlock) initialize(vm *VM, bytes []byte) error {
|
|
if err := ab.SingleDecisionBlock.initialize(vm, bytes); err != nil {
|
|
return err
|
|
}
|
|
return ab.Tx.initialize(vm)
|
|
}
|
|
|
|
// Reject implements the snowman.Block interface
|
|
func (ab *AtomicBlock) conflicts(s ids.Set) bool {
|
|
if ab.Status() == choices.Accepted {
|
|
return false
|
|
}
|
|
if ab.inputs.Overlaps(s) {
|
|
return true
|
|
}
|
|
return ab.parentBlock().conflicts(s)
|
|
}
|
|
|
|
// Verify this block performs a valid state transition.
|
|
//
|
|
// The parent block must be a proposal
|
|
//
|
|
// This function also sets onAcceptDB database if the verification passes.
|
|
func (ab *AtomicBlock) Verify() error {
|
|
parentBlock := ab.parentBlock()
|
|
|
|
ab.inputs = ab.Tx.InputUTXOs()
|
|
|
|
if parentBlock.conflicts(ab.inputs) {
|
|
return errConflictingParentTxs
|
|
}
|
|
|
|
// AtomicBlock is not a modifier on a proposal block, so its parent must be
|
|
// a decision.
|
|
parent, ok := parentBlock.(decision)
|
|
if !ok {
|
|
return errInvalidBlockType
|
|
}
|
|
|
|
pdb := parent.onAccept()
|
|
|
|
ab.onAcceptDB = versiondb.New(pdb)
|
|
if err := ab.Tx.SemanticVerify(ab.onAcceptDB); err != nil {
|
|
return err
|
|
}
|
|
|
|
ab.vm.currentBlocks[ab.ID().Key()] = ab
|
|
ab.parentBlock().addChild(ab)
|
|
return nil
|
|
}
|
|
|
|
// Accept implements the snowman.Block interface
|
|
func (ab *AtomicBlock) Accept() {
|
|
ab.vm.Ctx.Log.Verbo("Accepting block with ID %s", ab.ID())
|
|
|
|
ab.CommonBlock.Accept()
|
|
|
|
// Update the state of the chain in the database
|
|
if err := ab.onAcceptDB.Commit(); err != nil {
|
|
ab.vm.Ctx.Log.Error("unable to commit onAcceptDB")
|
|
}
|
|
|
|
batch, err := ab.vm.DB.CommitBatch()
|
|
if err != nil {
|
|
ab.vm.Ctx.Log.Fatal("unable to commit vm's DB")
|
|
}
|
|
defer ab.vm.DB.Abort()
|
|
|
|
if err := ab.Tx.Accept(batch); err != nil {
|
|
ab.vm.Ctx.Log.Error("unable to atomically commit block")
|
|
}
|
|
|
|
for _, child := range ab.children {
|
|
child.setBaseDatabase(ab.vm.DB)
|
|
}
|
|
if ab.onAcceptFunc != nil {
|
|
ab.onAcceptFunc()
|
|
}
|
|
|
|
parent := ab.parentBlock()
|
|
// remove this block and its parent from memory
|
|
parent.free()
|
|
ab.free()
|
|
}
|
|
|
|
// newAtomicBlock returns a new *AtomicBlock where the block's parent, a
|
|
// decision block, has ID [parentID].
|
|
func (vm *VM) newAtomicBlock(parentID ids.ID, tx AtomicTx) (*AtomicBlock, error) {
|
|
ab := &AtomicBlock{
|
|
SingleDecisionBlock: SingleDecisionBlock{CommonDecisionBlock: CommonDecisionBlock{
|
|
CommonBlock: CommonBlock{
|
|
Block: core.NewBlock(parentID),
|
|
vm: vm,
|
|
},
|
|
}},
|
|
Tx: tx,
|
|
}
|
|
|
|
// We serialize this block as a Block so that it can be deserialized into a
|
|
// Block
|
|
blk := Block(ab)
|
|
bytes, err := Codec.Marshal(&blk)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ab.Block.Initialize(bytes, vm.SnowmanVM)
|
|
return ab, nil
|
|
}
|