refactor(zen): replace arithmetic operators with StrictMathWrapper

Author: Federico2014

framework/src/main/java/org/tron/core/zen/ShieldedTRC20ParametersBuilder.java

@@ -14,6 +14,7 @@
 import org.tron.api.GrpcAPI;
 import org.tron.api.GrpcAPI.BytesMessage;
 import org.tron.api.GrpcAPI.ShieldedTRC20Parameters;
+import org.tron.common.math.StrictMathWrapper;
 import org.tron.common.utils.ByteArray;
 import org.tron.common.utils.ByteUtil;
 import org.tron.common.utils.Sha256Hash;
@@ -548,7 +549,7 @@ public void addSpend(
       byte[] path,
       long position) throws ZksnarkException {
     spends.add(new SpendDescriptionInfo(expsk, note, anchor, path, position));
-    valueBalance += note.getValue();
+    valueBalance = StrictMathWrapper.addExact(valueBalance, note.getValue());
   }
 
   public void addSpend(
@@ -559,7 +560,7 @@ public void addSpend(
       byte[] path,
       long position) {
     spends.add(new SpendDescriptionInfo(expsk, note, alpha, anchor, path, position));
-    valueBalance += note.getValue();
+    valueBalance = StrictMathWrapper.addExact(valueBalance, note.getValue());
   }
 
   public void addSpend(
@@ -571,22 +572,22 @@ public void addSpend(
       byte[] path,
       long position) {
     spends.add(new SpendDescriptionInfo(ak, nsk, note, alpha, anchor, path, position));
-    valueBalance += note.getValue();
+    valueBalance = StrictMathWrapper.addExact(valueBalance, note.getValue());
   }
 
   public void addOutput(byte[] ovk, PaymentAddress to, long value, byte[] memo)
       throws ZksnarkException {
     Note note = new Note(to, value);
     note.setMemo(memo);
     receives.add(new ReceiveDescriptionInfo(ovk, note));
-    valueBalance -= value;
+    valueBalance = StrictMathWrapper.subtractExact(valueBalance, value);
   }
 
   public void addOutput(byte[] ovk, DiversifierT d, byte[] pkD, long value, byte[] r, byte[] memo) {
     Note note = new Note(d, pkD, value, r);
     note.setMemo(memo);
     receives.add(new ReceiveDescriptionInfo(ovk, note));
-    valueBalance -= value;
+    valueBalance = StrictMathWrapper.subtractExact(valueBalance, value);
   }
 
   public static class SpendDescriptionInfo {

framework/src/main/java/org/tron/core/zen/ZenTransactionBuilder.java

@@ -10,6 +10,7 @@
 import lombok.Setter;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.ArrayUtils;
+import org.tron.common.math.StrictMathWrapper;
 import org.tron.common.utils.ByteArray;
 import org.tron.common.zksnark.IncrementalMerkleVoucherContainer;
 import org.tron.common.zksnark.JLibrustzcash;
@@ -67,7 +68,7 @@ public ZenTransactionBuilder() {
 
   public void addSpend(SpendDescriptionInfo spendDescriptionInfo) {
     spends.add(spendDescriptionInfo);
-    valueBalance += spendDescriptionInfo.note.getValue();
+    valueBalance = StrictMathWrapper.addExact(valueBalance, spendDescriptionInfo.note.getValue());
   }
 
   public void addSpend(
@@ -76,7 +77,7 @@ public void addSpend(
       byte[] anchor,
       IncrementalMerkleVoucherContainer voucher) throws ZksnarkException {
     spends.add(new SpendDescriptionInfo(expsk, note, anchor, voucher));
-    valueBalance += note.getValue();
+    valueBalance = StrictMathWrapper.addExact(valueBalance, note.getValue());
   }
 
   public void addSpend(
@@ -86,7 +87,7 @@ public void addSpend(
       byte[] anchor,
       IncrementalMerkleVoucherContainer voucher) {
     spends.add(new SpendDescriptionInfo(expsk, note, alpha, anchor, voucher));
-    valueBalance += note.getValue();
+    valueBalance = StrictMathWrapper.addExact(valueBalance, note.getValue());
   }
 
   public void addSpend(
@@ -98,22 +99,22 @@ public void addSpend(
       byte[] anchor,
       IncrementalMerkleVoucherContainer voucher) {
     spends.add(new SpendDescriptionInfo(ak, nsk, ovk, note, alpha, anchor, voucher));
-    valueBalance += note.getValue();
+    valueBalance = StrictMathWrapper.addExact(valueBalance, note.getValue());
   }
 
   public void addOutput(byte[] ovk, PaymentAddress to, long value, byte[] memo)
       throws ZksnarkException {
     Note note = new Note(to, value);
     note.setMemo(memo);
     receives.add(new ReceiveDescriptionInfo(ovk, note));
-    valueBalance -= value;
+    valueBalance = StrictMathWrapper.subtractExact(valueBalance, value);
   }
 
   public void addOutput(byte[] ovk, DiversifierT d, byte[] pkD, long value, byte[] r, byte[] memo) {
     Note note = new Note(d, pkD, value, r);
     note.setMemo(memo);
     receives.add(new ReceiveDescriptionInfo(ovk, note));
-    valueBalance -= value;
+    valueBalance = StrictMathWrapper.subtractExact(valueBalance, value);
   }
 
   public void setTransparentInput(byte[] address, long value) {