restrict number of block filter

Author: 317787106

common/src/main/java/org/tron/core/exception/jsonrpc/JsonRpcTooManyFilterException.java

@@ -0,0 +1,16 @@
+package org.tron.core.exception.jsonrpc;
+
+public class JsonRpcTooManyFilterException extends JsonRpcException {
+
+  public JsonRpcTooManyFilterException() {
+    super();
+  }
+
+  public JsonRpcTooManyFilterException(String message) {
+    super(message);
+  }
+
+  public JsonRpcTooManyFilterException(String message, Throwable cause) {
+    super(message, cause);
+  }
+}

framework/src/main/java/org/tron/core/services/jsonrpc/TronJsonRpc.java

@@ -23,13 +23,17 @@
 import org.tron.core.exception.jsonrpc.JsonRpcInvalidParamsException;
 import org.tron.core.exception.jsonrpc.JsonRpcInvalidRequestException;
 import org.tron.core.exception.jsonrpc.JsonRpcMethodNotFoundException;
+import org.tron.core.exception.jsonrpc.JsonRpcTooManyFilterException;
 import org.tron.core.exception.jsonrpc.JsonRpcTooManyResultException;
 import org.tron.core.services.jsonrpc.types.BlockResult;
 import org.tron.core.services.jsonrpc.types.BuildArguments;
 import org.tron.core.services.jsonrpc.types.CallArguments;
 import org.tron.core.services.jsonrpc.types.TransactionReceipt;
 import org.tron.core.services.jsonrpc.types.TransactionResult;
 
+/**
+ * Error code refers to https://www.quicknode.com/docs/ethereum/error-references
+ */
 @Component
 public interface TronJsonRpc {
 
@@ -293,9 +297,10 @@ String newFilter(FilterRequest fr) throws JsonRpcInvalidParamsException,
 
   @JsonRpcMethod("eth_newBlockFilter")
   @JsonRpcErrors({
+      @JsonRpcError(exception = JsonRpcTooManyFilterException.class, code = -32005, data = "{}"),
       @JsonRpcError(exception = JsonRpcMethodNotFoundException.class, code = -32601, data = "{}"),
   })
-  String newBlockFilter() throws JsonRpcMethodNotFoundException;
+  String newBlockFilter() throws JsonRpcTooManyFilterException, JsonRpcMethodNotFoundException;
 
   @JsonRpcMethod("eth_uninstallFilter")
   @JsonRpcErrors({

framework/src/main/java/org/tron/core/services/jsonrpc/TronJsonRpcImpl.java

@@ -65,6 +65,7 @@
 import org.tron.core.exception.jsonrpc.JsonRpcInvalidParamsException;
 import org.tron.core.exception.jsonrpc.JsonRpcInvalidRequestException;
 import org.tron.core.exception.jsonrpc.JsonRpcMethodNotFoundException;
+import org.tron.core.exception.jsonrpc.JsonRpcTooManyFilterException;
 import org.tron.core.exception.jsonrpc.JsonRpcTooManyResultException;
 import org.tron.core.services.NodeInfoService;
 import org.tron.core.services.http.JsonFormat;
@@ -112,6 +113,7 @@ public enum RequestSource {
 
   private static final String FILTER_NOT_FOUND = "filter not found";
   public static final int EXPIRE_SECONDS = 5 * 60;
+  private static final int MAX_FILTER_NUM = 100000;
   private static final Cache<LogFilterElement, LogFilterElement> logElementCache =
       CacheBuilder.newBuilder()
           .maximumSize(300_000L) // 300s * tps(1000) * 1 log/tx ≈ 300_000
@@ -1424,7 +1426,8 @@ public String newFilter(FilterRequest fr) throws JsonRpcInvalidParamsException,
   }
 
   @Override
-  public String newBlockFilter() throws JsonRpcMethodNotFoundException {
+  public String newBlockFilter() throws JsonRpcMethodNotFoundException,
+      JsonRpcTooManyFilterException {
     disableInPBFT("eth_newBlockFilter");
 
     Map<String, BlockFilterAndResult> blockFilter2Result;
@@ -1433,6 +1436,9 @@ public String newBlockFilter() throws JsonRpcMethodNotFoundException {
     } else {
       blockFilter2Result = blockFilter2ResultSolidity;
     }
+    if (blockFilter2Result.size() >= MAX_FILTER_NUM) {
+      throw new JsonRpcTooManyFilterException("Block filter's num exceeds " + MAX_FILTER_NUM);
+    }
 
     BlockFilterAndResult filterAndResult = new BlockFilterAndResult();
     String filterID = generateFilterId();