validate validator sets on initialization

Snowfork · Nov 27, 2023 · db994ef · db994ef
1 parent 4dd37e6
commit db994ef
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/contracts/src/BeefyClient.sol b/contracts/src/BeefyClient.sol
@@ -199,6 +199,7 @@ contract BeefyClient {
     error InvalidSignature();
     error InvalidTicket();
     error InvalidValidatorProof();
+    error InvalidValidatorSetData();
     error CommitmentNotRelevant();
     error NotEnoughClaims();
     error PrevRandaoAlreadyCaptured();
@@ -215,6 +216,9 @@ contract BeefyClient {
         ValidatorSet memory _initialValidatorSet,
         ValidatorSet memory _nextValidatorSet
     ) {
+        if (_nextValidatorSet.id != _initialValidatorSet.id + 1) {
+            revert InvalidValidatorSetData();
+        }
         randaoCommitDelay = _randaoCommitDelay;
         randaoCommitExpiration = _randaoCommitExpiration;
         minNumRequiredSignatures = _minNumRequiredSignatures;

diff --git a/contracts/test/BeefyClient.t.sol b/contracts/test/BeefyClient.t.sol
@@ -730,4 +730,18 @@ contract BeefyClientTest is Test {
     function testStorageToStorageCopies() public {
         beefyClient.copyCounters();
     }
+
+    function testFuzzInitializationValidation(uint128 currentId, uint128 nextId) public {
+        vm.assume(currentId < type(uint128).max);
+        vm.assume(currentId + 1 != nextId);
+        vm.expectRevert(BeefyClient.InvalidValidatorSetData.selector);
+        new BeefyClient(
+            randaoCommitDelay,
+            randaoCommitExpiration,
+            minNumRequiredSignatures,
+            0,
+            BeefyClient.ValidatorSet(currentId, 0, 0x0),
+            BeefyClient.ValidatorSet(nextId, 0, 0x0)
+        );
+    }
 }
diff --git a/contracts/test/mocks/BeefyClientMock.sol b/contracts/test/mocks/BeefyClientMock.sol
@@ -15,7 +15,7 @@ contract BeefyClientMock is BeefyClient {
             minNumRequiredSignatures,
             0,
             BeefyClient.ValidatorSet(0, 0, 0x0),
-            BeefyClient.ValidatorSet(0, 0, 0x0)
+            BeefyClient.ValidatorSet(1, 0, 0x0)
         )
     {}