func(test): add unit test for energy adjustment proposal

Author: yanghang8612

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

@@ -38,6 +38,7 @@
 import org.tron.core.utils.TransactionUtil;
 import org.tron.core.vm.EnergyCost;
 import org.tron.core.vm.LogInfoTriggerParser;
+import org.tron.core.vm.Op;
 import org.tron.core.vm.OperationRegistry;
 import org.tron.core.vm.VM;
 import org.tron.core.vm.VMConstant;
@@ -193,7 +194,7 @@ public void execute(Object object) throws ContractExeException {
           // If the last op consumed too much execution time, the CPU time limit for the whole tx can be exceeded.
           // This is not fair for other txs in the same block.
           // So when allowFairEnergyAdjustment is on, the CPU time limit will be checked at the end of tx execution.
-          program.checkCPUTimeLimit("TX_LAST_OP");
+          program.checkCPUTimeLimit(Op.getNameOf(program.getLastOp()) + "(TX_LAST_OP)");
         }
 
         if (TrxType.TRX_CONTRACT_CREATION_TYPE == trxType && !result.isRevert()) {

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

@@ -68,7 +68,7 @@ public static JumpTable getTable() {
     }
 
     if (VMConfig.allowEnergyAdjustment()) {
-      ajustForFairEnergy(table);
+      adjustForFairEnergy(table);
     }
 
     return table;
@@ -640,7 +640,7 @@ public static void appendShangHaiOperations(JumpTable table) {
         proposal));
   }
 
-  public static void ajustForFairEnergy(JumpTable table) {
+  public static void adjustForFairEnergy(JumpTable table) {
     table.set(new Operation(
         Op.VOTEWITNESS, 4, 1,
         EnergyCost::getVoteWitnessCost2,

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

@@ -275,6 +275,10 @@ public void setLastOp(byte op) {
     this.lastOp = op;
   }
 
+  public byte getLastOp() {
+    return this.lastOp;
+  }
+
   /**
    * Returns the last fully executed OP.
    */

framework/src/test/java/org/tron/common/runtime/vm/OperationsTest.java

@@ -3,6 +3,8 @@
 import static org.junit.Assert.assertEquals;
 
 import java.util.List;
+import java.util.Random;
+
 import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 import org.bouncycastle.util.encoders.Hex;
@@ -15,6 +17,7 @@
 import org.tron.common.BaseTest;
 import org.tron.common.parameter.CommonParameter;
 import org.tron.common.runtime.InternalTransaction;
+import org.tron.common.utils.DecodeUtil;
 import org.tron.core.Constant;
 import org.tron.core.config.args.Args;
 import org.tron.core.exception.ContractValidateException;
@@ -864,6 +867,50 @@ public void testPush0() throws ContractValidateException {
     VMConfig.initAllowTvmShangHai(0);
   }
 
+  @Test
+  public void testSuicideCost() throws ContractValidateException {
+    invoke = new ProgramInvokeMockImpl(StoreFactory.getInstance(), new byte[0], new byte[21]);
+    program = new Program(null, null, invoke,
+        new InternalTransaction(
+            Protocol.Transaction.getDefaultInstance(),
+            InternalTransaction.TrxType.TRX_UNKNOWN_TYPE));
+
+    byte[] receiver1 = generateRandomAddress();
+    program.stackPush(new DataWord(receiver1));
+    Assert.assertEquals(0, EnergyCost.getSuicideCost(program));
+    invoke.getDeposit().createAccount(receiver1, Protocol.AccountType.Normal);
+    Assert.assertEquals(0, EnergyCost.getSuicideCost(program));
+
+    byte[] receiver2 = generateRandomAddress();
+    program.stackPush(new DataWord(receiver2));
+    Assert.assertEquals(25000, EnergyCost.getSuicideCost2(program));
+    invoke.getDeposit().createAccount(receiver2, Protocol.AccountType.Normal);
+    Assert.assertEquals(0, EnergyCost.getSuicideCost2(program));
+  }
+
+  @Test
+  public void testVoteWitnessCost() throws ContractValidateException {
+    // Build stack environment, the stack from top to bottom is 0x00, 0x80, 0x00, 0x80
+    program = new Program(null, null, new ProgramInvokeMockImpl(),
+        new InternalTransaction(
+            Protocol.Transaction.getDefaultInstance(),
+            InternalTransaction.TrxType.TRX_UNKNOWN_TYPE));
+    program.stackPush(DataWord.of((byte) 0x80));
+    program.stackPush(DataWord.of((byte) 0x00));
+    program.stackPush(DataWord.of((byte) 0x80));
+    program.stackPush(DataWord.of((byte) 0x00));
+
+    // Test VoteWitness before EnergyAdjustment, should not have memory expand energy
+    Assert.assertEquals(30000, EnergyCost.getVoteWitnessCost(program));
+
+    // Test VoteWitness after EnergyAdjustment, should have memory expand energy
+    VMConfig.initAllowEnergyAdjustment(1);
+
+    long memWords = (0x80 + 32) / 32;
+    long memoryExpandEnergy = 3 * memWords + memWords * memWords / 512;
+    Assert.assertEquals(30000 + memoryExpandEnergy, EnergyCost.getVoteWitnessCost2(program));
+  }
+
   private void testOperations(Program program) {
     try {
       while (!program.isStopped()) {
@@ -954,4 +1001,10 @@ private byte[] compile(String code) {
     return new BytecodeCompiler().compile(code);
   }
 
+  private byte[] generateRandomAddress() {
+    byte[] address = new byte[21];
+    new Random().nextBytes(address);
+    address[0] = DecodeUtil.addressPreFixByte;
+    return address;
+  }
 }