Skip to content

Commit

Permalink
chore: move params to fhevm
Browse files Browse the repository at this point in the history 
Update with latest gas costs for decryption and div/rem.

Remove the dependency on go-ethereum in params.
  • Loading branch information
@dartdart26
dartdart26 committed Nov 3, 2023
1 parent 3a2f516 commit bd653e0
Show file tree
Hide file tree
Showing 2 changed files with 100 additions and 73 deletions.
80 changes: 54 additions & 26 deletions params/fhevm_params.go → fhevm/params.go
View file
Original file line number Diff line number Diff line change
@@ -1,21 +1,49 @@
package params
// BSD 3-Clause Clear License

import evm "github.com/ethereum/go-ethereum/params"
// Copyright © 2023 ZAMA.
// All rights reserved.

// Copyright 2015 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

package fhevm

// This file contains default gas costs of fhEVM-related operations.
// Users can change the values based on specific requirements in their blockchain.

// Base gas costs of existing EVM operations. Used for setting gas costs relative to them.
// These constants are used just for readability.
const EvmNetSstoreInitGas uint64 = 20000
const ColdSloadCostEIP2929 uint64 = 2100

var (
// FHE operation costs depend on tfhe-rs performance and hardware acceleration. These values will most certainly change.
FheUint8AddSubGas uint64 = 83000
FheUint16AddSubGas uint64 = 108000
FheUint32AddSubGas uint64 = 130000
FheUint8MulGas uint64 = 150000
FheUint16MulGas uint64 = 200000
FheUint32MulGas uint64 = 270000
FheUint8DivGas uint64 = 1370000
FheUint16DivGas uint64 = 3500000
FheUint32DivGas uint64 = 9120000
FheUint8RemGas uint64 = 1370000 // TODO: check again rem gas
FheUint16RemGas uint64 = 3500000
FheUint32RemGas uint64 = 9120000
FheUint8AddSubGas uint64 = 83000
FheUint16AddSubGas uint64 = 108000
FheUint32AddSubGas uint64 = 130000
FheUint8MulGas uint64 = 150000
FheUint16MulGas uint64 = 200000
FheUint32MulGas uint64 = 270000
// Div and Rem currently only support a plaintext divisor and below gas costs reflect that case only.
FheUint8DivGas uint64 = 200000
FheUint16DivGas uint64 = 250000
FheUint32DivGas uint64 = 350000
FheUint8RemGas uint64 = 200000
FheUint16RemGas uint64 = 250000
FheUint32RemGas uint64 = 350000
FheUint8BitwiseGas uint64 = 20000
FheUint16BitwiseGas uint64 = 21000
FheUint32BitwiseGas uint64 = 22000
Expand All @@ -33,22 +61,22 @@ var (
FheUint32NegNotGas uint64 = 130000

// TODO: Costs will depend on the complexity of doing reencryption/decryption by the oracle.
FheUint8ReencryptGas uint64 = 1000
FheUint16ReencryptGas uint64 = 1100
FheUint32ReencryptGas uint64 = 1200
FheUint8DecryptGas uint64 = 600
FheUint16DecryptGas uint64 = 700
FheUint32DecryptGas uint64 = 800
FheUint8ReencryptGas uint64 = 320000
FheUint16ReencryptGas uint64 = 320400
FheUint32ReencryptGas uint64 = 320800
FheUint8DecryptGas uint64 = 320000
FheUint16DecryptGas uint64 = 320400
FheUint32DecryptGas uint64 = 320800

// As of now, verification costs only cover ciphertext deserialization and assume there is no ZKPoK to verify.
FheUint8VerifyGas uint64 = 200
FheUint16VerifyGas uint64 = 300
FheUint32VerifyGas uint64 = 400

// TODO: Cost will depend on the complexity of doing decryption by the oracle.
FheUint8RequireGas uint64 = 170000
FheUint16RequireGas uint64 = 180000
FheUint32RequireGas uint64 = 190000
FheUint8RequireGas uint64 = 320000
FheUint16RequireGas uint64 = 320400
FheUint32RequireGas uint64 = 320800

// TODO: As of now, only support FheUint8. All optimistic require predicates are
// downcast to FheUint8 at the solidity level. Eventually move to ebool.
Expand All @@ -59,19 +87,19 @@ var (
FheUint8OptimisticRequireBitandGas uint64 = FheUint8BitwiseGas

// TODO: These will change once we have an FHE-based random generaration.
FheUint8RandGas uint64 = evm.NetSstoreInitGas + 1000
FheUint8RandGas uint64 = EvmNetSstoreInitGas + 1000
FheUint16RandGas uint64 = FheUint8RandGas + 1000
FheUint32RandGas uint64 = FheUint16RandGas + 1000

// TODO: The values here are chosen somewhat arbitrarily (at least the 8 bit ones). Also, we don't
// take into account whether a ciphertext existed (either "current" or "original") for the given handle.
// Finally, costs are likely to change in the future.
FheUint8ProtectedStorageSstoreGas uint64 = evm.NetSstoreInitGas + 2000
FheUint8ProtectedStorageSstoreGas uint64 = EvmNetSstoreInitGas + 2000
FheUint16ProtectedStorageSstoreGas uint64 = FheUint8ProtectedStorageSstoreGas * 2
FheUint32ProtectedStorageSstoreGas uint64 = FheUint16ProtectedStorageSstoreGas * 2

// TODO: We don't take whether the slot is cold or warm into consideration.
FheUint8ProtectedStorageSloadGas uint64 = evm.ColdSloadCostEIP2929 + 200
FheUint8ProtectedStorageSloadGas uint64 = ColdSloadCostEIP2929 + 200
FheUint16ProtectedStorageSloadGas uint64 = FheUint8ProtectedStorageSloadGas * 2
FheUint32ProtectedStorageSloadGas uint64 = FheUint16ProtectedStorageSloadGas * 2

Expand Down
93 changes: 46 additions & 47 deletions fhevm/precompiles.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@ import (
"github.com/ethereum/go-ethereum/crypto"
"github.com/holiman/uint256"
fhevm_crypto "github.com/zama-ai/fhevm-go/crypto"
"github.com/zama-ai/fhevm-go/params"
"golang.org/x/crypto/chacha20"
"golang.org/x/crypto/nacl/box"
)
Expand Down Expand Up @@ -287,87 +286,87 @@ func FheLibRun(environment EVMEnvironment, caller common.Address, addr common.Ad
}

var fheAddSubGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8AddSubGas,
FheUint16: params.FheUint16AddSubGas,
FheUint32: params.FheUint32AddSubGas,
FheUint8: FheUint8AddSubGas,
FheUint16: FheUint16AddSubGas,
FheUint32: FheUint32AddSubGas,
}

var fheDecryptGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8DecryptGas,
FheUint16: params.FheUint16DecryptGas,
FheUint32: params.FheUint32DecryptGas,
FheUint8: FheUint8DecryptGas,
FheUint16: FheUint16DecryptGas,
FheUint32: FheUint32DecryptGas,
}

var fheBitwiseOpGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8BitwiseGas,
FheUint16: params.FheUint16BitwiseGas,
FheUint32: params.FheUint32BitwiseGas,
FheUint8: FheUint8BitwiseGas,
FheUint16: FheUint16BitwiseGas,
FheUint32: FheUint32BitwiseGas,
}

var fheMulGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8MulGas,
FheUint16: params.FheUint16MulGas,
FheUint32: params.FheUint32MulGas,
FheUint8: FheUint8MulGas,
FheUint16: FheUint16MulGas,
FheUint32: FheUint32MulGas,
}

var fheDivGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8DivGas,
FheUint16: params.FheUint16DivGas,
FheUint32: params.FheUint32DivGas,
FheUint8: FheUint8DivGas,
FheUint16: FheUint16DivGas,
FheUint32: FheUint32DivGas,
}

var fheRemGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8RemGas,
FheUint16: params.FheUint16RemGas,
FheUint32: params.FheUint32RemGas,
FheUint8: FheUint8RemGas,
FheUint16: FheUint16RemGas,
FheUint32: FheUint32RemGas,
}

var fheShiftGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8ShiftGas,
FheUint16: params.FheUint16ShiftGas,
FheUint32: params.FheUint32ShiftGas,
FheUint8: FheUint8ShiftGas,
FheUint16: FheUint16ShiftGas,
FheUint32: FheUint32ShiftGas,
}

var fheLeGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8LeGas,
FheUint16: params.FheUint16LeGas,
FheUint32: params.FheUint32LeGas,
FheUint8: FheUint8LeGas,
FheUint16: FheUint16LeGas,
FheUint32: FheUint32LeGas,
}

var fheMinMaxGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8MinMaxGas,
FheUint16: params.FheUint16MinMaxGas,
FheUint32: params.FheUint32MinMaxGas,
FheUint8: FheUint8MinMaxGas,
FheUint16: FheUint16MinMaxGas,
FheUint32: FheUint32MinMaxGas,
}

var fheNegNotGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8NegNotGas,
FheUint16: params.FheUint16NegNotGas,
FheUint32: params.FheUint32NegNotGas,
FheUint8: FheUint8NegNotGas,
FheUint16: FheUint16NegNotGas,
FheUint32: FheUint32NegNotGas,
}

var fheReencryptGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8ReencryptGas,
FheUint16: params.FheUint16ReencryptGas,
FheUint32: params.FheUint32ReencryptGas,
FheUint8: FheUint8ReencryptGas,
FheUint16: FheUint16ReencryptGas,
FheUint32: FheUint32ReencryptGas,
}

var fheVerifyGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8VerifyGas,
FheUint16: params.FheUint16VerifyGas,
FheUint32: params.FheUint32VerifyGas,
FheUint8: FheUint8VerifyGas,
FheUint16: FheUint16VerifyGas,
FheUint32: FheUint32VerifyGas,
}

var fheTrivialEncryptGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8TrivialEncryptGas,
FheUint16: params.FheUint16TrivialEncryptGas,
FheUint32: params.FheUint32TrivialEncryptGas,
FheUint8: FheUint8TrivialEncryptGas,
FheUint16: FheUint16TrivialEncryptGas,
FheUint32: FheUint32TrivialEncryptGas,
}

var fheRandGasCosts = map[fheUintType]uint64{
FheUint8: params.FheUint8RandGas,
FheUint16: params.FheUint16RandGas,
FheUint32: params.FheUint32RandGas,
FheUint8: FheUint8RandGas,
FheUint16: FheUint16RandGas,
FheUint32: FheUint32RandGas,
}

// Gas costs
Expand Down Expand Up @@ -699,9 +698,9 @@ func optimisticRequireRequiredGas(environment EVMEnvironment, input []byte) uint
return 0
}
if len(environment.GetFhevmData().optimisticRequires) == 0 {
return params.FheUint8OptimisticRequireGas
return FheUint8OptimisticRequireGas
}
return params.FheUint8OptimisticRequireBitandGas
return FheUint8OptimisticRequireBitandGas
}

func castRequiredGas(environment EVMEnvironment, input []byte) uint64 {
Expand All @@ -711,7 +710,7 @@ func castRequiredGas(environment EVMEnvironment, input []byte) uint64 {
"len", len(input))
return 0
}
return params.FheCastGas
return FheCastGas
}

func decryptRequiredGas(environment EVMEnvironment, input []byte) uint64 {
Expand All @@ -729,7 +728,7 @@ func decryptRequiredGas(environment EVMEnvironment, input []byte) uint64 {
}

func fhePubKeyRequiredGas(environment EVMEnvironment, input []byte) uint64 {
return params.FhePubKeyGas
return FhePubKeyGas
}

func trivialEncryptRequiredGas(environment EVMEnvironment, input []byte) uint64 {
Expand Down

0 comments on commit bd653e0

Please sign in to comment.