Svarog c5707f64dc
[NOD-1420] Implement consensusStateManager (#985)
* [NOD-1420] Start working on ConsensusStateManager. Might be redundant due to recent changes

* [NOD-1420] Convert model to externalapi in utxo_algerbra helpers

* [NOD-1420] Add UTXO-diff algebra

* [NOD-1420] Prepare skeleton of calculateAcceptanceDataAndMultiset

* [NOD-1420] Added skeleton for AddBlockToVirtual

* [NOD-1420] Implement PopulateTransactionWithUTXOEntries

* [NOD-1420] Implement restorePastUTXO

* [NOD-1420] Implement finality check

* [NOD-1420] Move handling of tips to consensusStateManager

* [NOD-1420] Implement calculateAcceptanceDataAndMultiset

* [NOD-1420] Start implementing resolveBlockStatus

* [NOD-1420] Implement resolveBlockStatus

* [NOD-1420] Update related fields in end of resolveSingleBlockStatus

* [NOD-1420] Start working on selectVirtualParents

* [NOD-1420] Implemented BlockHeap

* [NOD-1420] Implement selectVirtualParents

* [NOD-1420] Implement updateVirtual

* [NOD-1420] Added comments where they were missing

* [NOD-1420] Place all consensusStateManager functions in correct files

* [NOD-1420] Return the missing outpoints from populateTransactionWithUTXOEntriesFromVirtualOrDiff

* [NOD-1420] Outpoint.ID -> TransactionID

* [NOD-1420] Fix Stringer tests

* [NOD-1420] Copy hash.FromString into utils

* [NOD-1420] SetParents should return an error

* [NOD-1420] Remove all reachabilityManager references from consensusStateManager

* [NOD-1420] Remove VirtualData. Get the info from the stores where needed

* [NOD-1420] Invert parameters to IsAncestorOf

* [NOD-1420] Use model.AcceptanceData

* [NOD-1420] Don't return accumulatedMassBefore in error cases

* [NOD-1420] Don't expect store functions to return nil when the requested data was found - instead add HasXXX functions

* [NOD-1420] addTransactionToMultiset sets isCoinbase properly

* [NOD-1420] expected hash string length is externalapi.DomainHashSize * 2

* [NOD-1420] Rename reachabilityTree -> reachabilityManager + updateReindexRoot if isNextVirtualSelectedParent

* [NOD-1420] ValidateCoinbaseTransaction in csm.verifyAndBuildUTXO

* [NOD-1420] Re-write HAsUTXODiffChild

* [NOD-1420] delete past_utxo.go.bak

* [NOD-1420] Implement validateCoinbaseTransaction in CSM

* [NOD-1420] Imlemented missing functionality in ValidateTransactionAndPopulateWithConsensusData

* [NOD-1420] Moved merge depth logic to MergeDepthManager

* [NOD-1420] Add logs
2020-11-02 16:18:53 +02:00

85 lines
2.6 KiB
Go

package utxoalgebra
import (
"reflect"
"github.com/kaspanet/kaspad/domain/consensus/utils/hashserialization"
"github.com/kaspanet/kaspad/domain/consensus/utils/transactionhelper"
"github.com/kaspanet/kaspad/domain/consensus/model"
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
"github.com/pkg/errors"
)
// DiffClone returns a new UTXODiff which is identical to the given diff
func DiffClone(diff *model.UTXODiff) *model.UTXODiff {
clone := &model.UTXODiff{
ToAdd: collectionClone(diff.ToAdd),
ToRemove: collectionClone(diff.ToRemove),
}
return clone
}
// DiffAddTransaction modifies the provided utxoDiff with provided transaction.
func DiffAddTransaction(utxoDiff *model.UTXODiff, transaction *externalapi.DomainTransaction, blockBlueScore uint64) error {
for _, input := range transaction.Inputs {
err := diffRemoveEntry(utxoDiff, &input.PreviousOutpoint, input.UTXOEntry)
if err != nil {
return err
}
}
isCoinbase := transactionhelper.IsCoinBase(transaction)
for i, output := range transaction.Outputs {
outpoint := &externalapi.DomainOutpoint{
TransactionID: *hashserialization.TransactionID(transaction),
Index: uint32(i),
}
entry := &externalapi.UTXOEntry{
Amount: output.Value,
ScriptPublicKey: output.ScriptPublicKey,
BlockBlueScore: blockBlueScore,
IsCoinbase: isCoinbase,
}
err := diffAddEntry(utxoDiff, outpoint, entry)
if err != nil {
return err
}
}
return nil
}
func diffAddEntry(diff *model.UTXODiff, outpoint *externalapi.DomainOutpoint, entry *externalapi.UTXOEntry) error {
if collectionContainsWithBlueScore(diff.ToRemove, outpoint, entry.BlockBlueScore) {
collectionRemove(diff.ToRemove, outpoint)
} else if _, exists := diff.ToAdd[*outpoint]; exists {
return errors.Errorf("AddEntry: Cannot add outpoint %s twice", outpoint)
} else {
collectionAdd(diff.ToAdd, outpoint, entry)
}
return nil
}
func diffRemoveEntry(diff *model.UTXODiff, outpoint *externalapi.DomainOutpoint, entry *externalapi.UTXOEntry) error {
if collectionContainsWithBlueScore(diff.ToAdd, outpoint, entry.BlockBlueScore) {
collectionRemove(diff.ToAdd, outpoint)
} else if _, exists := diff.ToRemove[*outpoint]; exists {
return errors.Errorf("removeEntry: Cannot remove outpoint %s twice", outpoint)
} else {
collectionAdd(diff.ToRemove, outpoint, entry)
}
return nil
}
func diffEqual(this *model.UTXODiff, other *model.UTXODiff) bool {
if this == nil || other == nil {
return this == other
}
return reflect.DeepEqual(this.ToAdd, other.ToAdd) &&
reflect.DeepEqual(this.ToRemove, other.ToRemove)
}