blockworks-foundation
/
solana-with-rpc-optimizations
mirror of https://github.com/blockworks-foundation/solana-with-rpc-optimizations.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[zk-token-proof] Add public compute unit variables for instructions (#33129) 

add public compute unit variables
This commit is contained in:
samkim-crypto 2023-09-07 20:55:56 -07:00 committed by GitHub
parent 0083e426b2
commit ece376fdec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 31 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
programs/zk-token-proof/src/lib.rs
View File

@ -16,6 +16,22 @@ use {
std::result::Result, std::result::Result,
}; };
pub const CLOSE_CONTEXT_STATE_COMPUTE_UNITS: u64 = 3_300;
pub const VERIFY_ZERO_BALANCE_COMPUTE_UNITS: u64 = 6012;
pub const VERIFY_WITHDRAW_COMPUTE_UNITS: u64 = 112_454;
pub const VERIFY_CIPHERTEXT_CIPHERTEXT_EQUALITY_COMPUTE_UNITS: u64 = 7_943;
pub const VERIFY_TRANSFER_COMPUTE_UNITS: u64 = 219_290;
pub const VERIFY_TRANSFER_WITH_FEE_COMPUTE_UNITS: u64 = 407_121;
pub const VERIFY_PUBKEY_VALIDITY_COMPUTE_UNITS: u64 = 2_619;
pub const VERIFY_RANGE_PROOF_U64_COMPUTE_UNITS: u64 = 105_066;
pub const VERIFY_BATCHED_RANGE_PROOF_U64_COMPUTE_UNITS: u64 = 111_478;
pub const VERIFY_BATCHED_RANGE_PROOF_U128_COMPUTE_UNITS: u64 = 204_512;
pub const VERIFY_BATCHED_RANGE_PROOF_U256_COMPUTE_UNITS: u64 = 368_000;
pub const VERIFY_CIPHERTEXT_COMMITMENT_EQUALITY_COMPUTE_UNITS: u64 = 6_424;
pub const VERIFY_GROUPED_CIPHERTEXT_2_HANDLES_VALIDITY_COMPUTE_UNITS: u64 = 6_440;
pub const VERIFY_BATCHED_GROUPED_CIPHERTEXT_2_HANDLES_VALIDITY_COMPUTE_UNITS: u64 = 12_575;
pub const VERIFY_FEE_SIGMA_COMPUTE_UNITS: u64 = 6_547;
fn process_verify_proof<T, U>(invoke_context: &mut InvokeContext) -> Result<(), InstructionError> fn process_verify_proof<T, U>(invoke_context: &mut InvokeContext) -> Result<(), InstructionError>
where where
T: Pod + ZkProofData<U>, T: Pod + ZkProofData<U>,
@ -136,7 +152,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::CloseContextState => { ProofInstruction::CloseContextState => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(3_300) .consume_checked(CLOSE_CONTEXT_STATE_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "CloseContextState"); ic_msg!(invoke_context, "CloseContextState");
@ -145,7 +161,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyZeroBalance => { ProofInstruction::VerifyZeroBalance => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(6_012) .consume_checked(VERIFY_ZERO_BALANCE_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyZeroBalance"); ic_msg!(invoke_context, "VerifyZeroBalance");
@ -154,7 +170,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyWithdraw => { ProofInstruction::VerifyWithdraw => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(112_454) .consume_checked(VERIFY_WITHDRAW_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyWithdraw"); ic_msg!(invoke_context, "VerifyWithdraw");
@ -163,7 +179,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyCiphertextCiphertextEquality => { ProofInstruction::VerifyCiphertextCiphertextEquality => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(7_943) .consume_checked(VERIFY_CIPHERTEXT_CIPHERTEXT_EQUALITY_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyCiphertextCiphertextEquality"); ic_msg!(invoke_context, "VerifyCiphertextCiphertextEquality");
@ -175,7 +191,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyTransfer => { ProofInstruction::VerifyTransfer => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(219_290) .consume_checked(VERIFY_TRANSFER_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyTransfer"); ic_msg!(invoke_context, "VerifyTransfer");
@ -184,7 +200,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyTransferWithFee => { ProofInstruction::VerifyTransferWithFee => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(407_121) .consume_checked(VERIFY_TRANSFER_WITH_FEE_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyTransferWithFee"); ic_msg!(invoke_context, "VerifyTransferWithFee");
@ -193,7 +209,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyPubkeyValidity => { ProofInstruction::VerifyPubkeyValidity => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(2_619) .consume_checked(VERIFY_PUBKEY_VALIDITY_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyPubkeyValidity"); ic_msg!(invoke_context, "VerifyPubkeyValidity");
@ -202,7 +218,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyRangeProofU64 => { ProofInstruction::VerifyRangeProofU64 => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(105_066) .consume_checked(VERIFY_RANGE_PROOF_U64_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyRangeProof"); ic_msg!(invoke_context, "VerifyRangeProof");
@ -211,7 +227,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyBatchedRangeProofU64 => { ProofInstruction::VerifyBatchedRangeProofU64 => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(111_478) .consume_checked(VERIFY_BATCHED_RANGE_PROOF_U64_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyBatchedRangeProof64"); ic_msg!(invoke_context, "VerifyBatchedRangeProof64");
@ -222,7 +238,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyBatchedRangeProofU128 => { ProofInstruction::VerifyBatchedRangeProofU128 => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(204_512) .consume_checked(VERIFY_BATCHED_RANGE_PROOF_U128_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyBatchedRangeProof128"); ic_msg!(invoke_context, "VerifyBatchedRangeProof128");
@ -233,7 +249,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
ProofInstruction::VerifyBatchedRangeProofU256 => { ProofInstruction::VerifyBatchedRangeProofU256 => {
if native_programs_consume_cu { if native_programs_consume_cu {
invoke_context invoke_context
.consume_checked(368_000) .consume_checked(VERIFY_BATCHED_RANGE_PROOF_U256_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
} }
ic_msg!(invoke_context, "VerifyBatchedRangeProof256"); ic_msg!(invoke_context, "VerifyBatchedRangeProof256");
@ -243,7 +259,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
} }
ProofInstruction::VerifyCiphertextCommitmentEquality => { ProofInstruction::VerifyCiphertextCommitmentEquality => {
invoke_context invoke_context
.consume_checked(6_424) .consume_checked(VERIFY_CIPHERTEXT_COMMITMENT_EQUALITY_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
ic_msg!(invoke_context, "VerifyCiphertextCommitmentEquality"); ic_msg!(invoke_context, "VerifyCiphertextCommitmentEquality");
process_verify_proof::< process_verify_proof::<
@ -253,7 +269,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
} }
ProofInstruction::VerifyGroupedCiphertext2HandlesValidity => { ProofInstruction::VerifyGroupedCiphertext2HandlesValidity => {
invoke_context invoke_context
.consume_checked(6_440) .consume_checked(VERIFY_GROUPED_CIPHERTEXT_2_HANDLES_VALIDITY_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
ic_msg!(invoke_context, "VerifyGroupedCiphertext2HandlesValidity"); ic_msg!(invoke_context, "VerifyGroupedCiphertext2HandlesValidity");
process_verify_proof::< process_verify_proof::<
@ -263,7 +279,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
} }
ProofInstruction::VerifyBatchedGroupedCiphertext2HandlesValidity => { ProofInstruction::VerifyBatchedGroupedCiphertext2HandlesValidity => {
invoke_context invoke_context
.consume_checked(12_575) .consume_checked(VERIFY_BATCHED_GROUPED_CIPHERTEXT_2_HANDLES_VALIDITY_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
ic_msg!( ic_msg!(
invoke_context, invoke_context,
@ -276,7 +292,7 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
} }
ProofInstruction::VerifyFeeSigma => { ProofInstruction::VerifyFeeSigma => {
invoke_context invoke_context
.consume_checked(6_547) .consume_checked(VERIFY_FEE_SIGMA_COMPUTE_UNITS)
.map_err(|_| InstructionError::ComputationalBudgetExceeded)?; .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
ic_msg!(invoke_context, "VerifyFeeSigma"); ic_msg!(invoke_context, "VerifyFeeSigma");
process_verify_proof::<FeeSigmaProofData, FeeSigmaProofContext>(invoke_context) process_verify_proof::<FeeSigmaProofData, FeeSigmaProofContext>(invoke_context)