forked from solana-labs/solana
-
Notifications
You must be signed in to change notification settings - Fork 231
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* add secp256r1 bench * cargo fmt * apply feature `openssl-vendored` for bench scripts
- Loading branch information
1 parent
6a867ad
commit 9c899a7
Showing
6 changed files
with
104 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,99 @@ | ||
#![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(); | ||
}); | ||
} |