recon-cli 0.81.2

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
64
65
66
67
68
69
70
71
72
73

//! `redis(url)` / `redis(url, command)` script binding.
//!
//! `redis(url)` sends PING. `redis(url, "GET foo")` or
//! `redis(url, "SET key \"hello world\"")` sends an arbitrary RESP2
//! command — tokens split shell-style (whitespace + `"…"` + `'…'` +
//! `\`-escapes), matching the CLI's `redis:// -d 'CMD args'` behaviour.
//!
//! Returns `#{ host, port, connect_ms, auth_reply, command, reply,
//! command_ms }`. AUTH failure exits 67; connect refused exits 7;
//! timeout exits 28.

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

pub fn register(engine: &mut Engine, defaults: ScriptDefaults) {
    {
        let d = defaults.clone();
        engine.register_fn(
            "redis",
            move |url: &str| -> Result<Map, Box<EvalAltResult>> { do_redis(url, None, d.connect_timeout) },
        );
    }
    {
        let d = defaults.clone();
        engine.register_fn(
            "redis",
            move |url: &str, command: &str| -> Result<Map, Box<EvalAltResult>> {
                let toks = redis_probe::shell_split(command)
                    .ok_or_else(|| err("redis: unbalanced quotes in command"))?;
                if toks.is_empty() {
                    return Err(err("redis: command must not be empty"));
                }
                do_redis(url, Some(toks), d.connect_timeout)
            },
        );
    }
    {
        let d = defaults.clone();
        engine.register_fn(
            "redis",
            move |url: &str, command: &str, opts: Map| -> Result<Map, Box<EvalAltResult>> {
                let toks = redis_probe::shell_split(command)
                    .ok_or_else(|| err("redis: unbalanced quotes in command"))?;
                if toks.is_empty() {
                    return Err(err("redis: command must not be empty"));
                }
                let timeout = opts_get_u64(&opts, "timeout").unwrap_or(d.connect_timeout);
                do_redis(url, Some(toks), timeout)
            },
        );
    }
}

fn do_redis(
    url: &str,
    command_args: Option<Vec<String>>,
    timeout_secs: u64,
) -> Result<Map, Box<EvalAltResult>> {
    let r = redis_probe::probe(url, command_args, 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("connect_ms".into(), r.connect_ms.into());
    if let Some(a) = r.auth_reply {
        m.insert("auth_reply".into(), a.into());
    }
    m.insert("command".into(), r.command_label.into());
    m.insert("reply".into(), r.reply.into());
    m.insert("command_ms".into(), r.command_ms.into());
    Ok(m)
}