Skip to content

Commit bf0a0c1

Browse files
committed
Shorten preset function names
1 parent 7817743 commit bf0a0c1

File tree

4 files changed

+60
-67
lines changed

4 files changed

+60
-67
lines changed

README.md

+4-2
Original file line numberDiff line numberDiff line change
@@ -25,15 +25,17 @@ Find a 196 bit prime returned in a 256-bit long `crypto_bigint::U256`:
2525

2626
```rust
2727
use crypto_bigint::U256;
28-
let prime = crypto_primes::generate_prime::<U256>(196);
28+
use rand_core::{OsRng, TryRngCore};
29+
let prime = crypto_primes::random_prime::<U256, _>(&mut OsRng.unwrap_err(), 196);
2930
assert!(crypto_primes::is_prime(&prime));
3031
```
3132

3233
Find a 64 bit safe prime returned in a `crypto_bigint::U1024`:
3334

3435
```rust
3536
use crypto_bigint::U1024;
36-
let prime = crypto_primes::generate_safe_prime::<U1024>(64);
37+
use rand_core::{OsRng, TryRngCore};
38+
let prime = crypto_primes::random_safe_prime::<U1024, _>(&mut OsRng.unwrap_err(), 64);
3739
assert!(crypto_primes::is_safe_prime(&prime));
3840
```
3941

benches/bench.rs

+21-21
Original file line numberDiff line numberDiff line change
@@ -12,16 +12,16 @@ use rug::{integer::Order, Integer as GmpInteger};
1212
use openssl::bn::BigNum;
1313

1414
use crypto_primes::{
15-
fips_is_prime_with_rng, generate_prime_with_rng, generate_safe_prime_with_rng,
15+
fips_is_prime,
1616
hazmat::{
1717
lucas_test, minimum_mr_iterations, random_odd_integer, AStarBase, BruteForceBase, LucasCheck, MillerRabin,
1818
SelfridgeBase, SetBits, SmallPrimesSieve,
1919
},
20-
is_prime, is_safe_prime,
20+
is_prime, is_safe_prime, random_prime, random_safe_prime,
2121
};
2222

2323
#[cfg(feature = "multicore")]
24-
use crypto_primes::{par_generate_prime_with_rng, par_generate_safe_prime_with_rng};
24+
use crypto_primes::{par_random_prime, par_random_safe_prime};
2525

2626
fn make_rng() -> ChaCha8Rng {
2727
ChaCha8Rng::from_seed(*b"01234567890123456789012345678901")
@@ -234,7 +234,7 @@ fn bench_presets(c: &mut Criterion) {
234234
group.bench_function("(U1024) Prime test (FIPS, 1/2^128 failure bound)", |b| {
235235
b.iter_batched(
236236
|| random_odd_uint::<U1024, _>(&mut OsRng.unwrap_err(), 1024),
237-
|num| fips_is_prime_with_rng(&mut OsRng.unwrap_err(), num.as_ref(), iters, false),
237+
|num| fips_is_prime(&mut OsRng.unwrap_err(), num.as_ref(), iters, false),
238238
BatchSize::SmallInput,
239239
)
240240
});
@@ -249,34 +249,34 @@ fn bench_presets(c: &mut Criterion) {
249249

250250
let mut rng = make_rng();
251251
group.bench_function("(U128) Random prime", |b| {
252-
b.iter(|| generate_prime_with_rng::<U128, _>(&mut rng, 128))
252+
b.iter(|| random_prime::<U128, _>(&mut rng, 128))
253253
});
254254

255255
let mut rng = make_rng();
256256
group.bench_function("(U1024) Random prime", |b| {
257-
b.iter(|| generate_prime_with_rng::<U1024, _>(&mut rng, 1024))
257+
b.iter(|| random_prime::<U1024, _>(&mut rng, 1024))
258258
});
259259

260260
let mut rng = make_rng();
261261
group.bench_function("(U128) Random safe prime", |b| {
262-
b.iter(|| generate_safe_prime_with_rng::<U128, _>(&mut rng, 128))
262+
b.iter(|| random_safe_prime::<U128, _>(&mut rng, 128))
263263
});
264264

265265
group.sample_size(20);
266266
let mut rng = make_rng();
267267
group.bench_function("(U1024) Random safe prime", |b| {
268-
b.iter(|| generate_safe_prime_with_rng::<U1024, _>(&mut rng, 1024))
268+
b.iter(|| random_safe_prime::<U1024, _>(&mut rng, 1024))
269269
});
270270

271271
let mut rng = make_rng();
272272
group.bench_function("(Boxed128) Random safe prime", |b| {
273-
b.iter(|| generate_safe_prime_with_rng::<BoxedUint, _>(&mut rng, 128))
273+
b.iter(|| random_safe_prime::<BoxedUint, _>(&mut rng, 128))
274274
});
275275

276276
group.sample_size(20);
277277
let mut rng = make_rng();
278278
group.bench_function("(Boxed1024) Random safe prime", |b| {
279-
b.iter(|| generate_safe_prime_with_rng::<BoxedUint, _>(&mut rng, 1024))
279+
b.iter(|| random_safe_prime::<BoxedUint, _>(&mut rng, 1024))
280280
});
281281

282282
group.finish();
@@ -286,19 +286,19 @@ fn bench_presets(c: &mut Criterion) {
286286

287287
let mut rng = make_rng();
288288
group.bench_function("(U128) Random safe prime", |b| {
289-
b.iter(|| generate_safe_prime_with_rng::<U128, _>(&mut rng, 128))
289+
b.iter(|| random_safe_prime::<U128, _>(&mut rng, 128))
290290
});
291291

292292
// The performance should scale with the prime size, not with the Uint size.
293293
// So we should strive for this test's result to be as close as possible
294294
// to that of the previous one and as far away as possible from the next one.
295295
group.bench_function("(U256) Random 128 bit safe prime", |b| {
296-
b.iter(|| generate_safe_prime_with_rng::<U256, _>(&mut rng, 128))
296+
b.iter(|| random_safe_prime::<U256, _>(&mut rng, 128))
297297
});
298298

299299
// The upper bound for the previous test.
300300
group.bench_function("(U256) Random 256 bit safe prime", |b| {
301-
b.iter(|| generate_safe_prime_with_rng::<U256, _>(&mut rng, 256))
301+
b.iter(|| random_safe_prime::<U256, _>(&mut rng, 256))
302302
});
303303

304304
group.finish();
@@ -311,23 +311,23 @@ fn bench_multicore_presets(c: &mut Criterion) {
311311
group.bench_function("(U128) Random prime", |b| {
312312
b.iter_batched(
313313
make_random_rng,
314-
|mut rng| par_generate_prime_with_rng::<U128, _>(&mut rng, 128, num_cpus::get()),
314+
|mut rng| par_random_prime::<U128, _>(&mut rng, 128, num_cpus::get()),
315315
BatchSize::SmallInput,
316316
)
317317
});
318318

319319
group.bench_function("(U1024) Random prime", |b| {
320320
b.iter_batched(
321321
make_random_rng,
322-
|mut rng| par_generate_prime_with_rng::<U1024, _>(&mut rng, 1024, num_cpus::get()),
322+
|mut rng| par_random_prime::<U1024, _>(&mut rng, 1024, num_cpus::get()),
323323
BatchSize::SmallInput,
324324
)
325325
});
326326

327327
group.bench_function("(U128) Random safe prime", |b| {
328328
b.iter_batched(
329329
make_random_rng,
330-
|mut rng| par_generate_safe_prime_with_rng::<U128, _>(&mut rng, 128, num_cpus::get()),
330+
|mut rng| par_random_safe_prime::<U128, _>(&mut rng, 128, num_cpus::get()),
331331
BatchSize::SmallInput,
332332
)
333333
});
@@ -336,15 +336,15 @@ fn bench_multicore_presets(c: &mut Criterion) {
336336
group.bench_function("(U1024) Random safe prime", |b| {
337337
b.iter_batched(
338338
make_random_rng,
339-
|mut rng| par_generate_safe_prime_with_rng::<U1024, _>(&mut rng, 1024, num_cpus::get()),
339+
|mut rng| par_random_safe_prime::<U1024, _>(&mut rng, 1024, num_cpus::get()),
340340
BatchSize::SmallInput,
341341
)
342342
});
343343

344344
group.bench_function("(Boxed128) Random safe prime", |b| {
345345
b.iter_batched(
346346
make_random_rng,
347-
|mut rng| par_generate_safe_prime_with_rng::<BoxedUint, _>(&mut rng, 128, num_cpus::get()),
347+
|mut rng| par_random_safe_prime::<BoxedUint, _>(&mut rng, 128, num_cpus::get()),
348348
BatchSize::SmallInput,
349349
)
350350
});
@@ -353,7 +353,7 @@ fn bench_multicore_presets(c: &mut Criterion) {
353353
group.bench_function("(Boxed1024) Random safe prime", |b| {
354354
b.iter_batched(
355355
make_random_rng,
356-
|mut rng| par_generate_safe_prime_with_rng::<BoxedUint, _>(&mut rng, 1024, num_cpus::get()),
356+
|mut rng| par_random_safe_prime::<BoxedUint, _>(&mut rng, 1024, num_cpus::get()),
357357
BatchSize::SmallInput,
358358
)
359359
});
@@ -539,7 +539,7 @@ fn bench_glass_pumpkin(c: &mut Criterion) {
539539

540540
let mut rng = make_rng();
541541
group.bench_function("(U1024) Random prime (crypto-primes default)", |b| {
542-
b.iter(|| generate_prime_with_rng::<BoxedUint, _>(&mut rng, 1024))
542+
b.iter(|| random_prime::<BoxedUint, _>(&mut rng, 1024))
543543
});
544544

545545
let mut rng = make_rng();
@@ -556,7 +556,7 @@ fn bench_glass_pumpkin(c: &mut Criterion) {
556556

557557
let mut rng = make_rng();
558558
group.bench_function("(U1024) Random safe prime (crypto-primes default)", |b| {
559-
b.iter(|| generate_safe_prime_with_rng::<BoxedUint, _>(&mut rng, 1024))
559+
b.iter(|| random_safe_prime::<BoxedUint, _>(&mut rng, 1024))
560560
});
561561

562562
let mut rng = make_rng();

src/lib.rs

+2-5
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,11 @@ mod presets;
2222
mod traits;
2323

2424
pub use generic::sieve_and_find;
25-
pub use presets::{
26-
fips_is_prime_with_rng, fips_is_safe_prime_with_rng, generate_prime_with_rng, generate_safe_prime_with_rng,
27-
is_prime, is_safe_prime,
28-
};
25+
pub use presets::{fips_is_prime, fips_is_safe_prime, is_prime, is_safe_prime, random_prime, random_safe_prime};
2926
pub use traits::SieveFactory;
3027

3128
#[cfg(feature = "multicore")]
32-
pub use presets::{par_generate_prime_with_rng, par_generate_safe_prime_with_rng};
29+
pub use presets::{par_random_prime, par_random_safe_prime};
3330

3431
#[cfg(feature = "multicore")]
3532
pub use generic::par_sieve_and_find;

0 commit comments

Comments
 (0)