add mutex

Author: codchen

store/gaskv/store.go

@@ -73,6 +73,9 @@ func (gs *Store) Set(key []byte, value []byte) {
 	gs.gasMeter.ConsumeGas(gs.gasConfig.WriteCostPerByte*types.Gas(len(key)), types.GasWritePerByteDesc)
 	gs.gasMeter.ConsumeGas(gs.gasConfig.WriteCostPerByte*types.Gas(len(value)), types.GasWritePerByteDesc)
 	gs.parent.Set(key, value)
+	if gs.tracer != nil {
+		gs.tracer.Set(key, value, gs.moduleName)
+	}
 }
 
 // Implements KVStore.
@@ -92,6 +95,9 @@ func (gs *Store) Delete(key []byte) {
 	// charge gas to prevent certain attack vectors even though space is being freed
 	gs.gasMeter.ConsumeGas(gs.gasConfig.DeleteCost, types.GasDeleteDesc)
 	gs.parent.Delete(key)
+	if gs.tracer != nil {
+		gs.tracer.Delete(key, gs.moduleName)
+	}
 }
 
 // Iterator implements the KVStore interface. It returns an iterator which

types/tracer.go

@@ -3,10 +3,12 @@ package types
 import (
 	"encoding/hex"
 	"encoding/json"
+	"sync"
 )
 
 type StoreTracer struct {
 	Modules map[string]*ModuleTrace
+	mu      *sync.Mutex
 }
 
 type ModuleTrace struct {
@@ -31,10 +33,13 @@ const (
 func NewStoreTracer() *StoreTracer {
 	return &StoreTracer{
 		Modules: map[string]*ModuleTrace{},
+		mu:      &sync.Mutex{},
 	}
 }
 
 func (st *StoreTracer) Get(key []byte, value []byte, module string) {
+	st.mu.Lock()
+	defer st.mu.Unlock()
 	accesses := st.getOrSetModuleTrace(module)
 	accesses.Accesses = append(accesses.Accesses, Access{
 		Op:    Get,
@@ -44,6 +49,8 @@ func (st *StoreTracer) Get(key []byte, value []byte, module string) {
 }
 
 func (st *StoreTracer) Set(key []byte, value []byte, module string) {
+	st.mu.Lock()
+	defer st.mu.Unlock()
 	accesses := st.getOrSetModuleTrace(module)
 	accesses.Accesses = append(accesses.Accesses, Access{
 		Op:    Set,
@@ -53,6 +60,8 @@ func (st *StoreTracer) Set(key []byte, value []byte, module string) {
 }
 
 func (st *StoreTracer) Has(key []byte, module string) {
+	st.mu.Lock()
+	defer st.mu.Unlock()
 	accesses := st.getOrSetModuleTrace(module)
 	accesses.Accesses = append(accesses.Accesses, Access{
 		Op:  Has,
@@ -61,6 +70,8 @@ func (st *StoreTracer) Has(key []byte, module string) {
 }
 
 func (st *StoreTracer) Delete(key []byte, module string) {
+	st.mu.Lock()
+	defer st.mu.Unlock()
 	accesses := st.getOrSetModuleTrace(module)
 	accesses.Accesses = append(accesses.Accesses, Access{
 		Op:  Delete,
@@ -90,6 +101,8 @@ type ModuleTraceDump struct {
 }
 
 func (st *StoreTracer) DerivePrestateToJson() []byte {
+	st.mu.Lock()
+	defer st.mu.Unlock()
 	d := StoreTraceDump{
 		Modules: make(map[string]ModuleTraceDump, len(st.Modules)),
 	}