NeoAnd/mcp-server-wazuh
1
0
Fork 0
mirror of https://github.com/gbrigandi/mcp-server-wazuh.git synced 2025-07-13 15:14:48 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Upgraded to latest wazuh client that fixes fetching critical vulnerabilities

Browse Source
This commit is contained in:
Gianluca Brigandi 2025-06-24 12:03:58 -07:00
parent 943fabddd3
commit 77914b5097
2 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@ -9,7 +9,7 @@ repository = "https://github.com/gbrigandi/mcp-server-wazuh"
readme = "README.md" readme = "README.md"
[dependencies] [dependencies]
wazuh-client = "0.1.3" wazuh-client = "0.1.4"
rmcp = { version = "0.1.5", features = ["server", "transport-io"] } rmcp = { version = "0.1.5", features = ["server", "transport-io"] }
tokio = { version = "1", features = ["full"] } tokio = { version = "1", features = ["full"] }
reqwest = { version = "0.12", features = ["json", "rustls-tls"], default-features = false } reqwest = { version = "0.12", features = ["json", "rustls-tls"], default-features = false }

17
src/main.rs
View File

@ -83,7 +83,7 @@ use std::env;
use clap::Parser; use clap::Parser;
use dotenv::dotenv; use dotenv::dotenv;
use wazuh_client::{WazuhClientFactory, WazuhIndexerClient, RulesClient, VulnerabilityClient, AgentsClient, LogsClient, ClusterClient, Port as WazuhPort}; use wazuh_client::{WazuhClientFactory, WazuhIndexerClient, RulesClient, VulnerabilityClient, VulnerabilitySeverity, AgentsClient, LogsClient, ClusterClient, Port as WazuhPort};
#[derive(Parser, Debug)] #[derive(Parser, Debug)]
#[command(name = "mcp-server-wazuh")] #[command(name = "mcp-server-wazuh")]
@ -532,7 +532,7 @@ impl WazuhToolsServer {
&agent_id, &agent_id,
Some(1000), // Get more results to filter Some(1000), // Get more results to filter
Some(offset), Some(offset),
params.severity.as_deref(), params.severity.as_deref().and_then(VulnerabilitySeverity::from_str)
).await { ).await {
Ok(all_vulns) => { Ok(all_vulns) => {
let filtered: Vec<_> = all_vulns let filtered: Vec<_> = all_vulns
@ -549,7 +549,7 @@ impl WazuhToolsServer {
&agent_id, &agent_id,
Some(limit), Some(limit),
Some(offset), Some(offset),
params.severity.as_deref(), params.severity.as_deref().and_then(VulnerabilitySeverity::from_str)
).await ).await
}; };
@ -566,12 +566,11 @@ impl WazuhToolsServer {
let mcp_content_items: Vec<Content> = vulnerabilities let mcp_content_items: Vec<Content> = vulnerabilities
.into_iter() .into_iter()
.map(|vuln| { .map(|vuln| {
let severity_indicator = match vuln.severity.to_lowercase().as_str() { let severity_indicator = match vuln.severity {
"critical" => "🔴 CRITICAL", VulnerabilitySeverity::Critical => "🔴 CRITICAL",
"high" => "🟠 HIGH", VulnerabilitySeverity::High => "🟠 HIGH",
"medium" => "🟡 MEDIUM", VulnerabilitySeverity::Medium => "🟡 MEDIUM",
"low" => "🟢 LOW", VulnerabilitySeverity::Low => "🟢 LOW",
_ => &vuln.severity,
}; };
let published_info = if let Some(published) = &vuln.published { let published_info = if let Some(published) = &vuln.published {