Fix data race b/c of lacking lock protection for "data" slice

Signed-off-by: Anton Litvinov <[email protected]>
mysteriumnetwork · Mar 18, 2024 · cc88890 · cc88890
1 parent 1297ff7
commit cc88890
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/session/pingpong/consumer_totals_storage.go b/session/pingpong/consumer_totals_storage.go
@@ -94,6 +94,9 @@ func (cts *ConsumerTotalsStorage) Store(chainID int64, id identity.Identity, her
 // Get fetches the amount as promised for the given channel.
 func (cts *ConsumerTotalsStorage) Get(chainID int64, id identity.Identity, hermesID common.Address) (*big.Int, error) {
 	key := cts.makeKey(chainID, id, hermesID)
+	cts.createLock.Lock()
+    defer cts.createLock.Unlock()
+
 	element, ok := cts.data[key]
 	if !ok {
 		return nil, ErrNotFound