Handle gas estimation for decryp() and reecrypt() as well

FhenixProtocol · Jan 2, 2024 · 9000a14 · 9000a14
1 parent c10d771
commit 9000a14
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/precompiles/contracts.go b/precompiles/contracts.go
@@ -140,6 +140,10 @@ func Reencrypt(input []byte, tp *TxParams) ([]byte, error) {
 		return nil, vm.ErrExecutionReverted
 	}
 
+	if tp.GasEstimation {
+		return []byte{1}, nil
+	}
+
 	decryptedValue, err := tfhe.Decrypt(*ct)
 	if err != nil {
 		logger.Error("failed decrypting ciphertext ", "error ", err)
@@ -178,6 +182,10 @@ func Decrypt(input []byte, tp *TxParams) (*big.Int, error) {
 		return nil, vm.ErrExecutionReverted
 	}
 
+	if tp.GasEstimation {
+		return new(big.Int).SetUint64(1), nil
+	}
+
 	decryptedValue, err := tfhe.Decrypt(*ct)
 	if err != nil {
 		logger.Error("failed decrypting ciphertext", " error ", err)