implement eth_simulateV1 JSON-RPC method for trading-flow case

Author: APshenkin

actuator/src/main/java/org/tron/core/actuator/VMActuator.java

@@ -54,6 +54,7 @@
 import org.tron.core.vm.program.ProgramPrecompile;
 import org.tron.core.vm.program.invoke.ProgramInvoke;
 import org.tron.core.vm.program.invoke.ProgramInvokeFactory;
+import org.tron.core.vm.program.listener.SimulationTracer;
 import org.tron.core.vm.repository.Repository;
 import org.tron.core.vm.repository.RepositoryImpl;
 import org.tron.core.vm.utils.MUtil;
@@ -95,6 +96,12 @@ public class VMActuator implements Actuator2 {
   @Setter
   private boolean enableEventListener;
 
+  @Setter
+  private Repository injectedRootRepository;
+
+  @Setter
+  private SimulationTracer simulationTracer;
+
   private LogInfoTriggerParser logInfoTriggerParser;
 
   public VMActuator(boolean isConstantCall) {
@@ -138,7 +145,9 @@ public void validate(Object object) throws ContractValidateException {
     //Route Type
     ContractType contractType = this.trx.getRawData().getContract(0).getType();
     //Prepare Repository
-    rootRepository = RepositoryImpl.createRoot(context.getStoreFactory());
+    rootRepository = injectedRootRepository != null
+        ? injectedRootRepository
+        : RepositoryImpl.createRoot(context.getStoreFactory());
 
     enableEventListener = context.isEventPluginLoaded();
 
@@ -414,6 +423,7 @@ private void create()
       if (VMConfig.allowTvmCompatibleEvm()) {
         this.program.setContractVersion(1);
       }
+      this.program.setSimulationTracer(simulationTracer);
       byte[] txId = TransactionUtil.getTransactionId(trx).getBytes();
       this.program.setRootTransactionId(txId);
       if (enableEventListener && isCheckTransaction()) {
@@ -437,10 +447,18 @@ private void create()
     // transfer from callerAddress to contractAddress according to callValue
     if (callValue > 0) {
       MUtil.transfer(rootRepository, callerAddress, contractAddress, callValue);
+      if (simulationTracer != null) {
+        simulationTracer.onTransfer(stripTronPrefix(callerAddress),
+            stripTronPrefix(contractAddress), callValue);
+      }
     }
     if (VMConfig.allowTvmTransferTrc10() && tokenValue > 0) {
       MUtil.transferToken(rootRepository, callerAddress, contractAddress, String.valueOf(tokenId),
           tokenValue);
+      if (simulationTracer != null) {
+        simulationTracer.onTokenTransfer(stripTronPrefix(callerAddress),
+            stripTronPrefix(contractAddress), tokenId, tokenValue);
+      }
     }
 
   }
@@ -529,6 +547,7 @@ private void call()
       if (VMConfig.allowTvmCompatibleEvm()) {
         this.program.setContractVersion(deployedContract.getContractVersion());
       }
+      this.program.setSimulationTracer(simulationTracer);
       byte[] txId = TransactionUtil.getTransactionId(trx).getBytes();
       this.program.setRootTransactionId(txId);
 
@@ -543,14 +562,31 @@ private void call()
 
     if (callValue > 0) {
       MUtil.transfer(rootRepository, callerAddress, contractAddress, callValue);
+      if (simulationTracer != null) {
+        simulationTracer.onTransfer(stripTronPrefix(callerAddress),
+            stripTronPrefix(contractAddress), callValue);
+      }
     }
     if (VMConfig.allowTvmTransferTrc10() && tokenValue > 0) {
       MUtil.transferToken(rootRepository, callerAddress, contractAddress, String.valueOf(tokenId),
           tokenValue);
+      if (simulationTracer != null) {
+        simulationTracer.onTokenTransfer(stripTronPrefix(callerAddress),
+            stripTronPrefix(contractAddress), tokenId, tokenValue);
+      }
     }
 
   }
 
+  private static byte[] stripTronPrefix(byte[] tronAddress) {
+    if (tronAddress == null || tronAddress.length != 21) {
+      return tronAddress;
+    }
+    byte[] evm = new byte[20];
+    System.arraycopy(tronAddress, 1, evm, 0, 20);
+    return evm;
+  }
+
   public long getAccountEnergyLimitWithFixRatio(AccountCapsule account, long feeLimit,
       long callValue) {
 

actuator/src/main/java/org/tron/core/vm/OperationActions.java

@@ -754,6 +754,9 @@ public static void logAction(Program program) {
         new LogInfo(address.getLast20Bytes(), topics, data);
 
     program.getResult().addLogInfo(logInfo);
+    if (program.getSimulationTracer() != null) {
+      program.getSimulationTracer().onLog(logInfo);
+    }
     program.step();
   }