Skip to content

Commit

Permalink
add secp256r1 bench
Browse files Browse the repository at this point in the history
  • Loading branch information
samkim-crypto committed Dec 2, 2024
1 parent f454a53 commit 10d78bd
Show file tree
Hide file tree
Showing 3 changed files with 98 additions and 0 deletions.
1 change: 1 addition & 0 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions sdk/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -189,6 +189,7 @@ anyhow = { workspace = true }
assert_matches = { workspace = true }
curve25519-dalek = { workspace = true }
hex = { workspace = true }
openssl = { workspace = true }
solana-logger = { workspace = true }
solana-program = { workspace = true, features = ["dev-context-only-utils"] }
solana-sdk = { path = ".", features = ["dev-context-only-utils"] }
Expand Down
96 changes: 96 additions & 0 deletions sdk/benches/secp256r1_instructions.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,96 @@
#![feature(test)]

extern crate test;
use {
openssl::{ec::{EcGroup, EcKey}, nid::Nid},
rand0_7::{thread_rng, Rng},
solana_feature_set::FeatureSet,
solana_sdk::{
hash::Hash,
signature::{Keypair, Signer},
transaction::Transaction,
},
solana_secp256r1_program::new_secp256r1_instruction,
test::Bencher,
};

// 5k transactions should be enough for benching loop
const TX_COUNT: u16 = 5120;

// prepare a bunch of unique txs
fn create_test_transactions(message_length: u16) -> Vec<Transaction> {
(0..TX_COUNT)
.map(|_| {
let mut rng = thread_rng();
let group = EcGroup::from_curve_name(Nid::X9_62_PRIME256V1).unwrap();
let secp_privkey = EcKey::generate(&group).unwrap();
let message: Vec<u8> = (0..message_length).map(|_| rng.gen_range(0, 255)).collect();
let secp_instruction = new_secp256r1_instruction(&message, secp_privkey).unwrap();
let mint_keypair = Keypair::new();

Transaction::new_signed_with_payer(
&[secp_instruction.clone()],
Some(&mint_keypair.pubkey()),
&[&mint_keypair],
Hash::default(),
)
})
.collect()
}

#[bench]
fn bench_secp256r1_len_032(b: &mut Bencher) {
let feature_set = FeatureSet::all_enabled();
let txs = create_test_transactions(32);
let mut tx_iter = txs.iter().cycle();
b.iter(|| {
tx_iter
.next()
.unwrap()
.verify_precompiles(&feature_set)
.unwrap();
});
}

#[bench]
fn bench_secp256r1_len_256(b: &mut Bencher) {
let feature_set = FeatureSet::all_enabled();
let txs = create_test_transactions(256);
let mut tx_iter = txs.iter().cycle();
b.iter(|| {
tx_iter
.next()
.unwrap()
.verify_precompiles(&feature_set)
.unwrap();
});
}

#[bench]
fn bench_secp256r1_len_32k(b: &mut Bencher) {
let feature_set = FeatureSet::all_enabled();
let txs = create_test_transactions(32 * 1024);
let mut tx_iter = txs.iter().cycle();
b.iter(|| {
tx_iter
.next()
.unwrap()
.verify_precompiles(&feature_set)
.unwrap();
});
}

#[bench]
fn bench_secp256r1_len_max(b: &mut Bencher) {
let required_extra_space = 113_u16; // len for pubkey, sig, and offsets
let feature_set = FeatureSet::all_enabled();
let txs = create_test_transactions(u16::MAX - required_extra_space);
let mut tx_iter = txs.iter().cycle();
b.iter(|| {
tx_iter
.next()
.unwrap()
.verify_precompiles(&feature_set)
.unwrap();
});
}

0 comments on commit 10d78bd

Please sign in to comment.