chg: [CEDetector] tag domains

pull/607/merge
terrtia 2024-10-16 11:35:16 +02:00
parent bc5a96b3ad
commit 35dd487281
No known key found for this signature in database
GPG Key ID: 1E1B1F50D84613D0
1 changed files with 5 additions and 0 deletions

View File

@ -25,6 +25,7 @@ sys.path.append(os.environ['AIL_BIN'])
##################################
from modules.abstract_module import AbstractModule
from lib.ConfigLoader import ConfigLoader
from lib.objects.Domains import Domain
class CEDetector(AbstractModule):
"""docstring for Onion module."""
@ -90,6 +91,10 @@ class CEDetector(AbstractModule):
print(f'CSAM DETECTED {content}')
# print()
self.add_message_to_queue(message=self.ce_tag, queue='Tags')
# Domains
for dom in self.obj.get_correlation('domain').get('domain', []):
domain = Domain(dom[1:])
self.add_message_to_queue(obj=domain, message=self.ce_tag, queue='Tags')
return to_tag