fix: Add exception handling code to HeightVoteSet for votes with null…

… power
planetarium · Oct 22, 2024 · 61ffe8c · 61ffe8c
1 parent 4113b97
commit 61ffe8c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/Libplanet.Net/Consensus/HeightVoteSet.cs b/src/Libplanet.Net/Consensus/HeightVoteSet.cs
@@ -132,8 +132,12 @@ public void AddVote(Vote vote)
                         vote);
                 }
 
-                if (vote.ValidatorPower is { } power &&
-                    _validatorSet.GetValidator(validatorKey).Power != power)
+                if (vote.ValidatorPower is not { } power)
+                {
+                    const string msg = "ValidatorPower of the vote cannot be null";
+                    throw new InvalidVoteException(msg, vote);
+                }
+                else if (_validatorSet.GetValidator(validatorKey).Power != power)
                 {
                     const string msg = "ValidatorPower of the vote is given and the value is " +
                                        "not the same with the one in the validator set";