state trie final fast pure latest

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

@@ -26,7 +26,6 @@
 import org.tron.common.utils.StorageUtils;
 import org.tron.common.utils.StringUtil;
 import org.tron.common.utils.WalletUtil;
-import org.tron.core.ChainBaseManager;
 import org.tron.core.capsule.AccountCapsule;
 import org.tron.core.capsule.BlockCapsule;
 import org.tron.core.capsule.ContractCapsule;
@@ -35,6 +34,7 @@
 import org.tron.core.db.TransactionContext;
 import org.tron.core.exception.ContractExeException;
 import org.tron.core.exception.ContractValidateException;
+import org.tron.core.exception.HeaderNotFound;
 import org.tron.core.utils.TransactionUtil;
 import org.tron.core.vm.EnergyCost;
 import org.tron.core.vm.LogInfoTriggerParser;
@@ -53,6 +53,7 @@
 import org.tron.core.vm.program.invoke.ProgramInvokeFactory;
 import org.tron.core.vm.repository.Repository;
 import org.tron.core.vm.repository.RepositoryImpl;
+import org.tron.core.vm.repository.RepositoryStateImpl;
 import org.tron.core.vm.utils.MUtil;
 import org.tron.protos.Protocol;
 import org.tron.protos.Protocol.Block;
@@ -94,6 +95,9 @@ public class VMActuator implements Actuator2 {
 
   private LogInfoTriggerParser logInfoTriggerParser;
 
+  private static final boolean isAllowStateRoot = CommonParameter.getInstance().getStorage()
+          .isAllowStateRoot();
+
   public VMActuator(boolean isConstantCall) {
     this.isConstantCall = isConstantCall;
     this.maxEnergyLimit = CommonParameter.getInstance().maxEnergyLimitForConstant;
@@ -110,21 +114,38 @@ private static long getEnergyFee(long callerEnergyUsage, long callerEnergyFrozen
 
   @Override
   public void validate(Object object) throws ContractValidateException {
-
     TransactionContext context = (TransactionContext) object;
     if (Objects.isNull(context)) {
       throw new RuntimeException("TransactionContext is null");
     }
 
-    // Load Config
-    ConfigLoader.load(context.getStoreFactory());
     // Warm up registry class
     OperationRegistry.init();
     trx = context.getTrxCap().getInstance();
+
+    // check whether is state query
+    boolean stateQuery = isAllowStateRoot && isConstantCall;
+    stateQuery = stateQuery && context.getBlockCap() != null;
+    try {
+      stateQuery = stateQuery && context.getBlockCap().getNum() < context.getStoreFactory()
+          .getChainBaseManager().getHead().getNum();
+    } catch (HeaderNotFound e) {
+      throw new ContractValidateException(e.getMessage());
+    }
+
+    //Prepare Repository
+    if (stateQuery) {
+      rootRepository = RepositoryStateImpl.createRoot(context.getBlockCap().getArchiveRoot());
+    } else {
+      rootRepository = RepositoryImpl.createRoot(context.getStoreFactory());
+    }
+    // Load Config
+    ConfigLoader.load(rootRepository);
+
     // If tx`s fee limit is set, use it to calc max energy limit for constant call
     if (isConstantCall && trx.getRawData().getFeeLimit() > 0) {
       maxEnergyLimit = Math.min(maxEnergyLimit, trx.getRawData().getFeeLimit()
-          / context.getStoreFactory().getChainBaseManager()
+          / rootRepository
           .getDynamicPropertiesStore().getEnergyFee());
     }
     blockCap = context.getBlockCap();
@@ -134,9 +155,6 @@ public void validate(Object object) throws ContractValidateException {
     }
     //Route Type
     ContractType contractType = this.trx.getRawData().getContract(0).getType();
-    //Prepare Repository
-    rootRepository = RepositoryImpl.createRoot(context.getStoreFactory());
-
     enableEventListener = context.isEventPluginLoaded();
 
     //set executorType type
@@ -297,7 +315,6 @@ public void execute(Object object) throws ContractExeException {
       String txHash = Hex.toHexString(rootInternalTx.getHash());
       VMUtils.saveProgramTraceFile(txHash, traceContent);
     }
-
   }
 
   private void create()
@@ -563,8 +580,7 @@ public long getAccountEnergyLimitWithFixRatio(AccountCapsule account, long feeLi
       long now = rootRepository.getHeadSlot();
       EnergyProcessor energyProcessor =
           new EnergyProcessor(
-              rootRepository.getDynamicPropertiesStore(),
-              ChainBaseManager.getInstance().getAccountStore());
+              rootRepository.getDynamicPropertiesStore(), rootRepository.getAccountStore());
       energyProcessor.updateUsage(account);
       account.setLatestConsumeTimeForEnergy(now);
       receipt.setCallerEnergyUsage(account.getEnergyUsage());
@@ -726,8 +742,7 @@ public long getTotalEnergyLimitWithFixRatio(AccountCapsule creator, AccountCapsu
       long now = rootRepository.getHeadSlot();
       EnergyProcessor energyProcessor =
           new EnergyProcessor(
-              rootRepository.getDynamicPropertiesStore(),
-              ChainBaseManager.getInstance().getAccountStore());
+              rootRepository.getDynamicPropertiesStore(), rootRepository.getAccountStore());
       energyProcessor.updateUsage(creator);
       creator.setLatestConsumeTimeForEnergy(now);
       receipt.setOriginEnergyUsage(creator.getEnergyUsage());

actuator/src/main/java/org/tron/core/vm/config/ConfigLoader.java

@@ -5,17 +5,17 @@
 import lombok.extern.slf4j.Slf4j;
 import org.tron.common.parameter.CommonParameter;
 import org.tron.core.store.DynamicPropertiesStore;
-import org.tron.core.store.StoreFactory;
+import org.tron.core.vm.repository.Repository;
 
 @Slf4j(topic = "VMConfigLoader")
 public class ConfigLoader {
 
   //only for unit test
   public static boolean disable = false;
 
-  public static void load(StoreFactory storeFactory) {
+  public static void load(Repository repository) {
     if (!disable) {
-      DynamicPropertiesStore ds = storeFactory.getChainBaseManager().getDynamicPropertiesStore();
+      DynamicPropertiesStore ds = repository.getDynamicPropertiesStore();
       VMConfig.setVmTrace(CommonParameter.getInstance().isVmTrace());
       if (ds != null) {
         VMConfig.initVmHardFork(checkForEnergyLimit(ds));

actuator/src/main/java/org/tron/core/vm/nativecontract/DelegateResourceProcessor.java

@@ -13,7 +13,6 @@
 import org.apache.commons.lang3.ArrayUtils;
 import org.tron.common.utils.DecodeUtil;
 import org.tron.common.utils.StringUtil;
-import org.tron.core.ChainBaseManager;
 import org.tron.core.actuator.ActuatorConstant;
 import org.tron.core.capsule.AccountCapsule;
 import org.tron.core.capsule.DelegatedResourceAccountIndexCapsule;
@@ -56,7 +55,9 @@ public void validate(DelegateResourceParam param, Repository repo) throws Contra
 
     switch (param.getResourceType()) {
       case BANDWIDTH: {
-        BandwidthProcessor processor = new BandwidthProcessor(ChainBaseManager.getInstance());
+        BandwidthProcessor processor = new BandwidthProcessor(dynamicStore,
+                repo.getAccountStore(),  repo.getAssetIssueStore(),
+                repo.getAssetIssueV2Store());
         processor.updateUsageForDelegated(ownerCapsule);
 
         long netUsage = (long) (ownerCapsule.getNetUsage() * TRX_PRECISION * ((double)
@@ -71,8 +72,7 @@ public void validate(DelegateResourceParam param, Repository repo) throws Contra
       }
       break;
       case ENERGY: {
-        EnergyProcessor processor =
-            new EnergyProcessor(dynamicStore, ChainBaseManager.getInstance().getAccountStore());
+        EnergyProcessor processor = new EnergyProcessor(dynamicStore, repo.getAccountStore());
         processor.updateUsage(ownerCapsule);
 
         long energyUsage = (long) (ownerCapsule.getEnergyUsage() * TRX_PRECISION * ((double)

actuator/src/main/java/org/tron/core/vm/nativecontract/UnDelegateResourceProcessor.java

@@ -12,7 +12,6 @@
 import lombok.extern.slf4j.Slf4j;
 import org.tron.common.utils.DecodeUtil;
 import org.tron.common.utils.StringUtil;
-import org.tron.core.ChainBaseManager;
 import org.tron.core.capsule.AccountCapsule;
 import org.tron.core.capsule.DelegatedResourceAccountIndexCapsule;
 import org.tron.core.capsule.DelegatedResourceCapsule;
@@ -100,7 +99,9 @@ public void execute(UnDelegateResourceParam param, Repository repo) {
     if (receiverCapsule != null) {
       switch (param.getResourceType()) {
         case BANDWIDTH:
-          BandwidthProcessor bandwidthProcessor = new BandwidthProcessor(ChainBaseManager.getInstance());
+          BandwidthProcessor bandwidthProcessor = new BandwidthProcessor(dynamicStore,
+                  repo.getAccountStore(),  repo.getAssetIssueStore(),
+                  repo.getAssetIssueV2Store());
           bandwidthProcessor.updateUsageForDelegated(receiverCapsule);
           /* For example, in a scenario where a regular account can be upgraded to a contract
           account through an interface, the account information will be cleared after the
@@ -125,8 +126,8 @@ public void execute(UnDelegateResourceParam param, Repository repo) {
           receiverCapsule.setLatestConsumeTime(now);
           break;
         case ENERGY:
-          EnergyProcessor energyProcessor =
-              new EnergyProcessor(dynamicStore, ChainBaseManager.getInstance().getAccountStore());
+          EnergyProcessor energyProcessor = new EnergyProcessor(dynamicStore,
+                  repo.getAccountStore());
           energyProcessor.updateUsage(receiverCapsule);
 
           if (receiverCapsule.getAcquiredDelegatedFrozenV2BalanceForEnergy()
@@ -165,7 +166,9 @@ public void execute(UnDelegateResourceParam param, Repository repo) {
         ownerCapsule.addDelegatedFrozenV2BalanceForBandwidth(-unDelegateBalance);
         ownerCapsule.addFrozenBalanceForBandwidthV2(unDelegateBalance);
 
-        BandwidthProcessor processor = new BandwidthProcessor(ChainBaseManager.getInstance());
+        BandwidthProcessor processor = new BandwidthProcessor(dynamicStore,
+                 repo.getAccountStore(),  repo.getAssetIssueStore(),
+                 repo.getAssetIssueV2Store());
         if (Objects.nonNull(receiverCapsule) && transferUsage > 0) {
           processor.unDelegateIncrease(ownerCapsule, receiverCapsule,
               transferUsage, BANDWIDTH, now);
@@ -179,7 +182,7 @@ public void execute(UnDelegateResourceParam param, Repository repo) {
         ownerCapsule.addFrozenBalanceForEnergyV2(unDelegateBalance);
 
         EnergyProcessor processor =
-            new EnergyProcessor(dynamicStore, ChainBaseManager.getInstance().getAccountStore());
+            new EnergyProcessor(dynamicStore, repo.getAccountStore());
         if (Objects.nonNull(receiverCapsule) && transferUsage > 0) {
           processor.unDelegateIncrease(ownerCapsule, receiverCapsule, transferUsage, ENERGY, now);
         }

actuator/src/main/java/org/tron/core/vm/program/ContractState.java

@@ -12,6 +12,7 @@
 import org.tron.core.capsule.DelegatedResourceCapsule;
 import org.tron.core.capsule.VotesCapsule;
 import org.tron.core.capsule.WitnessCapsule;
+import org.tron.core.store.AccountStore;
 import org.tron.core.store.AssetIssueStore;
 import org.tron.core.store.AssetIssueV2Store;
 import org.tron.core.store.DelegationStore;
@@ -46,6 +47,11 @@ public AssetIssueCapsule getAssetIssue(byte[] tokenId) {
     return repository.getAssetIssue(tokenId);
   }
 
+  @Override
+  public AccountStore getAccountStore() {
+    return repository.getAccountStore();
+  }
+
   @Override
   public AssetIssueV2Store getAssetIssueV2Store() {
     return repository.getAssetIssueV2Store();

actuator/src/main/java/org/tron/core/vm/program/Program.java

@@ -39,7 +39,6 @@
 import org.tron.common.utils.FastByteComparisons;
 import org.tron.common.utils.Utils;
 import org.tron.common.utils.WalletUtil;
-import org.tron.core.ChainBaseManager;
 import org.tron.core.Constant;
 import org.tron.core.capsule.AccountCapsule;
 import org.tron.core.capsule.BlockCapsule;
@@ -556,7 +555,8 @@ private long transferFrozenV2BalanceToInheritor(byte[] ownerAddr, byte[] inherit
             });
 
     // merge usage
-    BandwidthProcessor bandwidthProcessor = new BandwidthProcessor(ChainBaseManager.getInstance());
+    BandwidthProcessor bandwidthProcessor = new BandwidthProcessor(repo.getDynamicPropertiesStore()
+            , repo.getAccountStore(), repo.getAssetIssueStore(), repo.getAssetIssueV2Store());
     bandwidthProcessor.updateUsageForDelegated(ownerCapsule);
     ownerCapsule.setLatestConsumeTime(now);
     if (ownerCapsule.getNetUsage() > 0) {
@@ -566,7 +566,7 @@ private long transferFrozenV2BalanceToInheritor(byte[] ownerAddr, byte[] inherit
 
     EnergyProcessor energyProcessor =
         new EnergyProcessor(
-            repo.getDynamicPropertiesStore(), ChainBaseManager.getInstance().getAccountStore());
+            repo.getDynamicPropertiesStore(), repo.getAccountStore());
     energyProcessor.updateUsage(ownerCapsule);
     ownerCapsule.setLatestConsumeTimeForEnergy(now);
     if (ownerCapsule.getEnergyUsage() > 0) {

actuator/src/main/java/org/tron/core/vm/repository/Repository.java

@@ -9,6 +9,8 @@
 
 public interface Repository {
 
+  AccountStore getAccountStore();
+
   AssetIssueCapsule getAssetIssue(byte[] tokenId);
 
   AssetIssueV2Store getAssetIssueV2Store();