BW3-Core/module/descriptor.py

59 lines
2.2 KiB
Python
Raw Normal View History

2019-10-27 20:12:54 +01:00
#!/usr/bin/python
# -*- coding: utf-8 -*-
r"""!
2019-10-27 20:12:54 +01:00
____ ____ ______ __ __ __ _____
/ __ )/ __ \/ ___/ | / /___ _/ /______/ /_ |__ /
/ __ / / / /\__ \| | /| / / __ `/ __/ ___/ __ \ /_ <
/ /_/ / /_/ /___/ /| |/ |/ / /_/ / /_/ /__/ / / / ___/ /
/_____/\____//____/ |__/|__/\__,_/\__/\___/_/ /_/ /____/
German BOS Information Script
by Bastian Schroll
2019-10-27 21:08:23 +01:00
@file: descriptor.py
@date: 27.10.2019
2019-10-27 20:12:54 +01:00
@author: Bastian Schroll
2019-10-27 21:08:23 +01:00
@description: Module to add descriptions to bwPackets
2019-10-27 20:12:54 +01:00
"""
import logging
2019-10-28 21:20:05 +01:00
from module.moduleBase import ModuleBase
2019-10-27 20:12:54 +01:00
# ###################### #
# Custom plugin includes #
# ###################### #
logging.debug("- %s loaded", __name__)
2019-10-28 21:27:15 +01:00
class BoswatchModule(ModuleBase):
r"""!Adds descriptions to bwPackets"""
2019-10-27 20:12:54 +01:00
def __init__(self, config):
r"""!Do not change anything here!"""
2019-10-27 20:12:54 +01:00
super().__init__(__name__, config) # you can access the config class on 'self.config'
def onLoad(self):
r"""!Called by import of the plugin"""
2019-10-27 21:08:23 +01:00
for descriptor in self.config:
2019-10-27 21:53:00 +01:00
if descriptor.get("wildcard", default=None):
2019-10-27 21:08:23 +01:00
self.registerWildcard(descriptor.get("wildcard"), descriptor.get("descrField"))
2019-10-27 20:12:54 +01:00
def doWork(self, bwPacket):
r"""!start an run of the module.
2019-10-27 20:12:54 +01:00
@param bwPacket: A BOSWatch packet instance"""
2019-10-27 21:08:23 +01:00
for descriptor in self.config:
2020-04-18 14:14:56 +02:00
if not bwPacket.get(descriptor.get("scanField")):
break # scanField is not available in this packet
2020-04-14 22:34:17 +02:00
bwPacket.set(descriptor.get("descrField"), bwPacket.get(descriptor.get("scanField")))
2019-10-27 21:08:23 +01:00
for description in descriptor.get("descriptions"):
2019-10-27 21:58:06 +01:00
if str(description.get("for")) == bwPacket.get(descriptor.get("scanField")):
logging.debug("Description '%s' added in packet field '%s'",
description.get("add"), descriptor.get("descrField"))
2019-10-27 21:08:23 +01:00
bwPacket.set(descriptor.get("descrField"), description.get("add"))
break # this descriptor has found a description - run next descriptor
2019-10-27 20:12:54 +01:00
return bwPacket
def onUnload(self):
r"""!Called by destruction of the plugin"""
2019-10-27 20:12:54 +01:00
pass