core: fix light node synchronization issue (#30)

InsertChain() need to record the correct roundHeight mapping in order to
process snapshotRoundHeight() governance method correctly.

1 files changed, 11 insertions, 9 deletions

diff --git a/core/vm/governance.go b/core/vm/governance.go
index b38733e1a..f406294d2 100644
--- a/core/vm/governance.go
+++ b/core/vm/governance.go
@@ -1767,16 +1767,18 @@ func (g *GovernanceContract) transferOwnership(newOwner common.Address) ([]byte,
 }
 
 func (g *GovernanceContract) snapshotRound(round, height *big.Int) ([]byte, error) {
-	// Validate if this mapping is correct.
-	realHeight, ok := g.evm.Context.GetRoundHeight(round.Uint64())
-	if !ok {
-		g.penalize()
-		return nil, errExecutionReverted
-	}
+	// Validate if this mapping is correct. Only block proposer need to verify this.
+	if g.evm.IsBlockProposer() {
+		realHeight, ok := g.evm.Context.GetRoundHeight(round.Uint64())
+		if !ok {
+			g.penalize()
+			return nil, errExecutionReverted
+		}
 
-	if height.Cmp(new(big.Int).SetUint64(realHeight)) != 0 {
-		g.penalize()
-		return nil, errExecutionReverted
+		if height.Cmp(new(big.Int).SetUint64(realHeight)) != 0 {
+			g.penalize()
+			return nil, errExecutionReverted
+		}
 	}
 
 	// Only allow updating the next round.