zaphar/durnitisp
1
0
Fork 0
mirror of https://github.com/zaphar/durnitisp.git synced 2025-07-22 18:19:48 -04:00
Code Issues Packages Projects Releases Wiki Activity

Better handling of non echo packets for ICMPv4

Browse Source
This commit is contained in:
Jeremy Wall 2021-02-02 20:35:34 -05:00
parent 93faffbd42
commit 060e505ab0
1 changed files with 45 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
src/icmp.rs
View File

@ -25,7 +25,7 @@ use icmp_socket::{
packet::{Icmpv4Message, Icmpv6Message, WithEchoRequest}, packet::{Icmpv4Message, Icmpv6Message, WithEchoRequest},
IcmpSocket, IcmpSocket4, IcmpSocket6, Icmpv4Packet, Icmpv6Packet, IcmpSocket, IcmpSocket4, IcmpSocket6, Icmpv4Packet, Icmpv6Packet,
}; };
use log::{error, info}; use log::{debug, error, info};
use prometheus::{CounterVec, GaugeVec}; use prometheus::{CounterVec, GaugeVec};
use socket2::{self, SockAddr}; use socket2::{self, SockAddr};
@ -126,6 +126,9 @@ pub fn start_echo_loop(
.expect(&format!("Invalid IP Address {}", resolved)); .expect(&format!("Invalid IP Address {}", resolved));
let err_handler = |e: std::io::Error, send: bool| { let err_handler = |e: std::io::Error, send: bool| {
ping_counter
.with(&prometheus::labels! {"result" => "err", "domain" => domain_name})
.inc();
if send { if send {
error!( error!(
"ICMP: error sending to domain: {} and address: {} failed: {:?}, Trying again later", "ICMP: error sending to domain: {} and address: {} failed: {:?}, Trying again later",
@ -152,33 +155,46 @@ pub fn start_echo_loop(
Icmpv4Message::ParameterProblem { Icmpv4Message::ParameterProblem {
pointer: _, pointer: _,
padding: _, padding: _,
header: _, header,
} => { } => {
ping_counter let dest_addr =
.with(&prometheus::labels! {"result" => "parameter_problem", "domain" => domain_name}) Ipv4Addr::new(header[16], header[17], header[18], header[19]);
.inc(); if dest_addr == dest {
ping_counter
.with(&prometheus::labels! {"result" => "parameter_problem", "domain" => domain_name})
.inc();
} else {
return None;
}
} }
Icmpv4Message::Unreachable { Icmpv4Message::Unreachable { padding: _, header } => {
padding: _, let dest_addr =
header: _, Ipv4Addr::new(header[16], header[17], header[18], header[19]);
} => { if dest_addr == dest {
info!( info!(
"ICMP: Destination Unreachable {} from {}", "ICMP: Destination: {:?} Unreachable {} response from {}",
dest, dest_addr,
_s.as_inet().unwrap().ip() dest,
); _s.as_inet().unwrap().ip()
ping_counter );
.with(&prometheus::labels! {"result" => "unreachable", "domain" => domain_name}) ping_counter
.inc(); .with(&prometheus::labels! {"result" => "unreachable", "domain" => domain_name})
.inc();
} else {
return None;
}
} }
Icmpv4Message::TimeExceeded { Icmpv4Message::TimeExceeded { padding: _, header } => {
padding: _, let dest_addr =
header: _, Ipv4Addr::new(header[16], header[17], header[18], header[19]);
} => { if dest_addr == dest {
info!("ICMP: Timeout for {}", dest); info!("ICMP: Timeout for {}", dest);
ping_counter ping_counter
.with(&prometheus::labels! {"result" => "timeout", "domain" => domain_name}) .with(&prometheus::labels! {"result" => "timeout", "domain" => domain_name})
.inc(); .inc();
} else {
return None;
}
} }
Icmpv4Message::EchoReply { Icmpv4Message::EchoReply {
identifier, identifier,
@ -190,7 +206,10 @@ pub fn start_echo_loop(
return None; return None;
} }
if sequence != seq { if sequence != seq {
info!("ICMP: Discarding sequence {}", sequence); info!(
"ICMP: Discarding sequence {}, expected sequence {}",
sequence, seq
);
return None; return None;
} }
let elapsed = let elapsed =