feature: optimize trigger interface

chainbase/src/main/java/org/tron/core/db/TransactionTrace.java

@@ -255,7 +255,7 @@ public void pay() throws BalanceInsufficientException {
     AccountCapsule origin = accountStore.get(originAccount);
     AccountCapsule caller = accountStore.get(callerAccount);
     if (dynamicPropertiesStore.supportUnfreezeDelay()
-        && receipt.getReceipt().getResult().equals(contractResult.SUCCESS)) {
+        && getRuntimeResult().getException() == null && !getRuntimeResult().isRevert()) {
 
       // just fo caller is not origin, we set the related field for origin account
       if (origin != null && !caller.getAddress().equals(origin.getAddress())) {

framework/src/main/java/org/tron/core/Wallet.java

@@ -2842,7 +2842,7 @@ public Transaction triggerContract(TriggerSmartContract
         triggerSmartContract.getData().toByteArray());
 
     if (isConstant(abi, selector)) {
-      return callConstantContract(trxCap, builder, retBuilder);
+      return callConstantContract(trxCap, builder, retBuilder, false);
     } else {
       return trxCap.getInstance();
     }
@@ -2960,13 +2960,19 @@ private Transaction cleanContextAndTriggerConstantContract(
     txExtBuilder.clear();
     txRetBuilder.clear();
     transaction = triggerConstantContract(
-        triggerSmartContract, txCap, txExtBuilder, txRetBuilder);
+        triggerSmartContract, txCap, txExtBuilder, txRetBuilder, true);
     return transaction;
   }
 
   public Transaction triggerConstantContract(TriggerSmartContract triggerSmartContract,
       TransactionCapsule trxCap, Builder builder, Return.Builder retBuilder)
       throws ContractValidateException, ContractExeException, HeaderNotFound, VMIllegalException {
+    return triggerConstantContract(triggerSmartContract, trxCap, builder, retBuilder, false);
+  }
+
+  public Transaction triggerConstantContract(TriggerSmartContract triggerSmartContract,
+      TransactionCapsule trxCap, Builder builder, Return.Builder retBuilder, boolean isEstimating)
+      throws ContractValidateException, ContractExeException, HeaderNotFound, VMIllegalException {
 
     if (triggerSmartContract.getContractAddress().isEmpty()) { // deploy contract
       CreateSmartContract.Builder deployBuilder = CreateSmartContract.newBuilder();
@@ -2991,11 +2997,11 @@ public Transaction triggerConstantContract(TriggerSmartContract triggerSmartCont
         throw new ContractValidateException("Smart contract is not exist.");
       }
     }
-    return callConstantContract(trxCap, builder, retBuilder);
+    return callConstantContract(trxCap, builder, retBuilder, isEstimating);
   }
 
   public Transaction callConstantContract(TransactionCapsule trxCap,
-      Builder builder, Return.Builder retBuilder)
+      Builder builder, Return.Builder retBuilder, boolean isEstimating)
       throws ContractValidateException, ContractExeException, HeaderNotFound, VMIllegalException {
 
     if (!Args.getInstance().isSupportConstant()) {
@@ -3011,15 +3017,17 @@ public Transaction callConstantContract(TransactionCapsule trxCap,
       headBlock = blockCapsuleList.get(0).getInstance();
     }
 
-    TransactionContext context = new TransactionContext(new BlockCapsule(headBlock), trxCap,
+    BlockCapsule headBlockCapsule = new BlockCapsule(headBlock);
+    TransactionContext context = new TransactionContext(headBlockCapsule, trxCap,
         StoreFactory.getInstance(), true, false);
     VMActuator vmActuator = new VMActuator(true);
 
     vmActuator.validate(context);
     vmActuator.execute(context);
 
     ProgramResult result = context.getProgramResult();
-    if (result.getException() != null) {
+    if (!isEstimating && result.getException() != null
+        || result.getException() instanceof Program.OutOfTimeException) {
       RuntimeException e = result.getException();
       logger.warn("Constant call has an error {}", e.getMessage());
       throw e;
@@ -3028,6 +3036,9 @@ public Transaction callConstantContract(TransactionCapsule trxCap,
     TransactionResultCapsule ret = new TransactionResultCapsule();
     builder.setEnergyUsed(result.getEnergyUsed());
     builder.setEnergyPenalty(result.getEnergyPenaltyTotal());
+    builder.setBlockNumber(headBlockCapsule.getNum());
+    builder.setBlockHash(ByteString.copyFrom(headBlockCapsule.getBlockId().getBytes()));
+    builder.setEnergyPenalty(result.getEnergyPenaltyTotal());
     builder.addConstantResult(ByteString.copyFrom(result.getHReturn()));
     result.getLogInfoList().forEach(logInfo ->
         builder.addLogs(LogInfo.buildLog(logInfo)));

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

@@ -27,7 +27,11 @@
 @Slf4j(topic = "API")
 public class TriggerConstantContractServlet extends RateLimiterServlet {
 
-  private final String functionSelector = "function_selector";
+  private final String OWNER_ADDRESS = "owner_address";
+  private final String CONTRACT_ADDRESS = "contract_address";
+  private final String FUNCTION_SELECTOR = "function_selector";
+  private final String FUNCTION_PARAMETER = "parameter";
+  private final String CALL_DATA = "data";
 
   @Autowired
   private Wallet wallet;
@@ -37,13 +41,18 @@ protected void doGet(HttpServletRequest request, HttpServletResponse response) {
 
   protected void validateParameter(String contract) {
     JSONObject jsonObject = JSONObject.parseObject(contract);
-    if (!jsonObject.containsKey("owner_address")
-        || StringUtil.isNullOrEmpty(jsonObject.getString("owner_address"))) {
-      throw new InvalidParameterException("owner_address isn't set.");
+    if (StringUtil.isNullOrEmpty(jsonObject.getString(OWNER_ADDRESS))) {
+      throw new InvalidParameterException(OWNER_ADDRESS + " isn't set.");
     }
-    if (!jsonObject.containsKey("contract_address")
-        || StringUtil.isNullOrEmpty(jsonObject.getString("contract_address"))) {
-      throw new InvalidParameterException("contract_address isn't set.");
+    if (StringUtil.isNullOrEmpty(jsonObject.getString(CONTRACT_ADDRESS))
+        && StringUtil.isNullOrEmpty(jsonObject.getString(CALL_DATA))) {
+      throw new InvalidParameterException("At least one of "
+          + CONTRACT_ADDRESS + " and " + CALL_DATA + " must be set.");
+    }
+    if (!StringUtil.isNullOrEmpty(jsonObject.getString(FUNCTION_SELECTOR))
+        ^ StringUtil.isNullOrEmpty(jsonObject.getString(CALL_DATA))) {
+      throw new InvalidParameterException("Only one of "
+          + FUNCTION_SELECTOR + " and " + CALL_DATA + " can be set.");
     }
   }
 
@@ -62,17 +71,15 @@ protected void doPost(HttpServletRequest request, HttpServletResponse response)
       JsonFormat.merge(contract, build, visible);
       JSONObject jsonObject = JSONObject.parseObject(contract);
 
-      boolean isFunctionSelectorSet = jsonObject.containsKey(functionSelector)
-          && !StringUtil.isNullOrEmpty(jsonObject.getString(functionSelector));
-      String data;
+      boolean isFunctionSelectorSet =
+          !StringUtil.isNullOrEmpty(jsonObject.getString(FUNCTION_SELECTOR));
       if (isFunctionSelectorSet) {
-        String selector = jsonObject.getString(functionSelector);
-        String parameter = jsonObject.getString("parameter");
-        data = Util.parseMethod(selector, parameter);
+        String selector = jsonObject.getString(FUNCTION_SELECTOR);
+        String parameter = jsonObject.getString(FUNCTION_PARAMETER);
+        String data = Util.parseMethod(selector, parameter);
         build.setData(ByteString.copyFrom(ByteArray.fromHexString(data)));
-      } else {
-        build.setData(ByteString.copyFrom(new byte[0]));
       }
+
       TransactionCapsule trxCap = wallet
           .createTransactionCapsule(build.build(), ContractType.TriggerSmartContract);
 

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

@@ -27,7 +27,11 @@
 @Slf4j(topic = "API")
 public class TriggerSmartContractServlet extends RateLimiterServlet {
 
-  private final String functionSelector = "function_selector";
+  private final String OWNER_ADDRESS = "owner_address";
+  private final String CONTRACT_ADDRESS = "contract_address";
+  private final String FUNCTION_SELECTOR = "function_selector";
+  private final String FUNCTION_PARAMETER = "parameter";
+  private final String CALL_DATA = "data";
 
   @Autowired
   private Wallet wallet;
@@ -37,14 +41,17 @@ protected void doGet(HttpServletRequest request, HttpServletResponse response) {
 
   protected void validateParameter(String contract) {
     JSONObject jsonObject = JSONObject.parseObject(contract);
-    if (!jsonObject.containsKey("owner_address")
-        || StringUtil.isNullOrEmpty(jsonObject.getString("owner_address"))) {
+    if (StringUtil.isNullOrEmpty(jsonObject.getString(OWNER_ADDRESS))) {
       throw new InvalidParameterException("owner_address isn't set.");
     }
-    if (!jsonObject.containsKey("contract_address")
-        || StringUtil.isNullOrEmpty(jsonObject.getString("contract_address"))) {
+    if (StringUtil.isNullOrEmpty(jsonObject.getString(CONTRACT_ADDRESS))) {
       throw new InvalidParameterException("contract_address isn't set.");
     }
+    if (!StringUtil.isNullOrEmpty(jsonObject.getString(FUNCTION_SELECTOR))
+        ^ StringUtil.isNullOrEmpty(jsonObject.getString(CALL_DATA))) {
+      throw new InvalidParameterException("Only one of "
+          + FUNCTION_SELECTOR + " and " + CALL_DATA + " can be set.");
+    }
   }
 
   protected void doPost(HttpServletRequest request, HttpServletResponse response)
@@ -62,16 +69,13 @@ protected void doPost(HttpServletRequest request, HttpServletResponse response)
       JsonFormat.merge(contract, build, visible);
       JSONObject jsonObject = JSONObject.parseObject(contract);
 
-      boolean isFunctionSelectorSet = jsonObject.containsKey(functionSelector)
-          && !StringUtil.isNullOrEmpty(jsonObject.getString(functionSelector));
-      String data;
+      boolean isFunctionSelectorSet =
+          !StringUtil.isNullOrEmpty(jsonObject.getString(FUNCTION_SELECTOR));
       if (isFunctionSelectorSet) {
-        String selector = jsonObject.getString(functionSelector);
-        String parameter = jsonObject.getString("parameter");
-        data = Util.parseMethod(selector, parameter);
+        String selector = jsonObject.getString(FUNCTION_SELECTOR);
+        String parameter = jsonObject.getString(FUNCTION_PARAMETER);
+        String data = Util.parseMethod(selector, parameter);
         build.setData(ByteString.copyFrom(ByteArray.fromHexString(data)));
-      } else {
-        build.setData(ByteString.copyFrom(new byte[0]));
       }
 
       build.setCallTokenValue(Util.getJsonLongValue(jsonObject, "call_token_value"));

protocol/src/main/protos/api/api.proto

@@ -1315,6 +1315,8 @@ message TransactionExtention {
   repeated TransactionInfo.Log logs = 6;
   repeated InternalTransaction internal_transactions = 7;
   int64 energy_penalty = 8;
+  int64 block_number = 9;
+  bytes block_hash = 10;
 }
 
 message EstimateEnergyMessage {