added support for spent UTXO filter and index

Author: setavenger

src/common/types/spentoutpoints.go

@@ -0,0 +1,77 @@
+package types
+
+import (
+	"bytes"
+	"encoding/hex"
+	"errors"
+
+	"SilentPaymentAppBackend/src/common"
+)
+
+const LenOutpointHashShort = 8
+
+type SpentOutpointsIndex struct {
+	BlockHash   string                       `json:"block_hash"`
+	BlockHeight uint32                       `json:"block_height"`
+	Data        [][LenOutpointHashShort]byte `json:"data"`
+}
+
+func PairFactorySpentOutpointsIndex() Pair {
+	var filter Pair = &SpentOutpointsIndex{}
+	return filter
+}
+
+func (v *SpentOutpointsIndex) SerialiseKey() ([]byte, error) {
+	return GetDBKeyTweakIndex(v.BlockHash)
+}
+
+func (v *SpentOutpointsIndex) SerialiseData() ([]byte, error) {
+
+	// todo can this be made more efficiently?
+	totalLength := len(v.Data) * LenOutpointHashShort
+	flattened := make([]byte, 0, totalLength)
+
+	for _, byteArray := range v.Data {
+		flattened = append(flattened, byteArray[:]...)
+	}
+
+	return flattened, nil
+}
+
+func (v *SpentOutpointsIndex) DeSerialiseKey(key []byte) error {
+	if len(key) != 32 {
+		common.ErrorLogger.Printf("wrong key length: %+v", key)
+		return errors.New("key is wrong length. should not happen")
+	}
+
+	v.BlockHash = hex.EncodeToString(key)
+
+	return nil
+}
+
+func (v *SpentOutpointsIndex) DeSerialiseData(data []byte) error {
+	if len(data)%LenOutpointHashShort != 0 {
+		common.ErrorLogger.Printf("wrong data length: %+v", data)
+		return errors.New("data is wrong length. should not happen")
+	}
+
+	numArrays := len(data) / LenOutpointHashShort
+	v.Data = make([][LenOutpointHashShort]byte, numArrays)
+	// Iterate and copy segments from the flat slice into the new array of arrays
+	for i := 0; i < numArrays; i++ {
+		copy(v.Data[i][:], data[i*LenOutpointHashShort:(i+1)*LenOutpointHashShort])
+	}
+	return nil
+}
+
+func GetDBKeySpentSpentOutpointsIndex(blockHash string) ([]byte, error) {
+	var buf bytes.Buffer
+	blockHashBytes, err := hex.DecodeString(blockHash)
+	if err != nil {
+		common.ErrorLogger.Println(err)
+		return nil, err
+	}
+	buf.Write(blockHashBytes)
+
+	return buf.Bytes(), nil
+}

src/common/util.go

@@ -2,12 +2,14 @@ package common
 
 import (
 	"crypto/sha256"
-	"github.com/shopspring/decimal"
-	"golang.org/x/crypto/ripemd160"
 	"os/user"
 	"strings"
+
+	"github.com/shopspring/decimal"
+	"golang.org/x/crypto/ripemd160"
 )
 
+// ReverseBytes reverses the bytes inside the byte slice and returns the same slice. It does not return a copy.
 func ReverseBytes(bytes []byte) []byte {
 	for i, j := 0, len(bytes)-1; i < j; i, j = i+1, j-1 {
 		bytes[i], bytes[j] = bytes[j], bytes[i]

src/common/vars.go

@@ -55,12 +55,14 @@ var (
 
 // one has to call SetDirectories otherwise common.DBPath will be empty
 var (
-	DBPathHeaders    = DBPath + "/headers"
-	DBPathHeadersInv = DBPath + "/headers-inv" // for height to blockHash mapping
-	DBPathFilters    = DBPath + "/filters"
-	DBPathTweaks     = DBPath + "/tweaks"
-	DBPathTweakIndex = DBPath + "/tweak-index"
-	DBPathUTXOs      = DBPath + "/utxos"
+	DBPathHeaders              string
+	DBPathHeadersInv           string // for height to blockHash mapping
+	DBPathFilters              string
+	DBPathTweaks               string
+	DBPathTweakIndex           string
+	DBPathUTXOs                string
+	DBPathSpentOutpointsIndex  string
+	DBPathSpentOutpointsFilter string
 )
 
 // NumsH = 0x50929b74c1a04954b78b4b6035e97a5e078a5a0f28ec96d547bfee9ace803ac0
@@ -78,6 +80,8 @@ func SetDirectories() {
 	DBPathTweaks = DBPath + "/tweaks"
 	DBPathTweakIndex = DBPath + "/tweak-index"
 	DBPathUTXOs = DBPath + "/utxos"
+	DBPathSpentOutpointsIndex = DBPath + "/spent-index"
+	DBPathSpentOutpointsFilter = DBPath + "/spent-filter"
 }
 
 func HeaderMustSyncHeight() uint32 {

src/core/cfilter.go

@@ -3,14 +3,17 @@ package core
 import (
 	"SilentPaymentAppBackend/src/common"
 	"SilentPaymentAppBackend/src/common/types"
+	"bytes"
+	"encoding/binary"
 	"encoding/hex"
+
 	"github.com/btcsuite/btcd/btcutil/gcs/builder"
 	"github.com/btcsuite/btcd/chaincfg/chainhash"
 	"github.com/btcsuite/btcutil/gcs"
 )
 
 // BuildTaprootOnlyFilter creates the taproot only filter
-func BuildTaprootOnlyFilter(block *types.Block) (types.Filter, error) {
+func BuildNewUTXOsFilter(block *types.Block) (types.Filter, error) {
 	var taprootOutput [][]byte
 
 	for _, tx := range block.Txs {
@@ -64,3 +67,68 @@ func BuildTaprootOnlyFilter(block *types.Block) (types.Filter, error) {
 		BlockHash:   block.Hash,
 	}, nil
 }
+
+// BuildSpentUTXOsFilter creates a filter based on the spent
+func BuildSpentUTXOsFilter(spentOutpointsIndex types.SpentOutpointsIndex) (types.Filter, error) {
+
+	blockHashBytes, err := hex.DecodeString(spentOutpointsIndex.BlockHash)
+	if err != nil {
+		common.DebugLogger.Println("blockHash", spentOutpointsIndex.BlockHash)
+		common.ErrorLogger.Fatalln(err)
+		return types.Filter{}, err
+	}
+	c := chainhash.Hash{}
+
+	err = c.SetBytes(common.ReverseBytes(blockHashBytes))
+	if err != nil {
+		common.DebugLogger.Println("blockHash", spentOutpointsIndex.BlockHash)
+		common.ErrorLogger.Fatalln(err)
+		return types.Filter{}, err
+
+	}
+	key := builder.DeriveKey(&c)
+
+	// convert to slices
+	data := make([][]byte, len(spentOutpointsIndex.Data))
+	for i, outpointHash := range spentOutpointsIndex.Data {
+		var newHash [8]byte
+		copy(newHash[:], outpointHash[:])
+		data[i] = newHash[:]
+	}
+
+	filter, err := gcs.BuildGCSFilter(builder.DefaultP, builder.DefaultM, key, data)
+	if err != nil {
+		common.ErrorLogger.Fatalln(err)
+		return types.Filter{}, err
+	}
+
+	nBytes, err := filter.NBytes()
+	if err != nil {
+		common.ErrorLogger.Fatalln(err)
+		return types.Filter{}, err
+	}
+
+	return types.Filter{
+		FilterType:  4,
+		BlockHeight: spentOutpointsIndex.BlockHeight,
+		Data:        nBytes,
+		BlockHash:   spentOutpointsIndex.BlockHash,
+	}, nil
+}
+
+func SerialiseToOutpoint(utxo types.UTXO) ([]byte, error) {
+	var buf bytes.Buffer
+
+	txidBytes, err := hex.DecodeString(utxo.Txid)
+	if err != nil {
+		common.DebugLogger.Println(utxo.Txid)
+		common.ErrorLogger.Println(err)
+		return nil, err
+	}
+
+	// err is always nil
+	buf.Write(common.ReverseBytes(txidBytes))
+
+	binary.Write(&buf, binary.LittleEndian, utxo.Value)
+	return buf.Bytes(), err
+}

src/core/extractutxos.go

@@ -75,7 +75,6 @@ func extractSpentTaprootPubKeysFromTx(tx *types.Transaction, block *types.Block)
 				blockHash = headerInv.Hash
 			}
 
-			//832974
 			spentUTXOs = append(spentUTXOs, types.UTXO{
 				Txid:         vin.Txid,
 				Vout:         vin.Vout,

src/core/routine.go

@@ -12,19 +12,17 @@ import (
 func CheckForNewBlockRoutine() {
 	common.InfoLogger.Println("starting check_for_new_block_routine")
 	for {
-		select {
-		case <-time.NewTicker(3 * time.Second).C:
-			blockHash, err := GetBestBlockHash()
-			if err != nil {
-				common.ErrorLogger.Println(err)
-				// todo fail or restart after too many fails?
-				continue
-			}
-			err = FullProcessBlockHash(blockHash)
-			if err != nil {
-				common.ErrorLogger.Println(err)
-				return
-			}
+		<-time.NewTicker(3 * time.Second).C
+		blockHash, err := GetBestBlockHash()
+		if err != nil {
+			common.ErrorLogger.Println(err)
+			// todo fail or restart after too many fails?
+			continue
+		}
+		err = FullProcessBlockHash(blockHash)
+		if err != nil {
+			common.ErrorLogger.Println(err)
+			return
 		}
 	}
 }
@@ -58,7 +56,7 @@ func FullProcessBlockHash(blockHash string) error {
 func PullBlock(blockHash string) (*types.Block, error) {
 	if len(blockHash) != 64 {
 		common.ErrorLogger.Println("block_hash invalid:", blockHash)
-		return nil, errors.New(fmt.Sprintf("block_hash invalid: %s", blockHash))
+		return nil, fmt.Errorf("block_hash invalid: %s", blockHash)
 	}
 	// this method is preferred over lastHeader because then this function can be called for PreviousBlockHash
 	header, err := dblevel.FetchByBlockHashBlockHeader(blockHash)
@@ -185,12 +183,38 @@ func HandleBlock(block *types.Block) error {
 	}
 
 	// create special block filter
-	cFilterTaproot, err := BuildTaprootOnlyFilter(block)
+	cFilterNewUTXOs, err := BuildNewUTXOsFilter(block)
+	if err != nil {
+		common.ErrorLogger.Println(err)
+		return err
+	}
+
+	//
+	err = dblevel.InsertNewUTXOsFilter(cFilterNewUTXOs)
+	if err != nil {
+		common.ErrorLogger.Println(err)
+		return err
+	}
+
+	spentOutpointsIndex, err := BuildSpentUTXOIndex(taprootSpent, block)
+	if err != nil {
+		common.ErrorLogger.Println(err)
+		return err
+	}
+
+	err = dblevel.InsertSpentOutpointsIndex(&spentOutpointsIndex)
 	if err != nil {
 		common.ErrorLogger.Println(err)
 		return err
 	}
-	err = dblevel.InsertFilter(cFilterTaproot)
+
+	cFilterSpentUTXOs, err := BuildSpentUTXOsFilter(spentOutpointsIndex)
+	if err != nil {
+		common.ErrorLogger.Println(err)
+		return err
+	}
+
+	err = dblevel.InsertSpentOutpointsFilter(cFilterSpentUTXOs)
 	if err != nil {
 		common.ErrorLogger.Println(err)
 		return err

src/core/spentutxos.go

@@ -0,0 +1,36 @@
+package core
+
+import (
+	"SilentPaymentAppBackend/src/common"
+	"SilentPaymentAppBackend/src/common/types"
+	"crypto/sha256"
+	"encoding/hex"
+)
+
+func BuildSpentUTXOIndex(utxos []types.UTXO, block *types.Block) (types.SpentOutpointsIndex, error) {
+
+	blockHashBytes, err := hex.DecodeString(block.Hash)
+	if err != nil {
+		common.ErrorLogger.Println(err)
+		return types.SpentOutpointsIndex{}, err
+	}
+
+	spentOutpointsIndex := types.SpentOutpointsIndex{
+		BlockHash:   block.Hash,
+		BlockHeight: block.Height,
+	}
+
+	for _, utxo := range utxos {
+		var outpoint []byte
+		outpoint, err = SerialiseToOutpoint(utxo)
+		if err != nil {
+			common.ErrorLogger.Println(err)
+			return types.SpentOutpointsIndex{}, err
+		}
+
+		hashedOutpoint := sha256.Sum256(append(outpoint, blockHashBytes...))
+		spentOutpointsIndex.Data = append(spentOutpointsIndex.Data, [types.LenOutpointHashShort]byte(hashedOutpoint[:]))
+	}
+
+	return spentOutpointsIndex, err
+}

src/dataexport/exportcsv.go

@@ -78,7 +78,7 @@ func convertUTXOsToRecords(utxos []types.UTXO) ([][]string, error) {
 // Filters
 
 func ExportFilters(path string) error {
-	allEntries, err := dblevel.FetchAllFilters()
+	allEntries, err := dblevel.FetchAllNewUTXOsFilters()
 	if err != nil {
 		common.ErrorLogger.Println(err)
 		return err

src/db/dblevel/client.go

@@ -20,12 +20,14 @@ func (e NoEntryErr) Error() string {
 }
 
 var (
-	HeadersDB    *leveldb.DB
-	HeadersInvDB *leveldb.DB
-	FiltersDB    *leveldb.DB
-	TweaksDB     *leveldb.DB
-	TweakIndexDB *leveldb.DB
-	UTXOsDB      *leveldb.DB
+	HeadersDB              *leveldb.DB
+	HeadersInvDB           *leveldb.DB
+	NewUTXOsFiltersDB      *leveldb.DB
+	TweaksDB               *leveldb.DB
+	TweakIndexDB           *leveldb.DB
+	UTXOsDB                *leveldb.DB
+	SpentOutpointsIndexDB  *leveldb.DB
+	SpentOutpointsFilterDB *leveldb.DB
 )
 
 // OpenDBConnection opens a connection to the through path specified db instance
@@ -99,10 +101,10 @@ func retrieveByBlockHash(db *leveldb.DB, blockHash string, pair types.Pair) erro
 	}
 
 	data, err := db.Get(blockHashBytes, nil)
-	if err != nil && !errors.Is(err, NoEntryErr{}) { // todo this error probably exists as var/type somewhere
+	if err != nil && !errors.Is(err, leveldb.ErrNotFound) { // todo this error probably exists as var/type somewhere
 		common.ErrorLogger.Println(err)
 		return err
-	} else if err != nil && errors.Is(err, NoEntryErr{}) { // todo this error probably exists as var/type somewhere
+	} else if err != nil && errors.Is(err, leveldb.ErrNotFound) { // todo this error probably exists as var/type somewhere
 		// todo we don't need separate patterns if just return the errors anyways? or maybe just to avoid unnecessary logging
 		return NoEntryErr{}
 	}