From 35dd487281d5952107425188d497b6fc866f5ebe Mon Sep 17 00:00:00 2001
From: terrtia <or1994@hotmail.fr>
Date: Wed, 16 Oct 2024 11:35:16 +0200
Subject: [PATCH] chg: [CEDetector] tag domains

---
 bin/modules/CEDetector.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/bin/modules/CEDetector.py b/bin/modules/CEDetector.py
index 37b3d219..21b48ed7 100755
--- a/bin/modules/CEDetector.py
+++ b/bin/modules/CEDetector.py
@@ -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