update ferveo pk length

nucypher · Jul 3, 2023 · 3021892 · 3021892
1 parent 48d5794
commit 3021892
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/contracts/contracts/coordination/Coordinator.sol b/contracts/contracts/coordination/Coordinator.sol
@@ -127,8 +127,7 @@ contract Coordinator is AccessControlDefaultAdminRules {
     }
 
     function setProviderPublicKey(bytes calldata publicKey) external {
-        // TODO: Verify public key length
-        require(publicKey.length == 48, "Invalid public key length");
+        require(publicKey.length == 104, "Invalid public key length");
         providerPublicKey[msg.sender] = publicKey;
     }
 

diff --git a/tests/test_coordinator.py b/tests/test_coordinator.py
@@ -101,7 +101,7 @@ def test_invalid_initiate_ritual(coordinator, nodes, accounts, initiator):
         coordinator.initiateRitual(nodes, initiator, DURATION, sender=initiator)
 
     for node in nodes:
-        coordinator.setProviderPublicKey(os.urandom(48), sender=node)
+        coordinator.setProviderPublicKey(os.urandom(104), sender=node)
     with ape.reverts("Providers must be sorted"):
         coordinator.initiateRitual(nodes[1:] + [nodes[0]], initiator, DURATION, sender=initiator)
 
@@ -112,7 +112,7 @@ def test_invalid_initiate_ritual(coordinator, nodes, accounts, initiator):
 
 def initiate_ritual(coordinator, erc20, flat_rate_fee_model, initiator, nodes):
     for node in nodes:
-        coordinator.setProviderPublicKey(os.urandom(48), sender=node)
+        coordinator.setProviderPublicKey(os.urandom(104), sender=node)
     cost = flat_rate_fee_model.getRitualInitiationCost(nodes, DURATION)
     erc20.approve(coordinator.address, cost, sender=initiator)
     tx = coordinator.initiateRitual(nodes, initiator, DURATION, sender=initiator)