zpinger 0.6.0

Async, protocol-agnostic latency probe library: TCP / UDP / HTTP(S) / WebSocket / DNS / MQTT / gRPC / HLS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use std::net::SocketAddr;

#[path = "tests/test_pinger.rs"]
#[cfg(test)]
mod test_pinger;

// Always compiled regardless of features — the `Pinger` trait, the
// timed helper, the URI parser, and shared utilities. No protocol
// implementations live in here.
mod pinger;
pub mod uri;
mod util;

pub use crate::pinger::{timed, Pinger};

// TLS layer + rustls re-exports. Compiled whenever any protocol that
// needs TLS is enabled (http / ws / mqtt / hls).
#[cfg(feature = "_tls")]
mod tls;
#[cfg(feature = "_tls")]
pub use crate::tls::default_client_config;
#[cfg(feature = "_tls")]
pub use rustls::ClientConfig;

// Per-protocol modules + re-exports — each gated behind its own
// feature flag.
#[cfg(any(feature = "tcp", feature = "udp"))]
mod level4;
#[cfg(feature = "tcp")]
pub use crate::level4::TcpPinger;
#[cfg(feature = "udp")]
pub use crate::level4::UdpPinger;

#[cfg(feature = "dns")]
mod dns;
#[cfg(feature = "dns")]
pub use crate::dns::{DnsPinger, RecordType};

#[cfg(feature = "http")]
mod http;
#[cfg(feature = "http")]
pub use crate::http::{HttpMethod, HttpPinger};

#[cfg(feature = "ws")]
mod websocket;
#[cfg(feature = "ws")]
pub use crate::websocket::WebSocketPinger;

#[cfg(feature = "mqtt")]
mod mqtt;
#[cfg(feature = "mqtt")]
pub use crate::mqtt::{MqttPinger, MqttVersion};

#[cfg(feature = "grpc")]
mod grpc;
#[cfg(feature = "grpc")]
pub use crate::grpc::{GrpcPinger, GrpcStreamPinger};

#[cfg(feature = "hls")]
mod hls;
#[cfg(feature = "hls")]
pub use crate::hls::HlsPinger;

// `BUF_SIZE` is shared by `level4` (tcp / udp) and `http`.
// `HTTP_UNCONNECT_STATUS_CODE` is http-only.
#[cfg(any(feature = "tcp", feature = "udp", feature = "http"))]
pub(crate) const BUF_SIZE: usize = 0xFF;
#[cfg(feature = "http")]
pub(crate) const HTTP_UNCONNECT_STATUS_CODE: &[&str] = &["404", "501"];

/// Resolve `url`'s host:port to a list of socket addresses for display.
/// Falls back to scheme default ports (http → 80, https → 443) when
/// the URL has no explicit port. Returns an empty Vec if DNS lookup
/// fails or the host is empty — callers (the CLI in particular)
/// treat the result as informational and let the actual pinger
/// surface the real error.
pub async fn resolve(url: &str) -> Vec<SocketAddr> {
    let uri = uri::get_uri(url);
    if uri.domain.is_empty() {
        return Vec::new();
    }
    let port = if uri.port > 0 {
        uri.port as u16
    } else {
        match uri.scheme.to_ascii_lowercase().as_str() {
            "https" | "wss" => 443,
            _ => 80,
        }
    };
    match tokio::net::lookup_host(format!("{}:{}", uri.domain, port)).await {
        Ok(iter) => iter.collect(),
        Err(_) => Vec::new(),
    }
}