fix(platform): add limits to identity balances and proposed block cou…

…nts queries (#2148)

packages/rs-drive-abci/src/error/query.rs

@@ -40,6 +40,10 @@ pub enum QueryError {
     #[error("invalid argument error: {0}")]
     InvalidArgument(String),
 
+    /// Too many elements Error
+    #[error("too many elements error: {0}")]
+    TooManyElements(String),
+
     /// Not found Error
     #[error("not found error: {0}")]
     NotFound(String),

packages/rs-drive-abci/src/query/validator_queries/proposed_block_counts_by_evonode_ids/v0/mod.rs

@@ -21,6 +21,14 @@ impl<C> Platform<C> {
         platform_state: &PlatformState,
         platform_version: &PlatformVersion,
     ) -> Result<QueryValidationResult<GetEvonodesProposedEpochBlocksResponseV0>, Error> {
+        if ids.len() > self.config.drive.max_query_limit as usize {
+            return Ok(QueryValidationResult::new_with_error(
+                QueryError::TooManyElements(format!(
+                    "this query only supports up to {} ids at a time",
+                    ids.len()
+                )),
+            ));
+        }
         let evonode_ids = check_validation_result_with_data!(ids
             .into_iter()
             .map(|evonode_id_vec| {

packages/rs-drive/src/drive/identity/fetch/mod.rs

@@ -180,7 +180,7 @@ impl Drive {
             path: vec![vec![RootTree::Balances as u8]],
             query: SizedQuery {
                 query,
-                limit: None,
+                limit: Some(self.config.max_query_limit),
                 offset: None,
             },
         };