package state
import (
- "errors"
-
- "github.com/vapor/consensus"
- "github.com/vapor/database/storage"
- "github.com/vapor/protocol/bc"
+ "github.com/bytom/vapor/consensus"
+ "github.com/bytom/vapor/database/storage"
+ "github.com/bytom/vapor/errors"
+ "github.com/bytom/vapor/protocol/bc"
)
// UtxoViewpoint represents a view into the set of unspent transaction outputs
}
func (view *UtxoViewpoint) ApplyTransaction(block *bc.Block, tx *bc.Tx, statusFail bool) error {
- for _, prevout := range tx.SpentOutputIDs {
- spentOutput, err := tx.IntraChainOutput(prevout)
+ if err := view.applyCrossChainUtxo(block, tx); err != nil {
+ return err
+ }
+
+ if err := view.applySpendUtxo(block, tx, statusFail); err != nil {
+ return err
+ }
+
+ return view.applyOutputUtxo(block, tx, statusFail)
+}
+
+func (view *UtxoViewpoint) ApplyBlock(block *bc.Block, txStatus *bc.TransactionStatus) error {
+ for i, tx := range block.Transactions {
+ statusFail, err := txStatus.GetStatus(i)
if err != nil {
return err
}
- if statusFail && *spentOutput.Source.Value.AssetId != *consensus.BTMAssetID {
- continue
+
+ if err := view.ApplyTransaction(block, tx, statusFail); err != nil {
+ return err
}
+ }
+ return nil
+}
+
+func (view *UtxoViewpoint) CanSpend(hash *bc.Hash) bool {
+ entry := view.Entries[*hash]
+ return entry != nil && !entry.Spent
+}
+
+func (view *UtxoViewpoint) DetachTransaction(tx *bc.Tx, statusFail bool) error {
+ if err := view.detachCrossChainUtxo(tx); err != nil {
+ return err
+ }
+ if err := view.detachSpendUtxo(tx, statusFail); err != nil {
+ return err
+ }
+
+ return view.detachOutputUtxo(tx, statusFail)
+}
+
+func (view *UtxoViewpoint) DetachBlock(block *bc.Block, txStatus *bc.TransactionStatus) error {
+ for i := len(block.Transactions) - 1; i >= 0; i-- {
+ statusFail, err := txStatus.GetStatus(i)
+ if err != nil {
+ return err
+ }
+
+ if err := view.DetachTransaction(block.Transactions[i], statusFail); err != nil {
+ return err
+ }
+ }
+ return nil
+}
+
+func (view *UtxoViewpoint) HasUtxo(hash *bc.Hash) bool {
+ _, ok := view.Entries[*hash]
+ return ok
+}
+
+func (view *UtxoViewpoint) applyCrossChainUtxo(block *bc.Block, tx *bc.Tx) error {
+ for _, prevout := range tx.MainchainOutputIDs {
entry, ok := view.Entries[prevout]
if !ok {
- return errors.New("fail to find utxo entry")
+ return errors.New("fail to find mainchain output entry")
}
+
if entry.Spent {
- return errors.New("utxo has been spent")
- }
- if entry.IsCoinBase && entry.BlockHeight+consensus.CoinbasePendingBlockNumber > block.Height {
- return errors.New("coinbase utxo is not ready for use")
+ return errors.New("mainchain output has been spent")
}
+
+ entry.BlockHeight = block.Height
entry.SpendOutput()
}
+ return nil
+}
+func (view *UtxoViewpoint) applyOutputUtxo(block *bc.Block, tx *bc.Tx, statusFail bool) error {
for _, id := range tx.TxHeader.ResultIds {
- output, err := tx.IntraChainOutput(*id)
+ entryOutput, err := tx.Entry(*id)
if err != nil {
- // error due to it's a retirement, utxo doesn't care this output type so skip it
+ return err
+ }
+
+ var assetID bc.AssetID
+ utxoType := storage.NormalUTXOType
+ switch output := entryOutput.(type) {
+ case *bc.IntraChainOutput:
+ if output.Source.Value.Amount == uint64(0) {
+ continue
+ }
+ assetID = *output.Source.Value.AssetId
+ case *bc.VoteOutput:
+ assetID = *output.Source.Value.AssetId
+ utxoType = storage.VoteUTXOType
+ default:
+ // due to it's a retirement, utxo doesn't care this output type so skip it
continue
}
- if statusFail && *output.Source.Value.AssetId != *consensus.BTMAssetID {
+
+ if statusFail && assetID != *consensus.BTMAssetID {
continue
}
- isCoinbase := false
if block != nil && len(block.Transactions) > 0 && block.Transactions[0].ID == tx.ID {
- isCoinbase = true
+ utxoType = storage.CoinbaseUTXOType
}
- view.Entries[*id] = storage.NewUtxoEntry(isCoinbase, block.Height, false)
+ view.Entries[*id] = storage.NewUtxoEntry(utxoType, block.Height, false)
}
return nil
}
-func (view *UtxoViewpoint) ApplyBlock(block *bc.Block, txStatus *bc.TransactionStatus) error {
- for i, tx := range block.Transactions {
- statusFail, err := txStatus.GetStatus(i)
+func (view *UtxoViewpoint) applySpendUtxo(block *bc.Block, tx *bc.Tx, statusFail bool) error {
+ for _, prevout := range tx.SpentOutputIDs {
+ entryOutput, err := tx.Entry(prevout)
if err != nil {
return err
}
- if err := view.ApplyTransaction(block, tx, statusFail); err != nil {
- return err
- }
- }
- return nil
-}
-func (view *UtxoViewpoint) CanSpend(hash *bc.Hash) bool {
- entry := view.Entries[*hash]
- return entry != nil && !entry.Spent
-}
-
-func (view *UtxoViewpoint) DetachTransaction(tx *bc.Tx, statusFail bool) error {
- for _, prevout := range tx.SpentOutputIDs {
- spentOutput, err := tx.IntraChainOutput(prevout)
- if err != nil {
- return err
+ var assetID bc.AssetID
+ switch output := entryOutput.(type) {
+ case *bc.IntraChainOutput:
+ assetID = *output.Source.Value.AssetId
+ case *bc.VoteOutput:
+ assetID = *output.Source.Value.AssetId
+ default:
+ return errors.Wrapf(bc.ErrEntryType, "entry %x has unexpected type %T", prevout.Bytes(), entryOutput)
}
- if statusFail && *spentOutput.Source.Value.AssetId != *consensus.BTMAssetID {
+
+ if statusFail && assetID != *consensus.BTMAssetID {
continue
}
entry, ok := view.Entries[prevout]
- if ok && !entry.Spent {
- return errors.New("try to revert an unspent utxo")
+ if !ok {
+ return errors.New("fail to find utxo entry")
}
+
+ if entry.Spent {
+ return errors.New("utxo has been spent")
+ }
+
+ switch entry.Type {
+ case storage.CoinbaseUTXOType:
+ if (entry.BlockHeight + consensus.ActiveNetParams.CoinbasePendingBlockNumber) > block.Height {
+ return errors.New("coinbase utxo is not ready for use")
+ }
+
+ case storage.VoteUTXOType:
+ if (entry.BlockHeight + consensus.ActiveNetParams.VotePendingBlockNumber) > block.Height {
+ return errors.New("Coin is within the voting lock time")
+ }
+ }
+
+ entry.SpendOutput()
+ }
+ return nil
+}
+
+func (view *UtxoViewpoint) detachCrossChainUtxo(tx *bc.Tx) error {
+ for _, prevout := range tx.MainchainOutputIDs {
+ entry, ok := view.Entries[prevout]
if !ok {
- view.Entries[prevout] = storage.NewUtxoEntry(false, 0, false)
- continue
+ return errors.New("fail to find mainchain output entry")
+ }
+
+ if !entry.Spent {
+ return errors.New("mainchain output is unspent")
}
+
entry.UnspendOutput()
}
+ return nil
+}
+func (view *UtxoViewpoint) detachOutputUtxo(tx *bc.Tx, statusFail bool) error {
for _, id := range tx.TxHeader.ResultIds {
- output, err := tx.IntraChainOutput(*id)
+ entryOutput, err := tx.Entry(*id)
if err != nil {
- // error due to it's a retirement, utxo doesn't care this output type so skip it
+ return err
+ }
+
+ var assetID bc.AssetID
+ utxoType := storage.NormalUTXOType
+ switch output := entryOutput.(type) {
+ case *bc.IntraChainOutput:
+ if output.Source.Value.Amount == uint64(0) {
+ continue
+ }
+ assetID = *output.Source.Value.AssetId
+ case *bc.VoteOutput:
+ assetID = *output.Source.Value.AssetId
+ utxoType = storage.VoteUTXOType
+ default:
+ // due to it's a retirement, utxo doesn't care this output type so skip it
continue
}
- if statusFail && *output.Source.Value.AssetId != *consensus.BTMAssetID {
+
+ if statusFail && assetID != *consensus.BTMAssetID {
continue
}
- view.Entries[*id] = storage.NewUtxoEntry(false, 0, true)
+ view.Entries[*id] = storage.NewUtxoEntry(utxoType, 0, true)
}
return nil
}
-func (view *UtxoViewpoint) DetachBlock(block *bc.Block, txStatus *bc.TransactionStatus) error {
- for i := len(block.Transactions) - 1; i >= 0; i-- {
- statusFail, err := txStatus.GetStatus(i)
+func (view *UtxoViewpoint) detachSpendUtxo(tx *bc.Tx, statusFail bool) error {
+ for _, prevout := range tx.SpentOutputIDs {
+ entryOutput, err := tx.Entry(prevout)
if err != nil {
return err
}
- if err := view.DetachTransaction(block.Transactions[i], statusFail); err != nil {
- return err
+
+ var assetID bc.AssetID
+ utxoType := storage.NormalUTXOType
+ switch output := entryOutput.(type) {
+ case *bc.IntraChainOutput:
+ assetID = *output.Source.Value.AssetId
+ case *bc.VoteOutput:
+ assetID = *output.Source.Value.AssetId
+ utxoType = storage.VoteUTXOType
+ default:
+ return errors.Wrapf(bc.ErrEntryType, "entry %x has unexpected type %T", prevout.Bytes(), entryOutput)
}
+
+ if statusFail && assetID != *consensus.BTMAssetID {
+ continue
+ }
+
+ entry, ok := view.Entries[prevout]
+ if ok && !entry.Spent {
+ return errors.New("try to revert an unspent utxo")
+ }
+
+ if !ok {
+ view.Entries[prevout] = storage.NewUtxoEntry(utxoType, 0, false)
+ continue
+ }
+ entry.UnspendOutput()
}
return nil
}
-
-func (view *UtxoViewpoint) HasUtxo(hash *bc.Hash) bool {
- _, ok := view.Entries[*hash]
- return ok
-}