Skip to content

Commit 9262b12

Browse files
fix ping query params
1 parent 8dc42b9 commit 9262b12

3 files changed

Lines changed: 9 additions & 14 deletions

File tree

crates/network-scanner/examples/scan.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ fn main() -> anyhow::Result<()> {
3030
toggles: ScannerToggles {
3131
disable_broadcast: false,
3232
disable_subnet_scan: false,
33-
disable_ping_event: false,
33+
disable_ping_start: false,
3434
disable_resolve_dns: false,
3535
disable_zeroconf: false,
3636
},

crates/network-scanner/src/scanner.rs

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ impl NetworkScanner {
249249
let ping_interval = configs.ping_interval;
250250
let ping_timeout = configs.ping_timeout;
251251
let range_to_ping = configs.range_to_ping;
252-
let disable_ping_event = toggles.disable_ping_event;
252+
let disable_ping_start = toggles.disable_ping_start;
253253

254254
for ip_range in range_to_ping {
255255
let (task_manager, runtime, ip_sender) =
@@ -266,21 +266,15 @@ impl NetworkScanner {
266266

267267
while let Some(ping_event) = receiver.recv().await {
268268
debug!(ping_sent_ip = ?ping_event);
269-
if let crate::ping::PingEvent::Success { ip_addr, .. } = ping_event {
270-
ip_sender.send((ip_addr, None)).await?;
271-
};
272-
273-
if disable_ping_event {
274-
continue;
275-
}
276269

277270
match ping_event {
278271
crate::ping::PingEvent::Success { ip_addr, time } => {
272+
ip_sender.send((ip_addr, None)).await?;
279273
result_sender
280274
.send(ScanEntry::ScanEvent(ScanEvent::PingSuccess { ip_addr, time }))
281275
.await?;
282276
}
283-
crate::ping::PingEvent::Start { ip_addr } => {
277+
crate::ping::PingEvent::Start { ip_addr } if !disable_ping_start => {
284278
result_sender
285279
.send(ScanEntry::ScanEvent(ScanEvent::PingStart { ip_addr }))
286280
.await?;
@@ -290,6 +284,7 @@ impl NetworkScanner {
290284
.send(ScanEntry::ScanEvent(ScanEvent::PingFailed { ip_addr, reason }))
291285
.await?;
292286
}
287+
_ => {}
293288
}
294289
}
295290
}
@@ -471,7 +466,7 @@ impl Display for ScanMethod {
471466

472467
#[derive(Debug, Clone)]
473468
pub struct ScannerToggles {
474-
pub disable_ping_event: bool,
469+
pub disable_ping_start: bool,
475470
pub disable_broadcast: bool,
476471
pub disable_subnet_scan: bool,
477472
pub disable_zeroconf: bool,

devolutions-gateway/src/api/net.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -145,9 +145,9 @@ pub struct NetworkScanQueryParams {
145145
#[serde(default)]
146146
pub ports: Vec<u16>,
147147

148-
/// Disable the emission of ScanEvent::Ping
148+
/// Disable the emission of ScanEvent::Ping for status start
149149
#[serde(default = "default_true")]
150-
pub disable_ping_events: bool,
150+
pub disable_ping_start: bool,
151151

152152
/// Disable the execution of broadcast scan
153153
#[serde(default)]
@@ -200,7 +200,7 @@ impl TryFrom<NetworkScanQueryParams> for NetworkScannerParams {
200200
.collect::<Result<Vec<IpAddrRange>, anyhow::Error>>()?,
201201
},
202202
toggles: scanner::ScannerToggles {
203-
disable_ping_event: val.disable_ping_events,
203+
disable_ping_start: val.disable_ping_start,
204204
disable_broadcast: val.disable_boardcast,
205205
disable_subnet_scan: val.disable_subnet_scan,
206206
disable_zeroconf: val.disable_zeroconf,

0 commit comments

Comments
 (0)