NeoAnd/mcp-server-wazuh
1
0
Fork 0
mirror of https://github.com/gbrigandi/mcp-server-wazuh.git synced 2025-12-21 12:52:17 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: Refactor tools and upgrade wazuh-client

Browse Source 
This commit introduces a major refactoring of the tool implementation by splitting the tools into separate modules based on their domain (agents, alerts, rules, stats, vulnerabilities). This improves modularity and
maintainability.

Key changes:
- Upgraded wazuh-client to version 0.1.7 to leverage the new builder pattern for client instantiation.
- Refactored the main WazuhToolsServer to delegate tool calls to the new domain-specific tool modules.
- Created a tools module with submodules for each domain, each containing the relevant tool implementations and parameter structs.
- Updated the default limit for most tools from 100 to 300, while the vulnerability summary limit is set to 10,000 to ensure comprehensive scans.
- Removed a problematic manual test from the test script that was causing it to hang.
This commit is contained in:
Gianluca Brigandi
2025-07-10 14:56:37 -07:00
parent 4493362932
commit 8a6bd9e3c6
9 changed files with 61 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/tools/agents.rs
View File

@@ -15,7 +15,7 @@ use wazuh_client::{AgentsClient, Port as WazuhPort, VulnerabilityClient};
#[derive(Debug, serde::Deserialize, schemars::JsonSchema)]
pub struct GetAgentsParams {
#[schemars(description = "Maximum number of agents to retrieve (default: 100)")]
#[schemars(description = "Maximum number of agents to retrieve (default: 300)")]
pub limit: Option<u32>,
#[schemars(
description = "Agent status filter (active, disconnected, pending, never_connected)"
@@ -39,7 +39,7 @@ pub struct GetAgentProcessesParams {
description = "Agent ID to get processes for (required, e.g., \"0\", \"1\", \"001\")"
)]
pub agent_id: String,
#[schemars(description = "Maximum number of processes to retrieve (default: 100)")]
#[schemars(description = "Maximum number of processes to retrieve (default: 300)")]
pub limit: Option<u32>,
#[schemars(description = "Search string to filter processes by name or command (optional)")]
pub search: Option<String>,
@@ -51,7 +51,7 @@ pub struct GetAgentPortsParams {
description = "Agent ID to get network ports for (required, e.g., \"001\", \"002\", \"003\")"
)]
pub agent_id: String,
#[schemars(description = "Maximum number of ports to retrieve (default: 100)")]
#[schemars(description = "Maximum number of ports to retrieve (default: 300)")]
pub limit: Option<u32>,
#[schemars(description = "Protocol to filter by (e.g., \"tcp\", \"udp\")")]
pub protocol: String,
@@ -80,7 +80,7 @@ impl AgentTools {
&self,
params: GetAgentsParams,
) -> Result<CallToolResult, McpError> {
let limit = params.limit.unwrap_or(100);
let limit = params.limit.unwrap_or(300);
tracing::info!(
limit = %limit,
@@ -277,7 +277,7 @@ impl AgentTools {
return Self::error_result(err_msg);
}
};
let limit = params.limit.unwrap_or(100);
let limit = params.limit.unwrap_or(300);
let offset = 0;
tracing::info!(
@@ -401,7 +401,7 @@ impl AgentTools {
return Self::error_result(err_msg);
}
};
let limit = params.limit.unwrap_or(100);
let limit = params.limit.unwrap_or(300);
let offset = 0; // Default offset
tracing::info!(