Skip to content

Commit f5a6787

Browse files
Refactor port range handling in benchmarks
1 parent 6e8487c commit f5a6787

File tree

1 file changed

+6
-12
lines changed

1 file changed

+6
-12
lines changed

benches/benchmark_portscan.rs

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
use async_std::task::block_on;
22
use criterion::{criterion_group, criterion_main, Criterion};
3-
use rustscan::input::{Opts, PortRange, ScanOrder};
3+
use rustscan::input::{Opts, PortRanges, ScanOrder};
44
use rustscan::port_strategy::PortStrategy;
55
use rustscan::scanner::Scanner;
66
use std::hint::black_box;
@@ -20,11 +20,8 @@ fn bench_address() {
2020
}
2121

2222
fn bench_port_strategy() {
23-
let range = PortRange {
24-
start: 1,
25-
end: 1_000,
26-
};
27-
let _strategy = PortStrategy::pick(&Some(range.clone()), None, ScanOrder::Serial);
23+
let range = PortRanges(vec![(1, 1_000)]);
24+
let _strategy = PortStrategy::pick(Some(range.clone()), None, ScanOrder::Serial);
2825
}
2926

3027
fn bench_address_parsing() {
@@ -47,12 +44,9 @@ fn bench_address_parsing() {
4744

4845
fn criterion_benchmark(c: &mut Criterion) {
4946
let addrs = vec!["127.0.0.1".parse::<IpAddr>().unwrap()];
50-
let range = PortRange {
51-
start: 1,
52-
end: 1_000,
53-
};
54-
let strategy_tcp = PortStrategy::pick(&Some(range.clone()), None, ScanOrder::Serial);
55-
let strategy_udp = PortStrategy::pick(&Some(range.clone()), None, ScanOrder::Serial);
47+
let range = PortRanges(vec![(1, 1_000)]);
48+
let strategy_tcp = PortStrategy::pick(Some(range.clone()), None, ScanOrder::Serial);
49+
let strategy_udp = PortStrategy::pick(Some(range.clone()), None, ScanOrder::Serial);
5650

5751
let scanner_tcp = Scanner::new(
5852
&addrs,

0 commit comments

Comments
 (0)