diff options
Diffstat (limited to 'plugin')
-rw-r--r-- | plugin/evm/block.go | 15 | ||||
-rw-r--r-- | plugin/evm/database.go | 2 | ||||
-rw-r--r-- | plugin/evm/export_tx.go | 16 | ||||
-rw-r--r-- | plugin/evm/factory.go | 2 | ||||
-rw-r--r-- | plugin/evm/import_tx.go | 14 | ||||
-rw-r--r-- | plugin/evm/service.go | 10 | ||||
-rw-r--r-- | plugin/evm/static_service.go | 2 | ||||
-rw-r--r-- | plugin/evm/tx.go | 16 | ||||
-rw-r--r-- | plugin/evm/user.go | 6 | ||||
-rw-r--r-- | plugin/evm/vm.go | 150 | ||||
-rw-r--r-- | plugin/main.go | 2 |
11 files changed, 145 insertions, 90 deletions
diff --git a/plugin/evm/block.go b/plugin/evm/block.go index 5c30fe1..97ffc18 100644 --- a/plugin/evm/block.go +++ b/plugin/evm/block.go @@ -10,10 +10,10 @@ import ( "github.com/ava-labs/coreth/core/types" "github.com/ava-labs/go-ethereum/rlp" - "github.com/ava-labs/gecko/ids" - "github.com/ava-labs/gecko/snow/choices" - "github.com/ava-labs/gecko/snow/consensus/snowman" - "github.com/ava-labs/gecko/vms/components/missing" + "github.com/ava-labs/avalanche-go/ids" + "github.com/ava-labs/avalanche-go/snow/choices" + "github.com/ava-labs/avalanche-go/snow/consensus/snowman" + "github.com/ava-labs/avalanche-go/vms/components/missing" ) // Block implements the snowman.Block interface @@ -30,8 +30,11 @@ func (b *Block) ID() ids.ID { return b.id } func (b *Block) Accept() error { vm := b.vm - vm.ctx.Log.Verbo("Block %s is accepted", b.ID()) - vm.updateStatus(b.ID(), choices.Accepted) + vm.ctx.Log.Verbo("Block %s is accepted", b.id) + vm.updateStatus(b.id, choices.Accepted) + if err := vm.acceptedDB.Put(b.ethBlock.Number().Bytes(), b.id.Bytes()); err != nil { + return err + } tx := vm.getAtomicTx(b.ethBlock) if tx == nil { diff --git a/plugin/evm/database.go b/plugin/evm/database.go index de592e1..1300bbb 100644 --- a/plugin/evm/database.go +++ b/plugin/evm/database.go @@ -8,7 +8,7 @@ import ( "github.com/ava-labs/go-ethereum/ethdb" - "github.com/ava-labs/gecko/database" + "github.com/ava-labs/avalanche-go/database" ) var ( diff --git a/plugin/evm/export_tx.go b/plugin/evm/export_tx.go index 31b4681..77f4efe 100644 --- a/plugin/evm/export_tx.go +++ b/plugin/evm/export_tx.go @@ -10,14 +10,14 @@ import ( "github.com/ava-labs/coreth/core/state" "github.com/ava-labs/go-ethereum/log" - "github.com/ava-labs/gecko/chains/atomic" - "github.com/ava-labs/gecko/database" - "github.com/ava-labs/gecko/ids" - "github.com/ava-labs/gecko/snow" - "github.com/ava-labs/gecko/utils/crypto" - safemath "github.com/ava-labs/gecko/utils/math" - "github.com/ava-labs/gecko/vms/components/avax" - "github.com/ava-labs/gecko/vms/secp256k1fx" + "github.com/ava-labs/avalanche-go/chains/atomic" + "github.com/ava-labs/avalanche-go/database" + "github.com/ava-labs/avalanche-go/ids" + "github.com/ava-labs/avalanche-go/snow" + "github.com/ava-labs/avalanche-go/utils/crypto" + safemath "github.com/ava-labs/avalanche-go/utils/math" + "github.com/ava-labs/avalanche-go/vms/components/avax" + "github.com/ava-labs/avalanche-go/vms/secp256k1fx" ) // UnsignedExportTx is an unsigned ExportTx diff --git a/plugin/evm/factory.go b/plugin/evm/factory.go index 6ae62ae..4fe6047 100644 --- a/plugin/evm/factory.go +++ b/plugin/evm/factory.go @@ -4,7 +4,7 @@ package evm import ( - "github.com/ava-labs/gecko/ids" + "github.com/ava-labs/avalanche-go/ids" ) // ID this VM should be referenced by diff --git a/plugin/evm/import_tx.go b/plugin/evm/import_tx.go index 35ba8cc..e09f145 100644 --- a/plugin/evm/import_tx.go +++ b/plugin/evm/import_tx.go @@ -9,13 +9,13 @@ import ( "github.com/ava-labs/coreth/core/state" - "github.com/ava-labs/gecko/database" - "github.com/ava-labs/gecko/ids" - "github.com/ava-labs/gecko/snow" - "github.com/ava-labs/gecko/utils/crypto" - "github.com/ava-labs/gecko/utils/math" - "github.com/ava-labs/gecko/vms/components/avax" - "github.com/ava-labs/gecko/vms/secp256k1fx" + "github.com/ava-labs/avalanche-go/database" + "github.com/ava-labs/avalanche-go/ids" + "github.com/ava-labs/avalanche-go/snow" + "github.com/ava-labs/avalanche-go/utils/crypto" + "github.com/ava-labs/avalanche-go/utils/math" + "github.com/ava-labs/avalanche-go/vms/components/avax" + "github.com/ava-labs/avalanche-go/vms/secp256k1fx" "github.com/ava-labs/go-ethereum/common" ) diff --git a/plugin/evm/service.go b/plugin/evm/service.go index fdbfd91..2979f46 100644 --- a/plugin/evm/service.go +++ b/plugin/evm/service.go @@ -15,11 +15,11 @@ import ( "github.com/ava-labs/coreth" "github.com/ava-labs/coreth/core/types" - "github.com/ava-labs/gecko/api" - "github.com/ava-labs/gecko/utils/constants" - "github.com/ava-labs/gecko/utils/crypto" - "github.com/ava-labs/gecko/utils/formatting" - "github.com/ava-labs/gecko/utils/json" + "github.com/ava-labs/avalanche-go/api" + "github.com/ava-labs/avalanche-go/utils/constants" + "github.com/ava-labs/avalanche-go/utils/crypto" + "github.com/ava-labs/avalanche-go/utils/formatting" + "github.com/ava-labs/avalanche-go/utils/json" "github.com/ava-labs/go-ethereum/common" "github.com/ava-labs/go-ethereum/common/hexutil" ethcrypto "github.com/ava-labs/go-ethereum/crypto" diff --git a/plugin/evm/static_service.go b/plugin/evm/static_service.go index d3870ca..a379c44 100644 --- a/plugin/evm/static_service.go +++ b/plugin/evm/static_service.go @@ -8,7 +8,7 @@ import ( "encoding/json" "github.com/ava-labs/coreth/core" - "github.com/ava-labs/gecko/utils/formatting" + "github.com/ava-labs/avalanche-go/utils/formatting" ) // StaticService defines the static API services exposed by the evm diff --git a/plugin/evm/tx.go b/plugin/evm/tx.go index 789ce56..70386dd 100644 --- a/plugin/evm/tx.go +++ b/plugin/evm/tx.go @@ -9,14 +9,14 @@ import ( "github.com/ava-labs/coreth/core/state" - "github.com/ava-labs/gecko/database" - "github.com/ava-labs/gecko/ids" - "github.com/ava-labs/gecko/snow" - "github.com/ava-labs/gecko/utils/codec" - "github.com/ava-labs/gecko/utils/crypto" - "github.com/ava-labs/gecko/utils/hashing" - "github.com/ava-labs/gecko/vms/components/verify" - "github.com/ava-labs/gecko/vms/secp256k1fx" + "github.com/ava-labs/avalanche-go/database" + "github.com/ava-labs/avalanche-go/ids" + "github.com/ava-labs/avalanche-go/snow" + "github.com/ava-labs/avalanche-go/utils/codec" + "github.com/ava-labs/avalanche-go/utils/crypto" + "github.com/ava-labs/avalanche-go/utils/hashing" + "github.com/ava-labs/avalanche-go/vms/components/verify" + "github.com/ava-labs/avalanche-go/vms/secp256k1fx" "github.com/ava-labs/go-ethereum/common" ) diff --git a/plugin/evm/user.go b/plugin/evm/user.go index fbf2981..65dbe0c 100644 --- a/plugin/evm/user.go +++ b/plugin/evm/user.go @@ -7,9 +7,9 @@ import ( "errors" "fmt" - "github.com/ava-labs/gecko/database" - "github.com/ava-labs/gecko/ids" - "github.com/ava-labs/gecko/utils/crypto" + "github.com/ava-labs/avalanche-go/database" + "github.com/ava-labs/avalanche-go/ids" + "github.com/ava-labs/avalanche-go/utils/crypto" "github.com/ava-labs/go-ethereum/common" ) diff --git a/plugin/evm/vm.go b/plugin/evm/vm.go index b0073da..a1224d4 100644 --- a/plugin/evm/vm.go +++ b/plugin/evm/vm.go @@ -23,31 +23,34 @@ import ( "github.com/ava-labs/coreth/params" "github.com/ava-labs/go-ethereum/common" - ethcrypto "github.com/ava-labs/go-ethereum/crypto" "github.com/ava-labs/go-ethereum/rlp" "github.com/ava-labs/go-ethereum/rpc" + + ethcrypto "github.com/ava-labs/go-ethereum/crypto" + geckorpc "github.com/gorilla/rpc/v2" - "github.com/ava-labs/gecko/api/admin" - "github.com/ava-labs/gecko/cache" - "github.com/ava-labs/gecko/database" - "github.com/ava-labs/gecko/ids" - "github.com/ava-labs/gecko/snow" - "github.com/ava-labs/gecko/snow/choices" - "github.com/ava-labs/gecko/snow/consensus/snowman" - "github.com/ava-labs/gecko/utils/codec" - "github.com/ava-labs/gecko/utils/constants" - "github.com/ava-labs/gecko/utils/crypto" - "github.com/ava-labs/gecko/utils/formatting" - geckojson "github.com/ava-labs/gecko/utils/json" - "github.com/ava-labs/gecko/utils/logging" - "github.com/ava-labs/gecko/utils/timer" - "github.com/ava-labs/gecko/utils/units" - "github.com/ava-labs/gecko/utils/wrappers" - "github.com/ava-labs/gecko/vms/components/avax" - "github.com/ava-labs/gecko/vms/secp256k1fx" - - commonEng "github.com/ava-labs/gecko/snow/engine/common" + "github.com/ava-labs/avalanche-go/api/admin" + "github.com/ava-labs/avalanche-go/cache" + "github.com/ava-labs/avalanche-go/database" + "github.com/ava-labs/avalanche-go/database/prefixdb" + "github.com/ava-labs/avalanche-go/ids" + "github.com/ava-labs/avalanche-go/snow" + "github.com/ava-labs/avalanche-go/snow/choices" + "github.com/ava-labs/avalanche-go/snow/consensus/snowman" + "github.com/ava-labs/avalanche-go/utils/codec" + "github.com/ava-labs/avalanche-go/utils/constants" + "github.com/ava-labs/avalanche-go/utils/crypto" + "github.com/ava-labs/avalanche-go/utils/formatting" + "github.com/ava-labs/avalanche-go/utils/logging" + "github.com/ava-labs/avalanche-go/utils/timer" + "github.com/ava-labs/avalanche-go/utils/units" + "github.com/ava-labs/avalanche-go/utils/wrappers" + "github.com/ava-labs/avalanche-go/vms/components/avax" + "github.com/ava-labs/avalanche-go/vms/secp256k1fx" + + commonEng "github.com/ava-labs/avalanche-go/snow/engine/common" + geckojson "github.com/ava-labs/avalanche-go/utils/json" ) var ( @@ -60,6 +63,7 @@ var ( const ( lastAcceptedKey = "snowman_lastAccepted" + acceptedPrefix = "snowman_accepted" ) const ( @@ -67,7 +71,7 @@ const ( maxBlockTime = 1000 * time.Millisecond batchSize = 250 maxUTXOsToFetch = 1024 - blockCacheSize = 1 << 17 // 131072 + blockCacheSize = 1 << 10 // 1024 ) const ( @@ -157,6 +161,8 @@ type VM struct { networkChan chan<- commonEng.Message newTxPoolHeadChan chan core.NewTxPoolHeadEvent + acceptedDB database.Database + txPoolStabilizedHead common.Hash txPoolStabilizedOk chan struct{} txPoolStabilizedLock sync.Mutex @@ -229,6 +235,8 @@ func (vm *VM) Initialize( return err } + vm.acceptedDB = prefixdb.New([]byte(acceptedPrefix), db) + vm.chainID = g.Config.ChainID vm.txFee = txFee @@ -598,40 +606,56 @@ func (vm *VM) tryBlockGen() error { func (vm *VM) getCachedStatus(blockID ids.ID) choices.Status { vm.metalock.Lock() defer vm.metalock.Unlock() - status := choices.Processing if statusIntf, ok := vm.blockStatusCache.Get(blockID); ok { - status = statusIntf.(choices.Status) - } else { - wrappedBlk := vm.getBlock(blockID) - if wrappedBlk == nil { - return choices.Unknown + return statusIntf.(choices.Status) + } + + wrappedBlk := vm.getBlock(blockID) + if wrappedBlk == nil { + return choices.Unknown + } + blk := wrappedBlk.ethBlock + + heightKey := blk.Number().Bytes() + acceptedIDBytes, err := vm.acceptedDB.Get(heightKey) + if err == nil { + if acceptedID, err := ids.ToID(acceptedIDBytes); err != nil { + vm.ctx.Log.Error("snowman-eth: acceptedID bytes didn't match expected value: %s", err) + } else { + if acceptedID.Equals(blockID) { + vm.blockStatusCache.Put(blockID, choices.Accepted) + return choices.Accepted + } + vm.blockStatusCache.Put(blockID, choices.Rejected) + return choices.Rejected } - blk := wrappedBlk.ethBlock - acceptedBlk := vm.lastAccepted.ethBlock - - // TODO: There must be a better way of doing this. - // Traverse up the chain from the lower block until the indices match - highBlock := blk - lowBlock := acceptedBlk - if highBlock.Number().Cmp(lowBlock.Number()) < 0 { - highBlock, lowBlock = lowBlock, highBlock + } + + status := vm.getUncachedStatus(blk) + if status == choices.Accepted { + err := vm.acceptedDB.Put(heightKey, blockID.Bytes()) + if err != nil { + vm.ctx.Log.Error("snowman-eth: failed to write back acceptedID bytes: %s", err) } - for highBlock.Number().Cmp(lowBlock.Number()) > 0 { - parentBlock := vm.getBlock(ids.NewID(highBlock.ParentHash())) - if parentBlock == nil { - vm.blockStatusCache.Put(blockID, choices.Processing) - return choices.Processing + + tempBlock := wrappedBlk + for tempBlock.ethBlock != nil { + parentID := ids.NewID(tempBlock.ethBlock.ParentHash()) + tempBlock = vm.getBlock(parentID) + if tempBlock == nil || tempBlock.ethBlock == nil { + break } - highBlock = parentBlock.ethBlock - } - if highBlock.Hash() == lowBlock.Hash() { // on the same branch - if blk.Number().Cmp(acceptedBlk.Number()) <= 0 { - status = choices.Accepted + heightKey := tempBlock.ethBlock.Number().Bytes() + _, err := vm.acceptedDB.Get(heightKey) + if err == nil { + break + } + + if err := vm.acceptedDB.Put(heightKey, parentID.Bytes()); err != nil { + vm.ctx.Log.Error("snowman-eth: failed to write back acceptedID bytes: %s", err) } - } else { // on different branches - status = choices.Rejected } } @@ -639,6 +663,34 @@ func (vm *VM) getCachedStatus(blockID ids.ID) choices.Status { return status } +func (vm *VM) getUncachedStatus(blk *types.Block) choices.Status { + acceptedBlk := vm.lastAccepted.ethBlock + + // TODO: There must be a better way of doing this. + // Traverse up the chain from the lower block until the indices match + highBlock := blk + lowBlock := acceptedBlk + if highBlock.Number().Cmp(lowBlock.Number()) < 0 { + highBlock, lowBlock = lowBlock, highBlock + } + for highBlock.Number().Cmp(lowBlock.Number()) > 0 { + parentBlock := vm.getBlock(ids.NewID(highBlock.ParentHash())) + if parentBlock == nil { + return choices.Processing + } + highBlock = parentBlock.ethBlock + } + + if highBlock.Hash() != lowBlock.Hash() { // on different branches + return choices.Rejected + } + // on the same branch + if blk.Number().Cmp(acceptedBlk.Number()) <= 0 { + return choices.Accepted + } + return choices.Processing +} + func (vm *VM) getBlock(id ids.ID) *Block { if blockIntf, ok := vm.blockCache.Get(id); ok { return blockIntf.(*Block) diff --git a/plugin/main.go b/plugin/main.go index f5c9909..1f2f107 100644 --- a/plugin/main.go +++ b/plugin/main.go @@ -3,7 +3,7 @@ package main import ( "github.com/hashicorp/go-plugin" - "github.com/ava-labs/gecko/vms/rpcchainvm" + "github.com/ava-labs/avalanche-go/vms/rpcchainvm" "github.com/ava-labs/coreth/plugin/evm" ) |