fix(http): fix java.lang.IllegalStateException: STREAMED

framework/build.gradle

@@ -85,6 +85,7 @@ dependencies {
     compile group: 'org.pf4j', name: 'pf4j', version: '2.5.0'
 
     testImplementation group: 'org.springframework', name: 'spring-test', version: '5.2.0.RELEASE'
+    testImplementation group: 'org.springframework', name: 'spring-web', version: '5.2.0.RELEASE'
 
     compile group: 'org.zeromq', name: 'jeromq', version: '0.5.3'
     compile project(":chainbase")

framework/src/main/java/org/tron/core/services/http/Util.java

@@ -11,19 +11,22 @@
 import com.google.protobuf.GeneratedMessageV3;
 import com.google.protobuf.InvalidProtocolBufferException;
 import com.google.protobuf.Message;
+
+import java.io.BufferedReader;
 import java.io.IOException;
+import java.io.InputStreamReader;
 import java.lang.reflect.Constructor;
 import java.math.BigDecimal;
 import java.nio.charset.Charset;
 import java.security.InvalidParameterException;
 import java.util.ArrayList;
 import java.util.List;
-import java.util.stream.Collectors;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
 import org.bouncycastle.util.encoders.Hex;
+import org.eclipse.jetty.http.HttpMethod;
 import org.eclipse.jetty.util.StringUtil;
 import org.tron.api.GrpcAPI;
 import org.tron.api.GrpcAPI.BlockList;
@@ -70,6 +73,8 @@ public class Util {
   public static final String FUNCTION_SELECTOR = "function_selector";
   public static final String FUNCTION_PARAMETER = "parameter";
   public static final String CALL_DATA = "data";
+  public static final String APPLICATION_FORM_URLENCODED = "application/x-www-form-urlencoded";
+  public static final String APPLICATION_JSON = "application/json";
 
   public static String printTransactionFee(String transactionFee) {
     JSONObject jsonObject = new JSONObject();
@@ -498,16 +503,7 @@ public static void printAccount(Account reply, HttpServletResponse response, Boo
   public static byte[] getAddress(HttpServletRequest request) throws Exception {
     byte[] address = null;
     String addressParam = "address";
-    String addressStr = request.getParameter(addressParam);
-    if (StringUtils.isBlank(addressStr)) {
-      String input = request.getReader().lines()
-          .collect(Collectors.joining(System.lineSeparator()));
-      Util.checkBodySize(input);
-      JSONObject jsonObject = JSON.parseObject(input);
-      if (jsonObject != null) {
-        addressStr = jsonObject.getString(addressParam);
-      }
-    }
+    String addressStr = checkGetParam(request, addressParam);
     if (StringUtils.isNotBlank(addressStr)) {
       if (StringUtils.startsWith(addressStr, Constant.ADD_PRE_FIX_STRING_MAINNET)) {
         address = Hex.decode(addressStr);
@@ -518,6 +514,47 @@ public static byte[] getAddress(HttpServletRequest request) throws Exception {
     return address;
   }
 
+  private static String checkGetParam(HttpServletRequest request, String key) throws Exception {
+    String method = request.getMethod();
+    String value = null;
+
+    if (HttpMethod.GET.toString().toUpperCase() .equalsIgnoreCase(method)) {
+      return request.getParameter(key);
+    }
+    if (HttpMethod.POST.toString().toUpperCase().equals(method)) {
+      String contentType = request.getContentType();
+      if (StringUtils.isBlank(contentType)) {
+        throw new IllegalArgumentException("Invalid request parameter");
+      }
+      if (APPLICATION_JSON.toLowerCase().contains(contentType)){
+        value = getRequestValue(request);
+        if (StringUtils.isBlank(value)) {
+          return null;
+        }
+
+        JSONObject jsonObject = JSON.parseObject(value);
+        if (jsonObject != null) {
+          return jsonObject.getString(key);
+        }
+      } else if (APPLICATION_FORM_URLENCODED.toLowerCase().contains(contentType)) {
+        return request.getParameter(key);
+      } else {
+        return null;
+      }
+    }
+    return value;
+  }
+
+  public static String getRequestValue(HttpServletRequest request) throws IOException {
+    BufferedReader reader = new BufferedReader(new InputStreamReader(request.getInputStream()));
+    String line;
+    StringBuilder sb = new StringBuilder();
+    while ((line = reader.readLine()) != null) {
+      sb.append(line);
+    }
+    return sb.toString();
+  }
+
   public static List<Log> convertLogAddressToTronAddress(TransactionInfo transactionInfo) {
     List<Log> newLogList = new ArrayList<>();
 

framework/src/test/java/org/tron/core/services/http/GetBrokerageServletTest.java

@@ -0,0 +1,46 @@
+package org.tron.core.services.http;
+
+import static org.junit.Assert.assertEquals;
+
+import com.alibaba.fastjson.JSONObject;
+
+import java.io.UnsupportedEncodingException;
+
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+
+public class GetBrokerageServletTest {
+
+  private MockHttpServletRequest request;
+  private MockHttpServletResponse response;
+
+  @Before
+  public void setUp() {
+    request = new MockHttpServletRequest();
+    request.setMethod("POST");
+    request.setContentType("application/x-www-form-urlencoded");
+    request.setCharacterEncoding("UTF-8");
+
+    response = new MockHttpServletResponse();
+  }
+
+  @Test
+  public void getBrokerageTest() {
+    request.addParameter("address", "");
+    GetRewardServlet getRewardServlet = new GetRewardServlet();
+    getRewardServlet.doPost(request, response);
+    try {
+      String contentAsString = response.getContentAsString();
+      JSONObject result = JSONObject.parseObject(contentAsString);
+      int reward = (int)result.get("reward");
+      Assert.assertEquals(0, reward);
+      String content = (String) result.get("Error");
+      Assert.assertNull(content);
+    } catch (UnsupportedEncodingException e) {
+      e.printStackTrace();
+    }
+  }
+}

framework/src/test/java/org/tron/core/services/http/GetRewardServletTest.java

@@ -0,0 +1,44 @@
+package org.tron.core.services.http;
+
+import com.alibaba.fastjson.JSONObject;
+
+import java.io.UnsupportedEncodingException;
+
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.springframework.mock.web.MockHttpServletRequest;
+import org.springframework.mock.web.MockHttpServletResponse;
+
+public class GetRewardServletTest {
+
+  private MockHttpServletRequest request;
+  private MockHttpServletResponse response;
+
+  @Before
+  public void setUp() {
+    request = new MockHttpServletRequest();
+    request.setMethod("POST");
+    request.setContentType("application/x-www-form-urlencoded");
+    request.setCharacterEncoding("UTF-8");
+
+    response = new MockHttpServletResponse();
+  }
+
+  @Test
+  public void getRewardTest() {
+    request.addParameter("address", "");
+    GetRewardServlet getRewardServlet = new GetRewardServlet();
+    getRewardServlet.doPost(request, response);
+    try {
+      String contentAsString = response.getContentAsString();
+      JSONObject result = JSONObject.parseObject(contentAsString);
+      int reward = (int)result.get("reward");
+      Assert.assertEquals(0, reward);
+      String content = (String) result.get("Error");
+      Assert.assertNull(content);
+    } catch (UnsupportedEncodingException e) {
+      e.printStackTrace();
+    }
+  }
+}