mirror/rustnet
1
0
Fork 0
mirror of https://github.com/domcyrus/rustnet.git synced 2026-01-05 05:19:59 -06:00
Code Issues Packages Projects Releases Wiki Activity

adding dpi inormation to the TUI

Browse Source
This commit is contained in:
Marco Cadetg
2025-06-30 14:38:14 +02:00
parent 2ebabdd438
commit 44c2608ec4
2 changed files with 100 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/network/types.rs
View File

@@ -20,6 +20,37 @@ impl std::fmt::Display for Protocol {
}
}
impl std::fmt::Display for ApplicationProtocol {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
ApplicationProtocol::Http(info) => {
if let Some(host) = &info.host {
write!(f, "HTTP ({})", host)
} else {
write!(f, "HTTP")
}
}
ApplicationProtocol::Https(info) => {
if let Some(sni) = &info.sni {
write!(f, "HTTPS ({})", sni)
} else {
write!(f, "HTTPS")
}
}
ApplicationProtocol::Dns(info) => {
if let Some(query) = &info.query_name {
write!(f, "DNS ({})", query)
} else {
write!(f, "DNS")
}
}
ApplicationProtocol::Ssh => write!(f, "SSH"),
ApplicationProtocol::Quic => write!(f, "QUIC"),
ApplicationProtocol::Unknown => write!(f, "-"),
}
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum TcpState {
Listen,

72
src/ui.rs
View File

@@ -147,11 +147,12 @@ fn draw_connections_list(
) {
let widths = [
Constraint::Length(6), // Protocol
Constraint::Length(28), // Local Address
Constraint::Length(38), // Remote Address
Constraint::Length(24), // Local Address (reduced)
Constraint::Length(32), // Remote Address (reduced)
Constraint::Length(12), // State
Constraint::Length(10), // Service
Constraint::Length(22), // Bandwidth
Constraint::Length(16), // DPI/Application
Constraint::Length(18), // Bandwidth (reduced)
Constraint::Min(10), // Process
];
@@ -161,6 +162,7 @@ fn draw_connections_list(
"Remote Address",
"State",
"Service",
"Application",
"Down / Up",
"Process",
]
@@ -191,6 +193,12 @@ fn draw_connections_list(
let service_display = conn.service_name.clone().unwrap_or_else(|| "-".to_string());
// DPI/Application protocol display
let dpi_display = match &conn.dpi_info {
Some(dpi) => dpi.application.to_string(),
None => "-".to_string(),
};
let incoming_rate = format_rate(conn.current_incoming_rate_bps);
let outgoing_rate = format_rate(conn.current_outgoing_rate_bps);
let bandwidth_display = format!("{} / {}", incoming_rate, outgoing_rate);
@@ -201,6 +209,7 @@ fn draw_connections_list(
Cell::from(conn.remote_addr.to_string()),
Cell::from(conn.state()),
Cell::from(service_display),
Cell::from(dpi_display),
Cell::from(bandwidth_display),
Cell::from(process_display),
];
@@ -386,6 +395,63 @@ fn draw_connection_details(
Span::raw(conn.service_name.clone().unwrap_or_else(|| "-".to_string())),
]));
// Add DPI information
match &conn.dpi_info {
Some(dpi) => {
details_text.push(Line::from(vec![
Span::styled("Application: ", Style::default().fg(Color::Yellow)),
Span::raw(dpi.application.to_string()),
]));
// Add protocol-specific details
match &dpi.application {
crate::network::types::ApplicationProtocol::Http(info) => {
if let Some(method) = &info.method {
details_text.push(Line::from(vec![
Span::styled(" HTTP Method: ", Style::default().fg(Color::Cyan)),
Span::raw(method.clone()),
]));
}
if let Some(path) = &info.path {
details_text.push(Line::from(vec![
Span::styled(" HTTP Path: ", Style::default().fg(Color::Cyan)),
Span::raw(path.clone()),
]));
}
if let Some(status) = info.status_code {
details_text.push(Line::from(vec![
Span::styled(" HTTP Status: ", Style::default().fg(Color::Cyan)),
Span::raw(status.to_string()),
]));
}
}
crate::network::types::ApplicationProtocol::Https(info) => {
if let Some(version) = &info.version {
details_text.push(Line::from(vec![
Span::styled(" TLS Version: ", Style::default().fg(Color::Cyan)),
Span::raw(format!("{:?}", version)),
]));
}
}
crate::network::types::ApplicationProtocol::Dns(info) => {
if let Some(query_type) = &info.query_type {
details_text.push(Line::from(vec![
Span::styled(" DNS Type: ", Style::default().fg(Color::Cyan)),
Span::raw(format!("{:?}", query_type)),
]));
}
}
_ => {}
}
}
None => {
details_text.push(Line::from(vec![
Span::styled("Application: ", Style::default().fg(Color::Yellow)),
Span::raw("-".to_string()),
]));
}
}
let details = Paragraph::new(details_text)
.block(
Block::default()