modify the hard fork

Author: lvs007

chainbase/src/main/java/org/tron/common/utils/ForkUtils.java

@@ -37,6 +37,14 @@ public boolean pass(ForkBlockVersionEnum forkBlockVersionEnum) {
   }
 
   public synchronized boolean pass(int version) {
+    if (version > ForkBlockVersionEnum.VERSION_4_0.getValue()) {
+      return passNew(version);
+    } else {
+      return passOld(version);
+    }
+  }
+
+  private boolean passOld(int version) {
     if (version == ForkBlockVersionConsts.ENERGY_LIMIT) {
       return checkForEnergyLimit();
     }
@@ -45,6 +53,40 @@ public synchronized boolean pass(int version) {
     return check(stats);
   }
 
+  private boolean passNew(int version) {
+    ForkBlockVersionEnum versionEnum = ForkBlockVersionEnum.getForkBlockVersionEnum(version);
+    if (versionEnum == null) {
+      logger.error("not exist block version: {}", version);
+      return false;
+    }
+    long latestBlockTime = dynamicPropertiesStore.getLatestBlockHeaderTimestamp();
+    long maintenanceTimeInterval = dynamicPropertiesStore.getMaintenanceTimeInterval();
+    long hardForkTime = ((versionEnum.getHardForkTime() - 1) / maintenanceTimeInterval + 1)
+        * maintenanceTimeInterval;
+    if (latestBlockTime < hardForkTime) {
+      return false;
+    }
+    Boolean hardForkResult = dynamicPropertiesStore.getForked(version);
+    if (hardForkResult != null) {
+      return hardForkResult;
+    }
+    byte[] stats = dynamicPropertiesStore.statsByVersion(version);
+    if (stats == null || stats.length == 0) {
+      return false;
+    }
+    int count = 0;
+    for (int i = 0; i < stats.length; i++) {
+      if (check[i] == stats[i]) {
+        ++count;
+      }
+    }
+    hardForkResult = count >= versionEnum.getHardForkCount();
+    dynamicPropertiesStore.forked(version, hardForkResult);
+    logger.info("******** update block version {} hard fork result:{} ********", version,
+        hardForkResult);
+    return hardForkResult;
+  }
+
   // when block.version = 5,
   // it make block use new energy to handle transaction when block number >= 4727890L.
   // version !=5, skip this.

chainbase/src/main/java/org/tron/core/config/args/Parameter.java

@@ -5,19 +5,34 @@
 public class Parameter {
 
   public enum ForkBlockVersionEnum {
-    ENERGY_LIMIT(5),
-    VERSION_3_2_2(6),
-    VERSION_3_5(7),
-    VERSION_3_6(8),
-    VERSION_3_6_5(9),
-    VERSION_3_6_6(10),
-    VERSION_4_0(16);
+    ENERGY_LIMIT(5, 0L, 0),
+    VERSION_3_2_2(6, 0L, 0),
+    VERSION_3_5(7, 0L, 0),
+    VERSION_3_6(8, 0L, 0),
+    VERSION_3_6_5(9, 0L, 0),
+    VERSION_3_6_6(10, 0L, 0),
+    VERSION_4_0(16, 0L, 0);
 
     @Getter
     private int value;
+    @Getter
+    private long hardForkTime;
+    @Getter
+    private int hardForkCount;
 
-    ForkBlockVersionEnum(int value) {
+    ForkBlockVersionEnum(int value, long hardForkTime, int hardForkCount) {
       this.value = value;
+      this.hardForkTime = hardForkTime;
+      this.hardForkCount = hardForkCount;
+    }
+
+    public static ForkBlockVersionEnum getForkBlockVersionEnum(int value) {
+      for (ForkBlockVersionEnum versionEnum : values()) {
+        if (versionEnum.getValue() == value) {
+          return versionEnum;
+        }
+      }
+      return null;
     }
   }
 

chainbase/src/main/java/org/tron/core/store/DynamicPropertiesStore.java

@@ -91,7 +91,7 @@ public class DynamicPropertiesStore extends TronStoreWithRevoking<BytesCapsule>
   private static final byte[] TOTAL_STORAGE_TAX = "TOTAL_STORAGE_TAX".getBytes();
   private static final byte[] TOTAL_STORAGE_RESERVED = "TOTAL_STORAGE_RESERVED".getBytes();
   private static final byte[] STORAGE_EXCHANGE_TAX_RATE = "STORAGE_EXCHANGE_TAX_RATE".getBytes();
-  private static final byte[] FORK_CONTROLLER = "FORK_CONTROLLER".getBytes();
+  private static final String FORK_CONTROLLER = "FORK_CONTROLLER";
   private static final String FORK_PREFIX = "FORK_VERSION_";
   //This value is only allowed to be 0, 1, -1
   private static final byte[] REMOVE_THE_POWER_OF_THE_GR = "REMOVE_THE_POWER_OF_THE_GR".getBytes();
@@ -121,7 +121,7 @@ public class DynamicPropertiesStore extends TronStoreWithRevoking<BytesCapsule>
   private static final byte[] ALLOW_SHIELDED_TRANSACTION = "ALLOW_SHIELDED_TRANSACTION".getBytes();
   private static final byte[] ALLOW_SHIELDED_TRC20_TRANSACTION =
       "ALLOW_SHIELDED_TRC20_TRANSACTION"
-      .getBytes();
+          .getBytes();
   private static final byte[] ALLOW_TVM_CONSTANTINOPLE = "ALLOW_TVM_CONSTANTINOPLE".getBytes();
   private static final byte[] ALLOW_TVM_SOLIDITY_059 = "ALLOW_TVM_SOLIDITY_059".getBytes();
   private static final byte[] FORBID_TRANSFER_TO_CONTRACT = "FORBID_TRANSFER_TO_CONTRACT"
@@ -1767,8 +1767,9 @@ public void addTotalTransactionCost(long fee) {
     saveTotalTransactionCost(newValue);
   }
 
-  public void forked() {
-    put(FORK_CONTROLLER, new BytesCapsule(Boolean.toString(true).getBytes()));
+  public void forked(int version, boolean value) {
+    String forkKey = FORK_CONTROLLER + version;
+    put(forkKey.getBytes(), new BytesCapsule(Boolean.toString(value).getBytes()));
   }
 
   public void statsByVersion(int version, byte[] stats) {
@@ -1781,9 +1782,10 @@ public byte[] statsByVersion(int version) {
     return revokingDB.getUnchecked(statsKey.getBytes());
   }
 
-  public boolean getForked() {
-    byte[] value = revokingDB.getUnchecked(FORK_CONTROLLER);
-    return value == null ? Boolean.FALSE : Boolean.valueOf(new String(value));
+  public Boolean getForked(int version) {
+    String forkKey = FORK_CONTROLLER + version;
+    byte[] value = revokingDB.getUnchecked(forkKey.getBytes());
+    return value == null ? null : Boolean.valueOf(new String(value));
   }
 
   /**
@@ -1871,10 +1873,10 @@ private static class DynamicResourceProperties {
     private static final byte[] BLOCK_ENERGY_USAGE = "BLOCK_ENERGY_USAGE".getBytes();
     private static final byte[] ADAPTIVE_RESOURCE_LIMIT_MULTIPLIER =
         "ADAPTIVE_RESOURCE_LIMIT_MULTIPLIER"
-        .getBytes();
+            .getBytes();
     private static final byte[] ADAPTIVE_RESOURCE_LIMIT_TARGET_RATIO =
         "ADAPTIVE_RESOURCE_LIMIT_TARGET_RATIO"
-        .getBytes();
+            .getBytes();
   }
 
 }