ipaddress_strategy.py

"""
The strategy which searches for IP addresses.
"""
from core.strategies.wifi.base_wifi_strategy import BaseWiFiStrategy
from core.utils.datatypes import ConnectedDeviceResult, WiFiStrategyResult
from core.utils.logger import get_logger
from core.utils.program_launcher import ArpScanCommands, PingCommands, run_program

# Add logging support.
logger = get_logger(__name__)


class IpAddressStrategy(BaseWiFiStrategy):
    """
    The strategy which searches for IP addresses.
    """

    def __init__(self):
        super().__init__()
        # Check if the program is installed.
        try:
            run_program(PingCommands.GET_VERSION_INFO)
            logger.debug("Program is already installed.")
        except RuntimeError:
            logger.debug("Program is not installed. Installing...")
            run_program(PingCommands.INSTALL_PROGRAM)

    def check_protectors(self) -> WiFiStrategyResult:
        """This method checks if there are any protectors around."""
        for protector in self.protectors:
            # Send a ping to the protector.
            logger.debug("Sending ping to %s", protector.name)
            output = run_program(PingCommands.PING_TO + " " + protector.address)
            # Check if the ping was successful.
            if "Destination Host Unreachable" not in output:
                logger.debug("Protector found: %s", protector.name)
                return WiFiStrategyResult(protector, True)
        logger.debug("No protectors found.")
        return WiFiStrategyResult(None, False)

    # Internal methods
    def _get_all_connected(self) -> list[ConnectedDeviceResult]:
        """This method returns a list of addresses of the clients connected to the network."""
        output = run_program(ArpScanCommands.GET_CONNECTED_IP_ADDRESSES)
        ip_addrs = output.split('\n')
        ip_addrs = ip_addrs[:-1]
        logger.debug("Connected devices: %s", str(ip_addrs))
        return [ConnectedDeviceResult(mac_addr.upper()) for mac_addr in ip_addrs]