fix race conditions

Makefile

@@ -3,6 +3,9 @@ all: build
 build:
 	go build -o tendermint github.com/tendermint/tendermint/cmd
 
+build_race:
+	go build -race -o tendermint github.com/tendermint/tendermint/cmd
+
 test:
 	go test github.com/tendermint/tendermint/...
 

binary/reflect.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"io"
 	"reflect"
+	"sync"
 )
 
 type TypeInfo struct {
@@ -27,8 +28,23 @@ type HasTypeByte interface {
 	TypeByte() byte
 }
 
+// NOTE: do not access typeInfos directly, but call GetTypeInfo()
+var typeInfosMtx sync.Mutex
 var typeInfos = map[reflect.Type]*TypeInfo{}
 
+func GetTypeInfo(rt reflect.Type) *TypeInfo {
+	typeInfosMtx.Lock()
+	defer typeInfosMtx.Unlock()
+	info := typeInfos[rt]
+	if info == nil {
+		info = &TypeInfo{Type: rt}
+		RegisterType(info)
+	}
+	return info
+}
+
+// Registers and possibly modifies the TypeInfo.
+// NOTE: not goroutine safe, so only call upon program init.
 func RegisterType(info *TypeInfo) *TypeInfo {
 
 	// Register the type info
@@ -74,10 +90,7 @@ func readReflect(rv reflect.Value, rt reflect.Type, r io.Reader, n *int64, err *
 	}
 
 	// Get typeInfo
-	typeInfo := typeInfos[rt]
+	typeInfo := GetTypeInfo(rt)
-	if typeInfo == nil {
-		typeInfo = RegisterType(&TypeInfo{Type: rt})
-	}
 
 	// Custom decoder
 	if typeInfo.Decoder != nil {
@@ -178,10 +191,7 @@ func readReflect(rv reflect.Value, rt reflect.Type, r io.Reader, n *int64, err *
 func writeReflect(rv reflect.Value, rt reflect.Type, w io.Writer, n *int64, err *error) {
 
 	// Get typeInfo
-	typeInfo := typeInfos[rt]
+	typeInfo := GetTypeInfo(rt)
-	if typeInfo == nil {
-		typeInfo = RegisterType(&TypeInfo{Type: rt})
-	}
 
 	// Custom encoder, say for an interface type rt.
 	if typeInfo.Encoder != nil {
@@ -193,11 +203,8 @@ func writeReflect(rv reflect.Value, rt reflect.Type, w io.Writer, n *int64, err
 	if rt.Kind() == reflect.Interface {
 		rv = rv.Elem()
 		rt = rv.Type()
-		typeInfo = typeInfos[rt]
 		// If interface type, get typeInfo of underlying type.
-		if typeInfo == nil {
+		typeInfo = GetTypeInfo(rt)
-			typeInfo = RegisterType(&TypeInfo{Type: rt})
-		}
 	}
 
 	// Dereference pointer

consensus/reactor.go

@@ -323,7 +323,11 @@ OUTER_LOOP:
 
 		trySendVote := func(voteSet *VoteSet, peerVoteSet BitArray) (sent bool) {
 			// TODO: give priority to our vote.
+			// peerVoteSet BitArray is being accessed concurrently with
+			// writes from Receive() routines.  We must lock like so here:
+			ps.mtx.Lock()
 			index, ok := voteSet.BitArray().Sub(peerVoteSet).PickRandom()
+			ps.mtx.Unlock()
 			if ok {
 				vote := voteSet.GetByIndex(index)
 				// NOTE: vote may be a commit.