#!/usr/bin/env python3 """ A base plugin class for sensors. Anything installed on the target to collect system information and identify the attack """ import os from plugins.base.plugin_base import BasePlugin class SensorPlugin(BasePlugin): """ A sensor will be running on the target machine and monitor attacks. To remote control those sensors there are sensor plugins. This is the base class for them """ # Boilerplate name = None required_files = [] def __init__(self): super().__init__() # pylint:disable=useless-super-delegation self.debugit = False def set_sysconf(self, config): """ Set system config @param config: A dict with system configuration relevant for all plugins """ super().set_sysconf(config) def prime(self): """ prime sets hard core configs in the target. You can use it to call everything that permanently alters the OS by settings. If your prime function returns True the machine will be rebooted after prime-ing it. This is very likely what you want. Only use prime if install is not sufficient. """ return False def install(self): """ Install the sensor. Executed on the target. Take the sensor from the share and (maybe) copy it to its destination. Do some setup """ raise NotImplementedError def start(self, disown=None): """ Start the sensor. The connection to the client is disowned here. = Sent to background. This keeps the process running. @param disown: Send async into background """ raise NotImplementedError def stop(self): """ Stop the sensor """ raise NotImplementedError def __call_collect__(self, machine_path): """ Generate the data collect command @param machine_path: Machine specific path to collect data into """ path = os.path.join(machine_path, "sensors", self.name) os.makedirs(path) self.collect(path) def collect(self, path): """ Collect data from sensor. Copy it from sensor collection dir on target OS to the share @param path: The path to copy the data into """ raise NotImplementedError