Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-10-14 00:59:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NOD-400] Fix ECHM cache crashing on concurrent access. (#450)

Browse Source
This commit is contained in:
stasatdaglabs 2019-11-04 17:38:01 +02:00 committed by Ori Newman
parent 13f06ca293
commit a8f0d7b05b
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
blockdag/utxo_ecmh.go
View File

@ -6,11 +6,15 @@ import (
"github.com/daglabs/btcd/util/daghash" "github.com/daglabs/btcd/util/daghash"
"github.com/daglabs/btcd/wire" "github.com/daglabs/btcd/wire"
"github.com/golang/groupcache/lru" "github.com/golang/groupcache/lru"
"sync"
) )
const ecmhCacheSize = 4_000_000 const ecmhCacheSize = 4_000_000
var ecmhCache = lru.New(ecmhCacheSize) var (
ecmhCache = lru.New(ecmhCacheSize)
ecmhCacheLock sync.Mutex
)
func utxoMultiset(entry *UTXOEntry, outpoint *wire.Outpoint) (*btcec.Multiset, error) { func utxoMultiset(entry *UTXOEntry, outpoint *wire.Outpoint) (*btcec.Multiset, error) {
w := &bytes.Buffer{} w := &bytes.Buffer{}
@ -20,6 +24,10 @@ func utxoMultiset(entry *UTXOEntry, outpoint *wire.Outpoint) (*btcec.Multiset, e
} }
serializedUTXO := w.Bytes() serializedUTXO := w.Bytes()
utxoHash := daghash.DoubleHashH(serializedUTXO) utxoHash := daghash.DoubleHashH(serializedUTXO)
ecmhCacheLock.Lock()
defer ecmhCacheLock.Unlock()
if cachedMSPoint, ok := ecmhCache.Get(utxoHash); ok { if cachedMSPoint, ok := ecmhCache.Get(utxoHash); ok {
return cachedMSPoint.(*btcec.Multiset), nil return cachedMSPoint.(*btcec.Multiset), nil
} }