Stopping agents ungracefully feature

src/wazuh_qa_framework/global_variables/daemons.py

@@ -18,7 +18,7 @@
 WAZUH_DB_DAEMON = 'wazuh-db'
 AGENT_DAEMON = 'wazuh-agentd'
 
-WAZUH_ANGENT_WINDOWS_SERVICE_NAME = 'WazuhSvc'
+WAZUH_AGENT_WINDOWS_SERVICE_NAME = 'WazuhSvc'
 
 
 WAZUH_AGENT_DAEMONS = [LOGCOLLECTOR_DAEMON, SYSCHECK_DAEMON, EXEC_DAEMON, MODULES_DAEMON, AGENT_DAEMON]

src/wazuh_qa_framework/system/host_manager.py

@@ -632,6 +632,22 @@ def install_package(self, host, package_name, become=None, ignore_errors=False):
 
  return result
 
+ def stop_ungracefully_process(self, host, process):
+ """Stop ungracefully a process.
+
+ Args:
+ host (str): Hostname.
+ process (str): Name of the process to be stopped.
+ """
+ try:
+ if self.is_windows(host):
+ self.run_command(host, f'taskkill /F /IM {process}')
+ else:
+ self.run_command(host, f'pkill -9 {process}')
+
+ except Exception as e:
+ raise Exception(f"An error occurred while stopping the process. {e}")
+
  def uninstall_package(self, host, package_name, become=None, ignore_errors=False):
  """Uninstall a package on a host.
 

src/wazuh_qa_framework/system/wazuh_handler.py

@@ -4,10 +4,11 @@
 
 import os
 import re
+from functools import partial
 from multiprocessing.pool import ThreadPool
 
 from wazuh_qa_framework.generic_modules.logging.base_logger import BaseLogger
-from wazuh_qa_framework.global_variables.daemons import WAZUH_ANGENT_WINDOWS_SERVICE_NAME
+from wazuh_qa_framework.global_variables.daemons import WAZUH_AGENT_WINDOWS_SERVICE_NAME
 from wazuh_qa_framework.system.host_manager import HostManager
 
 
@@ -518,7 +519,7 @@ def restart_agent(self, host):
  host (str): Hostname
  """
  self.logger.debug(f'Restarting agent {host}')
- service_name = WAZUH_ANGENT_WINDOWS_SERVICE_NAME if self.is_windows(host) else 'wazuh-agent'
+ service_name = WAZUH_AGENT_WINDOWS_SERVICE_NAME if self.is_windows(host) else 'wazuh-agent'
  if self.is_agent(host):
  self.control_service(host, service_name, 'restarted')
  self.logger.debug(f'Agent {host} restarted successfully')
@@ -557,7 +558,7 @@ def restart_managers(self, manager_list, parallel=True):
  """Restart managers
 
  Args:
- manager_list (list): Managers list
+ manager_list (list): Managers list.
  parallel (bool, optional): Parallel execution. Defaults to True.
  """
  self.logger.info(f'Restarting managers: {manager_list}')
@@ -568,34 +569,43 @@ def restart_managers(self, manager_list, parallel=True):
  self.restart_manager(manager)
  self.logger.info(f'Managers restarted successfully: {manager_list}')
 
- def stop_agent(self, host):
- """Stop agent
+ def stop_agent(self, host, gracefully=True):
+ """Stop agent.
 
  Args:
- host (str): Hostname
+ host (str): Hostname.
+ gracefully (bool): Stop gracefully. Defaults to True.
  """
- self.logger.debug(f'Stopping agent {host}')
- service_name = WAZUH_ANGENT_WINDOWS_SERVICE_NAME if is_windows(host) else 'wazuh-agent'
+ type = 'gracefully' if gracefully else 'ungracefully'
+ self.logger.debug(f'Stopping {type} {host}')
+ service_name = WAZUH_AGENT_WINDOWS_SERVICE_NAME if self.is_windows(host) else 'wazuh-agent'
  if self.is_agent(host):
- self.control_service(host, service_name, 'stopped')
- self.logger.debug(f'Agent {host} stopped successfully')
+ if gracefully:
+ self.control_service(host, service_name, 'stopped')
+
+ if not gracefully:
+ process = 'wazuh-agent.exe' if self.is_windows(host) else 'wazuh-agent'
+ self.stop_ungracefully_process(host, process)
+
+ self.logger.info(f'Agent {host} stopped {type}')
+
  else:
- raise ValueError(f'Host {host} is not an agent')
+ raise ValueError(f'Host {host} is not an Agent')
 
- def stop_agents(self, agent_list=None, parallel=True):
- """Stop agents
+ def stop_agents(self, agent_list=None, parallel=True, gracefully=True):
+ """Stop agents.
 
  Args:
- agent_list(list, optional): Agents list. Defaults to None
+ agent_list (list, optional): Agents list. Defaults to None.
  parallel (bool, optional): Parallel execution. Defaults to True.
+ gracefully (bool): Stop gracefully. Defaults to True.
  """
- self.logger.info(f'Stopping agents: {agent_list}')
  if parallel:
- self.pool.map(self.stop_agent, agent_list)
+ stop_agent_with_type = partial(self.stop_agent, gracefully=gracefully)
+ self.pool.map(stop_agent_with_type, agent_list)
  else:
  for agent in agent_list:
- self.restart_agent(agent)
- self.logger.info(f'Agents stopped successfully: {agent_list}')
+ self.stop_agent(agent, gracefully)
 
  def stop_manager(self, host):
  """Stop manager
@@ -632,7 +642,7 @@ def start_agent(self, host):
  host (str): Hostname
  """
  self.logger.debug(f'Starting agent {host}')
- service_name = WAZUH_ANGENT_WINDOWS_SERVICE_NAME if is_windows(host) else 'wazuh-agent'
+ service_name = WAZUH_AGENT_WINDOWS_SERVICE_NAME if self.is_windows(host) else 'wazuh-agent'
  if self.is_agent(host):
  self.control_service(host, service_name, 'started')
  self.logger.debug(f'Agent {host} started successfully')
@@ -727,11 +737,11 @@ def stop_environment(self, parallel=True):
  self.pool.map(self.stop_agent, agent_list)
  else:
  self.logger.info(message='Stopping environment: Managers')
- for manager in get_managers():
+ for manager in agent_list:
  self.stop_manager(manager)
 
  self.logger.info(message='Stopping environment: Agents')
- for agent in get_agents():
+ for agent in agent_list:
  self.stop_agent(agent)
 
  self.logger.info('Stopping environment')
@@ -754,11 +764,11 @@ def start_environment(self, parallel=True):
  self.pool.map(self.start_agent, agent_list)
  else:
  self.logger.info(message='Starting environment: Managers')
- for manager in get_managers():
+ for manager in manager_list:
  self.start_manager(manager)
 
  self.logger.info(message='Starting environment: Agents')
- for agent in get_agents():
+ for agent in agent_list:
  self.start_agent(agent)
 
  self.logger.info('Environment started successfully')