recon-cli 0.80.6

Versatile network reconnaissance CLI: HTTP/TLS/DNS, multi-protocol probes, and a Rhai script engine
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

//! `ws(url)` / `wss(url)` script binding. Opens a WebSocket, sends a
//! Ping with a known nonce, waits for the matching Pong, closes cleanly.
//! Returns `#{ host, port, scheme, connect_ms, handshake_ms,
//! http_status, headers: #{ ... }, pong_nonce_matched, ping_ms }`.

use crate::script::convert::{anyhow_to_rhai, opts_get_u64};
use crate::script::defaults::ScriptDefaults;
use crate::ws_probe;
use rhai::{Engine, EvalAltResult, Map};

pub fn register(engine: &mut Engine, defaults: ScriptDefaults) {
    {
        let d = defaults.clone();
        engine.register_fn("ws", move |url: &str| -> Result<Map, Box<EvalAltResult>> {
            do_ws(url, d.connect_timeout)
        });
    }
    {
        let d = defaults.clone();
        engine.register_fn(
            "ws",
            move |url: &str, opts: Map| -> Result<Map, Box<EvalAltResult>> {
                let timeout = opts_get_u64(&opts, "timeout").unwrap_or(d.connect_timeout);
                do_ws(url, timeout)
            },
        );
    }
    {
        let d = defaults.clone();
        engine.register_fn("wss", move |url: &str| -> Result<Map, Box<EvalAltResult>> {
            do_ws(url, d.connect_timeout)
        });
    }
    {
        let d = defaults.clone();
        engine.register_fn(
            "wss",
            move |url: &str, opts: Map| -> Result<Map, Box<EvalAltResult>> {
                let timeout = opts_get_u64(&opts, "timeout").unwrap_or(d.connect_timeout);
                do_ws(url, timeout)
            },
        );
    }
}

fn do_ws(url: &str, timeout_secs: u64) -> Result<Map, Box<EvalAltResult>> {
    let r = ws_probe::probe(url, timeout_secs).map_err(anyhow_to_rhai)?;
    let mut m = Map::new();
    m.insert("host".into(), r.host.into());
    m.insert("port".into(), (r.port as i64).into());
    m.insert("scheme".into(), r.scheme.to_string().into());
    m.insert("connect_ms".into(), r.connect_ms.into());
    m.insert("handshake_ms".into(), r.handshake_ms.into());
    m.insert("http_status".into(), (r.http_status as i64).into());
    let mut headers_map = Map::new();
    for h in &r.headers {
        headers_map.insert(h.name.as_str().into(), h.value.clone().into());
    }
    m.insert("headers".into(), headers_map.into());
    m.insert("pong_nonce_matched".into(), r.pong_nonce_matched.into());
    m.insert("ping_ms".into(), r.ping_round_trip_ms.into());
    Ok(m)
}