evectl 0.1.1

EveCtl with Suricata and EveBox
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153

// SPDX-FileCopyrightText: (C) 2024 Jason Ish <jason@codemonkey.net>
// SPDX-License-Identifier: MIT

use colored::Colorize;

use crate::context::Context;
use crate::prelude::*;
use crate::prompt::Selections;
use crate::term;

#[derive(Clone)]
enum Options {
    Toggle,
    Interface,
    SensorName,
    Bpf,
    Exit,
}

pub(crate) fn menu(context: &mut Context) -> Result<()> {
    let config = &mut context.config;

    loop {
        term::clear();

        let mut selections = crate::prompt::Selections::new();

        if config.suricata.enabled {
            selections.push(Options::Toggle, "Disable Suricata");
        } else {
            selections.push(Options::Toggle, "Enable Suricata");
        }

        if config.suricata.interfaces.is_empty() {
            selections.push(Options::Interface, "Select Interface");
        } else {
            selections.push(
                Options::Interface,
                format!(
                    "Select Interface (current: {})",
                    &config.suricata.interfaces[0]
                ),
            );
        }

        //selections.push(Options::SensorName, "Set Sensor Name");

        selections.push(Options::SensorName, {
            if let Some(sensor_name) = &config.suricata.sensor_name {
                format!("Sensor Name (current: {})", sensor_name)
            } else {
                "Sensor Name (current: none)".to_string()
            }
        });

        let current_bpf = if let Some(bpf) = &config.suricata.bpf {
            format!(" (current: \"{}\")", bpf)
        } else {
            " (current: none)".to_string()
        };
        selections.push(Options::Bpf, format!("BPF filter{}", current_bpf));

        selections.push(Options::Exit, "Return");

        match inquire::Select::new("EveCtl: Configure Suricata", selections.to_vec()).prompt() {
            Ok(selection) => match selection.tag {
                Options::Toggle => {
                    toggle_enabled(config);
                }
                Options::Interface => {
                    let interface = select_interface("Select Interface")?;
                    config.suricata.interfaces = vec![interface.clone()];
                }
                Options::SensorName => {
                    set_sensor_name(config);
                }
                Options::Bpf => {
                    set_bpf_filter(config);
                }
                Options::Exit => break,
            },
            Err(_) => break,
        }
    }

    Ok(())
}

fn set_sensor_name(config: &mut Config) {
    let current = config.suricata.sensor_name.clone();
    if let Ok(sensor_name) = inquire::Text::new("Enter Sensor Name:").prompt() {
        if sensor_name.trim().is_empty() {
            if current.is_none() {
                return;
            }
            if inquire::Confirm::new("Clear Sensor Name?")
                .with_default(true)
                .prompt()
                .unwrap_or(false)
            {
                config.suricata.sensor_name = None;
            }
        } else {
            config.suricata.sensor_name = Some(sensor_name);
        }
    }
}

fn toggle_enabled(config: &mut Config) {
    config.suricata.enabled = !config.suricata.enabled;
    if config.suricata.enabled && config.suricata.interfaces.is_empty() {
        if let Ok(interface) = select_interface("Select Interface") {
            config.suricata.interfaces = vec![interface];
        }
    }
}

pub(crate) fn select_interface(prompt: &str) -> Result<String> {
    let interfaces = evectl::system::get_interfaces().unwrap();

    let mut selections = Selections::with_index();
    for interface in &interfaces {
        let address = interface
            .addr4
            .first()
            .map(|s| format!("-- {}", s.green().italic()))
            .unwrap_or("".to_string());
        selections.push(&interface.name, format!("{} {}", interface.name, address));
    }

    let iface = inquire::Select::new(prompt, selections.to_vec()).prompt()?;
    Ok(iface.tag.to_string())
}

fn set_bpf_filter(config: &mut Config) {
    let current = config.suricata.bpf.clone();
    if let Ok(filter) = inquire::Text::new("Enter BPF filter:").prompt() {
        if filter.is_empty() {
            if current.is_none() {
                return;
            }
            if inquire::Confirm::new("Clear BPF filter?")
                .with_default(true)
                .prompt()
                .unwrap_or(false)
            {
                config.suricata.bpf = None;
            }
        } else {
            config.suricata.bpf = Some(filter);
        }
    }
}