core: vm: add extra checks to prevent DKG spamming (#272)

Add two maps to check the uniqueness of DKGMasterPublicKey and DKGComplaints to
prevent malicious actors from spamming it.

1 files changed, 4 insertions, 4 deletions

diff --git a/core/governance.go b/core/governance.go
index 73fb0c923..db0e60b6c 100644
--- a/core/governance.go
+++ b/core/governance.go
@@ -93,10 +93,6 @@ func (g *Governance) GetStateAtRound(round uint64) *vm.GovernanceState {
 	return &vm.GovernanceState{StateDB: s}
 }
 
-func (g *Governance) GetRoundHeight(round uint64) uint64 {
-	return g.GetHeadState().RoundHeight(big.NewInt(int64(round))).Uint64()
-}
-
 func (g *Governance) Configuration(round uint64) *coreTypes.Config {
 	configHelper := g.GetStateForConfigAtRound(round)
 	c := configHelper.Configuration()
@@ -110,6 +106,10 @@ func (g *Governance) Configuration(round uint64) *coreTypes.Config {
 	}
 }
 
+func (g *Governance) GetRoundHeight(round uint64) uint64 {
+	return g.GetHeadState().RoundHeight(big.NewInt(int64(round))).Uint64()
+}
+
 func (g *Governance) GetStateForDKGAtRound(round uint64) *vm.GovernanceState {
 	dkgRound := g.GetHeadState().DKGRound().Uint64()
 	if round > dkgRound {