fix: [IPAdress module] empty queue if no IP ranges provided

This commit is contained in:
Terrtia 2023-06-19 15:17:41 +02:00
parent b64c70e0cd
commit c02b9f2a81
No known key found for this signature in database
GPG key ID: 1E1B1F50D84613D0

View file

@ -43,7 +43,8 @@ class IPAddress(AbstractModule):
networks = config_loader.get_config_str("IP", "networks") networks = config_loader.get_config_str("IP", "networks")
if not networks: if not networks:
print('No IP ranges provided') print('No IP ranges provided')
sys.exit(0) # sys.exit(0)
else:
try: try:
for network in networks.split(","): for network in networks.split(","):
self.ip_networks.add(IPv4Network(network)) self.ip_networks.add(IPv4Network(network))
@ -62,6 +63,9 @@ class IPAddress(AbstractModule):
self.logger.info(f"Module {self.module_name} initialized") self.logger.info(f"Module {self.module_name} initialized")
def compute(self, message, r_result=False): def compute(self, message, r_result=False):
if not self.ip_networks:
return None
item = Item(message) item = Item(message)
content = item.get_content() content = item.get_content()