diff --git a/contracts/validator-manager/ValidatorMessages.sol b/contracts/validator-manager/ValidatorMessages.sol index 95fa0028..a5998193 100644 --- a/contracts/validator-manager/ValidatorMessages.sol +++ b/contracts/validator-manager/ValidatorMessages.sol @@ -353,7 +353,7 @@ library ValidatorMessages { pure returns (bytes32, bool) { - if(input.length != 39) { + if (input.length != 39) { revert InvalidMessageLength(); } // Unpack the codec ID @@ -430,7 +430,7 @@ library ValidatorMessages { pure returns (bytes32, uint64, uint64) { - if(input.length != 54) { + if (input.length != 54) { revert InvalidMessageLength(); } @@ -530,7 +530,7 @@ library ValidatorMessages { for (uint256 i; i < 4; ++i) { typeID |= uint32(uint8(input[i + 2])) << uint32((8 * (3 - i))); } - + if (typeID != SUBNET_VALIDATOR_WEIGHT_UPDATE_MESSAGE_TYPE_ID) { revert InvalidMessageType(); } diff --git a/contracts/validator-manager/tests/PoSValidatorManagerTests.t.sol b/contracts/validator-manager/tests/PoSValidatorManagerTests.t.sol index e75f8e66..ffa98ce5 100644 --- a/contracts/validator-manager/tests/PoSValidatorManagerTests.t.sol +++ b/contracts/validator-manager/tests/PoSValidatorManagerTests.t.sol @@ -15,7 +15,6 @@ import { } from "@avalabs/subnet-evm-contracts@1.2.0/contracts/interfaces/IWarpMessenger.sol"; import {ValidatorMessages} from "../ValidatorMessages.sol"; import {ValidatorRegistrationInput, ValidatorStatus} from "../interfaces/IValidatorManager.sol"; -import {DelegatorStatus} from "../interfaces/IPoSValidatorManager.sol"; abstract contract PoSValidatorManagerTest is ValidatorManagerTest { uint64 public constant DEFAULT_UPTIME = uint64(100);