def startIPS():
receiverThread = Receiver('receiver')
collectorThread = Collector('collector')
preprocessorThread = Preprocessor('preprocessor')
receiverThread.start()
collectorThread.start()
preprocessorThread.start()
spectrumLoader = SpectrumLoader()
spectrumLoader.loadSpectrum()
locatingThread = Locator('locating')
locatingThread.start()
class Engine:
def __init__(self, config):
self.ap_list = config["ap_properties"]
self.poll_interval = config["monitoring_properties"]["poll_interval"]
self._thread_pool = Pool(len(config["ap_properties"]))
self._locator_queue = Queue()
self._mode = config["method"]
self._locator = Locator(self._locator_queue, config, self._mode)
def _fetch_probsup_dump(self, ap):
client = SSHClient()
client.set_missing_host_key_policy(AutoAddPolicy())
try:
client.connect(hostname=ap["ip_address"], port=ap["ssh_port"],
username=ap["login"], password=ap["password"], timeout=3)
except (NoValidConnectionsError, timeout):
return ""
connect = client.invoke_shell()
connect.send("wl -i %s probsup_dump\n" % ap["monitoring_interface"])
while not connect.recv_ready():
sleep(0.1)
probsup_dump = connect.recv(50000).decode()
connect.close()
client.close()
return probsup_dump
def shutdown(self):
self._thread_pool.close()
self._thread_pool.join()
self._locator.terminate()
self._locator.join()
def run(self):
self._locator.start()
while True:
probsup_dumps = self._thread_pool.map(self._fetch_probsup_dump, self.ap_list)
self._locator_queue.put(probsup_dumps)
sleep(self.poll_interval)