fix(dpp): add input validation

dashpay · Oct 2, 2024 · e03289d · e03289d
1 parent 3454692
commit e03289d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/rs-dpp/src/identity/identity_public_key/random.rs b/packages/rs-dpp/src/identity/identity_public_key/random.rs
@@ -583,6 +583,14 @@ impl IdentityPublicKey {
                 "at least 2 keys must be created".to_string(),
             ));
         }
+
+        if key_count > 16 {
+            return Err(ProtocolError::PublicKeyGenerationError(format!(
+                "too many keys requested: {}, max is 16",
+                key_count
+            )));
+        }
+
         //create a master and a high level key
         let mut main_keys = if key_count == 2 {
             vec![